def main(argv = None): if argv is None: argv = sys.argv argc = len(argv) self = "convert.py" if argv[0][-len(self):] != self: sys.exit("\n Rename script to %s\n" % self) if not (argc == 5 or (argc > 5 and ((argv[5].lower() == 'units' and argc == 9) or (argv[5].lower() == 'vacuum' and argc == 9) or (argv[5].lower() == 'supercell' and (argc == 7 or argc == 20)) or (argv[5].lower() == 'translate' and argc == 10) or (argv[5].lower() == 'rotate' and (argc == 8 or argc == 9)) or (argv[5].lower() == 'center' and argc == 8) or (argv[5].lower() == 'minbox' and argc == 7)))): sys.exit("\n Usage: %s ifn iff ofn off [action parameters]\n" % self + " ifn = input file name ofn = output file name\n" + " iff = input file format off = output file format\n" + " format = mat|paratec|vasp|espresso|siesta|tbpw|xyz|xsf|wien|povray\n" + " (wien & povray are only for output)\n\n" + " action parameters\n" + " ------------------\n" + " units lattice_constant_units lattice_vector_units atomic_position_units\n" + " vacuum axes separation_distance separation_distance_units\n" + " supercell ( sc_file | sc_origin sc_vectors sc_units sc_translation )\n" + " translate translation_vector translation_vector_units\n" + " rotate axis ( rotation_angle | adjacent_side opposite_side )\n" + " center axes center_type\n" + " minbox axes\n\n" + " axis = x | y | z rotation_angle in degrees\n" + " axes = x & y & z *_side in arbitrary cartesian units\n" + " sc_translation = true | false center_type = geometric | mass\n" + " *_units = bohr | angstrom | latconst | latvec\n") ifn = argv[1] ift = argv[2].lower() ofn = argv[3] oft = argv[4].lower() if ift == 'mat': ierr, mat = mat_read(ifn) elif ift == 'paratec': ierr, mat = paratec_read(ifn) elif ift == 'vasp': ierr, mat = vasp_read(ifn) elif ift == 'espresso': ierr, mat = espresso_read(ifn) elif ift == 'siesta': ierr, mat = siesta_read(ifn) elif ift == 'tbpw': ierr, mat = tbpw_read(ifn) elif ift == 'xyz': ierr, mat = xyz_read(ifn) elif ift == 'xsf': ierr, mat = xsf_read(ifn) else: sys.exit("\n Error: invalid input format %s\n" % ift) if ierr != 0: sys.exit("\n Error: unable to read file %s\n" % ifn) ierr = mat_check(mat) if ierr != 0: sys.exit("\n Error: invalid content in file %s\n" % ifn) if oft == 'povray': st = 1 umat = {} smin = [] smax = [] si = [] if argc != 5: if argv[5].lower() == 'units': ierr, fc = format_input(argv[6], 2) if ierr != 0: sys.exit("\n Error: invalid lattice_constant_units %s\n" % argv[6]) ierr, fo = format_input(argv[7], 3) if ierr != 0: sys.exit("\n Error: invalid lattice_vector_units %s\n" % argv[7]) ierr, fp = format_input(argv[8], 4) if ierr != 0: sys.exit("\n Error: invalid atomic_position_units %s\n" % argv[8]) if fo == format_index2mat[3]: fv = format_index2mat[2] else: fv = fo mat = mat_format(mat, fc, fo, fv, fp) elif argv[5].lower() == 'vacuum': axes = axes_input(argv[6]) try: distance = float(argv[7]) except: sys.exit("\n Error: invalid separation_distance %s\n" % argv[7]) ierr, format = format_input(argv[8], 3) if ierr != 0: sys.exit("\n Error: invalid separation_distance_units %s\n" % argv[8]) mat = mat_lattice(mat, axes, distance, format) elif argv[5].lower() == 'supercell': if argc == 7: ierr = 0 try: h = open(argv[6], 'r') except: ierr = 1 if ierr == 0: r = h.readlines() h.close() s = r[0] t = s.split() so = [float(t[0]), float(t[1]), float(t[2])] sv = [] for i in range(3): s = r[i + 1] t = s.split() sv.append([float(t[0]), float(t[1]), float(t[2])]) s = r[4] t = s.split() jerr, format = format_input(t[0], 4) ierr += jerr s = r[5] t = s.split() jerr, st = boolean_input(t[0]) ierr += jerr if ierr != 0: sys.exit("\n Error: unable to read supercell_file %s\n" % argv[6]) else: try: so = [float(argv[6]), float(argv[7]), float(argv[8])] except: sys.exit("\n Error: invalid supercell_origin %s %s %s\n" % (argv[6], argv[7], argv[8])) try: sv = [[float(argv[9]), float(argv[10]), float(argv[11])], [float(argv[12]), float(argv[13]), float(argv[14])], [float(argv[15]), float(argv[16]), float(argv[17])]] except: sys.exit("\n Error: invalid supercell_vectors %s %s %s %s %s %s %s %s %s\n" % (argv[9], argv[10], argv[11], argv[12], argv[13], argv[14], argv[15], argv[16], argv[17])) ierr, format = format_input(argv[18], 4) if ierr != 0: sys.exit("\n Error: invalid supercell_units %s\n" % argv[18]) ierr, st = boolean_input(argv[19]) if ierr != 0: sys.exit("\n Error: invalid supercell_translation %s\n" % argv[19]) if st == 0: umat = copy.deepcopy(mat) smin, smax, si, mat = mat_replicate(mat, so, sv, format) elif argv[5].lower() == 'translate': try: translation = [float(argv[6]), float(argv[7]), float(argv[8])] except: sys.exit("\n Error: invalid translation_vector %s %s %s\n" % (argv[6], argv[7], argv[8])) ierr, format = format_input(argv[9], 4) if ierr != 0: sys.exit("\n Error: invalid translation_vector_units %s\n" % argv[9]) mat = mat_translate(mat, translation, format) elif argv[5].lower() == 'rotate': ierr, axis = axis_input(argv[6]) if argc == 8: try: angle = float(argv[7]) * math.pi / 180.0 except: sys.exit("\n Error: invalid rotation_angle %s\n" % argv[7]) else: try: adjacent = float(argv[7]) except: sys.exit("\n Error: invalid adjacent_side %s\n" % argv[7]) try: opposite = float(argv[8]) except: sys.exit("\n Error: invalid opposite_side %s\n" % argv[8]) angle = math.atan2(opposite, adjacent) rotation = rotation_matrix(axis, angle) mat = mat_rotate(mat, rotation) elif argv[5].lower() == 'center': axes = axes_input(argv[6]) ierr, center_type = center_input(argv[7]) if ierr != 0: sys.exit("\n Error: invalid center_type %s\n" % argv[7]) format = mat['fp'] center = [] if center_type == 0: fr, rmin, rmax = mat_range(mat['na'], mat['fc'], mat['lc'], mat['fv'], mat['lv'], mat['fp'], mat['ap'], format) if fr != format: sys.exit("\n Error: failed center_type geometric\n") for j in range(3): center.append((rmin[j] + rmax[j]) / 2) else: if format == format_index2mat[3]: sys.exit("\n Error: center_type mass incompatible with atomic_position_units %s\n" % format) total_mass = 0.0 for j in range(3): center.append(0.0) for i in range(mat['na']): atomic_mass = periodic_table[index_by_number(mat['as'][i])]['mass'] total_mass += atomic_mass for j in range(3): center[j] += mat['ap'][i][j] * atomic_mass if abs(total_mass) > eps9: for j in range(3): center[j] /= total_mass else: sys.exit("\n Error: failed center_type mass\n") translation = [] for j in range(3): translation.append(-center[j] * axes[j]) mat = mat_translate(mat, translation, format) elif argv[5].lower() == 'minbox': axes = axes_input(argv[6]) for i in range(3): if axes[i] == 1: for j in range(3): if j != i: minwidth = inf9 minangle = 0.0 for k in range(180): angle = float(k) * math.pi / 180.0 rotation = rotation_matrix(j, angle) cmat = copy.deepcopy(mat) rmat = mat_rotate(cmat, rotation) format, rmin, rmax = mat_range(rmat['na'], rmat['fc'], rmat['lc'], rmat['fv'], rmat['lv'], rmat['fp'], rmat['ap'], 'bohr') cmat = [] rmat = [] width = rmax[i] - rmin[i] if width < minwidth - eps9: minwidth = width minangle = angle rotation = rotation_matrix(j, minangle) cmat = copy.deepcopy(mat) mat = mat_rotate(cmat, rotation) cmat = [] if oft == 'povray': box = 1 basis = 1 if st != 0: ierr, bonds = povray_bonds_st(mat) else: ierr, bonds = povray_bonds_ut(mat, umat, smin, smax, si) if ierr != 0: sys.exit("\n Error: failed povray_bonds\n") if oft == 'mat': ierr = mat_write(ofn, mat) elif oft == 'paratec': ierr = paratec_write(ofn, mat) elif oft == 'vasp': ierr = vasp_write(ofn, mat) elif oft == 'espresso': ierr = espresso_write(ofn, mat) elif oft == 'siesta': ierr = siesta_write(ofn, mat) elif oft == 'tbpw': ierr = tbpw_write(ofn, mat) elif oft == 'xyz': ierr = xyz_write(ofn, mat) elif oft == 'xsf': ierr = xsf_write(ofn, mat) elif oft == 'wien': ierr = wien_write(ofn, mat) elif oft == 'povray': ierr = povray_write(ofn, mat, box, basis, bonds) else: sys.exit("\n Error: invalid output format %s\n" % oft) if ierr != 0: sys.exit("\n Error: unable to write file %s\n" % ofn) return 0
def main(argv=None): if argv is None: argv = sys.argv argc = len(argv) self = "link.py" if argv[0][-len(self) :] != self: sys.exit("\n Rename script to %s\n" % self) if argc != 15: sys.exit( "\n Usage: %s fnc ffc fns ffs fna ffa icb icd isb isd acs facs rcs frcs\n" % self + " Description: links the side chain read from file fns to the core structure\n" + " read from file fnc and writes the obtained adduct to file fna. Variables\n" + " ffc, ffs, ffa set to mat|paratec|vasp|espresso|siesta|tbpw|xyz|xsf specify\n" + " the file formats. The core and side atoms with indices icb and isb are\n" + " bonded together, while the core and side atoms with indices icd and isd\n" + " are detached. The side chain is oriented in such a way that bonds icb-icd,\n" + " isd-isb, and icb-isb are all collinear. Variable acs gives the angle by\n" + " which the side chain is rotated around the icb-isb axis. Variable rcs\n" + " gives the distance between icb and isb atoms. Variables facs and frcs set\n" + " to radian or degree and bohr or angstrom define the units of acs and rcs.\n" ) fnc = argv[1] ffc = argv[2].lower() fns = argv[3] ffs = argv[4].lower() fna = argv[5] ffa = argv[6].lower() try: icb = int(argv[7]) - 1 except: icb = -1 try: icd = int(argv[8]) - 1 except: icd = -1 try: isb = int(argv[9]) - 1 except: isb = -1 try: isd = int(argv[10]) - 1 except: isd = -1 try: acs = float(argv[11]) except: sys.exit("\n Error: acs %s\n" % argv[11]) facs = argv[12].lower() if facs == "r" or facs == "rad" or facs == "radian": pass elif facs == "d" or facs == "deg" or facs == "degree": acs *= math.pi / 180.0 else: sys.exit("\n Error: facs %s\n" % argv[12]) try: rcs = float(argv[13]) except: rsc = -inf9 if rcs < eps9: sys.exit("\n Error: rcs %s\n" % argv[13]) frcs = argv[14].lower() if frcs == "b" or frcs == "bohr": format = format_index2mat[0] elif frcs == "a" or frcs == "ang" or frcs == "angstrom": format = format_index2mat[1] else: sys.exit("\n Error: frcs %s\n" % argv[14]) if ffc == "mat": ierr, matc = mat_read(fnc) elif ffc == "paratec": ierr, matc = paratec_read(fnc) elif ffc == "vasp": ierr, matc = vasp_read(fnc) elif ffc == "espresso": ierr, matc = espresso_read(fnc) elif ffc == "siesta": ierr, matc = siesta_read(fnc) elif ffc == "tbpw": ierr, matc = tbpw_read(fnc) elif ffc == "xyz": ierr, matc = xyz_read(fnc) elif ffc == "xsf": ierr, matc = xsf_read(fnc) else: sys.exit("\n Error: invalid input format %s\n" % ffc) if ierr != 0: sys.exit("\n Error: unable to read file %s\n" % fnc) ierr = mat_check(matc) if ierr != 0: sys.exit("\n Error: invalid content in file %s\n" % fnc) matc = mat_format(matc, format, format, format, format) if matc["fc"] != format or matc["fo"] != format or matc["fv"] != format or matc["fp"] != format: sys.exit("\n Error: unable to change format in file %s\n" % fnc) if icb < 0 or icb > matc["na"] - 1: sys.exit("\n Error: icb %s\n" % argv[7]) if icd < 0 or icd > matc["na"] - 1 or icd == icb: sys.exit("\n Error: icd %s\n" % argv[8]) vc = [] for j in range(3): vc.append(matc["ap"][icd][j] - matc["ap"][icb][j]) rc = 0.0 for j in range(3): rc += math.pow(vc[j], 2) rc = math.sqrt(rc) if rc < eps9: sys.exit("\n Error: atomic positions in file %s\n" % fnc) if ffs == "mat": ierr, mats = mat_read(fns) elif ffs == "paratec": ierr, mats = paratec_read(fns) elif ffs == "vasp": ierr, mats = vasp_read(fns) elif ffs == "espresso": ierr, mats = espresso_read(fns) elif ffs == "siesta": ierr, mats = siesta_read(fns) elif ffs == "tbpw": ierr, mats = tbpw_read(fns) elif ffs == "xyz": ierr, mats = xyz_read(fns) elif ffs == "xsf": ierr, mats = xsf_read(fns) else: sys.exit("\n Error: invalid input format %s\n" % ffs) if ierr != 0: sys.exit("\n Error: unable to read file %s\n" % fns) ierr = mat_check(matc) if ierr != 0: sys.exit("\n Error: invalid content in file %s\n" % fns) mats = mat_format(mats, format, format, format, format) if mats["fc"] != format or mats["fo"] != format or mats["fv"] != format or mats["fp"] != format: sys.exit("\n Error: unable to change format in file %s\n" % fns) if isb < 0 or isb > mats["na"] - 1: sys.exit("\n Error: isb %s\n" % argv[9]) if isd < 0 or isd > mats["na"] - 1 or isd == isb: sys.exit("\n Error: isd %s\n" % argv[10]) vs = [] for j in range(3): vs.append(mats["ap"][isb][j] - mats["ap"][isd][j]) rs = 0.0 for j in range(3): rs += math.pow(vs[j], 2) rs = math.sqrt(rs) if rs < eps9: sys.exit("\n Error: atomic positions in file %s\n" % fns) numberc, axesc, anglesc = align_vector(vc) numbers, axess, angless = align_vector(vs) number = numbers + 1 + numberc axes = [] angles = [] for i in range(numbers): axes.append(axess[i]) angles.append(angless[i]) axes.append(2) angles.append(acs) for i in range(numberc - 1, -1, -1): axes.append(axesc[i]) angles.append(-anglesc[i]) rotation = [[1.0, 0.0, 0.0], [0.0, 1.0, 0.0], [0.0, 0.0, 1.0]] for i in range(number): rotation = matrix_dot_matrix(rotation_matrix(axes[i], angles[i]), rotation) mats = mat_rotate(mats, rotation) translation = [0.0, 0.0, 0.0] for j in range(3): translation[j] += matc["ap"][icb][j] - mats["ap"][isb][j] + vc[j] * rcs / rc mats = mat_translate(mats, translation, format) matc = mat_remove(matc, icd) mats = mat_remove(mats, isd) mata = mat_merge(matc, mats) if ffa == "mat": ierr = mat_write(fna, mata) elif ffa == "paratec": ierr = paratec_write(fna, mata) elif ffa == "vasp": ierr = vasp_write(fna, mata) elif ffa == "espresso": ierr = espresso_write(fna, mata) elif ffa == "siesta": ierr = siesta_write(fna, mata) elif ffa == "tbpw": ierr = tbpw_write(fna, mata) elif ffa == "xyz": ierr = xyz_write(fna, mata) elif ffa == "xsf": ierr = xsf_write(fna, mata) else: sys.exit("\n Error: invalid output format %s\n" % ffa) if ierr != 0: sys.exit("\n Error: unable to write file %s\n" % fna) return 0
def main(argv = None): if argv is None: argv = sys.argv argc = len(argv) self = "volume.py" if argv[0][-len(self):] != self: sys.exit("\n Rename script to %s\n" % self) if not (argc == 9 or argc == 10): sys.exit("\n Usage: %s imfn imff ivfn ivff ovfn ovff phas cplx [hole]\n" % self + " imfn = input matter file name\n" + " imff = input matter file format\n" + " (mat|paratec|vasp|espresso|siesta|tbpw|xyz|xsf)\n" + " ivfn = input volumetric file name\n" + " ivff = input volumetric file format (a3dr)\n" + " ovfn = output volumetric file name\n" + " ovff = output volumetric file format (cube|xsf)\n" + " phas = remove wavefunction phase (true|false)\n" + " cplx = complex wavefunction (re|im|abs|abs2)\n" + " hole = plot hole (true|false)\n") imfn = argv[1] imff = argv[2].lower() ivfn = argv[3] ivff = argv[4].lower() ovfn = argv[5] ovff = argv[6].lower() if imff == 'mat': ierr, mat = mat_read(imfn) elif imff == 'paratec': ierr, mat = paratec_read(imfn) elif imff == 'vasp': ierr, mat = vasp_read(imfn) elif imff == 'espresso': ierr, mat = espresso_read(imfn) elif imff == 'siesta': ierr, mat = siesta_read(imfn) elif imff == 'tbpw': ierr, mat = tbpw_read(imfn) elif imff == 'xyz': ierr, mat = xyz_read(imfn) elif imff == 'xsf': ierr, mat = xsf_read(imfn) else: sys.exit("\n Error: invalid input format %s\n" % imff) if ierr != 0: sys.exit("\n Error: unable to read file %s\n" % imfn) ierr = mat_check(mat) if ierr != 0: sys.exit("\n Error: invalid content in file %s\n" % imfn) if ivff == 'a3dr': ierr, state, energy, holpos, latvec, grid, psi = a3dr_read(ivfn) else: sys.exit("\n Error: invalid input format %s\n" % ivff) if ierr != 0: sys.exit("\n Error: unable to read file %s\n" % ivfn) format = format_index2mat[0] so = [0.0, 0.0, 0.0] # temporary hack for a3dr format # lattice vectors in a3dr file are written in %.6f format # in paratec benzene example, lattice vectors read from a3dr file # differ from lattice vectors read from paratec input file # this results in identical atoms written to rho.cube # (total of 16 atoms instead of 12 atoms) # this hack computes the supercell dimensions and # replaces the lattice vectors read from a3dr file # with the lattice vectors read from the input file # multiplied by the supercell dimensions mat = mat_format(mat, mat['fc'], mat['fo'], format, mat['fp']) a1 = mat['lv'][0] a2 = mat['lv'][1] a3 = mat['lv'][2] b1 = latvec[0] b2 = latvec[1] b3 = latvec[2] n1 = int(round((a1[0] * b1[0] + a1[1] * b1[1] + a1[2] * b1[2]) / (a1[0]**2 + a1[1]**2 + a1[2]**2))) n2 = int(round((a2[0] * b2[0] + a2[1] * b2[1] + a2[2] * b2[2]) / (a2[0]**2 + a2[1]**2 + a2[2]**2))) n3 = int(round((a3[0] * b3[0] + a3[1] * b3[1] + a3[2] * b3[2]) / (a3[0]**2 + a3[1]**2 + a3[2]**2))) b1 = [n1 * a1[0], n1 * a1[1], n1 * a1[2]] b2 = [n2 * a2[0], n2 * a2[1], n2 * a2[2]] b3 = [n3 * a3[0], n3 * a3[1], n3 * a3[2]] latvec = [b1, b2, b3] # end of hack sv = latvec smin, smax, si, mat = mat_replicate(mat, so, sv, format) if mat['fo'] != format and mat['fv'] != format and mat['fp'] != format: sys.exit("\n Error: unable to make supercell\n") if argc > 9 and holpos != []: ierr, hole = boolean_input(argv[9]) if ierr != 0: sys.exit("\n Error: invalid hole %s\n" % argv[9]) if hole != 0: mat['as'].append(0) mat['ap'].append(holpos) mat['na'] += 1 mat['ns'] = len(mat_species(mat['na'], mat['as'])) ierr, phas = boolean_input(argv[7]) if ierr != 0: sys.exit("\n Error: invalid phase %s\n" % argv[7]) if phas != 0: rmax = 0.0 for i in range(grid[0]): for j in range(grid[1]): for k in range(grid[2]): r = math.sqrt(math.pow(psi[k][j][i][0], 2) + math.pow(psi[k][j][i][1], 2)) if r > rmax: rmax = r phase = math.atan2(psi[k][j][i][1], psi[k][j][i][0]) cs = math.cos(phase) sn = math.sin(phase) for i in range(grid[0]): for j in range(grid[1]): for k in range(grid[2]): x = psi[k][j][i][0] y = psi[k][j][i][1] psi[k][j][i][0] = cs * x + sn * y psi[k][j][i][1] = -sn * x + cs * y ierr, cplx = cplx_input(argv[8]) if ierr != 0: sys.exit("\n Error: invalid cplx %s\n" % argv[8]) phi = [] for i in range(grid[0]): phi.append([]) for j in range(grid[1]): phi[i].append([]) for k in range(grid[2]): if cplx == 0: chi = psi[k][j][i][0] elif cplx == 1: chi = psi[k][j][i][1] elif cplx == 2: chi = math.sqrt(math.pow(psi[k][j][i][0], 2) + math.pow(psi[k][j][i][1], 2)) elif cplx == 3: chi = math.pow(psi[k][j][i][0], 2) + math.pow(psi[k][j][i][1], 2) else: chi = 0.0 if phas != 0 and (cplx == 2 or cplx == 3) and psi[k][j][i][0] < 0.0: chi = -chi phi[i][j].append(chi) if ovff == 'cube': i = 0 elif ovff == 'xsf': i = 1 else: sys.exit("\n Error: invalid output format %s\n" % ovff) fc = format_index2mat[i] fo = format_index2mat[i] fv = format_index2mat[i] fp = format_index2mat[i] mat = mat_format(mat, fc, fo, fv, fp) if mat['fc'] != fc or mat['fo'] != fo or mat['fv'] != fv or mat['fp'] != fp: sys.exit("\n Error: invalid content in file %s\n" % imfn) na = mat['na'] aspecies = [] for i in range(na): aspecies.append(mat['as'][i]) acharge = [] for i in range(na): acharge.append(float(mat['as'][i])) aposition = [] for i in range(na): aposition.append([]) for j in range(3): aposition[i].append(mat['ap'][i][j]) origin = [] for j in range(3): origin.append(mat['lo'][j]) latvec = [] for i in range(3): latvec.append([]) for j in range(3): latvec[i].append(mat['lv'][i][j]) step = [] for i in range(3): step.append([]) for j in range(3): step[i].append(latvec[i][j] / float(grid[i])) if ovff == 'cube': ierr = cub_write(ovfn, state, energy, na, aspecies, acharge, aposition, grid, origin, step, phi) elif ovff == 'xsf': ierr = xsf_write(ovfn, state, energy, na, aspecies, aposition, grid, origin, latvec, phi) else: sys.exit("\n Error: invalid output format %s\n" % ovff) if ierr != 0: sys.exit("\n Error: unable to write file %s\n" % ovfn) return 0