Scripts: czone.py

#
# Color zone the first surface using the selected atoms and radius set below.
#
from chimera import selection as s, openModels as om
from _surface import SurfaceModel
radius = 5
surface = om.list(modelTypes = [SurfaceModel])[0]
xform_to_surface = surface.openState.xform.inverse()
atoms = s.currentAtoms()
bonds = s.currentBonds()
import ColorZone as cz
points, colors = cz.points_and_colors(atoms, bonds, xform_to_surface)
cz.color_zone(surface, points, colors, radius, auto_update = True)