示例#1
0
    def post_item_rename_event(self, data_obj, old_name=''):
        doc = AnimationEditor.get_active_document()
        if not doc:
            return

        if isinstance(data_obj, ctg.ae2.core.data.Set_Item):
            doc.event_manager.post_set_renamed_event([data_obj], old_name)

        elif isinstance(data_obj, ctg.ae2.core.data.Group_Item):
            doc.event_manager.post_group_renamed_event([data_obj], old_name)

        elif isinstance(data_obj, ctg.ae2.core.data.State_Tag_Item):
            doc.event_manager.post_state_tag_renamed_event([data_obj])

        elif isinstance(data_obj, ctg.ae2.core.data.Action_Tag_Item):
            doc.event_manager.post_action_tag_renamed_event([data_obj])

        elif isinstance(data_obj, ctg.ae2.core.data.Control_Filter_Item):
            doc.event_manager.post_control_filter_renamed_event([data_obj],
                                                                old_name)

        elif isinstance(data_obj, ctg.ae2.core.data.Control_Parameter_Item):
            doc.event_manager.post_control_parameter_renamed_event([data_obj],
                                                                   old_name)

        elif isinstance(data_obj, ctg.ae2.core.data.State_Machine_Item):
            doc.event_manager.post_state_machine_renamed_event([data_obj],
                                                               old_name)

        elif isinstance(data_obj, ctg.ae2.core.data.Blend_Tree_Item):
            doc.event_manager.post_blend_tree_renamed_event([data_obj],
                                                            old_name)
	def text_changed( self, event ):
		doc	= AnimationEditor.get_active_document( )
		if not doc:
			return

		sel_objs = doc.selection_manager.get_selected_objects()
		value 	= self.selected_name_text_ctrl.GetValue( )
		bg_color = wx.SystemSettings.GetColour( wx.SYS_COLOUR_LISTBOX )

		if sel_objs:
			if len( sel_objs ) == 1:
				if hasattr( sel_objs[0], 'validate_name' ):
					valid = sel_objs[0].validate_name( value )

					if not valid:
						bg_color = wx.Colour( 255, 0, 0 )

		try:
			self.selected_name_text_ctrl.SetBackgroundColour( bg_color )
		except:
			print('This fails when adding a state or action tag association')
		self.selected_name_text_ctrl.Refresh( )

		if event:
			event.Skip()
示例#3
0
    def text_changed(self, event):
        doc = AnimationEditor.get_active_document()
        if not doc:
            return

        sel_objs = doc.selection_manager.get_selected_objects()
        value = self.selected_name_text_ctrl.GetValue()
        bg_color = wx.SystemSettings.GetColour(wx.SYS_COLOUR_LISTBOX)

        if sel_objs:
            if len(sel_objs) == 1:
                if hasattr(sel_objs[0], 'validate_name'):
                    valid = sel_objs[0].validate_name(value)

                    if not valid:
                        bg_color = wx.Colour(255, 0, 0)

        try:
            self.selected_name_text_ctrl.SetBackgroundColour(bg_color)
        except:
            print('This fails when adding a state or action tag association')
        self.selected_name_text_ctrl.Refresh()

        if event:
            event.Skip()
    def get_text_style(self, grid, node, column_object, is_selected):
        style = wx.Font(pointSize=9,
                        family=wx.FONTFAMILY_DEFAULT,
                        style=wx.FONTSTYLE_NORMAL,
                        weight=wx.FONTWEIGHT_NORMAL,
                        underline=False,
                        face=u'Segoe UI',
                        encoding=wx.FONTENCODING_SYSTEM)

        if is_selected:
            if hasattr(node, 'get_render_colors'):
                node_color_set = node.get_render_colors(column_object)
                valid_column_ids = ['tag_assoc_tag', 'tag_assoc_clip']
                column_string_dict = {
                    'NODE_TYPE_TAG_STATE': 'tag_assoc_tag',
                    'NODE_TYPE_TAG_ACTION': 'tag_assoc_tag',
                    'NODE_TYPE_ANIM_CLIP': 'tag_assoc_clip'
                }

                if column_object.id_string in valid_column_ids:
                    doc = AnimationEditor.get_active_document()
                    if doc and (doc.selection_manager.
                                recent_sel_tag_associations_col
                                == column_object.id_string):
                        style = node_color_set.get_text_style(is_selected)

        return style
	def update_frame_count( self ):
		doc = AnimationEditor.get_active_document( )
		if not doc:
			return

		self.frame_count = doc.get_preview_clip_frame_count( )
		self.prg_render_progress.SetRange( ( self.frame_count - 1 ) )
	def post_item_rename_event( self, data_obj, old_name = '' ):
		doc	= AnimationEditor.get_active_document( )
		if not doc:
			return

		if isinstance( data_obj, ctg.ae2.core.data.Set_Item ):
			doc.event_manager.post_set_renamed_event( [ data_obj ], old_name )

		elif isinstance( data_obj, ctg.ae2.core.data.Group_Item ):
			doc.event_manager.post_group_renamed_event( [ data_obj ], old_name )

		elif isinstance( data_obj, ctg.ae2.core.data.State_Tag_Item ):
			doc.event_manager.post_state_tag_renamed_event( [ data_obj ] )

		elif isinstance( data_obj, ctg.ae2.core.data.Action_Tag_Item ):
			doc.event_manager.post_action_tag_renamed_event( [ data_obj ] )

		elif isinstance( data_obj, ctg.ae2.core.data.Control_Filter_Item ):
			doc.event_manager.post_control_filter_renamed_event( [ data_obj ], old_name )

		elif isinstance( data_obj, ctg.ae2.core.data.Control_Parameter_Item ):
			doc.event_manager.post_control_parameter_renamed_event( [ data_obj ], old_name )

		elif isinstance( data_obj, ctg.ae2.core.data.State_Machine_Item ):
			doc.event_manager.post_state_machine_renamed_event( [ data_obj ], old_name )

		elif isinstance( data_obj, ctg.ae2.core.data.Blend_Tree_Item ):
			doc.event_manager.post_blend_tree_renamed_event( [ data_obj ], old_name )
