wxPython/demo/CustomDragAndDrop.py

   1 # 11/5/2003 - Modified by grimmtooth@softhome.net (Jeff Grimmett)
   2 #
   3 # o Updated for wx namespace
   4 #
   5 # 11/24/2003 - Modified by grimmtooth@softhome.net (Jeff Grimmett)
   6 #
   7 # o Issues around line 167. I'm stuck.
   8 #
   9
  10 import  cPickle
  11 import  wx
  12
  13 #----------------------------------------------------------------------
  14
  15
  16 class DoodlePad(wx.Window):
  17     def __init__(self, parent, log):
  18         wx.Window.__init__(self, parent, -1, style=wx.SUNKEN_BORDER)
  19         self.log = log
  20         self.SetBackgroundColour(wx.WHITE)
  21         self.lines = []
  22         self.x = self.y = 0
  23         self.SetMode("Draw")
  24
  25         wx.EVT_LEFT_DOWN(self, self.OnLeftDown)
  26         wx.EVT_LEFT_UP(self, self.OnLeftUp)
  27         wx.EVT_RIGHT_UP(self, self.OnRightUp)
  28         wx.EVT_MOTION(self, self.OnMotion)
  29         wx.EVT_PAINT(self, self.OnPaint)
  30
  31
  32     def SetMode(self, mode):
  33         self.mode = mode
  34         if self.mode == "Draw":
  35             self.SetCursor(wx.StockCursor(wx.CURSOR_PENCIL))
  36         else:
  37             self.SetCursor(wx.STANDARD_CURSOR)
  38
  39
  40     def OnPaint(self, event):
  41         dc = wx.PaintDC(self)
  42         self.DrawSavedLines(dc)
  43
  44     def DrawSavedLines(self, dc):
  45         dc.BeginDrawing()
  46         dc.SetPen(wx.Pen(wx.BLUE, 3))
  47         for line in self.lines:
  48             for coords in line:
  49                 dc.DrawLine(*coords)
  50         dc.EndDrawing()
  51
  52
  53     def OnLeftDown(self, event):
  54         if self.mode == "Drag":
  55             self.StartDragOpperation()
  56         elif self.mode == "Draw":
  57             self.curLine = []
  58             self.x, self.y = event.GetPositionTuple()
  59             self.CaptureMouse()
  60         else:
  61             wx.Bell()
  62             self.log.write("unknown mode!\n")
  63
  64
  65     def OnLeftUp(self, event):
  66         self.lines.append(self.curLine)
  67         self.curLine = []
  68         self.ReleaseMouse()
  69
  70     def OnRightUp(self, event):
  71         self.lines = []
  72         self.Refresh()
  73
  74     def OnMotion(self, event):
  75         if event.Dragging() and not self.mode == "Drag":
  76             dc = wx.ClientDC(self)
  77             dc.BeginDrawing()
  78             dc.SetPen(wx.Pen(wx.BLUE, 3))
  79             coords = ((self.x, self.y), event.GetPositionTuple())
  80             self.curLine.append(coords)
  81             dc.DrawLine(*coords)
  82             self.x, self.y = event.GetPositionTuple()
  83             dc.EndDrawing()
  84
  85
  86     def StartDragOpperation(self):
  87         # pickle the lines list
  88         linesdata = cPickle.dumps(self.lines, 1)
  89
  90         # create our own data format and use it in a
  91         # custom data object
  92         ldata = wx.CustomDataObject(wx.CustomDataFormat("DoodleLines"))
  93         ldata.SetData(linesdata)
  94
  95         # Also create a Bitmap version of the drawing
  96         size = self.GetSize()
  97         bmp = wx.EmptyBitmap(size.width, size.height)
  98         dc = wx.MemoryDC()
  99         dc.SelectObject(bmp)
 100         dc.SetBackground(wx.WHITE_BRUSH)
 101         dc.Clear()
 102         self.DrawSavedLines(dc)
 103         dc.SelectObject(wx.NullBitmap)
 104
 105         # Now make a data object for the bitmap and also a composite
 106         # data object holding both of the others.
 107         bdata = wx.BitmapDataObject(bmp)
 108         data = wx.DataObjectComposite()
 109         data.Add(ldata)
 110         data.Add(bdata)
 111
 112         # And finally, create the drop source and begin the drag
 113         # and drop opperation
 114         dropSource = wx.DropSource(self)
 115         dropSource.SetData(data)
 116         self.log.WriteText("Begining DragDrop\n")
 117         result = dropSource.DoDragDrop(wx.Drag_AllowMove)
 118         self.log.WriteText("DragDrop completed: %d\n" % result)
 119
 120         if result == wx.DragMove:
 121             self.lines = []
 122             self.Refresh()
 123
 124
 125 #----------------------------------------------------------------------
 126
 127
 128 class DoodleDropTarget(wx.PyDropTarget):
 129     def __init__(self, window, log):
 130         wx.PyDropTarget.__init__(self)
 131         self.log = log
 132         self.dv = window
 133
 134         # specify the type of data we will accept
 135         self.df = wx.CustomDataFormat("DoodleLines")
 136         self.data = wx.CustomDataObject(self.df)
 137         self.SetDataObject(self.data)
 138
 139
 140     # some virtual methods that track the progress of the drag
 141     def OnEnter(self, x, y, d):
 142         self.log.WriteText("OnEnter: %d, %d, %d\n" % (x, y, d))
 143         return d
 144
 145     def OnLeave(self):
 146         self.log.WriteText("OnLeave\n")
 147
 148     def OnDrop(self, x, y):
 149         self.log.WriteText("OnDrop: %d %d\n" % (x, y))
 150         return True
 151
 152     def OnDragOver(self, x, y, d):
 153         #self.log.WriteText("OnDragOver: %d, %d, %d\n" % (x, y, d))
 154
 155         # The value returned here tells the source what kind of visual
 156         # feedback to give.  For example, if wxDragCopy is returned then
 157         # only the copy cursor will be shown, even if the source allows
 158         # moves.  You can use the passed in (x,y) to determine what kind
 159         # of feedback to give.  In this case we return the suggested value
 160         # which is based on whether the Ctrl key is pressed.
 161         return d
 162
 163
 164
 165     # Called when OnDrop returns True.  We need to get the data and
 166     # do something with it.
 167     def OnData(self, x, y, d):
 168         self.log.WriteText("OnData: %d, %d, %d\n" % (x, y, d))
 169
 170         # copy the data from the drag source to our data object
 171         if self.GetData():
 172             # convert it back to a list of lines and give it to the viewer
 173             linesdata = self.data.GetData()
 174             lines = cPickle.loads(linesdata)
 175             self.dv.SetLines(lines)
 176
 177         # what is returned signals the source what to do
 178         # with the original data (move, copy, etc.)  In this
 179         # case we just return the suggested value given to us.
 180         return d
 181
 182
 183
 184 class DoodleViewer(wx.Window):
 185     def __init__(self, parent, log):
 186         wx.Window.__init__(self, parent, -1, style=wx.SUNKEN_BORDER)
 187         self.log = log
 188         self.SetBackgroundColour(wx.WHITE)
 189         self.lines = []
 190         self.x = self.y = 0
 191         dt = DoodleDropTarget(self, log)
 192         self.SetDropTarget(dt)
 193         wx.EVT_PAINT(self, self.OnPaint)
 194
 195
 196     def SetLines(self, lines):
 197         self.lines = lines
 198         self.Refresh()
 199
 200     def OnPaint(self, event):
 201         dc = wx.PaintDC(self)
 202         self.DrawSavedLines(dc)
 203
 204     def DrawSavedLines(self, dc):
 205         dc.BeginDrawing()
 206         dc.SetPen(wx.Pen(wx.RED, 3))
 207
 208         for line in self.lines:
 209             for coords in line:
 210                 dc.DrawLine(*coords)
 211         dc.EndDrawing()
 212
 213 #----------------------------------------------------------------------
 214
 215 class CustomDnDPanel(wx.Panel):
 216     def __init__(self, parent, log):
 217         wx.Panel.__init__(self, parent, -1)
 218
 219         self.SetFont(wx.Font(10, wx.SWISS, wx.NORMAL, wx.BOLD, False))
 220
 221         # Make the controls
 222         text1 = wx.StaticText(self, -1,
 223                             "Draw a little picture in this window\n"
 224                             "then switch the mode below and drag the\n"
 225                             "picture to the lower window or to another\n"
 226                             "application that accepts BMP's as a drop\n"
 227                              "target.\n"
 228                             )
 229
 230         rb1 = wx.RadioButton(self, -1, "Draw", style=wx.RB_GROUP)
 231         rb1.SetValue(True)
 232         rb2 = wx.RadioButton(self, -1, "Drag")
 233         rb2.SetValue(False)
 234
 235         text2 = wx.StaticText(self, -1,
 236                              "The lower window is accepting a\n"
 237                              "custom data type that is a pickled\n"
 238                              "Python list of lines data.")
 239
 240         self.pad = DoodlePad(self, log)
 241         view = DoodleViewer(self, log)
 242
 243         # put them in sizers
 244         sizer = wx.BoxSizer(wx.HORIZONTAL)
 245         box = wx.BoxSizer(wx.VERTICAL)
 246         rbox = wx.BoxSizer(wx.HORIZONTAL)
 247
 248         rbox.Add(rb1)
 249         rbox.Add(rb2)
 250         box.Add(text1, 0, wx.ALL, 10)
 251         box.Add(rbox, 0, wx.ALIGN_CENTER)
 252         box.Add((10,90))
 253         box.Add(text2, 0, wx.ALL, 10)
 254
 255         sizer.Add(box)
 256
 257         dndsizer = wx.BoxSizer(wx.VERTICAL)
 258         dndsizer.Add(self.pad, 1, wx.EXPAND|wx.ALL, 5)
 259         dndsizer.Add(view, 1, wx.EXPAND|wx.ALL, 5)
 260
 261         sizer.Add(dndsizer, 1, wx.EXPAND)
 262
 263         self.SetAutoLayout(True)
 264         self.SetSizer(sizer)
 265
 266         # Events
 267         wx.EVT_RADIOBUTTON(self, rb1.GetId(), self.OnRadioButton)
 268         wx.EVT_RADIOBUTTON(self, rb2.GetId(), self.OnRadioButton)
 269
 270
 271     def OnRadioButton(self, evt):
 272         rb = self.FindWindowById(evt.GetId())
 273         self.pad.SetMode(rb.GetLabel())
 274
 275
 276 #----------------------------------------------------------------------
 277 #----------------------------------------------------------------------
 278
 279 class TestPanel(wx.Panel):
 280     def __init__(self, parent, log):
 281         wx.Panel.__init__(self, parent, -1)
 282
 283         self.SetAutoLayout(True)
 284         sizer = wx.BoxSizer(wx.VERTICAL)
 285
 286         msg = "Custom Drag-And-Drop"
 287         text = wx.StaticText(self, -1, "", style=wx.ALIGN_CENTRE)
 288         text.SetFont(wx.Font(24, wx.SWISS, wx.NORMAL, wx.BOLD, False))
 289         text.SetLabel(msg)
 290         w,h = text.GetTextExtent(msg)
 291         text.SetSize(wx.Size(w,h+1))
 292         text.SetForegroundColour(wx.BLUE)
 293         sizer.Add(text, 0, wx.EXPAND|wx.ALL, 5)
 294         sizer.Add(wx.StaticLine(self, -1), 0, wx.EXPAND)
 295
 296         sizer.Add(CustomDnDPanel(self, log), 1, wx.EXPAND)
 297
 298         self.SetSizer(sizer)
 299
 300
 301 #----------------------------------------------------------------------
 302
 303 def runTest(frame, nb, log):
 304     #win = TestPanel(nb, log)
 305     win = CustomDnDPanel(nb, log)
 306     return win
 307
 308
 309 if __name__ == '__main__':
 310     import sys
 311
 312     class DummyLog:
 313         def WriteText(self, text):
 314             sys.stdout.write(text)
 315
 316     class TestApp(wx.App):
 317         def OnInit(self):
 318             wx.InitAllImageHandlers()
 319             self.MakeFrame()
 320             return True
 321
 322         def MakeFrame(self, event=None):
 323             frame = wx.Frame(None, -1, "Custom Drag and Drop", size=(550,400))
 324             menu = wx.Menu()
 325             menu.Append(6543, "Window")
 326             mb = wx.MenuBar()
 327             mb.Append(menu, "New")
 328             frame.SetMenuBar(mb)
 329             wx.EVT_MENU(frame, 6543, self.MakeFrame)
 330             panel = TestPanel(frame, DummyLog())
 331             frame.Show(True)
 332             self.SetTopWindow(frame)
 333
 334     #----------------------------------------------------------------------
 335
 336     app = TestApp(0)
 337     app.MainLoop()
 338
 339 #----------------------------------------------------------------------
 340
 341
 342 overview = """<html><body>
 343 This demo shows Drag and Drop using a custom data type and a custom
 344 data object.  A type called "DoodleLines" is created and a Python
 345 Pickle of a list is actually transfered in the drag and drop
 346 opperation.
 347
 348 A second data object is also created containing a bitmap of the image
 349 and is made available to any drop target that accepts bitmaps, such as
 350 MS Word.
 351
 352 The two data objects are combined in a wx.DataObjectComposite and the
 353 rest is handled by the framework.
 354 </body></html>
 355 """
 356