Esempio n. 1
0
class NodeEditorDialog ( QtGui.QDialog ) :
  #
  # __init__
  #
  def __init__ ( self, node = None ):
    #
    QtGui.QDialog.__init__ ( self )

    self.editNode = node

    # self.removedLinks = []

    self.nodePropertiesEditor = None
    self.nodeParamEditor = None
    self.nodeLinkEditor = None

    self.nodeCodeEditor = None
    self.controlCodeEditor = None
    self.eventCodeEditor = None
    self.paramCodeEditor = None

    #self.setEditNode ( node )
    self.buildGui ()
    self.connectSignals ()

    self.ui.btn_save.setDefault ( False )
    self.ui.btn_close.setDefault ( True )
    self.updateGui ()
  #
  # connectSignals
  #
  def connectSignals ( self ) :
    #
    QtCore.QObject.connect ( self.ui.input_list, QtCore.SIGNAL ( 'selectionChanged' ), self.updateGui ) # onInputParamSelectionChanged )
    QtCore.QObject.connect ( self.ui.output_list, QtCore.SIGNAL ( 'selectionChanged' ), self.updateGui ) # onOutputParamSelectionChanged  )

    QtCore.QObject.connect ( self.ui.input_list, QtCore.SIGNAL ( 'addItem' ), self.onAddParam )
    QtCore.QObject.connect ( self.ui.output_list, QtCore.SIGNAL ( 'addItem' ), self.onAddParam )
    QtCore.QObject.connect ( self.ui.input_list, QtCore.SIGNAL ( 'renameItem' ), self.onRenameParam )
    QtCore.QObject.connect ( self.ui.output_list, QtCore.SIGNAL ( 'renameItem' ), self.onRenameParam )
    QtCore.QObject.connect ( self.ui.input_list, QtCore.SIGNAL ( 'removeItem' ), self.onRemoveParam )
    QtCore.QObject.connect ( self.ui.output_list, QtCore.SIGNAL ( 'removeItem' ), self.onRemoveParam )

    QtCore.QObject.connect ( self.ui.tabs_param_list, QtCore.SIGNAL ( 'currentChanged(int)' ), self.updateGui )

    if self.nodeParamEditor is not None :
      QtCore.QObject.connect ( self.nodeParamEditor, QtCore.SIGNAL ( 'changeParamName' ), self.onRenameParam )
      QtCore.QObject.connect ( self.nodeParamEditor, QtCore.SIGNAL ( 'changeParamLabel' ), self.onRenameParamLabel )

    if self.nodeCodeEditor is not None :
      QtCore.QObject.connect ( self.nodeCodeEditor.ui.textEdit, QtCore.SIGNAL ( 'textChanged()' ), self.onEditCode )

    if self.controlCodeEditor is not None :
      QtCore.QObject.connect ( self.controlCodeEditor.ui.textEdit, QtCore.SIGNAL ( 'textChanged()' ), self.onEditControlCode )
      
    if self.paramCodeEditor is not None :
      QtCore.QObject.connect ( self.paramCodeEditor.ui.textEdit, QtCore.SIGNAL ( 'textChanged()' ), self.onEditParamCode )
      
    if self.eventCodeEditor is not None :
      QtCore.QObject.connect ( self.eventCodeEditor.ui.textEdit, QtCore.SIGNAL ( 'textChanged()' ), self.onEditEventCode )

    QtCore.QObject.connect ( self.ui.internals_list, QtCore.SIGNAL ( 'addItem' ), self.onAddInternal )
    QtCore.QObject.connect ( self.ui.includes_list, QtCore.SIGNAL ( 'addItem' ), self.onAddInclude )

    QtCore.QObject.connect ( self.ui.internals_list, QtCore.SIGNAL ( 'renameItem' ), self.onRenameInternal )
    QtCore.QObject.connect ( self.ui.includes_list, QtCore.SIGNAL ( 'renameItem' ), self.onRenameInclude )

    QtCore.QObject.connect ( self.ui.internals_list, QtCore.SIGNAL ( 'removeItem' ), self.onRemoveInternal )
    QtCore.QObject.connect ( self.ui.includes_list, QtCore.SIGNAL ( 'removeItem' ), self.onRemoveInclude )
    
    QtCore.QObject.connect ( self.ui.handlers_list, QtCore.SIGNAL ( 'selectionChanged' ), self.updateGui )

  #
  # disconnectSignals
  #
  def disconnectSignals ( self ) :
    #
    QtCore.QObject.disconnect ( self.ui.input_list, QtCore.SIGNAL ( 'selectionChanged' ), self.updateGui ) # onInputParamSelectionChanged )
    QtCore.QObject.disconnect ( self.ui.output_list, QtCore.SIGNAL ( 'selectionChanged' ), self.updateGui ) # onOutputParamSelectionChanged  )

    QtCore.QObject.disconnect ( self.ui.input_list, QtCore.SIGNAL ( 'addItem' ), self.onAddParam )
    QtCore.QObject.disconnect ( self.ui.output_list, QtCore.SIGNAL ( 'addItem' ), self.onAddParam )
    QtCore.QObject.disconnect ( self.ui.input_list, QtCore.SIGNAL ( 'renameItem' ), self.onRenameParam )
    QtCore.QObject.disconnect ( self.ui.output_list, QtCore.SIGNAL ( 'renameItem' ), self.onRenameParam )
    QtCore.QObject.disconnect ( self.ui.input_list, QtCore.SIGNAL ( 'removeItem' ), self.onRemoveParam )
    QtCore.QObject.disconnect ( self.ui.output_list, QtCore.SIGNAL ( 'removeItem' ), self.onRemoveParam )

    QtCore.QObject.disconnect ( self.ui.tabs_param_list, QtCore.SIGNAL ( "currentChanged(int)" ), self.updateGui )

    if self.nodeParamEditor is not None :
      QtCore.QObject.disconnect ( self.nodeParamEditor, QtCore.SIGNAL ( 'changeParamName' ), self.onRenameParam )
      QtCore.QObject.disconnect ( self.nodeParamEditor, QtCore.SIGNAL ( 'changeParamLabel' ), self.onRenameParamLabel )

    if self.nodeCodeEditor is not None :
      QtCore.QObject.disconnect ( self.nodeCodeEditor.ui.textEdit, QtCore.SIGNAL ( 'textChanged()' ), self.onEditCode )

    if self.controlCodeEditor is not None :
      QtCore.QObject.disconnect ( self.controlCodeEditor.ui.textEdit, QtCore.SIGNAL ( 'textChanged()' ), self.onEditControlCode)

    if self.paramCodeEditor is not None :
      QtCore.QObject.disconnect ( self.paramCodeEditor.ui.textEdit, QtCore.SIGNAL ( 'textChanged()' ), self.onEditParamCode )
    
    if self.eventCodeEditor is not None :
      QtCore.QObject.disconnect ( self.eventCodeEditor.ui.textEdit, QtCore.SIGNAL ( 'textChanged()' ), self.onEditEventCode )

    QtCore.QObject.disconnect ( self.ui.internals_list, QtCore.SIGNAL ( 'addItem' ), self.onAddInternal )
    QtCore.QObject.disconnect ( self.ui.includes_list, QtCore.SIGNAL ( 'addItem' ), self.onAddInclude )

    QtCore.QObject.disconnect ( self.ui.internals_list, QtCore.SIGNAL ( 'renameItem' ), self.onRenameInternal )
    QtCore.QObject.disconnect ( self.ui.includes_list, QtCore.SIGNAL ( 'renameItem' ), self.onRenameInclude )

    QtCore.QObject.disconnect ( self.ui.internals_list, QtCore.SIGNAL ( 'removeItem' ), self.onRemoveInternal )
    QtCore.QObject.disconnect ( self.ui.includes_list, QtCore.SIGNAL ( 'removeItem' ), self.onRemoveInclude )
    
    QtCore.QObject.disconnect ( self.ui.handlers_list, QtCore.SIGNAL ( 'selectionChanged' ), self.updateGui )

  #
  # setEditNode
  #
  def setEditNode ( self, editNode ) :
    #
    if DEBUG_MODE : print '>> NodeEditorDialog: setEditNode'
    self.editNode = editNode
  #
  #  buildGui
  #
  def buildGui ( self ) :
    # build the gui created with QtDesigner
    self.ui = Ui_NodeEditorDialog ()
    self.ui.setupUi ( self )

    if self.editNode is not None :
      #
      self.setWindowTitle ( 'NodeEditor: %s (%s)' % ( self.editNode.label, self.editNode.name ) )
      
      linkedFont = QtGui.QFont ()
      linkedFont.setItalic ( True )
      linkedBrush = QtGui.QBrush ()
      linkedBrush.setColor ( QtCore.Qt.blue )
      
      # setup loacal variables list
      for name in self.editNode.internals :
        item = QtGui.QListWidgetItem ( name )
        self.ui.internals_list.ui.listWidget.addItem ( item )
      
      # setup includes list
      for name in self.editNode.includes :
        item = QtGui.QListWidgetItem ( name )
        self.ui.includes_list.ui.listWidget.addItem ( item )
      
      # setup input params list
      for param in self.editNode.inputParams :
        item = QtGui.QListWidgetItem ( param.name )
        if self.editNode.isInputParamLinked ( param ) :
          item.setFont ( linkedFont )
          item.setForeground ( linkedBrush )
        self.ui.input_list.ui.listWidget.addItem ( item )
  
      # setup output params list
      for param in self.editNode.outputParams :
        item = QtGui.QListWidgetItem ( param.name )
        if self.editNode.isOutputParamLinked ( param ) :
          item.setFont ( linkedFont )
          item.setForeground ( linkedBrush )
        self.ui.output_list.ui.listWidget.addItem ( item )
  
      # setup input links list
      for link in self.editNode.getInputLinks () :
        item = QtGui.QListWidgetItem ( 'id=%d' % link.id  )
        item.setData ( QtCore.Qt.UserRole, QVariant ( int ( link.id ) ) )
        self.ui.input_links_listWidget.addItem ( item )
  
      # setup output links list
      for link in self.editNode.getOutputLinks () :
        item = QtGui.QListWidgetItem ( 'id=%d' % link.id  )
        item.setData ( QtCore.Qt.UserRole, QVariant ( int ( link.id ) ) )
        self.ui.output_links_listWidget.addItem ( item )
        
      # setup event handlers list
      if self.editNode.event_code :
        for handler in self.editNode.event_code.keys () :
          item = QtGui.QListWidgetItem ( handler )
          self.ui.handlers_list.ui.listWidget.addItem ( item )
  
      self.nodeCodeEditor = self.ui.node_code
      self.nodeCodeEditor.setNodeCode ( self.editNode.code, 'SL' )
      
      self.controlCodeEditor = self.ui.control_code
      self.controlCodeEditor.setNodeCode ( self.editNode.control_code, 'python' )
      
      self.eventCodeEditor = self.ui.event_code
      #self.eventCodeEditor.setNodeCode ( self.editNode.event_code, 'python' )
      
      self.paramCodeEditor = self.ui.param_code
      #self.paramCodeEditor.setNodeCode ( self.editNode.param_code, 'python' )

      
      self.nodePropertiesEditor = self.ui.node
      self.nodePropertiesEditor.setNode ( self.editNode )
      
      self.nodeParamEditor = self.ui.param
      
      self.nodeLinkEditor = self.ui.link

      self.ui.tabWidget.setCurrentIndex ( TAB_NODE_CODE )
      self.ui.toolBox.setCurrentIndex ( IDX_INTERNALS )
      self.ui.tabs_param_list.setCurrentIndex ( 0 ) # input param tab

  #
  # updateGui
  #
  def updateGui ( self ) :
    #
    if self.editNode is not None :
      if DEBUG_MODE : print '>> NodeEditorDialog::updateGui'
        
      self.disconnectSignals ()
      
      self.eventCodeEditor.setNodeCode ( None, 'python' )
      self.paramCodeEditor.setNodeCode ( None, 'python' )
      
      idx = self.ui.toolBox.currentIndex ()
      
      if idx == IDX_PARAM :
        # Parameters
        tab_idx = self.ui.tabs_param_list.currentIndex ()
        param = None

        if tab_idx == 0 :
          # input parameters
          list_item = self.ui.input_list.ui.listWidget.currentItem ()

          if list_item is not None :
            param = self.editNode.getInputParamByName ( str ( list_item.text () ) )

        elif tab_idx == 1 :
          # output parametrs
          list_item = self.ui.output_list.ui.listWidget.currentItem ()

          if list_item is not None :
            param = self.editNode.getOutputParamByName ( str ( list_item.text () ) )

        self.nodeParamEditor.setParam ( param )
        if param is not None and param.type == 'control' :
          self.paramCodeEditor.setNodeCode ( param.code, 'python' )
          #print '*** set (%s).param.code :' % param.label
          #print param.code

      elif idx == IDX_HANDLERS :
        handler_item = self.ui.handlers_list.ui.listWidget.currentItem ()
        if handler_item is not None :
          handler = str ( handler_item.text () )
          handler_code = self.editNode.event_code [ handler ]
          self.eventCodeEditor.setNodeCode ( handler_code, 'python' )
        #else :
        #  print '** no selection in handlers_list'

      elif idx == IDX_LINKS :
        inputLinkSelected = False
        link_id = None
        tab_idx = self.ui.tabs_links_list.currentIndex ()

        if tab_idx == 0 :
          # input links
          inputLinkSelected = True
          links_item = self.ui.input_links_listWidget.currentItem ()
          if links_item is not None : ( link_id, ok ) = links_item.data ( QtCore.Qt.UserRole ).toInt ()

        else :
          # output links
          links_item = self.ui.output_links_listWidget.currentItem ()
          if links_item is not None : ( link_id, ok ) = links_item.data ( QtCore.Qt.UserRole ).toInt ()

        if link_id is not None :
          link = None

          if inputLinkSelected : link = self.editNode.getInputLinkByID ( link_id )
          else                 : link = self.editNode.getOutputLinkByID ( link_id )

          if link is not None :
            ( srcNode, srcParam ) = link.getSrc ()
            ( dstNode, dstParam ) = link.getDst ()
            self.nodeLinkEditor.ui.src_node_lineEdit.setText ( srcNode.label )
            self.nodeLinkEditor.ui.src_param_lineEdit.setText ( srcParam.label )
            self.nodeLinkEditor.ui.src_id_lineEdit.setText ( str ( srcNode.id ) )
            self.nodeLinkEditor.ui.dst_node_lineEdit.setText ( dstNode.label )
            self.nodeLinkEditor.ui.dst_param_lineEdit.setText ( dstParam.label )
            self.nodeLinkEditor.ui.dst_id_lineEdit.setText ( str ( dstNode.id ) )
      self.connectSignals ()
  #
  # onToolBoxIndexChanged
  #
  def onToolBoxIndexChanged ( self, idx ) :
    if DEBUG_MODE : print '>> NodeEditorDialog::onToolBoxIndexChanged (idx = %d)' % idx
    #
    #self.disconnectSignals ()

    if idx == IDX_PARAM :
      # Input, Output Parameters
      self.ui.tabWidget.setCurrentIndex ( TAB_PARAM )

    elif idx == IDX_LINKS :
      # Input, Output Links
      self.ui.tabWidget.setCurrentIndex ( TAB_LINK_INFO )

    elif idx == IDX_INTERNALS :
      # Includes, Local Names, Code
      self.ui.tabWidget.setCurrentIndex ( TAB_NODE_CODE )
      
    elif idx == IDX_HANDLERS :
      # Event Handlers code
      self.ui.tabWidget.setCurrentIndex ( TAB_EVENT_CODE )

    #self.connectSignals ()
    self.updateGui ()
  #
  # onInputParamSelectionChanged
  #
  def onInputParamSelectionChanged ( self, paramName ) :
    #
    if DEBUG_MODE : print '>> NodeEditorDialog::onInputParamSelectionChanged (%s)' % paramName
    param = self.editNode.getInputParamByName ( str ( paramName ) )
    self.nodeParamEditor.setParam ( param )
  #
  # onOutputParamSelectionChanged
  #
  def onOutputParamSelectionChanged ( self, paramName ) :
    #
    if DEBUG_MODE : print '>> NodeEditorDialog::onOutputParamSelectionChanged (%s)' % paramName
    param = self.editNode.getOutputParamByName ( str ( paramName ) )
    self.nodeParamEditor.setParam ( param )
  #
  # onCodeListIndexChanged
  #
  def onCodeListIndexChanged ( self, idx ) :
    # if DEBUG_MODE : print '>> NodeEditorDialog: onCodeListIndexChanged idx = %d' % idx
    self.updateGui ()
  #
  # onRemoveInternal
  #
  def onRemoveInternal ( self, internal ) :
    #
    internalsListWidget = self.ui.internals_list.ui.listWidget
    self.editNode.internals.remove ( internal )
    item = internalsListWidget.currentItem ()
    internalsListWidget.takeItem ( internalsListWidget.currentRow () )
    internalsListWidget.removeItemWidget ( item )
    internalsListWidget.clearSelection ()
    internalsListWidget.setCurrentItem ( None )
  #
  # onRemoveInclude
  #
  def onRemoveInclude ( self, include ) :
    #
    includesListWidget = self.ui.includes_list.ui.listWidget
    self.editNode.includes.remove ( include )
    item = includesListWidget.currentItem()
    includesListWidget.takeItem ( includesListWidget.currentRow () )
    includesListWidget.removeItemWidget ( item )
    includesListWidget.clearSelection ()
    includesListWidget.setCurrentItem ( None )
  #
  # onRemoveParam
  #
  def onRemoveParam ( self, paramName ) :
    #
    isInputParam = False
    tab_idx = self.ui.tabs_param_list.currentIndex ()
    if tab_idx == 0 : isInputParam = True
    param = None
    paramList = None
    linkList = None

    if isInputParam :
      param = self.editNode.getInputParamByName ( paramName )
      paramList = self.ui.input_list
      linkList = self.ui.input_links_listWidget
    else :
      param = self.editNode.getOutputParamByName ( paramName )
      paramList = self.ui.output_list
      linkList = self.ui.output_links_listWidget

    removedLinks = self.editNode.removeParam ( param )

    # remove item from links list and node network
    for link in removedLinks :
      item_name = 'id=%d' % link.id
      item = linkList.findItems ( item_name, QtCore.Qt.MatchExactly )[0]
      linkList.takeItem ( linkList.row ( item ) )
      link.nodenet.removeLink ( link )

    # remove item from param list
    item = paramList.ui.listWidget.currentItem ()
    paramList.ui.listWidget.takeItem ( paramList.ui.listWidget.currentRow () )
    paramList.ui.listWidget.removeItemWidget ( item )
    paramList.ui.listWidget.clearSelection ()
    paramList.ui.listWidget.setCurrentItem ( None )

    # remove item from code (invalidate code)
    pass
  #
  # onRenameInternal
  #
  def onRenameInternal ( self, oldName, newName ) :
    #
    internalsListWidget = self.ui.internals_list.ui.listWidget
    from core.meCommon import getUniqueName
    idx = self.editNode.internals.index ( oldName )
    newName = getUniqueName ( newName, self.editNode.internals )
    self.editNode.internals [ idx ] = newName

    item = internalsListWidget.findItems ( oldName, QtCore.Qt.MatchExactly )[0]
    item.setText ( newName )
    self.ui.internals_list.setName ( newName )
    internalsListWidget.clearSelection ()
    internalsListWidget.setCurrentItem ( item )
  #
  # onRenameInclude
  #
  def onRenameInclude ( self, oldName, newName ) :
    #
    includesListWidget = self.ui.includes_list.ui.listWidget
    from core.meCommon import getUniqueName
    idx = self.editNode.includes.index ( oldName )
    newName = getUniqueName ( newName, self.editNode.includes )
    self.editNode.includes [ idx ] = newName

    item = includesListWidget.findItems ( oldName, QtCore.Qt.MatchExactly )[0]
    item.setText ( newName )
    self.ui.includes_list.setName ( newName )
    includesListWidget.clearSelection ()
    includesListWidget.setCurrentItem ( item )
  #
  # onRenameParam
  #
  def onRenameParam ( self, oldName, newName ) :
    #
    isInputParam = False
    tab_idx = self.ui.tabs_param_list.currentIndex ()
    if tab_idx == 0 : isInputParam = True
    param = None
    paramList = None
    if isInputParam :
      param = self.editNode.getInputParamByName ( oldName )
      paramList = self.ui.input_list
    else :
      param = self.editNode.getOutputParamByName ( oldName )
      paramList = self.ui.output_list

    self.editNode.renameParamName ( param, newName )

    item = paramList.ui.listWidget.findItems ( oldName, QtCore.Qt.MatchExactly )[0]
    item.setText ( param.name )
    paramList.setName ( param.name )
    paramList.ui.listWidget.clearSelection ()
    paramList.ui.listWidget.setCurrentItem ( item )
  #
  # onRenameParamLabel
  #
  def onRenameParamLabel ( self, oldName, newName ) :
    #
    param = self.nodeParamEditor.param
    self.editNode.renameParamLabel ( param, newName )
    self.nodeParamEditor.ui.label_lineEdit.setText ( param.label )
  #
  # onAddInternal
  #
  def onAddInternal ( self, newName ) :
    #
    if DEBUG_MODE : print '>> NodeEditorDialog::onAddInternal (%s) ' % (newName)
    # name can be changed to be unique
    newName = self.editNode.addInternal ( newName )
    internalsListWidget = self.ui.internals_list.ui.listWidget
    internalsListWidget.addItem ( newName )
    internalsListWidget.setCurrentItem ( internalsListWidget.findItems ( newName, QtCore.Qt.MatchExactly )[0] )
  #
  # onAddInclude
  #
  def onAddInclude ( self, newName ) :
    #
    if DEBUG_MODE : print '>> NodeEditorDialog::onAddInclude (%s) ' % (newName)
    # name can be changed to be unique
    newName = self.editNode.addInclude ( newName )
    includesListWidget = self.ui.includes_list.ui.listWidget
    includesListWidget.addItem ( newName )
    includesListWidget.setCurrentItem ( includesListWidget.findItems ( newName, QtCore.Qt.MatchExactly )[0] )
  #
  # onAddParam
  #
  def onAddParam ( self, newName ) :
    #
    if DEBUG_MODE : print '>> NodeEditorDialog::onAddParam (%s) ' % (newName)
    isInputParam = False
    paramType = None
    isRibParam = ( self.editNode.type in VALID_RIB_NODE_TYPES )
    tab_idx = self.ui.tabs_param_list.currentIndex ()
    if tab_idx == 0 : isInputParam = True
    # ask user about param type
    typeDialog = QtGui.QDialog () # Qt.MSWindowsFixedSizeDialogHint
    typeDialog.setModal ( True )

    typeDialog.setWindowTitle ( 'Parameter Type' )
    typeDialog.resize (180, 100 )
    sizePolicy = QtGui.QSizePolicy ( QtGui.QSizePolicy.Fixed, QtGui.QSizePolicy.Fixed )
    sizePolicy.setHorizontalStretch ( 0 )
    sizePolicy.setVerticalStretch ( 0 )
    sizePolicy.setHeightForWidth ( typeDialog.sizePolicy().hasHeightForWidth() )
    typeDialog.setSizePolicy ( sizePolicy )
    typeDialog.setSizeGripEnabled ( False )

    typeDialog.verticalLayout = QtGui.QVBoxLayout ( typeDialog )
    typeDialog.verticalLayout.setSizeConstraint ( QtGui.QLayout.SetMinimumSize )
    typeDialog.type_comboBox = QtGui.QComboBox ( typeDialog )
    
    for label in VALID_PARAM_TYPES : typeDialog.type_comboBox.addItem ( label )
    
    typeDialog.verticalLayout.addWidget ( typeDialog.type_comboBox )

    typeDialog.btnBox = QtGui.QDialogButtonBox ( QtGui.QDialogButtonBox.Ok | QtGui.QDialogButtonBox.Cancel, parent = typeDialog )
    typeDialog.btnBox.setCenterButtons ( True )
    typeDialog.verticalLayout.addWidget ( typeDialog.btnBox )

    QtCore.QObject.connect ( typeDialog.btnBox, QtCore.SIGNAL ( 'accepted()' ), typeDialog.accept )
    QtCore.QObject.connect ( typeDialog.btnBox, QtCore.SIGNAL ( 'rejected()' ), typeDialog.reject )

    if typeDialog.exec_() == QtGui.QDialog.Accepted  :
      paramType = str ( typeDialog.type_comboBox.currentText () )

      if DEBUG_MODE : print '>> NodeEditorDialog::onAddParam typeDialog Accepted (%s)' % paramType
      # create empty xml node parameter
      dom = QtXml.QDomDocument ( newName )
      xmlnode = dom.createElement( 'property' )

      xmlnode.setAttribute ( 'name', newName )
      xmlnode.setAttribute ( 'label', newName )
      xmlnode.setAttribute ( 'type', paramType )
      param = createParamFromXml ( xmlnode, isRibParam, isInputParam )
      item = QtGui.QListWidgetItem ( param.name )

      paramListWidget = self.ui.input_list.ui.listWidget

      if isInputParam :
        self.editNode.addInputParam ( param )
      else :
        self.editNode.addOutputParam ( param )
        paramListWidget = self.ui.output_list.ui.listWidget

      paramListWidget.addItem ( param.name )
      paramListWidget.setCurrentItem ( paramListWidget.findItems ( param.name, QtCore.Qt.MatchExactly )[0] )
      #self.nodeParamEditor.setParam ( param )
  #
  # onEditCode
  #
  def onEditCode ( self ) :
    #
    if DEBUG_MODE : print '>> NodeEditorDialog::onEditCode'
    if self.nodeCodeEditor is not None :
      #self.nodeCodeEditor.ui.textEdit
      self.editNode.code = str ( self.nodeCodeEditor.ui.textEdit.toPlainText () )
  #
  # onEditControlCode
  #
  def onEditControlCode ( self ) :
    #
    if DEBUG_MODE : print '>> NodeEditorDialog::onEditControlCode'
    if self.controlCodeEditor is not None :
      self.editNode.control_code = str ( self.controlCodeEditor.ui.textEdit.toPlainText () )
  #
  # onEditParamCode
  #
  def onEditParamCode ( self ) :
    #
    if DEBUG_MODE : print '>> NodeEditorDialog::onEditParamCode'
    if self.paramCodeEditor is not None :
      param = self.nodeParamEditor.param
      if param is not None and param.type == 'control' :
        param.code = str ( self.paramCodeEditor.ui.textEdit.toPlainText () )
        #print '*** set (%s).param.code :' % param.label
        #print param.code
  #
  # onEditEventCode
  #
  def onEditEventCode ( self ) :
    #
    if DEBUG_MODE : print '>> NodeEditorDialog::onEditEventCode'
    if self.eventCodeEditor is not None :
      handler_item = self.ui.handlers_list.ui.listWidget.currentItem ()
      if handler_item is not None :
        handler = str ( handler_item.text () )
        self.editNode.event_code [ handler ] = str ( self.eventCodeEditor.ui.textEdit.toPlainText () )

  #
  # Ignore default Enter press event
  #
  def keyPressEvent ( self, event  ) :
    #
    #if DEBUG_MODE : print '>> NodeEditorDialog::keyPressEvent'
    if  event.key () == QtCore.Qt.Key_Enter or event.key () == QtCore.Qt.Key_Return :
      event.ignore ()
    else:
      QtGui.QDialog.keyPressEvent ( self, event )
  #
  # accept
  #
  def accept ( self ) :
    #
    if DEBUG_MODE : print '>> NodeEditorDialog::accept'
    self.done ( QtGui.QDialog.Accepted )
