def agent(self, agentName, id, agentSpec): '''If the given agent already exists, return it. Otherwise guess its agent type and id based on its name, and build a new agent with the appropriate AgentSpec. If no AgentSpec exists for that type, raise an error. This method also handles filtering out agents that do not exist in the user specified "selections"''' if not self._selectionGroup.contains( id ): # Filter out non-selected agents # return None try: agent = self._agents[agentName] except: # agentName does not already exist (probably no callsheet was # provided) - build one *if* the agentType is defined. The # agentType should have already been defined if a CDL file # was specified in the MAS file for the agentType. # if not agentSpec: agentSpec = self.scene.agentSpec(self._agentType(agentName)) agent = Agent.Agent() agent.name = agentName agent.id = id agent.agentSpec = agentSpec self._agents[agentName] = agent return agent
def read(amcFile, simData=None):
'''Load animation data from a .amc file and adds it to an SimData.Agent.
The name of the SimData.Agent is gotten from the .amc file name.
If a simData is provided, it will be queried to get SimData.Agent.
If the specified SimData.Agent does not exist, the read will exit
early. If no simData is provided a new SimData.Agent is created.
In both cases the target SimData.Agent is returned (although if
a SimData was provided it will also store the target SimData.Agent).
For now amcFile must be a path to a .amc file (as opposed to an open
file handle).'''
# AMC sim files are named: agentType.#.amc where agentType.# is
# the name of that particular agent instance
#
tokens = os.path.basename(amcFile).split(".")
agentName = Agent.formatAgentName(tokens[0], tokens[1])
# Python 2.4 limitation: try... except: finally: doesn't work,
# have to nest try... except: in try... finally:
try:
if simData:
agentSim = simData.agent(agentName)
if not agentSim:
# Agent is not in one of the chosen "selections"
#
return None
else:
agentSim = SimData.Agent(agentName)
path = os.path.dirname(amcFile)
fileHandle = open(amcFile, "r")
try:
frame = 0
for line in fileHandle:
tokens = line.strip().split()
if tokens:
if tokens[0][0] == ":":
# file options
continue
elif len(tokens) == 1:
# sample number
frame = int(tokens[0])
else:
jointName = tokens[0]
data = []
for i in range(1, len(tokens)):
data.append(float(tokens[i]))
agentSim.addSample(jointName, frame, data)
finally:
fileHandle.close()
except:
print >> sys.stderr, "Error reading AMC file: %s" % amcFile
raise
return agentSim
def read(amcFile, simData=None):
'''Load animation data from a .amc file and adds it to an SimData.Agent.
The name of the SimData.Agent is gotten from the .amc file name.
If a simData is provided, it will be queried to get SimData.Agent.
If the specified SimData.Agent does not exist, the read will exit
early. If no simData is provided a new SimData.Agent is created.
In both cases the target SimData.Agent is returned (although if
a SimData was provided it will also store the target SimData.Agent).
For now amcFile must be a path to a .amc file (as opposed to an open
file handle).'''
# AMC sim files are named: agentType.#.amc where agentType.# is
# the name of that particular agent instance
#
tokens = os.path.basename(amcFile).split(".")
agentName = Agent.formatAgentName(tokens[0], tokens[1])
# Python 2.4 limitation: try... except: finally: doesn't work,
# have to nest try... except: in try... finally:
try:
if simData:
agentSim = simData.agent(agentName)
if not agentSim:
# Agent is not in one of the chosen "selections"
#
return None
else:
agentSim = SimData.Agent(agentName)
path = os.path.dirname(amcFile)
fileHandle = open(amcFile, "r")
try:
frame = 0
for line in fileHandle:
tokens = line.strip().split()
if tokens:
if tokens[0][0] == ":":
# file options
continue
elif len(tokens) == 1:
# sample number
frame = int(tokens[0])
else:
jointName = tokens[0]
data = []
for i in range(1, len(tokens)):
data.append(float(tokens[i]))
agentSim.addSample(jointName, frame, data)
finally:
fileHandle.close()
except:
print >> sys.stderr, "Error reading AMC file: %s" % amcFile
raise
return agentSim
def build(self, scene, agentOptions):
if scene.mas().terrainFile:
self._factory.importObj(scene.mas().terrainFile, "terrain")
MayaPlacement.build(self._factory, scene.mas())
for agentSpec in scene.agentSpecs():
agent = Agent.Agent(instanced=False)
agent.agentSpec = agentSpec
agent.name = agentSpec.agentType
mayaAgent = MayaSceneAgent.MayaSceneAgent(agent, self._factory,
scene)
mayaAgent.build(agentOptions)
self._factory.addAgent(mayaAgent)
self._factory.cleanup()
# The layers are off by default to speed up load, turn them on
# now.
#
MayaAgent.showLayers()
def read(callsheet, sim):
"""Load the simmed values of agent variables"""
if isinstance(callsheet, basestring):
fileHandle = open(callsheet, "r")
else:
fileHandle = callsheet
try:
try:
for line in fileHandle:
tokens = line.strip().split()
if tokens:
# 0 : id
# 1 : name
# 2-17: placement matrix
# 18 : keyword "cdl"
# 19 : cdl file path
# 20-?: variable names and values
id = int(tokens[0])
agentName = Agent.formatAgentName(tokens[1])
agentSpec = sim.scene.agentSpec(sim.scene.resolvePath(tokens[19]))
agent = sim.agent(agentName, id, agentSpec)
if not agent:
# Agent is not in one of the chosen "selections"
#
continue
agent.placement = [float(token) for token in tokens[2:18]]
for i in range(20, len(tokens), 2):
agent.variableValues[tokens[i]] = float(tokens[i + 1])
finally:
if fileHandle != callsheet:
fileHandle.close()
except:
print >>sys.stderr, "Error reading callsheet."
raise
def read(callsheet, sim): '''Load the simmed values of agent variables''' if isinstance(callsheet, basestring): fileHandle = open(callsheet, "r") else: fileHandle = callsheet try: try: for line in fileHandle: tokens = line.strip().split() if tokens: # 0 : id # 1 : name # 2-17: placement matrix # 18 : keyword "cdl" # 19 : cdl file path # 20-?: variable names and values id = int(tokens[0]) agentName = Agent.formatAgentName(tokens[1]) agentSpec = sim.scene.agentSpec(sim.scene.resolvePath(tokens[19])) agent = sim.agent(agentName, id, agentSpec) if not agent: # Agent is not in one of the chosen "selections" # continue agent.placement = [ float(token) for token in tokens[2:18] ] for i in range(20, len(tokens), 2 ): agent.variableValues[tokens[i]] = float(tokens[i+1]) finally: if fileHandle != callsheet: fileHandle.close() except: print >> sys.stderr, "Error reading callsheet." raise
def compute(self, plug, dataBlock):
'''Calculate the output translate and rotate values. If any value is
queried, all of them are updated and cleaned.'''
try:
if ( plug == MsvSimLoader.aOutput or
plug == MsvSimLoader.aTranslate or
plug == MsvSimLoader.aRotate ):
#==============================================================
# Query Input Attributes
#==============================================================
simDir = self._asString( dataBlock.inputValue( MsvSimLoader.aSimDir ) )
if not simDir or not os.path.isdir(simDir):
raise npy.Errors.BadArgumentError("Please set the simDir attribute to a valid directory path.")
agentType = self._asString( dataBlock.inputValue( MsvSimLoader.aAgentType ) )
if not agentType:
raise npy.Errors.BadArgumentError("Please set the agentType attribute.")
instance = dataBlock.inputValue( MsvSimLoader.aInstance ).asInt()
frame = int(dataBlock.inputValue( MsvSimLoader.aTime ).asTime().value())
simType = self._asString( dataBlock.inputValue( MsvSimLoader.aSimType ) )
simType = ".%s" % simType
#==============================================================
# Query the cached sim data, or read it from disk if no cache
# exists
#==============================================================
try:
sim = MsvSimLoader.sims[simDir]
except:
sim = SimData.SimData()
SimReader.read( simDir, simType, sim )
MsvSimLoader.sims[simDir] = sim
#==============================================================
# Get the sim data for the target agent
#==============================================================
agentName = Agent.formatAgentName( agentType, instance )
agentSim = sim.agent(agentName)
if agentSim:
#==========================================================
# Iterate over the input joints and query their translate
# and rotate values from the sim data. Add the
# corresponding offset and stuff the result in 'output'.
#==========================================================
haJoints = dataBlock.inputArrayValue( MsvSimLoader.aJoints )
haOffsets = dataBlock.inputArrayValue( MsvSimLoader.aOffsets )
haOutput = dataBlock.outputArrayValue( MsvSimLoader.aOutput )
for i in range(haJoints.elementCount()):
# Assume that both the 'joints' and 'output' multis
# have the same number of elements and are not sparse
haJoints.jumpToArrayElement(i)
jointName = self._asString( haJoints.inputValue() )
jointSim = agentSim.joint(jointName)
# 'output' is a multi-compound with one entry for every
# entry in the 'joints' input multi. The compound
# children are also multis, with one element for every
# channel in the corresponding joint's sim data. By
# iterating over the elements in the 'translate' and
# 'rotate' compound children we know how many samples
# to query from the joint sim.
#
# 'offsets' is structured the same as 'output' and
# stores the translate and rotate offsets for a given
# channel.
sampleIndex = 0
haOffsets.jumpToArrayElement(i)
haOutput.jumpToArrayElement(i)
# The joint sim stores sample data as an array and
# without distinguishing between translate and rotate
# samples. 'sampleIndex' will keep track of where we
# are in that array.
haTranslateOffset = MArrayDataHandle( haOffsets.inputValue().child( MsvSimLoader.aTranslateOffset ) )
haTranslate = MArrayDataHandle( haOutput.outputValue().child( MsvSimLoader.aTranslate ) )
for j in range(haTranslate.elementCount()):
# Get the sample data
sample = jointSim.sampleByIndex( sampleIndex, frame )
# Add the offset
haTranslateOffset.jumpToArrayElement(j)
sample += haTranslateOffset.inputValue().asDistance().value()
# Stuff it in the output
distance = MDistance( sample )
haTranslate.jumpToArrayElement(j)
haTranslate.outputValue().setMDistance( distance )
# Increment sampleIndex so we know what sample to
# query from the jointSim
sampleIndex += 1
haTranslate.setAllClean()
haRotateOffset = MArrayDataHandle( haOffsets.inputValue().child( MsvSimLoader.aRotateOffset ) )
haRotate = MArrayDataHandle( haOutput.outputValue().child( MsvSimLoader.aRotate ) )
for j in range(haRotate.elementCount()):
# Get the sample data
sample = jointSim.sampleByIndex( sampleIndex, frame )
# Add the offset
haRotateOffset.jumpToArrayElement(j)
sample += haRotateOffset.inputValue().asAngle().value()
# Stuff it in the output
angle = MAngle( sample, MAngle.kDegrees )
haRotate.jumpToArrayElement(j)
haRotate.outputValue().setMAngle( angle )
# Increment sampleIndex so we know what sample to
# query from the jointSim
sampleIndex += 1
haRotate.setAllClean()
haOutput.setAllClean()
return MStatus.kSuccess
except ns.py.Errors.Error, e:
MGlobal.displayError("Error: %s" % e)
return MStatus.kFailure