示例#7
0
    def on_sync_file_pressed(self, event):
        doc = AnimationEditor.get_active_document()
        if not doc:
            return

        sel_objs = doc.selection_manager.get_recent_sel_clips()
        if sel_objs:
            if len(sel_objs) == 1:
                if hasattr(sel_objs[0], 'sync_file'):
                    synced = sel_objs[0].sync_file()
                    self.update_sync_button_bg_color(synced)
	def on_sync_file_pressed( self, event ):
		doc	= AnimationEditor.get_active_document( )
		if not doc:
			return

		sel_objs = doc.selection_manager.get_recent_sel_clips()
		if sel_objs:
			if len( sel_objs ) == 1:
				if hasattr( sel_objs[0], 'sync_file'):
					synced = sel_objs[0].sync_file( )
					self.update_sync_button_bg_color( synced )
	def get_prop_pane_type_to_display( self ):
		doc = AnimationEditor.get_active_document( )
		prop_pane_type_to_display = None

		if not doc:
			return prop_pane_type_to_display

		sel_objs = doc.selection_manager.get_selected_objects()
		if sel_objs:
			prop_pane_type_to_display = list( inspect.getmro( sel_objs[0].__class__ ) )[ -2 ].__name__

		return prop_pane_type_to_display
	def setup_dialog( self ):
		doc = AnimationEditor.get_active_document( )

		if self.IsShown( ) or not doc:
			return

		if hasattr( doc, 'preview_pane' ):
			self.set_position( )
			self.Show( True )
			self.Layout( )

		else:
			return
示例#11
0
    def get_prop_pane_type_to_display(self):
        doc = AnimationEditor.get_active_document()
        prop_pane_type_to_display = None

        if not doc:
            return prop_pane_type_to_display

        sel_objs = doc.selection_manager.get_selected_objects()
        if sel_objs:
            prop_pane_type_to_display = list(
                inspect.getmro(sel_objs[0].__class__))[-2].__name__

        return prop_pane_type_to_display
示例#12
0
    def get_property_panels(self):
        """
		This method gets and returns a list of property panes based on type of item
		selected.

		*Arguments:*
			* ``none``

		*Keyword Arguments:*
			* ``none``

		*Returns:*
			* ``none``
		"""
        #get active document
        doc = AnimationEditor.get_active_document()
        #get property panes subclasses
        subclasses = ctg.ui.panel.CTG_Animation_Properties_Panel.__subclasses__(
        )

        panes_to_show = []
        if not doc:
            return panes_to_show

        #update name and type controls and return a list of type names of selected objects
        type_names = self.update_name_type_controls()

        #get selected objects
        sel_objs = doc.selection_manager.get_selected_objects()

        if not sel_objs or len(type_names) > 1:
            return panes_to_show

        #get the panes and the display weight value
        prop_pane_type_name = self.get_prop_pane_type_to_display()
        for sub_class in subclasses:
            prop_class = None
            if hasattr(sub_class, 'PROPERTY_PANE_TYPE'):
                if sub_class.PROPERTY_PANE_TYPE == prop_pane_type_name:
                    if hasattr(sel_objs[0], 'get_prop_nodes'):
                        #get child prop nodes
                        if sel_objs[0].get_prop_nodes()[0]:
                            prop_class = sub_class
                    else:
                        prop_class = sub_class

            if prop_class:
                panes_to_show.append(prop_class)

        return panes_to_show
	def get_preview_window_screen_rect( self ):
		"""
		Returns the screen rect of the AE's preview window.
		"""
		doc = AnimationEditor.get_active_document( )
		if not doc:
			return None

		rect 				= wx.Rect( 0, 0, 0, 0 )
		screen_rect 	= doc.get_preview_rect( )
		rect.SetPosition( ( screen_rect[ 0 ], screen_rect[ 1 ] ) )
		rect.SetSize( ( screen_rect[ 2 ] - screen_rect[ 0 ], screen_rect[ 3 ] - screen_rect[ 1 ] ) )

		return rect
	def on_render_movie( self, event ):
		doc = AnimationEditor.get_active_document( )
		if not doc:
			return

		self.enable_controls( False )
		self.update_frame_count( )
		self.reset_values( )

		#callbacks
		ctg.CALLBACK_SYSTEM.register_callback( self.callback_string, 'ae preview updated', self.update_current_frame )

		if event:
			event.Skip( )
	def get_property_panels( self ):
		"""
		This method gets and returns a list of property panes based on type of item
		selected.

		*Arguments:*
			* ``none``

		*Keyword Arguments:*
			* ``none``

		*Returns:*
			* ``none``
		"""
		#get active document
		doc = AnimationEditor.get_active_document( )
		#get property panes subclasses
		subclasses = ctg.ui.panel.CTG_Animation_Properties_Panel.__subclasses__()

		panes_to_show 	= [ ]
		if not doc:
			return panes_to_show

		#update name and type controls and return a list of type names of selected objects
		type_names = self.update_name_type_controls( )

		#get selected objects
		sel_objs = doc.selection_manager.get_selected_objects()

		if not sel_objs or len( type_names ) > 1:
			return panes_to_show

		#get the panes and the display weight value
		prop_pane_type_name = self.get_prop_pane_type_to_display( )
		for sub_class in subclasses:
			prop_class = None
			if hasattr( sub_class, 'PROPERTY_PANE_TYPE' ):
				if sub_class.PROPERTY_PANE_TYPE == prop_pane_type_name:
					if hasattr( sel_objs[0], 'get_prop_nodes'):
						#get child prop nodes
						if sel_objs[0].get_prop_nodes()[0]:
							prop_class = sub_class
					else:
						prop_class = sub_class

			if prop_class:
				panes_to_show.append( prop_class )

		return panes_to_show
