X-Git-Url: https://harishankar.org/repos/?p=getaclue.git;a=blobdiff_plain;f=crosswordpuzzlecreator.py;h=1e62315ecd42f6fc4fb71ca8962e412a55e39455;hp=4098cd3e4e07f7ff9da1fcd2f85b458b8c377b97;hb=5f68ccac12372c5271c21e9045558d2d03e92a41;hpb=b1ba162f06177e9ccd5456e378069aafe6b2a916

diff --git a/crosswordpuzzlecreator.py b/crosswordpuzzlecreator.py
index 4098cd3..1e62315 100644
--- a/crosswordpuzzlecreator.py
+++ b/crosswordpuzzlecreator.py
@@ -4,6 +4,8 @@
 
 # Cross puzzle creator class
 import sys
+import cPickle
+import readline
 
 import crosswordpuzzle
 
@@ -30,6 +32,20 @@ class CrosswordPuzzleCreator:
 		self.current_file = None
 		self.puzzle = None
 
+	# save the puzzle
+	def save_puzzle (self):
+		if self.current_file:
+			fpuzzle = open (self.current_file, "wb")
+			cPickle.dump (self.puzzle, fpuzzle, cPickle.HIGHEST_PROTOCOL)
+			sys.stdout.write (self.BLUE + "Puzzle saved to: " +
+						self.current_file + "\n" + self.ENDCOL)
+
+	# load the puzzle
+	def load_puzzle (self):
+		if self.current_file:
+			fpuzzle = open (self.current_file, "rb")
+			self.puzzle = cPickle.load (fpuzzle)
+
 	# display the grid with words
 	def print_puzzle (self, no_words=False):
 		# if self.puzzle is not none
@@ -77,8 +93,76 @@ class CrosswordPuzzleCreator:
 				sys.stdout.write (' ' + self.BLUE + "%2d" % row + self.ENDCOL + "\n")
 			raw_input (self.BRICKRED + "Press <return> to continue" + self.ENDCOL)
 
+	# display existing clues
+	def on_display_clues (self):
+		try:
+			aclues = self.puzzle.get_clues_across ()
+			sys.stdout.write (self.BOLD + "\n------------\n")
+			sys.stdout.write ("Across words\n")
+			sys.stdout.write ("------------\n" + self.ENDCOL)
+
+			for word, clue in aclues:
+				sys.stdout.write (self.BOLD + word[0] + ": " + self.ENDCOL)
+				if clue:
+					sys.stdout.write (self.BLUE + clue + "\n" + self.ENDCOL)
+				else:
+					sys.stdout.write (self.BLUE + "(No clue yet)\n" + self.ENDCOL)
+		except crosswordpuzzle.NoWordsException:
+			sys.stderr.write ("No words across\n")
+
+		try:
+			dclues = self.puzzle.get_clues_down ()
+			sys.stdout.write (self.BOLD + "\n----------\n")
+			sys.stdout.write ("Down words\n")
+			sys.stdout.write ("----------\n" + self.ENDCOL)
+
+			for word, clue in dclues:
+				sys.stdout.write (self.BOLD + word[0] + ": " + self.ENDCOL)
+				if clue:
+					sys.stdout.write (self.BLUE + clue + "\n" + self.ENDCOL)
+				else:
+					sys.stdout.write (self.BLUE + "(No clue yet)\n" + self.ENDCOL)
+		except crosswordpuzzle.NoWordsException:
+			sys.stderr.write ("No words down\n")
+
+		raw_input (self.BRICKRED + "Press <return> to continue" + self.ENDCOL)
+
+	# set a clue to a word
+	def on_set_clue (self):
+		self.print_puzzle ()
+		# get the row and column
+		srow = raw_input (self.BRICKRED + "At row: " + self.ENDCOL)
+		scol = raw_input (self.BRICKRED + "At col: " + self.ENDCOL)
+		# try converting it to number
+		try:
+			row = int (srow)
+			col = int (scol)
+		except ValueError:
+			sys.stderr.write ("Invalid row or column\n")
+			return
+
+		try:
+			# across word set the clue if found
+			aword, arow, acol, alen = self.puzzle.get_word_across (row, col)
+			sys.stdout.write (self.BLUE + "Across word at position: " + aword + "\n" + self.ENDCOL)
+			clue = raw_input (self.BRICKRED + "Clue for across word: " + self.ENDCOL)
+			self.puzzle.data[arow][acol].clue_across = clue
+			sys.stdout.write (self.BLUE + "Set the clue: \n" + self.puzzle.data[arow][acol].clue_across)
+		except crosswordpuzzle.NoWordException:
+			sys.stderr.write ("No across word found at that position\n")
+
+		try:
+			# down word set the clue if found
+			dword, drow, dcol, dlen = self.puzzle.get_word_down (row, col)
+			sys.stdout.write (self.BLUE + "Down word at position: " + dword + "\n" + self.ENDCOL)
+			clue = raw_input (self.BRICKRED + "Clue for down word: " + self.ENDCOL)
+			self.puzzle.data[drow][dcol].clue_down = clue
+			sys.stdout.write (self.BLUE + "Set the clue: \n" + self.puzzle.data[drow][dcol].clue_down)
+		except crosswordpuzzle.NoWordException:
+			sys.stderr.write ("No down word found at that position\n")
+
 	# add a word to the puzzle
