|
MDStressLab++
|
Functions | |
| read_file (filename) | |
| Reads a .stress file containing atomistic stress data. | |
| plot_contour (data, output_fname, component_index, component_label) | |
| Plots a 2D contour of the selected stress component. | |
Variables | |
| dict | STRESS_COMPONENTS |
| Mapping from stress component names to (column index, label). | |
| filename = sys.argv[1] | |
| component_key = sys.argv[2].lower() | |
| component_index | |
| component_label | |
| base | |
| _ | |
| str | plt_fname = f"{base}_{component_key}.pdf" |
| data = read_file(filename) | |
| visualize.plot_contour | ( | data, | |
| output_fname, | |||
| component_index, | |||
| component_label | |||
| ) |
Plots a 2D contour of the selected stress component.
| data | List of rows with [x, y, z, σ_xx, ..., σ_yz]. |
| output_fname | Output PDF filename. |
| component_index | Index of the stress component to plot. |
| component_label | Label for the plot title. |
Uses SciPy's griddata to interpolate the scattered data, and matplotlib to create a filled contour plot with 100 levels using the RdBu_r colormap.
Definition at line 86 of file visualize.py.
| visualize.read_file | ( | filename | ) |
Reads a .stress file containing atomistic stress data.
| filename | The path to the .stress file. |
The file is expected to have:
Malformed or non-numeric lines are skipped.
Definition at line 61 of file visualize.py.
|
protected |
Definition at line 122 of file visualize.py.
| visualize.base |
Definition at line 122 of file visualize.py.
| visualize.component_index |
Definition at line 120 of file visualize.py.
| visualize.component_key = sys.argv[2].lower() |
Definition at line 114 of file visualize.py.
| visualize.component_label |
Definition at line 120 of file visualize.py.
Definition at line 125 of file visualize.py.
| visualize.filename = sys.argv[1] |
Definition at line 113 of file visualize.py.
| str visualize.plt_fname = f"{base}_{component_key}.pdf" |
Definition at line 123 of file visualize.py.
| dict visualize.STRESS_COMPONENTS |
Mapping from stress component names to (column index, label).
Definition at line 41 of file visualize.py.