1

     2

import os, sys

     3

import cPickle

     4

from types import TupleType

     5

     6

import wx, wx.grid

     7

import wx.aui

     8

     9

try:

    10

    import matplotlib

    11

    matplotlib.use('WX')

    12

    USE_MPL = True

    13

except:

    14

    USE_MPL = False

    15

    16

from editors.EditorPanel import EditorPanel

    17

from editors.SFCViewer import SFC_Viewer

    18

from editors.LDViewer import LD_Viewer

    19

from editors.TextViewer import TextViewer

    20

from editors.Viewer import Viewer, ZOOM_FACTORS

    21

from editors.GraphicViewer import GraphicViewer

    22

from editors.ResourceEditor import ConfigurationEditor, ResourceEditor

    23

from editors.DataTypeEditor import DataTypeEditor

    24

from PLCControler import *

    25

from controls import CustomTree, LibraryPanel, PouInstanceVariablesPanel, DebugVariablePanel, SearchResultPanel

    26

from dialogs import ProjectDialog, PouDialog, PouTransitionDialog, PouActionDialog, FindInPouDialog, SearchInProjectDialog

    27

from util.BitmapLibrary import GetBitmap

    28

    29

# Define PLCOpenEditor controls id

    30

[ID_PLCOPENEDITOR, ID_PLCOPENEDITORLEFTNOTEBOOK, 

    31

 ID_PLCOPENEDITORBOTTOMNOTEBOOK, ID_PLCOPENEDITORRIGHTNOTEBOOK, 

    32

 ID_PLCOPENEDITORPROJECTTREE, ID_PLCOPENEDITORMAINSPLITTER, 

    33

 ID_PLCOPENEDITORSECONDSPLITTER, ID_PLCOPENEDITORTHIRDSPLITTER, 

    34

 ID_PLCOPENEDITORLIBRARYPANEL, ID_PLCOPENEDITORLIBRARYSEARCHCTRL, 

    35

 ID_PLCOPENEDITORLIBRARYTREE, ID_PLCOPENEDITORLIBRARYCOMMENT, 

    36

 ID_PLCOPENEDITORTABSOPENED, ID_PLCOPENEDITORTABSOPENED, 

    37

 ID_PLCOPENEDITOREDITORMENUTOOLBAR, ID_PLCOPENEDITOREDITORTOOLBAR, 

    38

 ID_PLCOPENEDITORPROJECTPANEL, 

    39

] = [wx.NewId() for _init_ctrls in range(17)]

    40

    41

# Define PLCOpenEditor EditMenu extra items id

    42

[ID_PLCOPENEDITOREDITMENUENABLEUNDOREDO, ID_PLCOPENEDITOREDITMENUADDDATATYPE, 

    43

 ID_PLCOPENEDITOREDITMENUADDFUNCTION, ID_PLCOPENEDITOREDITMENUADDFUNCTIONBLOCK, 

    44

 ID_PLCOPENEDITOREDITMENUADDPROGRAM, ID_PLCOPENEDITOREDITMENUADDCONFIGURATION,

    45

 ID_PLCOPENEDITOREDITMENUFINDNEXT, ID_PLCOPENEDITOREDITMENUFINDPREVIOUS,

    46

 ID_PLCOPENEDITOREDITMENUSEARCHINPROJECT, 

    47

] = [wx.NewId() for _init_coll_EditMenu_Items in range(9)]

    48

    49

# Define PLCOpenEditor DisplayMenu extra items id

    50

[ID_PLCOPENEDITORDISPLAYMENURESETPERSPECTIVE, 

    51

] = [wx.NewId() for _init_coll_DisplayMenu_Items in range(1)]

    52

    53

#-------------------------------------------------------------------------------

    54

#                            EditorToolBar definitions

    55

#-------------------------------------------------------------------------------

    56

    57

# Define PLCOpenEditor Toolbar items id

    58

