def test_fe_6548(core_client):
from FabricEngine.Canvas.RTValEncoderDecoder import RTValEncoderDecoder
core_client.loadExtension('FileIO')
encdec = RTValEncoderDecoder(core_client)
filepath = core_client.RT.types.FilePath.create(
os.path.join('home', 'user', 'test.txt'))
print 'filepath:' + str(filepath)
filepath_str = encdec.getAsString(filepath)
print 'filepath_str:' + filepath_str
dec_filepath = encdec.getFromString('FilePath', filepath_str)
print 'dec_filepath:' + str(dec_filepath)
assert str(filepath) == str(dec_filepath)
vec3 = core_client.RT.types.Vec3(1, 2, 3)
print 'vec3:' + str(vec3)
vec3_str = encdec.getAsString(vec3)
print 'vec3_str:' + vec3_str
dec_vec3 = encdec.getFromString('Vec3', vec3_str)
print 'dec_vec3:' + str(dec_vec3)
assert str(vec3) == str(dec_vec3)
class BindingWrapper:
def __init__(self, client, binding, qUndoStack):
self.client = client
self.binding = binding
self.qUndoStack = qUndoStack
self.rtvalEncoderDecoder = RTValEncoderDecoder(self.client)
@staticmethod
def splitInts(packedIndices):
if not packedIndices:
return []
return map(
lambda indexStr: int(indexStr),
packedIndices.split('|')
)
@staticmethod
def splitNames(names):
if not names:
return []
return names.split('|')
@staticmethod
def splitPoss(posXs, posYs):
if not posXs:
return []
if isinstance(posXs, basestring):
posXs = map(
lambda str: float(str),
posXs.split('|')
)
elif not hasattr(posXs, '__iter__'):
posXs = [posXs]
if isinstance(posYs, basestring):
posYs = map(
lambda str: float(str),
posYs.split('|')
)
elif not hasattr(posYs, '__iter__'):
posYs = [posYs]
result = []
for i in range(0, len(posXs)):
result.append(QtCore.QPointF(posXs[i], posYs[i]))
return result
def decodePortType(self, portType):
portTypeLower = portType.lower()
if portTypeLower == "io":
return self.client.DFG.PortTypes.IO
elif portTypeLower == "out":
return self.client.DFG.PortTypes.Out
else:
return self.client.DFG.PortTypes.In
def undo(
self,
):
self.qUndoStack.undo()
def redo(
self,
):
self.qUndoStack.redo()
def instPreset(
self,
execPath,
presetPath,
posX,
posY
):
rootExec = self.binding.getExec()
exec_ = rootExec.getSubExec(execPath)
cmd = DFG.DFGUICmd_InstPreset(
self.binding,
execPath,
exec_,
presetPath,
QtCore.QPointF(posX, posY)
)
InvokeCmd(cmd, self.qUndoStack)
return cmd.getActualNodeName()
def addGraph(
self,
execPath,
title,
posX,
posY
):
rootExec = self.binding.getExec()
exec_ = rootExec.getSubExec(execPath)
cmd = DFG.DFGUICmd_AddGraph(
self.binding,
execPath,
exec_,
title,
QtCore.QPointF(posX, posY)
)
InvokeCmd(cmd, self.qUndoStack)
return cmd.getActualNodeName()
def addFunc(
self,
execPath,
title,
initialCode,
posX,
posY
):
rootExec = self.binding.getExec()
exec_ = rootExec.getSubExec(execPath)
cmd = DFG.DFGUICmd_AddFunc(
self.binding,
execPath,
exec_,
title,
initialCode,
QtCore.QPointF(posX, posY)
)
InvokeCmd(cmd, self.qUndoStack)
return cmd.getActualNodeName()
def addBackDrop(
self,
execPath,
title,
posX,
posY,
):
rootExec = self.binding.getExec()
exec_ = rootExec.getSubExec(execPath)
cmd = DFG.DFGUICmd_AddBackDrop(
self.binding,
execPath,
exec_,
title,
QtCore.QPointF(posX, posY)
)
InvokeCmd(cmd, self.qUndoStack)
return cmd.getActualNodeName()
def addVar(
self,
execPath,
desiredNodeName,
dataType,
extDep,
posX,
posY
):
rootExec = self.binding.getExec()
exec_ = rootExec.getSubExec(execPath)
cmd = DFG.DFGUICmd_AddVar(
self.binding,
execPath,
exec_,
desiredNodeName,
dataType,
extDep,
QtCore.QPointF(posX, posY)
)
InvokeCmd(cmd, self.qUndoStack)
return cmd.getActualNodeName()
def addGet(
self,
execPath,
desiredNodeName,
varPath,
posX,
posY
):
rootExec = self.binding.getExec()
exec_ = rootExec.getSubExec(execPath)
cmd = DFG.DFGUICmd_AddGet(
self.binding,
execPath,
exec_,
desiredNodeName,
varPath,
QtCore.QPointF(posX, posY)
)
InvokeCmd(cmd, self.qUndoStack)
return cmd.getActualNodeName()
def addSet(
self,
execPath,
desiredNodeName,
varPath,
posX,
posY
):
rootExec = self.binding.getExec()
exec_ = rootExec.getSubExec(execPath)
cmd = DFG.DFGUICmd_AddSet(
self.binding,
execPath,
exec_,
desiredNodeName,
varPath,
QtCore.QPointF(posX, posY)
)
InvokeCmd(cmd, self.qUndoStack)
return cmd.getActualNodeName()
def moveNodes(
self,
execPath,
packedNodeNames,
packedNewTopLeftPosXs,
packedNewTopLeftPosYs
):
rootExec = self.binding.getExec()
exec_ = rootExec.getSubExec(execPath)
nodeNames = BindingWrapper.splitNames(packedNodeNames)
newTopLeftPoss = BindingWrapper.splitPoss(
packedNewTopLeftPosXs,
packedNewTopLeftPosYs
)
cmd = DFG.DFGUICmd_MoveNodes(
self.binding,
execPath,
exec_,
nodeNames,
newTopLeftPoss
)
InvokeCmd(cmd, self.qUndoStack)
def resizeBackDrop(
self,
execPath,
backDropNodeName,
newTopLeftPosX,
newTopLeftPosY,
newSizeW,
newSizeH,
):
rootExec = self.binding.getExec()
exec_ = rootExec.getSubExec(execPath)
cmd = DFG.DFGUICmd_ResizeBackDrop(
self.binding,
execPath,
exec_,
backDropNodeName,
QtCore.QPointF(newTopLeftPosX, newTopLeftPosY),
QtCore.QSizeF(newSizeW, newSizeH),
)
InvokeCmd(cmd, self.qUndoStack)
def removeNodes(
self,
execPath,
packedNodeNames,
):
rootExec = self.binding.getExec()
exec_ = rootExec.getSubExec(execPath)
nodeNames = BindingWrapper.splitNames(packedNodeNames)
cmd = DFG.DFGUICmd_RemoveNodes(
self.binding,
execPath,
exec_,
nodeNames,
)
InvokeCmd(cmd, self.qUndoStack)
def implodeNodes(
self,
execPath,
packedNodeNames,
desiredNodeName
):
rootExec = self.binding.getExec()
exec_ = rootExec.getSubExec(execPath)
nodeNames = BindingWrapper.splitNames(packedNodeNames)
cmd = DFG.DFGUICmd_ImplodeNodes(
self.binding,
execPath,
exec_,
nodeNames,
desiredNodeName
)
InvokeCmd(cmd, self.qUndoStack)
return cmd.getActualImplodedNodeName()
def explodeNode(
self,
execPath,
nodeName
):
rootExec = self.binding.getExec()
exec_ = rootExec.getSubExec(execPath)
cmd = DFG.DFGUICmd_ExplodeNode(
self.binding,
execPath,
exec_,
nodeName
)
InvokeCmd(cmd, self.qUndoStack)
return cmd.getExplodedNodeNames()
def connect(
self,
execPath,
packedSrcPorts,
packedDstPorts,
):
rootExec = self.binding.getExec()
exec_ = rootExec.getSubExec(execPath)
srcPorts = BindingWrapper.splitNames(packedSrcPorts)
dstPorts = BindingWrapper.splitNames(packedDstPorts)
cmd = DFG.DFGUICmd_Connect(
self.binding,
execPath,
exec_,
srcPorts,
dstPorts
)
InvokeCmd(cmd, self.qUndoStack)
def disconnect(
self,
execPath,
packedSrcPorts,
packedDstPorts,
):
rootExec = self.binding.getExec()
exec_ = rootExec.getSubExec(execPath)
srcPorts = BindingWrapper.splitNames(packedSrcPorts)
dstPorts = BindingWrapper.splitNames(packedDstPorts)
cmd = DFG.DFGUICmd_Disconnect(
self.binding,
execPath,
exec_,
srcPorts,
dstPorts
)
InvokeCmd(cmd, self.qUndoStack)
def addPort(
self,
execPath,
desiredPortName,
portType,
typeSpec,
pathToConnect,
extDep,
metaData,
):
rootExec = self.binding.getExec()
exec_ = rootExec.getSubExec(execPath)
cmd = DFG.DFGUICmd_AddPort(
self.binding,
execPath,
exec_,
desiredPortName,
self.decodePortType(portType),
typeSpec,
pathToConnect,
extDep,
metaData,
)
InvokeCmd(cmd, self.qUndoStack)
return cmd.getActualPortName()
def addInstPort(
self,
execPath,
instName,
desiredPortName,
portType,
typeSpec,
pathToConnect,
connectType,
extDep,
metaData,
):
rootExec = self.binding.getExec()
exec_ = rootExec.getSubExec(execPath)
cmd = DFG.DFGUICmd_AddInstPort(
self.binding,
execPath,
exec_,
instName,
desiredPortName,
self.decodePortType(portType),
typeSpec,
pathToConnect,
self.decodePortType(connectType),
extDep,
metaData,
)
InvokeCmd(cmd, self.qUndoStack)
return cmd.getActualPortName()
def addInstBlockPort(
self,
execPath,
instName,
blockName,
desiredPortName,
typeSpec,
pathToConnect,
extDep,
metaData,
):
rootExec = self.binding.getExec()
exec_ = rootExec.getSubExec(execPath)
cmd = DFG.DFGUICmd_AddInstBlockPort(
self.binding,
execPath,
exec_,
instName,
blockName,
desiredPortName,
typeSpec,
pathToConnect,
extDep,
metaData,
)
InvokeCmd(cmd, self.qUndoStack)
return cmd.getActualPortName()
def editPort(
self,
execPath,
oldPortName,
desiredNewPortName,
portType,
typeSpec,
extDep,
metaData,
):
rootExec = self.binding.getExec()
exec_ = rootExec.getSubExec(execPath)
cmd = DFG.DFGUICmd_EditPort(
self.binding,
execPath,
exec_,
oldPortName,
desiredNewPortName,
self.decodePortType(portType),
typeSpec,
extDep,
metaData,
)
InvokeCmd(cmd, self.qUndoStack)
return cmd.getActualNewPortName()
def removePort(
self,
execPath,
packedPortNames,
):
rootExec = self.binding.getExec()
exec_ = rootExec.getSubExec(execPath)
portNames = BindingWrapper.splitNames(packedPortNames)
cmd = DFG.DFGUICmd_RemovePort(
self.binding,
execPath,
exec_,
portNames,
)
InvokeCmd(cmd, self.qUndoStack)
def createPreset(
self,
execPath,
nodeName,
presetDirPath,
presetName,
):
rootExec = self.binding.getExec()
exec_ = rootExec.getSubExec(execPath)
cmd = DFG.DFGUICmd_CreatePreset(
self.binding,
execPath,
exec_,
nodeName,
presetDirPath,
presetName,
)
InvokeCmd(cmd, self.qUndoStack)
return cmd.getPathname()
def setNodeComment(
self,
execPath,
nodeName,
comment,
):
rootExec = self.binding.getExec()
exec_ = rootExec.getSubExec(execPath)
cmd = DFG.DFGUICmd_SetNodeComment(
self.binding,
execPath,
exec_,
nodeName,
comment,
)
InvokeCmd(cmd, self.qUndoStack)
def setCode(
self,
execPath,
code,
):
rootExec = self.binding.getExec()
exec_ = rootExec.getSubExec(execPath)
cmd = DFG.DFGUICmd_SetCode(
self.binding,
execPath,
exec_,
code,
)
InvokeCmd(cmd, self.qUndoStack)
def editNode(
self,
execPath,
oldNodeName,
desiredNewNodeName,
nodeMetadata,
execMetadata,
):
rootExec = self.binding.getExec()
exec_ = rootExec.getSubExec(execPath)
cmd = DFG.DFGUICmd_EditNode(
self.binding,
execPath,
exec_,
oldNodeName,
desiredNewNodeName,
nodeMetadata,
execMetadata,
)
InvokeCmd(cmd, self.qUndoStack)
return cmd.getActualNewNodeName()
def renamePort(
self,
execPath,
portPath,
desiredNewPortName,
):
rootExec = self.binding.getExec()
exec_ = rootExec.getSubExec(execPath)
cmd = DFG.DFGUICmd_RenamePort(
self.binding,
execPath,
exec_,
portPath,
desiredNewPortName,
)
InvokeCmd(cmd, self.qUndoStack)
return cmd.getActualNewPortName()
def paste(
self,
execPath,
json,
cursorPosX,
cursorPosY,
):
rootExec = self.binding.getExec()
exec_ = rootExec.getSubExec(execPath)
cmd = DFG.DFGUICmd_Paste(
self.binding,
execPath,
exec_,
json,
QtCore.QPointF(cursorPosX, cursorPosY),
)
InvokeCmd(cmd, self.qUndoStack)
return cmd.getPastedItemNames()
def setArgValue(
self,
argName,
typeName,
valueJSON,
):
value = self.rtvalEncoderDecoder.getFromString(typeName, valueJSON)
cmd = DFG.DFGUICmd_SetArgValue(
self.binding,
argName,
value,
)
InvokeCmd(cmd, self.qUndoStack)
def setPortDefaultValue(
self,
execPath,
portPath,
typeName,
valueJSON,
):
rootExec = self.binding.getExec()
exec_ = rootExec.getSubExec(execPath)
value = self.rtvalEncoderDecoder.getFromString(typeName, valueJSON)
cmd = DFG.DFGUICmd_SetPortDefaultValue(
self.binding,
execPath,
exec_,
portPath,
value,
)
InvokeCmd(cmd, self.qUndoStack)
def setRefVarPath(
self,
execPath,
refName,
varPath,
):
rootExec = self.binding.getExec()
exec_ = rootExec.getSubExec(execPath)
cmd = DFG.DFGUICmd_SetRefVarPath(
self.binding,
execPath,
exec_,
refName,
varPath,
)
InvokeCmd(cmd, self.qUndoStack)
def reorderPorts(
self,
execPath,
itemPath,
packedIndices,
):
rootExec = self.binding.getExec()
exec_ = rootExec.getSubExec(execPath)
indices = BindingWrapper.splitInts(packedIndices)
cmd = DFG.DFGUICmd_ReorderPorts(
self.binding,
execPath,
exec_,
itemPath,
indices,
)
InvokeCmd(cmd, self.qUndoStack)
def dismissLoadDiags(
self,
packedIndices,
):
rootExec = self.binding.getExec()
indices = BindingWrapper.splitInts(packedIndices)
cmd = DFG.DFGUICmd_DismissLoadDiags(
self.binding,
indices,
)
InvokeCmd(cmd, self.qUndoStack)
def setExtDeps(
self,
execPath,
packedExtDeps,
):
rootExec = self.binding.getExec()
exec_ = rootExec.getSubExec(execPath)
extDeps = BindingWrapper.splitNames(packedExtDeps)
cmd = DFG.DFGUICmd_SetExtDeps(
self.binding,
execPath,
exec_,
extDeps,
)
InvokeCmd(cmd, self.qUndoStack)
def splitFromPreset(
self,
execPath,
):
rootExec = self.binding.getExec()
exec_ = rootExec.getSubExec(execPath)
cmd = DFG.DFGUICmd_SplitFromPreset(
self.binding,
execPath,
exec_,
)
InvokeCmd(cmd, self.qUndoStack)
def addBlock(
self,
execPath,
desiredName,
posX,
posY,
):
rootExec = self.binding.getExec()
exec_ = rootExec.getSubExec(execPath)
cmd = DFG.DFGUICmd_AddBlock(
self.binding,
execPath,
exec_,
desiredName,
QtCore.QPointF(posX, posY)
)
InvokeCmd(cmd, self.qUndoStack)
return cmd.getActualName()
def addBlockPort(
self,
execPath,
blockName,
desiredPortName,
portType,
typeSpec,
pathToConnect,
connectType,
extDep,
metaData,
):
rootExec = self.binding.getExec()
exec_ = rootExec.getSubExec(execPath)
cmd = DFG.DFGUICmd_AddBlockPort(
self.binding,
execPath,
exec_,
blockName,
desiredPortName,
self.decodePortType(portType),
typeSpec,
pathToConnect,
self.decodePortType(connectType),
extDep,
metaData,
)
InvokeCmd(cmd, self.qUndoStack)
return cmd.getActualPortName()