def main(): pdb = parse_PDB(sys.argv[1]) for name, xyz in zip( ['B', 'C', 'X'], [XYZ(7.5, 0, 0), XYZ(0, 7.5, 0), XYZ(7.5, 7.5, 7.5)]): new_chain = copy.deepcopy(pdb.chains['A']) new_chain.change_chain_name(name) new_chain.translate_xyz(xyz) pdb.add_chain(new_chain) pdb.renumber() write_PDB('test.pdb', pdb)
def read_RBs(file_name): results = [] with open(file_name, 'r') as fin: for l in fin: floats = [float(a) for a in l.split()[1:]] rb = RigidBody(XYZ(floats[0], floats[1], floats[2]), XYZ(floats[3], floats[4], floats[5]), XYZ(floats[6], floats[7], floats[8]), XYZ(floats[9], floats[10], floats[11])) results.append(rb) break print(results)