Method _get_field

python/solver.py:399–430

(self, f, band, bloch_phase)

        self.mode_solver.compute_field_divergence()

    def _get_field(self, f, band, bloch_phase):
        if self.mode_solver is None:
            raise ValueError(
                "Must call a run function before attempting to get a field"
            )

        if f == "b":
            self.mode_solver.get_bfield(band)
        elif f == "d":
            self.mode_solver.get_dfield(band)
        elif f == "e":
            self.mode_solver.get_efield(band)
        elif f == "h":
            self.mode_solver.get_hfield(band)

        dims = self.mode_solver.get_dims()

        while len(dims) < 3:
            dims += [1]

        dims += [3]
        arr = np.zeros(np.prod(dims), np.complex128)

        if bloch_phase:
            self.mode_solver.multiply_bloch_phase()

        self.mode_solver.get_curfield_cmplx(arr)

        arr = np.reshape(arr, dims)
        return MPBArray(
            arr, self.get_lattice(), self.current_k, bloch_phase=bloch_phase
        )

    def get_curfield_as_array(self, bloch_phase=True):
        dims = self.mode_solver.get_dims()