def CorrectShape(obj, cls, frame): prim = obj.ActivePrimitive tree = prim.ICETrees.Find("CorrectiveShape") exist = False basePos = None secondPos = None if tree: if cls.Properties("Frame"+str(frame)): exist = True toolkit = Dispatch("XSI.UIToolkit") buttonPressed = toolkit.MsgBox( "A corrective shape already exists for this frame\n Would you replace it??", constants.siMsgOkCancel, "Corrective Shape") if buttonPressed == constants.siMsgCancel: XSI.LogMessage("Store Secondary Shape cancelled by You, F**k TYou!!", constants.siInfo) return compound = tree.CompoundNodes("CorrectiveShape") if not exist: compound.InputPorts("Mute").Value = True XSI.DeactivateAbove(tree, True) basePoints = prim.Geometry.Points basePos = basePoints.PositionArray XSI.DeactivateAbove(tree, False) secondPoints = prim.Geometry.Points secondPos = secondPoints.PositionArray if not exist: compound.InputPorts("Mute").Value = False else: op = prim.ConstructionHistory.Find("AnimationMarker") XSI.DeactivateAbove(op, True) basePoints = prim.Geometry.Points basePos = basePoints.PositionArray XSI.DeactivateAbove(op, False) secondPoints = prim.Geometry.Points secondPos = secondPoints.PositionArray l = len(basePos[0]) delta = [0] * 3 * l for a in range(l): delta[a*3] = secondPos[0][a]-basePos[0][a] delta[a*3+1] = secondPos[1][a]-basePos[1][a] delta[a*3+2] = secondPos[2][a]-basePos[2][a] if not exist: shapeNode = Dispatch(cls.AddProperty("SimpleDeformShape", 0, "Frame"+str(frame))) shapeNode.Elements.Array = delta else: shapeNode = Dispatch(cls.Properties("Frame"+str(frame))) shapeNode.Elements.Array = delta if not tree: tree = ICETree.CreateIceTree(obj, "CorrectiveShape", 2) t = str(tree) set = XSI.AddICENode("SetOneDataNode", t) set.Parameters("Reference").Value = "Self.PointPosition" ifnode = XSI.AddIceNode("IfNode", t) XSI.ConnectICENodes(t + ".port1", str(ifnode) + ".Result") XSI.ConnectICENodes(str(ifnode) + ".IfFalse", str(set) + ".Value") get = XSI.AddICENode("GetDataNode", t) get.Parameters("Reference").Value = "Self.PointPosition" add = XSI.AddIceNode("AddNode", t) XSI.ConnectIceNodes(str(add) + ".Value1", str(get) + ".Value") XSI.ConnectIceNodes(str(set) + ".Source", str(add) + ".Result") compound = XSI.CreateICECompoundNode(str(get) + "," + str(set) + "," + str(add) + "," + str(ifnode), None) XSI.EditICECompoundProperties(str(compound), "CorrectiveShape", "", "", "", "", "", 1, 0, "", 4, 6732954) XSI.AddExposedParamToICECompoundNode(str(ifnode) + ".Condition", str(compound), None, "Mute") if not exist: compound = Dispatch(tree.Nodes("CorrectiveShape")) add = compound.Nodes.Filter("AddNode")(0) mult = XSI.AddIceNode("MultiplyByScalarNode", str(compound)) addInputs = add.InputPorts checkPorts = -1 p = 0 for i in addInputs: if not i.IsConnected: checkPorts = p break p += 1 if not checkPorts == -1: XSI.ConnectIceNodes(str(add) + ".Value" + str(checkPorts + 1), str(mult) + ".Result") else: index = add.InputPorts.Count XSI.AddPortToICENode(str(add) + ".Value" + str(index), constants.siNodePortDataInsertionLocationAfter) XSI.ConnectICENodes (str(add) + ".Value" + str(index + 1), str(mult) + ".Result") shapeStr = "Self.cls.CorrectiveShapeCls."+shapeNode.Name+".positions" getShape = XSI.AddIceNode("GetDataNode", str(tree)) getShape.Parameters("Reference").Value = shapeStr inputs = compound.InputPorts count = 0 for i in inputs: if i.Name.find("Position") > -1: count += 1 XSI.AddExposedParamToICECompoundNode(str(mult) + ".value", str(compound), None, "Position") XSI.EditExposedParamInICECompoundNode(str(compound) + ".Position", "Position" + str(frame), "0", "0", None, None, 0, "") XSI.AddExposedParamToICECompoundNode(str(mult) + ".Factor", str(compound), None, "Factor") XSI.EditExposedParamInICECompoundNode(str(compound) + ".Factor", "Frame" + str(frame), "0", "1", None, None, 0, "") XSI.ConnectICENodes(str(compound) + ".Position" + str(frame), str(getShape) + ".value") XSI.InspectObj(compound, None, None, constants.siLock) # delete all nodes between secondary shape modeling and animation markers for h in prim.ConstructionHistory: if h.FullName.find('marker') == -1: XSI.DeleteObj(h) if h.Name == 'Animation': break
def CorrectPush(obj, wmap, frame, exist): # check if tree exists prim = obj.ActivePrimitive tree = prim.ICETrees.Find("CorrectivePush") if not tree: tree = ICETree.CreateIceTree(obj, "CorrectivePush", 2) t = str(tree) set1 = XSI.AddICENode("SetOneDataNode", t) set1.Parameters("Reference").Value = "Self.PointPosition" ifnode = XSI.AddIceNode("IfNode", t) XSI.ConnectICENodes(t + ".port1", str(ifnode) + ".Result") XSI.ConnectICENodes(str(ifnode) + ".IfFalse", str(set1) + ".Value") get = XSI.AddICENode("GetDataNode", t) get.Parameters("Reference").Value = "Self.PointPosition" get1 = XSI.AddICENode("GetDataNode", t) get1.Parameters("Reference").Value = "Self.PointNormal" add = XSI.AddICENode("AddNode", t) XSI.ConnectICENodes(str(add) + ".Value1", str(get) + ".Value") add1 = XSI.AddIceNode("AddNode", t) mult = XSI.AddIceNode("MultiplyByScalarNode", t) mult1 = XSI.AddIceNode("MultiplyByScalarNode", t) XSI.ConnectICENodes(str(add) + ".Value2", str(mult) + ".Result") XSI.ConnectICENodes(str(mult) + ".Value", str(get1) + ".Value") XSI.ConnectICENodes(str(mult) + ".Factor", str(mult1) + ".Result") XSI.ConnectICENodes(str(mult1) + ".Value", str(add1) + ".Result") XSI.ConnectICENodes(str(set1) + ".Source", str(add) + ".Result") nodes = [get, get1, ifnode, add, add1, set1] compound = XSI.CreateICECompoundNode(",".join(nodes), None) XSI.EditICECompoundProperties(compound, "CorrectivePush", "", "", "", "", "", 1, 0, "", 4, 6732954) XSI.AddExposedParamToICECompoundNode(str(ifnode) + ".Condition", str(compound), None, "Mute") XSI.AddExposedParamToICECompoundNode(str(mult1) + ".Factor", str(compound), None, "Factor") if not exist: compound = tree.CompoundNodes("CorrectivePush") add = compound.Nodes.Filter("AddNode")(0) mult = XSI.AddIceNode("MultiplyByScalarNode", compound) addInputs = add.InputPorts checkPorts = -1 p = 0 for i in addInputs: if not i.IsConnected: checkPorts = p break p += 1 if not checkPorts == -1: XSI.ConnectIceNodes(str(add) + ".Value" + str(checkPorts + 1), str(mult) + ".Result") else: index = add.InputPorts.Count XSI.AddPortToICENode(str(add) + ".value" + str(index), constants.siNodePortDataInsertionLocationAfter) XSI.ConnectICENodes (str(add) + ".value" + str(index + 1), str(mult) + ".result") pushStr = "Self.cls.CorrectivePushCls."+wmap.Name+".Weights" getPushMap = XSI.AddIceNode("GetDataNode", str(tree)) getPushMap.Parameters("Reference").Value = pushStr inputs = compound.InputPorts count = 0 for i in inputs: if i.Name.find("Position")>-1: count += 1 XSI.AddExposedParamToICECompoundNode(str(mult) + ".value", str(compound), None, "Push") XSI.EditExposedParamInICECompoundNode(str(compound) + ".Push", "Push" + str(frame), "0", "0", None, None, 0, "") XSI.AddExposedParamToICECompoundNode(str(mult) + ".Factor", str(compound), None, "Frame" + str(frame)) XSI.EditExposedParamInICECompoundNode(str(compound) + ".Frame" + str(frame), "Frame" + str(frame), "0", "1", None, None, 0, "") XSI.ConnectICENodes(str(compound) + ".Push" + str(frame), str(getPushMap) + ".Value") XSI.InspectObj(compound, None, None, constants.siLock) XSI.SelectObj(wmap) XSI.PaintTool()
def CorrectSmooth(obj, wmap, frame, exist): # check if tree exists prim = obj.ActivePrimitive tree = prim.ICETrees.Find( "CorrectiveSmooth" ) if not tree: above = prim.ICETrees.Find("CorrectivePush") if not above: above = prim.ICETrees.Find("CorrectiveShape") if above: XSI.DeactivateAbove(above, True) tree = ICETree.CreateIceTree(obj, "CorrectiveSmooth", 2) if above: XSI.DeactivateAbove(above, False) t = str(tree) set1 = XSI.AddICENode("SetOneDataNode", t) set1.Parameters("Reference").Value = "Self.PointPosition" ifnode = XSI.AddIceNode("IfNode", t) repeat = XSI.AddICENode("RepeatNode", t) XSI.ConnectICENodes(str(repeat) + ".port", str(set1) + ".value") repeat.InputPorts("iterations").Value = 10 XSI.ConnectICENodes(t + ".port1", str(ifnode) + ".Result") XSI.ConnectICENodes(str(ifnode) + ".IfFalse", str(set1) + ".Value") XSI.ConnectICENodes(str(ifnode) + ".IfFalse", str(repeat) + ".Execute") get = XSI.AddICENode("GetDataNode", t) get.Parameters("Reference").Value = "Self.PointPosition" get1 = XSI.AddICENode("GetDataNode", t) get1.Parameters("Reference").Value = "Self.PointNeighbors" get2 = XSI.AddICENode("GetDataNode", t) get2.Parameters("Reference").Value = "PointPosition" XSI.ConnectICENodes(str(get2) + ".Source", str(get1) + ".Value") average = XSI.AddICENode("GetArrayAverageNode", t) XSI.ConnectICENodes(str(average) + ".Array", str(get2) + ".Value") interp = XSI.AddIceNode("LinearInterpolateNode", t) XSI.ConnectICENodes(str(interp) + ".First", str(get) + ".Value") XSI.ConnectICENodes(str(interp) + ".Second", str(average) + ".Result") XSI.ConnectIceNodes(str(set1) + ".Source", str(interp) + ".Result") clamp = XSI.AddICENode("ClampNode", t) XSI.ConnectICENodes(str(interp) + ".Blend", str(clamp) + ".Result") scalarZero = XSI.AddICENode("ScalarNode", t) scalarZero.InputPorts("Value").Value = 0 scalarOne = XSI.AddICENode("ScalarNode", t) scalarOne.InputPorts("Value").Value = 1 add = XSI.AddICENode("AddNode", t) XSI.ConnectICENodes(str(clamp) + ".Limit1", str(scalarZero) + ".Result") XSI.ConnectICENodes(str(clamp) + ".Limit2", str(scalarOne) + ".Result") XSI.ConnectICENodes(str(clamp) + ".Value", str(add) + ".Result") nodes = [get, set1, get1, get2, average, interp, repeat, ifnode, clamp, add] compound = XSI.CreateICECompoundNode(",".join(nodes), None) XSI.EditICECompoundProperties(compound, "CorrectiveSmooth", "", "", "", "", "", 1, 0, "", 4, 6732954) XSI.AddExposedParamToICECompoundNode(str(ifnode) + ".Condition", str(compound), None, "Mute") XSI.AddExposedParamToICECompoundNode(str(repeat) + ".Iterations", str(compound), None, "Repeat") if not exist: compound = tree.CompoundNodes("CorrectiveSmooth") add = compound.Nodes.Filter("AddNode")(0) mult = XSI.AddIceNode("MultiplyByScalarNode", str(compound)) addInputs = add.InputPorts checkPorts = -1 i = 0 for p in addInputs: if not p.IsConnected: checkPorts = i break if checkPorts > -1: XSI.ConnectIceNodes(str(add) + ".Value" + str(checkPorts + 1), str(mult) + ".Result") else: index = add.InputPorts.Count XSI.AddPortToICENode(str(add) + ".Value" + str(index), constants.siNodePortDataInsertionLocationAfter) XSI.ConnectICENodes(str(add) + ".Value" + str(index + 1), str(mult) + ".Result") t = str(tree) smoothStr = "Self.cls.CorrectiveSmoothCls." + wmap.Name + ".Weights" get = XSI.AddIceNode("GetDataNode", t) get.Parameters("Reference").Value = smoothStr inputs = compound.InputPorts count = 0 for i in inputs: if not i.Name.find("Smooth") == -1: count += 1 XSI.AddExposedParamToICECompoundNode(str(mult) + ".value", str(compound), None, "Smooth") XSI.EditExposedParamInICECompoundNode(str(compound) + ".Smooth", "Smooth" + str(frame), "0", "0", None, None, 0, "") XSI.AddExposedParamToICECompoundNode(str(mult) + ".Factor", str(compound), None, "Factor") XSI.EditExposedParamInICECompoundNode(str(compound) + ".Factor", "Frame" + str(frame), "0", "1", None, None, 0, "") XSI.ConnectICENodes(str(compound) + ".Smooth" + str(frame), str(get) + ".Value") XSI.InspectObj(compound, None, None, constants.siLock) XSI.SelectObj(wmap) XSI.PaintTool()