theon-1/PolyCubeCounting
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fixed import and infinite growth

Browse Source
This commit is contained in:
=
2024-02-06 17:02:16 +01:00
parent d0b161bfaa
commit 644ba22d1b
2 changed files with 19 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
python/polycube.py
View File

@@ -1,4 +1,4 @@
from python.tuple_tools import reorient_tuple, generate_neighbors
from tuple_tools import reorient_tuple, generate_neighbors
from itertools import combinations, chain
class PolyCube:
@@ -22,11 +22,12 @@ class PolyCube:
self.orientations = orientations
def generate_children(self, max_length:int):
max_growth = max_length - len(self.cubes)
growth_candidates = set([])
for last_addition in self.last_additions:
growth_candidates = growth_candidates.union(generate_neighbors(*last_addition))
growth_candidates = growth_candidates.difference(self.cubes)
realizable_growth = chain(combinations(growth_candidates, i) for i in range(1, max_length+1))
realizable_growth = chain(combinations(growth_candidates, i) for i in range(1, max_growth+1))
for addition in realizable_growth:
yield PolyCube(self.cubes.union(addition), addition)