Esempio n. 1
0
class StoryView(SceneEditorModule):
    name = 'story_view'
    dependency = ['mock']

    def onLoad(self):
        self.container = self.requestDockWindow(title='Story')
        self.widget = widget = self.container.addWidget(StoryGraphWidget())

        # self.canvas = addWidgetWithLayout(
        # 	MOAIEditCanvas( window.containerGraph )
        # )
        self.delegate = MOAILuaDelegate(self)
        self.delegate.load(_getModulePath('StoryView.lua'))

        # self.updateTimer        = self.container.startTimer( 60, self.onUpdateTimer )
        self.updatePending = False
        self.previewing = False
        self.previewUpdateTimer = False

    def onStart(self):
        self.container.show()

    def onSetFocus(self):
        self.getModule('scene_editor').setFocus()
        self.container.show()
        self.container.setFocus()

    def onUpdateTimer(self):
        if self.updatePending == True:
            self.updatePending = False
Esempio n. 2
0
    def onLoad(self):
        self.mainToolBar = self.addToolBar(
            'deckcanvas_tools',
            self.getMainWindow().requestToolBar('deckcanvas_tools'))

        self.addTool('deckcanvas_tools/tool_pen',
                     widget=SceneToolButton('deckcanvas_pen',
                                            label='Deck Canvas Editor',
                                            icon='deckcanvas/pen'))

        self.addTool('deckcanvas_tools/toggle_item_bounds_visible',
                     label='Toggle Item Bounds',
                     icon='deckcanvas/layer',
                     type='check')

        self.findTool('deckcanvas_tools/toggle_item_bounds_visible').setValue(
            True)

        self.delegate = MOAILuaDelegate(self)
        self.delegate.load(_getModulePath('DeckCanvasEditor.lua'))
        signals.connect('selection.changed', self.onSelectionChanged)

        self.toolWindow = self.requestToolWindow('DeckCanvasEditor',
                                                 title='DeckCanvas',
                                                 size=(120, 40),
                                                 minSize=(120, 40))

        self.targetCanvas = None
Esempio n. 3
0
    def onLoad(self):
        #UI
        self.windowTitle = 'Layers'
        self.window = self.requestDockWindow('LayerManager',
                                             title='Layers',
                                             size=(120, 120),
                                             minSize=(120, 120),
                                             dock='left')

        #Components
        self.treeFilter = self.window.addWidget(GenericTreeFilter(),
                                                expanding=False)
        self.tree = self.window.addWidget(
            LayerTreeWidget(self.window,
                            multiple_selection=False,
                            sorting=False,
                            editable=True,
                            drag_mode='internal'))
        self.tree.hasSoloLayer = False
        self.treeFilter.setTargetTree(self.tree)

        self.tool = self.addToolBar('layer_manager', self.window.addToolBar())
        self.delegate = MOAILuaDelegate(self)
        self.delegate.load(_getModulePath('LayerManager.lua'))

        self.addTool('layer_manager/add', label='add', icon='add')
        self.addTool('layer_manager/remove', label='remove', icon='remove')
        self.addTool('layer_manager/up', label='up', icon='arrow-up')
        self.addTool('layer_manager/down', label='down', icon='arrow-down')

        #SIGNALS
        signals.connect('moai.clean', self.onMoaiClean)
Esempio n. 4
0
    def onLoad(self):
        #UI
        self.window = self.requestDockWindow('SceneReferenceEditor',
                                             title='Scene Reference',
                                             size=(120, 120),
                                             minSize=(120, 120),
                                             dock='right')

        #Components
        self.tree = self.window.addWidget(
            SceneReferenceTreeWidget(multiple_selection=True,
                                     editable=True,
                                     drag_mode='internal'))

        self.tool = self.addToolBar('global_object_manager',
                                    self.window.addToolBar())
        self.delegate = MOAILuaDelegate(self)
        self.delegate.load(_getModulePath('SceneReferenceEditor.lua'))

        self.addTool('global_object_manager/add_group',
                     label='Add Group',
                     icon='add_folder')
        self.addTool('global_object_manager/add', label='Add', icon='add')
        self.addTool('global_object_manager/remove',
                     label='Remove',
                     icon='remove')

        if self.getModule('introspector'):
            import SceneReferenceNodeEditor

        registerSearchEnumerator(globalObjectNameSearchEnumerator)
        registerSearchEnumerator(globalObjectSearchEnumerator)
Esempio n. 5
0
    def onLoad(self):
        #UI
        self.window = self.requestDockWindow('GlobalObjectManager',
                                             title='Global Objects',
                                             size=(120, 120),
                                             minSize=(120, 120),
                                             dock='left')

        #Components
        self.treeFilter = self.window.addWidget(GenericTreeFilter(),
                                                expanding=False)
        self.tree = self.window.addWidget(
            GlobalObjectTreeWidget(multiple_selection=True,
                                   editable=True,
                                   drag_mode='internal'))
        self.treeFilter.setTargetTree(self.tree)

        self.tool = self.addToolBar('global_object_manager',
                                    self.window.addToolBar())
        self.delegate = MOAILuaDelegate(self)
        self.delegate.load(_getModulePath('GlobalObjectManager.lua'))

        # self.creatorMenu=self.addMenu(
        # 	'global_object_create',
        # 	{ 'label':'Create Global Object' }
        # 	)

        self.addTool('global_object_manager/add', label='Add', icon='add')
        self.addTool('global_object_manager/remove',
                     label='Remove',
                     icon='remove')
        self.addTool('global_object_manager/clone',
                     label='Clone',
                     icon='clone')
        self.addTool('global_object_manager/add_group',
                     label='Add Group',
                     icon='add_folder')
        self.addTool('global_object_manager/----')
        self.addTool('global_object_manager/refresh',
                     label='Refresh',
                     icon='refresh')

        # self.addMenuItem( 'main/find/find_global_object',
        # 	dict( label = 'Find In Global Objects', shortcut = 'ctrl+alt+f' ) )

        #SIGNALS
        signals.connect('moai.clean', self.onMoaiClean)

        signals.connect('global_object.added', self.onObjectAdded)
        signals.connect('global_object.removed', self.onObjectRemoved)
        signals.connect('global_object.renamed', self.onObjectRenamed)

        if self.getModule('introspector'):
            from . import GlobalObjectNodeEditor

        registerSearchEnumerator(globalObjectNameSearchEnumerator)
        registerSearchEnumerator(globalObjectSearchEnumerator)
Esempio n. 6
0
    def onLoad(self):
        self.container = self.requestDockWindow(title='Story')
        self.widget = widget = self.container.addWidget(StoryGraphWidget())

        # self.canvas = addWidgetWithLayout(
        # 	MOAIEditCanvas( window.containerGraph )
        # )
        self.delegate = MOAILuaDelegate(self)
        self.delegate.load(_getModulePath('StoryView.lua'))

        # self.updateTimer        = self.container.startTimer( 60, self.onUpdateTimer )
        self.updatePending = False
        self.previewing = False
        self.previewUpdateTimer = False
Esempio n. 7
0
	def onLoad( self ):
		#UI
		self.window = self.requestDockWindow( 'SceneReferenceEditor',
			title     = 'Scene Reference',
			size      = (120,120),
			minSize   = (120,120),
			dock      = 'right'
			)

		#Components
		self.tree = self.window.addWidget( 
				SceneReferenceTreeWidget( 
					multiple_selection = True,
					editable           = True,
					drag_mode          = 'internal'
				)
			)

		self.tool = self.addToolBar( 'global_object_manager', self.window.addToolBar() )
		self.delegate = MOAILuaDelegate( self )
		self.delegate.load( _getModulePath( 'SceneReferenceEditor.lua' ) )

		self.addTool( 'global_object_manager/add_group',    label = 'Add Group', icon = 'add_folder' )
		self.addTool( 'global_object_manager/add',    label = 'Add', icon = 'add' )
		self.addTool( 'global_object_manager/remove', label = 'Remove', icon = 'remove' )
		
		if self.getModule('introspector'):
			import SceneReferenceNodeEditor

		registerSearchEnumerator( globalObjectNameSearchEnumerator )
		registerSearchEnumerator( globalObjectSearchEnumerator )
Esempio n. 8
0
    def loadCommandScriptLua(self, path):
        print('loading lua command script', path)
        mtime0 = self.scriptTimestamps.get(path, 0)
        mtime1 = os.path.getmtime(path)
        if mtime1 <= mtime0: return  #ignore
        self.scriptTimestamps[path] = mtime1

        delegate = MOAILuaDelegate(self)
        if not delegate.load(path):
            return False

        name = delegate.getEnv('name')
        run = delegate.getEnv('run')
        if name and run:
            cmdObject = FunctionScriptCommand(run)
            ScriptCommandRegistry.get().registerCommand(name, cmdObject)
        return True
Esempio n. 9
0
	def onLoad( self ):
		#UI
		self.window = self.requestDockWindow( 'GlobalObjectManager',
			title     = 'Global Objects',
			size      = (120,120),
			minSize   = (120,120),
			dock      = 'left'
			)

		#Components
		self.tree = self.window.addWidget( 
				GlobalObjectTreeWidget( 
					multiple_selection = True,
					editable           = True,
					drag_mode          = 'internal'
				)
			)

		self.tool = self.addToolBar( 'global_object_manager', self.window.addToolBar() )
		self.delegate = MOAILuaDelegate( self )
		self.delegate.load( _getModulePath( 'GlobalObjectManager.lua' ) )

		# self.creatorMenu=self.addMenu(
		# 	'global_object_create',
		# 	{ 'label':'Create Global Object' }
		# 	)

		self.addTool( 'global_object_manager/add',    label = 'Add', icon = 'add' )
		self.addTool( 'global_object_manager/remove', label = 'Remove', icon = 'remove' )
		self.addTool( 'global_object_manager/clone',  label = 'Clone', icon = 'clone' )
		self.addTool( 'global_object_manager/add_group',    label = 'Add Group', icon = 'add_folder' )
		self.addTool( 'global_object_manager/----' )
		self.addTool( 'global_object_manager/refresh',    label = 'Refresh', icon = 'refresh' )
		
		# self.addMenuItem( 'main/find/find_global_object', 
		# 	dict( label = 'Find In Global Objects', shortcut = 'ctrl+alt+f' ) )

		#SIGNALS
		signals.connect( 'moai.clean', self.onMoaiClean )

		signals.connect( 'global_object.added',   self.onObjectAdded )
		signals.connect( 'global_object.removed', self.onObjectRemoved )
		signals.connect( 'global_object.renamed', self.onObjectRenamed )

		if self.getModule('introspector'):
			import GlobalObjectNodeEditor

		registerSearchEnumerator( globalObjectNameSearchEnumerator )
		registerSearchEnumerator( globalObjectSearchEnumerator )
Esempio n. 10
0
	def onLoad( self ):
		#UI
		self.windowTitle = 'Layers'
		self.window = self.requestDockWindow( 'LayerManager',
			title     = 'Layers',
			size      = (120,120),
			minSize   = (120,120),
			dock      = 'left'
			)

		#Components
		self.treeFilter = self.window.addWidget( GenericTreeFilter() )
		self.tree = self.window.addWidget( 
			LayerTreeWidget(
				self.window,
				multiple_selection = False,
				sorting            = False,
				editable           = True,
				drag_mode          = 'internal'
				)
			)
		self.tree.hasSoloLayer = False
		self.treeFilter.setTargetTree( self.tree )

		self.tool = self.addToolBar( 'layer_manager', self.window.addToolBar() )
		self.delegate = MOAILuaDelegate( self )
		self.delegate.load( _getModulePath( 'LayerManager.lua' ) )

		self.addTool( 'layer_manager/add',    label = 'add',  icon = 'add' )
		self.addTool( 'layer_manager/remove', label = 'remove', icon = 'remove' )
		self.addTool( 'layer_manager/up',     label = 'up', icon = 'arrow-up' )
		self.addTool( 'layer_manager/down',   label = 'down', icon = 'arrow-down' )

		
		#SIGNALS
		signals.connect( 'moai.clean', self.onMoaiClean )
