From 8e802b4a6d2d5af14810a2b6603fd53b2697c039 Mon Sep 17 00:00:00 2001 From: Harishankar 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 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