[wxWidgets.git] / wxPython / wx / tools / XRCed / tools.py

# Name:         tools.py
# Purpose:      XRC editor, toolbar
# Author:       Roman Rolinsky <rolinsky@mema.ucl.ac.be>
# Created:      19.03.2003
# RCS-ID:       $Id$

from xxx import *                       # xxx imports globals and params
from tree import ID_NEW

# Icons
import images

# Groups of controls
GROUPNUM = 4
GROUP_WINDOWS, GROUP_MENUS, GROUP_SIZERS, GROUP_CONTROLS = range(GROUPNUM)

# States depending on current selection and Control/Shift keys
STATE_ROOT, STATE_MENUBAR, STATE_TOOLBAR, STATE_MENU, STATE_STDDLGBTN, STATE_ELSE = range(6)
# Left toolbar for GUI elements
class Tools(wx.Panel):
    TOOL_SIZE = (30, 30)
    def __init__(self, parent):
        if wx.Platform == '__WXGTK__':
            wx.Panel.__init__(self, parent, -1,
                             style=wx.RAISED_BORDER|wx.WANTS_CHARS)
        else:
            wx.Panel.__init__(self, parent, -1, style=wx.WANTS_CHARS)
        # Create sizer for groups
        self.sizer = wx.BoxSizer(wx.VERTICAL)
        # Data to create buttons
        pullDownMenu = g.pullDownMenu
        self.groups = []
        self.ctrl = self.shift = False
        # Current state (what to enable/disable)
        self.state = None
        groups = [
            ["Windows",
             (ID_NEW.FRAME, images.getToolFrameBitmap()),
             (ID_NEW.DIALOG, images.getToolDialogBitmap()),
             (ID_NEW.PANEL, images.getToolPanelBitmap())],
            ["Menus",
             (ID_NEW.TOOL_BAR, images.getToolToolBarBitmap()),
             (ID_NEW.MENU_BAR, images.getToolMenuBarBitmap()),
             (ID_NEW.MENU, images.getToolMenuBitmap()),
             (ID_NEW.TOOL, images.getToolToolBitmap()),
             (ID_NEW.MENU_ITEM, images.getToolMenuItemBitmap()),
             (ID_NEW.SEPARATOR, images.getToolSeparatorBitmap())],
            ["Sizers",
             (ID_NEW.BOX_SIZER, images.getToolBoxSizerBitmap()),
             (ID_NEW.STATIC_BOX_SIZER, images.getToolStaticBoxSizerBitmap()),
             (ID_NEW.GRID_SIZER, images.getToolGridSizerBitmap()),
             (ID_NEW.FLEX_GRID_SIZER, images.getToolFlexGridSizerBitmap()),
             (ID_NEW.GRID_BAG_SIZER, images.getToolGridBagSizerBitmap()),
             (ID_NEW.SPACER, images.getToolSpacerBitmap())],
            ["Controls",
             (ID_NEW.STATIC_TEXT, images.getToolStaticTextBitmap()),
             (ID_NEW.STATIC_BITMAP, images.getToolStaticBitmapBitmap()),
             (ID_NEW.STATIC_LINE, images.getToolStaticLineBitmap()),
             
             (ID_NEW.BUTTON, images.getToolButtonBitmap()),
             (ID_NEW.BITMAP_BUTTON, images.getToolBitmapButtonBitmap()),
             (ID_NEW.STATIC_BOX, images.getToolStaticBoxBitmap()),
             
             (ID_NEW.TEXT_CTRL, images.getToolTextCtrlBitmap()),
             (ID_NEW.COMBO_BOX, images.getToolComboBoxBitmap()),
             (ID_NEW.CHOICE, images.getToolChoiceBitmap()),
             
             (ID_NEW.RADIO_BUTTON, images.getToolRadioButtonBitmap()),
             (ID_NEW.CHECK_BOX, images.getToolCheckBoxBitmap()),
             (ID_NEW.RADIO_BOX, images.getToolRadioBoxBitmap()),
             
             (ID_NEW.SPIN_CTRL, images.getToolSpinCtrlBitmap()),
             (ID_NEW.SPIN_BUTTON, images.getToolSpinButtonBitmap()),
             (ID_NEW.SCROLL_BAR, images.getToolScrollBarBitmap()),

             (ID_NEW.SLIDER, images.getToolSliderBitmap()),
             (ID_NEW.GAUGE, images.getToolGaugeBitmap()),
             (ID_NEW.TREE_CTRL, images.getToolTreeCtrlBitmap()),
             
             (ID_NEW.LIST_BOX, images.getToolListBoxBitmap()),
             (ID_NEW.CHECK_LIST, images.getToolCheckListBitmap()),
             (ID_NEW.LIST_CTRL, images.getToolListCtrlBitmap()),
             
             (ID_NEW.NOTEBOOK, images.getToolNotebookBitmap()),
             (ID_NEW.SPLITTER_WINDOW, images.getToolSplitterWindowBitmap()),

             (ID_NEW.UNKNOWN, images.getToolUnknownBitmap())]
            ]
        self.boxes = {}
        for grp in groups:
            self.AddGroup(grp[0])
            for b in grp[1:]:
                self.AddButton(b[0], b[1], g.pullDownMenu.createMap[b[0]])
        self.SetSizerAndFit(self.sizer)
        # Allow to be resized in vertical direction only
        self.SetSizeHints(self.GetSize()[0], -1)
        # Events
        wx.EVT_COMMAND_RANGE(self, ID_NEW.PANEL, ID_NEW.LAST,
                             wx.wxEVT_COMMAND_BUTTON_CLICKED, g.frame.OnCreate)
        wx.EVT_KEY_DOWN(self, self.OnKeyDown)
        wx.EVT_KEY_UP(self, self.OnKeyUp)
        self.Bind(wx.EVT_LEFT_DOWN, self.OnClickBox)

        self.drag = None

    def AddButton(self, id, image, text):
        from wx.lib import buttons
        button = buttons.GenBitmapButton(self, id, image, size=self.TOOL_SIZE,
                                         style=wx.NO_BORDER|wx.WANTS_CHARS)
        button.SetBezelWidth(0)
        wx.EVT_KEY_DOWN(button, self.OnKeyDown)
        wx.EVT_KEY_UP(button, self.OnKeyUp)
        wx.EVT_LEFT_DOWN(button, self.OnLeftDownOnButton)
        wx.EVT_MOTION(button, self.OnMotionOnButton)
        button.SetToolTipString(text)
        self.curSizer.Add(button)
        self.groups[-1][1][id] = button

    def AddGroup(self, name):
        # Each group is inside box
        id = wx.NewId()
        box = wx.StaticBox(self, id, '[+] '+name, style=wx.WANTS_CHARS)
        box.SetForegroundColour(wx.Colour(64, 64, 64))
