|
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