Scripts: surfsilhouette.py

# ----------------------------------------------------------------------------
# Save images of orthographically projected surface silhouettes for several
# orientations.
#
# Create surfaces then open this script in Chimera (File / Open) to create
# the images.  Tested in Chimera 1.4 daily build September 2, 2009.
#
def surface_silhouette_images(directory = '~/Desktop/silhouettes',
                              filename = 'sil_%04d.png',
                              axis = 'y', angle_step = 10, count = 36,
                              multilayer = False):

  # Find all surface models
  from chimera import openModels as om, runCommand as run, replyobj
  from _surface import SurfaceModel
  surfs = om.list(modelTypes = [SurfaceModel])
  if len(surfs) == 0:
    replyobj.warning('surfsilhouette.py: No surfaces')
    return

  # Turn off lighting for surfaces
  for s in surfs:
    for p in s.surfacePieces:
      p.useLighting = False

  # Color surfaces white, or transparent gray for multilayer.
  if multilayer:
    for s in surfs:
      s.oneTransparentLayer = False
      for p in s.surfacePieces:
        p.vertexColors = None
        p.color = (.1,.1,.1,0)
        p.tranparencyBlendMode = p.SRC_1_DST_1_MINUS_ALPHA # Add
  else:
    for s in surfs:
      for p in s.surfacePieces:
        p.vertexColors = None
        p.color = (1,1,1,1)
  
  # Turn off depth cueing.
  run('~set depth_cue')

  # Set orthographic projection
  run('set projection orthographic')

  # Turn off glossy lighting
  run('set light_quality normal')
  
  # Expand ~ to user's home directory.
  from OpenSave import tildeExpand
  directory = tildeExpand(directory)
  
  # Make directory if it doesn't exist.
  import os, os.path
  if not os.path.isdir(directory):
    os.mkdir(directory)
    
  # Save images.
  for i in range(count):
    path = os.path.join(directory, filename % i)
    run('copy file %s' % path)
    run('turn %s %f' % (axis, angle_step))

# ----------------------------------------------------------------------------
#
surface_silhouette_images()
#surface_silhouette_images(multilayer = True)