#        box.SetFont(g.smallerFont())
        box.show = True
        box.name = name
        box.gnum = len(self.groups)
        box.Bind(wx.EVT_LEFT_DOWN, self.OnClickBox)
        boxSizer = wx.StaticBoxSizer(box, wx.VERTICAL)
        boxSizer.Add((0, 4))
        self.boxes[id] = box
        self.curSizer = wx.GridSizer(0, 3)
        boxSizer.Add(self.curSizer)
        self.sizer.Add(boxSizer, 0, wx.TOP | wx.LEFT | wx.RIGHT | wx.EXPAND, 4)
        self.groups.append((box,{}))

    # Enable/disable group
    def EnableGroup(self, gnum, enable = True):
        grp = self.groups[gnum]
        #grp[0].Enable(enable)
        for b in grp[1].values(): b.Enable(enable)

    # Show/hide group
    def ShowGroup(self, gnum, show = True):
        grp = self.groups[gnum]
        grp[0].show = show
        for b in grp[1].values(): b.Show(show)

    # Enable/disable group item
    def EnableGroupItem(self, gnum, id, enable = True):
        grp = self.groups[gnum]
        grp[1][id].Enable(enable)

    # Enable/disable group items
    def EnableGroupItems(self, gnum, ids, enable = True):
        grp = self.groups[gnum]
        for id in ids:
            grp[1][id].Enable(enable)

    def OnClickBox(self, evt):
        if wx.Platform == '__WXMSW__':
            box = None
            for id,b in self.boxes.items():
                # Detect click on label
                if b.GetRect().Inside(evt.GetPosition()):
                    box = b
                    break
            if not box: return
        else:
            box = self.boxes[evt.GetId()]
        # Collapse/restore static box, change label
        self.ShowGroup(box.gnum, not box.show)
        if box.show: box.SetLabel('[+] ' + box.name)
        else: box.SetLabel('[-] ' + box.name)
        self.Layout()
        self.Refresh()
        #for b in self.boxes.items():

    # DaD
    def OnLeftDownOnButton(self, evt):
        self.posDown = evt.GetPosition()
        self.idDown = evt.GetId()
        self.btnDown = evt.GetEventObject()
        evt.Skip()

    def OnMotionOnButton(self, evt):
        # Detect dragging
        if evt.Dragging() and evt.LeftIsDown():
            d = evt.GetPosition() - self.posDown
            if max(abs(d[0]), abs(d[1])) >= 5:
                if self.btnDown.HasCapture(): 
                    # Generate up event to release mouse
                    evt = wx.MouseEvent(wx.EVT_LEFT_UP.typeId)
                    evt.SetId(self.idDown)
                    # Set flag to prevent normal button operation this time
                    self.drag = True
                    self.btnDown.ProcessEvent(evt)
                self.StartDrag()
        evt.Skip()

    def StartDrag(self):
        do = MyDataObject()
        do.SetData(str(self.idDown))
        bm = self.btnDown.GetBitmapLabel()
        # wxGTK requires wxIcon cursor, wxWIN and wxMAC require wxCursor
        if wx.Platform == '__WXGTK__':
            icon = wx.EmptyIcon()
            icon.CopyFromBitmap(bm)
            dragSource = wx.DropSource(self, icon)
        else:
            curs = wx.CursorFromImage(wx.ImageFromBitmap(bm))
            dragSource = wx.DropSource(self, curs)
        dragSource.SetData(do)
        g.frame.SetStatusText('Release the mouse button over the test window')
        dragSource.DoDragDrop()

    # Process key events
    def OnKeyDown(self, evt):
        if evt.GetKeyCode() == wx.WXK_CONTROL:
            g.tree.ctrl = True
        elif evt.GetKeyCode() == wx.WXK_SHIFT:
            g.tree.shift = True
        self.UpdateIfNeeded()
        evt.Skip()

    def OnKeyUp(self, evt):
        if evt.GetKeyCode() == wx.WXK_CONTROL:
            g.tree.ctrl = False
        elif evt.GetKeyCode() == wx.WXK_SHIFT:
            g.tree.shift = False
        self.UpdateIfNeeded()
        evt.Skip()

    def OnMouse(self, evt):
        # Update control and shift states
        g.tree.ctrl = evt.ControlDown()
        g.tree.shift = evt.ShiftDown()
        self.UpdateIfNeeded()
        evt.Skip()

    # Update UI after key presses, if necessary
    def UpdateIfNeeded(self):
        tree = g.tree
        if self.ctrl != tree.ctrl or self.shift != tree.shift:
            # Enabling is needed only for ctrl
            if self.ctrl != tree.ctrl: self.UpdateUI()
            self.ctrl = tree.ctrl
            self.shift = tree.shift
            if tree.ctrl:
                status = 'SBL'
            elif tree.shift:
                status = 'INS'
            else:
                status = ''
            g.frame.SetStatusText(status, 1)

    # Update interface
    def UpdateUI(self):
        if not self.IsShown(): return
        # Update status bar
        pullDownMenu = g.pullDownMenu
        tree = g.tree
        item = tree.selection
        # If nothing selected, disable everything and return
        if not item:
            # Disable everything
            for grp in range(GROUPNUM):
                self.EnableGroup(grp, False)
            self.state = None
            return
        if tree.ctrl: needInsert = True
        else: needInsert = tree.NeedInsert(item)
        # Enable depending on selection
        if item == tree.root or needInsert and tree.GetItemParent(item) == tree.root:
            state = STATE_ROOT
        else:            
            xxx = tree.GetPyData(item).treeObject()
            # Check parent for possible child nodes if inserting sibling
            if needInsert: xxx = xxx.parent
            if xxx.__class__ == xxxMenuBar:
                state = STATE_MENUBAR
            elif xxx.__class__ in [xxxToolBar, xxxTool] or \
                 xxx.__class__ == xxxSeparator and xxx.parent.__class__ == xxxToolBar:
                state = STATE_TOOLBAR
            elif xxx.__class__ in [xxxMenu, xxxMenuItem]:
                state = STATE_MENU
            elif xxx.__class__ == xxxStdDialogButtonSizer:
                state = STATE_STDDLGBTN
            else:
                state = STATE_ELSE

        # Enable depending on selection
        if state != self.state:
            # Disable everything
            for grp in range(GROUPNUM):
                self.EnableGroup(grp, False)
            # Enable some
            if state == STATE_ROOT:
                self.EnableGroup(GROUP_WINDOWS, True)
                self.EnableGroup(GROUP_MENUS, True)
                # But disable items
                self.EnableGroupItems(GROUP_MENUS,
                                      [ ID_NEW.TOOL,
                                        ID_NEW.MENU_ITEM,
                                        ID_NEW.SEPARATOR ],
                                      False)
            elif state == STATE_STDDLGBTN:
                pass                    # nothing can be added from toolbar
            elif state == STATE_MENUBAR:
                self.EnableGroup(GROUP_MENUS)
                self.EnableGroupItems(GROUP_MENUS,
                                      [ ID_NEW.TOOL_BAR,
                                        ID_NEW.MENU_BAR,
                                        ID_NEW.TOOL ],
                                      False)
            elif state == STATE_TOOLBAR:
                self.EnableGroup(GROUP_MENUS)
                self.EnableGroupItems(GROUP_MENUS,
                                      [ ID_NEW.TOOL_BAR,
                                        ID_NEW.MENU,
                                        ID_NEW.MENU_BAR,
                                        ID_NEW.MENU_ITEM ],
                                      False)
                self.EnableGroup(GROUP_CONTROLS)
                self.EnableGroupItems(GROUP_CONTROLS,
                                      [ ID_NEW.TREE_CTRL,
                                        ID_NEW.NOTEBOOK,
                                        ID_NEW.SPLITTER_WINDOW ],
                                      False)
            elif state == STATE_MENU:
                self.EnableGroup(GROUP_MENUS)
                self.EnableGroupItems(GROUP_MENUS,
                                      [ ID_NEW.TOOL_BAR,
                                        ID_NEW.MENU_BAR,
                                        ID_NEW.TOOL ],
                                      False)
            else:
                self.EnableGroup(GROUP_WINDOWS)
                self.EnableGroupItems(GROUP_WINDOWS,
                                      [ ID_NEW.FRAME,
                                        ID_NEW.DIALOG ],
                                      False)
                self.EnableGroup(GROUP_MENUS)
                self.EnableGroupItems(GROUP_MENUS,
                                      [ ID_NEW.MENU_BAR,
                                        ID_NEW.MENU_BAR,
                                        ID_NEW.MENU,
                                        ID_NEW.MENU_ITEM,
                                        ID_NEW.TOOL,
                                        ID_NEW.SEPARATOR ],
                                      False)
                self.EnableGroup(GROUP_SIZERS)
                self.EnableGroup(GROUP_CONTROLS)
        # Special case for *book (always executed)
        if state == STATE_ELSE:
            if xxx.__class__ in [xxxNotebook, xxxChoicebook, xxxListbook]:
                self.EnableGroup(GROUP_SIZERS, False)
            else:
                self.EnableGroup(GROUP_SIZERS)
                if not (xxx.isSizer or xxx.parent and xxx.parent.isSizer):
                    self.EnableGroupItem(GROUP_SIZERS, ID_NEW.SPACER, False)
            if xxx.__class__ == xxxFrame:
                self.EnableGroupItem(GROUP_MENUS, ID_NEW.MENU_BAR)
        # Save state
        self.state = state
