rasmus/nafuma
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add multiply and drawdown to plotting

Browse source
This commit is contained in:
rasmusthog 2022-10-09 18:42:05 +02:00
parent 8b0a5bcff7
commit 2e469909d7
1 changed files with 5 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

11
nafuma/xrd/plot.py
View file

@ -22,16 +22,14 @@ def plot_diffractogram(data, options={}):
Input: Input:
data (dict): Must include path = string to diffractogram data, and plot_kind = (recx, beamline, image)''' data (dict): Must include path = string to diffractogram data, and plot_kind = (recx, beamline, image)'''
# Update options
required_options = ['x_vals', 'y_vals', 'ylabel', 'xlabel', 'xunit', 'yunit', 'line', 'scatter', 'xlim', 'ylim', 'normalise', 'offset', 'offset_x', 'offset_y', 'offset_change',
'reflections_plot', 'reflections_indices', 'reflections_data', 'heatmap', 'cmap', 'plot_kind', 'palettes', 'highlight', 'highlight_colours', 'interactive', 'rc_params', 'format_params', 'interactive_session_active', 'plot_diff']
default_options = { default_options = {
'x_vals': '2th', 'y_vals': 'I', 'x_vals': '2th', 'y_vals': 'I',
'xlabel': '2$\\theta$', 'ylabel': None, 'xlabel': '2$\\theta$', 'ylabel': None,
'xunit': '$^{\circ}$', 'yunit': None, 'xunit': '$^{\circ}$', 'yunit': None,
'xlim': None, 'ylim': None, 'xlim': None, 'ylim': None,
'normalise': True, 'normalise': True,
'multiply': 1, # Factor to multiply the normalised data - only used if normalising.
'drawdown': False,
'offset': True, 'offset': True,
'offset_x': 0, 'offset_x': 0,
'offset_y': 1, 'offset_y': 1,
@ -42,6 +40,7 @@ def plot_diffractogram(data, options={}):
'reflections_indices': False, # whether to plot the reflection indices 'reflections_indices': False, # whether to plot the reflection indices
'reflections_data': None, # Should be passed as a list of dictionaries on the form {path: rel_path, reflection_indices: number of indices, colour: [r,g,b], min_alpha: 0-1] 'reflections_data': None, # Should be passed as a list of dictionaries on the form {path: rel_path, reflection_indices: number of indices, colour: [r,g,b], min_alpha: 0-1]
'heatmap': False, 'heatmap': False,
'heatmap_reverse': False,
'cmap': 'viridis', 'cmap': 'viridis',
'plot_kind': None, 'plot_kind': None,
'palettes': [('qualitative', 'Dark2_8')], 'palettes': [('qualitative', 'Dark2_8')],
@ -58,7 +57,7 @@ def plot_diffractogram(data, options={}):
if len(data['path']) > 10: if len(data['path']) > 10:
default_options['offset_y'] = 0.05 default_options['offset_y'] = 0.05
options = aux.update_options(options=options, required_options=required_options, default_options=default_options) options = aux.update_options(options=options, default_options=default_options)
#options['current_offset_y'] = options['offset_y'] #options['current_offset_y'] = options['offset_y']
# Convert data['path'] to list to allow iteration over this to accommodate both single and multiple diffractograms # Convert data['path'] to list to allow iteration over this to accommodate both single and multiple diffractograms
@ -322,7 +321,7 @@ def plot_diffractogram(data, options={}):
if options['plot_diff'] and len(data['path']) == 2: if options['plot_diff'] and len(data['path']) == 2:
diff = data['diffractogram'][0] diff = data['diffractogram'][0]
diff['I'] = diff['I'] - data['diffractogram'][1]['I'] diff['I'] = diff['I'] - data['diffractogram'][1]['I']
diff['I'] = diff['I'] - 0.5 diff['I'] = diff['I'] - 0.75
diff.plot(x=options['x_vals'], y=options['y_vals'], ax=ax, c=next(colours)) diff.plot(x=options['x_vals'], y=options['y_vals'], ax=ax, c=next(colours))