partage/lcm-crossword-generator
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

The generate_grid function now returns a much nicer dictionary.

Browse source
This commit is contained in:
Gonçalo S. Martins 2016-05-18 00:15:45 +01:00
parent 84b38a5ff0
commit 40d4875bf3
1 changed files with 8 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
crossword_generator.py
View file

@ -133,10 +133,10 @@ def generate_grid(words, dim):
certain completion level.
Return:
This function returns a list, in which list[0] is the grid, and list[1] is
the list of included words. The grid is a simple list of lists, where
zeroes represent the slots that were not filled in, with the remaining
slots containing a single letter each.
This function returns a dictionary, in which ["grid"] is the grid, and
"words" is the list of included words. The grid is a simple list of lists,
where zeroes represent the slots that were not filled in, with the
remaining slots containing a single letter each.
Assumptions:
Each possibility is a dictionary of the kind:
@ -191,7 +191,7 @@ def generate_grid(words, dim):
#print("Occupancy: {}.".format(occupancy))
# ... and return the grid
return [grid, added_words]
return {"grid": grid, "words": added_words}
def write_grid(grid, screen=False, out_file="table.tex"):
""" This function receives the generated grid and writes it to the file (or
@ -278,9 +278,9 @@ if __name__ == "__main__":
# Show grid
print("Final grid:")
write_grid(grid[0], True)
write_grid(grid["grid"], True)
print("Words:")
pprint.pprint(grid[1])
pprint.pprint(grid["words"])
# Print to file and compile
write_grid(grid[0])
write_grid(grid["words"])