Commit	Line	Data
d14a1e28 RD	1	# Name: tools.py
	2	# Purpose: XRC editor, toolbar
	3	# Author: Roman Rolinsky <rolinsky@mema.ucl.ac.be>
	4	# Created: 19.03.2003
	5	# RCS-ID: $Id$
1fded56b	6
d14a1e28 RD	7	from xxx import * # xxx imports globals and params
d14a1e28 RD	8	from tree import ID_NEW
1fded56b	9
d14a1e28 RD	10	# Icons
	11	import images
	12
	13	# Groups of controls
	14	GROUPNUM = 4
	15	GROUP_WINDOWS, GROUP_MENUS, GROUP_SIZERS, GROUP_CONTROLS = range(GROUPNUM)
	16
	17	# States depending on current selection and Control/Shift keys
c032d94e	18	STATE_ROOT, STATE_MENUBAR, STATE_TOOLBAR, STATE_MENU, STATE_STDDLGBTN, STATE_ELSE = range(6)
d14a1e28	19	# Left toolbar for GUI elements
29a41103	20	class Tools(wx.Panel):
d14a1e28 RD	21	TOOL_SIZE = (30, 30)
d14a1e28 RD	22	def __init__(self, parent):
29a41103 RD	23	if wx.Platform == '__WXGTK__':
	24	wx.Panel.__init__(self, parent, -1,
	25	style=wx.RAISED_BORDER\|wx.WANTS_CHARS)
