1

#!/usr/bin/env python

     2

# -*- coding: utf-8 -*-

     3

     4

# This file is part of Beremiz, a Integrated Development Environment for

     5

# programming IEC 61131-3 automates supporting plcopen standard and CanFestival.

     6

#

     7

# Copyright (C) 2007: Edouard TISSERANT and Laurent BESSARD

     8

# Copyright (C) 2017: Andrey Skvortsov

     9

#

    10

# See COPYING file for copyrights details.

    11

#

    12

# This program is free software; you can redistribute it and/or

    13

# modify it under the terms of the GNU General Public License

    14

# as published by the Free Software Foundation; either version 2

    15

# of the License, or (at your option) any later version.

    16

#

    17

# This program is distributed in the hope that it will be useful,

    18

# but WITHOUT ANY WARRANTY; without even the implied warranty of

    19

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

    20

# GNU General Public License for more details.

    21

#

    22

# You should have received a copy of the GNU General Public License

    23

# along with this program; if not, write to the Free Software

    24

# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.

    25

    26

    27

    28

    29

import os

    30

import sys

    31

import getopt

    32

    33

import wx

    34

import wx.adv

    35

    36

import version

    37

import util.paths as paths

    38

import util.ExceptionHandler

    39

from util.misc import InstallLocalRessources

    40

from docutil.docpdf import open_pdf

    41

from IDEFrame import IDEFrame, AppendMenu

    42

from IDEFrame import \

    43

    TITLE, \

    44

    EDITORTOOLBAR, \

    45

    FILEMENU, \

    46

    EDITMENU, \

    47

    DISPLAYMENU, \

    48

    PROJECTTREE, \

    49

    POUINSTANCEVARIABLESPANEL, \

    50

    LIBRARYTREE, \

    51

    PAGETITLES, \

    52

    DecodeFileSystemPath

    53

from editors.Viewer import Viewer

    54

from PLCControler import PLCControler

    55

from dialogs import ProjectDialog

    56

from dialogs.AboutDialog import ShowAboutDialog

    57

    58

    59

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

    60

#                            PLCOpenEditor Main Class

    61

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

    62

    63

# Define PLCOpenEditor FileMenu extra items id

    64

[

    65

    ID_PLCOPENEDITORFILEMENUGENERATE,

    66

    ID_PLCOPENEDITORFILEMENUGENERATEAS,

    67

] = [wx.NewIdRef() for _init_coll_FileMenu_Items in range(2)]

    68

    69

    70

beremiz_dir = paths.AbsDir(__file__)

    71

    72

    73

