Source

wxPython / wx / tools / XRCed / XMLTreeMenu.py

# Name:         XMLTreeMenu.py
# Purpose:      dynamic pulldown menu for XMLTree
# Author:       Roman Rolinsky <rolinsky@femagsoft.com>
# Created:      31.05.2007
# RCS-ID:       $Id: XMLTreeMenu.py 64107 2010-04-22 14:05:36Z ROL $

import wx
from globals import ID
from component import Manager

class XMLTreeMenu(wx.Menu):
    '''dynamic pulldown menu for XMLTree'''
    def __init__(self, container, comp, tree, createSibling, insertBefore):
        '''
        Create tree pull-down menu. createSibling flag is set if the
        child must be a sibling of the selected item, insertBefore
        flag is set if the child should be put before selected item in
        sibling mode or as the first child in non-sibling mode.
        '''
        wx.Menu.__init__(self)
        items = tree.GetSelections()
        if len(items) <= 1:
            item = tree.GetSelection()
            if not item: item = tree.root
            if not container:
                menu = self.CreateTopLevelMenu(comp)
            elif container is Manager.rootComponent and createSibling:
                menu = self.CreateTopLevelMenu(container)
            else:
                if createSibling:
                    menu = self.CreateSubMenus(container)
                else:
                    menu = self.CreateSubMenus(comp)
            # Select correct label for submenu
            if createSibling:
                if insertBefore:
                    self.AppendMenu(ID.SIBLING, 'Create Sibling', menu,
                                    'Create sibling before selected object')
                else:
                    self.AppendMenu(ID.SIBLING, 'Create Sibling', menu,
                                    'Create sibling after selected object')
            else:
                if insertBefore:
                    self.AppendMenu(ID.INSERT, 'Insert', menu,
                                    'Create object as the first child')
                else:
                    self.AppendMenu(ID.APPEND, 'Append', menu,
                                    'Create object as the last child')
            if comp is not Manager.rootComponent:
                self.Append(ID.SUBCLASS, 'Sublass...', 'Define subclass')
            self.AppendSeparator()
            
            if container:
                if container is Manager.rootComponent:
                    menu = self.CreateTopLevelMenu(container, ID.SHIFT)
                else:
                    menu = self.CreateSubMenus(container, ID.SHIFT)
                self.AppendMenu(ID.REPLACE, 'Replace With', menu,
                                'Replace selected object by another')
                self.AppendSeparator()
                
            self.Append(wx.ID_CUT, 'Cut', 'Cut to the clipboard')
            self.Append(wx.ID_COPY, 'Copy', 'Copy to the clipboard')
            if createSibling and item != tree.root:
                self.Append(ID.PASTE_SIBLING, 'Paste Sibling',
                            'Paste from the clipboard as a sibling')
            else:
                self.Append(ID.PASTE, 'Paste', 'Paste from the clipboard')
        if items:
            self.Append(wx.ID_DELETE, 'Delete', 'Delete selected objects')
        if comp.isContainer():
            self.AppendSeparator()
            self.Append(ID.EXPAND, 'Expand', 'Expand tree')
            self.Append(ID.COLLAPSE, 'Collapse', 'Collapse tree')

    def CreateTopLevelMenu(self, comp, idShift=0):
        m = wx.Menu()
        for index,component,label,help in Manager.menus['TOP_LEVEL']:
            if comp.canHaveChild(component):
                m.Append(component.id + idShift, label, help)
        m.AppendSeparator()
        m.Append(ID.REF, 'reference...', 'Create object_ref node')
        m.Append(ID.COMMENT, 'comment', 'Create comment node')        
        return m

    def CreateSubMenus(self, comp, idShift=0):
        menu = wx.Menu()
        for index,component,label,help in Manager.menus['ROOT']:
            if comp.canHaveChild(component):
                menu.Append(component.id + idShift, label, help)
        if menu.GetMenuItemCount():
            menu.AppendSeparator()
        for name in Manager.menuNames[2:]:
            # Skip empty menu groups
            if not Manager.menus.get(name, []): continue
            m = wx.Menu()
            for index,component,label,help in Manager.menus[name]:
                if comp.canHaveChild(component):
                    m.Append(component.id + idShift, label, help)
            if m.GetMenuItemCount():
                menu.AppendMenu(ID.MENU, name, m)
                menu.AppendSeparator()
            else:
                m.Destroy()
        menu.Append(ID.REF, 'reference...', 'Create object_ref node')
        menu.Append(ID.COMMENT, 'comment', 'Create comment node')
        return menu