Esempio n. 2
0
class NodeEditorDialog ( QtModule.QDialog ) :
	#
	# __init__
	#
	def __init__ ( self, node = None ):
		#
		QtModule.QDialog.__init__ ( self )

		self.editNode = node
		# self.removedLinks = []
		self.nodePropertiesEditor = None
		self.nodeParamEditor = None
		self.nodeLinkEditor = None

		self.nodeCodeEditor = None
		self.controlCodeEditor = None
		self.eventCodeEditor = None
		self.paramCodeEditor = None

		#self.setEditNode ( node )
		self.buildGui ()
		self.connectSignals ()

		self.ui.btn_save.setDefault ( False )
		self.ui.btn_close.setDefault ( True )
		self.updateGui ()
	#
	# connectSignals
	#
	def connectSignals ( self ) :
		#
		if usePyQt4 :
			QtCore.QObject.connect ( self.ui.input_list, QtCore.SIGNAL ( 'selectionChangedSignal' ), self.updateGui ) # onInputParamSelectionChanged )
			QtCore.QObject.connect ( self.ui.output_list, QtCore.SIGNAL ( 'selectionChangedSignal' ), self.updateGui ) # onOutputParamSelectionChanged  )
	
			QtCore.QObject.connect ( self.ui.input_list, QtCore.SIGNAL ( 'addItem' ), self.onAddParam )
			QtCore.QObject.connect ( self.ui.output_list, QtCore.SIGNAL ( 'addItem' ), self.onAddParam )
			
			QtCore.QObject.connect ( self.ui.input_list, QtCore.SIGNAL ( 'renameItem' ), self.onRenameParam )
			QtCore.QObject.connect ( self.ui.output_list, QtCore.SIGNAL ( 'renameItem' ), self.onRenameParam )
			
			QtCore.QObject.connect ( self.ui.input_list, QtCore.SIGNAL ( 'removeItem' ), self.onRemoveParam )
			QtCore.QObject.connect ( self.ui.output_list, QtCore.SIGNAL ( 'removeItem' ), self.onRemoveParam )
	
			QtCore.QObject.connect ( self.ui.tabs_param_list, QtCore.SIGNAL ( 'currentChanged(int)' ), self.updateGui )
	
			if self.nodeParamEditor is not None :
				QtCore.QObject.connect ( self.nodeParamEditor, QtCore.SIGNAL ( 'changeParamName' ), self.onRenameParam )
				QtCore.QObject.connect ( self.nodeParamEditor, QtCore.SIGNAL ( 'changeParamLabel' ), self.onRenameParamLabel )
	
			if self.nodeCodeEditor is not None :
				QtCore.QObject.connect ( self.nodeCodeEditor.ui.textEdit, QtCore.SIGNAL ( 'textChanged()' ), self.onEditCode )
	
			if self.controlCodeEditor is not None :
				QtCore.QObject.connect ( self.controlCodeEditor.ui.textEdit, QtCore.SIGNAL ( 'textChanged()' ), self.onEditControlCode )
				
			if self.paramCodeEditor is not None :
				QtCore.QObject.connect ( self.paramCodeEditor.ui.textEdit, QtCore.SIGNAL ( 'textChanged()' ), self.onEditParamCode )
				
			if self.eventCodeEditor is not None :
				QtCore.QObject.connect ( self.eventCodeEditor.ui.textEdit, QtCore.SIGNAL ( 'textChanged()' ), self.onEditEventCode )
	
			QtCore.QObject.connect ( self.ui.internals_list, QtCore.SIGNAL ( 'addItem' ), self.onAddInternal )
			QtCore.QObject.connect ( self.ui.includes_list, QtCore.SIGNAL ( 'addItem' ), self.onAddInclude )
	
			QtCore.QObject.connect ( self.ui.internals_list, QtCore.SIGNAL ( 'renameItem' ), self.onRenameInternal )
			QtCore.QObject.connect ( self.ui.includes_list, QtCore.SIGNAL ( 'renameItem' ), self.onRenameInclude )
	
			QtCore.QObject.connect ( self.ui.internals_list, QtCore.SIGNAL ( 'removeItem' ), self.onRemoveInternal )
			QtCore.QObject.connect ( self.ui.includes_list, QtCore.SIGNAL ( 'removeItem' ), self.onRemoveInclude )
			
			QtCore.QObject.connect ( self.ui.handlers_list, QtCore.SIGNAL ( 'selectionChangedSignal' ), self.updateGui )
		else :
			self.ui.input_list.selectionChangedSignal.connect ( self.updateGui ) # onInputParamSelectionChanged )
			self.ui.output_list.selectionChangedSignal.connect ( self.updateGui ) # onOutputParamSelectionChanged  )
	
			self.ui.input_list.addItem.connect ( self.onAddParam )
			self.ui.output_list.addItem.connect ( self.onAddParam )
			
			self.ui.input_list.renameItem.connect ( self.onRenameParam )
			self.ui.output_list.renameItem.connect ( self.onRenameParam )
			
			self.ui.input_list.removeItem.connect ( self.onRemoveParam )
			self.ui.output_list.removeItem.connect ( self.onRemoveParam )
	
			self.ui.tabs_param_list.currentChanged.connect ( self.updateGui )
	
			if self.nodeParamEditor is not None :
				self.nodeParamEditor.changeParamName.connect ( self.onRenameParam )
				self.nodeParamEditor.changeParamName.connect ( self.onRenameParamLabel )
	
			if self.nodeCodeEditor is not None :
				self.nodeCodeEditor.ui.textEdit.textChanged.connect ( self.onEditCode )
	
			if self.controlCodeEditor is not None :
				self.controlCodeEditor.ui.textEdit.textChanged.connect ( self.onEditControlCode )
				
			if self.paramCodeEditor is not None :
				self.paramCodeEditor.ui.textEdit.textChanged.connect ( self.onEditParamCode )
				
			if self.eventCodeEditor is not None :
				self.eventCodeEditor.ui.textEdit.textChanged.connect ( self.onEditEventCode )
	
			self.ui.internals_list.addItem.connect ( self.onAddInternal )
			self.ui.includes_list.addItem.connect ( self.onAddInclude )
	
			self.ui.internals_list.renameItem.connect ( self.onRenameInternal )
			self.ui.includes_list.renameItem.connect ( self.onRenameInclude )
	
			self.ui.internals_list.removeItem.connect ( self.onRemoveInternal )
			self.ui.includes_list.removeItem.connect ( self.onRemoveInclude )
			
			self.ui.handlers_list.selectionChangedSignal.connect ( self.updateGui )

	#
	# disconnectSignals
	#
	def disconnectSignals ( self ) :
		#
		if usePyQt4 :
			QtCore.QObject.disconnect ( self.ui.input_list, QtCore.SIGNAL ( 'selectionChangedSignal' ), self.updateGui ) # onInputParamSelectionChanged )
			QtCore.QObject.disconnect ( self.ui.output_list, QtCore.SIGNAL ( 'selectionChangedSignal' ), self.updateGui ) # onOutputParamSelectionChanged  )
	
			QtCore.QObject.disconnect ( self.ui.input_list, QtCore.SIGNAL ( 'addItem' ), self.onAddParam )
			QtCore.QObject.disconnect ( self.ui.output_list, QtCore.SIGNAL ( 'addItem' ), self.onAddParam )
			
			QtCore.QObject.disconnect ( self.ui.input_list, QtCore.SIGNAL ( 'renameItem' ), self.onRenameParam )
			QtCore.QObject.disconnect ( self.ui.output_list, QtCore.SIGNAL ( 'renameItem' ), self.onRenameParam )
			
			QtCore.QObject.disconnect ( self.ui.input_list, QtCore.SIGNAL ( 'removeItem' ), self.onRemoveParam )
			QtCore.QObject.disconnect ( self.ui.output_list, QtCore.SIGNAL ( 'removeItem' ), self.onRemoveParam )
	
			QtCore.QObject.disconnect ( self.ui.tabs_param_list, QtCore.SIGNAL ( "currentChanged(int)" ), self.updateGui )
	
			if self.nodeParamEditor is not None :
				QtCore.QObject.disconnect ( self.nodeParamEditor, QtCore.SIGNAL ( 'changeParamName' ), self.onRenameParam )
				QtCore.QObject.disconnect ( self.nodeParamEditor, QtCore.SIGNAL ( 'changeParamLabel' ), self.onRenameParamLabel )
	
			if self.nodeCodeEditor is not None :
				QtCore.QObject.disconnect ( self.nodeCodeEditor.ui.textEdit, QtCore.SIGNAL ( 'textChanged()' ), self.onEditCode )
	
			if self.controlCodeEditor is not None :
				QtCore.QObject.disconnect ( self.controlCodeEditor.ui.textEdit, QtCore.SIGNAL ( 'textChanged()' ), self.onEditControlCode)
	
			if self.paramCodeEditor is not None :
				QtCore.QObject.disconnect ( self.paramCodeEditor.ui.textEdit, QtCore.SIGNAL ( 'textChanged()' ), self.onEditParamCode )
			
			if self.eventCodeEditor is not None :
				QtCore.QObject.disconnect ( self.eventCodeEditor.ui.textEdit, QtCore.SIGNAL ( 'textChanged()' ), self.onEditEventCode )
	
			QtCore.QObject.disconnect ( self.ui.internals_list, QtCore.SIGNAL ( 'addItem' ), self.onAddInternal )
			QtCore.QObject.disconnect ( self.ui.includes_list, QtCore.SIGNAL ( 'addItem' ), self.onAddInclude )
	
			QtCore.QObject.disconnect ( self.ui.internals_list, QtCore.SIGNAL ( 'renameItem' ), self.onRenameInternal )
			QtCore.QObject.disconnect ( self.ui.includes_list, QtCore.SIGNAL ( 'renameItem' ), self.onRenameInclude )
	
			QtCore.QObject.disconnect ( self.ui.internals_list, QtCore.SIGNAL ( 'removeItem' ), self.onRemoveInternal )
			QtCore.QObject.disconnect ( self.ui.includes_list, QtCore.SIGNAL ( 'removeItem' ), self.onRemoveInclude )
			
			QtCore.QObject.disconnect ( self.ui.handlers_list, QtCore.SIGNAL ( 'selectionChangedSignal' ), self.updateGui )
		else :
			self.ui.input_list.selectionChangedSignal.disconnect ( self.updateGui ) # onInputParamSelectionChanged )
			self.ui.output_list.selectionChangedSignal.disconnect ( self.updateGui ) # onOutputParamSelectionChanged  )
	
			self.ui.input_list.addItem.disconnect ( self.onAddParam )
			self.ui.output_list.addItem.disconnect ( self.onAddParam )
			
			self.ui.input_list.renameItem.disconnect ( self.onRenameParam )
			self.ui.output_list.renameItem.disconnect ( self.onRenameParam )
			
			self.ui.input_list.removeItem.disconnect ( self.onRemoveParam )
			self.ui.output_list.removeItem.disconnect ( self.onRemoveParam )
	
			self.ui.tabs_param_list.currentChanged.disconnect ( self.updateGui )
	
			if self.nodeParamEditor is not None :
				self.nodeParamEditor.changeParamName.disconnect ( self.onRenameParam )
				self.nodeParamEditor.changeParamName.disconnect ( self.onRenameParamLabel )
	
			if self.nodeCodeEditor is not None :
				self.nodeCodeEditor.ui.textEdit.textChanged.disconnect ( self.onEditCode )
	
			if self.controlCodeEditor is not None :
				self.controlCodeEditor.ui.textEdit.textChanged.disconnect ( self.onEditControlCode )
				
			if self.paramCodeEditor is not None :
				self.paramCodeEditor.ui.textEdit.textChanged.disconnect ( self.onEditParamCode )
				
			if self.eventCodeEditor is not None :
				self.eventCodeEditor.ui.textEdit.textChanged.disconnect ( self.onEditEventCode )
	
			self.ui.internals_list.addItem.disconnect ( self.onAddInternal )
			self.ui.includes_list.addItem.disconnect ( self.onAddInclude )
	
			self.ui.internals_list.renameItem.disconnect ( self.onRenameInternal )
			self.ui.includes_list.renameItem.disconnect ( self.onRenameInclude )
	
			self.ui.internals_list.removeItem.disconnect ( self.onRemoveInternal )
			self.ui.includes_list.removeItem.disconnect ( self.onRemoveInclude )
			
			self.ui.handlers_list.selectionChangedSignal.disconnect ( self.updateGui )
	#
	# setEditNode
	#
	def setEditNode ( self, editNode ) :
		#
		if DEBUG_MODE : print '>> NodeEditorDialog: setEditNode'
		self.editNode = editNode
	#
	#  buildGui
	#
	def buildGui ( self ) :
		# build the gui created with QtDesigner
		self.ui = Ui_NodeEditorDialog ()
		self.ui.setupUi ( self )

		if self.editNode is not None :
			#
			self.setWindowTitle ( 'NodeEditor: %s (%s)' % ( self.editNode.label, self.editNode.name ) )
			
			linkedFont = QtGui.QFont ()
			linkedFont.setItalic ( True )
			linkedBrush = QtGui.QBrush ()
			linkedBrush.setColor ( QtCore.Qt.blue )
			
			# setup loacal variables list
			for name in self.editNode.internals :
				item = QtModule.QListWidgetItem ( name )
				self.ui.internals_list.ui.listWidget.addItem ( item )
			
			# setup includes list
			for name in self.editNode.includes :
				item = QtModule.QListWidgetItem ( name )
				self.ui.includes_list.ui.listWidget.addItem ( item )
			
			# setup input params list
			for param in self.editNode.inputParams :
				item = QtModule.QListWidgetItem ( param.name )
				if self.editNode.isInputParamLinked ( param ) :
					item.setFont ( linkedFont )
					item.setForeground ( linkedBrush )
				self.ui.input_list.ui.listWidget.addItem ( item )
	
			# setup output params list
			for param in self.editNode.outputParams :
				item = QtModule.QListWidgetItem ( param.name )
				if self.editNode.isOutputParamLinked ( param ) :
					item.setFont ( linkedFont )
					item.setForeground ( linkedBrush )
				self.ui.output_list.ui.listWidget.addItem ( item )
	
			# setup input links list
			for link in self.editNode.getInputLinks () :
				item = QtModule.QListWidgetItem ( 'id=%d' % link.id  )
				if not usePySide :
					item.setData ( QtCore.Qt.UserRole, QtCore.QVariant ( int ( link.id ) ) )
				else :
					item.setData ( QtCore.Qt.UserRole, int ( link.id ) )
				self.ui.input_links_listWidget.addItem ( item )
	
			# setup output links list
			for link in self.editNode.getOutputLinks () :
				item = QtModule.QListWidgetItem ( 'id=%d' % link.id  )
				if not usePySide :
					item.setData ( QtCore.Qt.UserRole, QtCore.QVariant ( int ( link.id ) ) )
				else :
					item.setData ( QtCore.Qt.UserRole, int ( link.id ) )
				self.ui.output_links_listWidget.addItem ( item )
				
			# setup event handlers list
			if self.editNode.event_code :
				for handler in self.editNode.event_code.keys () :
					item = QtModule.QListWidgetItem ( handler )
					self.ui.handlers_list.ui.listWidget.addItem ( item )
	
			self.nodeCodeEditor = self.ui.node_code
			self.nodeCodeEditor.setNodeCode ( self.editNode.code, 'SL' )
			
			self.controlCodeEditor = self.ui.control_code
			self.controlCodeEditor.setNodeCode ( self.editNode.control_code, 'python' )
			
			self.eventCodeEditor = self.ui.event_code
			#self.eventCodeEditor.setNodeCode ( self.editNode.event_code, 'python' )
			
			self.paramCodeEditor = self.ui.param_code
			#self.paramCodeEditor.setNodeCode ( self.editNode.param_code, 'python' )

			
			self.nodePropertiesEditor = self.ui.node
			self.nodePropertiesEditor.setNode ( self.editNode )
			
			self.nodeParamEditor = self.ui.param
			
			self.nodeLinkEditor = self.ui.link

			self.ui.tabWidget.setCurrentIndex ( TAB_NODE_CODE )
			self.ui.toolBox.setCurrentIndex ( IDX_INTERNALS )
			self.ui.tabs_param_list.setCurrentIndex ( 0 ) # input param tab
	#
	# updateGui
	#
	def updateGui ( self, foo_param = None ) :
		#
		if self.editNode is not None :
			if DEBUG_MODE : print '>> NodeEditorDialog::updateGui'
				
			self.disconnectSignals ()
			
			self.eventCodeEditor.setNodeCode ( None, 'python' )
			self.paramCodeEditor.setNodeCode ( None, 'python' )
			
			idx = self.ui.toolBox.currentIndex ()
			
			if idx == IDX_PARAM :
				# Parameters
				tab_idx = self.ui.tabs_param_list.currentIndex ()
				param = None

				if tab_idx == 0 :
					# input parameters
					list_item = self.ui.input_list.ui.listWidget.currentItem ()

					if list_item is not None :
						param = self.editNode.getInputParamByName ( str ( list_item.text () ) )

				elif tab_idx == 1 :
					# output parametrs
					list_item = self.ui.output_list.ui.listWidget.currentItem ()

					if list_item is not None :
						param = self.editNode.getOutputParamByName ( str ( list_item.text () ) )

				self.nodeParamEditor.setParam ( param )
				if param is not None and param.type == 'control' :
					self.paramCodeEditor.setNodeCode ( param.code, 'python' )
					#print '*** set (%s).param.code :' % param.label
					#print param.code

			elif idx == IDX_HANDLERS :
				handler_item = self.ui.handlers_list.ui.listWidget.currentItem ()
				if handler_item is not None :
					handler = str ( handler_item.text () )
					handler_code = self.editNode.event_code [ handler ]
					self.eventCodeEditor.setNodeCode ( handler_code, 'python' )
				#else :
				#  print '** no selection in handlers_list'

			elif idx == IDX_LINKS :
				inputLinkSelected = False
				link_id = None
				tab_idx = self.ui.tabs_links_list.currentIndex ()

				if tab_idx == 0 :
					# input links
					inputLinkSelected = True
					links_item = self.ui.input_links_listWidget.currentItem ()
					if links_item is not None : 
						if  not usePyQt5 :
							( link_id, ok ) = links_item.data ( QtCore.Qt.UserRole ).toInt ()
						else :
							link_id = links_item.data ( QtCore.Qt.UserRole )

				else :
					# output links
					links_item = self.ui.output_links_listWidget.currentItem ()
					if links_item is not None :
						if  not usePyQt5 : 
							( link_id, ok ) = links_item.data ( QtCore.Qt.UserRole ).toInt ()
						else:
							link_id = links_item.data ( QtCore.Qt.UserRole )

				if link_id is not None :
					link = None

					if inputLinkSelected : link = self.editNode.getInputLinkByID ( link_id )
					else                 : link = self.editNode.getOutputLinkByID ( link_id )

					if link is not None :
						( srcNode, srcParam ) = link.getSrc ()
						( dstNode, dstParam ) = link.getDst ()
						self.nodeLinkEditor.ui.src_node_lineEdit.setText ( srcNode.label )
						self.nodeLinkEditor.ui.src_param_lineEdit.setText ( srcParam.label )
						self.nodeLinkEditor.ui.src_id_lineEdit.setText ( str ( srcNode.id ) )
						self.nodeLinkEditor.ui.dst_node_lineEdit.setText ( dstNode.label )
						self.nodeLinkEditor.ui.dst_param_lineEdit.setText ( dstParam.label )
						self.nodeLinkEditor.ui.dst_id_lineEdit.setText ( str ( dstNode.id ) )
			self.connectSignals ()
	#
	# onToolBoxIndexChanged
	#
	def onToolBoxIndexChanged ( self, idx ) :
		if DEBUG_MODE : print '>> NodeEditorDialog::onToolBoxIndexChanged (idx = %d)' % idx
		#
		#self.disconnectSignals ()

		if idx == IDX_PARAM :
			# Input, Output Parameters
			self.ui.tabWidget.setCurrentIndex ( TAB_PARAM )

		elif idx == IDX_LINKS :
			# Input, Output Links
			self.ui.tabWidget.setCurrentIndex ( TAB_LINK_INFO )

		elif idx == IDX_INTERNALS :
			# Includes, Local Names, Code
			self.ui.tabWidget.setCurrentIndex ( TAB_NODE_CODE )
			
		elif idx == IDX_HANDLERS :
			# Event Handlers code
			self.ui.tabWidget.setCurrentIndex ( TAB_EVENT_CODE )

		#self.connectSignals ()
		self.updateGui ()
	#
	# onInputParamSelectionChanged
	#
	def onInputParamSelectionChanged ( self, paramName ) :
		#
		if DEBUG_MODE : print '>> NodeEditorDialog::onInputParamSelectionChanged (%s)' % paramName
		param = self.editNode.getInputParamByName ( str ( paramName ) )
		self.nodeParamEditor.setParam ( param )
	#
	# onOutputParamSelectionChanged
	#
	def onOutputParamSelectionChanged ( self, paramName ) :
		#
		if DEBUG_MODE : print '>> NodeEditorDialog::onOutputParamSelectionChanged (%s)' % paramName
		param = self.editNode.getOutputParamByName ( str ( paramName ) )
		self.nodeParamEditor.setParam ( param )
	#
	# onCodeListIndexChanged
	#
	def onCodeListIndexChanged ( self, idx ) :
		# if DEBUG_MODE : print '>> NodeEditorDialog: onCodeListIndexChanged idx = %d' % idx
		self.updateGui ()
	#
	# onRemoveInternal
	#
	def onRemoveInternal ( self, internal ) :
		#
		internalsListWidget = self.ui.internals_list.ui.listWidget
		self.editNode.internals.remove ( internal )
		item = internalsListWidget.currentItem ()
		internalsListWidget.takeItem ( internalsListWidget.currentRow () )
		internalsListWidget.removeItemWidget ( item )
		internalsListWidget.clearSelection ()
		internalsListWidget.setCurrentItem ( None )
	#
	# onRemoveInclude
	#
	def onRemoveInclude ( self, include ) :
		#
		includesListWidget = self.ui.includes_list.ui.listWidget
		self.editNode.includes.remove ( include )
		item = includesListWidget.currentItem()
		includesListWidget.takeItem ( includesListWidget.currentRow () )
		includesListWidget.removeItemWidget ( item )
		includesListWidget.clearSelection ()
		includesListWidget.setCurrentItem ( None )
	#
	# onRemoveParam
	#
	def onRemoveParam ( self, paramName ) :
		#
		isInputParam = False
		tab_idx = self.ui.tabs_param_list.currentIndex ()
		if tab_idx == 0 : isInputParam = True
		param = None
		paramList = None
		linkList = None

		if isInputParam :
			param = self.editNode.getInputParamByName ( paramName )
			paramList = self.ui.input_list
			linkList = self.ui.input_links_listWidget
		else :
			param = self.editNode.getOutputParamByName ( paramName )
			paramList = self.ui.output_list
			linkList = self.ui.output_links_listWidget

		removedLinks = self.editNode.removeParam ( param )

		# remove item from links list and node network
		for link in removedLinks :
			item_name = 'id=%d' % link.id
			item = linkList.findItems ( item_name, QtCore.Qt.MatchExactly )[0]
			linkList.takeItem ( linkList.row ( item ) )
			link.nodenet.removeLink ( link )

		# remove item from param list
		item = paramList.ui.listWidget.currentItem ()
		paramList.ui.listWidget.takeItem ( paramList.ui.listWidget.currentRow () )
		paramList.ui.listWidget.removeItemWidget ( item )
		paramList.ui.listWidget.clearSelection ()
		paramList.ui.listWidget.setCurrentItem ( None )

		# remove item from code (invalidate code)
	#
	# onRenameInternal
	#
	def onRenameInternal ( self, oldName, newName ) :
		#
		internalsListWidget = self.ui.internals_list.ui.listWidget
		from core.meCommon import getUniqueName
		idx = self.editNode.internals.index ( oldName )
		newName = getUniqueName ( newName, self.editNode.internals )
		self.editNode.internals [ idx ] = newName

		item = internalsListWidget.findItems ( oldName, QtCore.Qt.MatchExactly )[0]
		item.setText ( newName )
		self.ui.internals_list.setName ( newName )
		internalsListWidget.clearSelection ()
		internalsListWidget.setCurrentItem ( item )
	#
	# onRenameInclude
	#
	def onRenameInclude ( self, oldName, newName ) :
		#
		includesListWidget = self.ui.includes_list.ui.listWidget
		from core.meCommon import getUniqueName
		idx = self.editNode.includes.index ( oldName )
		newName = getUniqueName ( newName, self.editNode.includes )
		self.editNode.includes [ idx ] = newName

		item = includesListWidget.findItems ( oldName, QtCore.Qt.MatchExactly )[0]
		item.setText ( newName )
		self.ui.includes_list.setName ( newName )
		includesListWidget.clearSelection ()
		includesListWidget.setCurrentItem ( item )
	#
	# onRenameParam
	#
	def onRenameParam ( self, oldName, newName ) :
		#
		isInputParam = False
		tab_idx = self.ui.tabs_param_list.currentIndex ()
		if tab_idx == 0 : isInputParam = True
		param = None
		paramList = None
		if isInputParam :
			param = self.editNode.getInputParamByName ( oldName )
			paramList = self.ui.input_list
		else :
			param = self.editNode.getOutputParamByName ( oldName )
			paramList = self.ui.output_list

		self.editNode.renameParamName ( param, newName )

		item = paramList.ui.listWidget.findItems ( oldName, QtCore.Qt.MatchExactly )[0]
		item.setText ( param.name )
		paramList.setName ( param.name )
		paramList.ui.listWidget.clearSelection ()
		paramList.ui.listWidget.setCurrentItem ( item )
	#
	# onRenameParamLabel
	#
	def onRenameParamLabel ( self, oldName, newName ) :
		#
		param = self.nodeParamEditor.param
		self.editNode.renameParamLabel ( param, newName )
		self.nodeParamEditor.ui.label_lineEdit.setText ( param.label )
	#
	# onAddInternal
	#
	def onAddInternal ( self, newName ) :
		#
		if DEBUG_MODE : print '>> NodeEditorDialog::onAddInternal (%s) ' % (newName)
		# name can be changed to be unique
		newName = self.editNode.addInternal ( newName )
		internalsListWidget = self.ui.internals_list.ui.listWidget
		internalsListWidget.addItem ( newName )
		internalsListWidget.setCurrentItem ( internalsListWidget.findItems ( newName, QtCore.Qt.MatchExactly )[0] )
	#
	# onAddInclude
	#
	def onAddInclude ( self, newName ) :
		#
		if DEBUG_MODE : print '>> NodeEditorDialog::onAddInclude (%s) ' % (newName)
		# name can be changed to be unique
		newName = self.editNode.addInclude ( newName )
		includesListWidget = self.ui.includes_list.ui.listWidget
		includesListWidget.addItem ( newName )
		includesListWidget.setCurrentItem ( includesListWidget.findItems ( newName, QtCore.Qt.MatchExactly )[0] )
	#
	# onAddParam
	#
	def onAddParam ( self, newName ) :
		#
		if DEBUG_MODE : print '>> NodeEditorDialog::onAddParam (%s) ' % (newName)
		isInputParam = False
		paramType = None
		isRibParam = ( self.editNode.type in VALID_RIB_NODE_TYPES )
		tab_idx = self.ui.tabs_param_list.currentIndex ()
		if tab_idx == 0 : isInputParam = True
		# ask user about param type
		typeDialog = QtModule.QDialog () # Qt.MSWindowsFixedSizeDialogHint
		typeDialog.setModal ( True )

		typeDialog.setWindowTitle ( 'Parameter Type' )
		typeDialog.resize (180, 100 )
		sizePolicy = QtModule.QSizePolicy ( QtModule.QSizePolicy.Fixed, QtModule.QSizePolicy.Fixed )
		sizePolicy.setHorizontalStretch ( 0 )
		sizePolicy.setVerticalStretch ( 0 )
		sizePolicy.setHeightForWidth ( typeDialog.sizePolicy().hasHeightForWidth() )
		typeDialog.setSizePolicy ( sizePolicy )
		typeDialog.setSizeGripEnabled ( False )

		typeDialog.verticalLayout = QtGui.QVBoxLayout ( typeDialog )
		typeDialog.verticalLayout.setSizeConstraint ( QtGui.QLayout.SetMinimumSize )
		typeDialog.type_comboBox = QtGui.QComboBox ( typeDialog )
		
		for label in VALID_PARAM_TYPES : typeDialog.type_comboBox.addItem ( label )
		
		typeDialog.verticalLayout.addWidget ( typeDialog.type_comboBox )

		typeDialog.btnBox = QtModule.QDialogButtonBox ( QtModule.QDialogButtonBox.Ok | QtModule.QDialogButtonBox.Cancel, parent = typeDialog )
		typeDialog.btnBox.setCenterButtons ( True )
		typeDialog.verticalLayout.addWidget ( typeDialog.btnBox )

		if usePyQt4 :
			QtCore.QObject.connect ( typeDialog.btnBox, QtCore.SIGNAL ( 'accepted()' ), typeDialog.accept )
			QtCore.QObject.connect ( typeDialog.btnBox, QtCore.SIGNAL ( 'rejected()' ), typeDialog.reject )
		else :
			typeDialog.btnBox,accepted.connect ( typeDialog.accept )
			typeDialog.btnBox,rejected.connect ( typeDialog.reject )

		if typeDialog.exec_() == QtModule.QDialog.Accepted  :
			paramType = str ( typeDialog.type_comboBox.currentText () )

			if DEBUG_MODE : print '>> NodeEditorDialog::onAddParam typeDialog Accepted (%s)' % paramType
			# create empty xml node parameter
			dom = QtXml.QDomDocument ( newName )
			xmlnode = dom.createElement( 'property' )

			xmlnode.setAttribute ( 'name', newName )
			xmlnode.setAttribute ( 'label', newName )
			xmlnode.setAttribute ( 'type', paramType )
			param = createParamFromXml ( xmlnode, isRibParam, isInputParam )
			item = QtModule.QListWidgetItem ( param.name )

			paramListWidget = self.ui.input_list.ui.listWidget

			if isInputParam :
				self.editNode.addInputParam ( param )
			else :
				self.editNode.addOutputParam ( param )
				paramListWidget = self.ui.output_list.ui.listWidget

			paramListWidget.addItem ( param.name )
			paramListWidget.setCurrentItem ( paramListWidget.findItems ( param.name, QtCore.Qt.MatchExactly )[0] )
			#self.nodeParamEditor.setParam ( param )
	#
	# onEditCode
	#
	def onEditCode ( self ) :
		#
		if DEBUG_MODE : print '>> NodeEditorDialog::onEditCode'
		if self.nodeCodeEditor is not None :
			#self.nodeCodeEditor.ui.textEdit
			self.editNode.code = str ( self.nodeCodeEditor.ui.textEdit.toPlainText () )
	#
	# onEditControlCode
	#
	def onEditControlCode ( self ) :
		#
		if DEBUG_MODE : print '>> NodeEditorDialog::onEditControlCode'
		if self.controlCodeEditor is not None :
			self.editNode.control_code = str ( self.controlCodeEditor.ui.textEdit.toPlainText () )
	#
	# onEditParamCode
	#
	def onEditParamCode ( self ) :
		#
		if DEBUG_MODE : print '>> NodeEditorDialog::onEditParamCode'
		if self.paramCodeEditor is not None :
			param = self.nodeParamEditor.param
			if param is not None and param.type == 'control' :
				param.code = str ( self.paramCodeEditor.ui.textEdit.toPlainText () )
				#print '*** set (%s).param.code :' % param.label
				#print param.code
	#
	# onEditEventCode
	#
	def onEditEventCode ( self ) :
		#
		if DEBUG_MODE : print '>> NodeEditorDialog::onEditEventCode'
		if self.eventCodeEditor is not None :
			handler_item = self.ui.handlers_list.ui.listWidget.currentItem ()
			if handler_item is not None :
				handler = str ( handler_item.text () )
				self.editNode.event_code [ handler ] = str ( self.eventCodeEditor.ui.textEdit.toPlainText () )

	#
	# Ignore default Enter press event
	#
	def keyPressEvent ( self, event  ) :
		#
		#if DEBUG_MODE : print '>> NodeEditorDialog::keyPressEvent'
		if event.key () == QtCore.Qt.Key_Enter or event.key () == QtCore.Qt.Key_Return :
			event.ignore ()
		else:
			QtModule.QDialog.keyPressEvent ( self, event )
	#
	# accept
	#
	def accept ( self ) :
		#
		if DEBUG_MODE : print '>> NodeEditorDialog::accept'
		self.done ( QtModule.QDialog.Accepted )
