|
46 | 46 | "from matplotlib import pyplot as plt\n",
|
47 | 47 | "\n",
|
48 | 48 | "from jwave import FiniteDifferences, FourierSeries\n",
|
49 |
| - "from jwave.geometry import Domain, Medium, _circ_mask\n", |
50 |
| - "from jwave.utils import plot_complex_field, show_positive_field\n", |
| 49 | + "from jwave.geometry import Domain, Medium, circ_mask\n", |
| 50 | + "from jwave.utils import display_complex_field, show_positive_field\n", |
51 | 51 | "\n",
|
52 | 52 | "key = random.PRNGKey(42) # Random seed"
|
53 | 53 | ]
|
|
104 | 104 | "src = FourierSeries(jnp.expand_dims(src_field, -1), domain) * omega\n",
|
105 | 105 | "\n",
|
106 | 106 | "# Plotting\n",
|
107 |
| - "_ = plot_complex_field(src)" |
| 107 | + "_ = display_complex_field(src)" |
108 | 108 | ]
|
109 | 109 | },
|
110 | 110 | {
|
|
119 | 119 | "sound_speed = sound_speed.at[20:105, 20:200].set(1.0)\n",
|
120 | 120 | "sound_speed = (\n",
|
121 | 121 | " sound_speed\n",
|
122 |
| - " * (1 - _circ_mask(N, 90, [64, 180]))\n", |
123 |
| - " * (1 - _circ_mask(N, 50, [64, 22]))\n", |
| 122 | + " * (1 - circ_mask(N, 90, [64, 180]))\n", |
| 123 | + " * (1 - circ_mask(N, 50, [64, 22]))\n", |
124 | 124 | " * 0.5\n",
|
125 | 125 | " + 1\n",
|
126 | 126 | ")\n",
|
|
170 | 170 | }
|
171 | 171 | ],
|
172 | 172 | "source": [
|
173 |
| - "_ = plot_complex_field(field, max_intensity=2e5)" |
| 173 | + "_ = display_complex_field(field, max_intensity=2e5)" |
174 | 174 | ]
|
175 | 175 | },
|
176 | 176 | {
|
|
272 | 272 | "source": [
|
273 | 273 | "# Solve new problem\n",
|
274 | 274 | "field = solve_helmholtz(medium, params)\n",
|
275 |
| - "_ = plot_complex_field(field, max_intensity=2e5)" |
| 275 | + "_ = display_complex_field(field, max_intensity=2e5)" |
276 | 276 | ]
|
277 | 277 | },
|
278 | 278 | {
|
|
316 | 316 | "\n",
|
317 | 317 | "# Solve new problem\n",
|
318 | 318 | "field = solve_helmholtz(medium, params)\n",
|
319 |
| - "_ = plot_complex_field(field, max_intensity=2e5)" |
| 319 | + "_ = display_complex_field(field, max_intensity=2e5)" |
320 | 320 | ]
|
321 | 321 | },
|
322 | 322 | {
|
|
0 commit comments