[ID_PLCOPENEDITOREDITORTOOLBARSELECTION, ID_PLCOPENEDITOREDITORTOOLBARCOMMENT,

    59

 ID_PLCOPENEDITOREDITORTOOLBARVARIABLE, ID_PLCOPENEDITOREDITORTOOLBARBLOCK,

    60

 ID_PLCOPENEDITOREDITORTOOLBARCONNECTION, ID_PLCOPENEDITOREDITORTOOLBARWIRE,

    61

 ID_PLCOPENEDITOREDITORTOOLBARPOWERRAIL, ID_PLCOPENEDITOREDITORTOOLBARRUNG,

    62

 ID_PLCOPENEDITOREDITORTOOLBARCOIL, ID_PLCOPENEDITOREDITORTOOLBARCONTACT,

    63

 ID_PLCOPENEDITOREDITORTOOLBARBRANCH, ID_PLCOPENEDITOREDITORTOOLBARINITIALSTEP,

    64

 ID_PLCOPENEDITOREDITORTOOLBARSTEP, ID_PLCOPENEDITOREDITORTOOLBARTRANSITION,

    65

 ID_PLCOPENEDITOREDITORTOOLBARACTIONBLOCK, ID_PLCOPENEDITOREDITORTOOLBARDIVERGENCE,

    66

 ID_PLCOPENEDITOREDITORTOOLBARJUMP, ID_PLCOPENEDITOREDITORTOOLBARMOTION, 

    67

] = [wx.NewId() for _init_coll_DefaultEditorToolBar_Items in range(18)]

    68

    69

    70

    71

# Define behaviour of each Toolbar item according to current POU body type 

    72

# Informations meaning are in this order:

    73

#  - Item is toggled

    74

#  - PLCOpenEditor mode where item is displayed (could be more then one)

    75

#  - Item id

    76

#  - Item callback function name

    77

#  - Item icon filename

    78

#  - Item tooltip text

    79

