From 8e802b4a6d2d5af14810a2b6603fd53b2697c039 Mon Sep 17 00:00:00 2001
From: Harishankar <v.harishankar@gmail.com>
Date: Sun, 5 Dec 2010 14:13:39 +0530
Subject: [PATCH] Clues can now be entered for words

Finished implementing clues for words. Also written the interface
for getting an across or down word at a particular row/column.
---
 crosswordpuzzle.py        | 75 +++++++++++++++++++++++++++++++++++++++
 crosswordpuzzlecreator.py | 29 +++++++++++++--
 2 files changed, 102 insertions(+), 2 deletions(-)

diff --git a/crosswordpuzzle.py b/crosswordpuzzle.py
index 56807cd..9617897 100644
--- a/crosswordpuzzle.py
+++ b/crosswordpuzzle.py
@@ -45,6 +45,11 @@ class FrozenGridException (Exception):
 	def __init__ (self):
 		self.msg = "Grid is frozen and cannot be edited"
 
+# exception when no word is found at a position
+class NoWordException (Exception):
+	def __init__ (self, row, col):
+		self.pos = (row, col)
+
 class CrosswordPuzzle:
 	def __init__ (self, rows, cols):
 		# define number of rows and columns
@@ -63,6 +68,76 @@ class CrosswordPuzzle:
 			for j in range (cols):
 				self.data[i].append (GridItem ())
 
+	# getting a down word at a position
+	def get_word_down (self, row, col):
+		# if index is out of bounds
+		if row >= self.rows or col >= self.cols:
+			raise NoWordException (row, col)
+
+		# if there is no occupied down letter at that position
+		if self.data[row][col].occupied_down is False:
+			raise NoWordException (row, col)
+
+		# now traverse the grid to find the beginning of the word
+		i = row
+		while i >= 0:
+			# if it is occupied down and is the beginning of the word
+			if (self.data[i][col].occupied_down is True and
+				self.data[i][col].down_start is True):
+				start_row = i
+				break
+			i -= 1
+
+		i = start_row
+		word_chars = []
+		# now seek the end of the word
+		while i < self.rows:
+			if self.data[i][col].occupied_down is True:
+				word_chars.append (self.data[i][col].char)
+			else:
+				break
+			i += 1
+
+		word = "".join (word_chars)
+
+		# return the word, starting row, column and length as a tuple
+		return (word, start_row, col, len(word))
+
+	# getting an across word at a position
+	def get_word_across (self, row, col):
+		# if index is out of bounds
+		if row >= self.rows or col >= self.cols:
+			raise NoWordException (row, col)
+
+		# if there is no occupied across letter at that position
+		if self.data[row][col].occupied_across is False:
+			raise NoWordException (row, col)
+
+		# now traverse the grid to look for the beginning of the word
+		i = col
+		while i >= 0:
+			# if it is occupied across and is the beginning of the word
+			if (self.data[row][i].occupied_across is True and
+				self.data[row][i].across_start is True):
+				start_col = i
+				break
+			i -= 1
+
+		i = start_col
+		word_chars = []
+		# now seek the end of the word
+		while i < self.cols:
+			if self.data[row][i].occupied_across is True:
+				word_chars.append (self.data[row][i].char)
+			else:
+				break
+			i += 1
+
+		word = "".join (word_chars)
+
+		# return the word, starting column, row and length as a tuple
+		return (word, start_col, row, len(word))
+
 	# setting a down word
 	def set_word_down (self, row, col, word):
 		# if the grid is frozen the abort
diff --git a/crosswordpuzzlecreator.py b/crosswordpuzzlecreator.py
index 4098cd3..ed7e1d8 100644
--- a/crosswordpuzzlecreator.py
+++ b/crosswordpuzzlecreator.py
@@ -77,6 +77,30 @@ class CrosswordPuzzleCreator:
 				sys.stdout.write (' ' + self.BLUE + "%2d" % row + self.ENDCOL + "\n")
 			raw_input (self.BRICKRED + "Press <return> to continue" + self.ENDCOL)
 
+	# set a clue to a word
+	def 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:
+			aword, arow, acol, alen = self.puzzle.get_word_across (row, col)
+			sys.stdout.write (self.BLUE + "Word at position: " + aword + "\n" + self.ENDCOL)
+			clue = raw_input (self.BRICKRED + "Clue for word: " + self.ENDCOL)
+			if clue:
+				self.puzzle.data[arow][acol].clue_across = clue
+				sys.stdout.write (self.BLUE + "Set clue: \n" + self.puzzle.data[arow][acol].clue_across)
+		except crosswordpuzzle.NoWordException:
+			sys.stderr.write ("No across word found at that position")
+
 	# add a word to the puzzle
 	def add_word (self, across=True):
 		# first display the grid
@@ -122,8 +146,7 @@ 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 ("S. Save puzzle\n")
 				sys.stdout.write ("X. Exit to main menu\n" + self.ENDCOL)
 				ch = raw_input (self.BRICKRED + "Your choice: " + self.ENDCOL)
@@ -137,6 +160,8 @@ class CrosswordPuzzleCreator:
 					self.puzzle.freeze_grid ()
 				elif ch == "7":
 					self.puzzle.unfreeze_grid ()
+				elif ch == "8":
+					self.set_clue ()
 				elif ch == "X" or ch == "x":
 					break
 
-- 
2.20.1