class PLCOpenEditor(IDEFrame):

    74

    75

    def _init_coll_FileMenu_Items(self, parent):

    76

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

    77

                   kind=wx.ITEM_NORMAL, text=_('New') + '\tCTRL+N')

    78

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

    79

                   kind=wx.ITEM_NORMAL, text=_('Open') + '\tCTRL+O')

    80

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

    81

                   kind=wx.ITEM_NORMAL, text=_('Close Tab') + '\tCTRL+W')

    82

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

    83

                   kind=wx.ITEM_NORMAL, text=_('Close Project') + '\tCTRL+SHIFT+W')

    84

        parent.AppendSeparator()

    85

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

    86

                   kind=wx.ITEM_NORMAL, text=_('Save') + '\tCTRL+S')

    87

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

    88

                   kind=wx.ITEM_NORMAL, text=_('Save As...') + '\tCTRL+SHIFT+S')

    89

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

    90

                   kind=wx.ITEM_NORMAL, text=_('Generate Program') + '\tCTRL+G')

    91

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

    92

                   kind=wx.ITEM_NORMAL, text=_('Generate Program As...') + '\tCTRL+SHIFT+G')

    93

        parent.AppendSeparator()

    94

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

    95

                   kind=wx.ITEM_NORMAL, text=_('Page Setup') + '\tCTRL+ALT+P')

    96

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

    97

                   kind=wx.ITEM_NORMAL, text=_('Preview') + '\tCTRL+SHIFT+P')

    98

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

    99

                   kind=wx.ITEM_NORMAL, text=_('Print') + '\tCTRL+P')

   100

        parent.AppendSeparator()

   101

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

   102

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

   103

        parent.AppendSeparator()

   104

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

   105

                   kind=wx.ITEM_NORMAL, text=_('Quit') + '\tCTRL+Q')

   106

   107

        self.Bind(wx.EVT_MENU, self.OnNewProjectMenu, id=wx.ID_NEW)

   108

        self.Bind(wx.EVT_MENU, self.OnOpenProjectMenu, id=wx.ID_OPEN)

   109

        self.Bind(wx.EVT_MENU, self.OnCloseTabMenu, id=wx.ID_CLOSE)

   110

        self.Bind(wx.EVT_MENU, self.OnCloseProjectMenu, id=wx.ID_CLOSE_ALL)

   111

        self.Bind(wx.EVT_MENU, self.OnSaveProjectMenu, id=wx.ID_SAVE)

   112

        self.Bind(wx.EVT_MENU, self.OnSaveProjectAsMenu, id=wx.ID_SAVEAS)

   113

        self.Bind(wx.EVT_MENU, self.OnGenerateProgramMenu,

   114

                  id=ID_PLCOPENEDITORFILEMENUGENERATE)

   115

        self.Bind(wx.EVT_MENU, self.OnGenerateProgramAsMenu,

   116

                  id=ID_PLCOPENEDITORFILEMENUGENERATEAS)

   117

        self.Bind(wx.EVT_MENU, self.OnPageSetupMenu, id=wx.ID_PAGE_SETUP)

   118

        self.Bind(wx.EVT_MENU, self.OnPreviewMenu, id=wx.ID_PREVIEW)

   119

        self.Bind(wx.EVT_MENU, self.OnPrintMenu, id=wx.ID_PRINT)

   120

        self.Bind(wx.EVT_MENU, self.OnPropertiesMenu, id=wx.ID_PROPERTIES)

   121

        self.Bind(wx.EVT_MENU, self.OnQuitMenu, id=wx.ID_EXIT)

   122

   123

        self.AddToMenuToolBar([(wx.ID_NEW, "new", _('New'), None),

   124

                               (wx.ID_OPEN, "open", _('Open'), None),

   125

                               (wx.ID_SAVE, "save", _('Save'), None),

   126

                               (wx.ID_SAVEAS, "saveas", _('Save As...'), None),

   127

                               (wx.ID_PRINT, "print", _('Print'), None),

   128

                               (ID_PLCOPENEDITORFILEMENUGENERATE, "Build", _('Generate Program'), None)])

   129

   130

    def _init_coll_HelpMenu_Items(self, parent):

   131

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

   132

                   kind=wx.ITEM_NORMAL, text=_('PLCOpenEditor') + '\tF1')

   133

        # AppendMenu(parent, help='', id=wx.ID_HELP_CONTENTS,

   134

        #      kind=wx.ITEM_NORMAL, text=u'PLCOpen\tF2')

   135

        # AppendMenu(parent, help='', id=wx.ID_HELP_CONTEXT,

   136

        #      kind=wx.ITEM_NORMAL, text=u'IEC 61131-3\tF3')

   137

   138

        def handler(event):

   139

            return wx.MessageBox(

   140

                version.GetCommunityHelpMsg(),

   141

                _('Community support'),

   142

                wx.OK | wx.ICON_INFORMATION)

   143

   144

        menu_entry = parent.Append(help='', id=wx.ID_ANY, kind=wx.ITEM_NORMAL, text=_('Community support'))

   145

        self.Bind(wx.EVT_MENU, handler, menu_entry)

   146

   147

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

   148

                   kind=wx.ITEM_NORMAL, text=_('About'))

   149

        self.Bind(wx.EVT_MENU, self.OnPLCOpenEditorMenu, id=wx.ID_HELP)

   150

        # self.Bind(wx.EVT_MENU, self.OnPLCOpenMenu, id=wx.ID_HELP_CONTENTS)

   151

        self.Bind(wx.EVT_MENU, self.OnAboutMenu, id=wx.ID_ABOUT)

   152

   153

    def __init__(self, parent, fileOpen=None):

   154

        """ Constructor of the PLCOpenEditor class.

   155

   156

        :param parent: The parent window.

   157

        :param fileOpen: The filepath to open if no controler defined (default: None).

   158

        """

   159

        self.icon = wx.Icon(os.path.join(beremiz_dir, "images", "poe.ico"), wx.BITMAP_TYPE_ICO)

   160

        IDEFrame.__init__(self, parent)

   161

   162

        result = None

   163

   164

        # Open the filepath if defined

   165

        if fileOpen is not None:

   166

            fileOpen = DecodeFileSystemPath(fileOpen, False)

   167

            if os.path.isfile(fileOpen):

   168

                # Create a new controller

   169

                controler = PLCControler()

   170

                result = controler.OpenXMLFile(fileOpen)

   171

                self.Controler = controler

   172

                self.LibraryPanel.SetController(controler)

   173

                self.ProjectTree.Enable(True)

   174

                self.PouInstanceVariablesPanel.SetController(controler)

   175

                self._Refresh(PROJECTTREE, POUINSTANCEVARIABLESPANEL, LIBRARYTREE)

   176

   177

        # Define PLCOpenEditor icon

   178

        self.SetIcon(self.icon)

   179

   180

        self.Bind(wx.EVT_CLOSE, self.OnCloseFrame)

   181

   182

        self._Refresh(TITLE, EDITORTOOLBAR, FILEMENU, EDITMENU, DISPLAYMENU)

   183

   184

        if result is not None:

   185

            (num, line) = result

   186

            self.ShowErrorMessage(_("PLC syntax error at line {a1}:\n{a2}").format(a1=num, a2=line))

   187

   188

    def OnCloseFrame(self, event):

   189

        if self.Controler is None or self.CheckSaveBeforeClosing(_("Close Application")):

   190

            self.AUIManager.UnInit()

   191

   192

            self.SaveLastState()

   193

   194

            event.Skip()

   195

        else:

   196

            event.Veto()

   197

   198

    def RefreshTitle(self):

   199

        name = _("PLCOpenEditor")

   200

        if self.Controler is not None:

   201

            self.SetTitle("%s - %s" % (name, self.Controler.GetFilename()))

   202

        else:

   203

            self.SetTitle(name)

   204

   205

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

   206

    #                            File Menu Functions

   207

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

   208

   209

    def RefreshFileMenu(self):

   210

        MenuToolBar = self.Panes["MenuToolBar"]

   211

        if self.Controler is not None:

   212

            selected = self.TabsOpened.GetSelection()

   213

            if selected >= 0:

   214

                graphic_viewer = isinstance(self.TabsOpened.GetPage(selected), Viewer)

   215

            else:

   216

                graphic_viewer = False

   217

            if self.TabsOpened.GetPageCount() > 0:

   218

                self.FileMenu.Enable(wx.ID_CLOSE, True)

   219

                if graphic_viewer:

   220

                    self.FileMenu.Enable(wx.ID_PREVIEW, True)

   221

                    self.FileMenu.Enable(wx.ID_PRINT, True)

   222

                    MenuToolBar.EnableTool(wx.ID_PRINT, True)

   223

                else:

   224

                    self.FileMenu.Enable(wx.ID_PREVIEW, False)

   225

                    self.FileMenu.Enable(wx.ID_PRINT, False)

   226

                    MenuToolBar.EnableTool(wx.ID_PRINT, False)

   227

            else:

   228

                self.FileMenu.Enable(wx.ID_CLOSE, False)

   229

                self.FileMenu.Enable(wx.ID_PREVIEW, False)

   230

                self.FileMenu.Enable(wx.ID_PRINT, False)

   231

                MenuToolBar.EnableTool(wx.ID_PRINT, False)

   232

            self.FileMenu.Enable(wx.ID_PAGE_SETUP, True)

   233

            project_modified = not self.Controler.ProjectIsSaved()

   234

            self.FileMenu.Enable(wx.ID_SAVE, project_modified)

   235

            MenuToolBar.EnableTool(wx.ID_SAVE, project_modified)

   236

            self.FileMenu.Enable(wx.ID_PROPERTIES, True)

   237

            self.FileMenu.Enable(wx.ID_CLOSE_ALL, True)

   238

            self.FileMenu.Enable(wx.ID_SAVEAS, True)

   239

            MenuToolBar.EnableTool(wx.ID_SAVEAS, True)

   240

            self.FileMenu.Enable(ID_PLCOPENEDITORFILEMENUGENERATE, True)

   241

            MenuToolBar.EnableTool(ID_PLCOPENEDITORFILEMENUGENERATE, True)

   242

            self.FileMenu.Enable(ID_PLCOPENEDITORFILEMENUGENERATEAS, True)

   243

        else:

   244

            self.FileMenu.Enable(wx.ID_CLOSE, False)

   245

            self.FileMenu.Enable(wx.ID_PAGE_SETUP, False)

   246

            self.FileMenu.Enable(wx.ID_PREVIEW, False)

   247

            self.FileMenu.Enable(wx.ID_PRINT, False)

   248

            MenuToolBar.EnableTool(wx.ID_PRINT, False)

   249

            self.FileMenu.Enable(wx.ID_SAVE, False)

   250

            MenuToolBar.EnableTool(wx.ID_SAVE, False)

   251

            self.FileMenu.Enable(wx.ID_PROPERTIES, False)

   252

            self.FileMenu.Enable(wx.ID_CLOSE_ALL, False)

   253

            self.FileMenu.Enable(wx.ID_SAVEAS, False)

   254

            MenuToolBar.EnableTool(wx.ID_SAVEAS, False)

   255

            self.FileMenu.Enable(ID_PLCOPENEDITORFILEMENUGENERATE, False)

   256

            MenuToolBar.EnableTool(ID_PLCOPENEDITORFILEMENUGENERATE, False)

   257

            self.FileMenu.Enable(ID_PLCOPENEDITORFILEMENUGENERATEAS, False)

   258

   259

    def OnNewProjectMenu(self, event):

   260

        if self.Controler is not None and not self.CheckSaveBeforeClosing():

   261

            return

   262

        dialog = ProjectDialog(self)

   263

        if dialog.ShowModal() == wx.ID_OK:

   264

            properties = dialog.GetValues()

   265

            self.ResetView()

   266

            self.Controler = PLCControler()

   267

            self.Controler.CreateNewProject(properties)

   268

            self.LibraryPanel.SetController(self.Controler)

   269

            self.ProjectTree.Enable(True)

   270

            self._Refresh(TITLE, FILEMENU, EDITMENU, PROJECTTREE, POUINSTANCEVARIABLESPANEL,

   271

                          LIBRARYTREE)

   272

   273

    def OnOpenProjectMenu(self, event):

   274

        if self.Controler is not None and not self.CheckSaveBeforeClosing():

   275

            return

   276

        filepath = ""

   277

        if self.Controler is not None:

   278

            filepath = self.Controler.GetFilePath()

   279

        if filepath != "":

   280

            directory = os.path.dirname(filepath)

   281

        else:

   282

            directory = os.getcwd()

   283

   284

        result = None

   285

   286

        dialog = wx.FileDialog(self, _("Choose a file"), directory, "",  _("PLCOpen files (*.xml)|*.xml|All files|*.*"), wx.OPEN)

   287

        if dialog.ShowModal() == wx.ID_OK:

   288

            filepath = dialog.GetPath()

   289

            if os.path.isfile(filepath):

   290

                self.ResetView()

   291

                controler = PLCControler()

   292

                result = controler.OpenXMLFile(filepath)

   293

                self.Controler = controler

   294

                self.LibraryPanel.SetController(controler)

   295

                self.ProjectTree.Enable(True)

   296

                self.PouInstanceVariablesPanel.SetController(controler)

   297

                self._Refresh(PROJECTTREE, LIBRARYTREE)

   298

            self._Refresh(TITLE, EDITORTOOLBAR, FILEMENU, EDITMENU)

   299

        dialog.Destroy()

   300

   301

        if result is not None:

   302

            (num, line) = result

   303

            self.ShowErrorMessage(_("PLC syntax error at line {a1}:\n{a2}").format(a1=num, a2=line))

   304

   305

    def OnCloseProjectMenu(self, event):

   306

        if not self.CheckSaveBeforeClosing():

   307

            return

   308

        self.ResetView()

   309

        self._Refresh(TITLE, EDITORTOOLBAR, FILEMENU, EDITMENU)

   310

   311

    def OnSaveProjectMenu(self, event):

   312

        self.SaveProject()

   313

   314

    def OnSaveProjectAsMenu(self, event):

   315

        self.SaveProjectAs()

   316

   317

    def OnGenerateProgramMenu(self, event):

   318

        result = self.Controler.GetProgramFilePath()

   319

        if not result:

   320

            self.GenerateProgramAs()

   321

        else:

   322

            self.GenerateProgram(result)

   323

   324

    def OnGenerateProgramAsMenu(self, event):

   325

        self.GenerateProgramAs()

   326

   327

    def GenerateProgramAs(self):

   328

        dialog = wx.FileDialog(self, _("Choose a file"), os.getcwd(), os.path.basename(self.Controler.GetProgramFilePath()),  _("ST files (*.st)|*.st|All files|*.*"), wx.SAVE | wx.CHANGE_DIR)

   329

        if dialog.ShowModal() == wx.ID_OK:

   330

            self.GenerateProgram(dialog.GetPath())

   331

        dialog.Destroy()

   332

   333

    def GenerateProgram(self, filepath=None):

   334

        message_text = ""

   335

        header, icon = _("Done"), wx.ICON_INFORMATION

   336

        if os.path.isdir(os.path.dirname(filepath)):

   337

            _program, errors, warnings = self.Controler.GenerateProgram(filepath)

   338

            message_text += "".join([_("warning: %s\n") % warning for warning in warnings])

   339

            if len(errors) > 0:

   340

                message_text += "".join([_("error: %s\n") % error for error in errors])

   341

                message_text += _("Can't generate program to file %s!") % filepath

   342

                header, icon = _("Error"), wx.ICON_ERROR

   343

            else:

   344

                message_text += _("Program was successfully generated!")

   345

        else:

   346

            message_text += _("\"%s\" is not a valid folder!") % os.path.dirname(filepath)

   347

            header, icon = _("Error"), wx.ICON_ERROR

   348

        message = wx.MessageDialog(self, message_text, header, wx.OK | icon)

   349

        message.ShowModal()

   350

        message.Destroy()

   351

   352

    def OnPLCOpenEditorMenu(self, event):

   353

        wx.MessageBox(_("No documentation available.\nComing soon."))

   354

   355

    def OnPLCOpenMenu(self, event):

   356

        open_pdf(os.path.join(beremiz_dir, "plcopen", "TC6_XML_V101.pdf"))

   357

   358

    def OnAboutMenu(self, event):

   359

        info = wx.adv.AboutDialogInfo()

   360

        info = version.GetAboutDialogInfo(info)

   361

        info.Name = "PLCOpenEditor"

   362

        info.Description = _("PLCOpenEditor is part of Beremiz project.\n\n"

   363

                             "Beremiz is an ") + info.Description

   364

        info.Icon = wx.Icon(os.path.join(beremiz_dir, "images", "aboutlogo.png"), wx.BITMAP_TYPE_PNG)

   365

        ShowAboutDialog(self, info)

   366

   367

    def SaveProject(self):

   368

        result = self.Controler.SaveXMLFile()

   369

        if not result:

   370

            self.SaveProjectAs()

   371

        else:

   372

            self._Refresh(TITLE, FILEMENU, PAGETITLES)

   373

   374

    def SaveProjectAs(self):

   375

        filepath = self.Controler.GetFilePath()

   376

        if filepath != "":

   377

            directory, filename = os.path.split(filepath)

   378

        else:

   379

            directory, filename = os.getcwd(), "%(projectName)s.xml" % self.Controler.GetProjectProperties()

   380

        dialog = wx.FileDialog(self, _("Choose a file"), directory, filename,  _("PLCOpen files (*.xml)|*.xml|All files|*.*"), wx.SAVE | wx.OVERWRITE_PROMPT)

   381

        if dialog.ShowModal() == wx.ID_OK:

   382

            filepath = dialog.GetPath()

   383

            if os.path.isdir(os.path.dirname(filepath)):

   384

                result = self.Controler.SaveXMLFile(filepath)

   385

                if not result:

   386

                    self.ShowErrorMessage(_("Can't save project to file %s!") % filepath)

   387

            else:

   388

                self.ShowErrorMessage(_("\"%s\" is not a valid folder!") % os.path.dirname(filepath))

   389

            self._Refresh(TITLE, FILEMENU, PAGETITLES)

   390

        dialog.Destroy()

   391

   392

   393