d14a1e28	26	else:
29a41103	27	wx.Panel.__init__(self, parent, -1, style=wx.WANTS_CHARS)
d14a1e28	28	# Create sizer for groups
29a41103	29	self.sizer = wx.BoxSizer(wx.VERTICAL)
d14a1e28 RD	30	# Data to create buttons
	31	pullDownMenu = g.pullDownMenu
	32	self.groups = []
	33	self.ctrl = self.shift = False
	34	# Current state (what to enable/disable)
	35	self.state = None
	36	groups = [
	37	["Windows",
	38	(ID_NEW.FRAME, images.getToolFrameBitmap()),
	39	(ID_NEW.DIALOG, images.getToolDialogBitmap()),
	40	(ID_NEW.PANEL, images.getToolPanelBitmap())],
	41	["Menus",
	42	(ID_NEW.TOOL_BAR, images.getToolToolBarBitmap()),
	43	(ID_NEW.MENU_BAR, images.getToolMenuBarBitmap()),
	44	(ID_NEW.MENU, images.getToolMenuBitmap()),
	45	(ID_NEW.TOOL, images.getToolToolBitmap()),
	46	(ID_NEW.MENU_ITEM, images.getToolMenuItemBitmap()),
	47	(ID_NEW.SEPARATOR, images.getToolSeparatorBitmap())],
	48	["Sizers",
	49	(ID_NEW.BOX_SIZER, images.getToolBoxSizerBitmap()),
	50	(ID_NEW.STATIC_BOX_SIZER, images.getToolStaticBoxSizerBitmap()),
	51	(ID_NEW.GRID_SIZER, images.getToolGridSizerBitmap()),
	52	(ID_NEW.FLEX_GRID_SIZER, images.getToolFlexGridSizerBitmap()),
a4c013b2	53	(ID_NEW.GRID_BAG_SIZER, images.getToolGridBagSizerBitmap()),
d14a1e28 RD	54	(ID_NEW.SPACER, images.getToolSpacerBitmap())],
	55	["Controls",
	56	(ID_NEW.STATIC_TEXT, images.getToolStaticTextBitmap()),
	57	(ID_NEW.STATIC_BITMAP, images.getToolStaticBitmapBitmap()),
	58	(ID_NEW.STATIC_LINE, images.getToolStaticLineBitmap()),
	59
	60	(ID_NEW.BUTTON, images.getToolButtonBitmap()),
	61	(ID_NEW.BITMAP_BUTTON, images.getToolBitmapButtonBitmap()),
	62	(ID_NEW.STATIC_BOX, images.getToolStaticBoxBitmap()),
	63
	64	(ID_NEW.TEXT_CTRL, images.getToolTextCtrlBitmap()),
	65	(ID_NEW.COMBO_BOX, images.getToolComboBoxBitmap()),
	66	(ID_NEW.CHOICE, images.getToolChoiceBitmap()),
	67
	68	(ID_NEW.RADIO_BUTTON, images.getToolRadioButtonBitmap()),
	69	(ID_NEW.CHECK_BOX, images.getToolCheckBoxBitmap()),
	70	(ID_NEW.RADIO_BOX, images.getToolRadioBoxBitmap()),
	71
	72	(ID_NEW.SPIN_CTRL, images.getToolSpinCtrlBitmap()),
	73	(ID_NEW.SPIN_BUTTON, images.getToolSpinButtonBitmap()),
	74	(ID_NEW.SCROLL_BAR, images.getToolScrollBarBitmap()),
	75
	76	(ID_NEW.SLIDER, images.getToolSliderBitmap()),
	77	(ID_NEW.GAUGE, images.getToolGaugeBitmap()),
	78	(ID_NEW.TREE_CTRL, images.getToolTreeCtrlBitmap()),
	79
	80	(ID_NEW.LIST_BOX, images.getToolListBoxBitmap()),
	81	(ID_NEW.CHECK_LIST, images.getToolCheckListBitmap()),
	82	(ID_NEW.LIST_CTRL, images.getToolListCtrlBitmap()),
	83
	84	(ID_NEW.NOTEBOOK, images.getToolNotebookBitmap()),
68ae5821	85	(ID_NEW.SPLITTER_WINDOW, images.getToolSplitterWindowBitmap()),
d14a1e28 RD	86
	87	(ID_NEW.UNKNOWN, images.getToolUnknownBitmap())]
	88	]
