flightdata 0.3.2__tar.gz → 0.3.3__tar.gz

{flightdata-0.3.2 → flightdata-0.3.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: flightdata
-Version: 0.3.2
+Version: 0.3.3
 Summary: Python tools for handling flight data
 Author-email: Thomas David <thomasdavid0@gmail.com>
 License-File: LICENSE

{flightdata-0.3.2 → flightdata-0.3.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "flightdata"
-version = "0.3.2"
+version = "0.3.3"
 description = "Python tools for handling flight data"
 readme = "README.md"
 authors = [{ name = "Thomas David", email = "thomasdavid0@gmail.com" }]

{flightdata-0.3.2 → flightdata-0.3.3}/src/flightdata/base/table/labelgroup.py

@@ -324,24 +324,24 @@ class LabelGroup:
         elif index == len(self) - 1:
             side = "start"
         else:
-            next_space_fwd = np.argwhere(np.array(widths[index+1:]) > min_len)
-            next_space_fwd = next_space_fwd[0][0] if len(next_space_fwd) and len(next_space_fwd[0]) else None
-            next_space_bck = np.argwhere(np.array(widths[:index][::-1]) > min_len)
-            next_space_bck = next_space_bck[0][0] if len(next_space_bck) and len(next_space_bck[0]) else None
+            next_space_stop = np.argwhere(np.array(widths[index+1:]) > min_len)
+            next_space_stop = next_space_stop[0][0] if len(next_space_stop) and len(next_space_stop[0]) else None
+            next_space_start = np.argwhere(np.array(widths[:index][::-1]) > min_len)
+            next_space_start = next_space_start[0][0] if len(next_space_start) and len(next_space_start[0]) else None
             
-            if next_space_fwd is None and next_space_bck is None:
+            if next_space_stop is None and next_space_start is None:
                 raise ValueError(f"Cannot expand label {name}")
-            elif next_space_fwd is None:
+            elif next_space_stop is None:
                 side = "start"
-            elif next_space_bck is None:
+            elif next_space_start is None:
                 side= "stop"
-            elif next_space_fwd > next_space_bck:
-                side = "stop"
-            elif next_space_bck < next_space_fwd:
+            elif next_space_stop > 0 and next_space_start == 0:
                 side = "start"
+            elif next_space_start > 0 and next_space_stop == 0:
+                side = "stop"
             else:
-                #either next_space_fwd == next_space_bck
-                side = "stop" if widths[:index][::-1][next_space_bck] < widths[index+1:][next_space_fwd] else "start"
+                #either next_space_stop == next_space_start
+                side = "stop" if widths[:index][::-1][next_space_start] < widths[index+1:][next_space_stop] else "start"
        
         if side=="start":
             boundaries[index-1] -= 1

{flightdata-0.3.2 → flightdata-0.3.3}/src/flightdata/state/state.py

@@ -167,10 +167,10 @@ class State(Table):
 
         return plotsec(self, **(dict(nmodels=10, ribb=True) | kwargs))
 
-    def plotlabels(self, label: str):
+    def plotlabels(self, label: str, **kwargs):
         from plotting import plot_regions
 
-        return plot_regions(self, label)
+        return plot_regions(self, label, **kwargs)
 
     def extrapolate(self, duration: float, min_len=3) -> State:
         """Extrapolate the input state assuming uniform circular motion and small angles"""