Esempio n. 11
0
class SceneGraphEditor( SceneEditorModule ):
	def __init__(self):
		super( SceneGraphEditor, self ).__init__()
		self.sceneDirty = False
		self.activeSceneNode  = None
		self.refreshScheduled = False
		self.previewing       = False
		self.workspaceState   = None
		
	def getName( self ):
		return 'scenegraph_editor'

	def getDependency( self ):
		return [ 'scene_editor', 'mock' ]

	def onLoad( self ):
		#UI
		self.windowTitle = 'Scenegraph'
		self.container = self.requestDockWindow( 'SceneGraphEditor',
			title     = 'Scenegraph',
			size      = (200,200),
			minSize   = (200,200),
			dock      = 'left'
			)

		#Components
		self.treeFilter = self.container.addWidget(
			GenericTreeFilter(
				self.container
			),
			expanding = False
		)
		self.tree = self.container.addWidget( 
				SceneGraphTreeWidget( 
					self.container,
					sorting  = True,
					editable = True,
					multiple_selection = True,
					drag_mode = 'internal'
				)
			)
		self.treeFilter.setTargetTree( self.tree )
		self.tree.module = self
		self.tool = self.addToolBar( 'scene_graph', self.container.addToolBar() )
		self.delegate = MOAILuaDelegate( self )
		self.delegate.load( getModulePath( 'SceneGraphEditor.lua' ) )

		self.entityCreatorMenu=self.addMenu(
			'main/scene/entity_create',
			{ 'label':'Create Entity' }
			)

		self.componentCreatorMenu=self.addMenu(
			'main/scene/component_create',
			{ 'label':'Create Component' }
			)


		#menu
		self.addMenuItem(
			'main/file/open_scene', 
			dict( label = 'Open Scene', shortcut = 'ctrl+shift+o' )
			)

		self.addMenuItem( 'main/file/close_scene', 
			dict( label = 'Close Scene', shortcut = 'Ctrl+W' )
			)
		self.addMenuItem( 'main/scene/save_scene',
			dict( label = 'Save', shortcut = 'Ctrl+S' )
			)
		self.addMenuItem( 'main/scene/locate_scene_asset',
			dict( label = 'Locate Scene Asset' )
			)

		self.addMenu( 'main/scene/----' )
		
		self.addMenu( 'component_context', dict( label = 'Selected Component' ) )
		self.addMenuItem( 'component_context/remove_component', 
			dict( label = 'Remove' )
			)

		self.addMenuItem( 'component_context/----' )
		
		self.addMenuItem( 'component_context/copy_component', 
			dict( label = 'Copy' )
			)
		self.addMenuItem( 'component_context/paste_component', 
			dict( label = 'Paste Component Here' )
			)
		self.addMenuItem( 'component_context/----' )
		
		self.addMenuItem( 'component_context/move_component_up', 
			dict( label = 'Move Up' )
			)

		self.addMenuItem( 'component_context/move_component_down', 
			dict( label = 'Move Down' )
			)
		

		self.addMenu( 'main/entity', dict( label = 'Entity' ) )
		self.addMenuItem( 'main/entity/add_empty_entity',    dict( label = 'Create Empty', shortcut = 'ctrl+alt+N' ) )
		self.addMenuItem( 'main/entity/add_entity',          dict( label = 'Create',       shortcut = 'ctrl+shift+N' ) )
		self.addMenuItem( 'main/entity/----' )
		self.addMenuItem( 'main/entity/group_entity',        dict( label = 'Group Entites',    shortcut = 'ctrl+G' ) )
		self.addMenuItem( 'main/entity/create_group',        dict( label = 'Create Empty Group',    shortcut = 'ctrl+shift+G' ) )
		self.addMenuItem( 'main/entity/----' )
		self.addMenuItem( 'main/entity/load_prefab',         dict( label = 'Load Prefab', shortcut = 'ctrl+alt+shift+N' ) )
		self.addMenuItem( 'main/entity/load_prefab_in_container', dict( label = 'Load Prefab In Container', shortcut = 'ctrl+shift+=' ) )
		self.addMenuItem( 'main/entity/----' )
		self.addMenuItem( 'main/entity/remove_entity',       dict( label = 'Remove'  ) )
		self.addMenuItem( 'main/entity/clone_entity',        dict( label = 'Clone',  shortcut = 'ctrl+d' ) )
		self.addMenuItem( 'main/entity/----' )
		self.addMenuItem( 'main/entity/add_component',       dict( label = 'Add Component', shortcut = 'ctrl+alt+=' ) )
		self.addMenuItem( 'main/entity/assign_layer',        dict( label = 'Assign Layer', shortcut = 'ctrl+alt+L' ) )
		self.addMenuItem( 'main/entity/toggle_visibility',   dict( label = 'Toggle Visibility', shortcut = 'ctrl+/' ) )
		self.addMenuItem( 'main/entity/freeze_entity_pivot', dict( label = 'Freeze Pivot' ) )

		self.addMenuItem( 'main/entity/----' )
		self.addMenuItem( 'main/find/find_entity', dict( label = 'Find In Scene', shortcut = 'ctrl+f' ) )
		self.addMenuItem( 'main/find/find_entity_in_group', dict( label = 'Find In Group', shortcut = 'ctrl+shift+f' ) )
		self.addMenuItem( 'main/find/find_entity_group', dict( label = 'Find Group', shortcut = 'ctrl+alt+f' ) )

		#Toolbars
		self.addTool( 'scene_graph/select_scene',    label ='Select Scene', icon = 'settings' )
		self.addTool( 'scene_graph/----'  )
		self.addTool( 'scene_graph/create_group',    label ='+ Group', icon = 'add_folder' )
		self.addTool( 'scene_graph/----'  )
		self.addTool( 'scene_graph/make_proto',    label = 'Convert To Proto', icon = 'proto_make' )
		self.addTool( 'scene_graph/create_proto_instance',    label = 'Create Proto Instance', icon = 'proto_instantiate' )
		self.addTool( 'scene_graph/create_proto_container',    label = 'Create Proto Container', icon = 'proto_container' )
		self.addTool( 'scene_graph/----'  )
		self.addTool( 'scene_graph/fold_all',    label = 'F' )
		self.addTool( 'scene_graph/unfold_all',  label = 'U' )
		self.addTool( 'scene_graph/refresh_tree',  label = 'R' )
		# self.addTool( 'scene_graph/load_prefab', label = '+ P' )
		# self.addTool( 'scene_graph/save_prefab', label = '>>P' )

		self.addTool( 'scene/refresh', label = 'refresh', icon='refresh' )

		#SIGNALS
		signals.connect( 'moai.clean',        self.onMoaiClean        )

		signals.connect( 'scene.clear',       self.onSceneClear      )
		signals.connect( 'scene.change',      self.onSceneChange     )

		signals.connect( 'selection.changed', self.onSelectionChanged )
		signals.connect( 'selection.hint',    self.onSelectionHint    )

		signals.connect( 'preview.start',     self.onPreviewStart     )
		signals.connect( 'preview.stop' ,     self.onPreviewStop      )

		# signals.connect( 'animator.start',     self.onAnimatorStart     )
		# signals.connect( 'animator.stop' ,     self.onAnimatorStop      )

		signals.connect( 'entity.added',      self.onEntityAdded      )
		signals.connect( 'entity.removed',    self.onEntityRemoved    )
		signals.connect( 'entity.renamed',    self.onEntityRenamed    )
		signals.connect( 'entity.modified',    self.onEntityModified    )
		signals.connect( 'entity.visible_changed',    self.onEntityVisibleChanged )
		signals.connect( 'entity.pickable_changed',   self.onEntityPickableChanged )


		signals.connect( 'prefab.unlink',     self.onPrefabUnlink    )
		signals.connect( 'prefab.relink',     self.onPrefabRelink    )
		signals.connect( 'proto.unlink',      self.onPrefabUnlink    )
		signals.connect( 'proto.relink',      self.onPrefabRelink    )

		signals.connect( 'app.ready', self.postAppReady )


		signals.connect( 'component.added',   self.onComponentAdded   )
		signals.connect( 'component.removed', self.onComponentRemoved )

		signals.connect( 'project.presave',   self.preProjectSave )

		registerSearchEnumerator( sceneObjectSearchEnumerator )
		registerSearchEnumerator( entityNameSearchEnumerator )
		registerSearchEnumerator( componentNameSearchEnumerator )
		registerSearchEnumerator( layerNameSearchEnumerator )

	def onStart( self ):
		self.refreshCreatorMenu()

	def postAppReady( self ):
		self.openPreviousScene()

	def openPreviousScene( self ):
		previousScene = self.getConfig( 'previous_scene', None )
		if previousScene:
			node = self.getAssetLibrary().getAssetNode( previousScene )
			if node:
				node.edit()

	def onStop( self ):
		if self.activeSceneNode:
			self.setConfig( 'previous_scene', self.activeSceneNode.getNodePath() )
		else:
			self.setConfig( 'previous_scene', False )

	def onSetFocus( self ):
		self.container.show()
		self.container.raise_()
		self.container.setFocus()

	def getActiveScene( self ):
		return self.delegate.safeCallMethod( 'editor', 'getScene' )

	def getActiveSceneRootGroup( self ):
		scene = self.delegate.safeCallMethod( 'editor', 'getScene' )
		if scene:
			return scene.rootGroup
		else:
			return None

	def openScene( self, node, protoNode = None ):
		if self.activeSceneNode == node:			
			if self.getModule('scene_view'):
				self.getModule('scene_view').setFocus()

			if protoNode:
				self.delegate.safeCallMethod( 'editor', 'locateProto', protoNode.getPath() )
				if self.getModule('scene_view'):
					self.getModule('scene_view').focusSelection()
				
		else:
			if not self.closeScene(): return
			if self.getModule('scene_view'):
				self.getModule('scene_view').makeCanvasCurrent()
			self.activeSceneNode = node
			signals.emitNow( 'scene.pre_open', node )
			scene = self.delegate.safeCallMethod( 'editor', 'openScene', node.getPath() )
			if not scene:
				#todo: raise something
				alertMessage( 'error', 
					'%s\n\nfailed to open scene, see console for detailed information.' % node.getPath() )
				return False
			signals.emitNow( 'scene.open', self.activeSceneNode, scene )
			self.setFocus()
			self.editingProtoNode = protoNode		
			self.loadWorkspaceState( False )
			self.delegate.safeCallMethod( 'editor', 'postOpenScene' )

		
	def closeScene( self ):
		if not self.activeSceneNode: return True
		if self.sceneDirty:
			res = requestConfirm( 'scene modified!', 'save scene before close?' )
			if res == True:   #save
				self.saveScene()
			elif res == None: #cancel
				return
			elif res == False: #no save
				pass

		self.markSceneDirty( False )
		self.tree.clear()
		self.getApp().clearCommandStack( 'scene_editor' )
		self.getSelectionManager().removeSelection( self.getActiveScene() )
		signals.emitNow( 'scene.close', self.activeSceneNode )
		self.delegate.safeCallMethod( 'editor', 'closeScene' )		
		self.activeSceneNode = None
		return True

	def onSceneClear( self ):
		# self.tree.clear()
		pass

	def markSceneDirty( self, dirty = True ):
		if not self.previewing:
			self.sceneDirty = dirty

	def saveWorkspaceState( self ):
		self.retainWorkspaceState()
		treeFoldState      = self.workspaceState['tree_state']
		containerFoldState = self.workspaceState['container_state']
		entityLockState    = self.workspaceState['entity_lock_state']
		self.activeSceneNode.setMetaData( 'tree_state', treeFoldState )
		self.activeSceneNode.setMetaData( 'container_state', containerFoldState )
		self.activeSceneNode.setMetaData( 'entity_lock_state', entityLockState )

	def loadWorkspaceState( self, restoreState = True ):
		treeFoldState      = self.activeSceneNode.getMetaData( 'tree_state', None )		
		containerFoldState = self.activeSceneNode.getMetaData( 'container_state', None )		
		entityLockState    = self.activeSceneNode.getMetaData( 'entity_lock_state', None )	
		self.workspaceState = {
			'tree_state'        : treeFoldState,
			'container_state'   : containerFoldState,
			'entity_lock_state' : entityLockState
		}		
		if restoreState: self.restoreWorkspaceState()

	def retainWorkspaceState( self ):
		#tree node foldstate
		treeFoldState =  self.tree.saveFoldState()
		#save introspector foldstate
		introspectorFoldState = self.delegate.safeCallMethod( 'editor', 'saveIntrospectorFoldState' )
		entityLockState = self.delegate.safeCallMethod( 'editor', 'saveEntityLockState' )

		self.workspaceState = {
			'tree_state'        : treeFoldState,
			'container_state'   : introspectorFoldState,
			'entity_lock_state' : entityLockState
		}

	def restoreWorkspaceState( self ):
		if not self.workspaceState: return
		treeState = self.workspaceState.get( 'tree_state', None )
		if treeState:
			self.tree.loadFoldState( treeState )	
		
		containerState = self.workspaceState.get( 'container_state', None )
		if containerState:
			self.delegate.safeCallMethod( 'editor', 'loadIntrospectorFoldState', containerState )
		
		lockState = self.workspaceState.get( 'entity_lock_state', None )
		if lockState:
			self.delegate.safeCallMethod( 'editor', 'loadEntityLockState', lockState )


	def onSceneChange( self ):
		self.tree.hide()
		self.tree.rebuild()
		self.restoreWorkspaceState()
		self.tree.refreshAllContent()
		self.tree.verticalScrollBar().setValue( 0 )
		self.tree.show()
		if self.editingProtoNode:
			self.delegate.safeCallMethod( 'editor', 'locateProto', self.editingProtoNode.getPath() )
			self.editingProtoNode = None
			if self.getModule('scene_view'):
					self.getModule('scene_view').focusSelection()
		
	def saveScene( self ):
		if not self.activeSceneNode: return
		self.markSceneDirty( False )
		signals.emitNow( 'scene.save' )
		self.delegate.safeCallMethod( 'editor', 'saveScene', self.activeSceneNode.getAbsFilePath() )
		signals.emitNow( 'scene.saved' )
		self.saveWorkspaceState()

	def refreshScene( self ):
		if not self.activeSceneNode: return
		if self.previewing: return
		self.refreshScheduled = False
		node = self.activeSceneNode
		self.retainWorkspaceState()
		if self.delegate.safeCallMethod( 'editor', 'refreshScene' ):
			self.restoreWorkspaceState()
		self.refreshCreatorMenu()

	def scheduleRefreshScene( self ):
		if not self.activeSceneNode: return
		self.refreshScheduled = True

	def refreshCreatorMenu( self ):
		def addEntityMenuItem( name ):
			if name == '----': 
				self.entityCreatorMenu.addChild( '----' )
				return
			self.entityCreatorMenu.addChild({
					'name'     : 'create_entity_'+name,
					'label'    : name,
					'command'  : 'scene_editor/create_entity',
					'command_args' : dict( name = name )
				})

		def addComponentMenuItem( name ):
			if name == '----': 
				self.componentCreatorMenu.addChild( '----' )
				return
			self.componentCreatorMenu.addChild({
					'name'     : 'create_component_'+name,
					'label'    : name,
					'command'  : 'scene_editor/create_component',
					'command_args' : dict( name = name )
				})

		self.entityCreatorMenu.clear()
		self.componentCreatorMenu.clear()

		registry = _MOCK.getEntityRegistry()
		#entity
		keys = sorted( registry.keys() )
		addEntityMenuItem( 'Entity' )
		addEntityMenuItem( '----' )
		for entityName in sorted( registry.keys() ):
			if entityName!='Entity': addEntityMenuItem( entityName )

		#component
		registry = _MOCK.getComponentRegistry()
		for comName in sorted( registry.keys() ):
			addComponentMenuItem( comName )

	def needUpdate( self ):
		return True
		
	def onUpdate( self ):
		if self.refreshScheduled:
			self.refreshScene()

	def preProjectSave( self, prj ):
		if self.activeSceneNode:
			_MOCK.game.previewingScene = self.activeSceneNode.getNodePath()

	def onMoaiClean( self ):
		self.tree.clear()

	def onTool( self, tool ):
		name = tool.name
		
		if name == 'fold_all':
			self.tree.foldAllItems()

		elif name == 'unfold_all':
			self.tree.expandAllItems()

		elif name == 'refresh_tree':
			self.tree.rebuild()

		elif name == 'refresh':
			self.scheduleRefreshScene()

		elif name == 'make_proto':
			self.makeProto()

		elif name == 'create_proto_instance':
			requestSearchView( 
				info    = 'select a proto to instantiate',
				context = 'asset',
				type    = 'proto',
				on_selection = 
					lambda obj: 
						self.doCommand( 'scene_editor/create_proto_instance', proto = obj.getNodePath() )
				)

		elif name == 'create_proto_container':
			requestSearchView( 
				info    = 'select a proto to create contained instance',
				context = 'asset',
				type    = 'proto',
				on_selection = 
					lambda obj: 
						self.doCommand( 'scene_editor/create_proto_container', proto = obj.getNodePath() )
				)

		elif name == 'create_group':
			self.doCommand( 'scene_editor/entity_group_create' )

		elif name == 'group_entity':
			self.doCommand( 'scene_editor/group_entities' )
		
		elif name == 'select_scene':
			self.doCommand( 'scene_editor/select_scene' )
			

	def onMenu( self, menu ):
		name = menu.name
		if name == 'close_scene':
			if self.previewing:
				alertMessage( 'Warning', 'Stop previewing before closing scene' )
				return
			self.closeScene()

		elif name == 'open_scene':
			if self.previewing:
				alertMessage( 'Warning', 'Stop previewing before opening scene' )
				return
			requestSearchView( 
				info    = 'select scene to open',
				context = 'asset',
				type    = 'scene',
				on_selection = self.openScene
				)
			
		elif name == 'save_scene':
			if self.previewing:
				alertMessage( 'Warning', 'Stop previewing before saving' )
				return
			self.saveScene()

		elif name == 'locate_scene_asset':
			if self.activeSceneNode:
				assetBrowser = self.getModule( 'asset_browser' )
				if assetBrowser:
					assetBrowser.selectAsset( self.activeSceneNode )

		elif name == 'add_entity':
			requestSearchView( 
				info    = 'select entity type to create',
				context = 'entity_creation',
				on_selection = lambda obj: 
					self.doCommand( 'scene_editor/create_entity', name = obj )
				)

		elif name == 'add_component':
			requestSearchView( 
				info    = 'select component type to create',
				context = 'component_creation',
				on_selection = lambda obj: 
					self.doCommand( 'scene_editor/create_component', name = obj )
				)

		elif name == 'add_empty_entity':
			self.doCommand( 'scene_editor/create_entity', name = 'Entity' )

		elif name == 'load_prefab':
			requestSearchView( 
				info    = 'select a perfab node to instantiate',
				context = 'asset',
				type    = 'prefab',
				on_selection = 
					lambda obj: 
						self.doCommand( 'scene_editor/create_prefab_entity', prefab = obj.getNodePath() )
				)

		elif name == 'load_prefab_in_container':
			requestSearchView( 
				info    = 'select a perfab node to instantiate( PefabContainer )',
				context = 'asset',
				type    = 'prefab',
				on_selection = 
					lambda obj: 
						self.doCommand( 'scene_editor/create_prefab_container', prefab = obj.getNodePath() )
				)

		elif name == 'remove_entity':
			self.doCommand( 'scene_editor/remove_entity' )

		elif name == 'clone_entity':
			self.doCommand( 'scene_editor/clone_entity' )

		elif name == 'find_entity':
			requestSearchView( 
				info    = 'search for entity in current scene',
				context = 'scene',
				type    = 'entity',
				on_selection = lambda x: self.selectEntity( x, focus_tree = True ) ,
				on_test      = self.selectEntity
				)

		elif name == 'find_entity_in_group':
			requestSearchView( 
				info    = 'search for entity in current entity group',
				context = 'scene',
				type    = 'entity_in_group',
				on_selection = lambda x: self.selectEntity( x, focus_tree = True ) ,
				on_test      = self.selectEntity
				)

		elif name == 'find_entity_group':
			requestSearchView( 
				info    = 'search for group in current scene',
				context = 'scene',
				type    = 'group',
				on_selection = lambda x: self.selectEntity( x, focus_tree = True ) ,
				on_test      = self.selectEntity
				)

		elif name == 'create_group':
			self.doCommand( 'scene_editor/entity_group_create' )

		elif name == 'remove_component':
			context = menu.getContext()
			if context:
				self.doCommand( 'scene_editor/remove_component', target = context )

		elif name == 'copy_component':
			context = menu.getContext()
			if context:
				self.doCommand( 'scene_editor/copy_component', target = context )

		elif name == 'assign_layer':
			if not self.tree.getSelection(): return
			requestSearchView( 
				info    = 'select layer to assign',
				context = 'scene_layer',
				type    = _MOCK.Entity,
				on_selection = self.assignEntityLayer
				)

		elif name == 'toggle_visibility':
			self.doCommand( 'scene_editor/toggle_entity_visibility' )

		elif name == 'freeze_entity_pivot':
			self.doCommand( 'scene_editor/freeze_entity_pivot' )


	def onSelectionChanged( self, selection, key ):
		if key != 'scene': return
		if self.tree.syncSelection:
			self.tree.blockSignals( True )
			self.tree.selectNode( None )
			for e in selection:
				self.tree.selectNode( e, add = True)
			self.tree.blockSignals( False )

	def selectEntity( self, target, **option ):
		if option.get( 'focus_tree', False ):
			self.tree.setFocus()
		self.changeSelection( target )

	##----------------------------------------------------------------##
	def renameEntity( self, target, name ):
		#TODO:command pattern
		target.setName( target, name )
		signals.emit( 'entity.modified', target )

	def addEntityNode( self, entity ):
		self.tree.addNode( entity, expanded = False )
		self.tree.setNodeExpanded( entity, False )

	def removeEntityNode( self, entity ):
		self.tree.removeNode( entity )

	def assignEntityLayer( self, layerName ):
		#TODO:command pattern
		if not layerName: return
		self.doCommand( 'scene_editor/assign_layer', target = layerName )

	def onSelectionHint( self, selection ):
		if selection._entity:
			self.changeSelection( selection._entity )			
		else:
			self.changeSelection( selection )

	def onPreviewStart( self ):
		if not self.activeSceneNode: return
		self.retainWorkspaceState()
		self.delegate.safeCallMethod( 'editor', 'retainScene' )
		self.delegate.safeCallMethod( 'editor', 'startScenePreview' )
		self.previewing = True

	def onPreviewStop( self ):
		if not self.activeSceneNode: return
		self.changeSelection( None )
		self.tree.clear()
		self.delegate.safeCallMethod( 'editor', 'stopScenePreview' )
		self.previewing = False
		if self.delegate.safeCallMethod( 'editor', 'restoreScene' ):
			self.restoreWorkspaceState()

	def onAnimatorStart( self ):
		self.retainWorkspaceState()
		self.delegate.safeCallMethod( 'editor', 'retainScene' )

	def onAnimatorStop( self ):
		self.tree.clear()
		self.delegate.safeCallMethod( 'editor', 'clearScene' )
		if self.delegate.safeCallMethod( 'editor', 'restoreScene' ):
			self.restoreWorkspaceState()

	##----------------------------------------------------------------##
	def updateEntityPriority( self ):
		if not self.activeSceneNode: return
		self.markSceneDirty()

	def onEntityRenamed( self, entity, newname ):
		self.tree.refreshNodeContent( entity )
		self.markSceneDirty()

	def onEntityVisibleChanged( self, entity ):
		self.tree.refreshNodeContent( entity )

	def onEntityPickableChanged( self, entity ):
		self.tree.refreshNodeContent( entity )

	def onEntityAdded( self, entity, context = None ):
		if context == 'new':
			self.setFocus()
			pnode = entity.parent
			if pnode:
				self.tree.setNodeExpanded( pnode, True )
			self.tree.setFocus()
			self.tree.editNode( entity )
			self.tree.selectNode( entity )
		signals.emit( 'scene.update' )
		self.markSceneDirty()

	def onEntityRemoved( self, entity ):
		signals.emit( 'scene.update' )
		self.markSceneDirty()

	def onEntityModified( self, entity, context = None ):
		self.markSceneDirty()

	##----------------------------------------------------------------##
	def onComponentAdded( self, com, entity ):
		signals.emit( 'scene.update' )
		self.markSceneDirty()


	def onComponentRemoved( self, com, entity ):
		signals.emit( 'scene.update' )
		self.markSceneDirty()


	##----------------------------------------------------------------##
	def onPrefabUnlink( self, entity ):
		self.tree.refreshNodeContent( entity, updateChildren = True )

	def onPrefabRelink( self, entity ):
		self.tree.refreshNodeContent( entity, updateChildren = True )

	def createPrefab( self, targetPrefab ):
		selection = self.getSelection()
		if not selection: return
		if len( selection ) > 1:
			return alertMessage( 'multiple entities cannot be converted into prefab' )
		target = selection[0]
		self.doCommand( 'scene_editor/create_prefab', 
			entity = target, 
			prefab = targetPrefab.getNodePath(),
			file   = targetPrefab.getAbsFilePath()
		)

	def makeProto( self ):
		selection = self.getSelection()
		if not selection: return
		if len( selection ) > 1:
			return alertMessage( 'multiple entities cannot be converted into Proto' )
		target = selection[0]
		if not target: return
		if requestConfirm( 'convert proto', 'Convert this Entity into Proto?' ):
			self.doCommand( 'scene_editor/make_proto', 
				entity = target
			)
			self.tree.refreshNodeContent( target )

	def createProtoInstance( self ):
		pass


	##----------------------------------------------------------------##
	def onCopyEntity( self ):
		entityGroupData = self.delegate.callMethod( 'editor', 'makeSceneSelectionCopyData' )
		if not entityGroupData: return False
		clip = QtGui.QApplication.clipboard()
		mime = QtCore.QMimeData()
		text = ''
		for s in self.getSelection():
			if text == '':
				text = text + s.name
			else:
				text = text + '\n' + s.name
		mime.setText( text )
		mime.setData( GII_MIME_ENTITY_DATA, entityGroupData.encode('utf-8') )
		clip.setMimeData( mime )
		return True

	def onPasteEntity( self ):
		clip = QtGui.QApplication.clipboard()
		mime = clip.mimeData()
		if mime.hasFormat( GII_MIME_ENTITY_DATA ):
			data = mime.data( GII_MIME_ENTITY_DATA )
			self.doCommand( 'scene_editor/paste_entity',
				data = str(data).decode('utf-8')
			)

	##----------------------------------------------------------------##
	def onCopyComponent( self ):
		entityGroupData = self.delegate.callMethod( 'editor', 'makeEntityCopyData' )
		if not entityGroupData: return False
		clip = QtGui.QApplication.clipboard()
		mime = QtCore.QMimeData()
		text = ''
		for s in self.getSelection():
			if text == '':
				text = text + s.name
			else:
				text = text + '\n' + s.name
		mime.setText( text )
		mime.setData( GII_MIME_ENTITY_DATA, str(entityGroupData) )
		clip.setMimeData( mime )
		return True

	def onPasteComponent( self ):
		clip = QtGui.QApplication.clipboard()
		mime = clip.mimeData()
		if mime.hasFormat( GII_MIME_ENTITY_DATA ):
			data = mime.data( GII_MIME_ENTITY_DATA )
			self.doCommand( 'scene_editor/paste_entity',
				data = str(data)
			)
Esempio n. 12
0
    def onLoad(self):
        # UI
        self.windowTitle = "Animator"
        self.window = self.requestDockWindow(
            "AnimatorView", title="Animator", size=(120, 120), minSize=(120, 120), dock="bottom"
        )

        self.widget = AnimatorWidget()
        self.window.addWidget(self.widget)
        self.toolbarTarget = self.addToolBar("animator_target", self.widget.toolbarTarget)
        self.toolbarClips = self.addToolBar("animator_clips", self.widget.toolbarClips)
        self.toolbarPlay = self.addToolBar("animator_play", self.widget.toolbarPlay)
        self.toolbarTrack = self.addToolBar("animator_track", self.widget.toolbarTrack)
        # self.toolbarEdit  = self.addToolBar( 'animator_play',  self.widget.toolbarEdit )

        signals.connect("scene.close", self.onSceneClose)
        signals.connect("scene.save", self.preSceneSave)
        signals.connect("scene.saved", self.postSceneSave)

        # addWidgetWithLaytut( toolbar,
        # 	self.widget.containerEditTool )
        self.addTool("animator_target/change_context", label="Change Context", icon="in")
        self.addTool("animator_target/save_data", label="Save Data", icon="save")

        self.addTool("animator_clips/add_clip_group", label="add group", icon="add_folder")
        self.addTool("animator_clips/add_clip", label="add", icon="add")
        self.addTool("animator_clips/remove_clip", label="remove", icon="remove")
        self.addTool("animator_clips/clone_clip", label="clone", icon="clone")

        self.addTool("animator_play/goto_start", label="to start", icon="rewind")
        # self.addTool( 'animator_play/prev_key',   label = 'prev key',      icon = 'previous' )
        self.addTool("animator_play/stop", label="stop", icon="stop")
        self.addTool("animator_play/play", label="play", icon="play", type="check")
        # self.addTool( 'animator_play/next_key',   label = 'next key',      icon = 'next' )
        self.addTool("animator_play/goto_end", label="to end", icon="fast_forward")
        self.addTool("animator_play/----")
        self.addTool("animator_play/toggle_repeat", label="toggle repeat", icon="repeat", type="check")

        # SIGNALS
        self.addTool("animator_track/add_track_group", label="add group", icon="add_folder")
        self.addTool("animator_track/add_track", label="add", icon="add")
        self.addTool("animator_track/remove_track", label="remove", icon="remove")

        #
        signals.connect("selection.changed", self.onSceneSelectionChanged)

        self.delegate = MOAILuaDelegate(self)
        self.delegate.load(_getModulePath("AnimatorView.lua"))

        self.widget.setOwner(self)

        # playback
        self.previewing = False
        self.setEditing(False)

        self.targetAnimator = None
        self.targetClip = None
        self.targetAnimatorData = None
        self.currentTrack = None

        self.previewing = False
        self.previewLoop = False
        self.previewTime = 0.0
        self.previewStep = 1.0 / 60.0

        self.previewTimer = QtCore.QTimer(self.widget)
        self.previewTimer.setInterval(1000.0 / 65)
        self.previewTimer.stop()

        self.previewTimer.timeout.connect(self.onPreviewTimer)
