1

import wx

     2

import os, sys, shutil

     3

from xml.dom import minidom

     4

     5

from py_ext import PythonFileCTNMixin

     6

     7

class WxGladeHMI(PythonFileCTNMixin):

     8

     9

    ConfNodeMethods = [

    10

        {"bitmap" : "editWXGLADE",

    11

         "name" : _("WXGLADE GUI"),

    12

         "tooltip" : _("Edit a WxWidgets GUI with WXGlade"),

    13

         "method" : "_editWXGLADE"},

    14

    ]

    15

    16

    def ConfNodePath(self):

    17

        return os.path.join(os.path.dirname(__file__))

    18

    19

    def _getWXGLADEpath(self, project_path=None):

    20

        if project_path is None:

    21

            project_path = self.CTNPath()

    22

        # define name for wxGlade gui file

    23

        return os.path.join(project_path, "hmi.wxg")

    24

    25

    def launch_wxglade(self, options, wait=False):

    26

        from wxglade import __file__ as fileName

    27

        path = os.path.dirname(fileName)

    28

        glade = os.path.join(path, 'wxglade.py')

    29

        if wx.Platform == '__WXMSW__':

    30

            glade = "\"%s\""%glade

    31

        mode = {False:os.P_NOWAIT, True:os.P_WAIT}[wait]

    32

        os.spawnv(mode, sys.executable, ["\"%s\""%sys.executable] + [glade] + options)

    33

    34

    def OnCTNSave(self, from_project_path=None):

    35

        if from_project_path is not None:

    36

            shutil.copyfile(self._getWXGLADEpath(from_project_path),

    37

                            self._getWXGLADEpath())

    38

        return PythonFileCTNMixin.OnCTNSave(self, from_project_path)

    39

    40

    def CTNGenerate_C(self, buildpath, locations):

    41

        """

    42

        Return C code generated by iec2c compiler 

    43

        when _generate_softPLC have been called

    44

        @param locations: ignored

    45

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

    46

        """

    47

    48

        current_location = self.GetCurrentLocation()

    49

        # define a unique name for the generated C file

    50

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

    51

    52

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

    53

        runtimefile = open(runtimefile_path, 'w')

    54

    55

        hmi_frames = {}

    56

    57

        wxgfile_path=self._getWXGLADEpath()

    58

        if os.path.exists(wxgfile_path):

    59

            wxgfile = open(wxgfile_path, 'r')

    60

            wxgtree = minidom.parse(wxgfile)

    61

            wxgfile.close()

    62

    63

            for node in wxgtree.childNodes[1].childNodes:

    64

                if node.nodeType == wxgtree.ELEMENT_NODE:

    65

                    hmi_frames[node._attrs["name"].value] =  node._attrs["class"].value

    66

    67

            hmipyfile_path=os.path.join(self._getBuildPath(), "hmi.py")

    68

            if wx.Platform == '__WXMSW__':

    69

                wxgfile_path = "\"%s\""%wxgfile_path

    70

                wxghmipyfile_path = "\"%s\""%hmipyfile_path

    71

            else:

    72

                wxghmipyfile_path = hmipyfile_path

    73

            self.launch_wxglade(['-o', wxghmipyfile_path, '-g', 'python', wxgfile_path], wait=True)

    74

    75

            hmipyfile = open(hmipyfile_path, 'r')

    76

            runtimefile.write(hmipyfile.read())

    77

            hmipyfile.close()

    78

    79

        sections_code = self.GetSectionsCode()

    80

    81

        # Adding variables

    82

        runtimefile.write("## User variables reference\n" +

    83

                          sections_code["variables"] + "\n")

    84

    85

        # Adding user global variables and routines

    86

        runtimefile.write("## User internal user variables and routines\n" +

    87

                          sections_code["globals"] + "\n")

    88

    89

        for section in ["init", "cleanup"]:

    90

            if not sections_code[section]:

    91

                sections_code[section] = "    pass"

    92

    93

        sections_code.update({

    94

            "location": location_str,

    95

            "declare_hmi": "\n".join(map(lambda x:"%s = None" % x, hmi_frames.keys())),

    96

            "global_hmi": ",".join(hmi_frames.keys()),

    97

            "init_hmi": "\n".join(map(lambda x: """

    98

    %(name)s = %(class)s(None)

    99

    %(name)s.Bind(wx.EVT_CLOSE, OnCloseFrame)

   100

    %(name)s.Show()

   101

""" % {"name": x[0], "class": x[1]},

   102

                             hmi_frames.items())),

   103

            "cleanup_hmi": "\n    ".join(map(lambda x:"if %s is not None: %s.Destroy()" % (x,x), hmi_frames.keys()))})

   104

   105

        runtimefile.write("""

   106

%(declare_hmi)s

   107

   108

def _runtime_%(location)s_init():

   109

%(init)s

   110

   111

def _runtime_%(location)s_cleanup():

   112

%(cleanup)s

   113

   114

def _runtime_%(location)s_start():

   115

    global %(global_hmi)s

   116

   117

%(start)s

   118

   119

    def OnCloseFrame(evt):

   120

        wx.MessageBox(_("Please stop PLC to close"))

   121

   122

    %(init_hmi)s

   123

   124

def _runtime_%(location)s_stop():

   125

    global %(global_hmi)s

   126

   127

    %(cleanup_hmi)s

   128

   129

%(stop)s

   130

   131

""" % sections_code)

   132

        runtimefile.close()

   133

   134

        return [], "", False, ("runtime_%s.py"%location_str, file(runtimefile_path,"rb"))

   135

   136

    def _editWXGLADE(self):

   137

        wxg_filename = self._getWXGLADEpath()

   138

        open_wxglade = True

   139

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

   140

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

   141

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

   142

                                      _("Open wxGlade"),

   143

                                      wx.YES_NO|wx.ICON_QUESTION)

   144

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

   145

            dialog.Destroy()

   146

        if open_wxglade:

   147

            if not os.path.exists(wxg_filename):

   148

                hmi_name = self.BaseParams.getName()

   149

                open(wxg_filename,"w").write("""<?xml version="1.0"?>

   150

    <application path="" name="" class="" option="0" language="python" top_window="%(name)s" encoding="UTF-8" use_gettext="0" overwrite="0" use_new_namespace="1" for_version="2.8" is_template="0">

   151

        <object class="%(class)s" name="%(name)s" base="EditFrame">

   152

            <style>wxDEFAULT_FRAME_STYLE</style>

   153

            <title>frame_1</title>

   154

            <object class="wxBoxSizer" name="sizer_1" base="EditBoxSizer">

   155

                <orient>wxVERTICAL</orient>

   156

            <object class="sizerslot" />

   157

        </object>

   158

        </object>

   159

    </application>

   160

    """ % {"name": hmi_name, "class": "Class_%s" % hmi_name})

   161

            if wx.Platform == '__WXMSW__':

   162

                wxg_filename = "\"%s\""%wxg_filename

   163

            self.launch_wxglade([wxg_filename])

   164