552240ce	89	self.boxes = {}
d14a1e28 RD	90	for grp in groups:
	91	self.AddGroup(grp[0])
	92	for b in grp[1:]:
	93	self.AddButton(b[0], b[1], g.pullDownMenu.createMap[b[0]])
d14a1e28 RD	94	self.SetSizerAndFit(self.sizer)
	95	# Allow to be resized in vertical direction only
	96	self.SetSizeHints(self.GetSize()[0], -1)
	97	# Events
29a41103	98	wx.EVT_COMMAND_RANGE(self, ID_NEW.PANEL, ID_NEW.LAST,
5c82ed90	99	wx.wxEVT_COMMAND_BUTTON_CLICKED, g.frame.OnCreate)
29a41103 RD	100	wx.EVT_KEY_DOWN(self, self.OnKeyDown)
29a41103 RD	101	wx.EVT_KEY_UP(self, self.OnKeyUp)
9e20825c	102	self.Bind(wx.EVT_LEFT_DOWN, self.OnClickBox)
d14a1e28	103
ed650677 RR	104	self.drag = None
ed650677 RR	105
d14a1e28	106	def AddButton(self, id, image, text):
29a41103 RD	107	from wx.lib import buttons
29a41103 RD	108	button = buttons.GenBitmapButton(self, id, image, size=self.TOOL_SIZE,
5c82ed90	109	style=wx.NO_BORDER\|wx.WANTS_CHARS)
5e8b10e9	110	button.SetBezelWidth(0)
29a41103 RD	111	wx.EVT_KEY_DOWN(button, self.OnKeyDown)
29a41103 RD	112	wx.EVT_KEY_UP(button, self.OnKeyUp)
ed650677 RR	113	wx.EVT_LEFT_DOWN(button, self.OnLeftDownOnButton)
ed650677 RR	114	wx.EVT_MOTION(button, self.OnMotionOnButton)
d14a1e28 RD	115	button.SetToolTipString(text)
	116	self.curSizer.Add(button)
	117	self.groups[-1][1][id] = button
	118
	119	def AddGroup(self, name):
	120	# Each group is inside box