Esempio n. 13
0
class AnimatorView(SceneEditorModule):
    name = "animator"
    dependency = ["scene_editor", "mock"]

    def onLoad(self):
        # UI
        self.windowTitle = "Animator"
        self.window = self.requestDockWindow(
            "AnimatorView", title="Animator", size=(120, 120), minSize=(120, 120), dock="bottom"
        )

        self.widget = AnimatorWidget()
        self.window.addWidget(self.widget)
        self.toolbarTarget = self.addToolBar("animator_target", self.widget.toolbarTarget)
        self.toolbarClips = self.addToolBar("animator_clips", self.widget.toolbarClips)
        self.toolbarPlay = self.addToolBar("animator_play", self.widget.toolbarPlay)
        self.toolbarTrack = self.addToolBar("animator_track", self.widget.toolbarTrack)
        # self.toolbarEdit  = self.addToolBar( 'animator_play',  self.widget.toolbarEdit )

        signals.connect("scene.close", self.onSceneClose)
        signals.connect("scene.save", self.preSceneSave)
        signals.connect("scene.saved", self.postSceneSave)

        # addWidgetWithLaytut( toolbar,
        # 	self.widget.containerEditTool )
        self.addTool("animator_target/change_context", label="Change Context", icon="in")
        self.addTool("animator_target/save_data", label="Save Data", icon="save")

        self.addTool("animator_clips/add_clip_group", label="add group", icon="add_folder")
        self.addTool("animator_clips/add_clip", label="add", icon="add")
        self.addTool("animator_clips/remove_clip", label="remove", icon="remove")
        self.addTool("animator_clips/clone_clip", label="clone", icon="clone")

        self.addTool("animator_play/goto_start", label="to start", icon="rewind")
        # self.addTool( 'animator_play/prev_key',   label = 'prev key',      icon = 'previous' )
        self.addTool("animator_play/stop", label="stop", icon="stop")
        self.addTool("animator_play/play", label="play", icon="play", type="check")
        # self.addTool( 'animator_play/next_key',   label = 'next key',      icon = 'next' )
        self.addTool("animator_play/goto_end", label="to end", icon="fast_forward")
        self.addTool("animator_play/----")
        self.addTool("animator_play/toggle_repeat", label="toggle repeat", icon="repeat", type="check")

        # SIGNALS
        self.addTool("animator_track/add_track_group", label="add group", icon="add_folder")
        self.addTool("animator_track/add_track", label="add", icon="add")
        self.addTool("animator_track/remove_track", label="remove", icon="remove")

        #
        signals.connect("selection.changed", self.onSceneSelectionChanged)

        self.delegate = MOAILuaDelegate(self)
        self.delegate.load(_getModulePath("AnimatorView.lua"))

        self.widget.setOwner(self)

        # playback
        self.previewing = False
        self.setEditing(False)

        self.targetAnimator = None
        self.targetClip = None
        self.targetAnimatorData = None
        self.currentTrack = None

        self.previewing = False
        self.previewLoop = False
        self.previewTime = 0.0
        self.previewStep = 1.0 / 60.0

        self.previewTimer = QtCore.QTimer(self.widget)
        self.previewTimer.setInterval(1000.0 / 65)
        self.previewTimer.stop()

        self.previewTimer.timeout.connect(self.onPreviewTimer)

    def onStart(self):
        pass

    def setEditing(self, editing):
        self.widget.timeline.setEnabled(editing)
        self.widget.treeTracks.setEnabled(editing)
        self.findTool("animator_play").setEnabled(editing)
        self.findTool("animator_track").setEnabled(editing)
        self.findTool("animator_clips/add_clip_group").setEnabled(editing)
        self.findTool("animator_clips/add_clip").setEnabled(editing)
        self.findTool("animator_clips/remove_clip").setEnabled(editing)
        self.findTool("animator_clips/clone_clip").setEnabled(editing)

    def setTargetAnimator(self, target):
        self.saveAnimatorData()
        if target == self.targetAnimator:
            return
        if self.previewing:
            self.stopPreview()
        self.targetAnimator = target
        self.targetClip = None
        self.delegate.callMethod("view", "setTargetAnimator", target)
        self.targetAnimatorData = self.delegate.callMethod("view", "getTargetAnimatorData")
        self.widget.rebuild()
        if self.targetAnimator:
            self.setEditing(True)
            signals.emit("animator.start")
        else:
            self.setEditing(False)
            signals.emit("animator.stop")

        path = self.delegate.callMethod("view", "getTargetAnimatorDataPath")
        if path:
            self.window.setWindowTitle("Animator - %s" % path)
        else:
            self.window.setWindowTitle("Animator")
        clip = self.delegate.callMethod("view", "getPreviousTargeClip", target)
        self.enableTool("animator_play", False)
        self.enableTool("animator_track", False)
        if clip:
            self.widget.treeClips.selectNode(clip)
        else:
            self.widget.treeClips.selectFirstItem()
        self.applyTime(0, True)

    def setTargetClip(self, clip):
        wasPreviewing = self.previewing
        if self.previewing:
            self.stopPreview()

        self.targetClip = clip
        self.delegate.callMethod("view", "setTargetClip", clip)
        self.widget.rebuildTimeline()
        self.enableTool("animator_play", bool(clip))
        self.enableTool("animator_track", bool(clip))
        self.applyTime(0, True)
        if wasPreviewing:
            self.startPreview()

    def setCurrentTrack(self, track):
        self.currentTrack = track
        self.delegate.callMethod("view", "setCurrentTrack", track)

    def getClipList(self):
        if self.targetAnimatorData:
            clipList = self.targetAnimatorData.clips
            return [clip for clip in clipList.values()]
        else:
            return []

    def getRootClipGroup(self):
        if self.targetAnimatorData:
            return self.targetAnimatorData.getRootGroup(self.targetAnimatorData)

    def getTrackList(self):
        if self.targetClip:
            trackList = self.targetClip.getTrackList(self.targetClip)
            return [track for track in trackList.values()]
        else:
            return []

    def getMarkerList(self):
        if self.targetClip:
            markerList = self.targetClip.getMarkerList(self.targetClip)
            return [track for track in markerList.values()]
        else:
            return []

    def getClipRoot(self):
        if self.targetClip:
            return self.targetClip.getRoot(self.targetClip)
        else:
            return None

    def addClip(self):
        if not self.targetAnimatorData:
            return
        targetGroup = self.widget.getCurrentClipGroup()
        cmd = self.doCommand(
            "scene_editor/animator_add_clip", animator_data=self.targetAnimatorData, parent_group=targetGroup
        )
        clip = cmd.getResult()
        if clip:
            self.widget.addClip(clip, True)
        return clip

    def addClipGroup(self):
        if not self.targetAnimatorData:
            return
        targetGroup = self.widget.getCurrentClipGroup()
        cmd = self.doCommand(
            "scene_editor/animator_add_clip_group", animator_data=self.targetAnimatorData, parent_group=targetGroup
        )
        group = cmd.getResult()
        if group:
            self.widget.addClip(group, True)
        return group

    def removeClipNode(self):
        for clip in self.widget.treeClips.getSelection():
            if self.doCommand(
                "scene_editor/animator_remove_clip_node", animator_data=self.targetAnimatorData, target_node=clip
            ):
                self.widget.removeClip(clip)

    def cloneClipNode(self):
        if not self.targetClip:
            return
        result = []
        for clip in self.widget.treeClips.getSelection():
            cmd = self.doCommand(
                "scene_editor/animator_clone_clip_node", animator_data=self.targetAnimatorData, target_node=clip
            )
            if cmd:
                cloned = cmd.getResult()
                self.widget.addClip(cloned)
                result.append(cloned)
        return result

    def onObjectEdited(self, obj):
        if self.targetClip:
            self.delegate.callMethod("view", "clearPreviewState")
            self.delegate.callMethod("view", "markClipDirty")

    def onSceneSelectionChanged(self, selection, key):
        if key != "scene":
            return
        # find animator component
        # self.findTargetAnimator()

    def findTargetAnimator(self):
        target = self.delegate.callMethod("view", "findTargetAnimator")
        self.setTargetAnimator(target)
        return target

    def checkTargetAnimator(self):
        if not self.targetAnimator:
            alertMessage("No Animator", "No Animator Selected", "question")
            return False
        return True

    def addMarker(self):
        if not self.targetClip:
            return
        cmd = self.doCommand(
            "scene_editor/animator_add_marker", target_clip=self.targetClip, target_pos=self.widget.getCursorPos()
        )
        if cmd:
            marker = cmd.getResult()
            self.widget.addMarker(marker)

    def addKeyForField(self, target, fieldId):
        if not self.checkTargetAnimator():
            return

        if not self.targetClip:
            self.addClip()
            # alertMessage( 'No Clip', 'You need to select a Clip first', 'question' )
            # return False
        keys = self.delegate.callMethod("view", "addKeyForField", target, fieldId)
        if keys:
            for key in keys.values():
                self.widget.addKey(key, True)

    def addKeyForEvent(self, target, eventId):
        pass

    def addCustomAnimatorTrack(self, target, trackClasId):
        if not self.checkTargetAnimator():
            return

        track = self.delegate.callMethod("view", "addCustomAnimatorTrack", target, trackClasId)
        if track:
            self.widget.addTrack(track)

    def addKeyForSelectedTracks(self):
        # TODO: command
        selectedTracks = self.widget.getTrackSelection()
        for track in selectedTracks:
            keys = self.delegate.callMethod("view", "addKeyForSelectedTrack", track)
            if keys:
                for key in keys.values():
                    self.widget.addKey(key, True)

    def removeSelectedKeys(self):
        # TODO: command
        selectedKeys = self.widget.getKeySelection()
        for key in selectedKeys:
            self.widget.removeKey(key)

    def cloneSelectedKeys(self):
        # TODO: command
        selectedKeys = self.widget.getKeySelection()
        cloned = []
        for key in selectedKeys:
            clonedKey = self.delegate.callMethod("view", "cloneKey", key)
            if clonedKey:
                cloned.append(clonedKey)

        for clonedKey in cloned:
            self.widget.addKey(clonedKey, False)

    def onKeyRemoving(self, key):
        if self.delegate.callMethod("view", "removeKey", key) != False:
            return True

    def onTimelineKeyChanged(self, key, pos, length):
        self.delegate.callMethod("view", "updateTimelineKey", key, pos, length)

    def onTimelineKeyCurveValueChanged(self, key, value):
        self.delegate.callMethod("view", "updateTimelineKeyCurveValue", key, value)

    def onTimelineKeyTweenModeChanged(self, key, mode):
        self.delegate.callMethod("view", "updateTimelineKeyTweenMode", key, mode)

    def onTimelineKeyBezierPointChanged(self, key, bpx0, bpy0, bpx1, bpy1):
        self.delegate.callMethod("view", "updateTimelineKeyBezierPoint", key, bpx0, bpy0, bpx1, bpy1)

    def onTimelineMarkerChanged(self, marker, pos):
        self.delegate.callMethod("view", "updateTimelineMarker", marker, pos)

    def renameTrack(self, track, name):
        self.delegate.callMethod("view", "renameTrack", track, name)

    def renameClip(self, clip, name):
        self.delegate.callMethod("view", "renameClip", clip, name)

    def onTool(self, tool):
        name = tool.name
        if name == "change_context":
            target0 = self.targetAnimator
            target1 = self.findTargetAnimator()
            if (not target0) and (not target1):
                alertMessage("No Animator", "No Animator found in selected entity scope", "question")

        elif name == "save_data":
            self.saveAnimatorData()

        elif name == "add_clip":
            if self.checkTargetAnimator():
                self.addClip()

        elif name == "add_clip_group":
            if self.checkTargetAnimator():
                self.addClipGroup()

        elif name == "remove_clip":
            if self.checkTargetAnimator():
                self.removeClipNode()

        elif name == "clone_clip":
            if self.checkTargetAnimator():
                self.cloneClipNode()

        elif name == "add_track_group":
            group = self.delegate.callMethod("view", "addTrackGroup")
            if group:
                self.widget.addTrack(group, True)

        elif name == "remove_track":
            for track in self.widget.treeTracks.getSelection():
                self.delegate.callMethod("view", "removeTrack", track)
                self.widget.removeTrack(track)
                # preview
        elif name == "goto_start":
            self.gotoStart()
        elif name == "goto_end":
            self.gotoEnd()
        elif name == "play":
            if tool.getValue():
                self.startPreview()
            else:
                self.stopPreview(False)
        elif name == "stop":
            self.stopPreview(True)
        elif name == "toggle_repeat":
            self.delegate.callMethod("view", "togglePreviewRepeat", tool.getValue())

    def getActiveSceneView(self):
        return self.getModule("scene_view")

        # preview

    def startPreview(self):
        self.saveAnimatorData()
        if self.delegate.callMethod("view", "startPreview", self.previewTime):
            self.widget.setCursorMovable(False)
            self.previewing = True
            self.findTool("animator_play/play").setValue(True)
            self.previewTimer.start()
            self.getApp().setMinimalMainLoopBudget()

    def stopPreview(self, rewind=False):
        if self.previewing:
            self.delegate.callMethod("view", "stopPreview")
            self.getApp().resetMainLoopBudget()
            self.widget.setCursorMovable(True)
            self.previewing = False
            self.findTool("animator_play/play").setValue(False)
            self.previewTimer.stop()
            signals.emit("entity.modified", None, "")
        if rewind:
            self.gotoStart()

    def onPreviewTimer(self):
        playing, currentTime = self.delegate.callMethod("view", "doPreviewStep")
        self.previewTime = currentTime
        self.getActiveSceneView().forceUpdate()
        self.widget.setCursorPos(self.previewTime)
        if not playing:
            self.stopPreview()
            # signals.emit( 'entity.modified',  None , '' )

    def gotoStart(self):
        if self.previewing:
            self.delegate.callMethod("view", "applyTime", 0)
        else:
            self.widget.setCursorPos(0, True)

    def gotoEnd(self):
        if self.previewing:
            self.delegate.callMethod("view", "applyTime", 10)
        else:
            self.widget.setCursorPos(10, True)

    def applyTime(self, t, syncCursor=False):
        self.previewTime = self.delegate.callMethod("view", "applyTime", t)
        self.getActiveSceneView().forceUpdate()
        signals.emit("entity.modified", None, "")
        if syncCursor:
            self.widget.setCursorPos(t)

    def saveAnimatorData(self):
        if not self.targetAnimator:
            return
        self.delegate.callMethod("view", "saveData")

    def preSceneSave(self):
        if self.targetAnimator:
            self.delegate.callMethod("view", "restoreEntityState")

    def postSceneSave(self):
        if self.targetAnimator:
            self.applyTime(self.previewTime)

    def onSceneClose(self, scene):
        self.setTargetAnimator(None)

    def refreshTimeline(self):
        self.widget.rebuildTimeline()

    def refreshClipList(self):
        self.widget.rebuildClipList()

    def refreshAll(self):
        self.widget.rebuild()
Esempio n. 14
0
	def onLoad( self ):
		#UI
		self.windowTitle = 'Animator'
		self.window = self.requestDockWindow( 'AnimatorView',
			title     = 'Animator',
			size      = (120,120),
			minSize   = (120,120),
			dock      = 'bottom'
			)
		
		self.widget = AnimatorWidget()
		self.window.addWidget( self.widget )
		self.toolbarTarget = self.addToolBar( 'animator_target', self.widget.toolbarTarget )
		self.toolbarClips  = self.addToolBar( 'animator_clips', self.widget.toolbarClips )
		self.toolbarPlay   = self.addToolBar( 'animator_play',  self.widget.toolbarPlay )
		self.toolbarTrack  = self.addToolBar( 'animator_track', self.widget.toolbarTrack )
		# self.toolbarEdit  = self.addToolBar( 'animator_play',  self.widget.toolbarEdit )

		signals.connect( 'scene.close', self.onSceneClose )
		signals.connect( 'scene.save', self.preSceneSave )
		signals.connect( 'scene.saved', self.postSceneSave )

		# addWidgetWithLaytut( toolbar,
		# 	self.widget.containerEditTool )
		self.addTool( 'animator_target/change_context', label = 'Change Context', icon = 'in' )
		self.addTool( 'animator_target/save_data', label = 'Save Data', icon = 'save' )

		self.addTool( 'animator_clips/add_clip_group',   label = 'add group',    icon = 'add_folder' )
		self.addTool( 'animator_clips/add_clip',    label = 'add',    icon = 'add' )
		self.addTool( 'animator_clips/remove_clip', label = 'remove', icon = 'remove' )
		self.addTool( 'animator_clips/clone_clip', label = 'clone', icon = 'clone' )


		self.addTool( 'animator_play/goto_start', label = 'to start',  icon = 'rewind' )
		# self.addTool( 'animator_play/prev_key',   label = 'prev key',      icon = 'previous' )
		self.addTool( 'animator_play/stop',       label = 'stop',      icon = 'stop' )
		self.addTool( 'animator_play/play',       label = 'play',      icon = 'play',  type = 'check' )
		# self.addTool( 'animator_play/next_key',   label = 'next key',      icon = 'next' )
		self.addTool( 'animator_play/goto_end',   label = 'to end',    icon = 'fast_forward' )
		self.addTool( 'animator_play/toggle_repeat',  label = 'toggle repeat',  icon = 'repeat', type = 'check' )
		self.comboPreviewSpeed = QtGui.QComboBox()
		self.comboPreviewSpeed.addItems([ e[0] for e in PREVIEW_SPEED_OPTIONS ] )			
		self.comboPreviewSpeed.setCurrentIndex( 4 ) #1x
		self.comboPreviewSpeed.currentIndexChanged.connect( self.onPreviewSpeedChange )
		self.addTool( 'animator_play/preview_speed', widget = self.comboPreviewSpeed )
		
		#SIGNALS
		self.addTool( 'animator_track/locate_target', label = 'locate', icon = 'find' )
		self.addTool( 'animator_track/----' )
		self.addTool( 'animator_track/add_track_group',    label = 'add group',    icon = 'add_folder' )
		self.addTool( 'animator_track/add_track',    label = 'add',    icon = 'add' )
		self.addTool( 'animator_track/remove_track', label = 'remove', icon = 'remove' )

		#
		signals.connect( 'selection.changed', self.onSceneSelectionChanged )

		self.delegate = MOAILuaDelegate( self )
		self.delegate.load( _getModulePath( 'AnimatorView.lua' ) )

		self.widget.setOwner( self )

		#playback
		self.previewing = False
		self.setEditing( False )

		self.targetAnimator     = None
		self.targetClip         = None
		self.targetAnimatorData = None
		self.currentTrack       = None

		self.previewing  = False
		self.previewLoop = False
		self.previewTime = 0.0
		self.previewStep = 1.0/60.0

		self.previewTimer  = QtCore.QTimer( self.widget )
		self.previewTimer.setInterval( 1000.0/65 )
		self.previewTimer.stop()

		self.previewTimer.timeout.connect( self.onPreviewTimer )
Esempio n. 15
0
class AnimatorView( SceneEditorModule ):
	name = 'animator'
	dependency = [ 'scene_editor', 'mock' ]

	def onLoad( self ):
		#UI
		self.windowTitle = 'Animator'
		self.window = self.requestDockWindow( 'AnimatorView',
			title     = 'Animator',
			size      = (120,120),
			minSize   = (120,120),
			dock      = 'bottom'
			)
		
		self.widget = AnimatorWidget()
		self.window.addWidget( self.widget )
		self.toolbarTarget = self.addToolBar( 'animator_target', self.widget.toolbarTarget )
		self.toolbarClips  = self.addToolBar( 'animator_clips', self.widget.toolbarClips )
		self.toolbarPlay   = self.addToolBar( 'animator_play',  self.widget.toolbarPlay )
		self.toolbarTrack  = self.addToolBar( 'animator_track', self.widget.toolbarTrack )
		# self.toolbarEdit  = self.addToolBar( 'animator_play',  self.widget.toolbarEdit )

		signals.connect( 'scene.close', self.onSceneClose )
		signals.connect( 'scene.save', self.preSceneSave )
		signals.connect( 'scene.saved', self.postSceneSave )

		# addWidgetWithLaytut( toolbar,
		# 	self.widget.containerEditTool )
		self.addTool( 'animator_target/change_context', label = 'Change Context', icon = 'in' )
		self.addTool( 'animator_target/save_data', label = 'Save Data', icon = 'save' )

		self.addTool( 'animator_clips/add_clip_group',   label = 'add group',    icon = 'add_folder' )
		self.addTool( 'animator_clips/add_clip',    label = 'add',    icon = 'add' )
		self.addTool( 'animator_clips/remove_clip', label = 'remove', icon = 'remove' )
		self.addTool( 'animator_clips/clone_clip', label = 'clone', icon = 'clone' )


		self.addTool( 'animator_play/goto_start', label = 'to start',  icon = 'rewind' )
		# self.addTool( 'animator_play/prev_key',   label = 'prev key',      icon = 'previous' )
		self.addTool( 'animator_play/stop',       label = 'stop',      icon = 'stop' )
		self.addTool( 'animator_play/play',       label = 'play',      icon = 'play',  type = 'check' )
		# self.addTool( 'animator_play/next_key',   label = 'next key',      icon = 'next' )
		self.addTool( 'animator_play/goto_end',   label = 'to end',    icon = 'fast_forward' )
		self.addTool( 'animator_play/toggle_repeat',  label = 'toggle repeat',  icon = 'repeat', type = 'check' )
		self.comboPreviewSpeed = QtGui.QComboBox()
		self.comboPreviewSpeed.addItems([ e[0] for e in PREVIEW_SPEED_OPTIONS ] )			
		self.comboPreviewSpeed.setCurrentIndex( 4 ) #1x
		self.comboPreviewSpeed.currentIndexChanged.connect( self.onPreviewSpeedChange )
		self.addTool( 'animator_play/preview_speed', widget = self.comboPreviewSpeed )
		
		#SIGNALS
		self.addTool( 'animator_track/locate_target', label = 'locate', icon = 'find' )
		self.addTool( 'animator_track/----' )
		self.addTool( 'animator_track/add_track_group',    label = 'add group',    icon = 'add_folder' )
		self.addTool( 'animator_track/add_track',    label = 'add',    icon = 'add' )
		self.addTool( 'animator_track/remove_track', label = 'remove', icon = 'remove' )

		#
		signals.connect( 'selection.changed', self.onSceneSelectionChanged )

		self.delegate = MOAILuaDelegate( self )
		self.delegate.load( _getModulePath( 'AnimatorView.lua' ) )

		self.widget.setOwner( self )

		#playback
		self.previewing = False
		self.setEditing( False )

		self.targetAnimator     = None
		self.targetClip         = None
		self.targetAnimatorData = None
		self.currentTrack       = None

		self.previewing  = False
		self.previewLoop = False
		self.previewTime = 0.0
		self.previewStep = 1.0/60.0

		self.previewTimer  = QtCore.QTimer( self.widget )
		self.previewTimer.setInterval( 1000.0/65 )
		self.previewTimer.stop()

		self.previewTimer.timeout.connect( self.onPreviewTimer )

	def onStart( self ):
		pass

	def setEditing( self, editing ):
		self.widget.timeline.setEnabled( editing )
		self.widget.treeTracks.setEnabled( editing )
		self.findTool( 'animator_play'  ).setEnabled( editing )
		self.findTool( 'animator_track' ).setEnabled( editing )
		self.findTool( 'animator_clips/add_clip_group').setEnabled( editing )
		self.findTool( 'animator_clips/add_clip'      ).setEnabled( editing )
		self.findTool( 'animator_clips/remove_clip'   ).setEnabled( editing )
		self.findTool( 'animator_clips/clone_clip'    ).setEnabled( editing )

	def setTargetAnimator( self, target ):
		self.saveAnimatorData()
		if target == self.targetAnimator: return
		if self.previewing:
			self.stopPreview()
		self.targetAnimator = target
		self.targetClip     = None
		self.delegate.callMethod( 'view', 'setTargetAnimator', target )
		self.targetAnimatorData = self.delegate.callMethod( 'view', 'getTargetAnimatorData' )
		self.widget.rebuild()
		if self.targetAnimator:
			self.setEditing( True )
			signals.emit( 'animator.start' )
		else:
			self.setEditing( False )
			signals.emit( 'animator.stop' )
			
		path = self.delegate.callMethod( 'view', 'getTargetAnimatorDataPath' )
		if path:
			self.window.setWindowTitle( 'Animator - %s' % path )
		else:
			self.window.setWindowTitle( 'Animator' )
		clip = self.delegate.callMethod( 'view', 'getPreviousTargeClip', target )
		self.enableTool( 'animator_play' , False )
		self.enableTool( 'animator_track', False )
		if clip:
			self.widget.treeClips.selectNode( clip )
		else:
			self.widget.treeClips.selectFirstItem()
		self.applyTime( 0, True )

	def setTargetClip( self, clip ):
		wasPreviewing = self.previewing
		if self.previewing:
			self.stopPreview()

		self.targetClip = clip
		self.delegate.callMethod( 'view', 'setTargetClip', clip )
		self.widget.rebuildTimeline()
		self.enableTool( 'animator_play' , bool( clip ) )
		self.enableTool( 'animator_track', bool( clip ) )
		self.applyTime( 0, True )
		if wasPreviewing:
			self.startPreview()

	def setCurrentTrack( self, track ):
		self.currentTrack = track
		self.delegate.callMethod( 'view', 'setCurrentTrack', track )

	def getTargetClipLength( self ):
		return self.delegate.callMethod( 'view', 'getTargetClipLength' )

	def getClipList( self ):
		if self.targetAnimatorData:
			clipList = self.targetAnimatorData.clips
			return [ clip for clip in clipList.values()  ]
		else:
			return []

	def getRootClipGroup( self ):
		if self.targetAnimatorData:
			return self.targetAnimatorData.getRootGroup( self.targetAnimatorData )

	def getTrackList( self ):
		if self.targetClip:
			trackList = self.targetClip.getTrackList( self.targetClip )
			return [ track for track in trackList.values()  ]
		else:
			return []

	def getMarkerList( self ):
		if self.targetClip:
			markerList = self.targetClip.getMarkerList( self.targetClip )
			return [ track for track in markerList.values()  ]
		else:
			return []

	def getClipRoot( self ):
		if self.targetClip:
			return self.targetClip.getRoot( self.targetClip )
		else:
			return None

	def addClip( self ):
		if not self.targetAnimatorData: return
		targetGroup = self.widget.getCurrentClipGroup()
		cmd = self.doCommand( 'scene_editor/animator_add_clip',
			animator_data = self.targetAnimatorData,
			parent_group  = targetGroup
		  )
		clip = cmd.getResult()
		if clip:
			self.widget.addClip( clip, True )
		return clip

	def addClipGroup( self ):
		if not self.targetAnimatorData: return
		targetGroup = self.widget.getCurrentClipGroup()
		cmd = self.doCommand( 'scene_editor/animator_add_clip_group',
			animator_data = self.targetAnimatorData,
			parent_group  = targetGroup
		  )
		group = cmd.getResult()
		if group:
			self.widget.addClip( group, True )
		return group

	def removeClipNode( self ):
		for clip in self.widget.treeClips.getSelection():
			if self.doCommand( 'scene_editor/animator_remove_clip_node',
				animator_data = self.targetAnimatorData,
				target_node   = clip
			):
				self.widget.removeClip( clip )

	def cloneClipNode( self ):
		if not self.targetClip: return
		result = []
		for clip in self.widget.treeClips.getSelection():
			cmd = self.doCommand( 'scene_editor/animator_clone_clip_node',
				animator_data = self.targetAnimatorData,
				target_node   = clip
			)
			if cmd:
				cloned = cmd.getResult()
				self.widget.addClip( cloned )
				result.append( cloned )
		return result

	def onObjectEdited( self, obj ):
		if self.targetClip:
			self.delegate.callMethod( 'view', 'clearPreviewState' )
			self.delegate.callMethod( 'view', 'markClipDirty' )

	def onSceneSelectionChanged( self, selection, key ):
		if key != 'scene': return
		#find animator component
		# self.findTargetAnimator()

	def findTargetAnimator( self ):
		target = self.delegate.callMethod( 'view', 'findTargetAnimator' )
		self.setTargetAnimator( target )
		return target

	def checkTargetAnimator( self ):
		if not self.targetAnimator:
			alertMessage( 'No Animator', 'No Animator Selected', 'question' )
			return False
		return True

	def addMarker( self ):
		if not self.targetClip: return
		cmd = self.doCommand( 'scene_editor/animator_add_marker' ,
				target_clip = self.targetClip,
				target_pos  = self.widget.getCursorPos()
			)
		if cmd:
			marker = cmd.getResult()
			self.widget.addMarker( marker )

	def addKeyForField( self, target, fieldId ):
		if not self.checkTargetAnimator(): return 

		if not self.targetClip:
			self.addClip()
			# alertMessage( 'No Clip', 'You need to select a Clip first', 'question' )
			# return False
		keys = self.delegate.callMethod( 'view', 'addKeyForField', target, fieldId )
		if keys:
			for key in keys.values():
				self.widget.addKey( key, True )

	def addKeyForEvent( self, target, eventId ):
		pass

	def addCustomAnimatorTrack( self, target, trackClasId ):
		if not self.checkTargetAnimator(): return
			
		track = self.delegate.callMethod( 'view', 'addCustomAnimatorTrack', target, trackClasId )
		if track:
			self.widget.addTrack( track )

	def addKeyForSelectedTracks( self ):
		#TODO: command
		selectedTracks = self.widget.getTrackSelection()
		for track in selectedTracks:
			keys = self.delegate.callMethod( 'view', 'addKeyForSelectedTrack', track )
			if keys:
				for key in keys.values():
					self.widget.addKey( key, True )

	def removeSelectedKeys( self ):
		#TODO: command
		selectedKeys = self.widget.getKeySelection()
		for key in selectedKeys:
			self.widget.removeKey( key )

	def cloneSelectedKeys( self ):
		#TODO: command
		selectedKeys = self.widget.getKeySelection()
		cloned = []
		for key in selectedKeys:
			clonedKey = self.delegate.callMethod( 'view', 'cloneKey', key )
			if clonedKey:
				cloned.append( clonedKey )

		for clonedKey in cloned:
			self.widget.addKey( clonedKey, False )

	def onKeyRemoving( self, key ):
		if self.delegate.callMethod( 'view', 'removeKey', key ) != False:
			return True

	def onMarkerRemoving( self, marker ):
		if self.delegate.callMethod( 'view', 'removeMarker', marker ) != False:
			return True

	def onClipLengthChanging( self, t1 ):
		if self.delegate.callMethod( 'view', 'setTargetClipLength', t1 ) != False:
			return True

	def onTimelineKeyChanged( self, key, pos, length ):
		self.delegate.callMethod( 'view', 'updateTimelineKey', key, pos, length )

	def onTimelineKeyCurveValueChanged( self, key, value ):
		self.delegate.callMethod( 'view', 'updateTimelineKeyCurveValue', key, value )

	def onTimelineKeyTweenModeChanged( self, key, mode ):
		self.delegate.callMethod( 'view', 'updateTimelineKeyTweenMode', key, mode )

	def onTimelineKeyBezierPointChanged( self, key, bpx0, bpy0, bpx1, bpy1 ):
		self.delegate.callMethod( 'view', 'updateTimelineKeyBezierPoint', key, bpx0, bpy0, bpx1, bpy1 )

	def onTimelineMarkerChanged( self, marker, pos ):
		self.delegate.callMethod( 'view', 'updateTimelineMarker', marker, pos )

	def toggleTrackActive( self, track ):
		#TODO: command
		# self.module.doCommand( 'scene_editor/toggle_entity_visibility', target = node )
		self.delegate.callMethod( 'view', 'toggleTrackActive', track )


	def renameTrack( self, track, name ):
		self.delegate.callMethod( 'view', 'renameTrack', track, name )

	def renameClip( self, clip, name ):
		self.delegate.callMethod( 'view', 'renameClip', clip, name )

	def onTool( self, tool ):
		name = tool.name
		if name == 'change_context':
			target0 = self.targetAnimator
			target1 = self.findTargetAnimator()
			if ( not target0 ) and ( not target1 ):
				alertMessage( 'No Animator', 'No Animator found in selected entity scope', 'question' )
				
		elif name == 'save_data':
			self.saveAnimatorData()

		elif name == 'add_clip':
			if self.checkTargetAnimator():
				self.addClip()

		elif name == 'add_clip_group':
			if self.checkTargetAnimator():
				self.addClipGroup()

		elif name == 'remove_clip':
			if self.checkTargetAnimator():
				self.removeClipNode()			

		elif name == 'clone_clip':
			if self.checkTargetAnimator():
				self.cloneClipNode()			

		elif name == 'add_track_group':
			group = self.delegate.callMethod( 'view', 'addTrackGroup' )
			if group:
				self.widget.addTrack( group, True )

		elif name == 'remove_track':
			for track in self.widget.treeTracks.getSelection():
				self.delegate.callMethod( 'view', 'removeTrack', track )
				self.widget.removeTrack( track )
		elif name == 'locate_target':
			for track in self.widget.treeTracks.getSelection():
				sceneGraphEditor = self.getModule( 'scenegraph_editor')
				if sceneGraphEditor:
					targetEntity = self.delegate.callMethod( 'view', 'findTrackEntity', track )
					if targetEntity:
						sceneGraphEditor.selectEntity( targetEntity, focus_tree = True )
				#pass
				return

		#preview
		elif name == 'goto_start':
			self.gotoStart()
		elif name == 'goto_end':
			self.gotoEnd()
		elif name == 'play':
			if tool.getValue():
				self.startPreview()
			else:
				self.stopPreview( False )
		elif name == 'stop':
			self.stopPreview( True )
		elif name == 'toggle_repeat':
			self.delegate.callMethod( 'view', 'togglePreviewRepeat', tool.getValue() )
			

	def getActiveSceneView( self ):
		return self.getModule( 'scene_view' )

	#preview
	def startPreview( self ):
		self.saveAnimatorData()
		if self.delegate.callMethod( 'view', 'startPreview', self.previewTime ):
			self.widget.setCursorMovable( False )
			self.previewing = True
			self.findTool( 'animator_play/play' ).setValue( True )
			self.previewTimer.start()
			self.getApp().setMinimalMainLoopBudget()
			
	def stopPreview( self, rewind = False ):		
		if self.previewing:
			self.delegate.callMethod( 'view', 'stopPreview' )
			self.getApp().resetMainLoopBudget()
			self.widget.setCursorMovable( True )
			self.previewing = False
			self.findTool( 'animator_play/play' ).setValue( False )
			self.previewTimer.stop()
			signals.emit( 'entity.modified',  None , '' )
		if rewind:
			self.gotoStart()

	def onPreviewTimer( self ):
		playing, currentTime = self.delegate.callMethod( 'view', 'doPreviewStep' )
		self.previewTime = currentTime
		self.getActiveSceneView().forceUpdate()
		self.widget.setCursorPos( self.previewTime )
		if not playing:
			self.stopPreview()
		# signals.emit( 'entity.modified',  None , '' )

	def gotoStart( self ):
		if self.previewing:
			self.delegate.callMethod( 'view', 'applyTime', 0 )
		else:
			self.widget.setCursorPos( 0, True )

	def gotoEnd( self ):
		if self.previewing:
			self.delegate.callMethod( 'view', 'applyTime', 10 )
		else:
			self.widget.setCursorPos( 10, True )

	def applyTime( self, t, syncCursor = False ):
		self.previewTime = self.delegate.callMethod( 'view', 'applyTime', t )
		self.getActiveSceneView().forceUpdate()
		signals.emit( 'entity.modified',  None , '' )
		if syncCursor:
			self.widget.setCursorPos( t )

	def saveAnimatorData( self ):
		if not self.targetAnimator:
			return
		self.delegate.callMethod( 'view', 'saveData' )

	def preSceneSave( self ):
		if self.targetAnimator:
			self.delegate.callMethod( 'view', 'restoreEntityState' )

	def postSceneSave( self ):
		if self.targetAnimator:
			self.applyTime( self.previewTime )

	def onSceneClose( self, scene ):
		self.setTargetAnimator( None )

	def onPreviewSpeedChange( self, index ):
		label, throttle = PREVIEW_SPEED_OPTIONS[ index ]
		self.delegate.callMethod( 'view', 'setPreviewThrottle', throttle )

	def refreshTimeline( self ):
		self.widget.rebuildTimeline()

	def refreshClipList( self ):
		self.widget.rebuildClipList()

	def refreshAll( self ):
		self.widget.rebuild()