示例#16
0
    def on_name_changed(self, event):
        doc = AnimationEditor.get_active_document()
        if not doc:
            return

        sel_objs = doc.selection_manager.get_selected_objects()
        value = self.selected_name_text_ctrl.GetValue()

        if sel_objs:
            if len(sel_objs) == 1:
                if hasattr(sel_objs[0], 'validate_name'):
                    valid = sel_objs[0].validate_name(value)

                    if valid:
                        if hasattr(sel_objs[0], 'rename'):
                            did_rename = sel_objs[0].rename(value)

                            if did_rename:

                                #obj = doc.selection_manager.get_item_from_collection( sel_objs[0], value )
                                #self.post_item_rename_event( obj, self.sel_obj_name )
                                self.sel_obj_name = sel_objs[0].get_name()
                                self.selected_name_text_ctrl.SetValue(
                                    self.sel_obj_name)

                            else:
                                dlg = wx.MessageDialog(
                                    self, 'Renaming failed.',
                                    'Renaming Failed',
                                    wx.OK | wx.ICON_INFORMATION)

                                if dlg.ShowModal() == wx.ID_OK:
                                    self.selected_name_text_ctrl.SetValue(
                                        self.sel_obj_name)

                                dlg.Destroy()

                    else:
                        dlg = wx.MessageDialog(
                            self,
                            'The name is invalid either the name exist\n or has invalid characters.',
                            'Invalid Name', wx.OK | wx.ICON_INFORMATION)

                        if dlg.ShowModal() == wx.ID_OK:
                            self.selected_name_text_ctrl.SetValue(
                                self.sel_obj_name)

                        dlg.Destroy()
	def cancel_pressed( self, event ):
		doc = AnimationEditor.get_active_document( )
		if not doc:
			return

		self.enable_controls( False )

		if not self.movie_file_created:
			self.txt_prog_message.SetLabel( 'Rendering canceled... Closing...' )
			ctg.CALLBACK_SYSTEM.unregister_callbacks( self.callback_string )

		self.reset_values( )
		ctg.ui.util.set_editor_modal( False )

		self.Show( False )
		self.Destroy( )
	def get_sel_clips_names( self ):
		doc = AnimationEditor.get_active_document( )
		sel_clips_names = [ ]

		if not doc:
			return sel_clips_names

		sel_clip_objs = doc.selection_manager.get_recent_sel_clips()
		if not sel_clip_objs:
			sel_clips_names = [ NO_CLIP_SELECTED ]
			return sel_clips_names

		for clip_obj in sel_clip_objs:
			sel_clips_names.append( clip_obj.get_name( ) )

		return sel_clips_names
	def on_name_changed( self, event ):
		doc	= AnimationEditor.get_active_document( )
		if not doc:
			return

		sel_objs 	= doc.selection_manager.get_selected_objects()
		value 		= self.selected_name_text_ctrl.GetValue( )

		if sel_objs:
			if len( sel_objs ) == 1:
				if hasattr( sel_objs[0], 'validate_name' ):
					valid = sel_objs[0].validate_name( value )

					if valid:
						if hasattr( sel_objs[0], 'rename'):
							did_rename 	= sel_objs[0].rename( value )

							if did_rename:

								#obj = doc.selection_manager.get_item_from_collection( sel_objs[0], value )
								#self.post_item_rename_event( obj, self.sel_obj_name )
								self.sel_obj_name = sel_objs[0].get_name( )
								self.selected_name_text_ctrl.SetValue( self.sel_obj_name  )

							else:
								dlg = wx.MessageDialog(self, 'Renaming failed.','Renaming Failed', wx.OK | wx.ICON_INFORMATION )

								if dlg.ShowModal() == wx.ID_OK:
									self.selected_name_text_ctrl.SetValue( self.sel_obj_name )

								dlg.Destroy()

					else:
						dlg = wx.MessageDialog(self, 'The name is invalid either the name exist\n or has invalid characters.','Invalid Name',
													  wx.OK | wx.ICON_INFORMATION )

						if dlg.ShowModal() == wx.ID_OK:
							self.selected_name_text_ctrl.SetValue( self.sel_obj_name )

						dlg.Destroy()