EditorToolBarItems = {

    80

    "FBD" : [(True, FREEDRAWING_MODE|DRIVENDRAWING_MODE,

    81

              ID_PLCOPENEDITOREDITORTOOLBARMOTION, "OnMotionTool",

    82

              "move", _("Move the view")),

    83

             (True, FREEDRAWING_MODE|DRIVENDRAWING_MODE,

    84

              ID_PLCOPENEDITOREDITORTOOLBARCOMMENT, "OnCommentTool",

    85

              "add_comment", _("Create a new comment")),

    86

             (True, FREEDRAWING_MODE|DRIVENDRAWING_MODE,

    87

              ID_PLCOPENEDITOREDITORTOOLBARVARIABLE, "OnVariableTool",

    88

              "add_variable", _("Create a new variable")),

    89

             (True, FREEDRAWING_MODE|DRIVENDRAWING_MODE,

    90

              ID_PLCOPENEDITOREDITORTOOLBARBLOCK, "OnBlockTool",

    91

              "add_block", _("Create a new block")),

    92

             (True, FREEDRAWING_MODE|DRIVENDRAWING_MODE, 

    93

              ID_PLCOPENEDITOREDITORTOOLBARCONNECTION, "OnConnectionTool", 

    94

              "add_connection", _("Create a new connection"))],

    95

    "LD"  : [(True, FREEDRAWING_MODE|DRIVENDRAWING_MODE,

    96

              ID_PLCOPENEDITOREDITORTOOLBARMOTION, "OnMotionTool",

    97

              "move", _("Move the view")),

    98

             (True, FREEDRAWING_MODE, 

    99

              ID_PLCOPENEDITOREDITORTOOLBARCOMMENT, "OnCommentTool", 

   100

              "add_comment", _("Create a new comment")),

   101

             (True, FREEDRAWING_MODE, 

   102

              ID_PLCOPENEDITOREDITORTOOLBARPOWERRAIL, "OnPowerRailTool", 

   103

              "add_powerrail", _("Create a new power rail")),

   104

             (False, DRIVENDRAWING_MODE, 

   105

              ID_PLCOPENEDITOREDITORTOOLBARRUNG, "OnRungTool", 

   106

              "add_rung", _("Create a new rung")),

   107

             (True, FREEDRAWING_MODE, 

   108

              ID_PLCOPENEDITOREDITORTOOLBARCOIL, "OnCoilTool", 

   109

              "add_coil", _("Create a new coil")),

   110

             (False, FREEDRAWING_MODE|DRIVENDRAWING_MODE, 

   111

              ID_PLCOPENEDITOREDITORTOOLBARCONTACT, "OnContactTool", 

   112

              "add_contact", _("Create a new contact")),

   113

             (False, DRIVENDRAWING_MODE, 

   114

              ID_PLCOPENEDITOREDITORTOOLBARBRANCH, "OnBranchTool", 

   115

              "add_branch", _("Create a new branch")),

   116

             (True, FREEDRAWING_MODE, 

   117

              ID_PLCOPENEDITOREDITORTOOLBARVARIABLE, "OnVariableTool", 

   118

              "add_variable", _("Create a new variable")),

   119

             (False, FREEDRAWING_MODE|DRIVENDRAWING_MODE, 

   120

              ID_PLCOPENEDITOREDITORTOOLBARBLOCK, "OnBlockTool", 

   121

              "add_block", _("Create a new block")),

   122

             (True, FREEDRAWING_MODE, 

   123

              ID_PLCOPENEDITOREDITORTOOLBARCONNECTION, "OnConnectionTool", 

   124

              "add_connection", _("Create a new connection"))],

   125

    "SFC" : [(True, FREEDRAWING_MODE|DRIVENDRAWING_MODE,

   126

              ID_PLCOPENEDITOREDITORTOOLBARMOTION, "OnMotionTool",

   127

              "move", _("Move the view")),

   128

             (True, FREEDRAWING_MODE|DRIVENDRAWING_MODE, 

   129

              ID_PLCOPENEDITOREDITORTOOLBARCOMMENT, "OnCommentTool", 

   130

              "add_comment", _("Create a new comment")),

   131

             (True, FREEDRAWING_MODE|DRIVENDRAWING_MODE, 

   132

              ID_PLCOPENEDITOREDITORTOOLBARINITIALSTEP, "OnInitialStepTool", 

   133

              "add_initial_step", _("Create a new initial step")),

   134

             (False, FREEDRAWING_MODE|DRIVENDRAWING_MODE, 

   135

              ID_PLCOPENEDITOREDITORTOOLBARSTEP, "OnStepTool", 

   136

              "add_step", _("Create a new step")),

   137

             (True, FREEDRAWING_MODE, 

   138

              ID_PLCOPENEDITOREDITORTOOLBARTRANSITION, "OnTransitionTool", 

   139

              "add_transition", _("Create a new transition")),

   140

             (False, FREEDRAWING_MODE|DRIVENDRAWING_MODE, 

   141

              ID_PLCOPENEDITOREDITORTOOLBARACTIONBLOCK, "OnActionBlockTool", 

   142

              "add_action", _("Create a new action block")),

   143

             (False, FREEDRAWING_MODE|DRIVENDRAWING_MODE, 

   144

              ID_PLCOPENEDITOREDITORTOOLBARDIVERGENCE, "OnDivergenceTool", 

   145

              "add_divergence", _("Create a new divergence")),

   146

             (False, FREEDRAWING_MODE|DRIVENDRAWING_MODE, 

   147

              ID_PLCOPENEDITOREDITORTOOLBARJUMP, "OnJumpTool", 

   148

              "add_jump", _("Create a new jump")),

   149

             (True, FREEDRAWING_MODE, 

   150

              ID_PLCOPENEDITOREDITORTOOLBARVARIABLE, "OnVariableTool", 

   151

              "add_variable", _("Create a new variable")),

   152

             (True, FREEDRAWING_MODE, 

   153

              ID_PLCOPENEDITOREDITORTOOLBARBLOCK, "OnBlockTool", 

   154

              "add_block", _("Create a new block")),

   155

             (True, FREEDRAWING_MODE, 

   156

              ID_PLCOPENEDITOREDITORTOOLBARCONNECTION, "OnConnectionTool", 

   157

              "add_connection", _("Create a new connection")),

   158

             (True, FREEDRAWING_MODE, 

   159

              ID_PLCOPENEDITOREDITORTOOLBARPOWERRAIL, "OnPowerRailTool", 

   160

              "add_powerrail", _("Create a new power rail")),

   161

             (True, FREEDRAWING_MODE, 

   162

              ID_PLCOPENEDITOREDITORTOOLBARCONTACT, "OnContactTool", 

   163

              "add_contact", _("Create a new contact"))],

   164

    "ST"  : [],

   165

    "IL"  : [],

   166

    "debug": [(True, FREEDRAWING_MODE|DRIVENDRAWING_MODE,

   167

              ID_PLCOPENEDITOREDITORTOOLBARMOTION, "OnMotionTool",

   168

              "move", _("Move the view"))],

   169

}

   170

   171

#-------------------------------------------------------------------------------

   172

#                               Helper Functions

   173

#-------------------------------------------------------------------------------

   174

   175

import base64

   176

   177

def EncodeFileSystemPath(path, use_base64=True):

   178

    path = path.encode(sys.getfilesystemencoding())

   179

    if use_base64:

   180

        return base64.encodestring(path)

   181

    return path

   182

   183

def DecodeFileSystemPath(path, is_base64=True):

   184

    if is_base64:

   185

        path = base64.decodestring(path)

   186

    return unicode(path, sys.getfilesystemencoding())

   187

   188

# Compatibility function for wx versions < 2.6

   189

def AppendMenu(parent, help, id, kind, text):

   190

    if wx.VERSION >= (2, 6, 0):

   191

        parent.Append(help=help, id=id, kind=kind, text=text)

   192

    else:

   193

        parent.Append(helpString=help, id=id, kind=kind, item=text)

   194

   195

