# --- UCSF Chimera Copyright ---
# Copyright (c) 2000 Regents of the University of California.
# All rights reserved.  This software provided pursuant to a
# license agreement containing restrictions on its disclosure,
# duplication and use.  This notice must be embedded in or
# attached to all copies, including partial copies, of the
# software or any revisions or derivations thereof.
# --- UCSF Chimera Copyright ---
#
# $Id: v16.py 31310 2010-09-01 22:03:44Z gregc $

from v15 import reportRestoreError, restoreWindowSize, init, \
	restoreOpenModelsAttrs, RemapDialog, restoreLabels, \
	sessionID, registerAfterModelsCB, makeAfterModelsCBs, findFile, \
	idLookup, getColor, restoreVRML, restoreSurfaces, \
	restoreOpenStates, restoreColors, restoreVdw, \
	restoreDispChanged, restoreSelections, \
	restorePseudoBondGroups, restoreMolecules, \
	noAutoRestore, autoRestorable, restoreModelClip, beginRestore, \
	endRestore, restoreSecondaryStructure, restoreMiscAttrs

import globals # so that various version files can easily access same variables
import chimera

def restoreDrawModes(reprDict, xSections, ribScalings):
	from chimera import Residue
	crossSections = []
	for outline, a1, a2, a3 in xSections:
		cx = chimera.RibbonXSection(a1, a2, a3)
		cx.outline = outline
		crossSections.append(cx)
	for itemID, info in reprDict.items():
		item = idLookup(itemID)
		if isinstance(item, Residue):
			drawMode, xsi, styleClassName, size = info
			item.ribbonDrawMode = drawMode
			if xsi is not None:
				item.ribbonXSection = crossSections[xsi]
			if styleClassName is not None:
				klass = eval("chimera.%s" % styleClassName)
				item.ribbonStyle = klass(size)
		else:
			item.drawMode = info
	scalings = map(chimera.Residue.getDefaultRibbonStyle, [
		chimera.Residue.RS_TURN, chimera.Residue.RS_HELIX,
		chimera.Residue.RS_SHEET, chimera.Residue.RS_ARROW,
		chimera.Residue.RS_NUCLEIC])
	for i, rs in enumerate(scalings):
		width, thickness = ribScalings[i]
		if i == 3:
			arrowSW, arrowST = width, thickness
			continue
		rs.size = [width, thickness]
	# arrow has two sets...
	arrowEW, arrowET = ribScalings[-1]
	arrow = scalings[3]
	arrow.size = [arrowSW, arrowST, arrowEW, arrowET]

def restoreCamera(detail, fogColor, viewerBG, viewerHL,
					viewerLB, viewerAttrs, cameraAttrs):
	import chimera
	chimera.LODControl.get().quality = detail
	# globals.colorMap is deleted by end of session so look up colors now
	fogColor = getColor(fogColor)
	viewerBG = getColor(viewerBG)
	viewerHL = getColor(viewerHL)
	def delay(arg1, arg2, arg3, fogColor=fogColor, viewerBG=viewerBG,
			viewerHL=viewerHL,
			viewerAttrs=viewerAttrs, cameraAttrs=cameraAttrs):
		viewer = chimera.viewer
		viewer.depthCueColor = fogColor
		viewer.background = viewerBG
		viewer.highlightColor = viewerHL
		import v1
		v1.fixViewerAttrs(viewerAttrs)
		for attr, val in viewerAttrs.items():
			if attr in ("viewSize", "scaleFactor"):
				continue
			try:
				setattr(viewer, attr, val)
			except ValueError:
				# ignore highlight errors
				if attr != 'highlight':
					raise
		viewer.setViewSizeAndScaleFactor(viewerAttrs["viewSize"],
						viewerAttrs["scaleFactor"])
		camera = viewer.camera
		for attr, val in cameraAttrs.items():
			setattr(camera, attr, val)
		camera.fieldOfView = 25
		from chimera.triggerSet import ONESHOT
		return ONESHOT
	from SimpleSession import END_RESTORE_SESSION
	chimera.triggers.addHandler(END_RESTORE_SESSION, delay, None)