示例#20
0
	def get_text_style( self, grid, node, column_object, is_selected ):
		style = wx.Font( pointSize = 9,
		                 family = wx.FONTFAMILY_DEFAULT,
		                 style = wx.FONTSTYLE_NORMAL,
		                 weight = wx.FONTWEIGHT_NORMAL,
		                 underline = False,
		                 face = u'Segoe UI',
		                 encoding = wx.FONTENCODING_SYSTEM )

		if is_selected:
			if hasattr( node, 'get_render_colors' ):
				node_color_set = node.get_render_colors( column_object )
				valid_column_ids = [ 'tag_assoc_tag', 'tag_assoc_clip' ]
				column_string_dict = { 'NODE_TYPE_TAG_STATE'	: 'tag_assoc_tag',
					                    'NODE_TYPE_TAG_ACTION': 'tag_assoc_tag',
					                    'NODE_TYPE_ANIM_CLIP'	: 'tag_assoc_clip' }


				if column_object.id_string in valid_column_ids:
					doc = AnimationEditor.get_active_document( )
					if doc and ( doc.selection_manager.recent_sel_tag_associations_col == column_object.id_string ):
						style = node_color_set.get_text_style( is_selected )

		return style
示例#21
0
    def Draw(self, grid, attr, dc, rect, row, col, isSelected):

        # Get the column_collection
        column_collection = grid.get_column_collection()

        # Draw Background
        renderer = grid.get_renderer()
        renderer.draw_background(grid, dc, rect, row, col, isSelected)

        # Draw Column Highlight
        column_object = column_collection.column_order[col]

        # Get the node
        node = grid.get_row_as_node(row)
        if node.repair():
            # This node was in need of repair and was successfully repaired
            pass

        obj = node.get_data_obj()
        obj_val = column_object.get_value(node)
        obj_locked = grid.is_node_read_only(node)

        isSortColumn = (column_object == column_collection.sort_column)

        # set the proper bitmap
        bitmap = None

        # if the object is a state machine or blendtree set the appropriate icon
        doc = AnimationEditor.get_active_document()
        if doc:
            if node.is_clip_valid:
                clip_name = obj.get_tag().get_clip_name()
                if clip_name:
                    if doc.is_blend_tree(clip_name):
                        bitmap = self.bmp_blendtree
                    elif doc.is_state_machine(clip_name):
                        bitmap = self.bmp_statemachine

        #clip_name = obj.get_tag( ).get_clip_name( )
        #if clip_name:
        ## get the name lists
        #if clip_name.endswith( ' sm' ):
        #bitmap = self.bmp_statemachine
        #elif clip_name.endswith( ' bt' ):
        #bitmap = self.bmp_blendtree

        # see if the object has an error if it does override the icon with the error
        if not node.is_clip_valid or not node.is_tag_valid:
            bitmap = self.bmp_error

        if bitmap:
            image = wx.MemoryDC()
            image.SelectObject(bitmap)

        # set the bullet color
        if obj_val == True:
            color = wx.GREEN
        else:
            color = wx.LIGHT_GREY

        # set the colors
        dc.SetBrush(wx.Brush(wx.GREEN))
        dc.SetPen(wx.BLACK_PEN)

        #dc.DrawRectangleRect( rect )

        # copy the image but only to the size of the grid cell
        if bitmap:
            width, height = bitmap.GetWidth(), bitmap.GetHeight()

            if width > rect.width - 2:
                width = rect.width - 2

            if height > rect.height - 2:
                height = rect.height - 2

            dc.Blit(rect.x + 6, rect.y + 3, width, height, image, 0, 0,
                    wx.COPY, True)
