def main(): args = parse_command_line_arguments() coordinate_types = { 'd' : 'Direct', 'direct' : 'Direct', 'c' : 'Cartesian', 'cartesian' : 'Cartesian' } coordinate_type = coordinate_types[ args.coordinate_type ] # initialise poscar = Poscar() # read POSCAR file poscar.read_from( args.poscar ) if args.scale: poscar.cell.matrix *= poscar.scaling poscar.scaling = 1.0 if args.supercell: # generate supercell if args.group: # check that if grouping is switched on, we are asking for a supercell that allows a "3D-chequerboard" pattern. for (i,axis) in zip( args.supercell, range(3) ): if i%2 == 1 and i > 1: raise Exception( "odd supercell expansions != 1 are incompatible with automatic grouping" ) poscar = poscar.replicate( *args.supercell, group=args.group ) if args.bohr: poscar = poscar.in_bohr() # output to stdout output_opts = { 'label' : args.label, 'numbered' : args.number_atoms, 'coordinates_only' : args.coordinates_only, 'selective': args.selective } poscar.output( coordinate_type=coordinate_type, opts=output_opts )
def main(): args = parse_command_line_arguments() coordinate_types = { 'd': 'Direct', 'direct': 'Direct', 'c': 'Cartesian', 'cartesian': 'Cartesian' } coordinate_type = coordinate_types[args.coordinate_type] # initialise poscar = Poscar() # read POSCAR file poscar.read_from(args.poscar) if args.scale: poscar.cell.matrix *= poscar.scaling poscar.scaling = 1.0 if args.supercell: # generate supercell if args.group: # check that if grouping is switched on, we are asking for a supercell that allows a "3D-chequerboard" pattern. for (i, axis) in zip(args.supercell, range(3)): if i % 2 == 1 and i > 1: raise Exception( "odd supercell expansions != 1 are incompatible with automatic grouping" ) poscar = poscar.replicate(*args.supercell, group=args.group) if args.bohr: poscar = poscar.in_bohr() # output to stdout output_opts = { 'label': args.label, 'numbered': args.number_atoms, 'coordinates_only': args.coordinates_only, 'selective': args.selective } poscar.output(coordinate_type=coordinate_type, opts=output_opts)
parser.add_argument( '-b', '--bohr', action='store_true', help='assumes the input file is in Angstrom, and converts everything to bohr') parser.add_argument( '-n', '--number-atoms', action='store_true', help='label coordinates with atom number' ) args = parser.parse_args() return( args ) if __name__ == "__main__": args = parse_command_line_arguments() coordinate_types = { 'd' : 'Direct', 'direct' : 'Direct', 'c' : 'Cartesian', 'cartesian' : 'Cartesian' } coordinate_type = coordinate_types[ args.coordinate_type ] # initialise poscar = Poscar() # read POSCAR file poscar.read_from( args.poscar ) if args.supercell: # generate supercell if args.group: # check that if grouping is switched on, we are asking for a supercell that allows a "3D-chequerboard" pattern. for (i,axis) in zip( args.supercell, range(3) ): if i%2 == 1 and i > 1: raise Exception( "odd supercell expansions != 1 are incompatible with automatic grouping" ) poscar = poscar.replicate( *args.supercell, group=args.group ) if args.bohr: poscar = poscar.in_bohr() # output to stdout output_opts = { 'label' : args.label, 'numbered' : args.number_atoms, 'coordinates_only' : args.coordinates_only } poscar.output( coordinate_type = coordinate_type, opts = output_opts )
return( args ) if __name__ == "__main__": args = parse_command_line_arguments() coordinate_types = { 'd' : 'Direct', 'direct' : 'Direct', 'c' : 'Cartesian', 'cartesian' : 'Cartesian' } coordinate_type = coordinate_types[ args.coordinate_type ] # initialise poscar = Poscar() # read POSCAR file poscar.read_from( args.poscar ) if args.scale: poscar.cell.matrix *= poscar.scaling poscar.scaling = 1.0 if args.supercell: # generate supercell if args.group: # check that if grouping is switched on, we are asking for a supercell that allows a "3D-chequerboard" pattern. for (i,axis) in zip( args.supercell, range(3) ): if i%2 == 1 and i > 1: raise Exception( "odd supercell expansions != 1 are incompatible with automatic grouping" ) poscar = poscar.replicate( *args.supercell, group=args.group ) if args.bohr: poscar = poscar.in_bohr() # output to stdout output_opts = { 'label' : args.label, 'numbered' : args.number_atoms, 'coordinates_only' : args.coordinates_only } poscar.output( coordinate_type = coordinate_type, opts = output_opts )