552240ce RR	121	id = wx.NewId()
552240ce RR	122	box = wx.StaticBox(self, id, '[+] '+name, style=wx.WANTS_CHARS)
9e20825c RR	123	box.SetForegroundColour(wx.Colour(64, 64, 64))
9e20825c RR	124	# box.SetFont(g.smallerFont())
552240ce RR	125	box.show = True
	126	box.name = name
	127	box.gnum = len(self.groups)
552240ce	128	box.Bind(wx.EVT_LEFT_DOWN, self.OnClickBox)
29a41103	129	boxSizer = wx.StaticBoxSizer(box, wx.VERTICAL)
d14a1e28	130	boxSizer.Add((0, 4))
552240ce	131	self.boxes[id] = box
29a41103	132	self.curSizer = wx.GridSizer(0, 3)
d14a1e28	133	boxSizer.Add(self.curSizer)
552240ce	134	self.sizer.Add(boxSizer, 0, wx.TOP \| wx.LEFT \| wx.RIGHT \| wx.EXPAND, 4)
d14a1e28 RD	135	self.groups.append((box,{}))
	136
	137	# Enable/disable group
	138	def EnableGroup(self, gnum, enable = True):
	139	grp = self.groups[gnum]
9e20825c	140	#grp[0].Enable(enable)
d14a1e28 RD	141	for b in grp[1].values(): b.Enable(enable)
d14a1e28 RD	142
552240ce RR	143	# Show/hide group
	144	def ShowGroup(self, gnum, show = True):
	145	grp = self.groups[gnum]
	146	grp[0].show = show
	147	for b in grp[1].values(): b.Show(show)
	148
d14a1e28 RD	149	# Enable/disable group item
	150	def EnableGroupItem(self, gnum, id, enable = True):
	151	grp = self.groups[gnum]
	152	grp[1][id].Enable(enable)
	153
	154	# Enable/disable group items
	155	def EnableGroupItems(self, gnum, ids, enable = True):
	156	grp = self.groups[gnum]
	157	for id in ids:
	158	grp[1][id].Enable(enable)
	159
552240ce	160	def OnClickBox(self, evt):
9e20825c RR	161	if wx.Platform == '__WXMSW__':
	162	box = None
	163	for id,b in self.boxes.items():
	164	# Detect click on label
	165	if b.GetRect().Inside(evt.GetPosition()):
	166	box = b
	167	break
	168	if not box: return
	169	else:
	170	box = self.boxes[evt.GetId()]
552240ce RR	171	# Collapse/restore static box, change label
	172	self.ShowGroup(box.gnum, not box.show)
	173	if box.show: box.SetLabel('[+] ' + box.name)
	174	else: box.SetLabel('[-] ' + box.name)
	175	self.Layout()
