flightdata 0.2.10__tar.gz → 0.2.12__tar.gz

{flightdata-0.2.10 → flightdata-0.2.12}/.github/workflows/publish_pypi.yml

@@ -4,7 +4,7 @@ name: Publish to PyPI and TestPyPI
 on:
   push:
     tags:
-      - '**'
+      - 'v[0-9]+.[0-9]+.[0-9]+'
 jobs:
   build:
     name: Build distribution

{flightdata-0.2.10/flightdata.egg-info → flightdata-0.2.12}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: flightdata
-Version: 0.2.10
+Version: 0.2.12
 Summary: Module for handling UAV flight log data
 Author-email: Thomas David <thomasdavid0@gmail.com>
 License: GNU GPL v3
@@ -34,3 +34,16 @@ The State object is a table representing the position and orientation of the air
 
 
 Further documentation will be provided here: https://pfcdocumentation.readthedocs.io/pyflightcoach/flightdata.html
+
+
+### Installation
+
+```bash
+pip install pfc-geometry
+```
+
+### Setup from source
+
+```bash
+pip install .
+```

{flightdata-0.2.10 → flightdata-0.2.12}/README.md

@@ -12,3 +12,16 @@ The State object is a table representing the position and orientation of the air
 
 
 Further documentation will be provided here: https://pfcdocumentation.readthedocs.io/pyflightcoach/flightdata.html
+
+
+### Installation
+
+```bash
+pip install pfc-geometry
+```
+
+### Setup from source
+
+```bash
+pip install .
+```

{flightdata-0.2.10 → flightdata-0.2.12}/flightdata/base/table.py

@@ -21,7 +21,6 @@ class Table:
         self.label_cols = [c for c in data.columns if c not in self.constructs.cols()]
     
         self.data = data
-        #self.data.index = self.data.index - self.data.index[0]
         
         if fill:
             missing = self.constructs.missing(self.data.columns)
@@ -81,40 +80,16 @@ class Table:
         else:
             raise TypeError(f"Expected Number or slice, got {sli.__class__.__name__}")
         
-
     def slice_raw_t(self, sli):
         inds = self.data.reset_index(names="t2").set_index("t").loc[sli].t2.to_numpy()#set_index("t", drop=False).columns
 
         return self.__class__(self.data.loc[inds])
 
-    @classmethod
-    def stack(Cls, sections: list, overlap: int=1) -> Self:
-        """Stack a list of Tables on top of each other. 
-        The overlap is the number of rows to overlap between each section
-        """
-        # first build list of index offsets, to be added to each dataframe
-        if overlap > 0:
-            offsets = np.cumsum([0] + [sec.data.index[-overlap] for sec in sections[:-1]])
-            dfs = [section.data.iloc[:-overlap] for section in sections[:-1]] + [sections[-1].data]
-        elif overlap == 0:
-            offsets = np.cumsum([0] + [sec.duration + sec.dt[-1] for sec in sections[:-1]])
-            dfs = [section.data for section in sections]
-        else:
-            raise AttributeError("Overlap must be >= 0")
-
-        for df, offset in zip(dfs, offsets):
-            df.index = np.array(df.index) - df.index[0] + offset
-        combo = pd.concat(dfs)
-        combo.index.name = "t"
-
-        combo["t"] = combo.index
-
-        return Cls(combo)
-
     def __iter__(self):
         for ind in list(self.data.index):
             yield self[ind]
 
+
     @classmethod
     def from_constructs(cls, *args,**kwargs) -> Self:
         kwargs = dict(
@@ -133,7 +108,7 @@ class Table:
         )
 
         return cls(df)
-
+    
     def __repr__(self):
         return f"{self.__class__.__name__} Table(duration = {self.duration})"
 
@@ -161,6 +136,30 @@ class Table:
             ignore_index=True
         ).set_index("t", drop=False))
       
