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

python now works, jl can leave

Browse Source
This commit is contained in:
=
2024-02-06 18:54:53 +01:00
parent 60bef312b5
commit ca34ab8ba3
9 changed files with 32 additions and 299 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
python/polycube.py
View File

@@ -1,11 +1,11 @@
from tuple_tools import reorient_tuple, generate_neighbors
from tuple_tools import reorient_tuple, generate_neighbors, subtract
from itertools import combinations, chain
class PolyCube:
#cubes set{tuple}
#orientations list{set{tuple{}}}
#last_additions set{tuple}
#cubes set{tuple}
#oriented_offsets list{list{tuple{}}}
#last_additions set{tuple}
def __init__(self, cubes_list, last_additions = []) -> None:
self.cubes = set(cubes_list)
@@ -15,11 +15,21 @@ class PolyCube:
def _compute_reorientations(self):
n_cubes = len(self.cubes)
n_orientations = 24
orientations = [set()]*n_orientations
orientations = [[]]*n_orientations
oriented_offsets = [[]]*n_orientations
for i in range(n_orientations):
for cube in self.cubes:
orientations[i].add(reorient_tuple(cube, i))
self.orientations = orientations
if n_cubes == 1:
oriented_offsets[i] = []
else:
orientations[i] = []
for cube in self.cubes:
orientations[i].append(reorient_tuple(cube, i))
orientations[i].sort()
oriented_offsets[i] = [(0, 0, 0)]*(n_cubes-1)
last_cube = orientations[i][0]
for j in range(n_cubes-1):
oriented_offsets[i][j] = subtract(*last_cube, *orientations[i][j+1])
self.oriented_offsets = oriented_offsets
def generate_children(self, max_length:int):
max_growth = max_length - len(self.cubes)