blob: 9c5cb96bd777edbed1c3f762e2193f302f70c36a (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
|
*****************
Schrodinger Plot
*****************
Plots the solutions of schrodinger_solve. Visualises the given potential together
with the eigenstates and the probability density. Also plots the standard
deviation of each energy level.
Solutions of schrodinger_solve must be in a directory together for schrodinger_plot
to visualise them. Solution files must keep names given by schrodinger_solve.
The plot generated by schrodinger_plot can be altered with different scale and
axis limits to better visualize the solution. If no custom scale or limits are given
schrodinger_solve will limit the plot by the limits of the potential as well as the
energies of the solution for a better initial result.
**Usage**
.. code-block:: text
schrodinger_plot.py [-h] [-s SOLUTION_DIR] [-o OUTPUT_DIR] [--show SHOW] [-e EXPORT] [--scale SCALE] [-x XLIM] [-y1 ENERGY_LIM] [-y2 UNCERTAINTY_LIM]
**Command line options**
.. code-block:: text
-h, --help show a help message
-s SOLUTION_DIR --solution_dir
the path of the solution directory (default: None)
-o OUTPUT_DIR --output_dir
the path where the pdf should be saved (default: None)
--show SHOW Boolean, if True the plot is shown directly (default:
True)
-e EXPORT --export
Boolean, if True the plot is exported as a pdf
(default: True)
--scale SCALE Float, scales the wave functions (default: 1.0)
-x XLIM --xlim Limit of the x-axis of the left plot. None or
tuple[float, float] of shape (x_min, x_max)(default:
None)
-y1 ENERGY_LIM --energy_lim
Limit of the y-axis of the left plot. None or
tuple[float, float] of shape (y_min, y_max)(default:
None)
-y2 UNCERTAINTY_LIM --uncertainty_lim
Limit of the y-axis of the right plot. None or
tuple[float, float] of shape (y_min, y_max)(default:
None)
|
