core: Added get_length method to labels

aimat-lab · May 28, 2024 · 96bddf9 · 96bddf9
1 parent cad70b3
commit 96bddf9
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 8 deletions.
diff --git a/xrdpattern/core/labels.py b/xrdpattern/core/labels.py
@@ -139,6 +139,11 @@ def primary_wavelength(self) -> float:
     def secondary_wavelength(self) -> float:
         return self.artifacts.secondary_wavelength
 
+    @classmethod
+    def get_length(cls):
+        empty = cls.make_empty()
+        return len(empty.list_repr)
+
 
 @dataclass
 class Artifacts:

diff --git a/xrdpattern/core/structure/base.py b/xrdpattern/core/structure/base.py
@@ -41,14 +41,6 @@ def in_unit_interval(val : float) -> bool:
             in_unit_interval = False
         return in_unit_interval
 
-    @staticmethod
-    def shift_into_unit(x : float):
-        if x >= 0:
-            n = int(x)
-        else:
-            n = int(x-1)
-        return x - n
-
     @classmethod
     def make_void(cls) -> AtomicSite:
         return cls(x=torch.nan, y=torch.nan, z=torch.nan, occupancy=0, species=Void())