1

     2

import os

     3

import types

     4

     5

import wx

     6

     7

from EditorPanel import EditorPanel

     8

     9

from IDEFrame import TITLE, FILEMENU, PROJECTTREE, PAGETITLES

    10

    11

from controls import TextCtrlAutoComplete

    12

from dialogs import BrowseValuesLibraryDialog

    13

from util.BitmapLibrary import GetBitmap

    14

    15

if wx.Platform == '__WXMSW__':

    16

    faces = { 'times': 'Times New Roman',

    17

              'mono' : 'Courier New',

    18

              'helv' : 'Arial',

    19

              'other': 'Comic Sans MS',

    20

              'size' : 16,

    21

             }

    22

else:

    23

    faces = { 'times': 'Times',

    24

              'mono' : 'Courier',

    25

              'helv' : 'Helvetica',

    26

              'other': 'new century schoolbook',

    27

              'size' : 18,

    28

             }

    29

    30

SCROLLBAR_UNIT = 10

    31

    32

class GenBitmapTextButton(wx.lib.buttons.GenBitmapTextButton):

    33

    def _GetLabelSize(self):

    34

        """ used internally """

    35

        w, h = self.GetTextExtent(self.GetLabel())

    36

        if not self.bmpLabel:

    37

            return w, h, False       # if there isn't a bitmap use the size of the text

    38

    39

        w_bmp = self.bmpLabel.GetWidth()+2

    40

        h_bmp = self.bmpLabel.GetHeight()+2

    41

        height = h + h_bmp

    42

        if w_bmp > w:

    43

            width = w_bmp

    44

        else:

    45

            width = w

    46

        return width, height, False

    47

    48

    def DrawLabel(self, dc, width, height, dw=0, dy=0):

    49

        bmp = self.bmpLabel

    50

        if bmp != None:     # if the bitmap is used

    51

            if self.bmpDisabled and not self.IsEnabled():

    52

                bmp = self.bmpDisabled

    53

            if self.bmpFocus and self.hasFocus:

    54

                bmp = self.bmpFocus

    55

            if self.bmpSelected and not self.up:

    56

                bmp = self.bmpSelected

    57

            bw,bh = bmp.GetWidth(), bmp.GetHeight()

    58

            if not self.up:

    59

                dw = dy = self.labelDelta

    60

            hasMask = bmp.GetMask() != None

    61

        else:

    62

            bw = bh = 0     # no bitmap -> size is zero

    63

    64

        dc.SetFont(self.GetFont())

    65

        if self.IsEnabled():

    66

            dc.SetTextForeground(self.GetForegroundColour())

    67

        else:

    68

            dc.SetTextForeground(wx.SystemSettings.GetColour(wx.SYS_COLOUR_GRAYTEXT))

    69

    70

        label = self.GetLabel()

    71

        tw, th = dc.GetTextExtent(label)        # size of text

    72

        if not self.up:

    73

            dw = dy = self.labelDelta

    74

    75

        pos_x = (width-bw)/2+dw      # adjust for bitmap and text to centre

    76

        pos_y = (height-bh-th)/2+dy

    77

        if bmp !=None:

    78

            dc.DrawBitmap(bmp, pos_x, pos_y, hasMask) # draw bitmap if available

    79

            pos_x = (width-tw)/2+dw      # adjust for bitmap and text to centre

    80

            pos_y += bh + 2

    81

    82

        dc.DrawText(label, pos_x, pos_y)      # draw the text

    83

    84

    85

