Test for wx.TB_HORZ_LAYOUT too

[wxWidgets.git] / wxPython / demo / DragImage.py

diff --git a/wxPython/demo/DragImage.py b/wxPython/demo/DragImage.py
index c03fd9292c15ad2f5910234a90d7e3dc9025aa63..2592c977286a77d481e60f69b61fc401195f906e 100644 (file)
--- a/wxPython/demo/DragImage.py
+++ b/wxPython/demo/DragImage.py
@@ -47,6 +47,7 @@ class DragCanvas(wx.ScrolledWindow):
 
         self.SetCursor(wx.StockCursor(wx.CURSOR_ARROW))
         self.bg_bmp = images.getBackgroundBitmap()
+        self.SetBackgroundStyle(wx.BG_STYLE_CUSTOM)
 
         # Make a shape from an image and mask.  This one will demo
         # dragging outside the window
@@ -149,7 +150,7 @@ class DragCanvas(wx.ScrolledWindow):
         dc = evt.GetDC()
 
         if not dc:
-            dc = wxClientDC(self)
+            dc = wx.ClientDC(self)
             rect = self.GetUpdateRegion().GetBox()
             dc.SetClippingRect(rect)
         self.TileBackground(dc)
@@ -184,10 +185,8 @@ class DragCanvas(wx.ScrolledWindow):
         self.dragImage.EndDrag()
         self.dragImage = None
 
-        dc = wx.ClientDC(self)
-
         if self.hiliteShape:
-            self.hiliteShape.Draw(dc)
+            self.RefreshRect(self.hiliteShape.GetRect())
             self.hiliteShape = None
 
         # reposition and draw the shape
@@ -211,9 +210,10 @@ class DragCanvas(wx.ScrolledWindow):
             )
             
         self.dragShape.shown = True
-        self.dragShape.Draw(dc)
+        self.RefreshRect(self.dragShape.GetRect())
         self.dragShape = None
 
+
     # The mouse is moving
     def OnMotion(self, evt):
         # Ignore mouse movement if we're not dragging.