+    @classmethod
+    def stack(Cls, sections: list, overlap: int=1) -> Self:
+        """Stack a list of Tables on top of each other. 
+        The overlap is the number of rows to overlap between each section
+        """
+        # first build list of index offsets, to be added to each dataframe
+        if overlap > 0:
+            offsets = np.cumsum([0] + [sec.data.index[-overlap] for sec in sections[:-1]])
+            dfs = [section.data.iloc[:-overlap] for section in sections[:-1]] + [sections[-1].data]
+        elif overlap == 0:
+            offsets = np.cumsum([0] + [sec.duration + sec.dt[-1] for sec in sections[:-1]])
+            dfs = [section.data for section in sections]
+        else:
+            raise AttributeError("Overlap must be >= 0")
+
+        for df, offset in zip(dfs, offsets):
+            df.index = np.array(df.index) - df.index[0] + offset
+        combo = pd.concat(dfs)
+        combo.index.name = "t"
+
+        combo["t"] = combo.index
+
+        return Cls(combo)
+
     def label(self, **kwargs) -> Self:
         return self.__class__(self.data.assign(**kwargs))
 
@@ -177,9 +176,11 @@ class Table:
     
     def get_subset_df(self, **kwargs) -> pd.DataFrame:
         dfo = self.data
+        sel = np.full(len(self.data), True)
         for k, v in kwargs.items():
-            dfo = dfo.loc[dfo[k] == v, :]            
-        return dfo
+            sel = sel & (dfo[k] == v)
+
+        return self.data.loc[sel + (sel.astype(int).diff() == -1)]
 
     def get_label_subset(self, min_len=1, **kwargs) -> Self:
         return self.__class__(self.get_subset_df(**kwargs), min_len=min_len)
@@ -244,13 +245,14 @@ class Table:
         return self.label(**labels.to_dict(orient='list'))
     
     @classmethod
-    def shift_multi(Cls, steps: int, tb1: Self, tb2: Self, min_len=2) -> Tuple(Self, Self):
+    def shift_multi(Cls, steps: int, tb1: Self, tb2: Self, min_len=1) -> Tuple[Self, Self]:
         '''Take datapoints off the start of tb2 and add to the end tb1'''
