(
sim: Simulation,
ax: Optional[Axes] = None,
fields: Optional = None,
output_plane: Optional[Volume] = None,
field_parameters: Optional[dict] = None,
colorbar_parameters: Optional[dict] = None,
)
| 842 | |
| 843 | |
| 844 | def plot_fields( |
| 845 | sim: Simulation, |
| 846 | ax: Optional[Axes] = None, |
| 847 | fields: Optional = None, |
| 848 | output_plane: Optional[Volume] = None, |
| 849 | field_parameters: Optional[dict] = None, |
| 850 | colorbar_parameters: Optional[dict] = None, |
| 851 | ) -> Union[Axes, Any]: |
| 852 | components = { |
| 853 | mp.Ex, |
| 854 | mp.Ey, |
| 855 | mp.Ez, |
| 856 | mp.Er, |
| 857 | mp.Ep, |
| 858 | mp.Dx, |
| 859 | mp.Dy, |
| 860 | mp.Dz, |
| 861 | mp.Dr, |
| 862 | mp.Dp, |
| 863 | mp.Hx, |
| 864 | mp.Hy, |
| 865 | mp.Hz, |
| 866 | mp.Hr, |
| 867 | mp.Hp, |
| 868 | mp.Bx, |
| 869 | mp.By, |
| 870 | mp.Bz, |
| 871 | mp.Br, |
| 872 | mp.Bp, |
| 873 | mp.Sx, |
| 874 | mp.Sy, |
| 875 | mp.Sz, |
| 876 | mp.Sr, |
| 877 | mp.Sp, |
| 878 | } |
| 879 | |
| 880 | if not sim._is_initialized: |
| 881 | sim.init_sim() |
| 882 | |
| 883 | if fields is None: |
| 884 | return ax |
| 885 | |
| 886 | if field_parameters is None: |
| 887 | field_parameters = default_field_parameters |
| 888 | else: |
| 889 | field_parameters = dict(default_field_parameters, **field_parameters) |
| 890 | |
| 891 | # user specifies a field component |
| 892 | if fields not in components: |
| 893 | raise ValueError("Please specify a valid field component (mp.Ex, mp.Ey, ...") |
| 894 | |
| 895 | # Get domain measurements |
| 896 | sim_center, sim_size = get_2D_dimensions(sim, output_plane) |
| 897 | |
| 898 | xmin, xmax, ymin, ymax, zmin, zmax = box_vertices( |
| 899 | sim_center, sim_size, sim.is_cylindrical |
| 900 | ) |
| 901 |
no test coverage detected