9e20825c RR	176	self.Refresh()
9e20825c RR	177	#for b in self.boxes.items():
552240ce	178
ed650677 RR	179	# DaD
	180	def OnLeftDownOnButton(self, evt):
	181	self.posDown = evt.GetPosition()
	182	self.idDown = evt.GetId()
	183	self.btnDown = evt.GetEventObject()
	184	evt.Skip()
	185
	186	def OnMotionOnButton(self, evt):
	187	# Detect dragging
	188	if evt.Dragging() and evt.LeftIsDown():
	189	d = evt.GetPosition() - self.posDown
	190	if max(abs(d[0]), abs(d[1])) >= 5:
	191	if self.btnDown.HasCapture():
	192	# Generate up event to release mouse
	193	evt = wx.MouseEvent(wx.EVT_LEFT_UP.typeId)
	194	evt.SetId(self.idDown)
	195	# Set flag to prevent normal button operation this time
	196	self.drag = True
	197	self.btnDown.ProcessEvent(evt)
	198	self.StartDrag()
	199	evt.Skip()
	200
	201	def StartDrag(self):
5c82ed90	202	do = MyDataObject()
ed650677 RR	203	do.SetData(str(self.idDown))
ed650677 RR	204	bm = self.btnDown.GetBitmapLabel()
a7d8ec1e RR	205	# wxGTK requires wxIcon cursor, wxWIN and wxMAC require wxCursor
a7d8ec1e RR	206	if wx.Platform == '__WXGTK__':
5c82ed90 RR	207	icon = wx.EmptyIcon()
	208	icon.CopyFromBitmap(bm)
	209	dragSource = wx.DropSource(self, icon)
a7d8ec1e	210	else:
5c82ed90 RR	211	curs = wx.CursorFromImage(wx.ImageFromBitmap(bm))
	212	dragSource = wx.DropSource(self, curs)
	213	dragSource.SetData(do)
	214	g.frame.SetStatusText('Release the mouse button over the test window')
	215	dragSource.DoDragDrop()
	216
d14a1e28 RD	217	# Process key events
d14a1e28 RD	218	def OnKeyDown(self, evt):
29a41103	219	if evt.GetKeyCode() == wx.WXK_CONTROL:
d14a1e28	220	g.tree.ctrl = True
29a41103	221	elif evt.GetKeyCode() == wx.WXK_SHIFT:
d14a1e28 RD	222	g.tree.shift = True
	223	self.UpdateIfNeeded()
	224	evt.Skip()
	225
	226	def OnKeyUp(self, evt):
29a41103	227	if evt.GetKeyCode() == wx.WXK_CONTROL:
d14a1e28	228	g.tree.ctrl = False
29a41103	229	elif evt.GetKeyCode() == wx.WXK_SHIFT:
d14a1e28 RD	230	g.tree.shift = False
	231	self.UpdateIfNeeded()
	232	evt.Skip()
	233
	234	def OnMouse(self, evt):
	235	# Update control and shift states
	236	g.tree.ctrl = evt.ControlDown()
	237	g.tree.shift = evt.ShiftDown()
	238	self.UpdateIfNeeded()
	239	evt.Skip()
	240
	241	# Update UI after key presses, if necessary
	242	def UpdateIfNeeded(self):
	243	tree = g.tree
	244	if self.ctrl != tree.ctrl or self.shift != tree.shift:
	245	# Enabling is needed only for ctrl
	246	if self.ctrl != tree.ctrl: self.UpdateUI()
	247	self.ctrl = tree.ctrl
	248	self.shift = tree.shift
	249	if tree.ctrl:
	250	status = 'SBL'
	251	elif tree.shift:
	252	status = 'INS'
	253	else:
	254	status = ''
	255	g.frame.SetStatusText(status, 1)
	256
	257	# Update interface
	258	def UpdateUI(self):
	259	if not self.IsShown(): return
	260	# Update status bar
	261	pullDownMenu = g.pullDownMenu
	262	tree = g.tree
	263	item = tree.selection
	264	# If nothing selected, disable everything and return
	265	if not item:
	266	# Disable everything
	267	for grp in range(GROUPNUM):
	268	self.EnableGroup(grp, False)
	269	self.state = None
	270	return
	271	if tree.ctrl: needInsert = True
	272	else: needInsert = tree.NeedInsert(item)
	273	# Enable depending on selection
	274	if item == tree.root or needInsert and tree.GetItemParent(item) == tree.root:
	275	state = STATE_ROOT
	276	else:
	277	xxx = tree.GetPyData(item).treeObject()
	278	# Check parent for possible child nodes if inserting sibling
	279	if needInsert: xxx = xxx.parent
	280	if xxx.__class__ == xxxMenuBar:
	281	state = STATE_MENUBAR
	282	elif xxx.__class__ in [xxxToolBar, xxxTool] or \
	283	xxx.__class__ == xxxSeparator and xxx.parent.__class__ == xxxToolBar:
	284	state = STATE_TOOLBAR
	285	elif xxx.__class__ in [xxxMenu, xxxMenuItem]:
	286	state = STATE_MENU