示例#22
0
    def __init__(self, parent, size=(-1, -1)):
        wx.Panel.__init__(self, parent, size=size)

        #self.SetDoubleBuffered( True )

        self.folding_panes_dict = {}
        self.prop_panes_dict = {}
        self.child_prop_panes_dict = {}
        self.fold_pane_to_show = []
        self.fold_pane_to_hide = []
        self.last_shown_panes = []
        self.prop_panes_to_show = []
        self.sel_obj_name = ''
        self.fold_pane_names_dict = {}

        #main pane sizer
        main_sizer = wx.BoxSizer(wx.VERTICAL)
        self.name_inner_sizer = wx.BoxSizer(wx.HORIZONTAL)

        #folding panels images
        self.fp_images = wx.ImageList(16, 16)
        expanded = vlib.image.cache.load(
            ctg.ui.bitmaps.STR_BMP_EXPANDED_ICON_16)
        self.fp_images.Add(expanded.get_wx_bitmap())

        collapsed = vlib.image.cache.load(
            ctg.ui.bitmaps.STR_BMP_COLLAPSED_ICON_16)
        self.fp_images.Add(collapsed.get_wx_bitmap())

        #folding panels style setup
        self.fp_style = wx.lib.agw.foldpanelbar.CaptionBarStyle()
        self.fp_style.SetCaptionStyle(
            wx.lib.agw.foldpanelbar.CAPTIONBAR_FILLED_RECTANGLE)
        self.fp_style.SetFirstColour(wx.Colour(160, 160, 160, 255))
        self.fp_style.SetSecondColour(wx.Colour(0, 0, 0, 255))
        self.fp_style.SetCaptionColour(wx.Colour(0, 0, 0, 255))

        #label to display selected name
        self.name_label = wx.StaticText(self, -1, 'Name:')
        self.name_label.SetFont(
            wx.Font(8, wx.FONTFAMILY_DEFAULT, wx.NORMAL, wx.BOLD))
        self.selected_name_text_ctrl = wx.TextCtrl(self,
                                                   -1,
                                                   '',
                                                   size=(240, 20),
                                                   style=wx.TE_PROCESS_ENTER)
        self.sync_bt_ctrl = wx.Button(self, -1, '', size=(20, 20))
        self.sync_bt_ctrl.SetBackgroundColour(wx.Colour(255, 0, 0))

        #label to display selected type
        self.type_label = wx.StaticText(self, -1, 'Type:')
        self.type_label.SetFont(
            wx.Font(8, wx.FONTFAMILY_DEFAULT, wx.NORMAL, wx.BOLD))
        self.selected_type_label = wx.StaticText(self, -1, '')

        #setup name and type sizer
        self.name_inner_sizer.Add(self.selected_name_text_ctrl)
        self.name_inner_sizer.AddSpacer(3)
        self.name_inner_sizer.Add(self.sync_bt_ctrl, wx.ALIGN_TOP)

        self.name_type_sizer = wx.FlexGridSizer(cols=2, hgap=9, vgap=4)
        self.name_type_sizer.Add(self.name_label)
        self.name_type_sizer.Add(self.name_inner_sizer)
        self.name_type_sizer.Add(self.type_label)
        self.name_type_sizer.Add(self.selected_type_label)

        # scroll panel for properties folding panels
        self.scroll_panel = wx.ScrolledWindow(self,
                                              -1,
                                              pos=wx.DefaultPosition,
                                              size=self.GetSize(),
                                              style=wx.SUNKEN_BORDER)

        self.pnl_page = wx.Panel(self.scroll_panel, wx.ID_ANY)
        self.pnl_sizer = wx.BoxSizer(wx.VERTICAL)
        self.pnl_page.SetSizer(self.pnl_sizer)

        # create folding panel
        self.folding_panels = wx.lib.agw.foldpanelbar.FoldPanelBar(
            self.pnl_page, -1, wx.DefaultPosition, wx.DefaultSize,
            wx.lib.agw.foldpanelbar.FPB_VERTICAL)
        self.pnl_sizer.Add(self.folding_panels, 1, wx.EXPAND | wx.ALL, 2)
        self.scroll_panel.SetScrollRate(20, 20)

        #setup main panel sizer
        self.main_sizer = wx.BoxSizer(wx.VERTICAL)
        self.main_sizer.AddSpacer(5)
        self.main_sizer.Add(self.name_type_sizer, 0, wx.ALL, 2)
        self.main_sizer.Add(self.scroll_panel, 1, wx.EXPAND | wx.ALL, 2)
        self.SetSizer(self.main_sizer)

        #bind event to update scroll bars
        self.folding_panels.Bind(wx.lib.agw.foldpanelbar.EVT_CAPTIONBAR,
                                 self.update_scrolling_bars)
        self.Bind(wx.EVT_SIZE, self.on_panel_resize)
        self.selected_name_text_ctrl.Bind(wx.EVT_TEXT_ENTER,
                                          self.on_name_changed)
        self.selected_name_text_ctrl.Bind(wx.EVT_TEXT, self.text_changed)
        self.sync_bt_ctrl.Bind(wx.EVT_BUTTON, self.on_sync_file_pressed)
        self.scroll_panel.Bind(wx.EVT_CHILD_FOCUS,
                               self.on_folding_panel_gain_focus)
        self.Bind(wx.EVT_KEY_DOWN, self.on_key_down)

        #refresh pane
        self.refresh_ui()

        doc = AnimationEditor.get_active_document()
        setattr(doc, 'animation_properties_pane', self)
	def update_name_type_controls( self ):
		"""
		This method updates name and type controls.

		*Arguments:*
			* ``none``

		*Keyword Arguments:*
			* ``none``

		*Returns:*
			* ``type_names`` Names of all types
		"""
		doc	= AnimationEditor.get_active_document( )
		if not doc:
			return

		type_names  		= [ ]
		obj_names   		= [ ]
		enable 				= False

		sel_objs = doc.selection_manager.get_selected_objects()

		if sel_objs:
			for obj in sel_objs:

				#object type
				if hasattr( obj, "object_type_name") and obj.object_type_name != None:
					type_name 	= "{0}_{1}".format( str( obj.object_type_name ).replace(" ", "_"), "Node" )
				else:
					type_name 	= obj.__class__.__name__

				#object name
				if hasattr( obj, 'get_name'):
					obj_name = obj.get_name( )
				elif hasattr( obj, 'get_bone_name'):
					obj_name = obj.get_bone_name( )
				elif hasattr( obj, 'name'):
					obj_name = obj.name

				#add object type to list of object type
				if type_name not in type_names:
					type_names.append( type_name )

				#add object name to list of object names
				if obj_name not in obj_names:
					obj_names.append( obj_name )

			if len( sel_objs ) == 1:
				if type_names[0] in [ 'Clip_Item', 'Tag_Association_Item', 'Control_Parameter_Node', 'Rig_Bone_Item' ] or obj_names[0].lower() == 'Default'.lower():
					enable = False
				else:
					enable = True
		else:
			enable = False

		if not obj_names or len( obj_names ) > 1:
			self.selected_name_text_ctrl.SetValue( '' )
			self.sel_obj_name  = ''
		else:
			self.selected_name_text_ctrl.SetValue( obj_names[0] )
			self.sel_obj_name  = obj_names[0]

		if not type_names:
			self.selected_type_label.SetLabel( '' )
		else:
			if len( type_names ) == 1:
				self.selected_type_label.SetLabel( type_names[0] )
			else:
				self.selected_type_label.SetLabel( 'Multiple Types Selected' )

		#enable
		self.selected_name_text_ctrl.Enable( enable )
		self.name_inner_sizer.Hide( self.sync_bt_ctrl )
		self.name_inner_sizer.Layout( )
		self.name_type_sizer.Layout( )

		return type_names
