added interpolation for force and impedance data

medunigraz · Apr 16, 2024 · 233a5b5 · 233a5b5
1 parent 42aeca4
commit 233a5b5
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/pyceps/datatypes/study.py b/pyceps/datatypes/study.py
@@ -573,7 +573,8 @@ def interpolate_data(self, which):
 
         Parameters:
             which : str
-                parameter to interpolate, options: ['uni', 'bip', 'act']
+                parameter to interpolate
+                options: ['uni', 'bip', 'lat', 'imp', 'frc']
 
         Raises:
             KeyError : If parameter to interpolate is unknown
@@ -609,6 +610,10 @@ def interpolate_data(self, which):
             data = np.asarray([p.bipVoltage for p in valid_points])
         elif which.lower() == 'uni':
             data = np.asarray([p.uniVoltage for p in valid_points])
+        elif which.lower() == 'imp':
+            data = np.asarray([p.impedance for p in valid_points])
+        elif which.lower() == 'frc':
+            data = np.asarray([p.force for p in valid_points])
         else:
             raise KeyError()
 

diff --git a/pyceps/fileio/cartoio.py b/pyceps/fileio/cartoio.py
@@ -1194,6 +1194,8 @@ def import_map(self, egm_names_from_pos=False):
         self.interpolate_data('lat')
         self.interpolate_data('bip')
         self.interpolate_data('uni')
+        self.interpolate_data('imp')
+        self.interpolate_data('frc')
         self.bsecg = self.get_map_ecg(method=['median', 'mse', 'ccf'])
 
     def load_mesh(self):