Esempio n. 3
0
  def buildGui ( self ) :
    # build the gui created with QtDesigner
    self.ui = Ui_NodeEditorDialog ()
    self.ui.setupUi ( self )

    if self.editNode is not None :
      #
      self.setWindowTitle ( 'NodeEditor: %s (%s)' % ( self.editNode.label, self.editNode.name ) )
      
      linkedFont = QtGui.QFont ()
      linkedFont.setItalic ( True )
      linkedBrush = QtGui.QBrush ()
      linkedBrush.setColor ( QtCore.Qt.blue )
      
      # setup loacal variables list
      for name in self.editNode.internals :
        item = QtGui.QListWidgetItem ( name )
        self.ui.internals_list.ui.listWidget.addItem ( item )
      
      # setup includes list
      for name in self.editNode.includes :
        item = QtGui.QListWidgetItem ( name )
        self.ui.includes_list.ui.listWidget.addItem ( item )
      
      # setup input params list
      for param in self.editNode.inputParams :
        item = QtGui.QListWidgetItem ( param.name )
        if self.editNode.isInputParamLinked ( param ) :
          item.setFont ( linkedFont )
          item.setForeground ( linkedBrush )
        self.ui.input_list.ui.listWidget.addItem ( item )
  
      # setup output params list
      for param in self.editNode.outputParams :
        item = QtGui.QListWidgetItem ( param.name )
        if self.editNode.isOutputParamLinked ( param ) :
          item.setFont ( linkedFont )
          item.setForeground ( linkedBrush )
        self.ui.output_list.ui.listWidget.addItem ( item )
  
      # setup input links list
      for link in self.editNode.getInputLinks () :
        item = QtGui.QListWidgetItem ( 'id=%d' % link.id  )
        item.setData ( QtCore.Qt.UserRole, QVariant ( int ( link.id ) ) )
        self.ui.input_links_listWidget.addItem ( item )
  
      # setup output links list
      for link in self.editNode.getOutputLinks () :
        item = QtGui.QListWidgetItem ( 'id=%d' % link.id  )
        item.setData ( QtCore.Qt.UserRole, QVariant ( int ( link.id ) ) )
        self.ui.output_links_listWidget.addItem ( item )
        
      # setup event handlers list
      if self.editNode.event_code :
        for handler in self.editNode.event_code.keys () :
          item = QtGui.QListWidgetItem ( handler )
          self.ui.handlers_list.ui.listWidget.addItem ( item )
  
      self.nodeCodeEditor = self.ui.node_code
      self.nodeCodeEditor.setNodeCode ( self.editNode.code, 'SL' )
      
      self.controlCodeEditor = self.ui.control_code
      self.controlCodeEditor.setNodeCode ( self.editNode.control_code, 'python' )
      
      self.eventCodeEditor = self.ui.event_code
      #self.eventCodeEditor.setNodeCode ( self.editNode.event_code, 'python' )
      
      self.paramCodeEditor = self.ui.param_code
      #self.paramCodeEditor.setNodeCode ( self.editNode.param_code, 'python' )

      
      self.nodePropertiesEditor = self.ui.node
      self.nodePropertiesEditor.setNode ( self.editNode )
      
      self.nodeParamEditor = self.ui.param
      
      self.nodeLinkEditor = self.ui.link

      self.ui.tabWidget.setCurrentIndex ( TAB_NODE_CODE )
      self.ui.toolBox.setCurrentIndex ( IDX_INTERNALS )
      self.ui.tabs_param_list.setCurrentIndex ( 0 ) # input param tab