Esempio n. 16
0
    def onLoad(self):
        self.configPath = self.getProject().getConfigPath(_DEPLOY_CONFIG_FILE)
        #UI
        self.container = self.requestDocumentWindow('DeployManager',
                                                    title='Deployment Manager',
                                                    allowDock=False,
                                                    minSize=(300, 300),
                                                    maxSize=(300, 300))

        #Components
        self.window = self.container.addWidgetFromFile(
            _getModulePath('DeployManager.ui'))

        self.delegate = MOAILuaDelegate(self)
        self.delegate.load(_getModulePath('DeployManager.lua'))

        #scene tree
        layout = QtGui.QVBoxLayout()
        self.window.containerSceneTree.setLayout(layout)
        layout.setSpacing(0)
        layout.setMargin(0)

        self.treeScene = DeploySceneTree(self.window.containerSceneTree,
                                         editable=True,
                                         sorting=False,
                                         multiple_selection=False)
        self.treeScene.manager = self
        layout.addWidget(self.treeScene)

        sceneToolbar = QtGui.QToolBar(self.window.containerSceneTree)
        layout.addWidget(sceneToolbar)
        self.sceneTool = self.addToolBar('deploy_scene', sceneToolbar)
        self.addTool('deploy_scene/add_scene', label='add', icon='add')
        self.addTool('deploy_scene/remove_scene',
                     label='remove',
                     icon='remove')
        self.addTool('deploy_scene/move_up_scene', label='up', icon='arrow-up')
        self.addTool('deploy_scene/move_down_scene',
                     label='down',
                     icon='arrow-down')
        self.addTool('deploy_scene/----')
        self.addTool('deploy_scene/edit_scene',
                     label='change target scene',
                     icon='pencil')
        self.addTool('deploy_scene/----')
        self.addTool('deploy_scene/set_entry_scene',
                     label='set as entry',
                     icon='flag')

        #deploy target tree
        layout = QtGui.QVBoxLayout()
        self.window.containerTargetTree.setLayout(layout)
        layout.setSpacing(0)
        layout.setMargin(0)

        self.treeTarget = DeployTargetTree(self.window.containerTargetTree,
                                           editable=True,
                                           multiple_selection=False)
        self.treeTarget.manager = self
        layout.addWidget(self.treeTarget)

        targetToolbar = QtGui.QToolBar(self.window.containerTargetTree)
        layout.addWidget(targetToolbar)
        self.targetTool = self.addToolBar('deploy_target', targetToolbar)
        self.addTool('deploy_target/add_target', label='+')
        self.addTool('deploy_target/remove_target', label='-')

        #target property
        self.propertyTarget = addWidgetWithLayout(
            PropertyEditor(self.window.containerTargetProp))

        #menu
        self.addMenuItem('main/file/----')
        self.addMenuItem('main/file/deploy_manager',
                         dict(label='Deploy Manager', shortcut='F11'))

        self.addMenuItem('main/file/deploy_build',
                         dict(label='Deploy Build', shortcut='Ctrl+F11'))

        # self.container.show()
        self.window.buttonOK.clicked.connect(self.onButtonOK)

        #other
        registerSearchEnumerator(deployTargetSearchEnumerator)

        signals.connect('project.pre_deploy', self.preDeploy)
        signals.connect('project.deploy', self.onDeploy)
        signals.connect('project.post_deploy', self.postDeploy)
Esempio n. 17
0
class SceneReferenceEditor(SceneEditorModule):
    def __init__(self):
        super(SceneReferenceEditor, self).__init__()
        self.refreshScheduled = False

    def getName(self):
        return 'global_object_manager'

    def getDependency(self):
        return ['scene_editor', 'mock']

    def onLoad(self):
        #UI
        self.window = self.requestDockWindow('SceneReferenceEditor',
                                             title='Scene Reference',
                                             size=(120, 120),
                                             minSize=(120, 120),
                                             dock='right')

        #Components
        self.tree = self.window.addWidget(
            SceneReferenceTreeWidget(multiple_selection=True,
                                     editable=True,
                                     drag_mode='internal'))

        self.tool = self.addToolBar('global_object_manager',
                                    self.window.addToolBar())
        self.delegate = MOAILuaDelegate(self)
        self.delegate.load(_getModulePath('SceneReferenceEditor.lua'))

        self.addTool('global_object_manager/add_group',
                     label='Add Group',
                     icon='add_folder')
        self.addTool('global_object_manager/add', label='Add', icon='add')
        self.addTool('global_object_manager/remove',
                     label='Remove',
                     icon='remove')

        if self.getModule('introspector'):
            import SceneReferenceNodeEditor

        registerSearchEnumerator(globalObjectNameSearchEnumerator)
        registerSearchEnumerator(globalObjectSearchEnumerator)

    def onAppReady(self):
        self.tree.rebuild()

    def onMoaiClean(self):
        self.tree.clear()

    def onTool(self, tool):
        name = tool.name
        if name == 'add':
            requestSearchView(
                info='select global object class to create',
                context='global_object_class',
                on_selection=lambda objName: self.createGlobalObject(objName))

        if name == 'add_group':
            group = self.delegate.safeCall('addGroup')
            self.tree.addNode(group)
            self.tree.editNode(group)
            self.tree.selectNode(group)

        elif name == 'remove':
            for node in self.tree.getSelection():
                self.doCommand('scene_editor/remove_global_object',
                               target=node)
                self.tree.removeNode(node)

        elif name == 'refresh':
            self.scheduleRefreshObject()

    def onMenu(self, menu):
        if menu.name == 'find_global_object':
            requestSearchView(
                info='search for global object',
                context='global_object',
                on_selection=lambda node: self.selectObject(node, True))

    def needUpdate(self):
        return True

    def onUpdate(self):
        if self.refreshScheduled:
            self.refreshScheduled = False
            self.delegate.safeCall('reloadObjects')
            self.tree.rebuild()

    def scheduleRefreshObject(self):
        self.refreshScheduled = True

    def selectObject(self, target, updateTree=False):
        self.changeSelection(target)
        if updateTree:
            self.tree.blockSignals(True)
            self.tree.selectNode(None)
            if isinstance(target, list):
                for e in target:
                    self.tree.selectNode(e, add=True)
            else:
                self.tree.selectNode(target)
            self.tree.blockSignals(False)

    def renameObject(self, obj, name):
        obj.setName(obj, name)

    def createGlobalObject(self, objName):
        self.doCommand('scene_editor/create_global_object', name=objName)

    def onObjectAdded(self, node, reason='new'):
        self.tree.addNode(node)
        if reason == 'new':
            self.tree.editNode(node)
            self.tree.selectNode(node)

    def onObjectRemoved(self, node):
        self.tree.removeNode(node)

    def onObjectRenamed(self, node, name):
        self.tree.refreshNodeContent(node)
