ColorByDistance: Difference between revisions
Jump to navigation
Jump to search
mNo edit summary |
|||
(One intermediate revision by the same user not shown) | |||
Line 3: | Line 3: | ||
|author = [[User:zhentg|Zhenting Gao]] | |author = [[User:zhentg|Zhenting Gao]] | ||
}} | }} | ||
== Introduction == | |||
Color the surface of a binding site by the distance to a specific atom. | |||
==Code== | == Usage == | ||
*Open PyMOL | |||
*Load PDB files | |||
*run this Python script inside PyMOL | |||
*call the function | |||
**colorByDistance anchor, bindingSite | |||
== Required Arguments == | |||
Text | |||
== Optional Arguments == | |||
Text | |||
== Examples == | |||
Text | |||
== The Code == | |||
<source> | <source> | ||
def colorByDistance(anchor='anchor', site='site'): | def colorByDistance(anchor='anchor', site='site'): | ||
Line 11: | Line 28: | ||
#Update: 11/12/2018 | #Update: 11/12/2018 | ||
#Aim: Color atoms of a binding site based on their distance from a point | #Aim: Color atoms of a binding site based on their distance from a point | ||
#May also refer to https://pymolwiki.org/index.php/Ramp_New#Ramp_.2B_Distance_Measure | |||
# returns the length of the distance between atom A and atom B | # returns the length of the distance between atom A and atom B | ||
Line 25: | Line 43: | ||
# create the pseudoatom at the origin | # create the pseudoatom at the origin | ||
cmd.pseudoatom("pOrig", anchor, label="origin") | |||
# these are special PyMOL variables that will hold # the coordinates of | # these are special PyMOL variables that will hold # the coordinates of | ||
Line 35: | Line 53: | ||
# copy the coordinates into those special variables | # copy the coordinates into those special variables | ||
cmd.iterate_state(1, | cmd.iterate_state(1, "pOrig", 'stored.origCoord.append((x,y,z))') | ||
cmd.iterate_state(1, site, 'stored.distCoord.append((x,y,z))') | cmd.iterate_state(1, site, 'stored.distCoord.append((x,y,z))') | ||
Line 45: | Line 63: | ||
stored.newB = map(lambda x,y: diff_len(x,y), stored.distCoord, stored.origCoord) | stored.newB = map(lambda x,y: diff_len(x,y), stored.distCoord, stored.origCoord) | ||
print(stored.newB) | #print(stored.newB) | ||
# put them into the b-factor of the protein | # put them into the b-factor of the protein | ||
cmd.alter( | cmd.alter( site, "b=stored.newB.pop(0)") | ||
# color by rainbow_rev or any other | # color by rainbow_rev or any other | ||
Line 56: | Line 74: | ||
cmd.set("surface_color","-1",site) #color the surface of the binding site by corresponding atom colors | cmd.set("surface_color","-1",site) #color the surface of the binding site by corresponding atom colors | ||
cmd.extend('colorByDistance', colorByDistance) | cmd.extend('colorByDistance', colorByDistance) | ||
</source> | </source> |
Latest revision as of 06:08, 12 November 2018
Type | Python Script |
---|---|
Download | |
Author(s) | Zhenting Gao |
License | |
Introduction
Color the surface of a binding site by the distance to a specific atom.
Usage
- Open PyMOL
- Load PDB files
- run this Python script inside PyMOL
- call the function
- colorByDistance anchor, bindingSite
Required Arguments
Text
Optional Arguments
Text
Examples
Text
The Code
def colorByDistance(anchor='anchor', site='site'):
#Based on: https://pymolwiki.org/index.php/Spectrum#Intermediate
#Author: Zhenting Gao (zhentgpicasa@gmail.com)
#Update: 11/12/2018
#Aim: Color atoms of a binding site based on their distance from a point
#May also refer to https://pymolwiki.org/index.php/Ramp_New#Ramp_.2B_Distance_Measure
# returns the length of the distance between atom A and atom B
diff_len = lambda x,y : math.sqrt((x[0]-y[0])*(x[0]-y[0]) + (x[1]-y[1])*(x[1]-y[1]) + (x[2]-y[2])*(x[2]-y[2]))
# fetch site from the PDB
#fetch site, async=0
# show it as surface
#as surface
# create the pseudoatom at the origin
cmd.pseudoatom("pOrig", anchor, label="origin")
# these are special PyMOL variables that will hold # the coordinates of
# the atoms and the pseudoatom
stored.origCoord = []
stored.distCoord = []
# copy the coordinates into those special variables
cmd.iterate_state(1, "pOrig", 'stored.origCoord.append((x,y,z))')
cmd.iterate_state(1, site, 'stored.distCoord.append((x,y,z))')
# extend origCoord to be the same length as the other
stored.origCoord *= len(stored.distCoord)
# calculate the distances
stored.newB = map(lambda x,y: diff_len(x,y), stored.distCoord, stored.origCoord)
#print(stored.newB)
# put them into the b-factor of the protein
cmd.alter( site, "b=stored.newB.pop(0)")
# color by rainbow_rev or any other
# palette listed in "help spectrum"
cmd.spectrum(expression="b", palette="rainbow", selection=site)
cmd.set("surface_color","-1",site) #color the surface of the binding site by corresponding atom colors
cmd.extend('colorByDistance', colorByDistance)