示例#24
0
    def update_name_type_controls(self):
        """
		This method updates name and type controls.

		*Arguments:*
			* ``none``

		*Keyword Arguments:*
			* ``none``

		*Returns:*
			* ``type_names`` Names of all types
		"""
        doc = AnimationEditor.get_active_document()
        if not doc:
            return

        type_names = []
        obj_names = []
        enable = False

        sel_objs = doc.selection_manager.get_selected_objects()

        if sel_objs:
            for obj in sel_objs:

                #object type
                if hasattr(
                        obj,
                        "object_type_name") and obj.object_type_name != None:
                    type_name = "{0}_{1}".format(
                        str(obj.object_type_name).replace(" ", "_"), "Node")
                else:
                    type_name = obj.__class__.__name__

                #object name
                if hasattr(obj, 'get_name'):
                    obj_name = obj.get_name()
                elif hasattr(obj, 'get_bone_name'):
                    obj_name = obj.get_bone_name()
                elif hasattr(obj, 'name'):
                    obj_name = obj.name

                #add object type to list of object type
                if type_name not in type_names:
                    type_names.append(type_name)

                #add object name to list of object names
                if obj_name not in obj_names:
                    obj_names.append(obj_name)

            if len(sel_objs) == 1:
                if type_names[0] in [
                        'Clip_Item', 'Tag_Association_Item',
                        'Control_Parameter_Node', 'Rig_Bone_Item'
                ] or obj_names[0].lower() == 'Default'.lower():
                    enable = False
                else:
                    enable = True
        else:
            enable = False

        if not obj_names or len(obj_names) > 1:
            self.selected_name_text_ctrl.SetValue('')
            self.sel_obj_name = ''
        else:
            self.selected_name_text_ctrl.SetValue(obj_names[0])
            self.sel_obj_name = obj_names[0]

        if not type_names:
            self.selected_type_label.SetLabel('')
        else:
            if len(type_names) == 1:
                self.selected_type_label.SetLabel(type_names[0])
            else:
                self.selected_type_label.SetLabel('Multiple Types Selected')

        #enable
        self.selected_name_text_ctrl.Enable(enable)
        self.name_inner_sizer.Hide(self.sync_bt_ctrl)
        self.name_inner_sizer.Layout()
        self.name_type_sizer.Layout()

        return type_names
	def __init__( self, parent, size = (-1, -1 ) ):
		wx.Panel.__init__( self, parent, size=size )

		#self.SetDoubleBuffered( True )

		self.folding_panes_dict 		= { }
		self.prop_panes_dict 			= { }
		self.child_prop_panes_dict 	= { }
		self.fold_pane_to_show  		= [ ]
		self.fold_pane_to_hide 			= [ ]
		self.last_shown_panes 			= [ ]
		self.prop_panes_to_show       = [ ]
		self.sel_obj_name 				= ''
		self.fold_pane_names_dict     = { }

		#main pane sizer
		main_sizer = wx.BoxSizer( wx.VERTICAL )
		self.name_inner_sizer = wx.BoxSizer( wx.HORIZONTAL )

		#folding panels images
		self.fp_images = wx.ImageList( 16, 16 )
		expanded = vlib.image.cache.load( ctg.ui.bitmaps.STR_BMP_EXPANDED_ICON_16 )
		self.fp_images.Add( expanded.get_wx_bitmap( ) )

		collapsed = vlib.image.cache.load( ctg.ui.bitmaps.STR_BMP_COLLAPSED_ICON_16 )
		self.fp_images.Add( collapsed.get_wx_bitmap( ) )

		#folding panels style setup
		self.fp_style = wx.lib.agw.foldpanelbar.CaptionBarStyle( )
		self.fp_style.SetCaptionStyle( wx.lib.agw.foldpanelbar.CAPTIONBAR_FILLED_RECTANGLE )
		self.fp_style.SetFirstColour( wx.Colour( 160, 160, 160, 255 ) )
		self.fp_style.SetSecondColour( wx.Colour( 0, 0, 0, 255 ) )
		self.fp_style.SetCaptionColour( wx.Colour( 0, 0, 0, 255 ) )

		#label to display selected name
		self.name_label = wx.StaticText( self, -1, 'Name:' )
		self.name_label.SetFont( wx.Font( 8, wx.FONTFAMILY_DEFAULT, wx.NORMAL, wx.BOLD ))
		self.selected_name_text_ctrl = wx.TextCtrl( self, -1, '', size = (240, 20), style = wx.TE_PROCESS_ENTER )
		self.sync_bt_ctrl   	= wx.Button( self, -1, '', size=(20, 20))
		self.sync_bt_ctrl.SetBackgroundColour( wx.Colour( 255, 0, 0 ) )

		#label to display selected type
		self.type_label = wx.StaticText( self, -1, 'Type:' )
		self.type_label.SetFont( wx.Font( 8, wx.FONTFAMILY_DEFAULT, wx.NORMAL, wx.BOLD ))
		self.selected_type_label = wx.StaticText( self, -1, '' )

		#setup name and type sizer
		self.name_inner_sizer.Add( self.selected_name_text_ctrl )
		self.name_inner_sizer.AddSpacer( 3 )
		self.name_inner_sizer.Add( self.sync_bt_ctrl, wx.ALIGN_TOP )

		self.name_type_sizer = wx.FlexGridSizer(cols=2, hgap=9, vgap=4)
		self.name_type_sizer.Add( self.name_label )
		self.name_type_sizer.Add( self.name_inner_sizer )
		self.name_type_sizer.Add( self.type_label )
		self.name_type_sizer.Add( self.selected_type_label )

		# scroll panel for properties folding panels
		self.scroll_panel = wx.ScrolledWindow( self, -1, pos=wx.DefaultPosition, size=self.GetSize( ), style=wx.SUNKEN_BORDER )

		self.pnl_page  = wx.Panel( self.scroll_panel, wx.ID_ANY )
		self.pnl_sizer = wx.BoxSizer( wx.VERTICAL )
		self.pnl_page.SetSizer( self.pnl_sizer )

		# create folding panel
		self.folding_panels = wx.lib.agw.foldpanelbar.FoldPanelBar( self.pnl_page, -1, wx.DefaultPosition, wx.DefaultSize, wx.lib.agw.foldpanelbar.FPB_VERTICAL )
		self.pnl_sizer.Add( self.folding_panels, 1, wx.EXPAND|wx.ALL, 2 )
		self.scroll_panel.SetScrollRate( 20, 20 )

		#setup main panel sizer
		self.main_sizer = wx.BoxSizer( wx.VERTICAL )
		self.main_sizer.AddSpacer( 5 )
		self.main_sizer.Add( self.name_type_sizer, 0, wx.ALL, 2 )
		self.main_sizer.Add( self.scroll_panel, 1, wx.EXPAND|wx.ALL, 2  )
		self.SetSizer( self.main_sizer )

		#bind event to update scroll bars
		self.folding_panels.Bind( wx.lib.agw.foldpanelbar.EVT_CAPTIONBAR, self.update_scrolling_bars )
		self.Bind( wx.EVT_SIZE, self.on_panel_resize )
		self.selected_name_text_ctrl.Bind( wx.EVT_TEXT_ENTER, self.on_name_changed )
		self.selected_name_text_ctrl.Bind( wx.EVT_TEXT, self.text_changed )
		self.sync_bt_ctrl.Bind( wx.EVT_BUTTON, self.on_sync_file_pressed )
		self.scroll_panel.Bind( wx.EVT_CHILD_FOCUS, self.on_folding_panel_gain_focus )
		self.Bind( wx.EVT_KEY_DOWN, self.on_key_down )

		#refresh pane
		self.refresh_ui( )

		doc = AnimationEditor.get_active_document( )
		setattr( doc, 'animation_properties_pane', self )
	def __init__( self, parent ):
		ctg.ui.dialogs.CTG_Dialog.__init__( self, parent, title = 'Render Preview Movies', size = ( 374, 350 ), lock_min_size=False, style = wx.DEFAULT_DIALOG_STYLE ) #wx.FRAME_TOOL_WINDOW )

		doc = AnimationEditor.get_active_document( )
		self.current_frame	      	= 0
		self.finished			      	= False

		self.movie_file_created			= False
		self.sel_rendered_movie_name 	= None

		self.animx_name					= self.get_sel_clips_names( )[0]
		self.frame_count					= doc.get_preview_clip_frame_count( )
		self.image_files					= [ ]
		self.path_movie_name_dict  	= { }

		self.ToggleWindowStyle( wx.RESIZE_BORDER )

		# BUILD THE UI
		#==========================================
		self.main_sizer 					= wx.BoxSizer( wx.VERTICAL )
		self.btn_sizer 					= wx.BoxSizer( wx.HORIZONTAL )
		self.output_folder_sizer 		= wx.BoxSizer( wx.HORIZONTAL )
		self.movie_ext_sizer 			= wx.BoxSizer( wx.HORIZONTAL )
		self.rendered_movies_box 		= wx.StaticBox( self, -1, 'Recent rendered movies:' )
		self.rendered_movies_sizer 	= wx.StaticBoxSizer( self.rendered_movies_box, wx.VERTICAL )


		self.txt_prog_message 			= wx.StaticText( self, -1, 'Rendering movie: {0}{1}'.format( self.animx_name, '.animx' ) )
		self.txt_output_folder 			= wx.StaticText( self, -1, 'Output Folder: ')
		self.txt_movie_ext 				= wx.StaticText( self, -1, 'Movie Ext: ')
		self.prg_render_progress 		= wx.Gauge( self, -1, self.frame_count, style = wx.GA_SMOOTH )

		self.output_folder_control 	= ctg.ui.widgets.misc.Directory_Control( self, -1, message='Choose Path:',
		                                                                       show_full_path=True,
		                                                                       default_path = OUTPUT_FOLDER,
		                                                                       size=( 224, 25 ) )


		self.btn_cancel 				= wx.Button( self, -1, "Cancel Rendering" )
		self.btn_render 				= wx.Button( self, -1, "Render Movie" )
		self.btn_close 				= wx.Button( self, -1, size=( 60,-1), label = "Close" )
		self.btn_play 					= wx.Button( self, -1, size=( 85, -1), label = "Play Movie" )
		self.btn_play.Enable( False )

		self.movie_type_control 	= wx.Choice( self, wx.ID_ANY, wx.DefaultPosition, ( 60, -1 ), ['mov', 'avi', 'mpeg'] , 0 )
		self.render_movies 			= wx.ListBox(self, -1, (-1,-1 ), (320, 80), [ ], wx.LB_SINGLE )
		warning_str = wx.StaticText( self, -1, "WARNING: While these movie is rendering,you cannot move any\nother window on top of the AE's preview window. Rendering data\nwill be lost if this happens." )

		self.movie_ext_sizer.AddSpacer( 22 )
		self.movie_ext_sizer.Add( self.txt_movie_ext, 0, wx.ALIGN_CENTER, 5 )
		self.movie_ext_sizer.Add( self.movie_type_control, 1, wx.EXPAND | wx.ALIGN_CENTER, 5 )

		self.output_folder_sizer.Add( self.txt_output_folder, 0, wx.ALIGN_CENTER, 5 )
		self.output_folder_sizer.Add( self.output_folder_control, 1, wx.EXPAND | wx.ALIGN_CENTER, 5 )

		self.btn_sizer.Add( self.btn_render, 0, wx.ALIGN_CENTER, 5 )
		self.btn_sizer.Add( self.btn_cancel, 0, wx.ALIGN_CENTER, 5 )
		self.btn_sizer.Add( self.btn_play, 0, wx.ALIGN_CENTER, 5 )
		self.btn_sizer.Add( self.btn_close, 0, wx.ALIGN_CENTER, 5 )

		self.rendered_movies_sizer.Add( self.render_movies, 1, wx.EXPAND | wx.ALIGN_CENTER, 5 )

		self.main_sizer.Add( self.txt_prog_message, 0, wx.EXPAND | wx.ALL, 8 )
		self.main_sizer.Add( warning_str, 0, wx.EXPAND | wx.ALL, 8 )
		self.main_sizer.AddSpacer( 6 )
		self.main_sizer.Add( self.output_folder_sizer, 1, wx.EXPAND | wx.LEFT | wx.RIGHT, 1 )
		self.main_sizer.AddSpacer( 4 )
		self.main_sizer.Add( self.movie_ext_sizer, 1, wx.EXPAND | wx.LEFT | wx.RIGHT, 1 )
		self.main_sizer.AddSpacer( 4 )
		self.main_sizer.Add( self.rendered_movies_sizer, 1, wx.EXPAND )
		self.main_sizer.Add( self.prg_render_progress, 1, wx.EXPAND| wx.ALIGN_CENTER, 10 )
		self.main_sizer.Add( self.btn_sizer, 0, wx.ALIGN_CENTER | wx.TOP, 5 )
		self.main_sizer.AddSpacer( 7 )
		#==========================================

		#bind events
		self.btn_cancel.Bind( wx.EVT_BUTTON, self.cancel_pressed )
		self.Bind( wx.EVT_CLOSE, self.cancel_pressed )
		self.btn_render.Bind( wx.EVT_BUTTON, self.on_render_movie )
		self.btn_close.Bind( wx.EVT_BUTTON, self.close )
		self.btn_play.Bind( wx.EVT_BUTTON, self.on_play_movie )
		self.render_movies.Bind( wx.EVT_LISTBOX, self.on_movie_name_selected )
		self.output_folder_control.Bind( ctg.ui.widgets.misc.EVT_DIRECTORY_SELECTED, self.on_output_folder_changed  )
		self.movie_type_control.Bind( wx.EVT_CHOICE, self.on_movie_ext_changed  )

		#callbacks
		self.callback_string = 'animation_preview_render_dialog_callbacks_%s' % id( doc )

		#set default rendering values and enable controls
		self.set_default_values( )
		self.enable_controls( True )

		#setup main sizer
		self.SetSizer( self.main_sizer )
