/usr/share/pyshared/Bio/PDB/Superimposer.py is in python-biopython 1.58-1.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 | # Copyright (C) 2002, Thomas Hamelryck (thamelry@binf.ku.dk)
# This code is part of the Biopython distribution and governed by its
# license. Please see the LICENSE file that should have been included
# as part of this package.
"""Superimpose two structures."""
import numpy
from Bio.SVDSuperimposer import SVDSuperimposer
from Bio.PDB.PDBExceptions import PDBException
class Superimposer(object):
"""
Rotate/translate one set of atoms on top of another,
thereby minimizing the RMSD.
"""
def __init__(self):
self.rotran=None
self.rms=None
def set_atoms(self, fixed, moving):
"""
Put (translate/rotate) the atoms in fixed on the atoms in
moving, in such a way that the RMSD is minimized.
@param fixed: list of (fixed) atoms
@param moving: list of (moving) atoms
@type fixed,moving: [L{Atom}, L{Atom},...]
"""
if not (len(fixed)==len(moving)):
raise PDBException("Fixed and moving atom lists differ in size")
l=len(fixed)
fixed_coord=numpy.zeros((l, 3))
moving_coord=numpy.zeros((l, 3))
for i in range(0, len(fixed)):
fixed_coord[i]=fixed[i].get_coord()
moving_coord[i]=moving[i].get_coord()
sup=SVDSuperimposer()
sup.set(fixed_coord, moving_coord)
sup.run()
self.rms=sup.get_rms()
self.rotran=sup.get_rotran()
def apply(self, atom_list):
"""
Rotate/translate a list of atoms.
"""
if self.rotran is None:
raise PDBException("No transformation has been calculated yet")
rot, tran=self.rotran
rot=rot.astype('f')
tran=tran.astype('f')
for atom in atom_list:
atom.transform(rot, tran)
if __name__=="__main__":
import sys
from Bio.PDB import PDBParser, Selection
p=PDBParser()
s1=p.get_structure("FIXED", sys.argv[1])
fixed=Selection.unfold_entities(s1, "A")
s2=p.get_structure("MOVING", sys.argv[1])
moving=Selection.unfold_entities(s2, "A")
rot=numpy.identity(3).astype('f')
tran=numpy.array((1.0, 2.0, 3.0), 'f')
for atom in moving:
atom.transform(rot, tran)
sup=Superimposer()
sup.set_atoms(fixed, moving)
print sup.rotran
print sup.rms
sup.apply(moving)
|