adding show_legend to total_traj_plots

track_analyzer/plotting.py

@@ -1377,7 +1377,7 @@ def plot_total_traj(data_dir, df, dim=3, plot_dir=None, plot_fn=None, plot_confi
     cmap_lim = specific_config['cmap_lim']
     subset_order = specific_config['subset_order']
     transparency = specific_config['transparency']
-    show_legend = True
+    show_legend = specific_config['show_legend']
 
     # get info
     info = tpr.get_info(data_dir)

track_analyzer/prepare.py

@@ -1347,6 +1347,7 @@ def make_traj_config(data_dir=None, export_config=True):
                          'cmap_lim': None,  # pass custom colormap limits
                          'subset_order': None,  # if color_code is group, order of group in color cycle
                          'transparency': 1,  #plot transparency
+                         'show_legend': True  # show legend
                          }
 
     # config of voronoi plot

track_analyzer/scripts/analyze_maps.py

@@ -199,8 +199,8 @@ def map_analysis(data_dir, data=None, image=None, refresh=False, parallelize=Fal
                 plot_dir = osp.join(sub_dir, field)
                 tpr.safe_mkdir(plot_dir)
             map_param_ = dict(map_param)
-            map_param_['vlim'] = scalar_fields_[field]['vlim']
-            map_param_['cmap'] = scalar_fields_[field]['cmap']
+            map_param_['vlim'] = scalar_fields_[field]['vlim'] if 'vlim' in scalar_fields_[field].keys() else None
+            map_param_['cmap'] = scalar_fields_[field]['cmap'] if 'cmap' in scalar_fields_[field].keys() else 'plasma'
             tpl.plot_all_scalar_fields(data_dir, df, field_data, field, image=image, map_param=map_param_,
                                        plot_dir=plot_dir, plot_config=plot_config, dont_print_count=False)
         for field in vector_fields.keys():