示例#27
0
	def Draw( self, grid, attr, dc, rect, row, col, isSelected ):

		# Get the column_collection
		column_collection = grid.get_column_collection( )

		# Draw Background
		renderer = grid.get_renderer( )
		renderer.draw_background( grid, dc, rect, row, col, isSelected )

		# Draw Column Highlight
		column_object = column_collection.column_order[ col ]

		# Get the node
		node = grid.get_row_as_node( row )
		if node.repair( ):
			# This node was in need of repair and was successfully repaired
			pass

		obj = node.get_data_obj( )
		obj_val = column_object.get_value( node )
		obj_locked = grid.is_node_read_only( node )

		isSortColumn = ( column_object == column_collection.sort_column )

		# set the proper bitmap
		bitmap = None

		# if the object is a state machine or blendtree set the appropriate icon
		doc = AnimationEditor.get_active_document( )
		if doc:
			if node.is_clip_valid:
				clip_name = obj.get_tag( ).get_clip_name( )
				if clip_name:
					if doc.is_blend_tree( clip_name ):
						bitmap = self.bmp_blendtree
					elif doc.is_state_machine( clip_name ):
						bitmap = self.bmp_statemachine

		#clip_name = obj.get_tag( ).get_clip_name( )
		#if clip_name:
			## get the name lists
			#if clip_name.endswith( ' sm' ):
				#bitmap = self.bmp_statemachine
			#elif clip_name.endswith( ' bt' ):
				#bitmap = self.bmp_blendtree

		# see if the object has an error if it does override the icon with the error
		if not node.is_clip_valid or not node.is_tag_valid:
			bitmap = self.bmp_error

		if bitmap:
			image = wx.MemoryDC( )
			image.SelectObject( bitmap )

		# set the bullet color
		if obj_val == True:
			color = wx.GREEN
		else:
			color = wx.LIGHT_GREY

		# set the colors
		dc.SetBrush( wx.Brush( wx.GREEN ) )
		dc.SetPen( wx.BLACK_PEN )

		#dc.DrawRectangleRect( rect )

		# copy the image but only to the size of the grid cell
		if bitmap:
			width, height = bitmap.GetWidth(), bitmap.GetHeight()

			if width > rect.width-2:
				width = rect.width-2

			if height > rect.height-2:
				height = rect.height-2

			dc.Blit(rect.x+6, rect.y+3, width, height, image, 0, 0, wx.COPY, True)