1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186
|
#-------------------------------------------------------------------------------
#
# Defines the DockWindowShell class used to house drag and drag DockWindow
# items that are dropped on the desktop or on the DockWindowShell window.
#
# Written by: David C. Morrill
#
# Date: 11/04/2005
#
# (c) Copyright 2005 by Enthought, Inc.
#
#-------------------------------------------------------------------------------
#-------------------------------------------------------------------------------
# Imports:
#-------------------------------------------------------------------------------
import wx
from enthought.traits.api \
import HasPrivateTraits, Instance
from enthought.traits.ui.api \
import View, Group
from enthought.pyface.image_resource \
import ImageResource
from dock_window \
import DockWindow
from dock_sizer \
import DockSizer, DockSection, DockRegion, DockControl, DOCK_RIGHT, \
TabHeight, NBMarginLeft, NBMarginRight, NBMarginTop, NBMarginBottom
#-------------------------------------------------------------------------------
# Constants:
#-------------------------------------------------------------------------------
# Color used for background of windows (like dialog background color):
WindowColor = wx.Colour( 236, 233, 216 )
# DockWindowShell frame icon:
FrameIcon = ImageResource( 'shell.ico' )
#-------------------------------------------------------------------------------
# 'DockWindowShell' class:
#-------------------------------------------------------------------------------
class DockWindowShell ( HasPrivateTraits ):
#---------------------------------------------------------------------------
# Trait definitions:
#---------------------------------------------------------------------------
# The wx.Frame window which is the actual shell:
control = Instance( wx.Frame )
#---------------------------------------------------------------------------
# Initializes the object:
#---------------------------------------------------------------------------
def __init__ ( self, dock_control, use_mouse = False, **traits ):
super( DockWindowShell, self ).__init__( **traits )
parent = wx.GetTopLevelParent( dock_control.control )
while True:
next_parent = parent.GetParent()
if next_parent is None:
break
parent = next_parent
self.control = shell = wx.Frame( parent, -1, dock_control.name,
style = wx.DEFAULT_FRAME_STYLE |
wx.FRAME_FLOAT_ON_PARENT |
wx.FRAME_NO_TASKBAR )
shell.SetIcon( FrameIcon.create_icon() )
shell.SetBackgroundColour( WindowColor )
wx.EVT_CLOSE( shell, self._on_close )
self._dock_window = dw = DockWindow( shell, auto_close = True ).set(
style = 'tab' )
sizer = wx.BoxSizer( wx.VERTICAL )
sizer.Add( dw.control, 1, wx.EXPAND )
shell.SetSizer( sizer )
old_control = dock_control.control
if use_mouse:
x, y = wx.GetMousePosition()
else:
x, y = old_control.GetPositionTuple()
x, y = old_control.GetParent().ClientToScreenXY( x, y )
dx, dy = old_control.GetSize()
dx += (NBMarginLeft + NBMarginRight)
dy += (TabHeight + NBMarginTop + NBMarginBottom)
self.add_control( dock_control )
# Set the correct window size and position, accounting for the tab size
# and window borders:
shell.SetDimensions( x, y, dx, dy )
cdx, cdy = shell.GetClientSizeTuple()
ex_dx = dx - cdx
ex_dy = dy - cdy
shell.SetDimensions( x - (ex_dx / 2) - NBMarginLeft,
y - ex_dy + (ex_dx / 2) - TabHeight - NBMarginTop,
dx + ex_dx, dy + ex_dy )
shell.Show()
#---------------------------------------------------------------------------
# Adds a new DockControl to the shell window:
#---------------------------------------------------------------------------
def add_control ( self, dock_control ):
""" Adds a new DockControl to the shell window.
"""
dw = self._dock_window.control
dockable = dock_control.dockable
# If the current DockControl should be closed, then do it:
close = dockable.dockable_should_close()
if close:
dock_control.close( force = True )
# Create the new control:
control = dockable.dockable_get_control( dw )
# If the DockControl was closed, then reset it to point to the new
# control:
if close:
dock_control.set( control = control, style = 'tab' )
else:
# Create a DockControl to describe the new control:
dock_control = DockControl( control = control,
name = dock_control.name,
export = dock_control.export,
style = 'tab',
image = dock_control.image,
closeable = True )
# Finish initializing the DockControl:
dockable.dockable_init_dockcontrol( dock_control )
# Get the current DockSizer:
sizer = dw.GetSizer()
if sizer is None:
# Create the initial sizer:
dw.SetSizer( DockSizer( DockSection( contents = [ DockRegion(
contents = [ dock_control ] ) ] ) ) )
else:
# Sizer exists already, try to add the DockControl as a new
# notebook tab. If the user has reorganized the layout, then just
# dock it on the right side somewhere:
section = sizer.GetContents()
region = section.contents[0]
if isinstance( region, DockRegion ):
region.add( dock_control )
else:
section.add( dock_control, region, DOCK_RIGHT )
# Force the control to update:
dw.Layout()
dw.Refresh()
#---------------------------------------------------------------------------
# Handles the user attempting to close the window:
#---------------------------------------------------------------------------
def _on_close ( self, event ):
""" Handles the user attempting to close the window.
"""
window = self._dock_window.control
section = window.GetSizer().GetContents()
n = len( section.contents )
# Try to close each individual control:
for control in section.get_controls():
control.close( layout = False )
# If some, but not all, were closed, make sure the window gets updated:
if 0 < len( section.contents ) < n:
window.Layout()
window.Refresh()
|