class GenStaticBitmap(wx.StaticBitmap):

    87

    and accept image name as __init__ parameter, fail silently if file do not exist"""

    88

    def __init__(self, parent, ID, bitmapname,

    89

                 pos = wx.DefaultPosition, size = wx.DefaultSize,

    90

                 style = 0,

    91

                 name = "genstatbmp"):

    93

        bitmap = GetBitmap(bitmapname)

    94

        if bitmap is None:

    95

            bitmap = wx.EmptyBitmap(0, 0)

    98

                 bitmap,

    99

                 pos, size,

   100

                 style,

   101

                 name)

   102

   103

class ConfTreeNodeEditor(EditorPanel):

   105

    SHOW_BASE_PARAMS = True

   106

    SHOW_PARAMS = True

   107

    CONFNODEEDITOR_TABS = []

   109

    def _init_Editor(self, parent):

   110

        tabs_num = len(self.CONFNODEEDITOR_TABS)

   111

        if self.SHOW_PARAMS and len(self.Controler.GetParamsAttributes()) > 0:

   112

            tabs_num += 1

   114

        if tabs_num > 1 or self.SHOW_BASE_PARAMS:

   116

                style=wx.SUNKEN_BORDER|wx.SP_3D)

   118

            self.MainSizer = wx.BoxSizer(wx.VERTICAL)

   120

            if self.SHOW_BASE_PARAMS:

   121

                baseparamseditor_sizer = wx.BoxSizer(wx.HORIZONTAL)

   123

                      flag=wx.GROW|wx.ALL)

   125

                self.FullIECChannel = wx.StaticText(self.Editor, -1)

   126

                self.FullIECChannel.SetFont(

   128

                            wx.BOLD, faceName = faces["helv"]))

   130

                      flag=wx.ALIGN_CENTER_VERTICAL)

   132

                updownsizer = wx.BoxSizer(wx.VERTICAL)

   134

                      flag=wx.LEFT|wx.ALIGN_CENTER_VERTICAL)

   137

                      bitmap=GetBitmap('IECCDown'), size=wx.Size(16, 16), style=wx.NO_BORDER)

   139

                      self.IECCUpButton)

   140

                updownsizer.AddWindow(self.IECCUpButton, flag=wx.ALIGN_LEFT)

   143

                      bitmap=GetBitmap('IECCUp'), size=wx.Size(16, 16), style=wx.NO_BORDER)

   145

                      self.IECCDownButton)

   146

                updownsizer.AddWindow(self.IECCDownButton, flag=wx.ALIGN_LEFT)

   149

                      size=wx.Size(150, 25))

   150

                self.ConfNodeName.SetFont(

   152

                            wx.BOLD, faceName = faces["helv"]))

   155

                      self.ConfNodeName)

   157

                      flag=wx.LEFT|wx.RIGHT|wx.ALIGN_CENTER_VERTICAL)

   159

                buttons_sizer = self.GenerateMethodButtonSizer()

   160

                baseparamseditor_sizer.AddSizer(buttons_sizer, flag=wx.ALIGN_CENTER)

   162

            if tabs_num > 1:

   163

                self.ConfNodeNoteBook = wx.Notebook(self.Editor)

   164

                parent = self.ConfNodeNoteBook

   165

                self.MainSizer.AddWindow(self.ConfNodeNoteBook, 1, flag=wx.GROW)

   166

            else:

   167

                parent = self.Editor

   168

                self.ConfNodeNoteBook = None

   170

            self.Editor.SetSizer(self.MainSizer)

   171

        else:

   172

            self.ConfNodeNoteBook = None

   173

            self.Editor = None

   175

        for title, create_func_name in self.CONFNODEEDITOR_TABS:

   176

            editor = getattr(self, create_func_name)(parent)

   177

            if self.ConfNodeNoteBook is not None:

   178

                self.ConfNodeNoteBook.AddPage(editor, title)

   179

            elif self.SHOW_BASE_PARAMS:

   180

                self.MainSizer.AddWindow(editor, 1, flag=wx.GROW)

   181

            else:

   182

                self.Editor = editor

   184

        if self.SHOW_PARAMS and len(self.Controler.GetParamsAttributes()) > 0:

   186

            panel_style = wx.TAB_TRAVERSAL|wx.HSCROLL|wx.VSCROLL

   187

            if self.ConfNodeNoteBook is None and parent != self.Editor:

   188

                panel_style |= wx.SUNKEN_BORDER

   190

                  style=panel_style)

   191

            self.ParamsEditor.Bind(wx.EVT_SIZE, self.OnParamsEditorResize)

   192

            self.ParamsEditor.Bind(wx.EVT_SCROLLWIN, self.OnParamsEditorScroll)

   194

            self.ParamsEditorSizer = wx.FlexGridSizer(cols=1, hgap=0, rows=1, vgap=5)

   195

            self.ParamsEditorSizer.AddGrowableCol(0)

   196

            self.ParamsEditorSizer.AddGrowableRow(0)

   197

            self.ParamsEditor.SetSizer(self.ParamsEditorSizer)

   199

            self.ConfNodeParamsSizer = wx.BoxSizer(wx.VERTICAL)

   201

                  flag=wx.LEFT|wx.RIGHT|wx.BOTTOM)

   203

            self.RefreshConfNodeParamsSizer()

   205

            if self.ConfNodeNoteBook is not None:

   206

                self.ConfNodeNoteBook.AddPage(self.ParamsEditor, _("Config"))

   207

            elif self.SHOW_BASE_PARAMS:

   208

                self.MainSizer.AddWindow(self.ParamsEditor, 1, flag=wx.GROW)

   209

            else:

   210

                self.Editor = self.ParamsEditor

   211

        else:

   212

            self.ParamsEditor = None

   214

    def __init__(self, parent, controler, window, tagname=""):

   215

        EditorPanel.__init__(self, parent, tagname, window, controler)

   217

        icon_name = self.Controler.GetIconName()

   218

        if icon_name is not None:

   219

            self.SetIcon(GetBitmap(icon_name))

   220

        else:

   221

            self.SetIcon(GetBitmap("Extension"))

   223

    def __del__(self):

   224

        self.Controler.OnCloseEditor(self)

   226

    def GetTagName(self):

   227

        return self.Controler.CTNFullName()

   229

    def GetTitle(self):

   230

        fullname = self.Controler.CTNFullName()

   231

        if self.Controler.CTNTestModified():

   232

            return "~%s~" % fullname

   233

        return fullname

   235

    def HasNoModel(self):

   236

        return False

   238

    def GetBufferState(self):

   239

        return False, False

   241

    def Undo(self):

   242

        pass

   244

    def Redo(self):

   245

        pass

   247

    def RefreshView(self):

   248

        EditorPanel.RefreshView(self)

   249

        if self.SHOW_BASE_PARAMS:

   250

            self.ConfNodeName.ChangeValue(self.Controler.MandatoryParams[1].getName())

   251

            self.RefreshIECChannelControlsState()

   252

        if self.ParamsEditor is not None:

   253

            self.RefreshConfNodeParamsSizer()

   254

            self.RefreshScrollbars()

   256

    def RefreshIECChannelControlsState(self):

   257

        self.FullIECChannel.SetLabel(self.Controler.GetFullIEC_Channel())

   258

        self.IECCDownButton.Enable(self.Controler.BaseParams.getIEC_Channel() > 0)

   259

        self.MainSizer.Layout()

   261

    def RefreshConfNodeParamsSizer(self):

   262

        self.Freeze()

   263

        self.ConfNodeParamsSizer.Clear(True)

   265

        confnode_infos = self.Controler.GetParamsAttributes()

   266

        if len(confnode_infos) > 0:

   267

            self.GenerateSizerElements(self.ConfNodeParamsSizer, confnode_infos, None, False)

   269

        self.ParamsEditorSizer.Layout()

   270

        self.Thaw()

   272

    def GenerateMethodButtonSizer(self):

   273

        normal_bt_font=wx.Font(faces["size"] / 3, wx.DEFAULT, wx.NORMAL, wx.NORMAL, faceName = faces["helv"])

   274

        mouseover_bt_font=wx.Font(faces["size"] / 3, wx.DEFAULT, wx.NORMAL, wx.NORMAL, underline=True, faceName = faces["helv"])

   276

        msizer = wx.BoxSizer(wx.HORIZONTAL)

   278

        for confnode_method in self.Controler.ConfNodeMethods:

   279

            if "method" in confnode_method and confnode_method.get("shown",True):

   280

                button = GenBitmapTextButton(self.Editor,

   282

                    label=confnode_method["name"], style=wx.NO_BORDER)

   283

                button.SetFont(normal_bt_font)

   284

                button.SetToolTipString(confnode_method["tooltip"])

   285

                if confnode_method.get("push", False):

   286

                    button.Bind(wx.EVT_LEFT_DOWN, self.GetButtonCallBackFunction(confnode_method["method"], True))

   287

                else:

   288

                    button.Bind(wx.EVT_BUTTON, self.GetButtonCallBackFunction(confnode_method["method"]), button)

   289

                # a fancy underline on mouseover

   290

                def setFontStyle(b, s):

   291

                    def fn(event):

   292

                        b.SetFont(s)

   293

                        b.Refresh()

   294

                        event.Skip()

   295

                    return fn

   296

                button.Bind(wx.EVT_ENTER_WINDOW, setFontStyle(button, mouseover_bt_font))

   297

                button.Bind(wx.EVT_LEAVE_WINDOW, setFontStyle(button, normal_bt_font))

   298

                #hack to force size to mini

   299

                if not confnode_method.get("enabled",True):

   300

                    button.Disable()

   301

                msizer.AddWindow(button, flag=wx.ALIGN_CENTER)

   302

        return msizer

   304

    def GenerateSizerElements(self, sizer, elements, path, clean = True):

   305

        if clean:

   306

            sizer.Clear(True)

   307

        first = True

   308

        for element_infos in elements:

   309

            if path:

   310

                element_path = "%s.%s"%(path, element_infos["name"])

   311

            else:

   312

                element_path = element_infos["name"]

   313

            if element_infos["type"] == "element":

   314

                name = element_infos["name"]

   315

                value = element_infos["value"]

   316

                label = _(name)

   317

                if value is not None:

   318

                    label += " - %s" % _(value)

   320

                      label=_(label), size=wx.Size(10, 0))

   321

                staticboxsizer = wx.StaticBoxSizer(staticbox, wx.VERTICAL)

   322

                if first:

   324

                          flag=wx.GROW|wx.TOP|wx.BOTTOM)

   325

                else:

   327

                          flag=wx.GROW|wx.BOTTOM)

   330

                                           element_path)

   331

            else:

   332

                boxsizer = wx.FlexGridSizer(cols=3, rows=1)

   333

                boxsizer.AddGrowableCol(1)

   334

                if first:

   336

                          flag=wx.GROW|wx.ALL)

   337

                else:

   339

                          flag=wx.GROW|wx.LEFT|wx.RIGHT|wx.BOTTOM)

   341

                staticbitmap = GenStaticBitmap(ID=-1, bitmapname=element_infos["name"],

   342

                    name="%s_bitmap"%element_infos["name"], parent=self.ParamsEditor,

   343

                    pos=wx.Point(0, 0), size=wx.Size(24, 24), style=0)

   344

                boxsizer.AddWindow(staticbitmap, border=5, flag=wx.RIGHT)

   347

                      label="%s:"%_(element_infos["name"]))

   349

                      flag=wx.ALIGN_CENTER_VERTICAL|wx.RIGHT)

   351

                if isinstance(element_infos["type"], types.ListType):

   352

                    if isinstance(element_infos["value"], types.TupleType):

   353

                        browse_boxsizer = wx.BoxSizer(wx.HORIZONTAL)

   354

                        boxsizer.AddSizer(browse_boxsizer)

   357

                              size=wx.Size(275, -1), style=wx.TE_READONLY)

   358

                        if element_infos["value"] is not None:

   359

                            textctrl.SetValue(element_infos["value"][0])

   360

                            value_infos = element_infos["value"][1]

   361

                        else:

   362

                            value_infos = None

   363

                        browse_boxsizer.AddWindow(textctrl)

   366

                              label="...", size=wx.Size(25, 25))

   367

                        browse_boxsizer.AddWindow(button)

   371

                                    button)

   372

                    else:

   374

                              size=wx.Size(300, -1), style=wx.CB_READONLY)

   375

                        boxsizer.AddWindow(combobox)

   377

                        if element_infos["use"] == "optional":

   378

                            combobox.Append("")

   379

                        if len(element_infos["type"]) > 0 and isinstance(element_infos["type"][0], types.TupleType):

   380

                            for choice, xsdclass in element_infos["type"]:

   381

                                combobox.Append(choice)

   382

                            name = element_infos["name"]

   383

                            value = element_infos["value"]

   386

                                  label="%s - %s"%(_(name), _(value)), size=wx.Size(10, 0))

   387

                            staticboxsizer = wx.StaticBoxSizer(staticbox, wx.VERTICAL)

   388

                            sizer.AddSizer(staticboxsizer, border=5, flag=wx.GROW|wx.BOTTOM)

   389

                            self.GenerateSizerElements(staticboxsizer, element_infos["children"], element_path)

   390

                            callback = self.GetChoiceContentCallBackFunction(combobox, staticboxsizer, element_path)

   391

                        else:

   392

                            for choice in element_infos["type"]:

   393

                                combobox.Append(choice)

   394

                            callback = self.GetChoiceCallBackFunction(combobox, element_path)

   395

                        if element_infos["value"] is None:

   396

                            combobox.SetStringSelection("")

   397

                        else:

   398

                            combobox.SetStringSelection(element_infos["value"])

   399

                        combobox.Bind(wx.EVT_COMBOBOX, callback, combobox)

   401

                elif isinstance(element_infos["type"], types.DictType):

   402

                    scmin = -(2**31)

   403

                    scmax = 2**31-1

   404

                    if "min" in element_infos["type"]:

   405

                        scmin = element_infos["type"]["min"]

   406

                    if "max" in element_infos["type"]:

   407

                        scmax = element_infos["type"]["max"]

   409

                          size=wx.Size(300, -1), style=wx.SP_ARROW_KEYS|wx.ALIGN_RIGHT)

   410

                    spinctrl.SetRange(scmin, scmax)

   411

                    boxsizer.AddWindow(spinctrl)

   412

                    if element_infos["value"] is not None:

   413

                        spinctrl.SetValue(element_infos["value"])

   415

                                  self.GetTextCtrlCallBackFunction(spinctrl, element_path),

   416

                                  spinctrl)

   418

                else:

   419

                    if element_infos["type"] == "boolean":

   420

                        checkbox = wx.CheckBox(self.ParamsEditor, size=wx.Size(17, 25))

   421

                        boxsizer.AddWindow(checkbox)

   422

                        if element_infos["value"] is not None:

   423

                            checkbox.SetValue(element_infos["value"])

   426

                                      checkbox)

   428

                    elif element_infos["type"] in ["unsignedLong", "long","integer"]:

   429

                        if element_infos["type"].startswith("unsigned"):

   430

                            scmin = 0

   431

                        else:

   432

                            scmin = -(2**31)

   433

                        scmax = 2**31-1

   435

                              size=wx.Size(300, -1), style=wx.SP_ARROW_KEYS|wx.ALIGN_RIGHT)

   436

                        spinctrl.SetRange(scmin, scmax)

   437

                        boxsizer.AddWindow(spinctrl)

   438

                        if element_infos["value"] is not None:

   439

                            spinctrl.SetValue(element_infos["value"])

   442

                                      spinctrl)

   444

                    else:

   445

                        choices = self.ParentWindow.GetConfigEntry(element_path, [""])

   449

                                                        element_path=element_path,

   450

                                                        size=wx.Size(300, -1))

   452

                        boxsizer.AddWindow(textctrl)

   453

                        if element_infos["value"] is not None:

   454

                            textctrl.ChangeValue(str(element_infos["value"]))

   455

                        callback = self.GetTextCtrlCallBackFunction(textctrl, element_path)

   456

                        textctrl.Bind(wx.EVT_TEXT_ENTER, callback)

   457

                        textctrl.Bind(wx.EVT_KILL_FOCUS, callback)

   458

            first = False

   461

    def GetItemChannelChangedFunction(self, dir):

   462

        def OnConfNodeTreeItemChannelChanged(event):

   463

            confnode_IECChannel = self.Controler.BaseParams.getIEC_Channel()

   464

            res = self.SetConfNodeParamsAttribute("BaseParams.IEC_Channel", confnode_IECChannel + dir)

   465

            wx.CallAfter(self.RefreshIECChannelControlsState)

   466

            wx.CallAfter(self.ParentWindow._Refresh, TITLE, FILEMENU, PROJECTTREE)

   467

            event.Skip()

   468

        return OnConfNodeTreeItemChannelChanged

   470

    def SetConfNodeParamsAttribute(self, *args, **kwargs):

   471

        res, StructChanged = self.Controler.SetParamsAttribute(*args, **kwargs)

   472

        if StructChanged and self.ParamsEditor is not None:

   473

            wx.CallAfter(self.RefreshConfNodeParamsSizer)

   474

        wx.CallAfter(self.ParentWindow._Refresh, TITLE, FILEMENU)

   475

        return res

   477

    def GetButtonCallBackFunction(self, method, push=False):

   478

        """ Generate the callbackfunc for a given confnode method"""

   479

        def OnButtonClick(event):

   481

            event.GetEventObject().Disable()

   482

            # Call

   483

            getattr(self.Controler,method)()

   485

            event.GetEventObject().Enable()

   487

            if not push:

   488

                event.Skip()

   489

        return OnButtonClick

   491

    def GetChoiceCallBackFunction(self, choicectrl, path):

   492

        def OnChoiceChanged(event):

   493

            res = self.SetConfNodeParamsAttribute(path, choicectrl.GetStringSelection())

   494

            if res is None:

   495

                res = ""

   496