c032d94e RR	287	elif xxx.__class__ == xxxStdDialogButtonSizer:
c032d94e RR	288	state = STATE_STDDLGBTN
d14a1e28 RD	289	else:
	290	state = STATE_ELSE
	291
	292	# Enable depending on selection
	293	if state != self.state:
	294	# Disable everything
	295	for grp in range(GROUPNUM):
	296	self.EnableGroup(grp, False)
	297	# Enable some
	298	if state == STATE_ROOT:
	299	self.EnableGroup(GROUP_WINDOWS, True)
	300	self.EnableGroup(GROUP_MENUS, True)
	301	# But disable items
	302	self.EnableGroupItems(GROUP_MENUS,
	303	[ ID_NEW.TOOL,
	304	ID_NEW.MENU_ITEM,
	305	ID_NEW.SEPARATOR ],
	306	False)
306b6fe9	307	elif state == STATE_STDDLGBTN:
c032d94e	308	pass # nothing can be added from toolbar
d14a1e28 RD	309	elif state == STATE_MENUBAR:
	310	self.EnableGroup(GROUP_MENUS)
	311	self.EnableGroupItems(GROUP_MENUS,
	312	[ ID_NEW.TOOL_BAR,
	313	ID_NEW.MENU_BAR,
	314	ID_NEW.TOOL ],
	315	False)
	316	elif state == STATE_TOOLBAR:
	317	self.EnableGroup(GROUP_MENUS)
	318	self.EnableGroupItems(GROUP_MENUS,
	319	[ ID_NEW.TOOL_BAR,
	320	ID_NEW.MENU,
	321	ID_NEW.MENU_BAR,
	322	ID_NEW.MENU_ITEM ],
	323	False)
	324	self.EnableGroup(GROUP_CONTROLS)
	325	self.EnableGroupItems(GROUP_CONTROLS,
	326	[ ID_NEW.TREE_CTRL,
68ae5821 RD	327	ID_NEW.NOTEBOOK,
68ae5821 RD	328	ID_NEW.SPLITTER_WINDOW ],
d14a1e28 RD	329	False)
	330	elif state == STATE_MENU:
	331	self.EnableGroup(GROUP_MENUS)
	332	self.EnableGroupItems(GROUP_MENUS,
	333	[ ID_NEW.TOOL_BAR,
	334	ID_NEW.MENU_BAR,
	335	ID_NEW.TOOL ],
	336	False)
	337	else:
	338	self.EnableGroup(GROUP_WINDOWS)
	339	self.EnableGroupItems(GROUP_WINDOWS,
	340	[ ID_NEW.FRAME,
	341	ID_NEW.DIALOG ],
	342	False)
	343	self.EnableGroup(GROUP_MENUS)
	344	self.EnableGroupItems(GROUP_MENUS,
	345	[ ID_NEW.MENU_BAR,
	346	ID_NEW.MENU_BAR,
	347	ID_NEW.MENU,
	348	ID_NEW.MENU_ITEM,
	349	ID_NEW.TOOL,
	350	ID_NEW.SEPARATOR ],
	351	False)
	352	self.EnableGroup(GROUP_SIZERS)
	353	self.EnableGroup(GROUP_CONTROLS)
306b6fe9	354	# Special case for *book (always executed)
d14a1e28	355	if state == STATE_ELSE:
306b6fe9	356	if xxx.__class__ in [xxxNotebook, xxxChoicebook, xxxListbook]:
d14a1e28 RD	357	self.EnableGroup(GROUP_SIZERS, False)
	358	else:
	359	self.EnableGroup(GROUP_SIZERS)
	360	if not (xxx.isSizer or xxx.parent and xxx.parent.isSizer):
	361	self.EnableGroupItem(GROUP_SIZERS, ID_NEW.SPACER, False)
f65bb0f8 RD	362	if xxx.__class__ == xxxFrame:
f65bb0f8 RD	363	self.EnableGroupItem(GROUP_MENUS, ID_NEW.MENU_BAR)
d14a1e28 RD	364	# Save state
d14a1e28 RD	365	self.state = state