-	def add_word (self, across=True):
+	def on_add_word (self, across=True):
 		# first display the grid
 		self.print_puzzle ()
 		# get the row and column
@@ -107,11 +191,52 @@ class CrosswordPuzzleCreator:
 		except crosswordpuzzle.FrozenGridException:
 			sys.stderr.write ("Word cannot be added to a frozen puzzle.\n")
 
+	# Export to image/HTML
+	def on_export_image (self, solution=True):
+		try:
+			sys.stdout.write (self.BLUE + "Exporting puzzle to image/HTML\n")
+			pngfile = raw_input (self.BRICKRED + "Filename (PNG): " + self.ENDCOL)
+			if solution is False:
+				htmlfile = raw_input (self.BRICKRED + "Filename (HTML): " +
+							self.ENDCOL)
+				puztitle = raw_input (self.BRICKRED + "Title of puzzle: " +
+							self.ENDCOL)
+				self.puzzle.export_image (pngfile, htmlfile, puztitle, solution)
+			else:
+				self.puzzle.export_image (pngfile)
+
+			sys.stdout.write (self.BLUE + "Successfully exported!")
+		except crosswordpuzzle.FrozenGridException:
+			sys.stderr.write ("Cannot export as grid is not frozen/finalized\n")
+		except crosswordpuzzle.NoWordsException:
+			sys.stderr.write ("No words to export!\n")
+
+	# Export to across lite
+	def on_export_acrosslite (self):
+		try:
+			sys.stdout.write (self.BLUE + "Exporting to AcrossLite(tm) Format\n" +
+						self.ENDCOL)
+			title = raw_input (self.BRICKRED + "Puzzle title: " + self.ENDCOL)
+			name = raw_input (self.BRICKRED + "Author name: " + self.ENDCOL)
+			copyright = raw_input (self.BRICKRED + "Copyright: " + self.ENDCOL)
+			exportfile = raw_input (self.BRICKRED + "Export to file: " + self.ENDCOL)
+
+			acrosslite_str = self.puzzle.export_acrosslite (title, name, copyright)
+			fexport = open (exportfile, "w")
+			fexport.write (acrosslite_str)
+			fexport.close ()
+			sys.stdout.write (self.BLUE + "Exported AcrossLite(tm) File: " +
+				exportfile + "\n" + self.ENDCOL)
+		except crosswordpuzzle.FrozenGridException:
+			sys.stderr.write ("Cannot export as grid is not frozen/finalized\n")
+		except crosswordpuzzle.NoWordsException:
+			sys.stderr.write ("No words to export!\n")
+
 	# Puzzle loop
 	def do_puzzle_loop (self):
 		# there is a current file
