electrostatic_potential Namespace Reference

Functions
def	draw_atoms

Variables
tuple	parser = ArgumentParser()
tuple	args = parser.parse_args()
string	output_dir = "tmp"
string	path_name = "/home/svenni/Dropbox/projects/programming/hartree-fock/build-hartree-fock-stan-Desktop_Qt_5_2_1_GCC_64bit-Release/app"
tuple	atoms_data_file = h5py.File(join(path_name, "results.h5"))
tuple	atom_meta = atoms_data_file.get("atomMeta")
tuple	atoms = atoms_data_file.get("state")
string	file_name = "electrostatic_potential.h5"
tuple	density_file = h5py.File(join(path_name, file_name))
tuple	data = density_file.get("dataset")
tuple	n_electrons = draw_atoms(atoms, atom_meta)
tuple	data_max_min_diff = (data.max() - data.min())
list	levels = [0.0003, 0.008]
list	contours = []
tuple	iso = mlab.contour3d(X, Y, Z, data, vmin=contours[0], vmax=contours[-1], opacity=0.5, contours=contours)

Function Documentation

def electrostatic_potential.draw_atoms	(		atoms,
			atom_meta
	)

Definition at line 28 of file electrostatic_potential.py.

def draw_atoms(atoms, atom_meta):
    n_electrons = 0
    counter = 0
    for atom in atoms:
        if atom_meta[counter]["type"] == 1:
            color = (1, 1, 1)
        elif atom_meta[counter]["type"] == 8:
            color = (1, 0, 0)
        else:
            color = (1, 1, 0)
        mlab.points3d(atom[0], atom[1], atom[2],
                          scale_factor=0.3,
                          resolution=20,
                          color=color,
                          scale_mode='none')
        counter += 1
    return n_electrons