fdtd.py 文件源码

def init_material_hy(self):
        """Set up the update mechanism for Hy field.

        Set up the update mechanism for Hy field and stores the result
        at self.pw_material[Hy].

        """
        self.pw_material[Hy] = {}
        shape = self.hy.shape
        for idx in ndindex(shape):
            spc = self.space.hy_index_to_space(*idx)
            mat_obj, underneath = self.geom_tree.material_of_point(spc)
            if idx[2] == 0 or idx[0] == 0:
                mat_obj = Dummy(mat_obj.eps_inf, mat_obj.mu_inf)
            pw_obj = mat_obj.get_pw_material_hy(idx, spc, underneath, self.cmplx)

            if self.pw_material[Hy].has_key(type(pw_obj)):
                self.pw_material[Hy][type(pw_obj)].merge(pw_obj)
            else:
                self.pw_material[Hy][type(pw_obj)] = pw_obj