Commit babd627c authored by amichaut's avatar amichaut
Browse files

adding show_legend to total_traj_plots

parent 4d2a5702
Pipeline #74094 passed with stages
in 29 seconds
......@@ -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)
......
......@@ -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
......
......@@ -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():
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment