From PyMOLWiki
Revision as of 02:36, 13 July 2009 by Shiven (talk | contribs) (Code)
Jump to: navigation, search


An attempt to perform a coloring of two structures by RMS deviation as calculated by PyMol's internal Rms_Cur command.


WARNING: This is still a work in progress, and is almost useless right now! It is just a proof of principle at this stage and mainly an attempt at using the "alignment" object to iterate over the aligned objects. However, if you make any improvements, please do edit this page to reflect the same.

--- ColorByRMSD: RMSD based coloring --- 
Author  : Shivender Shandilya
Program : ColByRMS
Date    : July 2009
Version : 0.0.1 (very very alpha!)
Mail    :

Keywords: color rms rmsd colorbyrms colorbyrmsd
 This email from Warren -
 DeLano, W.L. The PyMOL Molecular Graphics System (2002) DeLano Scientific, San Carlos, CA, USA.

import pymol
import cmd

# Code for a special case first
cmd.load("D:/PyMOL-1.2r1/data/tut/1hpv.pdb", "1hpv")

cmd.create("chA", "1hpv and polymer and chain A")
cmd.create("chB", "1hpv and polymer and chain B")

# We need the alignment object, but PyMol refuses to give us that unless
# we (needlessly) specify all the other values in the cmd.align() function
# all the 0's are needlessly forced to be specified...
# PyMol is not smart enough to use the defaults
# cmd.align("chA and name CA", "chB and name CA",0,0,0,0,0,object="aln")
# So, lets use "super" instead
cmd.super("chA", "chB",object="aln")

# Utter the magic word

# Arrays to store all residue identifiers in "aln"
stored.alnAres = []
stored.alnBres = []

# Now interrogate "aln" to get residue numbers for each object
cmd.iterate("chA and aln","stored.alnAres.append(resi)")
cmd.iterate("chB and aln","stored.alnBres.append(resi)")

# The main function that assigns "cur_rms"*10 as the new b-factor
def colbyRMS(objA, alnAri, objB, alnBri):
    cmd.alter(str(objA), "chain='Z'")
    cmd.alter(str(objB), "chain='Z'")
    for x in range(len(alnAri)):
        rmsd = cmd.rms_cur(str(objA)+" and i. "+str(alnAri[x]), str(objB)+" and i. "+str(alnBri[x]))
        cmd.alter(str(objA)+" and resi "+str(alnAri[x]), "b = "+str(rmsd*10))
        cmd.alter(str(objB)+" and resi "+str(alnBri[x]), "b = "+str(rmsd*10))

cmd.extend("colbyRMS", colbyRMS)

# Run the just defined function
if len(stored.alnAres) > len(stored.alnBres):

# Arrays to store the NEW b-factors
stored.alnAnb = []
stored.alnBnb = []

# Store the NEW b-factors
cmd.iterate("chA and aln","stored.alnAnb.append(b)")
cmd.iterate("chB and aln","stored.alnBnb.append(b)")

# Assign the just stored NEW b-factors to the original object
for x in range(len(stored.alnAres)):
    cmd.alter("1hpv and chain A and resi "+str(stored.alnAres[x]), "b = "+str(stored.alnAnb[x]))
for x in range(len(stored.alnBres)):
    cmd.alter("1hpv and chain B and resi "+str(stored.alnBres[x]), "b = "+str(stored.alnBnb[x]))

# Get rid of all intermediate objects etc.

# Showcase what you did
cmd.hide("all")"cartoon", "1hpv")
print "\n"
print "Colored by 'overall' RMSD...\n"
cmd.spectrum("b", "rainbow", "1hpv")