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

      })

   129

        runtimefile.close()

   130

   131

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

   132

   133

        return res

   134

   135

    def _ImportSVG(self):

   136

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

   137

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

   138

            svgpath = dialog.GetPath()

   139

            if os.path.isfile(svgpath):

   140

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

   141

            else:

   142

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

   143

        dialog.Destroy()

   144

   145

    def _StartInkscape(self):

   146

        svgfile = self._getSVGpath()

   147

        open_inkscape = True

   148

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

   149

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

   150

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

   151

                                      _("Open Inkscape"),

   152

                                      wx.YES_NO | wx.ICON_QUESTION)

   153

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

   154

            dialog.Destroy()

   155

        if open_inkscape:

   156

            if not os.path.isfile(svgfile):

   157

                svgfile = None

   158

            open_svg(svgfile)