ChernoffFaceWidget.py

# ChernoffFaceWidget.py
# (C)2013
# Scott Ernst, Peter McKay, Ryan Leonard.

import nimble
from math import radians, sin, cos
from nimble import cmds
from pyglass.widgets.PyGlassWidget import PyGlassWidget
import re
import FaceController as fc
from geo import getCensusData

def _validateCoordString(coord_str):
    if not re.match("^(\-?\d+(\.\d+)?)$", coord_str):
        raise Exception("Unable to use this coordinate!")
    return coord_str



#___________________________________________________________________________________________________ ChernoffFaceWidget
class ChernoffFaceWidget(PyGlassWidget):
	"""A class for the finaly Project"""

#===================================================================================================
#                                                                                       C L A S S
#___________________________________________________________________________________________________ __init__
	def __init__(self, parent, **kwargs):
		"""Creates a new instance of ChernoffFaceWidget."""
		super(ChernoffFaceWidget, self).__init__(parent, **kwargs)
	# manual pane
		self.eyebrowAngleSlider.valueChanged.connect(self._handleEyebrowChange)
		self.mouthAngleSlider.valueChanged.connect(self._handleMouthAngleChange)
		self.eyeSizeSlider.valueChanged.connect(self._handleEyeSizeChange)
		self.mouthSizeSlider.valueChanged.connect(self._handleMouthSizeChange)
		self.headShapeSlider.valueChanged.connect(self._handleHeadShapeChange)
		self.eyeSpacingSlider.valueChanged.connect(self._handleEyeSpacingChange)
	# nav pane
		self.homeBtn.clicked.connect(self._handleReturnHome)
	# census data pane
		self.calculateCensusBtn.clicked.connect(self._handleCalculateBtn)
	# set all to default values to reflect current state of sliders
		self._updateMayaBasedOnSliders()
	#END __init__

	def _updateMayaBasedOnSliders(self):
		self._handleEyebrowChange()
		self._handleMouthAngleChange()
		self._handleEyeSizeChange()
		self._handleMouthSizeChange()
		self._handleHeadShapeChange()
		self._handleEyeSpacingChange()

	#===================================================================================================
	#                                                                                 H A N D L E R S
  
	#___________________________________________________________________________________________________ _handleValChange
	def _handleEyebrowChange(self):
		eyeAngle = self.eyebrowAngleSlider.value() * 10
		print("eyebrow angle: "+str(eyeAngle))
		fc.eyeBrow(eyeAngle)

  
	#___________________________________________________________________________________________________ _handleValChange
	def _handleEyeSpacingChange(self):
		eyeSpace = self.eyeSpacingSlider.value() * 10
		print("eye spacing: "+str(eyeSpace))
		fc.eyeDistance(eyeSpace)

  
	#___________________________________________________________________________________________________ _handleValChange
	def _handleMouthSizeChange(self):
		mouthSize = self.mouthSizeSlider.value() * 10
		print("mouth size: "+str(mouthSize))
		fc.mouthEnbiggen(mouthSize)

  
	#___________________________________________________________________________________________________ _handleValChange
	def _handleEyeSizeChange(self):
		eyeSize = self.eyeSizeSlider.value() * 10
		print("eye size: "+str(eyeSize))
		fc.eyeEnbiggen(eyeSize)

  
	#___________________________________________________________________________________________________ _handleValChange
	def _handleMouthAngleChange(self):
		mouthAngle = self.mouthAngleSlider.value() * 10
		print("Mouth Angle: "+str(mouthAngle))
		fc.mouth(mouthAngle)

	#___________________________________________________________________________________________________ _handleValChange
	def _handleHeadShapeChange(self):
		headShape = self.headShapeSlider.value() * 10
		print("head shape: "+str(headShape))
		fc.head(headShape)

	#___________________________________________________________________________________________________ _handleCalculateBtn
	def _handleCalculateBtn(self):
		# collect data based on a lat/long provided by user
		latStr = _validateCoordString(self.latitudeLineEdit.text())
		longStr = _validateCoordString(self.longitudeLineEdit.text())
		data = getCensusData(float(latStr), float(longStr))

		# apply that data to sliders
		self.eyebrowAngleSlider.setValue(data[0])
		self.mouthAngleSlider.setValue(data[1])
		self.eyeSizeSlider.setValue(data[2])
		self.mouthSizeSlider.setValue(data[3])
		self.headShapeSlider.setValue(data[4])
		self.eyeSpacingSlider.setValue(data[5])

		# update maya scene based on sliders
		self._updateMayaBasedOnSliders()

	#___________________________________________________________________________________________________ _handleRunBtn
	def _handleRunBtn(self):
		print("Generate!")

	#___________________________________________________________________________________________________ _handleReturnHome
	def _handleReturnHome(self):
		self.mainWindow.setActiveWidget('home')