Esempio n. 4
0
	def buildGui ( self ) :
		# build the gui created with QtDesigner
		self.ui = Ui_NodeEditorDialog ()
		self.ui.setupUi ( self )

		if self.editNode is not None :
			#
			self.setWindowTitle ( 'NodeEditor: %s (%s)' % ( self.editNode.label, self.editNode.name ) )
			
			linkedFont = QtGui.QFont ()
			linkedFont.setItalic ( True )
			linkedBrush = QtGui.QBrush ()
			linkedBrush.setColor ( QtCore.Qt.blue )
			
			# setup loacal variables list
			for name in self.editNode.internals :
				item = QtModule.QListWidgetItem ( name )
				self.ui.internals_list.ui.listWidget.addItem ( item )
			
			# setup includes list
			for name in self.editNode.includes :
				item = QtModule.QListWidgetItem ( name )
				self.ui.includes_list.ui.listWidget.addItem ( item )
			
			# setup input params list
			for param in self.editNode.inputParams :
				item = QtModule.QListWidgetItem ( param.name )
				if self.editNode.isInputParamLinked ( param ) :
					item.setFont ( linkedFont )
					item.setForeground ( linkedBrush )
				self.ui.input_list.ui.listWidget.addItem ( item )
	
			# setup output params list
			for param in self.editNode.outputParams :
				item = QtModule.QListWidgetItem ( param.name )
				if self.editNode.isOutputParamLinked ( param ) :
					item.setFont ( linkedFont )
					item.setForeground ( linkedBrush )
				self.ui.output_list.ui.listWidget.addItem ( item )
	
			# setup input links list
			for link in self.editNode.getInputLinks () :
				item = QtModule.QListWidgetItem ( 'id=%d' % link.id  )
				if not usePySide :
					item.setData ( QtCore.Qt.UserRole, QtCore.QVariant ( int ( link.id ) ) )
				else :
					item.setData ( QtCore.Qt.UserRole, int ( link.id ) )
				self.ui.input_links_listWidget.addItem ( item )
	
			# setup output links list
			for link in self.editNode.getOutputLinks () :
				item = QtModule.QListWidgetItem ( 'id=%d' % link.id  )
				if not usePySide :
					item.setData ( QtCore.Qt.UserRole, QtCore.QVariant ( int ( link.id ) ) )
				else :
					item.setData ( QtCore.Qt.UserRole, int ( link.id ) )
				self.ui.output_links_listWidget.addItem ( item )
				
			# setup event handlers list
			if self.editNode.event_code :
				for handler in self.editNode.event_code.keys () :
					item = QtModule.QListWidgetItem ( handler )
					self.ui.handlers_list.ui.listWidget.addItem ( item )
	
			self.nodeCodeEditor = self.ui.node_code
			self.nodeCodeEditor.setNodeCode ( self.editNode.code, 'SL' )
			
			self.controlCodeEditor = self.ui.control_code
			self.controlCodeEditor.setNodeCode ( self.editNode.control_code, 'python' )
			
			self.eventCodeEditor = self.ui.event_code
			#self.eventCodeEditor.setNodeCode ( self.editNode.event_code, 'python' )
			
			self.paramCodeEditor = self.ui.param_code
			#self.paramCodeEditor.setNodeCode ( self.editNode.param_code, 'python' )

			
			self.nodePropertiesEditor = self.ui.node
			self.nodePropertiesEditor.setNode ( self.editNode )
			
			self.nodeParamEditor = self.ui.param
			
			self.nodeLinkEditor = self.ui.link

			self.ui.tabWidget.setCurrentIndex ( TAB_NODE_CODE )
			self.ui.toolBox.setCurrentIndex ( IDX_INTERNALS )
			self.ui.tabs_param_list.setCurrentIndex ( 0 ) # input param tab