def init(self, view, filePath):
self._graph = Graph()
self._graphWrapper = GraphWrapper(self._graph, view)
self._buttlePath = filePath
for index in range(1, 10):
self._mapViewerIndextoNodeName[str(index)] = None
return self
def init(self, view, filePath):
self._graph = Graph()
self._graphWrapper = GraphWrapper(self._graph, view)
self._buttlePath = filePath
for index in range(1, 10):
self._mapViewerIndextoNodeName[str(index)] = None
return self
class ButtleData(QtCore.QObject):
"""
Class ButtleData defined by:
- _graphWrapper : the graphWrapper
- _graph : the graph (core)
- _currentViewerNodeName : the name of the node currently in the viewer
- _currentSelectedNodeNames : list of the names of the nodes currently selected
- _currentParamNodeName : the name of the node currently displayed in the paramEditor
- _currentConnectionId : the list of the id of the connections currently selected
- _currentCopiedNodesInfo : the list of buttle info for the current node(s) copied
- _mapNodeNameToComputedImage : this map makes the correspondance between a gloablHash and a computed image (max 20 images stored)
- _buttlePath : the path of the root directory (usefull to import images)
- _graphCanBeSaved : a boolean indicating if the graph had changed since the last saving (usefull for the display of the icon "save graph")
This class containts all data we need to manage the application.
"""
_graph = None
_graphWrapper = None
_graphCanBeSaved = False
# the current params
_currentParamNodeName = None
_currentSelectedNodeNames = []
# for the connections
_currentConnectionId = None
# to eventually save current nodes data
_currentCopiedNodesInfo = {}
# for the viewer
_currentViewerIndex = 1
_currentViewerFrame = 0
_currentViewerNodeName = None
_mapViewerIndextoNodeName = {}
_mapNodeNameToComputedImage = {}
# boolean used in viewerManager
_videoIsPlaying = False
# processGraph used in viewerManager to compute images
_processGraph = None
_timeRange = None
def init(self, view, filePath):
self._graph = Graph()
self._graphWrapper = GraphWrapper(self._graph, view)
self._buttlePath = filePath
for index in range(1, 10):
self._mapViewerIndextoNodeName[str(index)] = None
return self
################################################## GETTERS ET SETTERS ##################################################
#################### getters ####################
def getGraph(self):
return self._graph
def getGraphWrapper(self):
return self._graphWrapper
def getButtlePath(self):
return self._buttlePath
### current data ###
def getCurrentParamNodeName(self):
"""
Returns the name of the current param node.
"""
return self._currentParamNodeName
def getCurrentSelectedNodeNames(self):
"""
Returns the names of the current selected nodes.
"""
return self._currentSelectedNodeNames
@QtCore.Slot()
def getCurrentViewerNodeName(self):
"""
Returns the name of the current viewer node.
"""
return self._currentViewerNodeName
def getCurrentCopiedNodesInfo(self):
"""
Returns the list of buttle info for the current node(s) copied.
"""
return self._currentCopiedNodesInfo
### current data wrapper ###
def getCurrentParamNodeWrapper(self):
"""
Returns the current param nodeWrapper.
"""
return self._graphWrapper.getNodeWrapper(self.getCurrentParamNodeName())
def getCurrentSelectedNodeWrappers(self):
"""
Returns the current selected nodeWrappers as a QObjectListModel.
"""
currentSelectedNodeWrappers = QObjectListModel(self)
currentSelectedNodeWrappers.setObjectList([self.getGraphWrapper().getNodeWrapper(nodeName) for nodeName in self.getCurrentSelectedNodeNames()])
return currentSelectedNodeWrappers
def getCurrentViewerIndex(self):
"""
Returns the current viewer index.
"""
return self._currentViewerIndex
@QtCore.Slot(int, result=QtCore.QObject)
def getNodeWrapperByViewerIndex(self, index):
"""
Returns the nodeWrapper of the node contained in the viewer at the corresponding index.
"""
# We get the infos of this node. It's a tuple (fodeName, frame), so we have to return the first element nodeViewerInfos[0].
nodeViewerInfos = self._mapViewerIndextoNodeName[str(index)]
if nodeViewerInfos is None:
return None
else:
return self._graphWrapper.getNodeWrapper(nodeViewerInfos[0])
@QtCore.Slot(int, result=int)
def getFrameByViewerIndex(self, index):
"""
Returns the frame of the node contained in the viewer at the corresponding index.
"""
# We get the infos of this node. It's a tuple (fodeName, frame), so we have to return the second element nodeViewerInfos[1].
nodeViewerInfos = self._mapViewerIndextoNodeName[str(index)]
if nodeViewerInfos is None:
return 0
else:
return nodeViewerInfos[1]
def getCurrentViewerNodeWrapper(self):
"""
Returns the current viewer nodeWrapper.
"""
return self._graphWrapper.getNodeWrapper(self.getCurrentViewerNodeName())
def getCurrentViewerFrame(self):
"""
Returns the frame of the current viewer nodeWrapper.
"""
return self._currentViewerFrame
def getCurrentConnectionWrapper(self):
"""
Returns the current currentConnectionWrapper.
"""
return self._graphWrapper.getConnectionWrapper(self._currentConnectionId)
def getMapNodeNameToComputedImage(self):
"""
Returns the map of images already computed.
"""
return self._mapNodeNameToComputedImage
def graphCanBeSaved(self):
"""
Returns the value of the boolean self._graphCanBeSaved
"""
return self._graphCanBeSaved
#################### setters ####################
### current data ###
def setCurrentParamNodeName(self, nodeName):
self._currentParamNodeName = nodeName
def setCurrentSelectedNodeNames(self, nodeNames):
self._currentSelectedNodeNames = nodeNames
def setCurrentViewerNodeName(self, nodeName):
self._currentViewerNodeName = nodeName
self.currentViewerNodeChanged.emit()
def setCurrentViewerFrame(self, frame):
self._currentViewerFrame = frame
self.currentViewerFrameChanged.emit()
def setCurrentCopiedNodesInfo(self, nodesInfo):
self._currentCopiedNodesInfo = nodesInfo
### current data wrapper ###
def setCurrentParamNodeWrapper(self, nodeWrapper):
"""
Changes the current param node and emits the change.
"""
if self._currentParamNodeName == nodeWrapper.getName():
return
self._currentParamNodeName = nodeWrapper.getName()
# Emit signal
self.currentParamNodeChanged.emit()
def setCurrentSelectedNodeWrappers(self, nodeWrappers):
self.setCurrentSelectedNodeNames([nodeWrapper.getName() for nodeWrapper in nodeWrappers])
self.currentSelectedNodesChanged.emit()
def setCurrentViewerIndex(self, index):
"""
Set the value of the current viewer index.
"""
# Update value of the current viewer index
self._currentViewerIndex = index
# Emit signal
self.currentViewerIndexChanged.emit()
def setCurrentViewerNodeWrapper(self, nodeWrapper):
"""
Changes the current viewer node and emits the change.
"""
if nodeWrapper is None:
self._currentViewerNodeName = None
elif self._currentViewerNodeName == nodeWrapper.getName():
return
else:
self._currentViewerNodeName = nodeWrapper.getName()
# emit signal
self.currentViewerNodeChanged.emit()
def setCurrentConnectionWrapper(self, connectionWrapper):
"""
Changes the current conenctionWrapper and emits the change.
"""
if self._currentConnectionId == connectionWrapper.getId():
self._currentConnectionId = None
else:
self._currentConnectionId = connectionWrapper.getId()
self.currentConnectionWrapperChanged.emit()
def setGraphCanBeSaved(self, canBeSaved):
"""
Set the value of the boolean self._graphCanBeSaved.
"""
self._graphCanBeSaved = canBeSaved
self.graphCanBeSavedChanged.emit()
############################################### VIDEO FONCTIONS ##################################################
def getVideoIsPlaying(self):
return self._videoIsPlaying
def setVideoIsPlaying(self, valueBool):
self._videoIsPlaying = valueBool
def getProcessGraph(self):
return self._processGraph
def setProcessGraph(self, processGraph):
self._processGraph = processGraph
############################################### ADDITIONAL FUNCTIONS ##################################################
@QtCore.Slot(QtCore.QObject)
def appendToCurrentSelectedNodeWrappers(self, nodeWrapper):
self.appendToCurrentSelectedNodeNames(nodeWrapper.getName())
def appendToCurrentSelectedNodeNames(self, nodeName):
if nodeName in self._currentSelectedNodeNames:
self._currentSelectedNodeNames.remove(nodeName)
else:
self._currentSelectedNodeNames.append(nodeName)
# emit signal
self.currentSelectedNodesChanged.emit()
@QtCore.Slot("QVariant", result=bool)
def nodeIsSelected(self, nodeWrapper):
"""
Returns True if the node is selected (=if nodeName is in the list _currentSelectedNodeNames), else False.
"""
for nodeName in self._currentSelectedNodeNames:
if nodeName == nodeWrapper.getName():
return True
return False
@QtCore.Slot(int, int, int, int)
def addNodeWrappersInRectangleSelection(self, xRect, yRect, widthRect, heightRect):
"""
Selects the nodes which are is the rectangle selection area.
"""
for nodeW in self.getGraphWrapper().getNodeWrappers():
xNode = nodeW.getNode().getCoord()[0]
yNode = nodeW.getNode().getCoord()[1]
widthNode = nodeW.getWidth()
heightNode = nodeW.getHeight()
# we project the boundin-boxes on the axes and we check if the segments overlap
horizontalOverlap = (xNode < xRect + widthRect) and (xRect < xNode + widthNode)
verticalOverlap = (yNode < yRect + heightRect) and (yRect < yNode + heightNode)
overlap = horizontalOverlap and verticalOverlap
# if the bounding-boxes overlap then the node is in the selection area
if overlap:
self.appendToCurrentSelectedNodeNames(nodeW.getName())
@QtCore.Slot(QtCore.QObject, int)
def assignNodeToViewerIndex(self, nodeWrapper, frame):
"""
Assigns a node to the _mapViewerIndextoNodeName at the current viewerIndex.
It adds a tuple (nodeName, frame).
"""
if nodeWrapper:
self._mapViewerIndextoNodeName.update({str(self._currentViewerIndex): (nodeWrapper.getName(), frame)})
@QtCore.Slot()
def clearCurrentSelectedNodeNames(self):
self._currentSelectedNodeNames[:] = []
self.currentSelectedNodesChanged.emit()
@QtCore.Slot()
def clearCurrentConnectionId(self):
self._currentConnectionId = None
self.currentConnectionWrapperChanged.emit()
def clearCurrentCopiedNodesInfo(self):
self._currentCopiedNodesInfo.clear()
def canPaste(self):
"""
Returns True if we can paste (= if there is at least one node selected).
"""
return self._currentCopiedNodesInfo != {}
################################################## PLUGIN LIST #####################################################
@QtCore.Slot(str, result=QtCore.QObject)
def getQObjectPluginsIdentifiersByParentPath(self, pathname):
"""
Returns a QObjectListModel of all the PluginsIdentifiers (String) we expect to find after the submenu 'pathname'.
"""
pluginsIds = QObjectListModel(self)
pluginsIds.setObjectList(tuttleTools.getPluginsIdentifiersByParentPath(pathname))
return pluginsIds
@QtCore.Slot(str, result=bool)
def isAPlugin(self, pluginId):
"""
Returns if a string is a plugin identifier.
"""
return pluginId in tuttleTools.getPluginsIdentifiers()
################################################## SAVE / LOAD ##################################################
@QtCore.Slot(str)
@QtCore.Slot()
def saveData(self, url='buttleofx/backup/data.bofx'):
"""
Saves all data in a json file (default file : buttleofx/backup/data.bofx)
"""
with io.open(url, 'w', encoding='utf-8') as f:
dictJson = {
"date": {},
"window": {},
"graph": {},
"paramEditor": {},
"viewer": {
"other_views": {},
"current_view": {}
}
}
# date
today = datetime.today().strftime("%A, %d. %B %Y %I:%M%p")
dictJson["date"]["creation"] = today
# graph
dictJson["graph"] = self.getGraph().object_to_dict()
# graph : currentSeletedNodes
for node in self.getGraph().getNodes():
if node.getName() in self.getCurrentSelectedNodeNames():
dictJson["graph"]["currentSelectedNodes"].append(node.getName())
# paramEditor : currentParamNodeName
dictJson["paramEditor"] = self.getCurrentParamNodeName()
# viewer : currentViewerNodeName
for num_view, view in self._mapViewerIndextoNodeName.iteritems():
if view is not None:
(nodeName, frame) = view
if self.getCurrentViewerNodeName() == nodeName:
dictJson["viewer"]["current_view"][str(num_view)] = {}
dictJson["viewer"]["current_view"][str(num_view)]["nodeName"] = nodeName
dictJson["viewer"]["current_view"][str(num_view)]["frame"] = frame
else:
dictJson["viewer"]["other_views"][str(num_view)] = {}
dictJson["viewer"]["other_views"][str(num_view)]["nodeName"] = nodeName
dictJson["viewer"]["other_views"][str(num_view)]["frame"] = frame
# write dictJson in a file
f.write(unicode(json.dumps(dictJson, sort_keys=True, indent=2, ensure_ascii=False)))
f.closed
# Finally we update the savedGraphIndex of the CommandManager : it must be equal to the current index
CommandManager().setSavedGraphIndex(CommandManager().getIndex())
@QtCore.Slot(str)
@QtCore.Slot()
def loadData(self, url='buttleofx/backup/data.bofx'):
"""
Loads all data from a Json file (the default Json file if no url is given)
"""
with open(unicode(url), 'r') as f:
read_data = f.read()
decoded = json.loads(read_data, object_hook=_decode_dict)
# create the graph
self.getGraph().dict_to_object(decoded["graph"])
# graph : currentSeletedNodes
for currentSeletedNode in decoded["graph"]["currentSelectedNodes"]:
self.appendToCurrentSelectedNodeNames(currentSeletedNode)
self.currentSelectedNodesChanged.emit()
# paramEditor : currentParamNodeName
self.setCurrentParamNodeName(decoded["paramEditor"])
self.currentParamNodeChanged.emit()
# viewer : other views
for index, view in decoded["viewer"]["other_views"].iteritems():
nodeName, frame = view["nodeName"], view["frame"]
self._mapViewerIndextoNodeName.update({index: (nodeName, frame)})
# viewer : currentViewerNodeName
for index, current_view in decoded["viewer"]["current_view"].iteritems():
nodeName, frame = current_view["nodeName"], current_view["frame"]
self._mapViewerIndextoNodeName.update({index: (nodeName, frame)})
# #The next commands doesn't work : we need to click on the viewer's number to see the image in the viewer. Need to be fixed.
# self.setCurrentViewerIndex(int(index))
# self.setCurrentViewerNodeName(current_view)
# self.setCurrentViewerNodeWrapper = self.getNodeWrapperByViewerIndex(int(index))
# self.setCurrentViewerFrame(frame)
# ButtleEvent().emitViewerChangedSignal()
f.closed
################################################## DATA EXPOSED TO QML ##################################################
# graphWrapper
graphWrapper = QtCore.Property(QtCore.QObject, getGraphWrapper, constant=True)
# filePath
buttlePath = QtCore.Property(str, getButtlePath, constant=True)
# current param, view, and selected node
currentParamNodeChanged = QtCore.Signal()
currentParamNodeWrapper = QtCore.Property(QtCore.QObject, getCurrentParamNodeWrapper, setCurrentParamNodeWrapper, notify=currentParamNodeChanged)
currentViewerNodeChanged = QtCore.Signal()
currentViewerNodeWrapper = QtCore.Property(QtCore.QObject, getCurrentViewerNodeWrapper, setCurrentViewerNodeWrapper, notify=currentViewerNodeChanged)
currentViewerFrameChanged = QtCore.Signal()
currentViewerFrame = QtCore.Property(int, getCurrentViewerFrame, setCurrentViewerFrame, notify=currentViewerFrameChanged)
currentSelectedNodesChanged = QtCore.Signal()
currentSelectedNodeWrappers = QtCore.Property("QVariant", getCurrentSelectedNodeWrappers, setCurrentSelectedNodeWrappers, notify=currentSelectedNodesChanged)
currentConnectionWrapperChanged = QtCore.Signal()
currentConnectionWrapper = QtCore.Property(QtCore.QObject, getCurrentConnectionWrapper, setCurrentConnectionWrapper, notify=currentConnectionWrapperChanged)
currentViewerIndexChanged = QtCore.Signal()
currentViewerIndex = QtCore.Property(int, getCurrentViewerIndex, setCurrentViewerIndex, notify=currentViewerIndexChanged)
# paste possibility
pastePossibilityChanged = QtCore.Signal()
canPaste = QtCore.Property(bool, canPaste, notify=pastePossibilityChanged)
# possibility to save graph
graphCanBeSavedChanged = QtCore.Signal()
graphCanBeSaved = QtCore.Property(bool, graphCanBeSaved, notify=graphCanBeSavedChanged)
ParamInt(defaultValue=11, minimum=5, maximum=500, text="something"),
ParamInt(defaultValue=50, minimum=1, maximum=52, text="truc"),
ParamString(defaultValue="something.jpg", stringType="filename"),
ParamInt(defaultValue=7, minimum=5, maximum=12),
ParamString(defaultValue="somethingelse.jpg", stringType="type2")
]
# create undo-redo context
cmdManager = CommandManager()
cmdManager.setActive()
cmdManager.clean()
# data
# graph and graphWrapper
graph = Graph()
graphWrapper = GraphWrapper(graph, view)
# test node creation
graph.createNode("Blur", cmdManager)
graphWrapper.__str__()
# params
paramList = []
for node in graph.getNodes():
nodeName = ParamString(defaultValue=node.getName(), stringType="Name")
nodeType = ParamString(defaultValue=node.getType(), stringType="Type")
nodeCoord_x = ParamInt(defaultValue=node.getCoord()[0], minimum=0, maximum=1000, text="Coord x")
nodeCoord_y = ParamInt(defaultValue=node.getCoord()[1], minimum=0, maximum=1000, text="Coord y")
nodeColor_r = ParamInt(defaultValue=node.getColor().red(), minimum=0, maximum=255, text="Color red")
nodeColor_g = ParamInt(defaultValue=node.getColor().green(), minimum=0, maximum=255, text="Color green")
class ButtleData(QtCore.QObject):
"""
Class ButtleData defined by:
- _graphWrapper : the graphWrapper
- _graph : the graph (core)
- _currentViewerNodeName : the name of the node currently in the viewer
- _currentSelectedNodeNames : list of the names of the nodes currently selected
- _currentParamNodeName : the name of the node currently displayed in the paramEditor
- _currentConnectionId : the list of the id of the connections currently selected
- _currentCopiedNodesInfo : the list of buttle info for the current node(s) copied
- _mapNodeNameToComputedImage : this map makes the correspondance between a gloablHash and a computed image (max 20 images stored)
- _buttlePath : the path of the root directory (usefull to import images)
- _graphCanBeSaved : a boolean indicating if the graph had changed since the last saving (usefull for the display of the icon "save graph")
This class containts all data we need to manage the application.
"""
_graph = None
_graphWrapper = None
_graphCanBeSaved = False
# the current params
_currentParamNodeName = None
_currentSelectedNodeNames = []
# for the connections
_currentConnectionId = None
# to eventually save current nodes data
_currentCopiedNodesInfo = {}
# for the viewer
_currentViewerIndex = 1
_currentViewerFrame = 0
_currentViewerNodeName = None
_mapViewerIndextoNodeName = {}
_mapNodeNameToComputedImage = {}
# boolean used in viewerManager
_videoIsPlaying = False
# processGraph used in viewerManager to compute images
_processGraph = None
_timeRange = None
def init(self, view, filePath):
self._graph = Graph()
self._graphWrapper = GraphWrapper(self._graph, view)
self._buttlePath = filePath
for index in range(1, 10):
self._mapViewerIndextoNodeName[str(index)] = None
return self
################################################## GETTERS ET SETTERS ##################################################
#################### getters ####################
def getGraph(self):
return self._graph
def getGraphWrapper(self):
return self._graphWrapper
def getButtlePath(self):
return self._buttlePath
### current data ###
def getCurrentParamNodeName(self):
"""
Returns the name of the current param node.
"""
return self._currentParamNodeName
def getCurrentSelectedNodeNames(self):
"""
Returns the names of the current selected nodes.
"""
return self._currentSelectedNodeNames
@QtCore.Slot()
def getCurrentViewerNodeName(self):
"""
Returns the name of the current viewer node.
"""
return self._currentViewerNodeName
def getCurrentCopiedNodesInfo(self):
"""
Returns the list of buttle info for the current node(s) copied.
"""
return self._currentCopiedNodesInfo
### current data wrapper ###
def getCurrentParamNodeWrapper(self):
"""
Returns the current param nodeWrapper.
"""
return self._graphWrapper.getNodeWrapper(
self.getCurrentParamNodeName())
def getCurrentSelectedNodeWrappers(self):
"""
Returns the current selected nodeWrappers as a QObjectListModel.
"""
currentSelectedNodeWrappers = QObjectListModel(self)
currentSelectedNodeWrappers.setObjectList([
self.getGraphWrapper().getNodeWrapper(nodeName)
for nodeName in self.getCurrentSelectedNodeNames()
])
return currentSelectedNodeWrappers
def getCurrentViewerIndex(self):
"""
Returns the current viewer index.
"""
return self._currentViewerIndex
@QtCore.Slot(int, result=QtCore.QObject)
def getNodeWrapperByViewerIndex(self, index):
"""
Returns the nodeWrapper of the node contained in the viewer at the corresponding index.
"""
# We get the infos of this node. It's a tuple (fodeName, frame), so we have to return the first element nodeViewerInfos[0].
nodeViewerInfos = self._mapViewerIndextoNodeName[str(index)]
if nodeViewerInfos is None:
return None
else:
return self._graphWrapper.getNodeWrapper(nodeViewerInfos[0])
@QtCore.Slot(int, result=int)
def getFrameByViewerIndex(self, index):
"""
Returns the frame of the node contained in the viewer at the corresponding index.
"""
# We get the infos of this node. It's a tuple (fodeName, frame), so we have to return the second element nodeViewerInfos[1].
nodeViewerInfos = self._mapViewerIndextoNodeName[str(index)]
if nodeViewerInfos is None:
return 0
else:
return nodeViewerInfos[1]
def getCurrentViewerNodeWrapper(self):
"""
Returns the current viewer nodeWrapper.
"""
return self._graphWrapper.getNodeWrapper(
self.getCurrentViewerNodeName())
def getCurrentViewerFrame(self):
"""
Returns the frame of the current viewer nodeWrapper.
"""
return self._currentViewerFrame
def getCurrentConnectionWrapper(self):
"""
Returns the current currentConnectionWrapper.
"""
return self._graphWrapper.getConnectionWrapper(
self._currentConnectionId)
def getMapNodeNameToComputedImage(self):
"""
Returns the map of images already computed.
"""
return self._mapNodeNameToComputedImage
def graphCanBeSaved(self):
"""
Returns the value of the boolean self._graphCanBeSaved
"""
return self._graphCanBeSaved
#################### setters ####################
### current data ###
def setCurrentParamNodeName(self, nodeName):
self._currentParamNodeName = nodeName
def setCurrentSelectedNodeNames(self, nodeNames):
self._currentSelectedNodeNames = nodeNames
def setCurrentViewerNodeName(self, nodeName):
self._currentViewerNodeName = nodeName
self.currentViewerNodeChanged.emit()
def setCurrentViewerFrame(self, frame):
self._currentViewerFrame = frame
self.currentViewerFrameChanged.emit()
def setCurrentCopiedNodesInfo(self, nodesInfo):
self._currentCopiedNodesInfo = nodesInfo
### current data wrapper ###
def setCurrentParamNodeWrapper(self, nodeWrapper):
"""
Changes the current param node and emits the change.
"""
if self._currentParamNodeName == nodeWrapper.getName():
return
self._currentParamNodeName = nodeWrapper.getName()
# Emit signal
self.currentParamNodeChanged.emit()
def setCurrentSelectedNodeWrappers(self, nodeWrappers):
self.setCurrentSelectedNodeNames(
[nodeWrapper.getName() for nodeWrapper in nodeWrappers])
self.currentSelectedNodesChanged.emit()
def setCurrentViewerIndex(self, index):
"""
Set the value of the current viewer index.
"""
# Update value of the current viewer index
self._currentViewerIndex = index
# Emit signal
self.currentViewerIndexChanged.emit()
def setCurrentViewerNodeWrapper(self, nodeWrapper):
"""
Changes the current viewer node and emits the change.
"""
if nodeWrapper is None:
self._currentViewerNodeName = None
elif self._currentViewerNodeName == nodeWrapper.getName():
return
else:
self._currentViewerNodeName = nodeWrapper.getName()
# emit signal
self.currentViewerNodeChanged.emit()
def setCurrentConnectionWrapper(self, connectionWrapper):
"""
Changes the current conenctionWrapper and emits the change.
"""
if self._currentConnectionId == connectionWrapper.getId():
self._currentConnectionId = None
else:
self._currentConnectionId = connectionWrapper.getId()
self.currentConnectionWrapperChanged.emit()
def setGraphCanBeSaved(self, canBeSaved):
"""
Set the value of the boolean self._graphCanBeSaved.
"""
self._graphCanBeSaved = canBeSaved
self.graphCanBeSavedChanged.emit()
############################################### VIDEO FONCTIONS ##################################################
def getVideoIsPlaying(self):
return self._videoIsPlaying
def setVideoIsPlaying(self, valueBool):
self._videoIsPlaying = valueBool
def getProcessGraph(self):
return self._processGraph
def setProcessGraph(self, processGraph):
self._processGraph = processGraph
############################################### ADDITIONAL FUNCTIONS ##################################################
@QtCore.Slot(QtCore.QObject)
def appendToCurrentSelectedNodeWrappers(self, nodeWrapper):
self.appendToCurrentSelectedNodeNames(nodeWrapper.getName())
def appendToCurrentSelectedNodeNames(self, nodeName):
if nodeName in self._currentSelectedNodeNames:
self._currentSelectedNodeNames.remove(nodeName)
else:
self._currentSelectedNodeNames.append(nodeName)
# emit signal
self.currentSelectedNodesChanged.emit()
@QtCore.Slot("QVariant", result=bool)
def nodeIsSelected(self, nodeWrapper):
"""
Returns True if the node is selected (=if nodeName is in the list _currentSelectedNodeNames), else False.
"""
for nodeName in self._currentSelectedNodeNames:
if nodeName == nodeWrapper.getName():
return True
return False
@QtCore.Slot(int, int, int, int)
def addNodeWrappersInRectangleSelection(self, xRect, yRect, widthRect,
heightRect):
"""
Selects the nodes which are is the rectangle selection area.
"""
for nodeW in self.getGraphWrapper().getNodeWrappers():
xNode = nodeW.getNode().getCoord()[0]
yNode = nodeW.getNode().getCoord()[1]
widthNode = nodeW.getWidth()
heightNode = nodeW.getHeight()
# we project the boundin-boxes on the axes and we check if the segments overlap
horizontalOverlap = (xNode < xRect + widthRect) and (
xRect < xNode + widthNode)
verticalOverlap = (yNode < yRect + heightRect) and (
yRect < yNode + heightNode)
overlap = horizontalOverlap and verticalOverlap
# if the bounding-boxes overlap then the node is in the selection area
if overlap:
self.appendToCurrentSelectedNodeNames(nodeW.getName())
@QtCore.Slot(QtCore.QObject, int)
def assignNodeToViewerIndex(self, nodeWrapper, frame):
"""
Assigns a node to the _mapViewerIndextoNodeName at the current viewerIndex.
It adds a tuple (nodeName, frame).
"""
if nodeWrapper:
self._mapViewerIndextoNodeName.update({
str(self._currentViewerIndex): (nodeWrapper.getName(), frame)
})
@QtCore.Slot()
def clearCurrentSelectedNodeNames(self):
self._currentSelectedNodeNames[:] = []
self.currentSelectedNodesChanged.emit()
@QtCore.Slot()
def clearCurrentConnectionId(self):
self._currentConnectionId = None
self.currentConnectionWrapperChanged.emit()
def clearCurrentCopiedNodesInfo(self):
self._currentCopiedNodesInfo.clear()
def canPaste(self):
"""
Returns True if we can paste (= if there is at least one node selected).
"""
return self._currentCopiedNodesInfo != {}
################################################## PLUGIN LIST #####################################################
@QtCore.Slot(str, result=QtCore.QObject)
def getQObjectPluginsIdentifiersByParentPath(self, pathname):
"""
Returns a QObjectListModel of all the PluginsIdentifiers (String) we expect to find after the submenu 'pathname'.
"""
pluginsIds = QObjectListModel(self)
pluginsIds.setObjectList(
tuttleTools.getPluginsIdentifiersByParentPath(pathname))
return pluginsIds
@QtCore.Slot(str, result=bool)
def isAPlugin(self, pluginId):
"""
Returns if a string is a plugin identifier.
"""
return pluginId in tuttleTools.getPluginsIdentifiers()
################################################## SAVE / LOAD ##################################################
@QtCore.Slot(str)
@QtCore.Slot()
def saveData(self, url='buttleofx/backup/data.bofx'):
"""
Saves all data in a json file (default file : buttleofx/backup/data.bofx)
"""
with io.open(url, 'w', encoding='utf-8') as f:
dictJson = {
"date": {},
"window": {},
"graph": {},
"paramEditor": {},
"viewer": {
"other_views": {},
"current_view": {}
}
}
# date
today = datetime.today().strftime("%A, %d. %B %Y %I:%M%p")
dictJson["date"]["creation"] = today
# graph
dictJson["graph"] = self.getGraph().object_to_dict()
# graph : currentSeletedNodes
for node in self.getGraph().getNodes():
if node.getName() in self.getCurrentSelectedNodeNames():
dictJson["graph"]["currentSelectedNodes"].append(
node.getName())
# paramEditor : currentParamNodeName
dictJson["paramEditor"] = self.getCurrentParamNodeName()
# viewer : currentViewerNodeName
for num_view, view in self._mapViewerIndextoNodeName.iteritems():
if view is not None:
(nodeName, frame) = view
if self.getCurrentViewerNodeName() == nodeName:
dictJson["viewer"]["current_view"][str(num_view)] = {}
dictJson["viewer"]["current_view"][str(
num_view)]["nodeName"] = nodeName
dictJson["viewer"]["current_view"][str(
num_view)]["frame"] = frame
else:
dictJson["viewer"]["other_views"][str(num_view)] = {}
dictJson["viewer"]["other_views"][str(
num_view)]["nodeName"] = nodeName
dictJson["viewer"]["other_views"][str(
num_view)]["frame"] = frame
# write dictJson in a file
f.write(
unicode(
json.dumps(dictJson,
sort_keys=True,
indent=2,
ensure_ascii=False)))
f.closed
# Finally we update the savedGraphIndex of the CommandManager : it must be equal to the current index
CommandManager().setSavedGraphIndex(CommandManager().getIndex())
@QtCore.Slot(str)
@QtCore.Slot()
def loadData(self, url='buttleofx/backup/data.bofx'):
"""
Loads all data from a Json file (the default Json file if no url is given)
"""
with open(unicode(url), 'r') as f:
read_data = f.read()
decoded = json.loads(read_data, object_hook=_decode_dict)
# create the graph
self.getGraph().dict_to_object(decoded["graph"])
# graph : currentSeletedNodes
for currentSeletedNode in decoded["graph"]["currentSelectedNodes"]:
self.appendToCurrentSelectedNodeNames(currentSeletedNode)
self.currentSelectedNodesChanged.emit()
# paramEditor : currentParamNodeName
self.setCurrentParamNodeName(decoded["paramEditor"])
self.currentParamNodeChanged.emit()
# viewer : other views
for index, view in decoded["viewer"]["other_views"].iteritems():
nodeName, frame = view["nodeName"], view["frame"]
self._mapViewerIndextoNodeName.update(
{index: (nodeName, frame)})
# viewer : currentViewerNodeName
for index, current_view in decoded["viewer"][
"current_view"].iteritems():
nodeName, frame = current_view["nodeName"], current_view[
"frame"]
self._mapViewerIndextoNodeName.update(
{index: (nodeName, frame)})
# #The next commands doesn't work : we need to click on the viewer's number to see the image in the viewer. Need to be fixed.
# self.setCurrentViewerIndex(int(index))
# self.setCurrentViewerNodeName(current_view)
# self.setCurrentViewerNodeWrapper = self.getNodeWrapperByViewerIndex(int(index))
# self.setCurrentViewerFrame(frame)
# ButtleEvent().emitViewerChangedSignal()
f.closed
################################################## DATA EXPOSED TO QML ##################################################
# graphWrapper
graphWrapper = QtCore.Property(QtCore.QObject,
getGraphWrapper,
constant=True)
# filePath
buttlePath = QtCore.Property(str, getButtlePath, constant=True)
# current param, view, and selected node
currentParamNodeChanged = QtCore.Signal()
currentParamNodeWrapper = QtCore.Property(QtCore.QObject,
getCurrentParamNodeWrapper,
setCurrentParamNodeWrapper,
notify=currentParamNodeChanged)
currentViewerNodeChanged = QtCore.Signal()
currentViewerNodeWrapper = QtCore.Property(QtCore.QObject,
getCurrentViewerNodeWrapper,
setCurrentViewerNodeWrapper,
notify=currentViewerNodeChanged)
currentViewerFrameChanged = QtCore.Signal()
currentViewerFrame = QtCore.Property(int,
getCurrentViewerFrame,
setCurrentViewerFrame,
notify=currentViewerFrameChanged)
currentSelectedNodesChanged = QtCore.Signal()
currentSelectedNodeWrappers = QtCore.Property(
"QVariant",
getCurrentSelectedNodeWrappers,
setCurrentSelectedNodeWrappers,
notify=currentSelectedNodesChanged)
currentConnectionWrapperChanged = QtCore.Signal()
currentConnectionWrapper = QtCore.Property(
QtCore.QObject,
getCurrentConnectionWrapper,
setCurrentConnectionWrapper,
notify=currentConnectionWrapperChanged)
currentViewerIndexChanged = QtCore.Signal()
currentViewerIndex = QtCore.Property(int,
getCurrentViewerIndex,
setCurrentViewerIndex,
notify=currentViewerIndexChanged)
# paste possibility
pastePossibilityChanged = QtCore.Signal()
canPaste = QtCore.Property(bool, canPaste, notify=pastePossibilityChanged)
# possibility to save graph
graphCanBeSavedChanged = QtCore.Signal()
graphCanBeSaved = QtCore.Property(bool,
graphCanBeSaved,
notify=graphCanBeSavedChanged)
ParamInt(defaultValue=11, minimum=5, maximum=500, text="something"),
ParamInt(defaultValue=50, minimum=1, maximum=52, text="truc"),
ParamString(defaultValue="something.jpg", stringType="filename"),
ParamInt(defaultValue=7, minimum=5, maximum=12),
ParamString(defaultValue="somethingelse.jpg", stringType="type2")
]
# create undo-redo context
cmdManager = CommandManager()
cmdManager.setActive()
cmdManager.clean()
# data
# graph and graphWrapper
graph = Graph()
graphWrapper = GraphWrapper(graph, view)
# test node creation
graph.createNode("Blur", cmdManager)
graphWrapper.__str__()
# params
paramList = []
for node in graph.getNodes():
nodeName = ParamString(defaultValue=node.getName(), stringType="Name")
nodeType = ParamString(defaultValue=node.getType(), stringType="Type")
nodeCoord_x = ParamInt(defaultValue=node.getCoord()[0],
minimum=0,
maximum=1000,
text="Coord x")
nodeCoord_y = ParamInt(defaultValue=node.getCoord()[1],
