ブログ - 1月, 2018

import AtomsMaya
import tempfile

# parameters for the script
node = "tcAgentGroupNode1"
rig_scene = "/path/to/rig"
start = 1
end = 50

# find the agent ids, if no agent is selected we use all the agents
ids = cmds.getAttr("%s.agentSelection" % node)
if not ids:
    ids = list(range(len(AtomsMaya.getAgentGroup(node).agents())))

# exporting the fbxs
_, temp_fbx = tempfile.mkstemp('.fbx')
for i in ids:
    cmds.tcAtomsFbxExporter(fileName=temp_fbx, agentGroup=node, agentId=ids, frameStart=start, frameEnd=end)

# retargeting the fbxs on the animation rig
ma_files = []
for i in ids:
    _, temp_ma = tempfile.mkstemp('.ma')
	# the fbx exporter create fbx files with 4 digit padding
    file_path = temp_fbx.replace(".fbx", "%04d.fbx" % i)
    cmds.file(rig_scene, o=True, f=True)
    cmds.tcAnimationRigRetarget(fileName=file_path)
	# saving the file to a different temp location
    cmds.file(rn=temp_ma)
    cmds.file(save=True)
    ma_files.append(temp_ma)
    
# importing all the retargeted characters in a brand new scene
cmds.file(new=True, f=True)
for m in ma_files:
    cmds.file(m, i=True)
    
    

import Atoms
import AtomsCore
import AtomsMath
import os
import xbox
import maya


class XBoxControllerModule(Atoms.BehaviourModule):
    def __init__(self):
        Atoms.BehaviourModule.__init__(self)

		# Adding the module metadatas
        self.addAttribute("cachedAngle", AtomsCore.Vector3Metadata(AtomsMath.V3d(0.0,0.0,0.0)))
        self.addAttribute("cachedState", AtomsCore.IntMetadata(0))
        self.addAttribute("record", AtomsCore.BoolMetadata(False))
        self.addAttribute("turnAngle", AtomsCore.DoubleMetadata(3.0))

		# initializing the class members
        self.state = 0
        self.old_state = 0
        self.old_dir = [0.0,0.0,0.0]
		# lists for caching the values
        self.stateKeys=[]
        self.angleKeys=[]

    def initSimulation(self, agentGroup):
        # Emptying the two lists and setting initial values
        self.stateKeys=[]
        self.angleKeys=[]
        
        self.angleKeys.append([maya.cmds.currentTime(q=1), [0.0,0.0,0.0]])
        self.stateKeys.append([maya.cmds.currentTime(q=1), 0])

    def initFrame(self, agents, agentGroup):
		# getting the module parameters
        turnAngle = self.attributes()["turnAngle"].value()
        record = self.attributes()["record"].value()

		# getting the xbox control state
        state = xbox.getControlState()
        angle = state["LX"]
        buttonA = state["A"]
        buttonB = state["B"]
        buttonX = state["X"]
        
		# finding the right state depending on the pressed button
        if buttonA:
            self.state = 0
        if buttonB:
            self.state = 1
        if buttonX:
            self.state = 2
        
        if record:
			# if we are recording we cache the values
            if self.old_dir[0] != state["LX"] or self.old_dir[1] != state["LY"]:
                self.angleKeys.append([maya.cmds.currentTime(q=1), [state["LX"],state["LY"],0.0]])
            if self.state != self.old_state:
                self.stateKeys.append([maya.cmds.currentTime(q=1), self.state])
            self.old_dir = [state["LX"],state["LY"],0.0]
            self.old_state = self.state
        else:
			# otherwise we get the default values from the module parameters
            angle = self.attributes()["cachedAngle"].value().x
            self.state = self.attributes()["cachedState"].value()
            
		# for each agent changing the state and direction
        for agent in agents:
            if self.state:
                agent.metadata()["state"].set(self.state)
            dir = agent.metadata()["direction"].value()
            q = AtomsMath.Quatd()
            q.setAxisAngle(AtomsMath.V3d(0.0,1.0,0.0), -angle * turnAngle * 3.1456 / 180.0)
            dir = dir * q.toMatrix44()
            agent.metadata()["direction"].set(dir)

    def endSimulation(self, agents, agentGroup):
        # setting all the keyframes		
        for v in self.angleKeys:
            maya.cmds.setKeyframe(agentGroup.name() + ".atoms_XBoxControllerModule_cachedAngle.atoms_XBoxControllerModule_cachedAngleX",v=v[1][0],t=[v[0]])
            maya.cmds.setKeyframe(agentGroup.name() + ".atoms_XBoxControllerModule_cachedAngle.atoms_XBoxControllerModule_cachedAngleY",v=v[1][1],t=[v[0]])
            maya.cmds.setKeyframe(agentGroup.name() + ".atoms_XBoxControllerModule_cachedAngle.atoms_XBoxControllerModule_cachedAngleZ",v=v[1][02],t=[v[0]])
        for v in self.stateKeys:
            maya.cmds.setKeyframe(agentGroup.name() + ".atoms_XBoxControllerModule_cachedState",v=v[1],t=[v[0]])
            

import AtomsMaya
import maya


from PySide2 import QtWidgets, QtCore, QtGui

class XboxControllerModuleWidget(QtWidgets.QWidget):
    def __init__(self, parent=None):
        super(XboxControllerModuleWidget, self).__init__(parent)
        self.ag_name = maya.cmds.ls(sl=1,dag=1,lf=1,shapes=1)[0]
        self.v_layout = QtWidgets.QVBoxLayout()
        self.start = QtWidgets.QPushButton("Start")
        self.stop = QtWidgets.QPushButton("Stop")
        self.v_layout.addWidget(self.start)
        self.v_layout.addWidget(self.stop)        
        self.setLayout(self.v_layout)
        
        self.start.clicked.connect(self._start)
        self.stop.clicked.connect(self._stop)
        self.start.setEnabled(True)
        self.stop.setEnabled(False)
        
    def _start(self):
        ag_name = self.ag_name
        maya.cmds.setAttr(ag_name + ".atoms_XBoxControllerModule_record",1)
        maya.mel.eval("playButtonStart;")
        maya.mel.eval("playButtonForward;")
        self.start.setEnabled(False)
        self.stop.setEnabled(True)
        
    def _stop(self):
        ag_name = self.ag_name
        ag = AtomsMaya.getAgentGroup(ag_name)
        mod = ag.behaviourModule("XBoxControllerModule")
        mod.endSimulation(ag.agents(), ag)
        maya.cmds.setAttr(ag_name + ".atoms_XBoxControllerModule_record",0)
        maya.mel.eval("playButtonForward;")
        self.start.setEnabled(True)
        self.stop.setEnabled(False)
        
w = XboxControllerModuleWidget()
w.show()