[sphand_ros] Add reflectance_param to format_printed_prox_to_csv

start-jsk · Feb 12, 2020 · 811e482 · 811e482
1 parent 36e4188
commit 811e482
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/demos/sphand_ros/sphand_driver/scripts/format_printed_prox_to_csv b/demos/sphand_ros/sphand_driver/scripts/format_printed_prox_to_csv
@@ -35,7 +35,18 @@ def main():
             # elif 'init_value' in line:
             #     init = yaml.load(line, Loader=yaml.SafeLoader)['init_value']
             #     print(init + diff)
-                print(diff)
+                print('{},'.format(diff), end='')
+                cnt = cnt + 1
+        elif cnt == 3:
+            if 'prop_const' in line:
+                # Old msg
+                print(yaml.load(line,
+                                Loader=yaml.SafeLoader)['prop_const'])
+                cnt = 0
+            elif 'reflectance_param' in line:
+                # New msg
+                print(yaml.load(line,
+                                Loader=yaml.SafeLoader)['reflectance_param'])
                 cnt = 0
         line = f.readline()
     f.close()