Esempio n. 18
0
class SceneGraphEditor(SceneEditorModule):
    def __init__(self):
        super(SceneGraphEditor, self).__init__()
        self.sceneDirty = False
        self.activeSceneNode = None
        self.refreshScheduled = False
        self.previewing = False
        self.workspaceState = None

    def getName(self):
        return 'scenegraph_editor'

    def getDependency(self):
        return ['scene_editor', 'mock']

    def onLoad(self):
        #UI
        self.windowTitle = 'Scenegraph'
        self.container = self.requestDockWindow('SceneGraphEditor',
                                                title='Scenegraph',
                                                size=(200, 200),
                                                minSize=(200, 200),
                                                dock='left')

        #Components
        self.treeFilter = self.container.addWidget(GenericTreeFilter(
            self.container),
                                                   expanding=False)
        self.tree = self.container.addWidget(
            SceneGraphTreeWidget(self.container,
                                 sorting=True,
                                 editable=True,
                                 multiple_selection=True,
                                 drag_mode='internal'))
        self.treeFilter.setTargetTree(self.tree)
        self.tree.module = self
        self.tool = self.addToolBar('scene_graph', self.container.addToolBar())
        self.delegate = MOAILuaDelegate(self)
        self.delegate.load(getModulePath('SceneGraphEditor.lua'))

        self.entityCreatorMenu = self.addMenu('main/scene/entity_create',
                                              {'label': 'Create Entity'})

        self.componentCreatorMenu = self.addMenu('main/scene/component_create',
                                                 {'label': 'Create Component'})

        #menu
        self.addMenuItem('main/file/open_scene',
                         dict(label='Open Scene', shortcut='ctrl+shift+o'))

        self.addMenuItem('main/file/close_scene',
                         dict(label='Close Scene', shortcut='Ctrl+W'))
        self.addMenuItem('main/scene/save_scene',
                         dict(label='Save', shortcut='Ctrl+S'))
        self.addMenuItem('main/scene/locate_scene_asset',
                         dict(label='Locate Scene Asset'))

        self.addMenu('main/scene/----')

        self.addMenu('component_context', dict(label='Selected Component'))
        self.addMenuItem('component_context/remove_component',
                         dict(label='Remove'))

        self.addMenuItem('component_context/----')

        self.addMenuItem('component_context/copy_component',
                         dict(label='Copy'))
        self.addMenuItem('component_context/paste_component',
                         dict(label='Paste Component Here'))
        self.addMenuItem('component_context/----')

        self.addMenuItem('component_context/move_component_up',
                         dict(label='Move Up'))

        self.addMenuItem('component_context/move_component_down',
                         dict(label='Move Down'))

        self.addMenuItem('component_context/----')
        self.addMenuItem('component_context/edit_component_alias',
                         dict(label='Edit Alias'))

        self.addMenu('main/entity', dict(label='Entity'))
        self.addMenuItem('main/entity/add_empty_entity',
                         dict(label='Create Empty', shortcut='ctrl+alt+N'))
        self.addMenuItem('main/entity/add_entity',
                         dict(label='Create', shortcut='ctrl+shift+N'))
        self.addMenuItem('main/entity/----')
        self.addMenuItem('main/entity/group_entity',
                         dict(label='Group Entites', shortcut='ctrl+G'))
        self.addMenuItem(
            'main/entity/create_group',
            dict(label='Create Empty Group', shortcut='ctrl+shift+G'))
        self.addMenuItem('main/entity/----')
        self.addMenuItem(
            'main/entity/load_prefab',
            dict(label='Load Prefab', shortcut='ctrl+alt+shift+N'))
        self.addMenuItem(
            'main/entity/load_prefab_in_container',
            dict(label='Load Prefab In Container', shortcut='ctrl+shift+='))
        self.addMenuItem('main/entity/----')
        self.addMenuItem('main/entity/remove_entity', dict(label='Remove'))
        self.addMenuItem('main/entity/clone_entity',
                         dict(label='Clone', shortcut='ctrl+d'))
        self.addMenuItem('main/entity/----')
        self.addMenuItem('main/entity/add_component',
                         dict(label='Add Component', shortcut='ctrl+alt+='))
        self.addMenuItem('main/entity/assign_layer',
                         dict(label='Assign Layer', shortcut='ctrl+alt+L'))
        self.addMenuItem('main/entity/toggle_visibility',
                         dict(label='Toggle Visibility', shortcut='ctrl+/'))
        self.addMenuItem('main/entity/freeze_entity_pivot',
                         dict(label='Freeze Pivot'))

        self.addMenuItem('main/entity/----')
        self.addMenuItem('main/find/find_entity',
                         dict(label='Find In Scene', shortcut='ctrl+f'))
        self.addMenuItem('main/find/find_entity_in_group',
                         dict(label='Find In Group', shortcut='ctrl+shift+f'))
        self.addMenuItem('main/find/find_entity_group',
                         dict(label='Find Group', shortcut='ctrl+alt+f'))

        #Toolbars
        self.addTool('scene_graph/select_scene',
                     label='Select Scene',
                     icon='settings')
        self.addTool('scene_graph/----')
        self.addTool('scene_graph/create_group',
                     label='+ Group',
                     icon='add_folder')
        self.addTool('scene_graph/----')
        self.addTool('scene_graph/make_proto',
                     label='Convert To Proto',
                     icon='proto_make')
        self.addTool('scene_graph/create_proto_instance',
                     label='Create Proto Instance',
                     icon='proto_instantiate')
        self.addTool('scene_graph/create_proto_container',
                     label='Create Proto Container',
                     icon='proto_container')
        self.addTool('scene_graph/----')
        self.addTool('scene_graph/fold_all', label='F')
        self.addTool('scene_graph/unfold_all', label='U')
        self.addTool('scene_graph/refresh_tree', label='R')
        # self.addTool( 'scene_graph/load_prefab', label = '+ P' )
        # self.addTool( 'scene_graph/save_prefab', label = '>>P' )

        self.addTool('scene/refresh', label='refresh', icon='refresh')

        #SIGNALS
        signals.connect('moai.clean', self.onMoaiClean)

        signals.connect('scene.clear', self.onSceneClear)
        signals.connect('scene.change', self.onSceneChange)

        signals.connect('selection.changed', self.onSelectionChanged)
        signals.connect('selection.hint', self.onSelectionHint)

        signals.connect('preview.start', self.onPreviewStart)
        signals.connect('preview.stop', self.onPreviewStop)

        # signals.connect( 'animator.start',     self.onAnimatorStart     )
        # signals.connect( 'animator.stop' ,     self.onAnimatorStop      )

        signals.connect('entity.added', self.onEntityAdded)
        signals.connect('entity.removed', self.onEntityRemoved)
        signals.connect('entity.renamed', self.onEntityRenamed)
        signals.connect('entity.modified', self.onEntityModified)
        signals.connect('entity.visible_changed', self.onEntityVisibleChanged)
        signals.connect('entity.pickable_changed',
                        self.onEntityPickableChanged)

        signals.connect('prefab.unlink', self.onPrefabUnlink)
        signals.connect('prefab.relink', self.onPrefabRelink)
        signals.connect('proto.unlink', self.onPrefabUnlink)
        signals.connect('proto.relink', self.onPrefabRelink)

        signals.connect('app.ready', self.postAppReady)

        signals.connect('component.added', self.onComponentAdded)
        signals.connect('component.removed', self.onComponentRemoved)

        signals.connect('project.presave', self.preProjectSave)

        registerSearchEnumerator(sceneObjectSearchEnumerator)
        registerSearchEnumerator(entityNameSearchEnumerator)
        registerSearchEnumerator(componentNameSearchEnumerator)
        registerSearchEnumerator(layerNameSearchEnumerator)

    def onStart(self):
        self.refreshCreatorMenu()

    def postAppReady(self):
        self.openPreviousScene()

    def openPreviousScene(self):
        previousScene = self.getConfig('previous_scene', None)
        if previousScene:
            node = self.getAssetLibrary().getAssetNode(previousScene)
            if node:
                node.edit()

    def onStop(self):
        if self.activeSceneNode:
            self.setConfig('previous_scene',
                           self.activeSceneNode.getNodePath())
        else:
            self.setConfig('previous_scene', False)

    def onSetFocus(self):
        self.container.show()
        self.container.raise_()
        self.container.setFocus()

    def getActiveScene(self):
        return self.delegate.safeCallMethod('editor', 'getScene')

    def getActiveSceneRootGroup(self):
        scene = self.delegate.safeCallMethod('editor', 'getScene')
        if scene:
            return scene.rootGroup
        else:
            return None

    def openScene(self, node, protoNode=None):
        if self.activeSceneNode == node:
            if self.getModule('scene_view'):
                self.getModule('scene_view').setFocus()

            if protoNode:
                self.delegate.safeCallMethod('editor', 'locateProto',
                                             protoNode.getPath())
                if self.getModule('scene_view'):
                    self.getModule('scene_view').focusSelection()

        else:
            if not self.closeScene(): return
            if self.getModule('scene_view'):
                self.getModule('scene_view').makeCanvasCurrent()
            self.activeSceneNode = node
            signals.emitNow('scene.pre_open', node)
            scene = self.delegate.safeCallMethod('editor', 'openScene',
                                                 node.getPath())
            if not scene:
                #todo: raise something
                alertMessage(
                    'error',
                    '%s\n\nfailed to open scene, see console for detailed information.'
                    % node.getPath())
                return False
            signals.emitNow('scene.open', self.activeSceneNode, scene)
            self.setFocus()
            self.editingProtoNode = protoNode
            self.loadWorkspaceState(False)
            self.delegate.safeCallMethod('editor', 'postOpenScene')

    def closeScene(self):
        if not self.activeSceneNode: return True
        if self.sceneDirty:
            res = requestConfirm('scene modified!', 'save scene before close?')
            if res == True:  #save
                self.saveScene()
            elif res == None:  #cancel
                return
            elif res == False:  #no save
                pass

        self.markSceneDirty(False)
        self.tree.clear()
        self.getApp().clearCommandStack('scene_editor')
        self.getSelectionManager().removeSelection(self.getActiveScene())
        signals.emitNow('scene.close', self.activeSceneNode)
        self.delegate.safeCallMethod('editor', 'closeScene')
        self.activeSceneNode = None
        return True

    def onSceneClear(self):
        # self.tree.clear()
        pass

    def markSceneDirty(self, dirty=True):
        if not self.previewing:
            self.sceneDirty = dirty

    def saveWorkspaceState(self):
        self.retainWorkspaceState()
        treeFoldState = self.workspaceState['tree_state']
        containerFoldState = self.workspaceState['container_state']
        entityLockState = self.workspaceState['entity_lock_state']
        self.activeSceneNode.setMetaData('tree_state', treeFoldState)
        self.activeSceneNode.setMetaData('container_state', containerFoldState)
        self.activeSceneNode.setMetaData('entity_lock_state', entityLockState)

    def loadWorkspaceState(self, restoreState=True):
        treeFoldState = self.activeSceneNode.getMetaData('tree_state', None)
        containerFoldState = self.activeSceneNode.getMetaData(
            'container_state', None)
        entityLockState = self.activeSceneNode.getMetaData(
            'entity_lock_state', None)
        self.workspaceState = {
            'tree_state': treeFoldState,
            'container_state': containerFoldState,
            'entity_lock_state': entityLockState
        }
        if restoreState: self.restoreWorkspaceState()

    def retainWorkspaceState(self):
        #tree node foldstate
        treeFoldState = self.tree.saveFoldState()
        #save introspector foldstate
        introspectorFoldState = self.delegate.safeCallMethod(
            'editor', 'saveIntrospectorFoldState')
        entityLockState = self.delegate.safeCallMethod('editor',
                                                       'saveEntityLockState')

        self.workspaceState = {
            'tree_state': treeFoldState,
            'container_state': introspectorFoldState,
            'entity_lock_state': entityLockState
        }

    def restoreWorkspaceState(self):
        if not self.workspaceState: return
        treeState = self.workspaceState.get('tree_state', None)
        if treeState:
            self.tree.loadFoldState(treeState)

        containerState = self.workspaceState.get('container_state', None)
        if containerState:
            self.delegate.safeCallMethod('editor', 'loadIntrospectorFoldState',
                                         containerState)

        lockState = self.workspaceState.get('entity_lock_state', None)
        if lockState:
            self.delegate.safeCallMethod('editor', 'loadEntityLockState',
                                         lockState)

    def onSceneChange(self):
        self.tree.hide()
        self.tree.rebuild()
        self.restoreWorkspaceState()
        self.tree.refreshAllContent()
        self.tree.verticalScrollBar().setValue(0)
        self.tree.show()
        if self.editingProtoNode:
            self.delegate.safeCallMethod('editor', 'locateProto',
                                         self.editingProtoNode.getPath())
            self.editingProtoNode = None
            if self.getModule('scene_view'):
                self.getModule('scene_view').focusSelection()

    def saveScene(self):
        if not self.activeSceneNode: return
        self.markSceneDirty(False)
        signals.emitNow('scene.save')
        self.delegate.safeCallMethod('editor', 'saveScene',
                                     self.activeSceneNode.getAbsFilePath())
        signals.emitNow('scene.saved')
        self.saveWorkspaceState()

    def refreshScene(self):
        if not self.activeSceneNode: return
        if self.previewing: return
        self.refreshScheduled = False
        node = self.activeSceneNode
        self.retainWorkspaceState()
        if self.delegate.safeCallMethod('editor', 'refreshScene'):
            self.restoreWorkspaceState()
        self.refreshCreatorMenu()

    def scheduleRefreshScene(self):
        if not self.activeSceneNode: return
        self.refreshScheduled = True

    def refreshCreatorMenu(self):
        def addEntityMenuItem(name):
            if name == '----':
                self.entityCreatorMenu.addChild('----')
                return
            self.entityCreatorMenu.addChild({
                'name': 'create_entity_' + name,
                'label': name,
                'command': 'scene_editor/create_entity',
                'command_args': dict(name=name)
            })

        def addComponentMenuItem(name):
            if name == '----':
                self.componentCreatorMenu.addChild('----')
                return
            self.componentCreatorMenu.addChild({
                'name': 'create_component_' + name,
                'label': name,
                'command': 'scene_editor/create_component',
                'command_args': dict(name=name)
            })

        self.entityCreatorMenu.clear()
        self.componentCreatorMenu.clear()

        registry = _MOCK.getEntityRegistry()
        #entity
        keys = sorted(registry.keys())
        addEntityMenuItem('Entity')
        addEntityMenuItem('----')
        for entityName in sorted(registry.keys()):
            if entityName != 'Entity': addEntityMenuItem(entityName)

        #component
        registry = _MOCK.getComponentRegistry()
        for comName in sorted(registry.keys()):
            addComponentMenuItem(comName)

    def needUpdate(self):
        return True

    def onUpdate(self):
        if self.refreshScheduled:
            self.refreshScene()

    def preProjectSave(self, prj):
        if self.activeSceneNode:
            _MOCK.game.previewingScene = self.activeSceneNode.getNodePath()

    def onMoaiClean(self):
        self.tree.clear()

    def onTool(self, tool):
        name = tool.name

        if name == 'fold_all':
            self.tree.foldAllItems()

        elif name == 'unfold_all':
            self.tree.expandAllItems()

        elif name == 'refresh_tree':
            self.tree.rebuild()

        elif name == 'refresh':
            self.scheduleRefreshScene()

        elif name == 'make_proto':
            self.makeProto()

        elif name == 'create_proto_instance':
            requestSearchView(info='select a proto to instantiate',
                              context='asset',
                              type='proto',
                              on_selection=lambda obj: self.doCommand(
                                  'scene_editor/create_proto_instance',
                                  proto=obj.getNodePath()))

        elif name == 'create_proto_container':
            requestSearchView(
                info='select a proto to create contained instance',
                context='asset',
                type='proto',
                on_selection=lambda obj: self.doCommand(
                    'scene_editor/create_proto_container',
                    proto=obj.getNodePath()))

        elif name == 'create_group':
            self.doCommand('scene_editor/entity_group_create')

        elif name == 'group_entity':
            self.doCommand('scene_editor/group_entities')

        elif name == 'select_scene':
            self.doCommand('scene_editor/select_scene')

    def onMenu(self, menu):
        name = menu.name
        if name == 'close_scene':
            if self.previewing:
                alertMessage('Warning', 'Stop previewing before closing scene')
                return
            self.closeScene()

        elif name == 'open_scene':
            if self.previewing:
                alertMessage('Warning', 'Stop previewing before opening scene')
                return
            requestSearchView(info='select scene to open',
                              context='asset',
                              type='scene',
                              on_selection=self.openScene)

        elif name == 'save_scene':
            if self.previewing:
                alertMessage('Warning', 'Stop previewing before saving')
                return
            self.saveScene()

        elif name == 'locate_scene_asset':
            if self.activeSceneNode:
                assetBrowser = self.getModule('asset_browser')
                if assetBrowser:
                    assetBrowser.selectAsset(self.activeSceneNode)

        elif name == 'add_entity':
            requestSearchView(info='select entity type to create',
                              context='entity_creation',
                              on_selection=lambda obj: self.doCommand(
                                  'scene_editor/create_entity', name=obj))

        elif name == 'add_component':
            requestSearchView(info='select component type to create',
                              context='component_creation',
                              on_selection=lambda obj: self.doCommand(
                                  'scene_editor/create_component', name=obj))

        elif name == 'add_empty_entity':
            self.doCommand('scene_editor/create_entity', name='Entity')

        elif name == 'load_prefab':
            requestSearchView(info='select a perfab node to instantiate',
                              context='asset',
                              type='prefab',
                              on_selection=lambda obj: self.doCommand(
                                  'scene_editor/create_prefab_entity',
                                  prefab=obj.getNodePath()))

        elif name == 'load_prefab_in_container':
            requestSearchView(
                info='select a perfab node to instantiate( PefabContainer )',
                context='asset',
                type='prefab',
                on_selection=lambda obj: self.doCommand(
                    'scene_editor/create_prefab_container',
                    prefab=obj.getNodePath()))

        elif name == 'remove_entity':
            self.doCommand('scene_editor/remove_entity')

        elif name == 'clone_entity':
            self.doCommand('scene_editor/clone_entity')

        elif name == 'find_entity':
            requestSearchView(
                info='search for entity in current scene',
                context='scene',
                type='entity',
                on_selection=lambda x: self.selectEntity(x, focus_tree=True),
                on_test=self.selectEntity)

        elif name == 'find_entity_in_group':
            requestSearchView(
                info='search for entity in current entity group',
                context='scene',
                type='entity_in_group',
                on_selection=lambda x: self.selectEntity(x, focus_tree=True),
                on_test=self.selectEntity)

        elif name == 'find_entity_group':
            requestSearchView(
                info='search for group in current scene',
                context='scene',
                type='group',
                on_selection=lambda x: self.selectEntity(x, focus_tree=True),
                on_test=self.selectEntity)

        elif name == 'create_group':
            self.doCommand('scene_editor/entity_group_create')

        elif name == 'remove_component':
            context = menu.getContext()
            if context:
                self.doCommand('scene_editor/remove_component', target=context)

        elif name == 'copy_component':
            context = menu.getContext()
            if context:
                self.doCommand('scene_editor/copy_component', target=context)

        elif name == 'edit_component_alias':
            context = menu.getContext()
            if context:
                oldAlias = context._alias or ''
                alias = requestString('Edit Alias', 'Enter Alias:', oldAlias)
                if alias != None:
                    if not alias: alias = False
                    self.doCommand('scene_editor/rename_component',
                                   target=context,
                                   alias=alias)

        elif name == 'assign_layer':
            if not self.tree.getSelection(): return
            requestSearchView(info='select layer to assign',
                              context='scene_layer',
                              type=_MOCK.Entity,
                              on_selection=self.assignEntityLayer)

        elif name == 'toggle_visibility':
            self.doCommand('scene_editor/toggle_entity_visibility')

        elif name == 'freeze_entity_pivot':
            self.doCommand('scene_editor/freeze_entity_pivot')

    def onSelectionChanged(self, selection, key):
        if key != 'scene': return
        if self.tree.syncSelection:
            self.tree.blockSignals(True)
            self.tree.selectNode(None)
            for e in selection:
                self.tree.selectNode(e, add=True)
            self.tree.blockSignals(False)

    def selectEntity(self, target, **option):
        if option.get('focus_tree', False):
            self.tree.setFocus()
        self.changeSelection(target)

    ##----------------------------------------------------------------##
    def renameEntity(self, target, name):
        #TODO:command pattern
        target.setName(target, name)
        signals.emit('entity.modified', target)

    def addEntityNode(self, entity):
        self.tree.addNode(entity, expanded=False)
        self.tree.setNodeExpanded(entity, False)

    def removeEntityNode(self, entity):
        self.tree.removeNode(entity)

    def assignEntityLayer(self, layerName):
        #TODO:command pattern
        if not layerName: return
        self.doCommand('scene_editor/assign_layer', target=layerName)

    def onSelectionHint(self, selection):
        if selection._entity:
            self.changeSelection(selection._entity)
        else:
            self.changeSelection(selection)

    def onPreviewStart(self):
        if not self.activeSceneNode: return
        self.retainWorkspaceState()
        self.delegate.safeCallMethod('editor', 'retainScene')
        self.delegate.safeCallMethod('editor', 'startScenePreview')
        self.previewing = True

    def onPreviewStop(self):
        if not self.activeSceneNode: return
        self.changeSelection(None)
        self.tree.clear()
        self.delegate.safeCallMethod('editor', 'stopScenePreview')
        self.previewing = False
        if self.delegate.safeCallMethod('editor', 'restoreScene'):
            self.restoreWorkspaceState()

    def onAnimatorStart(self):
        self.retainWorkspaceState()
        self.delegate.safeCallMethod('editor', 'retainScene')

    def onAnimatorStop(self):
        self.tree.clear()
        self.delegate.safeCallMethod('editor', 'clearScene')
        if self.delegate.safeCallMethod('editor', 'restoreScene'):
            self.restoreWorkspaceState()

    ##----------------------------------------------------------------##
    def updateEntityPriority(self):
        if not self.activeSceneNode: return
        self.markSceneDirty()

    def onEntityRenamed(self, entity, newname):
        self.tree.refreshNodeContent(entity)
        self.markSceneDirty()

    def onEntityVisibleChanged(self, entity):
        self.tree.refreshNodeContent(entity)

    def onEntityPickableChanged(self, entity):
        self.tree.refreshNodeContent(entity)

    def onEntityAdded(self, entity, context=None):
        if context == 'new':
            self.setFocus()
            pnode = entity.parent
            if pnode:
                self.tree.setNodeExpanded(pnode, True)
            self.tree.setFocus()
            self.tree.editNode(entity)
            self.tree.selectNode(entity)
        signals.emit('scene.update')
        self.markSceneDirty()

    def onEntityRemoved(self, entity):
        signals.emit('scene.update')
        self.markSceneDirty()

    def onEntityModified(self, entity, context=None):
        self.markSceneDirty()

    ##----------------------------------------------------------------##
    def onComponentAdded(self, com, entity):
        signals.emit('scene.update')
        self.markSceneDirty()

    def onComponentRemoved(self, com, entity):
        signals.emit('scene.update')
        self.markSceneDirty()

    ##----------------------------------------------------------------##
    def onPrefabUnlink(self, entity):
        self.tree.refreshNodeContent(entity, updateChildren=True)

    def onPrefabRelink(self, entity):
        self.tree.refreshNodeContent(entity, updateChildren=True)

    def createPrefab(self, targetPrefab):
        selection = self.getSelection()
        if not selection: return
        if len(selection) > 1:
            return alertMessage(
                'multiple entities cannot be converted into prefab')
        target = selection[0]
        self.doCommand('scene_editor/create_prefab',
                       entity=target,
                       prefab=targetPrefab.getNodePath(),
                       file=targetPrefab.getAbsFilePath())

    def makeProto(self):
        selection = self.getSelection()
        if not selection: return
        if len(selection) > 1:
            return alertMessage(
                'multiple entities cannot be converted into Proto')
        target = selection[0]
        if not target: return
        if requestConfirm('convert proto', 'Convert this Entity into Proto?'):
            self.doCommand('scene_editor/make_proto', entity=target)
            self.tree.refreshNodeContent(target)

    def createProtoInstance(self):
        pass

    ##----------------------------------------------------------------##
    def onCopyEntity(self):
        entityGroupData = self.delegate.callMethod(
            'editor', 'makeSceneSelectionCopyData')
        if not entityGroupData: return False
        clip = QtGui.QApplication.clipboard()
        mime = QtCore.QMimeData()
        text = ''
        for s in self.getSelection():
            if text == '':
                text = text + s.name
            else:
                text = text + '\n' + s.name
        mime.setText(text)
        mime.setData(GII_MIME_ENTITY_DATA, entityGroupData.encode('utf-8'))
        clip.setMimeData(mime)
        return True

    def onPasteEntity(self):
        clip = QtGui.QApplication.clipboard()
        mime = clip.mimeData()
        if mime.hasFormat(GII_MIME_ENTITY_DATA):
            data = mime.data(GII_MIME_ENTITY_DATA)
            self.doCommand('scene_editor/paste_entity',
                           data=str(data).decode('utf-8'))

    ##----------------------------------------------------------------##
    def onCopyComponent(self):
        entityGroupData = self.delegate.callMethod('editor',
                                                   'makeEntityCopyData')
        if not entityGroupData: return False
        clip = QtGui.QApplication.clipboard()
        mime = QtCore.QMimeData()
        text = ''
        for s in self.getSelection():
            if text == '':
                text = text + s.name
            else:
                text = text + '\n' + s.name
        mime.setText(text)
        mime.setData(GII_MIME_ENTITY_DATA, str(entityGroupData))
        clip.setMimeData(mime)
        return True

    def onPasteComponent(self):
        clip = QtGui.QApplication.clipboard()
        mime = clip.mimeData()
        if mime.hasFormat(GII_MIME_ENTITY_DATA):
            data = mime.data(GII_MIME_ENTITY_DATA)
            self.doCommand('scene_editor/paste_entity', data=str(data))
Esempio n. 19
0
class DeployManager( SceneEditorModule ):
	def __init__(self):
		super( DeployManager, self ).__init__()

	def getName( self ):
		return 'deploy_manager'

	def getDependency( self ):
		return [ 'mock' ]

	def onLoad( self ):
		self.configPath = self.getProject().getConfigPath( _DEPLOY_CONFIG_FILE )
		#UI
		self.container = self.requestDocumentWindow( 'DeployManager',
			title     = 'Deployment Manager',
			allowDock = False,
			minSize   = ( 300, 300 ),
			maxSize   = ( 300, 300 )
			)

		#Components
		self.window = self.container.addWidgetFromFile( _getModulePath('DeployManager.ui') )

		self.delegate = MOAILuaDelegate( self )
		self.delegate.load( _getModulePath( 'DeployManager.lua' ) )

		#scene tree
		layout = QtGui.QVBoxLayout()
		self.window.containerSceneTree.setLayout( layout )
		layout.setSpacing( 0 )
		layout.setMargin( 0 )

		self.treeScene = DeploySceneTree( 
			self.window.containerSceneTree,
			editable = True,
			sorting  = False,
			multiple_selection = False
			)
		self.treeScene.manager = self
		layout.addWidget( self.treeScene )

		sceneToolbar = QtGui.QToolBar( self.window.containerSceneTree )
		layout.addWidget( sceneToolbar )
		self.sceneTool = self.addToolBar( 'deploy_scene', sceneToolbar )
		self.addTool( 'deploy_scene/add_scene',         label = 'add'     ,icon = 'add'    )
		self.addTool( 'deploy_scene/remove_scene',      label = 'remove'  ,icon = 'remove' )
		self.addTool( 'deploy_scene/move_up_scene',     label = 'up'      ,icon = 'arrow-up'     )
		self.addTool( 'deploy_scene/move_down_scene',   label = 'down'    ,icon = 'arrow-down'   )
		self.addTool( 'deploy_scene/----' )
		self.addTool( 'deploy_scene/edit_scene',   label = 'change target scene' ,icon = 'pencil'   )
		self.addTool( 'deploy_scene/----' )
		self.addTool( 'deploy_scene/set_entry_scene',   label = 'set as entry' ,icon = 'flag'   )

		#deploy target tree
		layout = QtGui.QVBoxLayout()
		self.window.containerTargetTree.setLayout( layout )
		layout.setSpacing( 0 )		
		layout.setMargin( 0 )

		self.treeTarget = DeployTargetTree( 
			self.window.containerTargetTree,
			editable = True,
			multiple_selection = False
			)
		self.treeTarget.manager = self
		layout.addWidget( self.treeTarget )

		targetToolbar = QtGui.QToolBar( self.window.containerTargetTree )
		layout.addWidget( targetToolbar )
		self.targetTool = self.addToolBar( 'deploy_target', targetToolbar )
		self.addTool( 'deploy_target/add_target',    label = '+' )
		self.addTool( 'deploy_target/remove_target', label = '-' )

		#target property
		self.propertyTarget = addWidgetWithLayout(
			PropertyEditor( self.window.containerTargetProp )
		)

		#menu
		self.addMenuItem( 'main/file/----' )
		self.addMenuItem( 'main/file/deploy_manager', 
			dict( label = 'Deploy Manager', shortcut = 'F11' )
			)

		self.addMenuItem( 'main/file/deploy_build', 
			dict( label = 'Deploy Build', shortcut = 'Ctrl+F11' )
			)
		
		
		# self.container.show()
		self.window.buttonOK.clicked.connect( self.onButtonOK )

		#other
		registerSearchEnumerator( deployTargetSearchEnumerator )

		signals.connect( 'project.pre_deploy', self.preDeploy )
		signals.connect( 'project.deploy', self.onDeploy )
		signals.connect( 'project.post_deploy', self.postDeploy )


	def onStart( self ):
		#load config
		self.loadConfig()
		#fill trees
		self.treeTarget.rebuild()
		self.treeScene.rebuild()

	def onStop( self ):
		self.saveConfig()

	def loadConfig( self ):
		self.delegate.safeCall( 'loadDeployManagerConfig', self.configPath )

	def saveConfig( self ):
		self.delegate.safeCall( 'saveDeployManagerConfig', self.configPath )		

	def getDeployTargetTypes( self ):
		registry = self.delegate.safeCall( 'getDeployTargetTypeRegistry' )
		return [ name for name in registry.keys() ]

	def getDeployTargets( self ):
		targets =  self.delegate.safeCallMethod( 'config', 'getTargets' )
		return [ obj for obj in targets.values() ]

	def addDeployTarget( self, targetType ):
		target = self.delegate.safeCallMethod( 'config', 'addDeployTarget', targetType )
		self.treeTarget.addNode( target )
		self.treeTarget.editNode( target )

	def changeDeployScene( self, targetScene ):
		for sceneEntry in self.treeScene.getSelection():
			self.delegate.safeCallMethod( 'config', 'changeTargetScene', sceneEntry, targetScene.getPath() )
			self.treeScene.refreshNode( sceneEntry )
			return

	def renameDeployTarget( self, target, name ):
		target.name = name #avoid duplicated name

	def addDeployScene( self, sceneNode ):
		if not sceneNode: return
		entry = self.delegate.safeCallMethod( 'config', 'addDeployScene', sceneNode.getNodePath() )
		self.treeScene.addNode( entry )
		self.treeScene.editNode( entry )

	def renameDeployScene( self, entry, alias ):
		entry.alias = alias #TODO: avoid duplicated name

	def getDeployScenes( self ):
		scenes =  self.delegate.safeCallMethod( 'config', 'getScenes' )
		return [ obj for obj in scenes.values() ]

	def updateGameConfig( self ):
		self.delegate.safeCallMethod( 'config', 'updateGameConfig' )

	def preDeploy( self, context ):
		self.updateGameConfig()

	def onDeploy( self, context ):
		pass

	def postDeploy( self, context ):
		pass

	def onTool( self, tool ):
		name = tool.name
		if name == 'add_target':
			requestSearchView( 
				info    = 'select deploy target type',
				context = 'deploy_target_type',
				on_selection = self.addDeployTarget
				)

		elif name == 'remove_target':
			for target in self.treeTarget.getSelection():
				self.treeTarget.removeNode( target )
				self.delegate.safeCallMethod( 'config', 'removeDeployTarget', target )

		elif name == 'add_scene':
			requestSearchView( 
				info    = 'select scene to deploy',
				context = 'asset',
				type    = 'scene',
				on_selection = self.addDeployScene
				)

		elif name == 'edit_scene':
			requestSearchView( 
				info    = 'select new target scene ',
				context = 'asset',
				type    = 'scene',
				on_selection = self.changeDeployScene
				)

		elif name == 'remove_scene':
			for entry in self.treeScene.getSelection():	
				self.delegate.safeCallMethod( 'config', 'removeDeployScene', entry )
				self.treeScene.removeNode( entry )
			self.treeScene.refreshAllContent()

		elif name == 'set_entry_scene':
			for entry in self.treeScene.getSelection():
				self.delegate.safeCallMethod( 'config', 'setEntryScene', entry )
				break
			self.treeScene.refreshAllContent()

		elif name == 'move_up_scene':
			for target in self.treeScene.getSelection():
				self.delegate.safeCallMethod( 'config', 'moveSceneUp', target )
				self.treeScene.rebuild()
				self.treeScene.selectNode( target )
				break

		elif name == 'move_down_scene':
			for target in self.treeScene.getSelection():
				self.delegate.safeCallMethod( 'config', 'moveSceneDown', target )
				self.treeScene.rebuild()
				self.treeScene.selectNode( target )
				break

	def onMenu( self, node ):
		name = node.name
		if name == 'deploy_manager' :
			self.onSetFocus()

		elif name == 'deploy_build':
			app.getProject().deploy()

	def onSetFocus( self ):
		self.container.show()
		self.container.raise_()
	
	def onButtonOK( self ):
		self.saveConfig()
		self.container.hide()
