X-Git-Url: https://harishankar.org/repos/?p=getaclue.git;a=blobdiff_plain;f=crosswordpuzzle.py;h=7d3b2709933e604a71b04171726a1f359ce3291c;hp=694a7660b19884e377ed5ff0e8435a1ba9c272b0;hb=8bf14bbd585036c0cfda88c2ace825f1aa471730;hpb=f9926c428d81940df4d22a6caeffa9865057b12c
diff --git a/crosswordpuzzle.py b/crosswordpuzzle.py
index 694a766..7d3b270 100644
--- a/crosswordpuzzle.py
+++ b/crosswordpuzzle.py
@@ -4,6 +4,9 @@
# Class for the puzzle data representation
+# for export to PNG image
+import cairo
+
class GridItem:
# initialize the item
def __init__ (self, item_char='.', across_start = False, down_start = False,
@@ -73,9 +76,99 @@ class CrosswordPuzzle:
for j in range (cols):
self.data[i].append (GridItem ())
+ # export to an image
+ def export_image (self, pngfile, htmlfile=None, puztitle="Crossword Puzzle",
+ solution=True):
+ # don't export if grid is not frozen
+ if self.frozen_grid is False:
+ raise FrozenGridException
+
+ # create cairo image surface and context
+ px = 30
+ surf = cairo.ImageSurface (cairo.FORMAT_RGB24, self.cols*px, self.rows*px)
+ ctx = cairo.Context (surf)
+
+ ctx.set_source_rgb (1, 1, 1)
+ ctx.rectangle (0, 0, self.cols*px, self.rows*px)
+ ctx.fill ()
+
+ # get the clues across and down
+ clues_across = self.get_clues_across ()
+ clues_down = self.get_clues_down ()
+
+
+ # traverse through the grid
+ for row in range (self.rows):
+ for col in range (self.cols):
+ # if grid is un-occupied
+ if (self.data[row][col].occupied_across is False and
+ self.data[row][col].occupied_down is False):
+ ctx.set_source_rgb (0, 0, 0)
+ ctx.rectangle (col*px, row*px, px, px)
+ ctx.fill ()
+ # grid is occupied
+ else:
+ ctx.set_source_rgb (1, 1, 1)
+ ctx.rectangle (col*px, row*px, px, px)
+ ctx.fill ()
+ ctx.set_source_rgb (0, 0, 0)
+ ctx.rectangle (col*px, row*px, px, px)
+ ctx.stroke ()
+ # if solution is not to be provided, number the grid
+ if solution is False:
+ if self.data[row][col].numbered <> 0:
+ ctx.select_font_face ("Serif")
+ ctx.set_font_size (10)
+ ctx.move_to (col*px+5, row*px+10)
+ ctx.show_text (str(self.data[row][col].numbered))
+ # display the words
+ else:
+ ctx.select_font_face ("Serif")
+ ctx.set_font_size (16)
+ ctx.move_to (col*px+10, row*px+20)
+ ctx.show_text (self.data[row][col].char)
+
+ surf.write_to_png (open (pngfile, "wb"))
+
+ # if solution is false, publish the clues and the image in a HTML file
+ if htmlfile and solution is False:
+ html_contents = ["", "", ""]
+ html_contents.append (puztitle)
+ html_contents.append ("")
+ html_contents.append ("")
+ html_contents.append ("")
+ html_contents.append ("" + puztitle + "
")
+ html_contents.append ('
')
+
+ html_contents.append ("Across clues
")
+ html_contents.append ("")
+ for word, clue in clues_across:
+ clue_str = str (self.data[word[1]][word[2]].numbered) + " - " \
+ + clue
+ html_contents.append (clue_str)
+ html_contents.append ("
")
+ html_contents.append ("
")
+
+ html_contents.append ("Down clues
")
+ html_contents.append ("")
+ for word, clue in clues_down:
+ clue_str = str (self.data[word[1]][word[2]].numbered) + " - " \
+ + clue
+ html_contents.append (clue_str)
+ html_contents.append ("
")
+ html_contents.append ("
")
+ html_contents.append ("")
+ html_contents.append ("")
+
+ html_str = "\r\n".join (html_contents)
+
+ fhtml = open (htmlfile, "wb")
+ fhtml.write (html_str)
+ fhtml.close ()
+
# get the AcrossLite(TM) data for exporting
def export_acrosslite (self, title, author, copyright):
- # don't export if grid is frozen
+ # don't export if grid is not frozen
if self.frozen_grid is False:
raise FrozenGridException