Method _output_scalar_field

python/solver.py:707–752

(self, curfield_type, fname_prefix)

                f[name] = np.imag(component_field)

    def _output_scalar_field(self, curfield_type, fname_prefix):
        components = ["x", "y", "z"]

        if curfield_type == "n":
            fname = "epsilon"
            description = "dielectric function, epsilon"
        elif curfield_type == "m":
            fname = "mu"
            description = "permeability mu"
        else:
            kpoint_index = self.mode_solver.get_kpoint_index()
            curfield_band = self.mode_solver.curfield_band
            fname = "{}pwr.k{:02d}.b{:02d}".format(
                curfield_type.lower(), kpoint_index, curfield_band
            )
            descr_fmt = "{} field energy density, kpoint {}, band {}, freq={:.6g}"
            description = descr_fmt.format(
                curfield_type,
                kpoint_index,
                curfield_band,
                self.freqs[curfield_band - 1],
            )

        parity_suffix = curfield_type not in "mn"
        fname = self._create_fname(fname, fname_prefix, parity_suffix)
        if verbosity.mpb > 0:
            print(f"Outputting {fname}...")

        with h5py.File(fname, "w") as f:
            f["description"] = description.encode()
            self._create_h5_dataset(f, "data")
            self._write_lattice_vectors(f)

            if curfield_type == "n":
                for inv in [False, True]:
                    inv_str = "epsilon_inverse" if inv else "epsilon"
                    for c1 in range(3):
                        for c2 in range(c1, 3):
                            self.mode_solver.get_epsilon_tensor(c1, c2, 0, inv)
                            dataname = f"{inv_str}.{components[c1]}{components[c2]}"
                            self._create_h5_dataset(f, dataname)

                            if with_hermitian_epsilon() and c1 != c2:
                                self.mode_solver.get_epsilon_tensor(c1, c2, 1, inv)
                                dataname += ".i"
                                self._create_h5_dataset(f, dataname)

    def _write_lattice_vectors(self, h5file):
        lattice = np.zeros((3, 3))