[TITLE, EDITORTOOLBAR, FILEMENU, EDITMENU, DISPLAYMENU, PROJECTTREE, 

   196

 POUINSTANCEVARIABLESPANEL, LIBRARYTREE, SCALING, PAGETITLES

   197

] = range(10)

   198

   199

def GetShortcutKeyCallbackFunction(viewer_function):

   200

    def ShortcutKeyFunction(self, event):

   201

        control = self.FindFocus()

   202

        if control is not None and control.GetName() in ["Viewer", "TextViewer"]:

   203

            getattr(control.ParentWindow, viewer_function)()

   204

        elif isinstance(control, wx.stc.StyledTextCtrl):

   205

            getattr(control, viewer_function)()

   206

        elif isinstance(control, wx.TextCtrl):

   207

            control.ProcessEvent(event)

   208

    return ShortcutKeyFunction

   209

   210

def GetDeleteElementFunction(remove_function, parent_type=None, check_function=None):

   211

    def DeleteElementFunction(self, selected):

   212

        name = self.ProjectTree.GetItemText(selected)

   213

        if check_function is None or check_function(name):

   214

            if parent_type is not None:

   215

                item_infos = self.ProjectTree.GetPyData(selected)

   216

                parent_name = item_infos["tagname"].split("::")[1]

   217

                remove_function(self.Controler, parent_name, name)

   218

            else:

   219

                remove_function(self.Controler, name)

   220

    return DeleteElementFunction

   221

   222

if wx.Platform == '__WXMSW__':

   223

    TAB_BORDER = 6

   224

    NOTEBOOK_BORDER = 6

   225

else:

   226

    TAB_BORDER = 7

   227

    NOTEBOOK_BORDER = 2

   228

   229

def SimplifyTabLayout(tabs, rect):

   230

    for tab in tabs:

   231

        if tab["pos"][0] == rect.x:

   232

            others = [t for t in tabs if t != tab]

   233

            others.sort(lambda x,y: cmp(x["pos"][0], y["pos"][0]))

   234

            for other in others:

   235

                if (other["pos"][1] == tab["pos"][1] and 

   236

                    other["size"][1] == tab["size"][1] and

   237

                    other["pos"][0] == tab["pos"][0] + tab["size"][0] + TAB_BORDER):

   238

   239

                    tab["size"] = (tab["size"][0] + other["size"][0] + TAB_BORDER, tab["size"][1])

   240

                    tab["pages"].extend(other["pages"])

   241

                    tabs.remove(other)

   242

   243

                    if tab["size"][0] == rect.width:

   244

                        return True

   245

   246

        elif tab["pos"][1] == rect.y:

   247

            others = [t for t in tabs if t != tab]

   248

            others.sort(lambda x,y: cmp(x["pos"][1], y["pos"][1]))

   249

            for other in others:

   250

                if (other["pos"][0] == tab["pos"][0] and 

   251

                    other["size"][0] == tab["size"][0] and

   252

                    other["pos"][1] == tab["pos"][1] + tab["size"][1] + TAB_BORDER):

   253

   254

                    tab["size"] = (tab["size"][0], tab["size"][1] + other["size"][1] + TAB_BORDER)

   255

                    tab["pages"].extend(other["pages"])

   256

                    tabs.remove(other)

   257

   258

                    if tab["size"][1] == rect.height:

   259

                        return True

   260

    return False

   261

   262

def ComputeTabsLayout(tabs, rect):

   263

    if len(tabs) == 0:

   264

        return tabs

   265

    if len(tabs) == 1:

   266

        return tabs[0]

   267

    split = None

   268

    for idx, tab in enumerate(tabs):

   269

        if len(tab["pages"]) == 0:

   270

            raise ValueError, "Not possible"

   271

        if tab["size"][0] == rect.width:

   272

            if tab["pos"][1] == rect.y:

   273

                split = (wx.TOP, float(tab["size"][1]) / float(rect.height))

   274

                split_rect = wx.Rect(rect.x, rect.y + tab["size"][1] + TAB_BORDER, 

   275

                                     rect.width, rect.height - tab["size"][1] - TAB_BORDER)

   276

            elif tab["pos"][1] == rect.height + 1 - tab["size"][1]:

   277

                split = (wx.BOTTOM, 1.0 - float(tab["size"][1]) / float(rect.height))

   278

                split_rect = wx.Rect(rect.x, rect.y, 

   279

                                     rect.width, rect.height - tab["size"][1] - TAB_BORDER)

   280

            break

   281

        elif tab["size"][1] == rect.height:

   282

            if tab["pos"][0] == rect.x:

   283

                split = (wx.LEFT, float(tab["size"][0]) / float(rect.width))

   284

                split_rect = wx.Rect(rect.x + tab["size"][0] + TAB_BORDER, rect.y, 

   285

                                     rect.width - tab["size"][0] - TAB_BORDER, rect.height)

   286

            elif tab["pos"][0] == rect.width + 1 - tab["size"][0]:

   287

                split = (wx.RIGHT, 1.0 - float(tab["size"][0]) / float(rect.width))

   288

                split_rect = wx.Rect(rect.x, rect.y, 

   289

                                     rect.width - tab["size"][0] - TAB_BORDER, rect.height)

   290

            break

   291

    if split != None:

   292

        split_tab = tabs.pop(idx)

   293

        return {"split": split,

   294

                "tab": split_tab, 

   295

                "others": ComputeTabsLayout(tabs, split_rect)}

   296

    else:

   297

        if SimplifyTabLayout(tabs, rect):

   298

            return ComputeTabsLayout(tabs, rect)

   299

    return tabs

   300

   301

