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

now generate childern up to max size

Browse Source
This commit is contained in:
=
2024-02-06 16:50:04 +01:00
parent 2a87f9c46a
commit 028df78c3e
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
python/polycube.py
View File

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