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.