#-------------------------------------------------------------------------------

   302

#                              IDEFrame Base Class

   303

#-------------------------------------------------------------------------------

   304

   305

UNEDITABLE_NAMES_DICT = dict([(_(name), name) for name in UNEDITABLE_NAMES])

   306

   307

class IDEFrame(wx.Frame):

   308

   309

    # Compatibility function for wx versions < 2.6

   310

    if wx.VERSION < (2, 6, 0):

   311

        def Bind(self, event, function, id = None):

   312

            if id is not None:

   313

                event(self, id, function)

   314

            else:

   315

                event(self, function)

   316

   317

    def _init_coll_MenuBar_Menus(self, parent):

   318

        parent.Append(menu=self.FileMenu, title=_(u'&File'))

   319

        parent.Append(menu=self.EditMenu, title=_(u'&Edit'))

   320

        parent.Append(menu=self.DisplayMenu, title=_(u'&Display'))

   321

        parent.Append(menu=self.HelpMenu, title=_(u'&Help'))

   322

   323

    def _init_coll_FileMenu_Items(self, parent):

   324

        pass

   325

   326

    def _init_coll_AddMenu_Items(self, parent, add_config=True):

   327

        AppendMenu(parent, help='', id=ID_PLCOPENEDITOREDITMENUADDDATATYPE,

   328

              kind=wx.ITEM_NORMAL, text=_(u'&Data Type'))

   329

        AppendMenu(parent, help='', id=ID_PLCOPENEDITOREDITMENUADDFUNCTION,

   330

              kind=wx.ITEM_NORMAL, text=_(u'&Function'))

   331

        AppendMenu(parent, help='', id=ID_PLCOPENEDITOREDITMENUADDFUNCTIONBLOCK,

   332

              kind=wx.ITEM_NORMAL, text=_(u'Function &Block'))

   333

        AppendMenu(parent, help='', id=ID_PLCOPENEDITOREDITMENUADDPROGRAM,

   334

              kind=wx.ITEM_NORMAL, text=_(u'&Program'))

   335

        if add_config:

   336

            AppendMenu(parent, help='', id=ID_PLCOPENEDITOREDITMENUADDCONFIGURATION,

   337

                  kind=wx.ITEM_NORMAL, text=_(u'&Configuration'))

   338

   339

    def _init_coll_EditMenu_Items(self, parent):

   340

        AppendMenu(parent, help='', id=wx.ID_UNDO,

   341

              kind=wx.ITEM_NORMAL, text=_(u'Undo') + '\tCTRL+Z')

   342

        AppendMenu(parent, help='', id=wx.ID_REDO,

   343

              kind=wx.ITEM_NORMAL, text=_(u'Redo') + '\tCTRL+Y')

   344

        #AppendMenu(parent, help='', id=ID_PLCOPENEDITOREDITMENUENABLEUNDOREDO,

   345

        #      kind=wx.ITEM_CHECK, text=_(u'Enable Undo/Redo'))

   346

        enable_undo_redo = _(u'Enable Undo/Redo') # Keeping text in translations for possible menu reactivation

   347

        parent.AppendSeparator()

   348

        AppendMenu(parent, help='', id=wx.ID_CUT,

   349

              kind=wx.ITEM_NORMAL, text=_(u'Cut') + '\tCTRL+X')

   350

        AppendMenu(parent, help='', id=wx.ID_COPY,

   351

              kind=wx.ITEM_NORMAL, text=_(u'Copy') + '\tCTRL+C')

   352

        AppendMenu(parent, help='', id=wx.ID_PASTE,

   353

              kind=wx.ITEM_NORMAL, text=_(u'Paste') + '\tCTRL+V')

   354

        parent.AppendSeparator()

   355

        AppendMenu(parent, help='', id=wx.ID_FIND,

   356

              kind=wx.ITEM_NORMAL, text=_(u'Find') + '\tCTRL+F')

   357

        AppendMenu(parent, help='', id=ID_PLCOPENEDITOREDITMENUFINDNEXT,

   358

              kind=wx.ITEM_NORMAL, text=_(u'Find Next') + '\tCTRL+K')

   359

        AppendMenu(parent, help='', id=ID_PLCOPENEDITOREDITMENUFINDPREVIOUS,

   360

              kind=wx.ITEM_NORMAL, text=_(u'Find Previous') + '\tCTRL+SHIFT+K')

   361

        parent.AppendSeparator()

   362

        AppendMenu(parent, help='', id=ID_PLCOPENEDITOREDITMENUSEARCHINPROJECT,

   363

              kind=wx.ITEM_NORMAL, text=_(u'Search in Project') + '\tCTRL+SHIFT+F')

   364

        parent.AppendSeparator()

   365

        add_menu = wx.Menu(title='')

   366

        self._init_coll_AddMenu_Items(add_menu)

   367

        parent.AppendMenu(wx.ID_ADD, _(u"&Add Element"), add_menu)

   368

        AppendMenu(parent, help='', id=wx.ID_SELECTALL,

   369

              kind=wx.ITEM_NORMAL, text=_(u'Select All') + '\tCTRL+A')

   370

        AppendMenu(parent, help='', id=wx.ID_DELETE,

   371

              kind=wx.ITEM_NORMAL, text=_(u'&Delete'))

   372

        self.Bind(wx.EVT_MENU, self.OnUndoMenu, id=wx.ID_UNDO)

   373

        self.Bind(wx.EVT_MENU, self.OnRedoMenu, id=wx.ID_REDO)

   374

        #self.Bind(wx.EVT_MENU, self.OnEnableUndoRedoMenu, id=ID_PLCOPENEDITOREDITMENUENABLEUNDOREDO)

   375

        self.Bind(wx.EVT_MENU, self.OnCutMenu, id=wx.ID_CUT)

   376

        self.Bind(wx.EVT_MENU, self.OnCopyMenu, id=wx.ID_COPY)

   377

        self.Bind(wx.EVT_MENU, self.OnPasteMenu, id=wx.ID_PASTE)

   378

        self.Bind(wx.EVT_MENU, self.OnFindMenu, id=wx.ID_FIND)

   379

        self.Bind(wx.EVT_MENU, self.OnFindNextMenu, 

   380

              id=ID_PLCOPENEDITOREDITMENUFINDNEXT)

   381

        self.Bind(wx.EVT_MENU, self.OnFindPreviousMenu, 

   382

              id=ID_PLCOPENEDITOREDITMENUFINDPREVIOUS)

   383

        self.Bind(wx.EVT_MENU, self.OnSearchInProjectMenu, 

   384

              id=ID_PLCOPENEDITOREDITMENUSEARCHINPROJECT)

   385

        self.Bind(wx.EVT_MENU, self.OnSearchInProjectMenu, 

   386

              id=ID_PLCOPENEDITOREDITMENUSEARCHINPROJECT)

   387

        self.Bind(wx.EVT_MENU, self.OnAddDataTypeMenu,

   388

              id=ID_PLCOPENEDITOREDITMENUADDDATATYPE)

   389

        self.Bind(wx.EVT_MENU, self.GenerateAddPouFunction("function"),

   390

              id=ID_PLCOPENEDITOREDITMENUADDFUNCTION)

   391

        self.Bind(wx.EVT_MENU, self.GenerateAddPouFunction("functionBlock"),

   392

              id=ID_PLCOPENEDITOREDITMENUADDFUNCTIONBLOCK)

   393

        self.Bind(wx.EVT_MENU, self.GenerateAddPouFunction("program"),

   394

              id=ID_PLCOPENEDITOREDITMENUADDPROGRAM)

   395

        self.Bind(wx.EVT_MENU, self.OnAddConfigurationMenu,

   396

              id=ID_PLCOPENEDITOREDITMENUADDCONFIGURATION)

   397

        self.Bind(wx.EVT_MENU, self.OnSelectAllMenu, id=wx.ID_SELECTALL)

   398

        self.Bind(wx.EVT_MENU, self.OnDeleteMenu, id=wx.ID_DELETE)

   399

   400

        self.AddToMenuToolBar([(wx.ID_UNDO, "undo", _(u'Undo'), None),

   401

                               (wx.ID_REDO, "redo", _(u'Redo'), None),

   402

                               None,

   403

                               (wx.ID_CUT, "cut", _(u'Cut'), None),

   404

                               (wx.ID_COPY, "copy", _(u'Copy'), None),

   405

                               (wx.ID_PASTE, "paste", _(u'Paste'), None),

   406

                               None,

   407

                               (ID_PLCOPENEDITOREDITMENUSEARCHINPROJECT, "find", _(u'Search in Project'), None)])

   408

   409

    def _init_coll_DisplayMenu_Items(self, parent):

   410

        AppendMenu(parent, help='', id=wx.ID_REFRESH,

   411

              kind=wx.ITEM_NORMAL, text=_(u'Refresh') + '\tCTRL+R')

   412

        if self.EnableDebug:

   413

            AppendMenu(parent, help='', id=wx.ID_CLEAR,

   414

                  kind=wx.ITEM_NORMAL, text=_(u'Clear Errors') + '\tCTRL+K')

   415

        parent.AppendSeparator()

   416

        zoommenu = wx.Menu(title='')

   417

        parent.AppendMenu(wx.ID_ZOOM_FIT, _("Zoom"), zoommenu)

   418

        for idx, value in enumerate(ZOOM_FACTORS):

   419

            new_id = wx.NewId()

   420

            AppendMenu(zoommenu, help='', id=new_id,

   421

                  kind=wx.ITEM_RADIO, text=str(int(round(value * 100))) + "%")

   422

            self.Bind(wx.EVT_MENU, self.GenerateZoomFunction(idx), id=new_id)

   423

   424

        parent.AppendSeparator()

   425

        AppendMenu(parent, help='', id=ID_PLCOPENEDITORDISPLAYMENURESETPERSPECTIVE,

   426

              kind=wx.ITEM_NORMAL, text=_(u'Reset Perspective'))

   427

        self.Bind(wx.EVT_MENU, self.OnResetPerspective, id=ID_PLCOPENEDITORDISPLAYMENURESETPERSPECTIVE)

   428

   429

        self.Bind(wx.EVT_MENU, self.OnRefreshMenu, id=wx.ID_REFRESH)

   430

        if self.EnableDebug:

   431

            self.Bind(wx.EVT_MENU, self.OnClearErrorsMenu, id=wx.ID_CLEAR)

   432

   433

    def _init_coll_HelpMenu_Items(self, parent):

   434

        pass

   435

   436

    def _init_utils(self):

   437

        self.MenuBar = wx.MenuBar()

   438

   439

        self.FileMenu = wx.Menu(title='')

   440

        self.EditMenu = wx.Menu(title='')

   441

        self.DisplayMenu = wx.Menu(title='')

   442

        self.HelpMenu = wx.Menu(title='')

   443

   444

        self._init_coll_MenuBar_Menus(self.MenuBar)

   445

        self._init_coll_FileMenu_Items(self.FileMenu)

   446

        self._init_coll_EditMenu_Items(self.EditMenu)

   447

        self._init_coll_DisplayMenu_Items(self.DisplayMenu)

   448

        self._init_coll_HelpMenu_Items(self.HelpMenu)

   449

   450

    def _init_ctrls(self, prnt):

   451

        wx.Frame.__init__(self, id=ID_PLCOPENEDITOR, name='IDEFrame',

   452

              parent=prnt, pos=wx.DefaultPosition, size=wx.Size(1000, 600),

   453

              style=wx.DEFAULT_FRAME_STYLE)

   454

        self.SetClientSize(wx.Size(1000, 600))

   455

        self.Bind(wx.EVT_ACTIVATE, self.OnActivated)

   456

   457

        self.TabsImageList = wx.ImageList(31, 16)

   458

        self.TabsImageListIndexes = {}

   459

   460

        #-----------------------------------------------------------------------

   461

        #                          Creating main structure

   462

        #-----------------------------------------------------------------------

   463

   464

        self.AUIManager = wx.aui.AuiManager(self)

   465

        self.AUIManager.SetDockSizeConstraint(0.5, 0.5)

   466

        self.Panes = {}

   467

   468

        self.LeftNoteBook = wx.aui.AuiNotebook(self, ID_PLCOPENEDITORLEFTNOTEBOOK,

   469

              style=wx.aui.AUI_NB_TOP|wx.aui.AUI_NB_TAB_SPLIT|wx.aui.AUI_NB_TAB_MOVE|

   470

                    wx.aui.AUI_NB_SCROLL_BUTTONS|wx.aui.AUI_NB_TAB_EXTERNAL_MOVE)

   471

        self.LeftNoteBook.Bind(wx.aui.EVT_AUINOTEBOOK_ALLOW_DND, 

   472

                self.OnAllowNotebookDnD)

   473

        self.AUIManager.AddPane(self.LeftNoteBook, 

   474

              wx.aui.AuiPaneInfo().Name("ProjectPane").

   475

              Left().Layer(1).

   476

              BestSize(wx.Size(300, 500)).CloseButton(False))

   477

   478

        self.BottomNoteBook = wx.aui.AuiNotebook(self, ID_PLCOPENEDITORBOTTOMNOTEBOOK,

   479

              style=wx.aui.AUI_NB_TOP|wx.aui.AUI_NB_TAB_SPLIT|wx.aui.AUI_NB_TAB_MOVE|

   480

                    wx.aui.AUI_NB_SCROLL_BUTTONS|wx.aui.AUI_NB_TAB_EXTERNAL_MOVE)

   481

        self.BottomNoteBook.Bind(wx.aui.EVT_AUINOTEBOOK_ALLOW_DND, 

   482

                self.OnAllowNotebookDnD)

   483

        self.AUIManager.AddPane(self.BottomNoteBook, 

   484

              wx.aui.AuiPaneInfo().Name("ResultPane").

   485

              Bottom().Layer(0).

   486

              BestSize(wx.Size(800, 300)).CloseButton(False))

   487

   488

        self.RightNoteBook = wx.aui.AuiNotebook(self, ID_PLCOPENEDITORRIGHTNOTEBOOK,

   489

              style=wx.aui.AUI_NB_TOP|wx.aui.AUI_NB_TAB_SPLIT|wx.aui.AUI_NB_TAB_MOVE|

   490

                    wx.aui.AUI_NB_SCROLL_BUTTONS|wx.aui.AUI_NB_TAB_EXTERNAL_MOVE)

   491

        self.RightNoteBook.Bind(wx.aui.EVT_AUINOTEBOOK_ALLOW_DND, 

   492

                self.OnAllowNotebookDnD)

   493

        self.AUIManager.AddPane(self.RightNoteBook, 

   494

              wx.aui.AuiPaneInfo().Name("LibraryPane").

   495

              Right().Layer(0).

   496

              BestSize(wx.Size(250, 400)).CloseButton(False))

   497

   498

        self.TabsOpened = wx.aui.AuiNotebook(self, ID_PLCOPENEDITORTABSOPENED, 

   499

              style=wx.aui.AUI_NB_DEFAULT_STYLE|wx.aui.AUI_NB_WINDOWLIST_BUTTON)

   500

        self.TabsOpened.Bind(wx.aui.EVT_AUINOTEBOOK_PAGE_CHANGING,

   501

              self.OnPouSelectedChanging)

   502

        self.TabsOpened.Bind(wx.aui.EVT_AUINOTEBOOK_PAGE_CHANGED,

   503

              self.OnPouSelectedChanged)

   504

        self.TabsOpened.Bind(wx.aui.EVT_AUINOTEBOOK_PAGE_CLOSE,

   505

              self.OnPageClose)

   506

        self.TabsOpened.Bind(wx.aui.EVT_AUINOTEBOOK_END_DRAG,

   507

              self.OnPageDragged)

   508

        self.AUIManager.AddPane(self.TabsOpened, 

   509

              wx.aui.AuiPaneInfo().CentrePane().Name("TabsPane"))

   510

   511

        #-----------------------------------------------------------------------

   512

        #                    Creating PLCopen Project Types Tree

   513

        #-----------------------------------------------------------------------

   514

   515

        self.MainTabs = {}

   516

   517

        self.ProjectPanel = wx.SplitterWindow(id=ID_PLCOPENEDITORPROJECTPANEL,

   518

                  name='ProjectPanel', parent=self.LeftNoteBook, point=wx.Point(0, 0),

   519

                  size=wx.Size(0, 0), style=wx.SP_3D)

   520

   521

        self.ProjectTree = CustomTree(id=ID_PLCOPENEDITORPROJECTTREE,

   522

                  name='ProjectTree', parent=self.ProjectPanel, 

   523

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

   524

                  style=wx.TR_HAS_BUTTONS|wx.TR_SINGLE|wx.SUNKEN_BORDER|wx.TR_EDIT_LABELS)

   525

        self.ProjectTree.SetBackgroundBitmap(GetBitmap("custom_tree_background"),

   526

                                             wx.ALIGN_RIGHT|wx.ALIGN_BOTTOM)

   527

        add_menu = wx.Menu()

   528

        self._init_coll_AddMenu_Items(add_menu)

   529