from Qt import QtCore, QtWidgets
import nodz_main
app = QtWidgets.QApplication([])
nodz = nodz_main.Nodz(None)
# nodz.loadConfig(filePath='')
nodz.initialize()
nodz.show()
######################################################################
# Test signals
######################################################################
# Nodes
@QtCore.Slot(str)
def on_nodeCreated(nodeName):
print 'node created : ', nodeName
@QtCore.Slot(str)
def on_nodeDeleted(nodeName):
print 'node deleted : ', nodeName
@QtCore.Slot(str, str)
def on_nodeEdited(nodeName, newName):
print 'node edited : {0}, new name : {1}'.format(nodeName, newName)
def __init__(self, type, controller):
"""
Initialize the graphics view.
"""
super(GUI, self).__init__()
self.controller = controller
self.layout = QtWidgets.QHBoxLayout(self)
self.leftSide = QtWidgets.QVBoxLayout(self)
self.generateWidget = Generate(self, self.controller)
self.leftSide.addWidget(self.generateWidget)
self.nodePickerWidget = NodePickerWidget(self, self.controller)
self.leftSide.addWidget(self.nodePickerWidget)
self.editor = Editor(self, self.controller)
self.editorTranslate = EditorTranslate(self, self.controller)
self.editorRotate = EditorRotate(self, self.controller)
self.editorScale = EditorScale(self, self.controller)
self.editorInitial = EditorInitial(self, self.controller)
self.editorMesh = EditorMesh(self, self.controller)
self.editorSplit = EditorSplit(self, self.controller)
self.editorSplitSegment = EditorSplitSegment(self, self.controller)
self.editorRepeat = EditorRepeat(self, self.controller)
self.editorWidget = QtWidgets.QStackedWidget(self)
self.editorWidget.addWidget(self.editorTranslate) #0
self.editorWidget.addWidget(self.editorRotate) #1
self.editorWidget.addWidget(self.editorScale) #2
self.editorWidget.addWidget(self.editorInitial) #3
self.editorWidget.addWidget(self.editorMesh) #4
self.editorWidget.addWidget(self.editorSplit) #5
self.editorWidget.addWidget(self.editor) #6
self.editorWidget.addWidget(self.editorSplitSegment) #7
self.editorWidget.addWidget(self.editorRepeat) #8
self.leftSide.addWidget(self.editorWidget)
self.editorWidget.setCurrentIndex(6)
self.nodzWidget = nodz_main.Nodz(self, self.controller)
self.nodzWidget.initialize()
self.nodzWidget.setMinimumSize(1000, 800)
# Put the left side layout into a widget so we can restrict
# the size of the widget because we can't restrict layouts
self.leftSideWidget = QtWidgets.QWidget()
self.leftSideWidget.setMaximumWidth(200)
self.leftSideWidget.setLayout(self.leftSide)
self.layout.addWidget(self.leftSideWidget)
self.layout.addWidget(self.nodzWidget)
self.window = QtWidgets.QWidget(self)
self.window.setLayout(self.layout)
self.setCentralWidget(self.window)
self.setWindowTitle("ProBuilder")
def do_shit():
win = hou.qt.mainWindow()
dialog = QtWidgets.QDialog(parent=win)
lay = QtWidgets.QHBoxLayout()
dialog.setLayout(lay)
print 'starting'
nodz = nodz_main.Nodz(None)
lay.addWidget(nodz)
# nodz.loadConfig(filePath='')
print 'initializing'
nodz.initialize()
print 'showing'
dialog.show()
print 'shown'
######################################################################
# Test signals
######################################################################
# Nodes
@QtCore.Slot(str)
def on_nodeCreated(nodeName):
print('node created : ', nodeName)
@QtCore.Slot(str)
def on_nodeDeleted(nodeName):
print('node deleted : ', nodeName)
@QtCore.Slot(str, str)
def on_nodeEdited(nodeName, newName):
print('node edited : {0}, new name : {1}'.format(nodeName, newName))
@QtCore.Slot(str)
def on_nodeSelected(nodesName):
print('node selected : ', nodesName)
@QtCore.Slot(str, object)
def on_nodeMoved(nodeName, nodePos):
print('node {0} moved to {1}'.format(nodeName, nodePos))
@QtCore.Slot(str)
def on_nodeDoubleClick(nodeName):
print('double click on node : {0}'.format(nodeName))
# Attrs
@QtCore.Slot(str, int)
def on_attrCreated(nodeName, attrId):
print('attr created : {0} at index : {1}'.format(nodeName, attrId))
@QtCore.Slot(str, int)
def on_attrDeleted(nodeName, attrId):
print('attr Deleted : {0} at old index : {1}'.format(nodeName, attrId))
@QtCore.Slot(str, int, int)
def on_attrEdited(nodeName, oldId, newId):
print('attr Edited : {0} at old index : {1}, new index : {2}'.format(nodeName, oldId, newId))
# Connections
@QtCore.Slot(str, str, str, str)
def on_connected(srcNodeName, srcPlugName, destNodeName, dstSocketName):
print('connected src: "{0}" at "{1}" to dst: "{2}" at "{3}"'.format(srcNodeName, srcPlugName, destNodeName, dstSocketName))
@QtCore.Slot(str, str, str, str)
def on_disconnected(srcNodeName, srcPlugName, destNodeName, dstSocketName):
print('disconnected src: "{0}" at "{1}" from dst: "{2}" at "{3}"'.format(srcNodeName, srcPlugName, destNodeName, dstSocketName))
# Graph
@QtCore.Slot()
def on_graphSaved():
print('graph saved !')
@QtCore.Slot()
def on_graphLoaded():
print('graph loaded !')
@QtCore.Slot()
def on_graphCleared():
print('graph cleared !')
@QtCore.Slot()
def on_graphEvaluated():
print('graph evaluated !')
# Other
@QtCore.Slot(object)
def on_keyPressed(key):
print('key pressed : ', key)
# nodz.signal_NodeCreated.connect(on_nodeCreated)
# nodz.signal_NodeDeleted.connect(on_nodeDeleted)
# nodz.signal_NodeEdited.connect(on_nodeEdited)
# nodz.signal_NodeSelected.connect(on_nodeSelected)
# nodz.signal_NodeMoved.connect(on_nodeMoved)
# nodz.signal_NodeDoubleClicked.connect(on_nodeDoubleClick)
#
# nodz.signal_AttrCreated.connect(on_attrCreated)
# nodz.signal_AttrDeleted.connect(on_attrDeleted)
# nodz.signal_AttrEdited.connect(on_attrEdited)
#
# nodz.signal_PlugConnected.connect(on_connected)
# nodz.signal_SocketConnected.connect(on_connected)
# nodz.signal_PlugDisconnected.connect(on_disconnected)
# nodz.signal_SocketDisconnected.connect(on_disconnected)
#
# nodz.signal_GraphSaved.connect(on_graphSaved)
# nodz.signal_GraphLoaded.connect(on_graphLoaded)
# nodz.signal_GraphCleared.connect(on_graphCleared)
# nodz.signal_GraphEvaluated.connect(on_graphEvaluated)
#
# nodz.signal_KeyPressed.connect(on_keyPressed)
######################################################################
# Test API
######################################################################
# Node A
nodeA = nodz.createNode(name='nodeA', preset='node_preset_1', position=None)
nodz.createAttribute(node=nodeA, name='Aattr1', index=-1, preset='attr_preset_1',
plug=True, socket=False, dataType=str)
nodz.createAttribute(node=nodeA, name='Aattr2', index=-1, preset='attr_preset_1',
plug=False, socket=False, dataType=int)
nodz.createAttribute(node=nodeA, name='Aattr3', index=-1, preset='attr_preset_2',
plug=True, socket=True, dataType=int)
nodz.createAttribute(node=nodeA, name='Aattr4', index=-1, preset='attr_preset_2',
plug=True, socket=True, dataType=str)
nodz.createAttribute(node=nodeA, name='Aattr5', index=-1, preset='attr_preset_3',
plug=True, socket=True, dataType=int, plugMaxConnections=1, socketMaxConnections=-1)
nodz.createAttribute(node=nodeA, name='Aattr6', index=-1, preset='attr_preset_3',
plug=True, socket=True, dataType=int, plugMaxConnections=1, socketMaxConnections=-1)
# Node B
nodeB = nodz.createNode(name='nodeB', preset='node_preset_1')
nodz.createAttribute(node=nodeB, name='Battr1', index=-1, preset='attr_preset_1',
plug=True, socket=False, dataType=str)
nodz.createAttribute(node=nodeB, name='Battr2', index=-1, preset='attr_preset_1',
plug=True, socket=False, dataType=int)
nodz.createAttribute(node=nodeB, name='Battr3', index=-1, preset='attr_preset_2',
plug=True, socket=False, dataType=int)
nodz.createAttribute(node=nodeB, name='Battr4', index=-1, preset='attr_preset_3',
plug=True, socket=False, dataType=int, plugMaxConnections=1, socketMaxConnections=-1)
# Node C
nodeC = nodz.createNode(name='nodeC', preset='node_preset_1')
nodz.createAttribute(node=nodeC, name='Cattr1', index=-1, preset='attr_preset_1',
plug=False, socket=True, dataType=str)
nodz.createAttribute(node=nodeC, name='Cattr2', index=-1, preset='attr_preset_1',
plug=True, socket=False, dataType=int)
nodz.createAttribute(node=nodeC, name='Cattr3', index=-1, preset='attr_preset_1',
plug=True, socket=False, dataType=str)
nodz.createAttribute(node=nodeC, name='Cattr4', index=-1, preset='attr_preset_2',
plug=False, socket=True, dataType=str)
nodz.createAttribute(node=nodeC, name='Cattr5', index=-1, preset='attr_preset_2',
plug=False, socket=True, dataType=int)
nodz.createAttribute(node=nodeC, name='Cattr6', index=-1, preset='attr_preset_3',
plug=True, socket=False, dataType=str)
nodz.createAttribute(node=nodeC, name='Cattr7', index=-1, preset='attr_preset_3',
plug=True, socket=False, dataType=str)
nodz.createAttribute(node=nodeC, name='Cattr8', index=-1, preset='attr_preset_3',
plug=True, socket=False, dataType=int)
# Please note that this is a local test so once the graph is cleared
# and reloaded, all the local variables are not valid anymore, which
# means the following code to alter nodes won't work but saving/loading/
# clearing/evaluating will.
# Connection creation
nodz.createConnection('nodeB', 'Battr2', 'nodeA', 'Aattr3')
nodz.createConnection('nodeB', 'Battr1', 'nodeA', 'Aattr4')
# Attributes Edition
nodz.editAttribute(node=nodeC, index=0, newName=None, newIndex=-1)
nodz.editAttribute(node=nodeC, index=-1, newName='NewAttrName', newIndex=None)
# Attributes Deletion
nodz.deleteAttribute(node=nodeC, index=-1)
# Nodes Edition
nodz.editNode(node=nodeC, newName='newNodeName')
# Nodes Deletion
# nodz.deleteNode(node=nodeC)
# Graph
print( nodz.evaluateGraph())