Esempio n. 20
0
class LayerManager( SceneEditorModule ):
	def __init__(self):
		super( LayerManager, self ).__init__()

	def getName( self ):
		return 'layer_manager'

	def getDependency( self ):
		return [ 'scene_editor', 'mock' ]

	def onLoad( self ):
		#UI
		self.windowTitle = 'Layers'
		self.window = self.requestDockWindow( 'LayerManager',
			title     = 'Layers',
			size      = (120,120),
			minSize   = (120,120),
			dock      = 'left'
			)

		#Components
		self.treeFilter = self.window.addWidget( GenericTreeFilter() )
		self.tree = self.window.addWidget( 
			LayerTreeWidget(
				self.window,
				multiple_selection = False,
				sorting            = False,
				editable           = True,
				drag_mode          = 'internal'
				)
			)
		self.tree.hasSoloLayer = False
		self.treeFilter.setTargetTree( self.tree )

		self.tool = self.addToolBar( 'layer_manager', self.window.addToolBar() )
		self.delegate = MOAILuaDelegate( self )
		self.delegate.load( _getModulePath( 'LayerManager.lua' ) )

		self.addTool( 'layer_manager/add',    label = 'add',  icon = 'add' )
		self.addTool( 'layer_manager/remove', label = 'remove', icon = 'remove' )
		self.addTool( 'layer_manager/up',     label = 'up', icon = 'arrow-up' )
		self.addTool( 'layer_manager/down',   label = 'down', icon = 'arrow-down' )

		
		#SIGNALS
		signals.connect( 'moai.clean', self.onMoaiClean )

	def onAppReady( self ):
		self.tree.rebuild()

	def onMoaiClean( self ):
		self.tree.clear()

	def onTool( self, tool ):
		name = tool.name
		if name == 'add':
			layer = self.delegate.safeCall( 'addLayer' )
			self.tree.addNode( layer )
			self.tree.editNode( layer )
			self.tree.selectNode( layer )
			
		elif name == 'remove':
			for l in self.tree.getSelection():
				self.delegate.safeCall( 'removeLayer', l )
				self.tree.removeNode( l )
		elif name == 'up':
			for l in self.tree.getSelection():
				self.delegate.safeCall( 'moveLayerUp', l )
				self.tree.rebuild()
				self.tree.selectNode( l )
				break
		elif name == 'down':
			for l in self.tree.getSelection():
				self.delegate.safeCall( 'moveLayerDown', l )		
				self.tree.rebuild()
				self.tree.selectNode( l )
				break

	def changeLayerName( self, layer, name ):
		layer.setName( layer, name )

	def toggleHidden( self, layer ):
		layer.setVisible( layer, not layer.isVisible( layer ) )
		self.tree.refreshNodeContent( layer )
		signals.emit( 'scene.update' )

	def toggleEditVisible( self, layer ):
		layer.setEditorVisible( layer, not layer.isEditorVisible( layer ) )
		self.tree.refreshNodeContent( layer )
		signals.emit( 'scene.update' )

	def toggleLock( self, layer ):
		layer.locked = not layer.locked
		self.tree.refreshNodeContent( layer )
		signals.emit( 'scene.update' )

	def toggleSolo( self, layer ):
		solo = layer.isEditorSolo( layer )
		if solo:
			self.tree.hasSoloLayer = False
			for l in _MOCK.game.layers.values():
				if l.name == '_GII_EDITOR_LAYER' : continue
				if l != layer:
					l.editorSolo = False
			layer.setEditorSolo( layer, False )
		else:
			self.tree.hasSoloLayer = True
			for l in _MOCK.game.layers.values():
				if l.name == '_GII_EDITOR_LAYER' : continue
				if l != layer:
					l.editorSolo = 'hidden'
			layer.setEditorSolo( layer, True )
		self.tree.refreshAllContent()
		signals.emit( 'scene.update' )
Esempio n. 21
0
class AnimatorView(SceneEditorModule):
    name = 'animator'
    dependency = ['scene_editor', 'mock']

    def onLoad(self):
        #UI
        self.windowTitle = 'Animator'
        self.window = self.requestDockWindow('AnimatorView',
                                             title='Animator',
                                             size=(120, 120),
                                             minSize=(120, 120),
                                             dock='bottom',
                                             icon='play')

        self.widget = AnimatorWidget()
        self.window.addWidget(self.widget)
        self.toolbarTarget = self.addToolBar('animator_target',
                                             self.widget.toolbarTarget)
        self.toolbarClips = self.addToolBar('animator_clips',
                                            self.widget.toolbarClips)
        self.toolbarPlay = self.addToolBar('animator_play',
                                           self.widget.toolbarPlay)
        self.toolbarTrack = self.addToolBar('animator_track',
                                            self.widget.toolbarTrack)
        self.toolbarClipTree = self.addToolBar('animator_clip_tree',
                                               self.widget.toolbarClipTree)
        # self.toolbarEdit  = self.addToolBar( 'animator_play',  self.widget.toolbarEdit )

        signals.connect('scene.close', self.onSceneClose)
        signals.connect('scene.save', self.preSceneSave)
        signals.connect('scene.saved', self.postSceneSave)

        # addWidgetWithLaytut( toolbar,
        # 	self.widget.containerEditTool )
        self.addTool('animator_target/find_animator',
                     label='Find Animator',
                     icon='find')
        self.addTool('animator_target/change_animator',
                     label='Edit Selected Animator',
                     icon='in')
        # self.addTool( 'animator_target/close_animator', label = 'Close Animator', icon = 'in' )
        self.addTool('animator_target/----')
        self.addTool('animator_target/save_data',
                     label='Save Data',
                     icon='save')
        self.addTool('animator_target/----')
        self.addTool('animator_target/extra', label='Extra', icon='dots')

        #
        self.addTool('animator_clips/add_clip_group',
                     label='add group',
                     icon='add_folder')
        self.addTool('animator_clips/add_clip', label='add', icon='add')
        self.addTool('animator_clips/add_clip_tree',
                     label='add tree',
                     icon='add_clip_tree')
        # self.addTool( 'animator_clips/add_clip_list',   label = 'add list',    icon = 'add_clip_list' )
        self.addTool('animator_clips/----')
        self.addTool('animator_clips/remove_clip',
                     label='remove',
                     icon='remove')
        self.addTool('animator_clips/clone_clip', label='clone', icon='clone')

        #
        self.addTool('animator_play/goto_start',
                     label='to start',
                     icon='rewind')
        self.addTool('animator_play/stop', label='stop', icon='stop')
        self.addTool('animator_play/play',
                     label='play',
                     icon='play',
                     type='check')
        self.addTool('animator_play/goto_end',
                     label='to end',
                     icon='fast_forward')
        self.addTool('animator_play/toggle_repeat',
                     label='toggle repeat',
                     icon='repeat',
                     type='check')

        self.comboPreviewSpeed = QtWidgets.QComboBox()
        self.comboPreviewSpeed.addItems([e[0] for e in PREVIEW_SPEED_OPTIONS])
        self.comboPreviewSpeed.setCurrentIndex(4)  #1x
        self.comboPreviewSpeed.currentIndexChanged.connect(
            self.onPreviewSpeedChange)
        self.addTool('animator_play/preview_speed',
                     widget=self.comboPreviewSpeed)

        #
        self.addTool('animator_clip_tree/stop_tree', label='stop', icon='stop')
        self.addTool('animator_clip_tree/play_tree',
                     label='play',
                     icon='play',
                     type='check')
        self.addTool('animator_clip_tree/----', )
        self.addTool('animator_clip_tree/add_clip_tree_node',
                     label='Add node',
                     icon='add')
        self.addTool('animator_clip_tree/remove_clip_tree_node',
                     label='Remove node',
                     icon='remove')

        #SIGNALS
        self.addTool('animator_track/fold_tracks',
                     label='fold all',
                     icon='collapse')
        self.addTool('animator_track/unfold_tracks',
                     label='unfold all',
                     icon='expand')
        self.addTool('animator_track/----')
        self.addTool('animator_track/locate_target',
                     label='locate',
                     icon='find')
        self.addTool('animator_track/retarget',
                     label='retarget',
                     icon='compose')
        self.addTool('animator_track/----')
        self.addTool('animator_track/add_track_group',
                     label='add group',
                     icon='add_folder')
        self.addTool('animator_track/add_track', label='add', icon='add')
        self.addTool('animator_track/remove_track',
                     label='remove',
                     icon='remove')

        #
        self.addShortcut(self.widget, 'Space', self.togglePreview)
        self.addShortcut(self.widget, 'shift+Space', self.restartPreview)
        self.addShortcut(self.widget, 'Escape', self.resetPreview)

        signals.connect('selection.changed', self.onSceneSelectionChanged)

        self.delegate = MOAILuaDelegate(self)
        self.delegate.load(_getModulePath('AnimatorView.lua'))

        self.widget.setOwner(self)

        #playback
        self.previewing = False
        self.setEditing(False)
        self.setReadOnly(False)

        self.targetAnimator = None
        self.targetClip = None
        self.targetAnimatorData = None
        self.currentTrack = None

        self.previewing = False
        self.previewLoop = False
        self.previewTime = 0.0
        self.previewStep = 1.0 / 60.0

        self.previewTimer = QtCore.QTimer()
        self.previewTimer.setTimerType(Qt.PreciseTimer)
        # self.previewTimer  = self.addTimer()
        self.previewTimer.setInterval(1000.0 / 65)
        self.previewTimer.stop()
        self.previewTimer.timeout.connect(self.onPreviewTimer)

        # self.previewTimer.onTick = self.onPreviewTimer

    def onStart(self):
        pass

    def setEditing(self, editing):
        self.widget.timeline.setEnabled(editing)
        self.widget.treeTracks.setEnabled(editing)
        self.findTool('animator_play').setEnabled(editing)
        self.findTool('animator_track').setEnabled(editing)
        self.findTool('animator_clips/add_clip_group').setEnabled(editing)
        self.findTool('animator_clips/add_clip_tree').setEnabled(editing)
        self.findTool('animator_clips/add_clip').setEnabled(editing)
        self.findTool('animator_clips/remove_clip').setEnabled(editing)
        self.findTool('animator_clips/clone_clip').setEnabled(editing)

    def setReadOnly(self, readonly):
        self.readOnly = readonly
        #TODO

    def setTargetAnimator(self, target):
        self.saveAnimatorData()
        if target == self.targetAnimator: return
        if self.previewing:
            self.stopPreview()
        self.targetAnimator = target
        self.targetClip = None
        self.delegate.callMethod('view', 'setTargetAnimator', target)
        self.targetAnimatorData = self.delegate.callMethod(
            'view', 'getTargetAnimatorData')
        self.widget.rebuild()
        if self.targetAnimator:
            self.setEditing(True)
            signals.emit('animator.start')
        else:
            self.setEditing(False)
            signals.emit('animator.stop')

        path = self.delegate.callMethod('view', 'getTargetAnimatorDataPath')
        if path:
            self.window.setWindowTitle('Animator - %s' % path)
        else:
            self.window.setWindowTitle('Animator')
        clip = self.delegate.callMethod('view', 'getPreviousTargeClip', target)
        self.enableTool('animator_play', False)
        self.enableTool('animator_track', False)
        if clip:
            self.widget.treeClips.selectNode(clip)
        else:
            self.widget.treeClips.selectFirstItem()
        self.applyTime(0, True)

    def setTargetClip(self, clip):
        wasPreviewing = self.previewing
        if self.previewing:
            self.stopPreview()

        self.targetClip = clip
        self.delegate.callMethod('view', 'setTargetClip', clip)
        self.widget.rebuildTimeline()
        self.widget.rebuildClipTree()
        self.enableTool('animator_play', bool(clip))
        self.enableTool('animator_track', bool(clip))
        self.applyTime(0, True)
        if wasPreviewing:
            self.startPreview()

    def setCurrentTrack(self, track):
        self.currentTrack = track
        self.delegate.callMethod('view', 'setCurrentTrack', track)

    def getTargetClipLength(self):
        return self.delegate.callMethod('view', 'getTargetClipLength')

    def getClipList(self):
        if self.targetAnimatorData:
            clipList = self.targetAnimatorData.clips
            return [clip for clip in list(clipList.values())]
        else:
            return []

    def getRootClipGroup(self):
        if self.targetAnimatorData:
            return self.targetAnimatorData.getRootGroup(
                self.targetAnimatorData)

    def getTrackList(self):
        if self.targetClip:
            trackList = self.targetClip.getTrackList(self.targetClip)
            return [track for track in list(trackList.values())]
        else:
            return []

    def getMarkerList(self):
        if self.targetClip:
            markerList = self.targetClip.getMarkerList(self.targetClip)
            return [track for track in list(markerList.values())]
        else:
            return []

    def getClipRoot(self):
        if self.targetClip:
            return self.targetClip.getRoot(self.targetClip)
        else:
            return None

    def getClipTreeRoot(self):
        if self.targetClip and isMockInstance(self.targetClip,
                                              'AnimatorClipTree'):
            return self.targetClip.getTreeRoot(self.targetClip)

    def addClip(self):
        if not self.targetAnimatorData: return
        targetGroup = self.widget.getCurrentClipGroup()
        cmd = self.doCommand('scene_editor/animator_add_clip',
                             animator_data=self.targetAnimatorData,
                             parent_group=targetGroup)
        clip = cmd.getResult()
        clip.setFixedLength(clip, 3)
        if clip:
            self.widget.addClip(clip, True)
        return clip

    def addClipTree(self):
        if not self.targetAnimatorData: return
        targetGroup = self.widget.getCurrentClipGroup()
        cmd = self.doCommand('scene_editor/animator_add_clip_tree',
                             animator_data=self.targetAnimatorData,
                             parent_group=targetGroup)
        clip = cmd.getResult()
        if clip:
            self.widget.addClip(clip, True)
        return clip

    def addClipGroup(self):
        if not self.targetAnimatorData: return
        targetGroup = self.widget.getCurrentClipGroup()
        cmd = self.doCommand('scene_editor/animator_add_clip_group',
                             animator_data=self.targetAnimatorData,
                             parent_group=targetGroup)
        group = cmd.getResult()
        if group:
            self.widget.addClip(group, True)
        return group

    def removeClipNode(self):
        for clip in self.widget.treeClips.getSelection():
            if self.doCommand('scene_editor/animator_remove_clip_node',
                              animator_data=self.targetAnimatorData,
                              target_node=clip):
                self.widget.removeClip(clip)

    def cloneClipNode(self):
        if not self.targetClip: return
        result = []
        for clip in self.widget.treeClips.getSelection():
            cmd = self.doCommand('scene_editor/animator_clone_clip_node',
                                 animator_data=self.targetAnimatorData,
                                 target_node=clip)
            if cmd:
                cloned = cmd.getResult()
                self.widget.addClip(cloned, True)
                result.append(cloned)
        return result

    def onObjectEdited(self, obj):
        if self.targetClip:
            self.delegate.callMethod('view', 'clearPreviewState')
            self.delegate.callMethod('view', 'markClipDirty')
            self.refreshPreview()

    def onSceneSelectionChanged(self, selection, key):
        if key != 'scene': return
        #find animator component
        # self.findTargetAnimator()

    def findTargetAnimator(self):
        target = self.delegate.callMethod('view', 'findTargetAnimator')
        self.setTargetAnimator(target)
        return target

    def checkTargetAnimator(self):
        if not self.targetAnimator:
            alertMessage('No Animator', 'No Animator Selected', 'question')
            return False
        return True

    def addMarker(self):
        if not self.targetClip: return
        cmd = self.doCommand('scene_editor/animator_add_marker',
                             target_clip=self.targetClip,
                             target_pos=self.widget.getCursorPos())
        if cmd:
            marker = cmd.getResult()
            self.widget.addMarker(marker)

    def addKeyForField(self, target, fieldId):
        if not self.checkTargetAnimator(): return

        if not self.targetClip:
            self.addClip()
            # alertMessage( 'No Clip', 'You need to select a Clip first', 'question' )
            # return False
        keys = self.delegate.callMethod('view', 'addKeyForField', target,
                                        fieldId)
        if keys:
            for key in list(keys.values()):
                self.widget.addKey(key, True)

    def addKeyForEvent(self, target, eventId):
        pass

    def addCustomAnimatorTrack(self, target, trackClasId):
        if not self.checkTargetAnimator(): return

        track = self.delegate.callMethod('view', 'addCustomAnimatorTrack',
                                         target, trackClasId)
        if track:
            self.widget.addTrack(track)

    def addKeyForSelectedTracks(self):
        #TODO: command
        selectedTracks = self.widget.getTrackSelection()
        for track in selectedTracks:
            keys = self.delegate.callMethod('view', 'addKeyForSelectedTrack',
                                            track)
            if keys:
                for key in list(keys.values()):
                    self.widget.addKey(key, True)

    def removeSelectedKeys(self):
        #TODO: command
        selectedKeys = self.widget.getKeySelection()
        for key in selectedKeys:
            self.widget.removeKey(key)

    def cloneSelectedKeys(self):
        #TODO: command
        selectedKeys = self.widget.getKeySelection()
        cloned = []
        for key in selectedKeys:
            clonedKey = self.delegate.callMethod('view', 'cloneKey', key)
            if clonedKey:
                cloned.append(clonedKey)

        for clonedKey in cloned:
            keyItem = self.widget.addKey(clonedKey, False)
            keyItem.fitIntoNextEmptySpan()

    def onKeyRemoving(self, key):
        if self.delegate.callMethod('view', 'removeKey', key) != False:
            self.refreshPreview()
            return True

    def onMarkerRemoving(self, marker):
        if self.delegate.callMethod('view', 'removeMarker', marker) != False:
            return True

    def onClipLengthChanging(self, t1):
        if self.delegate.callMethod('view', 'setTargetClipLength',
                                    t1) != False:
            self.refreshPreview()
            return True

    def onTimelineKeyChanged(self, key, pos, length):
        self.delegate.callMethod('view', 'updateTimelineKey', key, pos, length)
        self.refreshPreview()

    def onTimelineKeyCurveValueChanged(self, key, value):
        self.delegate.callMethod('view', 'updateTimelineKeyCurveValue', key,
                                 value)
        self.refreshPreview()

    def onTimelineKeyTweenModeChanged(self, key, mode):
        self.delegate.callMethod('view', 'updateTimelineKeyTweenMode', key,
                                 mode)
        self.refreshPreview()

    def onTimelineKeyBezierPointChanged(self, key, bpx0, bpy0, bpx1, bpy1):
        self.delegate.callMethod('view', 'updateTimelineKeyBezierPoint', key,
                                 bpx0, bpy0, bpx1, bpy1)
        self.refreshPreview()

    def onTimelineMarkerChanged(self, marker, pos):
        self.delegate.callMethod('view', 'updateTimelineMarker', marker, pos)

    def toggleTrackActive(self, track):
        #TODO: command
        # self.module.doCommand( 'scene_editor/toggle_entity_visibility', target = node )
        self.delegate.callMethod('view', 'toggleTrackActive', track)
        self.refreshPreview()
        self.refreshPreview()

    def renameTrack(self, track, name):
        self.delegate.callMethod('view', 'renameTrack', track, name)

    def renameClip(self, clip, name):
        self.delegate.callMethod('view', 'renameClip', clip, name)

    def createClipTreeNode(self, nodeTypeName):
        if not self.targetClip: return None
        contextNode = self.widget.treeClipTree.getFirstSelection()
        cmd = self.doCommand('scene_editor/animator_add_clip_tree_node',
                             context_node=contextNode,
                             parent_tree=self.targetClip,
                             node_type=nodeTypeName)
        node = cmd.getResult()
        if node:
            self.widget.addClipTreeNode(node, True)
        return node

    def listClipTreeNodeTypes(self, typeId, context, option):
        contextNode = self.widget.treeClipTree.getFirstSelection()
        parentTree = self.targetClip
        if not contextNode:
            contextNode = parentTree.getTreeRoot(parentTree)
        res = _MOCK_EDIT.requestAvailAnimatorClipTreeNodeTypes(contextNode)
        entries = []
        for n in list(res.values()):
            entry = (n, n, 'AnimatorClipTreeNode', 'animator_clip_tree/' + n)
            entries.append(entry)
        return entries

    def retargetTrack(self, entity):
        for track in self.widget.treeTracks.getSelection():
            cmd = self.doCommand('scene_editor/animator_retarget_track',
                                 target_track=track,
                                 target_entity=entity,
                                 animator=self.targetAnimator)
            self.widget.treeTracks.refreshNodeContent(track)
        self.delegate.callMethod('view', 'clearPreviewState')
        self.delegate.callMethod('view', 'markClipDirty')
        self.refreshPreview()

    def selectAnimatorEntity(self, com):
        entity = com._entity
        if not entity: return
        self.changeSelection(entity)
        self.findTargetAnimator()

    def openTargetContextMenu(self):
        menu = QtWidgets.QMenu("Animator Target Context")
        itemAddKey = menu.addAction('Relocate Targets')
        itemAddKey.triggered.connect(self.onRelocateTargets)
        menu.exec_(QtGui.QCursor.pos())

    def onRelocateTargets(self):
        if not self.targetAnimator: return
        self.delegate.callMethod('view', 'relocateTargets')

    def onTool(self, tool):
        name = tool.name
        if name == 'find_animator':
            requestSearchView(context='scene',
                              type=_MOCK.Animator,
                              on_selection=self.selectAnimatorEntity)

        elif name == 'change_animator':
            target0 = self.targetAnimator
            target1 = self.findTargetAnimator()
            if (not target0) and (not target1):
                alertMessage('No Animator',
                             'No Animator found in selected entity scope',
                             'question')

        elif name == 'save_data':
            self.saveAnimatorData()

        elif name == 'extra':
            self.openTargetContextMenu()

        elif name == 'add_clip':
            if self.checkTargetAnimator():
                self.addClip()

        elif name == 'add_clip_tree':
            if self.checkTargetAnimator():
                self.addClipTree()

        elif name == 'add_clip_group':
            if self.checkTargetAnimator():
                self.addClipGroup()

        elif name == 'remove_clip':
            if self.checkTargetAnimator():
                self.removeClipNode()

        elif name == 'clone_clip':
            if self.checkTargetAnimator():
                self.cloneClipNode()

        elif name == 'add_track_group':
            group = self.delegate.callMethod('view', 'addTrackGroup')
            if group:
                self.widget.addTrack(group, True)

        elif name == 'remove_track':
            for track in self.widget.treeTracks.getSelection():
                self.delegate.callMethod('view', 'removeTrack', track)
                self.widget.removeTrack(track)
            self.refreshPreview()

        elif name == 'fold_tracks':
            self.widget.treeTracks.foldAllItems()

        elif name == 'unfold_tracks':
            self.widget.treeTracks.expandAllItems()

        elif name == 'locate_target':
            for track in self.widget.treeTracks.getSelection():
                sceneGraphEditor = self.getModule('scenegraph_editor')
                if sceneGraphEditor:
                    targetEntity = self.delegate.callMethod(
                        'view', 'findTrackEntity', track)
                    if targetEntity:
                        sceneGraphEditor.selectEntity(targetEntity,
                                                      focus_tree=True)
                #pass
                return

        elif name == 'retarget':
            requestSearchView(
                info='retarget animator track to...',
                context='scene',
                type='entity',
                multiple_selection=False,
                on_selection=self.retargetTrack,
            )

        elif name == 'add_clip_tree_node':
            requestSearchView(info='adding AnimatorClipTree node...',
                              context='clip_tree_node',
                              type=None,
                              multiple_selection=False,
                              on_selection=self.createClipTreeNode,
                              on_search=self.listClipTreeNodeTypes)

        elif name == 'remove_clip_tree_node':
            for node in self.widget.treeClipTree.getSelection():
                if node.isVirtual(node): continue
                self.doCommand('scene_editor/animator_remove_clip_tree_node',
                               target_node=node)
            self.widget.rebuildClipTree()
            self.refreshPreview()

        #preview
        elif name == 'goto_start':
            self.gotoStart()
        elif name == 'goto_end':
            self.gotoEnd()
        elif name == 'play':
            if tool.getValue():
                self.startPreview()
            else:
                self.stopPreview(False)
        elif name == 'stop':
            self.stopPreview(True)
        elif name == 'play_tree':
            if tool.getValue():
                self.startPreview()
            else:
                self.stopPreview(False)
        elif name == 'stop_tree':
            self.stopPreview(True)
        elif name == 'toggle_repeat':
            self.delegate.callMethod('view', 'togglePreviewRepeat',
                                     tool.getValue())

    def getActiveSceneView(self):
        return self.getModule('scene_view')

    #preview
    def togglePreview(self):
        if self.previewing:
            self.stopPreview()
        else:
            self.startPreview()

    def restartPreview(self):
        self.stopPreview(True)
        self.startPreview()

    def resetPreview(self):
        self.stopPreview(True)

    def startPreview(self):
        # self.saveAnimatorData() #WHY??????
        if self.delegate.callMethod('view', 'startPreview', self.previewTime):
            self.widget.setCursorMovable(False)
            self.previewing = True
            self.findTool('animator_play/play').setValue(True)
            self.findTool('animator_clip_tree/play_tree').setValue(True)
            self.previewTimer.start()
            self.getApp().setMinimalMainLoopBudget()

    def stopPreview(self, rewind=False):
        if self.previewing:
            self.delegate.callMethod('view', 'stopPreview')
            self.getApp().resetMainLoopBudget()
            self.widget.setCursorMovable(True)
            self.previewing = False
            self.findTool('animator_play/play').setValue(False)
            self.findTool('animator_clip_tree/play_tree').setValue(False)
            self.previewTimer.stop()
            signals.emit('entity.modified', None, '')
        if rewind:
            self.gotoStart()

    def onPreviewTimer(self):
        playing, currentTime = self.delegate.callMethod(
            'view', 'doPreviewStep')
        self.previewTime = currentTime
        self.getActiveSceneView().forceUpdate()
        self.widget.setCursorPos(self.previewTime)
        if not playing:
            self.stopPreview()

    def gotoStart(self):
        if self.previewing:
            self.delegate.callMethod('view', 'applyTime', 0)
        else:
            self.widget.setCursorPos(0, True)

    def gotoEnd(self):
        if self.previewing:
            self.delegate.callMethod('view', 'applyTime', 10)
        else:
            self.widget.setCursorPos(10, True)

    def applyTime(self, t, syncCursor=False):
        self.previewTime = self.delegate.callMethod('view', 'applyTime', t)
        # self.getActiveSceneView().scheduleUpdate()
        self.getActiveSceneView().forceUpdate()
        # signals.emit( 'entity.modified',  None , '' )
        if syncCursor:
            self.widget.setCursorPos(t)

    def saveAnimatorData(self):
        if not self.targetAnimator:
            return
        self.delegate.callMethod('view', 'saveData')

    def preSceneSave(self):
        if self.targetAnimator:
            self.delegate.callMethod('view', 'restoreEntityState')

    def postSceneSave(self):
        if self.targetAnimator:
            self.refreshPreview()

    def onSceneClose(self, scene):
        self.setTargetAnimator(None)

    def onPreviewSpeedChange(self, index):
        label, throttle = PREVIEW_SPEED_OPTIONS[index]
        self.delegate.callMethod('view', 'setPreviewThrottle', throttle)

    def refreshPreview(self):
        self.applyTime(self.previewTime)
        self.applyTime(self.previewTime)

    def refreshTimeline(self):
        self.widget.rebuildTimeline()

    def refreshClipList(self):
        self.widget.rebuildClipList()

    def refreshAll(self):
        self.widget.rebuild()