-		while True:
-			if self.current_file and self.puzzle:
+		if self.current_file and self.puzzle:
+			while True:
 				sys.stdout.write (self.BOLD + "\n-----------------------------------\n")
 				sys.stdout.write ("Puzzle: " + self.current_file + "\n")
 				sys.stdout.write ("-----------------------------------" + self.ENDCOL + "\n")
@@ -122,21 +247,39 @@ class CrosswordPuzzleCreator:
 				sys.stdout.write ("5. Remove down word\n")
 				sys.stdout.write ("6. Freeze grid\n")
 				sys.stdout.write ("7. Unfreeze grid\n")
-				sys.stdout.write ("8. Set a clue\n")
-				sys.stdout.write ("9. Remove clue\n")
+				sys.stdout.write ("8. Set clue for word\n")
+				sys.stdout.write ("9. Display clues\n")
+				sys.stdout.write ("R. Reset grid\n")
 				sys.stdout.write ("S. Save puzzle\n")
+				sys.stdout.write ("E. Export to AcrossLite(TM) format\n")
+				sys.stdout.write ("H. Export puzzle as image/HTML\n")
+				sys.stdout.write ("I. Export solution as image\n")
 				sys.stdout.write ("X. Exit to main menu\n" + self.ENDCOL)
 				ch = raw_input (self.BRICKRED + "Your choice: " + self.ENDCOL)
 				if ch == "1":
 					self.print_puzzle ()
 				elif ch == "2":
-					self.add_word ()
+					self.on_add_word ()
 				elif ch == "3":
-					self.add_word (False)
+					self.on_add_word (False)
 				elif ch == "6":
 					self.puzzle.freeze_grid ()
 				elif ch == "7":
 					self.puzzle.unfreeze_grid ()
+				elif ch == "8":
+					self.on_set_clue ()
+				elif ch == "9":
+					self.on_display_clues ()
+				elif ch == "R" or ch == "r":
+					self.on_reset_grid ()
+				elif ch == "S" or ch == "s":
+					self.save_puzzle ()
+				elif ch == "E" or ch == "e":
+					self.on_export_acrosslite ()
+				elif ch == "H" or ch == "h":
+					self.on_export_image (False)
+				elif ch == "I" or ch == "i":
+					self.on_export_image ()
 				elif ch == "X" or ch == "x":
 					break
 
@@ -155,6 +298,22 @@ class CrosswordPuzzleCreator:
 		self.puzzle = crosswordpuzzle.CrosswordPuzzle (rows, cols)
 		self.do_puzzle_loop ()
 
+	# when user chooses to load puzzle
+	def on_load_puzzle (self):
+		self.current_file = raw_input (self.BRICKRED + "Puzzle to load: "
+						+ self.ENDCOL)
+		self.load_puzzle ()
+		self.do_puzzle_loop ()
+
+	# when user chooses to reset grid
+	def on_reset_grid (self):
+		ans = raw_input (self.BRICKRED +
+			"This will clear the entire grid! Are you sure (Y/N)? " + self.ENDCOL)
+		if ans == "y" or ans == "Y":
+			self.puzzle.reset_grid ()
+			sys.stdout.write (self.BLUE + "Grid has been cleared of all data!"
+						+ self.ENDCOL + "\n")
+
 	# Main application loop
 	def do_main_loop (self):
 		# display the menu
@@ -168,5 +327,7 @@ class CrosswordPuzzleCreator:
 			ch = raw_input (self.BRICKRED + "Your choice: " + self.ENDCOL)
 			if ch == '1':
 				self.on_new_puzzle ()
+			if ch == '2':
+				self.on_load_puzzle ()
 			if ch == 'x' or ch == 'X':
 				break