| 142 | |
| 143 | |
| 144 | def view_sim(): |
| 145 | sim = setup_sim(8) |
| 146 | xy0 = mp.Volume( |
| 147 | center=mp.Vector3(0, 0, 0), size=mp.Vector3(sim.cell_size.x, sim.cell_size.y, 0) |
| 148 | ) |
| 149 | xy1 = mp.Volume( |
| 150 | center=mp.Vector3(0, 0, 1), size=mp.Vector3(sim.cell_size.x, sim.cell_size.y, 0) |
| 151 | ) |
| 152 | yz0 = mp.Volume( |
| 153 | center=mp.Vector3(0, 0, 0), size=mp.Vector3(0, sim.cell_size.y, sim.cell_size.z) |
| 154 | ) |
| 155 | yz1 = mp.Volume( |
| 156 | center=mp.Vector3(1, 0, 0), size=mp.Vector3(0, sim.cell_size.y, sim.cell_size.z) |
| 157 | ) |
| 158 | xz0 = mp.Volume( |
| 159 | center=mp.Vector3(0, 0, 0), size=mp.Vector3(sim.cell_size.x, 0, sim.cell_size.z) |
| 160 | ) |
| 161 | xz1 = mp.Volume( |
| 162 | center=mp.Vector3(0, 1, 0), size=mp.Vector3(sim.cell_size.x, 0, sim.cell_size.z) |
| 163 | ) |
| 164 | vols = [xy0, xy1, yz0, yz1, xz0, xz1] |
| 165 | titles = ["xy0", "xy1", "yz0", "yz1", "xz0", "xz1"] |
| 166 | xlabel = ["x", "x", "y", "y", "x", "x"] |
| 167 | ylabel = ["y", "y", "z", "z", "z", "z"] |
| 168 | for k in range(len(vols)): |
| 169 | ax = plt.subplot(2, 3, k + 1) |
| 170 | sim.plot2D(ax=ax, output_plane=vols[k]) |
| 171 | ax.set_xlabel(xlabel[k]) |
| 172 | ax.set_ylabel(ylabel[k]) |
| 173 | ax.set_title(titles[k]) |
| 174 | plt.tight_layout() |
| 175 | plt.show() |
| 176 | |
| 177 | |
| 178 | class TestVisualization(unittest.TestCase): |