def _InitUI( self ):
"""
"""
dmgr = self.state.dataModelMgr
image_list = wx.ImageList( 16, 16, initialCount = 4 )
for n in ( 'unselected', 'selected', 'unselected', 'selected' ):
name = 'item_{0}_16x16.png'.format( n )
im = wx.Image( os.path.join( Config.GetResDir(), name ) )
image_list.Add( im.ConvertToBitmap() )
self.AssignStateImageList( image_list )
root_item = self.AddRoot( 'Datasets' )
#r root_item = self.AddRoot( 'Files' )
self.SetItemState( root_item, wx.TREE_ITEMSTATE_NONE )
if self.showSelectedDataSet:
item = self.AppendItem( root_item, LABEL_selectedDataSet, 0, 1 )
self.SetItemPyData( item, NAME_selectedDataSet )
self.selectedDataSetItem = item
# -- Each model
# --
if dmgr.GetDataModelCount() == 1:
dmodel = dmgr.GetFirstDataModel()
model_item = self.AppendItem( root_item, 'Dataset Types' )
self.SetItemPyData( model_item, dmodel )
#r self.SetItemText( root_item, 'Types' )
self._CreateModelItems( model_item, dmodel )
self.Expand( model_item )
elif dmgr.GetDataModelCount() > 1:
for name in dmgr.GetDataModelNames():
dmodel = dmgr.GetDataModel( name )
# model_item = self.AppendItem( root_item, name, 0, 1 )
model_item = self.AppendItem( root_item, name )
self.SetItemPyData( model_item, dmodel )
self._CreateModelItems( model_item, dmodel )
#self.Bind( wx.EVT_TREE_SEL_CHANGED, self._OnSelectionChanged )
self.Bind( wx.EVT_TREE_KEY_DOWN, self._OnKeyDown )
self.Bind( wx.EVT_TREE_STATE_IMAGE_CLICK, self._OnItemClick )