class PLCOpenEditorApp(wx.App):

   394

    # def SetOpenFile(

   395

   396

    def PrintUsage(self):

   397

        print("\nUsage of PLCOpenEditor.py :")

   398

        print("\n   %s [Filepath]\n" % sys.argv[0])

   399

   400

    def ParseCommandLine(self):

   401

        # Parse options given to PLCOpenEditor in command line

   402

        try:

   403

            opts, args = getopt.getopt(sys.argv[1:], "h", ["help"])

   404

        except getopt.GetoptError:

   405

            # print help information and exit:

   406

            self.PrintUsage()

   407

            sys.exit(2)

   408

   409

        # Extract if help has been requested

   410

        for o, _a in opts:

   411

            if o in ("-h", "--help"):

   412

                self.PrintUsage()

   413

                sys.exit()

   414

   415

        # Extract the optional filename to open

   416

        self.fileOpen = None

   417

        if len(args) > 1:

   418

            self.PrintUsage()

   419

            sys.exit()

   420

        elif len(args) == 1:

   421

            self.fileOpen = args[0]

   422

   423

    def OnInit(self):

   424

        self.SetAppName('plcopeneditor')

   425

        self.ParseCommandLine()

   426

        InstallLocalRessources(beremiz_dir)

   427

        util.ExceptionHandler.AddExceptHook(version.app_version)

   428

        self.frame = PLCOpenEditor(None, fileOpen=self.fileOpen)

   429

        return True

   430

   431

    def Show(self):

   432

        self.frame.Show()

   433

   434

   435

if __name__ == '__main__':

   436

    app = PLCOpenEditorApp()

   437

    app.Show()

   438

    app.MainLoop()