Esempio n. 22
0
class DeckCanvasEditor(SceneEditorModule):
    name = 'deckcanvas_editor'
    dependency = ['mock']

    def onLoad(self):
        self.mainToolBar = self.addToolBar(
            'deckcanvas_tools',
            self.getMainWindow().requestToolBar('deckcanvas_tools'))

        self.addTool('deckcanvas_tools/tool_pen',
                     widget=SceneToolButton('deckcanvas_pen',
                                            label='Deck Canvas Editor',
                                            icon='deckcanvas/pen'))

        self.addTool('deckcanvas_tools/toggle_item_bounds_visible',
                     label='Toggle Item Bounds',
                     icon='deckcanvas/layer',
                     type='check')

        self.findTool('deckcanvas_tools/toggle_item_bounds_visible').setValue(
            True)

        self.delegate = MOAILuaDelegate(self)
        self.delegate.load(_getModulePath('DeckCanvasEditor.lua'))
        signals.connect('selection.changed', self.onSelectionChanged)

        self.toolWindow = self.requestToolWindow('DeckCanvasEditor',
                                                 title='DeckCanvas',
                                                 size=(120, 40),
                                                 minSize=(120, 40))

        self.targetCanvas = None

    def onStart(self):
        self.setEditActive(True)

    def onSetFocus(self):
        self.getModule('scene_editor').setFocus()
        self.container.show()
        self.container.setFocus()

    def updateSelection(self):
        target = self.delegate.callMethod('editor', 'findTargetDeckCanvas')
        self.setTargetCanvas(target)

    def onSelectionChanged(self, selection, key):
        if key == 'scene':
            self.updateSelection()

        elif key == 'asset':
            decks = []
            for node in selection:
                if node.getType().startswith('deck2d.'):
                    decks.append(node.getNodePath())

            self.delegate.callMethod('editor', 'changeDeckSelection', decks)

    def setEditActive(self, active):
        self.enableTool('deckcanvas_tools/tool_pen', active)

    def setTargetCanvas(self, canvas):
        self.delegate.callMethod('editor', 'setTargetCanvas', canvas)
        self.targetCanvas = canvas
        # if not self.targetCanvas:
        # 	self.setEditActive( False )
        # 	return
        # self.setEditActive( True )

    def showToolWindow(self):
        # self.toolWindow.show()
        self.getModule('scene_view').setFocus()

    def hideToolWindow(self):
        # self.toolWindow.hide()
        pass

    def startPenTool(self):
        self.delegate.callMethod('editor', 'startPenTool')

    def onTool(self, tool):
        name = tool.name
        if name == 'toggle_item_bounds_visible':
            _MOCK.setDeckCanvasItemBoundsVisible(bool(tool.getValue()))
            signals.emit('scene.update')
Esempio n. 23
0
    def onLoad(self):
        #UI
        self.windowTitle = 'Animator'
        self.window = self.requestDockWindow('AnimatorView',
                                             title='Animator',
                                             size=(120, 120),
                                             minSize=(120, 120),
                                             dock='bottom',
                                             icon='play')

        self.widget = AnimatorWidget()
        self.window.addWidget(self.widget)
        self.toolbarTarget = self.addToolBar('animator_target',
                                             self.widget.toolbarTarget)
        self.toolbarClips = self.addToolBar('animator_clips',
                                            self.widget.toolbarClips)
        self.toolbarPlay = self.addToolBar('animator_play',
                                           self.widget.toolbarPlay)
        self.toolbarTrack = self.addToolBar('animator_track',
                                            self.widget.toolbarTrack)
        self.toolbarClipTree = self.addToolBar('animator_clip_tree',
                                               self.widget.toolbarClipTree)
        # self.toolbarEdit  = self.addToolBar( 'animator_play',  self.widget.toolbarEdit )

        signals.connect('scene.close', self.onSceneClose)
        signals.connect('scene.save', self.preSceneSave)
        signals.connect('scene.saved', self.postSceneSave)

        # addWidgetWithLaytut( toolbar,
        # 	self.widget.containerEditTool )
        self.addTool('animator_target/find_animator',
                     label='Find Animator',
                     icon='find')
        self.addTool('animator_target/change_animator',
                     label='Edit Selected Animator',
                     icon='in')
        # self.addTool( 'animator_target/close_animator', label = 'Close Animator', icon = 'in' )
        self.addTool('animator_target/----')
        self.addTool('animator_target/save_data',
                     label='Save Data',
                     icon='save')
        self.addTool('animator_target/----')
        self.addTool('animator_target/extra', label='Extra', icon='dots')

        #
        self.addTool('animator_clips/add_clip_group',
                     label='add group',
                     icon='add_folder')
        self.addTool('animator_clips/add_clip', label='add', icon='add')
        self.addTool('animator_clips/add_clip_tree',
                     label='add tree',
                     icon='add_clip_tree')
        # self.addTool( 'animator_clips/add_clip_list',   label = 'add list',    icon = 'add_clip_list' )
        self.addTool('animator_clips/----')
        self.addTool('animator_clips/remove_clip',
                     label='remove',
                     icon='remove')
        self.addTool('animator_clips/clone_clip', label='clone', icon='clone')

        #
        self.addTool('animator_play/goto_start',
                     label='to start',
                     icon='rewind')
        self.addTool('animator_play/stop', label='stop', icon='stop')
        self.addTool('animator_play/play',
                     label='play',
                     icon='play',
                     type='check')
        self.addTool('animator_play/goto_end',
                     label='to end',
                     icon='fast_forward')
        self.addTool('animator_play/toggle_repeat',
                     label='toggle repeat',
                     icon='repeat',
                     type='check')

        self.comboPreviewSpeed = QtWidgets.QComboBox()
        self.comboPreviewSpeed.addItems([e[0] for e in PREVIEW_SPEED_OPTIONS])
        self.comboPreviewSpeed.setCurrentIndex(4)  #1x
        self.comboPreviewSpeed.currentIndexChanged.connect(
            self.onPreviewSpeedChange)
        self.addTool('animator_play/preview_speed',
                     widget=self.comboPreviewSpeed)

        #
        self.addTool('animator_clip_tree/stop_tree', label='stop', icon='stop')
        self.addTool('animator_clip_tree/play_tree',
                     label='play',
                     icon='play',
                     type='check')
        self.addTool('animator_clip_tree/----', )
        self.addTool('animator_clip_tree/add_clip_tree_node',
                     label='Add node',
                     icon='add')
        self.addTool('animator_clip_tree/remove_clip_tree_node',
                     label='Remove node',
                     icon='remove')

        #SIGNALS
        self.addTool('animator_track/fold_tracks',
                     label='fold all',
                     icon='collapse')
        self.addTool('animator_track/unfold_tracks',
                     label='unfold all',
                     icon='expand')
        self.addTool('animator_track/----')
        self.addTool('animator_track/locate_target',
                     label='locate',
                     icon='find')
        self.addTool('animator_track/retarget',
                     label='retarget',
                     icon='compose')
        self.addTool('animator_track/----')
        self.addTool('animator_track/add_track_group',
                     label='add group',
                     icon='add_folder')
        self.addTool('animator_track/add_track', label='add', icon='add')
        self.addTool('animator_track/remove_track',
                     label='remove',
                     icon='remove')

        #
        self.addShortcut(self.widget, 'Space', self.togglePreview)
        self.addShortcut(self.widget, 'shift+Space', self.restartPreview)
        self.addShortcut(self.widget, 'Escape', self.resetPreview)

        signals.connect('selection.changed', self.onSceneSelectionChanged)

        self.delegate = MOAILuaDelegate(self)
        self.delegate.load(_getModulePath('AnimatorView.lua'))

        self.widget.setOwner(self)

        #playback
        self.previewing = False
        self.setEditing(False)
        self.setReadOnly(False)

        self.targetAnimator = None
        self.targetClip = None
        self.targetAnimatorData = None
        self.currentTrack = None

        self.previewing = False
        self.previewLoop = False
        self.previewTime = 0.0
        self.previewStep = 1.0 / 60.0

        self.previewTimer = QtCore.QTimer()
        self.previewTimer.setTimerType(Qt.PreciseTimer)
        # self.previewTimer  = self.addTimer()
        self.previewTimer.setInterval(1000.0 / 65)
        self.previewTimer.stop()
        self.previewTimer.timeout.connect(self.onPreviewTimer)
Esempio n. 24
0
class GlobalObjectManager(SceneEditorModule):
    def __init__(self):
        super(GlobalObjectManager, self).__init__()
        self.refreshScheduled = False

    def getName(self):
        return 'global_object_manager'

    def getDependency(self):
        return ['scene_editor', 'mock']

    def onLoad(self):
        #UI
        self.window = self.requestDockWindow('GlobalObjectManager',
                                             title='Global Objects',
                                             size=(120, 120),
                                             minSize=(120, 120),
                                             dock='left')

        #Components
        self.treeFilter = self.window.addWidget(GenericTreeFilter(),
                                                expanding=False)
        self.tree = self.window.addWidget(
            GlobalObjectTreeWidget(multiple_selection=True,
                                   editable=True,
                                   drag_mode='internal'))
        self.treeFilter.setTargetTree(self.tree)

        self.tool = self.addToolBar('global_object_manager',
                                    self.window.addToolBar())
        self.delegate = MOAILuaDelegate(self)
        self.delegate.load(_getModulePath('GlobalObjectManager.lua'))

        # self.creatorMenu=self.addMenu(
        # 	'global_object_create',
        # 	{ 'label':'Create Global Object' }
        # 	)

        self.addTool('global_object_manager/add', label='Add', icon='add')
        self.addTool('global_object_manager/remove',
                     label='Remove',
                     icon='remove')
        self.addTool('global_object_manager/clone',
                     label='Clone',
                     icon='clone')
        self.addTool('global_object_manager/add_group',
                     label='Add Group',
                     icon='add_folder')
        self.addTool('global_object_manager/----')
        self.addTool('global_object_manager/refresh',
                     label='Refresh',
                     icon='refresh')

        # self.addMenuItem( 'main/find/find_global_object',
        # 	dict( label = 'Find In Global Objects', shortcut = 'ctrl+alt+f' ) )

        #SIGNALS
        signals.connect('moai.clean', self.onMoaiClean)

        signals.connect('global_object.added', self.onObjectAdded)
        signals.connect('global_object.removed', self.onObjectRemoved)
        signals.connect('global_object.renamed', self.onObjectRenamed)

        if self.getModule('introspector'):
            from . import GlobalObjectNodeEditor

        registerSearchEnumerator(globalObjectNameSearchEnumerator)
        registerSearchEnumerator(globalObjectSearchEnumerator)

    def onAppReady(self):
        self.tree.rebuild()
        self.tree.setColumnWidth(0, 150)

    def onMoaiClean(self):
        self.tree.clear()

    def onTool(self, tool):
        name = tool.name
        if name == 'add':
            requestSearchView(
                info='select global object class to create',
                context='global_object_class',
                on_selection=lambda objName: self.createGlobalObject(objName))

        if name == 'add_group':
            group = self.delegate.safeCall('addGroup')
            self.tree.addNode(group)
            self.tree.editNode(group)
            self.tree.selectNode(group)

        elif name == 'remove':
            for node in self.tree.getSelection():
                self.doCommand('scene_editor/remove_global_object',
                               target=node)
                self.tree.removeNode(node)

        elif name == 'refresh':
            self.scheduleRefreshObject()

    def onMenu(self, menu):
        if menu.name == 'find_global_object':
            requestSearchView(
                info='search for global object',
                context='global_object',
                on_selection=lambda node: self.selectObject(node, True))

    def needUpdate(self):
        return True

    def onUpdate(self):
        if self.refreshScheduled:
            self.refreshScheduled = False
            self.delegate.safeCall('reloadObjects')
            self.tree.rebuild()

    def scheduleRefreshObject(self):
        self.refreshScheduled = True

    def selectObject(self, target, updateTree=False):
        self.changeSelection(target)
        if updateTree:
            self.tree.blockSignals(True)
            self.tree.selectNode(None)
            if isinstance(target, list):
                for e in target:
                    self.tree.selectNode(e, add=True)
            else:
                self.tree.selectNode(target)
            self.tree.blockSignals(False)

    def renameObject(self, obj, name):
        obj.setName(obj, name)

    def createGlobalObject(self, objName):
        self.doCommand('scene_editor/create_global_object', name=objName)

    def reparentObject(self, sources, target):
        if self.doCommand('scene_editor/reparent_global_object',
                          sources=sources,
                          target=target):
            return True

    def onObjectAdded(self, node, reason='new'):
        self.tree.addNode(node)
        if reason == 'new':
            self.tree.editNode(node)
            self.tree.selectNode(node)

    def onObjectRemoved(self, node):
        self.tree.removeNode(node)

    def onObjectRenamed(self, node, name):
        self.tree.refreshNodeContent(node)
Esempio n. 25
0
	def onLoad( self ):
		self.configPath = self.getProject().getConfigPath( _DEPLOY_CONFIG_FILE )
		#UI
		self.container = self.requestDocumentWindow( 'DeployManager',
			title     = 'Deployment Manager',
			allowDock = False,
			minSize   = ( 300, 300 ),
			maxSize   = ( 300, 300 )
			)

		#Components
		self.window = self.container.addWidgetFromFile( _getModulePath('DeployManager.ui') )

		self.delegate = MOAILuaDelegate( self )
		self.delegate.load( _getModulePath( 'DeployManager.lua' ) )

		#scene tree
		layout = QtGui.QVBoxLayout()
		self.window.containerSceneTree.setLayout( layout )
		layout.setSpacing( 0 )
		layout.setMargin( 0 )

		self.treeScene = DeploySceneTree( 
			self.window.containerSceneTree,
			editable = True,
			sorting  = False,
			multiple_selection = False
			)
		self.treeScene.manager = self
		layout.addWidget( self.treeScene )

		sceneToolbar = QtGui.QToolBar( self.window.containerSceneTree )
		layout.addWidget( sceneToolbar )
		self.sceneTool = self.addToolBar( 'deploy_scene', sceneToolbar )
		self.addTool( 'deploy_scene/add_scene',         label = 'add'     ,icon = 'add'    )
		self.addTool( 'deploy_scene/remove_scene',      label = 'remove'  ,icon = 'remove' )
		self.addTool( 'deploy_scene/move_up_scene',     label = 'up'      ,icon = 'arrow-up'     )
		self.addTool( 'deploy_scene/move_down_scene',   label = 'down'    ,icon = 'arrow-down'   )
		self.addTool( 'deploy_scene/----' )
		self.addTool( 'deploy_scene/edit_scene',   label = 'change target scene' ,icon = 'pencil'   )
		self.addTool( 'deploy_scene/----' )
		self.addTool( 'deploy_scene/set_entry_scene',   label = 'set as entry' ,icon = 'flag'   )

		#deploy target tree
		layout = QtGui.QVBoxLayout()
		self.window.containerTargetTree.setLayout( layout )
		layout.setSpacing( 0 )		
		layout.setMargin( 0 )

		self.treeTarget = DeployTargetTree( 
			self.window.containerTargetTree,
			editable = True,
			multiple_selection = False
			)
		self.treeTarget.manager = self
		layout.addWidget( self.treeTarget )

		targetToolbar = QtGui.QToolBar( self.window.containerTargetTree )
		layout.addWidget( targetToolbar )
		self.targetTool = self.addToolBar( 'deploy_target', targetToolbar )
		self.addTool( 'deploy_target/add_target',    label = '+' )
		self.addTool( 'deploy_target/remove_target', label = '-' )

		#target property
		self.propertyTarget = addWidgetWithLayout(
			PropertyEditor( self.window.containerTargetProp )
		)

		#menu
		self.addMenuItem( 'main/file/----' )
		self.addMenuItem( 'main/file/deploy_manager', 
			dict( label = 'Deploy Manager', shortcut = 'F11' )
			)

		self.addMenuItem( 'main/file/deploy_build', 
			dict( label = 'Deploy Build', shortcut = 'Ctrl+F11' )
			)
		
		
		# self.container.show()
		self.window.buttonOK.clicked.connect( self.onButtonOK )

		#other
		registerSearchEnumerator( deployTargetSearchEnumerator )

		signals.connect( 'project.pre_deploy', self.preDeploy )
		signals.connect( 'project.deploy', self.onDeploy )
		signals.connect( 'project.post_deploy', self.postDeploy )
Esempio n. 26
0
	def onLoad( self ):
		#UI
		self.windowTitle = 'Scenegraph'
		self.container = self.requestDockWindow( 'SceneGraphEditor',
			title     = 'Scenegraph',
			size      = (200,200),
			minSize   = (200,200),
			dock      = 'left'
			)

		#Components
		self.treeFilter = self.container.addWidget(
			GenericTreeFilter(
				self.container
			),
			expanding = False
		)
		self.tree = self.container.addWidget( 
				SceneGraphTreeWidget( 
					self.container,
					sorting  = True,
					editable = True,
					multiple_selection = True,
					drag_mode = 'internal'
				)
			)
		self.treeFilter.setTargetTree( self.tree )
		self.tree.module = self
		self.tool = self.addToolBar( 'scene_graph', self.container.addToolBar() )
		self.delegate = MOAILuaDelegate( self )
		self.delegate.load( getModulePath( 'SceneGraphEditor.lua' ) )

		self.entityCreatorMenu=self.addMenu(
			'main/scene/entity_create',
			{ 'label':'Create Entity' }
			)

		self.componentCreatorMenu=self.addMenu(
			'main/scene/component_create',
			{ 'label':'Create Component' }
			)


		#menu
		self.addMenuItem(
			'main/file/open_scene', 
			dict( label = 'Open Scene', shortcut = 'ctrl+shift+o' )
			)

		self.addMenuItem( 'main/file/close_scene', 
			dict( label = 'Close Scene', shortcut = 'Ctrl+W' )
			)
		self.addMenuItem( 'main/scene/save_scene',
			dict( label = 'Save', shortcut = 'Ctrl+S' )
			)
		self.addMenuItem( 'main/scene/locate_scene_asset',
			dict( label = 'Locate Scene Asset' )
			)

		self.addMenu( 'main/scene/----' )
		
		self.addMenu( 'component_context', dict( label = 'Selected Component' ) )
		self.addMenuItem( 'component_context/remove_component', 
			dict( label = 'Remove' )
			)

		self.addMenuItem( 'component_context/----' )
		
		self.addMenuItem( 'component_context/copy_component', 
			dict( label = 'Copy' )
			)
		self.addMenuItem( 'component_context/paste_component', 
			dict( label = 'Paste Component Here' )
			)
		self.addMenuItem( 'component_context/----' )
		
		self.addMenuItem( 'component_context/move_component_up', 
			dict( label = 'Move Up' )
			)

		self.addMenuItem( 'component_context/move_component_down', 
			dict( label = 'Move Down' )
			)
		

		self.addMenu( 'main/entity', dict( label = 'Entity' ) )
		self.addMenuItem( 'main/entity/add_empty_entity',    dict( label = 'Create Empty', shortcut = 'ctrl+alt+N' ) )
		self.addMenuItem( 'main/entity/add_entity',          dict( label = 'Create',       shortcut = 'ctrl+shift+N' ) )
		self.addMenuItem( 'main/entity/----' )
		self.addMenuItem( 'main/entity/group_entity',        dict( label = 'Group Entites',    shortcut = 'ctrl+G' ) )
		self.addMenuItem( 'main/entity/create_group',        dict( label = 'Create Empty Group',    shortcut = 'ctrl+shift+G' ) )
		self.addMenuItem( 'main/entity/----' )
		self.addMenuItem( 'main/entity/load_prefab',         dict( label = 'Load Prefab', shortcut = 'ctrl+alt+shift+N' ) )
		self.addMenuItem( 'main/entity/load_prefab_in_container', dict( label = 'Load Prefab In Container', shortcut = 'ctrl+shift+=' ) )
		self.addMenuItem( 'main/entity/----' )
		self.addMenuItem( 'main/entity/remove_entity',       dict( label = 'Remove'  ) )
		self.addMenuItem( 'main/entity/clone_entity',        dict( label = 'Clone',  shortcut = 'ctrl+d' ) )
		self.addMenuItem( 'main/entity/----' )
		self.addMenuItem( 'main/entity/add_component',       dict( label = 'Add Component', shortcut = 'ctrl+alt+=' ) )
		self.addMenuItem( 'main/entity/assign_layer',        dict( label = 'Assign Layer', shortcut = 'ctrl+alt+L' ) )
		self.addMenuItem( 'main/entity/toggle_visibility',   dict( label = 'Toggle Visibility', shortcut = 'ctrl+/' ) )
		self.addMenuItem( 'main/entity/freeze_entity_pivot', dict( label = 'Freeze Pivot' ) )

		self.addMenuItem( 'main/entity/----' )
		self.addMenuItem( 'main/find/find_entity', dict( label = 'Find In Scene', shortcut = 'ctrl+f' ) )
		self.addMenuItem( 'main/find/find_entity_in_group', dict( label = 'Find In Group', shortcut = 'ctrl+shift+f' ) )
		self.addMenuItem( 'main/find/find_entity_group', dict( label = 'Find Group', shortcut = 'ctrl+alt+f' ) )

		#Toolbars
		self.addTool( 'scene_graph/select_scene',    label ='Select Scene', icon = 'settings' )
		self.addTool( 'scene_graph/----'  )
		self.addTool( 'scene_graph/create_group',    label ='+ Group', icon = 'add_folder' )
		self.addTool( 'scene_graph/----'  )
		self.addTool( 'scene_graph/make_proto',    label = 'Convert To Proto', icon = 'proto_make' )
		self.addTool( 'scene_graph/create_proto_instance',    label = 'Create Proto Instance', icon = 'proto_instantiate' )
		self.addTool( 'scene_graph/create_proto_container',    label = 'Create Proto Container', icon = 'proto_container' )
		self.addTool( 'scene_graph/----'  )
		self.addTool( 'scene_graph/fold_all',    label = 'F' )
		self.addTool( 'scene_graph/unfold_all',  label = 'U' )
		self.addTool( 'scene_graph/refresh_tree',  label = 'R' )
		# self.addTool( 'scene_graph/load_prefab', label = '+ P' )
		# self.addTool( 'scene_graph/save_prefab', label = '>>P' )

		self.addTool( 'scene/refresh', label = 'refresh', icon='refresh' )

		#SIGNALS
		signals.connect( 'moai.clean',        self.onMoaiClean        )

		signals.connect( 'scene.clear',       self.onSceneClear      )
		signals.connect( 'scene.change',      self.onSceneChange     )

		signals.connect( 'selection.changed', self.onSelectionChanged )
		signals.connect( 'selection.hint',    self.onSelectionHint    )

		signals.connect( 'preview.start',     self.onPreviewStart     )
		signals.connect( 'preview.stop' ,     self.onPreviewStop      )

		# signals.connect( 'animator.start',     self.onAnimatorStart     )
		# signals.connect( 'animator.stop' ,     self.onAnimatorStop      )

		signals.connect( 'entity.added',      self.onEntityAdded      )
		signals.connect( 'entity.removed',    self.onEntityRemoved    )
		signals.connect( 'entity.renamed',    self.onEntityRenamed    )
		signals.connect( 'entity.modified',    self.onEntityModified    )
		signals.connect( 'entity.visible_changed',    self.onEntityVisibleChanged )
		signals.connect( 'entity.pickable_changed',   self.onEntityPickableChanged )


		signals.connect( 'prefab.unlink',     self.onPrefabUnlink    )
		signals.connect( 'prefab.relink',     self.onPrefabRelink    )
		signals.connect( 'proto.unlink',      self.onPrefabUnlink    )
		signals.connect( 'proto.relink',      self.onPrefabRelink    )

		signals.connect( 'app.ready', self.postAppReady )


		signals.connect( 'component.added',   self.onComponentAdded   )
		signals.connect( 'component.removed', self.onComponentRemoved )

		signals.connect( 'project.presave',   self.preProjectSave )

		registerSearchEnumerator( sceneObjectSearchEnumerator )
		registerSearchEnumerator( entityNameSearchEnumerator )
		registerSearchEnumerator( componentNameSearchEnumerator )
		registerSearchEnumerator( layerNameSearchEnumerator )
