Atom("C", array([0.0, 2 * a, 0.0])), ] four_atoms_cell = deepcopy(system.atoms) shift_r = array([0.0, 3.0 * a, 0.0]) for i in range(1, n): for atom in four_atoms_cell: new_atom = deepcopy(atom) new_atom.r = new_atom.r + i * shift_r system.atoms.append(new_atom) system.spin_multiplier = 1 system.k_points = [array([-pi / sqrt(3) / a, 0.0, 0.0]), array([0.0, 0.0, 0.0]), array([pi / sqrt(3) / a, 0, 0])] system.make_k_mesh(150) system.parameters = { "C": {"ep": 1.2057, "ed": 24.1657, "lambda": 0.001}, "CC": {"Vppp": -3.26, "Vpps": 0.0, "Vpds": 0.0, "Vpdp": 2.4, "Vdds": 0.0, "Vddp": 3.6, "Vddd": -7.4}, } for i in xrange(len(system.atoms)): system.atoms[i].orbitals = ["pz"] # ,'dxy', 'dyz', 'dxz', 'dx2-y2', 'dz2'] system.just_do_main_magic() lst = system.find_indeces_for_ldos(atom_idx=0) print lst plt = Plotter(system.name) plt.plot_energy_bands_from_file() doser = LDOSCalculator(1, system.name, 200, 0, indeces_list=lst) doser.f()