|
14 | 14 |
|
15 | 15 | def slice(fnames:List[str], fields:List[str], |
16 | 16 | loc: str = "top", widthScale: float = 3.0, |
17 | | - dr: Optional: float = 0.15, |
| 17 | + dr: Optional[float] = None, |
18 | 18 | theta: Optional[float] = None) -> None: |
19 | 19 | """ |
20 | 20 | A slice plot of the datasets for different field parameters for Spherical2D geometry. |
@@ -58,9 +58,14 @@ def slice(fnames:List[str], fields:List[str], |
58 | 58 | # Process information for each dataset |
59 | 59 |
|
60 | 60 | # Some geometry properties |
61 | | - rr = ds.domain_right_edge[0].in_units("km") |
62 | 61 | rl = ds.domain_left_edge[0].in_units("km") |
63 | | - dr = dr * rr.units |
| 62 | + if dr is None: |
| 63 | + rr = ds.domain_right_edge[0].in_units("km") |
| 64 | + dr = rr - rl |
| 65 | + else: |
| 66 | + dr = dr * rl.units |
| 67 | + rr = rl + dr |
| 68 | + |
64 | 69 | r_center = 0.5 * dr + rl |
65 | 70 |
|
66 | 71 | thetar = ds.domain_right_edge[1] |
@@ -122,7 +127,8 @@ def slice(fnames:List[str], fields:List[str], |
122 | 127 | [rr*np.sin(theta), rr*np.cos(theta)], |
123 | 128 | coord_system="plot", |
124 | 129 | color="k", |
125 | | - linestyle="--") |
| 130 | + linewidth=1.5, |
| 131 | + linestyle="-.") |
126 | 132 |
|
127 | 133 | plot = sp.plots[field] |
128 | 134 | plot.figure = fig |
@@ -181,7 +187,7 @@ def slice(fnames:List[str], fields:List[str], |
181 | 187 | parser.add_argument('-t', '--theta', type=float, |
182 | 188 | help="""user defined theta center location of the plot domain. |
183 | 189 | Alternative way of defining plotting center""") |
184 | | - parser.add_argument('-r', '--dr', default=0.15, type=float, |
| 190 | + parser.add_argument('-r', '--dr', type=float, |
185 | 191 | help="""Distance between upper r and lower r shown in the SlicePlot. |
186 | 192 | Assumed in unit km. This is used to control center and width of the SlicePlot""") |
187 | 193 | parser.add_argument('-w', '--width', default=4.0, type=float, |
|
0 commit comments