Esempio n. 27
0
    def onLoad(self):
        #UI
        self.windowTitle = 'Scenegraph'
        self.container = self.requestDockWindow('SceneGraphEditor',
                                                title='Scenegraph',
                                                size=(200, 200),
                                                minSize=(200, 200),
                                                dock='left')

        #Components
        self.treeFilter = self.container.addWidget(GenericTreeFilter(
            self.container),
                                                   expanding=False)
        self.tree = self.container.addWidget(
            SceneGraphTreeWidget(self.container,
                                 sorting=True,
                                 editable=True,
                                 multiple_selection=True,
                                 drag_mode='internal'))
        self.treeFilter.setTargetTree(self.tree)
        self.tree.module = self
        self.tool = self.addToolBar('scene_graph', self.container.addToolBar())
        self.delegate = MOAILuaDelegate(self)
        self.delegate.load(getModulePath('SceneGraphEditor.lua'))

        self.entityCreatorMenu = self.addMenu('main/scene/entity_create',
                                              {'label': 'Create Entity'})

        self.componentCreatorMenu = self.addMenu('main/scene/component_create',
                                                 {'label': 'Create Component'})

        #menu
        self.addMenuItem('main/file/open_scene',
                         dict(label='Open Scene', shortcut='ctrl+shift+o'))

        self.addMenuItem('main/file/close_scene',
                         dict(label='Close Scene', shortcut='Ctrl+W'))
        self.addMenuItem('main/scene/save_scene',
                         dict(label='Save', shortcut='Ctrl+S'))
        self.addMenuItem('main/scene/locate_scene_asset',
                         dict(label='Locate Scene Asset'))

        self.addMenu('main/scene/----')

        self.addMenu('component_context', dict(label='Selected Component'))
        self.addMenuItem('component_context/remove_component',
                         dict(label='Remove'))

        self.addMenuItem('component_context/----')

        self.addMenuItem('component_context/copy_component',
                         dict(label='Copy'))
        self.addMenuItem('component_context/paste_component',
                         dict(label='Paste Component Here'))
        self.addMenuItem('component_context/----')

        self.addMenuItem('component_context/move_component_up',
                         dict(label='Move Up'))

        self.addMenuItem('component_context/move_component_down',
                         dict(label='Move Down'))

        self.addMenuItem('component_context/----')
        self.addMenuItem('component_context/edit_component_alias',
                         dict(label='Edit Alias'))

        self.addMenu('main/entity', dict(label='Entity'))
        self.addMenuItem('main/entity/add_empty_entity',
                         dict(label='Create Empty', shortcut='ctrl+alt+N'))
        self.addMenuItem('main/entity/add_entity',
                         dict(label='Create', shortcut='ctrl+shift+N'))
        self.addMenuItem('main/entity/----')
        self.addMenuItem('main/entity/group_entity',
                         dict(label='Group Entites', shortcut='ctrl+G'))
        self.addMenuItem(
            'main/entity/create_group',
            dict(label='Create Empty Group', shortcut='ctrl+shift+G'))
        self.addMenuItem('main/entity/----')
        self.addMenuItem(
            'main/entity/load_prefab',
            dict(label='Load Prefab', shortcut='ctrl+alt+shift+N'))
        self.addMenuItem(
            'main/entity/load_prefab_in_container',
            dict(label='Load Prefab In Container', shortcut='ctrl+shift+='))
        self.addMenuItem('main/entity/----')
        self.addMenuItem('main/entity/remove_entity', dict(label='Remove'))
        self.addMenuItem('main/entity/clone_entity',
                         dict(label='Clone', shortcut='ctrl+d'))
        self.addMenuItem('main/entity/----')
        self.addMenuItem('main/entity/add_component',
                         dict(label='Add Component', shortcut='ctrl+alt+='))
        self.addMenuItem('main/entity/assign_layer',
                         dict(label='Assign Layer', shortcut='ctrl+alt+L'))
        self.addMenuItem('main/entity/toggle_visibility',
                         dict(label='Toggle Visibility', shortcut='ctrl+/'))
        self.addMenuItem('main/entity/freeze_entity_pivot',
                         dict(label='Freeze Pivot'))

        self.addMenuItem('main/entity/----')
        self.addMenuItem('main/find/find_entity',
                         dict(label='Find In Scene', shortcut='ctrl+f'))
        self.addMenuItem('main/find/find_entity_in_group',
                         dict(label='Find In Group', shortcut='ctrl+shift+f'))
        self.addMenuItem('main/find/find_entity_group',
                         dict(label='Find Group', shortcut='ctrl+alt+f'))

        #Toolbars
        self.addTool('scene_graph/select_scene',
                     label='Select Scene',
                     icon='settings')
        self.addTool('scene_graph/----')
        self.addTool('scene_graph/create_group',
                     label='+ Group',
                     icon='add_folder')
        self.addTool('scene_graph/----')
        self.addTool('scene_graph/make_proto',
                     label='Convert To Proto',
                     icon='proto_make')
        self.addTool('scene_graph/create_proto_instance',
                     label='Create Proto Instance',
                     icon='proto_instantiate')
        self.addTool('scene_graph/create_proto_container',
                     label='Create Proto Container',
                     icon='proto_container')
        self.addTool('scene_graph/----')
        self.addTool('scene_graph/fold_all', label='F')
        self.addTool('scene_graph/unfold_all', label='U')
        self.addTool('scene_graph/refresh_tree', label='R')
        # self.addTool( 'scene_graph/load_prefab', label = '+ P' )
        # self.addTool( 'scene_graph/save_prefab', label = '>>P' )

        self.addTool('scene/refresh', label='refresh', icon='refresh')

        #SIGNALS
        signals.connect('moai.clean', self.onMoaiClean)

        signals.connect('scene.clear', self.onSceneClear)
        signals.connect('scene.change', self.onSceneChange)

        signals.connect('selection.changed', self.onSelectionChanged)
        signals.connect('selection.hint', self.onSelectionHint)

        signals.connect('preview.start', self.onPreviewStart)
        signals.connect('preview.stop', self.onPreviewStop)

        # signals.connect( 'animator.start',     self.onAnimatorStart     )
        # signals.connect( 'animator.stop' ,     self.onAnimatorStop      )

        signals.connect('entity.added', self.onEntityAdded)
        signals.connect('entity.removed', self.onEntityRemoved)
        signals.connect('entity.renamed', self.onEntityRenamed)
        signals.connect('entity.modified', self.onEntityModified)
        signals.connect('entity.visible_changed', self.onEntityVisibleChanged)
        signals.connect('entity.pickable_changed',
                        self.onEntityPickableChanged)

        signals.connect('prefab.unlink', self.onPrefabUnlink)
        signals.connect('prefab.relink', self.onPrefabRelink)
        signals.connect('proto.unlink', self.onPrefabUnlink)
        signals.connect('proto.relink', self.onPrefabRelink)

        signals.connect('app.ready', self.postAppReady)

        signals.connect('component.added', self.onComponentAdded)
        signals.connect('component.removed', self.onComponentRemoved)

        signals.connect('project.presave', self.preProjectSave)

        registerSearchEnumerator(sceneObjectSearchEnumerator)
        registerSearchEnumerator(entityNameSearchEnumerator)
        registerSearchEnumerator(componentNameSearchEnumerator)
        registerSearchEnumerator(layerNameSearchEnumerator)
Esempio n. 28
0
class GlobalObjectManager( SceneEditorModule ):
	def __init__(self):
		super( GlobalObjectManager, self ).__init__()
		self.refreshScheduled = False

	def getName( self ):
		return 'global_object_manager'

	def getDependency( self ):
		return [ 'scene_editor', 'mock' ]

	def onLoad( self ):
		#UI
		self.window = self.requestDockWindow( 'GlobalObjectManager',
			title     = 'Global Objects',
			size      = (120,120),
			minSize   = (120,120),
			dock      = 'left'
			)

		#Components
		self.tree = self.window.addWidget( 
				GlobalObjectTreeWidget( 
					multiple_selection = True,
					editable           = True,
					drag_mode          = 'internal'
				)
			)

		self.tool = self.addToolBar( 'global_object_manager', self.window.addToolBar() )
		self.delegate = MOAILuaDelegate( self )
		self.delegate.load( _getModulePath( 'GlobalObjectManager.lua' ) )

		# self.creatorMenu=self.addMenu(
		# 	'global_object_create',
		# 	{ 'label':'Create Global Object' }
		# 	)

		self.addTool( 'global_object_manager/add',    label = 'Add', icon = 'add' )
		self.addTool( 'global_object_manager/remove', label = 'Remove', icon = 'remove' )
		self.addTool( 'global_object_manager/clone',  label = 'Clone', icon = 'clone' )
		self.addTool( 'global_object_manager/add_group',    label = 'Add Group', icon = 'add_folder' )
		self.addTool( 'global_object_manager/----' )
		self.addTool( 'global_object_manager/refresh',    label = 'Refresh', icon = 'refresh' )
		
		# self.addMenuItem( 'main/find/find_global_object', 
		# 	dict( label = 'Find In Global Objects', shortcut = 'ctrl+alt+f' ) )

		#SIGNALS
		signals.connect( 'moai.clean', self.onMoaiClean )

		signals.connect( 'global_object.added',   self.onObjectAdded )
		signals.connect( 'global_object.removed', self.onObjectRemoved )
		signals.connect( 'global_object.renamed', self.onObjectRenamed )

		if self.getModule('introspector'):
			import GlobalObjectNodeEditor

		registerSearchEnumerator( globalObjectNameSearchEnumerator )
		registerSearchEnumerator( globalObjectSearchEnumerator )

	def onAppReady( self ):
		self.tree.rebuild()
		self.tree.setColumnWidth( 0, 150 )

	def onMoaiClean( self ):
		self.tree.clear()

	def onTool( self, tool ):
		name = tool.name
		if name == 'add':
			requestSearchView( 
				info    = 'select global object class to create',
				context = 'global_object_class',
				on_selection = lambda objName: self.createGlobalObject( objName )				
				)

		if name == 'add_group':
			group = self.delegate.safeCall( 'addGroup' )
			self.tree.addNode( group )
			self.tree.editNode( group )
			self.tree.selectNode( group )

		elif name == 'remove':
			for node in self.tree.getSelection():
				self.doCommand( 'scene_editor/remove_global_object', target = node )
				self.tree.removeNode( node )

		elif name == 'refresh':
			self.scheduleRefreshObject()

	def onMenu( self, menu ):
		if menu.name == 'find_global_object':
			requestSearchView( 
				info    = 'search for global object',
				context = 'global_object',				
				on_selection = lambda node: self.selectObject( node, True )
				)

	def needUpdate( self ):
		return True
		
	def onUpdate( self ):
		if self.refreshScheduled :
			self.refreshScheduled = False
			self.delegate.safeCall( 'reloadObjects' )
			self.tree.rebuild()

	def scheduleRefreshObject( self ):
		self.refreshScheduled = True

	def selectObject( self, target, updateTree = False ):
		self.changeSelection( target )
		if updateTree:
			self.tree.blockSignals( True )
			self.tree.selectNode( None )
			if isinstance( target, list ):
				for e in target:
					self.tree.selectNode( e, add = True)
			else:
				self.tree.selectNode( target )
			self.tree.blockSignals( False )

	def renameObject( self, obj, name ):
		obj.setName( obj, name )

	def createGlobalObject( self, objName ):
		self.doCommand( 'scene_editor/create_global_object', name = objName )

	def onObjectAdded( self, node, reason = 'new' ):
		self.tree.addNode( node )
		if reason == 'new':
			self.tree.editNode( node )
			self.tree.selectNode( node )

	def onObjectRemoved( self, node ):
		self.tree.removeNode( node )

	def onObjectRenamed( self, node, name ):
		self.tree.refreshNodeContent( node )
Esempio n. 29
0
class LayerManager(SceneEditorModule):
    def __init__(self):
        super(LayerManager, self).__init__()

    def getName(self):
        return 'layer_manager'

    def getDependency(self):
        return ['scene_editor', 'mock']

    def onLoad(self):
        #UI
        self.windowTitle = 'Layers'
        self.window = self.requestDockWindow('LayerManager',
                                             title='Layers',
                                             size=(120, 120),
                                             minSize=(120, 120),
                                             dock='left')

        #Components
        self.treeFilter = self.window.addWidget(GenericTreeFilter(),
                                                expanding=False)
        self.tree = self.window.addWidget(
            LayerTreeWidget(self.window,
                            multiple_selection=False,
                            sorting=False,
                            editable=True,
                            drag_mode='internal'))
        self.tree.hasSoloLayer = False
        self.treeFilter.setTargetTree(self.tree)

        self.tool = self.addToolBar('layer_manager', self.window.addToolBar())
        self.delegate = MOAILuaDelegate(self)
        self.delegate.load(_getModulePath('LayerManager.lua'))

        self.addTool('layer_manager/add', label='add', icon='add')
        self.addTool('layer_manager/remove', label='remove', icon='remove')
        self.addTool('layer_manager/up', label='up', icon='arrow-up')
        self.addTool('layer_manager/down', label='down', icon='arrow-down')

        #SIGNALS
        signals.connect('moai.clean', self.onMoaiClean)

    def onAppReady(self):
        self.tree.rebuild()

    def onMoaiClean(self):
        self.tree.clear()

    def onTool(self, tool):
        name = tool.name
        if name == 'add':
            layer = self.delegate.safeCall('addLayer')
            self.tree.addNode(layer)
            self.tree.editNode(layer)
            self.tree.selectNode(layer)

        elif name == 'remove':
            for l in self.tree.getSelection():
                self.delegate.safeCall('removeLayer', l)
                self.tree.removeNode(l)
        elif name == 'up':
            for l in self.tree.getSelection():
                self.delegate.safeCall('moveLayerUp', l)
                self.tree.rebuild()
                self.tree.selectNode(l)
                break
        elif name == 'down':
            for l in self.tree.getSelection():
                self.delegate.safeCall('moveLayerDown', l)
                self.tree.rebuild()
                self.tree.selectNode(l)
                break

    def changeLayerName(self, layer, name):
        layer.setName(layer, name)

    def toggleHidden(self, layer):
        layer.setVisible(layer, not layer.isVisible(layer))
        self.tree.refreshNodeContent(layer)
        signals.emit('scene.update')

    def toggleEditVisible(self, layer):
        layer.setEditorVisible(layer, not layer.isEditorVisible(layer))
        self.tree.refreshNodeContent(layer)
        signals.emit('scene.update')

    def toggleLock(self, layer):
        layer.locked = not layer.locked
        self.tree.refreshNodeContent(layer)
        signals.emit('scene.update')

    def toggleSolo(self, layer):
        solo = layer.isEditorSolo(layer)
        if solo:
            self.tree.hasSoloLayer = False
            for l in list(_MOCK.game.layers.values()):
                if l.name == '_GII_EDITOR_LAYER': continue
                if l != layer:
                    l.editorSolo = False
            layer.setEditorSolo(layer, False)
        else:
            self.tree.hasSoloLayer = True
            for l in list(_MOCK.game.layers.values()):
                if l.name == '_GII_EDITOR_LAYER': continue
                if l != layer:
                    l.editorSolo = 'hidden'
            layer.setEditorSolo(layer, True)
        self.tree.refreshAllContent()
        signals.emit('scene.update')
Esempio n. 30
0
class DeployManager(SceneEditorModule):
    def __init__(self):
        super(DeployManager, self).__init__()

    def getName(self):
        return 'deploy_manager'

    def getDependency(self):
        return ['mock']

    def onLoad(self):
        self.configPath = self.getProject().getConfigPath(_DEPLOY_CONFIG_FILE)
        #UI
        self.container = self.requestDocumentWindow('DeployManager',
                                                    title='Deployment Manager',
                                                    allowDock=False,
                                                    minSize=(300, 300),
                                                    maxSize=(300, 300))

        #Components
        self.window = self.container.addWidgetFromFile(
            _getModulePath('DeployManager.ui'))

        self.delegate = MOAILuaDelegate(self)
        self.delegate.load(_getModulePath('DeployManager.lua'))

        #scene tree
        layout = QtGui.QVBoxLayout()
        self.window.containerSceneTree.setLayout(layout)
        layout.setSpacing(0)
        layout.setMargin(0)

        self.treeScene = DeploySceneTree(self.window.containerSceneTree,
                                         editable=True,
                                         sorting=False,
                                         multiple_selection=False)
        self.treeScene.manager = self
        layout.addWidget(self.treeScene)

        sceneToolbar = QtGui.QToolBar(self.window.containerSceneTree)
        layout.addWidget(sceneToolbar)
        self.sceneTool = self.addToolBar('deploy_scene', sceneToolbar)
        self.addTool('deploy_scene/add_scene', label='add', icon='add')
        self.addTool('deploy_scene/remove_scene',
                     label='remove',
                     icon='remove')
        self.addTool('deploy_scene/move_up_scene', label='up', icon='arrow-up')
        self.addTool('deploy_scene/move_down_scene',
                     label='down',
                     icon='arrow-down')
        self.addTool('deploy_scene/----')
        self.addTool('deploy_scene/edit_scene',
                     label='change target scene',
                     icon='pencil')
        self.addTool('deploy_scene/----')
        self.addTool('deploy_scene/set_entry_scene',
                     label='set as entry',
                     icon='flag')

        #deploy target tree
        layout = QtGui.QVBoxLayout()
        self.window.containerTargetTree.setLayout(layout)
        layout.setSpacing(0)
        layout.setMargin(0)

        self.treeTarget = DeployTargetTree(self.window.containerTargetTree,
                                           editable=True,
                                           multiple_selection=False)
        self.treeTarget.manager = self
        layout.addWidget(self.treeTarget)

        targetToolbar = QtGui.QToolBar(self.window.containerTargetTree)
        layout.addWidget(targetToolbar)
        self.targetTool = self.addToolBar('deploy_target', targetToolbar)
        self.addTool('deploy_target/add_target', label='+')
        self.addTool('deploy_target/remove_target', label='-')

        #target property
        self.propertyTarget = addWidgetWithLayout(
            PropertyEditor(self.window.containerTargetProp))

        #menu
        self.addMenuItem('main/file/----')
        self.addMenuItem('main/file/deploy_manager',
                         dict(label='Deploy Manager', shortcut='F11'))

        self.addMenuItem('main/file/deploy_build',
                         dict(label='Deploy Build', shortcut='Ctrl+F11'))

        # self.container.show()
        self.window.buttonOK.clicked.connect(self.onButtonOK)

        #other
        registerSearchEnumerator(deployTargetSearchEnumerator)

        signals.connect('project.pre_deploy', self.preDeploy)
        signals.connect('project.deploy', self.onDeploy)
        signals.connect('project.post_deploy', self.postDeploy)

    def onStart(self):
        #load config
        self.loadConfig()
        #fill trees
        self.treeTarget.rebuild()
        self.treeScene.rebuild()

    def onStop(self):
        self.saveConfig()

    def loadConfig(self):
        self.delegate.safeCall('loadDeployManagerConfig', self.configPath)

    def saveConfig(self):
        self.delegate.safeCall('saveDeployManagerConfig', self.configPath)

    def getDeployTargetTypes(self):
        registry = self.delegate.safeCall('getDeployTargetTypeRegistry')
        return [name for name in registry.keys()]

    def getDeployTargets(self):
        targets = self.delegate.safeCallMethod('config', 'getTargets')
        return [obj for obj in targets.values()]

    def addDeployTarget(self, targetType):
        target = self.delegate.safeCallMethod('config', 'addDeployTarget',
                                              targetType)
        self.treeTarget.addNode(target)
        self.treeTarget.editNode(target)

    def changeDeployScene(self, targetScene):
        for sceneEntry in self.treeScene.getSelection():
            self.delegate.safeCallMethod('config', 'changeTargetScene',
                                         sceneEntry, targetScene.getPath())
            self.treeScene.refreshNode(sceneEntry)
            return

    def renameDeployTarget(self, target, name):
        target.name = name  #avoid duplicated name

    def addDeployScene(self, sceneNode):
        if not sceneNode: return
        entry = self.delegate.safeCallMethod('config', 'addDeployScene',
                                             sceneNode.getNodePath())
        self.treeScene.addNode(entry)
        self.treeScene.editNode(entry)

    def renameDeployScene(self, entry, alias):
        entry.alias = alias  #TODO: avoid duplicated name

    def getDeployScenes(self):
        scenes = self.delegate.safeCallMethod('config', 'getScenes')
        return [obj for obj in scenes.values()]

    def updateGameConfig(self):
        self.delegate.safeCallMethod('config', 'updateGameConfig')

    def preDeploy(self, context):
        self.updateGameConfig()

    def onDeploy(self, context):
        pass

    def postDeploy(self, context):
        pass

    def onTool(self, tool):
        name = tool.name
        if name == 'add_target':
            requestSearchView(info='select deploy target type',
                              context='deploy_target_type',
                              on_selection=self.addDeployTarget)

        elif name == 'remove_target':
            for target in self.treeTarget.getSelection():
                self.treeTarget.removeNode(target)
                self.delegate.safeCallMethod('config', 'removeDeployTarget',
                                             target)

        elif name == 'add_scene':
            requestSearchView(info='select scene to deploy',
                              context='asset',
                              type='scene',
                              on_selection=self.addDeployScene)

        elif name == 'edit_scene':
            requestSearchView(info='select new target scene ',
                              context='asset',
                              type='scene',
                              on_selection=self.changeDeployScene)

        elif name == 'remove_scene':
            for entry in self.treeScene.getSelection():
                self.delegate.safeCallMethod('config', 'removeDeployScene',
                                             entry)
                self.treeScene.removeNode(entry)
            self.treeScene.refreshAllContent()

        elif name == 'set_entry_scene':
            for entry in self.treeScene.getSelection():
                self.delegate.safeCallMethod('config', 'setEntryScene', entry)
                break
            self.treeScene.refreshAllContent()

        elif name == 'move_up_scene':
            for target in self.treeScene.getSelection():
                self.delegate.safeCallMethod('config', 'moveSceneUp', target)
                self.treeScene.rebuild()
                self.treeScene.selectNode(target)
                break

        elif name == 'move_down_scene':
            for target in self.treeScene.getSelection():
                self.delegate.safeCallMethod('config', 'moveSceneDown', target)
                self.treeScene.rebuild()
                self.treeScene.selectNode(target)
                break

    def onMenu(self, node):
        name = node.name
        if name == 'deploy_manager':
            self.onSetFocus()

        elif name == 'deploy_build':
            app.getProject().deploy()

    def onSetFocus(self):
        self.container.show()
        self.container.raise_()

    def onButtonOK(self):
        self.saveConfig()
        self.container.hide()