add lattice as a scene property and make structure automatically assign it

Author: orionarcher

crystal_toolkit/core/scene.py

@@ -45,6 +45,7 @@ class Scene:
     contents: list = field(default_factory=list)
     origin: List[float] = field(default=(0, 0, 0))
     visible: bool = True
+    lattice: Optional[List[List[float]]] = None
     _meta: Dict = None
 
     def __add__(self, other):
@@ -61,6 +62,7 @@ def __add__(self, other):
             contents=self.contents + other.contents,
             origin=self.origin,
             visible=self.visible,
+            lattice=self.lattice,
             _meta={self.name: self._meta, other.name: other._meta},
         )
 
@@ -91,6 +93,7 @@ def to_json(self):
             name=self.name,
             contents=self.merge_primitives(self.contents),
             origin=self.origin,
+            lattice=self.lattice,
         )
 
         def remove_defaults(scene_dict):

crystal_toolkit/renderables/structure.py

@@ -69,7 +69,7 @@ def get_structure_scene(
     Args:
         self:  Structure object
         origin: fractional coordinate of the origin
-        legend: Legend for the sites
+        legend: Legend for the sites\
         draw_image_atoms: If true draw image atoms that are just outside the
         periodic boundary
 
@@ -95,16 +95,18 @@ def get_structure_scene(
                 site.lattice,
                 properties=site.properties,
             )
-
         site_scene = site.get_scene(legend=legend,)
         for scene in site_scene.contents:
             primitives[scene.name] += scene.contents
 
     primitives["unit_cell"].append(self.lattice.get_scene())
 
+    lattice_vectors = [list(array) for array in self.lattice.matrix]
+
     return Scene(
         name="Structure",
         origin=origin,
+        lattice=lattice_vectors,
         contents=[
             Scene(name=k, contents=v, origin=origin) for k, v in primitives.items()
         ],