-        if (steps>0 and len(tb2)-min_len<steps) or (steps<0 and min_len - len(tb1) > steps):
-            raise ValueError(f'Cannot Squash a Table to less than {min_len} datapoints')
-        dfj = Cls.stack([tb1, tb2], overlap=0).data
-        return Cls(dfj.iloc[:len(tb1)+steps, :]), \
-               Cls(dfj.iloc[len(tb1)+steps:, :])
+        #if (steps>0 and len(tb2)-min_len<steps) or (steps<0 and min_len - len(tb1) > steps):
+        #    raise ValueError(f'Cannot Squash a Table to less than {min_len} datapoints')
+        tj = Cls.stack([tb1, tb2]).shift_label(steps, min_len, **dict(tb1.labels.iloc[0]))
+        
+        return Cls(tj.get_subset_df(**dict(tb1.labels.iloc[0]))), \
+             Cls(tj.get_subset_df(**dict(tb2.labels.iloc[0])))
     
     def shift_label_ratio(self, ratio: float, min_len=None, **kwargs) -> Self:
         '''shift a label within its allowable bounds, with a ratio of

{flightdata-0.2.10 → flightdata-0.2.12}/flightdata/flight/flight.py

@@ -473,11 +473,19 @@ class Flight:
             wind_N = df['wN'] if 'wN' in df.columns else None,
             wind_E = df['wE'] if 'wE' in df.columns else None,
         )
-        origin = Origin('fcj_origin',GPS(
-            fc_json['parameters']['originLat'], 
-            fc_json['parameters']['originLng'], 
-            fc_json['parameters']['originAlt']
-        ), 0)
+        origin = Origin(
+            'fcj_origin',
+            GPS(
+                float(fc_json['parameters']['pilotLat']), 
+                float(fc_json['parameters']['pilotLng']), 
+                float(fc_json['parameters']['pilotAlt'])
+            ).offset(-Point(
+                fc_json['parameters']['moveNorth'],
+                fc_json['parameters']['moveEast'],
+                0,
+            )), 
+            0
+        )
         
         return Flight(df.set_index('time_flight', drop=False), None, origin, 'position')#.remove_time_flutter()
 

{flightdata-0.2.10 → flightdata-0.2.12}/flightdata/scripts/collect_logs.py

@@ -25,4 +25,7 @@ def main():
         if not (folder / file.name).exists():
             print(f'copying {file} to {folder / file.name}')
             shutil.copyfile(file, folder / file.name )
-        
+        
+
+if __name__ == '__main__':
+    main()

{flightdata-0.2.10 → flightdata-0.2.12}/flightdata/scripts/flightline.py

@@ -54,4 +54,7 @@ def main():
     else:
         box = box_from_log(Flight.from_log(plog), args.input)
 
-    box.to_f3a_zone(Path(args.logdir) / f'box_{plog.stem}.f3a')
+    box.to_f3a_zone(Path(args.logdir) / f'box_{plog.stem}.f3a')
+
+if __name__ == '__main__':
+    main()

{flightdata-0.2.10 → flightdata-0.2.12}/flightdata/state.py

@@ -217,7 +217,7 @@ class State(Table):
         
     def get_meid(self: State, manid: int, elid: int=None):
         st = self.get_manoeuvre(manid)
-        if not elid is None:
+        if elid is not None:
             return st.get_element(elid)
         else:
             return st

{flightdata-0.2.10 → flightdata-0.2.12/flightdata.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: flightdata
-Version: 0.2.10
+Version: 0.2.12
 Summary: Module for handling UAV flight log data
 Author-email: Thomas David <thomasdavid0@gmail.com>
 License: GNU GPL v3
@@ -34,3 +34,16 @@ The State object is a table representing the position and orientation of the air
 
 
 Further documentation will be provided here: https://pfcdocumentation.readthedocs.io/pyflightcoach/flightdata.html
+
+
+### Installation
+
+```bash
+pip install pfc-geometry
+```
+
+### Setup from source
+
+```bash
+pip install .
+```

{flightdata-0.2.10 → flightdata-0.2.12}/test/base/test_table.py

@@ -8,7 +8,7 @@ from pytest import fixture
 
 @fixture
 def df():
-    df = pd.DataFrame(np.linspace(0,100, 100), columns=['t'])
+    df = pd.DataFrame(np.linspace(0,5, 6), columns=['t'])
     return df.set_index("t", drop=False)
 
 @fixture
@@ -54,12 +54,15 @@ def test_get_subset_df(tab_full, labst):
     assert len(tab_full) == len(df)
 
 
-def test_shift_labels_ratios(tab_full):
-    tf = tab_full.label(element='e0', manoeuvre='m0')
-    tf.data.loc[tf.duration/2:,'element'] = 'e1'
-    
+def test_shift_labels_ratios(tab_full: Table):
+    tf: Table = Table.stack([
+        tab_full.label(element='e0', manoeuvre='m0'),
+        tab_full.label(element='e1', manoeuvre='m0')
+
+    ])
+        
     assert sum(tf.shift_labels_ratios([0.5],2).element == 'e1') < sum(tf.element == 'e1')
-    tf
+    
     
 def test_stack(tab_full):
     tfn = Table.stack([tab_full.label(element='e0'), tab_full.label(element='e1')], overlap=0)
@@ -72,18 +75,18 @@ def test_stack(tab_full):
 
 
 def test_shift_multi(tab_full):
-    t1, t2 = tab_full.label(element='e1'), tab_full.label(element='e2')
-    tb1, tb2 = Table.shift_multi(2, t1, t2)
+    tabs = Table.stack([tab_full.label(element='e0'), tab_full.label(element='e1')], overlap=1).split_labels()
+    tb1, tb2 = Table.shift_multi(2, tabs['e0'], tabs['e1'])
     
     assert len(tb1) == len(tab_full) + 2
     assert len(tb2) == len(tab_full) - 2
     assert tb2.duration == tab_full.data.index[-3]
     
-    tb1, tb2 = Table.shift_multi(-5, t1, t2)
+    tb1, tb2 = Table.shift_multi(-3, tabs['e0'], tabs['e1'])
     
-    assert len(tb1) == len(tab_full) - 5
-    assert len(tb2) == len(tab_full) + 5
-    assert tb1.duration == tab_full.data.index[-6]
+    assert len(tb1) == len(tab_full) - 3
+    assert len(tb2) == len(tab_full) + 3
+    assert tb1.duration == tab_full.data.index[-4]
 
 
 def test_table_cumulative_labels(tab_full):
@@ -102,6 +105,6 @@ def test_table_cumulative_labels(tab_full):
     indexes = np.unique(res, return_index=True)[1]
     np.testing.assert_array_equal(
         [res[index] for index in sorted(indexes)],
-        np.array(['a1_b1', 'a2_b1', 'a2_b2', 'a1_b1_1'])
+        np.array(['a1_b1_0', 'a2_b1_0', 'a2_b2_0', 'a1_b1_1'])
     )
     pass