def initialize_problem(self, row_domains, column_domains, constraints):
rows = len(row_domains)
columns = len(column_domains)
self.guigrid = GUIGrid(self.width, self.height*2//3, rows, columns, self.screen)
rowVIs = []
for i in range(rows):
v = VI((True,i), row_domains[i])
rowVIs.append(v)
colVIs = []
for i in range(columns):
v = VI((False,i), column_domains[i])
v.neighbors = rowVIs
colVIs.append(v)
for vi in rowVIs:
vi.neighbors = colVIs
self.astarGAC = NonogramSolver(rowVIs + colVIs, row_domains + column_domains, constraints)
self.currentState = self.astarGAC.search()
self.screen.fill(self.WHITE)
self.guigrid.reset()
if self.currentState:
for var in self.currentState.viList:
var.drawColorsToGUI(self.guigrid)
self.guigrid.draw()
pygame.display.flip()
class Window:
def update_cell(self, row, column, state):
self.grid.update_cell(row, column, state)
def __init__(self, width=500, height=750):
pygame.init()
self.WHITE = (255,255,255)
self.width = width
self.height = height
self.screen = pygame.display.set_mode((self.width, self.height))
self.prevState = None
def initialize_problem(self, row_domains, column_domains, constraints):
rows = len(row_domains)
columns = len(column_domains)
self.guigrid = GUIGrid(self.width, self.height*2//3, rows, columns, self.screen)
rowVIs = []
for i in range(rows):
v = VI((True,i), row_domains[i])
rowVIs.append(v)
colVIs = []
for i in range(columns):
v = VI((False,i), column_domains[i])
v.neighbors = rowVIs
colVIs.append(v)
for vi in rowVIs:
vi.neighbors = colVIs
self.astarGAC = NonogramSolver(rowVIs + colVIs, row_domains + column_domains, constraints)
self.currentState = self.astarGAC.search()
self.screen.fill(self.WHITE)
self.guigrid.reset()
if self.currentState:
for var in self.currentState.viList:
var.drawColorsToGUI(self.guigrid)
self.guigrid.draw()
pygame.display.flip()
def loop(self):
clock = pygame.time.Clock()
results = [None, None, None]
active_result_index = 0
active_result = results[active_result_index]
while True:
pygame.event.pump()
time.sleep(0.3)
if not self.currentState.isSolution():
self.prevState = self.currentState
self.currentState = self.astarGAC.iterateSearch()
if self.currentState is not None:
self.screen.fill(self.WHITE)
self.guigrid.reset()
if self.currentState:
for var in self.currentState.viList:
var.drawColorsToGUI(self.guigrid)
self.guigrid.draw()
for event in pygame.event.get():
if event.type == pygame.QUIT:
sys.exit()
pygame.display.flip()
clock.tick(60)
