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

import wx

    27

import os

    28

import sys

    29

import shutil

    30

    31

from pyjs import translate

    32

    33

import util.paths as paths

    34

from POULibrary import POULibrary

    35

from docutil import open_svg

    36

from py_ext import PythonFileCTNMixin

    37

    38

    39

class SVGUILibrary(POULibrary):

    40

    def GetLibraryPath(self):

    41

        return paths.AbsNeighbourFile(__file__, "pous.xml")

    42

    43

    44

class SVGUI(PythonFileCTNMixin):

    45

    46

    ConfNodeMethods = [

    47

        {

    48

            "bitmap":    "ImportSVG",

    49

            "name":    _("Import SVG"),

    50

            "tooltip": _("Import SVG"),

    51

            "method":   "_ImportSVG"

    52

        },

    53

        {

    54

            "bitmap":    "ImportSVG",  # should be something different

    55

            "name":    _("Inkscape"),

    56

            "tooltip": _("Create HMI"),

    57

            "method":   "_StartInkscape"

    58

        },

    59

    ]

    60

    61

    def ConfNodePath(self):

    62

        return paths.AbsDir(__file__)

    63

    64

    def _getSVGpath(self, project_path=None):

    65

        if project_path is None:

    66

            project_path = self.CTNPath()

    67

        # define name for SVG file containing gui layout

    68

        return os.path.join(project_path, "gui.svg")

    69

    70

    def _getSVGUIserverpath(self):

    71

        return paths.AbsNeighbourFile(__file__, "svgui_server.py")

    72

    73

    def OnCTNSave(self, from_project_path=None):

    74

        if from_project_path is not None:

    75

            shutil.copyfile(self._getSVGpath(from_project_path),

    76

                            self._getSVGpath())

    77

        return PythonFileCTNMixin.OnCTNSave(self, from_project_path)

    78

    79

    def CTNGenerate_C(self, buildpath, locations):

    80

        """

    81

        Return C code generated by iec2c compiler

    82

        when _generate_softPLC have been called

    83

        @param locations: ignored

    84

        @return: [(C_file_name, CFLAGS),...] , LDFLAGS_TO_APPEND

    85

        """

    86

    87

        current_location = self.GetCurrentLocation()

    88

        # define a unique name for the generated C file

    89

        location_str = "_".join(map(lambda x: str(x), current_location))

    90

    91

        res = ([], "", False)

    92

    93

        svgfile = self._getSVGpath()

    94

        if os.path.exists(svgfile):

    95

            res += (("gui.svg", file(svgfile, "rb")),)

    96

    97

        svguiserverfile = open(self._getSVGUIserverpath(), 'r')

    98

        svguiservercode = svguiserverfile.read()

    99

        svguiserverfile.close()

   100

   101

        svguilibpath = os.path.join(self._getBuildPath(), "svguilib.js")

   102

        svguilibfile = open(svguilibpath, 'w')

   103

        fpath = paths.AbsDir(__file__)

   104

        svguilibfile.write(translate(os.path.join(fpath, "pyjs", "lib", "sys.py"), "sys"))

   105

        svguilibfile.write(open(os.path.join(fpath, "pyjs", "lib", "_pyjs.js"), 'r').read())

   106

        svguilibfile.write(translate(os.path.join(fpath, "pyjs", "lib", "pyjslib.py"), "pyjslib"))

   107

        svguilibfile.write(translate(os.path.join(fpath, "svguilib.py"), "svguilib"))

   108

        svguilibfile.write("pyjslib();\nsvguilib();\n")

   109

        svguilibfile.write(open(os.path.join(fpath, "pyjs", "lib", "json.js"), 'r').read())

   110

        svguilibfile.write(open(os.path.join(fpath, "livesvg.js"), 'r').read())

   111

        svguilibfile.close()

   112

        jsmodules = {"LiveSVGPage": "svguilib.js"}

   113

        res += (("svguilib.js", file(svguilibpath, "rb")),)

   114

   115

        runtimefile_path = os.path.join(buildpath, "runtime_%s.py" % location_str)

   116

        runtimefile = open(runtimefile_path, 'w')

   117

        runtimefile.write(svguiservercode % {"svgfile": "gui.svg"})

   118

        runtimefile.write("""

   119

def _runtime_%(location)s_start():

   120

    website.LoadHMI(%(svgui_class)s, %(jsmodules)s)

   121

   122

def _runtime_%(location)s_stop():

   123

    website.UnLoadHMI()

   124

   125

        """ % {"location": location_str,

   126

               "svgui_class": "SVGUI_HMI",

   127

               "jsmodules": str(jsmodules)})

   128

        runtimefile.close()

   129

   130

        res += (("runtime_%s.py" % location_str, file(runtimefile_path, "rb")),)

   131

   132

        return res

   133

   134

    def _ImportSVG(self):

   135

        dialog = wx.FileDialog(self.GetCTRoot().AppFrame, _("Choose a SVG file"), os.getcwd(), "",  _("SVG files (*.svg)|*.svg|All files|*.*"), wx.OPEN)

   136

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

   137

            svgpath = dialog.GetPath()

   138

            if os.path.isfile(svgpath):

   139

                shutil.copy(svgpath, self._getSVGpath())

   140

            else:

   141

                self.GetCTRoot().logger.write_error(_("No such SVG file: %s\n") % svgpath)

   142

        dialog.Destroy()

   143

   144

    def _StartInkscape(self):

   145

        svgfile = self._getSVGpath()

   146

        open_inkscape = True

   147

        if not self.GetCTRoot().CheckProjectPathPerm():

   148

            dialog = wx.MessageDialog(self.GetCTRoot().AppFrame,

   149

                                      _("You don't have write permissions.\nOpen Inkscape anyway ?"),

   150

                                      _("Open Inkscape"),

   151

                                      wx.YES_NO | wx.ICON_QUESTION)

   152

            open_inkscape = dialog.ShowModal() == wx.ID_YES

   153

            dialog.Destroy()

   154

        if open_inkscape:

   155

            if not os.path.isfile(svgfile):

   156

                svgfile = None

   157

            open_svg(svgfile)