Con force i

bicycledataprocessor/database.py

@@ -93,9 +93,27 @@ def __init__(self, **kwargs):
         self.processedCols = ['FrameAccelerationX',
                               'FrameAccelerationY',
                               'FrameAccelerationZ',
+                              'LongitudinalRearContactAcceleration'
+                              'LateralRearContactAcceleration'
+                              'DownwardRearContactAcceleration'
+                              'LongitudinalFrontContactAcceleration'
+                              'LateralFrontContactAcceleration'
+                              'DownwardFrontContactAcceleration'
+                              'YawAngle'
+                              'FrontWheelYawAngle'
+                              'SteerTorque_Slip'
+                              'LongitudinalRearContactForce_Slip'
+                              'LateralRearContactForce_Slip'
+                              'LongitudinalFrontContactForce_Slip'
+                              'LateralFrontContactForce_Slip'
+                              'LongitudinalRearContactForce_Nonslip'
+                              'LateralRearContactForce_Nonslip'
+                              'LongitudinalFrontContactForce_Nonslip'
+                              'LateralFrontContactForce_Nonslip'
                               'PitchRate',
                               'PullForce',
                               'RearWheelRate',
+                              'FrontWheelRate',
                               'RollAngle',
                               'RollRate',
                               'ForwardSpeed',
@@ -361,6 +379,45 @@ def create_task_table(self):
             pass
         self.close()
 
+    def select_runs(self, riders, maneuvers, environments):
+        """Returns a list of runs given a set of conditions.
+
+        Parameters
+        ----------
+        riders : list
+        All or a subset of ['Charlie', 'Jason', 'Luke'].
+        maneuvers : list
+        All or a subset of ['Balance', 'Balance With Disturbance', 
+                'Track Straight Line', 'Track Straight Line With Disturbance'].
+        environments : list
+        All or a subset of ['Horse Treadmill', 'Pavillion Floor'].
+
+        Returns
+        -------
+        runs : list
+        List of run numbers for the given conditions.
+
+        """
+
+        self.open()
+
+        table = self.database.root.runTable
+
+        runs = []
+        for row in table.iterrows():
+            con = []
+            con.append(row['Rider'] in riders)
+            con.append(row['Maneuver'] in maneuvers)
+            con.append(row['Environment'] in environments)
+            con.append(row['corrupt'] is not True)
+            con.append(int(row['RunID']) > 100)
+            if False not in con:
+                runs.append(row['RunID'])
+
+        self.close()
+
+        return runs
+
     def sync_data(self, directory='exports/'):
         """Synchronizes data to the biosport website."""
         user = 'biosport'