PyCAD/InputModeCanvas.py at master · danheeks/PyCAD · GitHub

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
from PropertiesCanvas import PropertiesCanvas
from PropertiesCanvas import PropertiesObserver
import cad
import ToolImage
import wx
import Key

EXTRA_TOOLBAR_HEIGHT = 7

class InputModeObserver(PropertiesObserver):
    def __init__(self, window):
        PropertiesObserver.__init__(self, window)

    def OnSelectionChanged(self, added, removed):
        self.window.objects = cad.GetSelectedObjects()
        #self.make_initial_properties_in_refresh = True
        self.window.RemoveAndAddAll()
        #self.make_initial_properties_in_refresh = False

# This is a property grid for listing a cad object's properties

class InputModeCanvas(PropertiesCanvas):
    def __init__(self, parent):
        PropertiesCanvas.__init__(self, parent)
        self.toolBar = None
        self.inRemoveAndAddAll = False
        self.objects = []
        #self.make_initial_properties_in_refresh = False
        self.observer = InputModeObserver(self)
        cad.RegisterObserver(self.observer)
        self.AddToolBar()
        self.Bind(wx.EVT_SIZE, self.OnSize)
        self.pg.Bind( wx.EVT_KEY_DOWN, self.OnKeyDown )

    def AddToolBar(self):
        self.toolBar = wx.ToolBar(self.panel, style=wx.TB_NODIVIDER | wx.TB_FLAT)
        self.toolBar.SetToolBitmapSize(wx.Size(ToolImage.GetBitmapSize(), ToolImage.GetBitmapSize()))
        self.toolBar.Realize()

    def UpdateTitleProperty(self):
        p = wx.GetApp().frame.input_mode_canvas.pg.GetProperty('Input Mode')
        if p != None:
            p.SetValue(wx.GetApp().input_mode_object.GetTitle())
        wx.GetApp().Repaint()

    def RemoveAndAddAll(self):
        if self.in_OnPropGridChange:
            return

        if self.inRemoveAndAddAll:
            cad.MessageBox('recursion = RemoveAndAddAll')
            return
        self.inRemoveAndAddAll = True

        self.ClearProperties()

        properties = []

        # add the input_mode mode's properties
        title = cad.PropertyStringReadOnly('Input Mode', wx.GetApp().input_mode_object.GetTitle())
        # to do         if(wxGetApp().input_mode_object->TitleHighlighted())title->m_highlighted = true;
        properties.append(title)
        properties += wx.GetApp().input_mode_object.GetProperties()

        for property in properties:
            self.AddProperty(property)

        self.RecreateToolbar()

        self.SizeCode()

        self.inRemoveAndAddAll = False

    def RecreateToolbar(self):
        # add toolbar buttons
        tools = wx.GetApp().GetInputModeTools()

        self.toolBar.ClearTools()
        for tool in tools:
            image = wx.Image(tool.BitmapPath())
            image.Rescale(24, 24)
            button = self.toolBar.AddTool(wx.ID_ANY, tool.GetTitle(), wx.Bitmap(image))
            self.Bind(wx.EVT_TOOL, tool.Run, button)

        self.toolBar.Realize()

    def OnKeyDown(self, event):
        if not wx.GetApp().OnKeyDown(event):
            event.Skip()

    def SizeCode(self):
        size = self.GetClientSize()
        self.panel.SetSize(0,0,size.x, size.y)
        if self.toolBar.GetToolsCount() > 0:
            toolbar_size = self.toolBar.GetClientSize()
            toolbar_height = ToolImage.GetBitmapSize() + EXTRA_TOOLBAR_HEIGHT
            self.pg.SetSize(0, 0, size.x, size.y - toolbar_height)
            self.toolBar.SetSize(0, size.y - toolbar_height , size.x, toolbar_height )
            self.toolBar.Show()
        else:
            self.pg.SetSize(0,0, size.x, size.y)
            self.toolBar.Show(False)

    def OnSize(self, event):
        self.SizeCode()