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 GetIconName(self):

    17

        return "wxGlade"

    18

    19

    def ConfNodePath(self):

    20

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

    21

    22

    def _getWXGLADEpath(self, project_path=None):

    23

        if project_path is None:

    24

            project_path = self.CTNPath()

    25

        # define name for wxGlade gui file

    26

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

    27

    28

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

    29

        from wxglade import __file__ as fileName

    30

        path = os.path.dirname(fileName)

    31

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

    32

        if wx.Platform == '__WXMSW__':

    33

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

    34

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

    35

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

    36

    37

    def OnCTNSave(self, from_project_path=None):

    38

        if from_project_path is not None:

    39

            shutil.copyfile(self._getWXGLADEpath(from_project_path),

    40

                            self._getWXGLADEpath())

    41

        return PythonFileCTNMixin.OnCTNSave(self, from_project_path)

    42

    43

    def CTNGenerate_C(self, buildpath, locations):

    44

    45

        hmi_frames = []

    46

    47

        wxgfile_path=self._getWXGLADEpath()

    48

        if os.path.exists(wxgfile_path):

    49

            wxgfile = open(wxgfile_path, 'r')

    50

            wxgtree = minidom.parse(wxgfile)

    51

            wxgfile.close()

    52

    53

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

    54

                if node.nodeType == wxgtree.ELEMENT_NODE:

    55

                    hmi_frames.append({

    56

                        "name" : node.getAttribute("name"),

    57

                        "class" : node.getAttribute("class"),

    58

                        "handlers" : [

    59

                            hnode.firstChild.data for hnode in 

    60

                            node.getElementsByTagName("handler")]})

    61

    62

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

    63

            if wx.Platform == '__WXMSW__':

    64

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

    65

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

    66

            else:

    67

                wxghmipyfile_path = hmipyfile_path

    68

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

    69

    70

            hmipyfile = open(hmipyfile_path, 'r')

    71

            define_hmi = hmipyfile.read().decode('utf-8')

    72

            hmipyfile.close()

    73

    74

        else:

    75

            define_hmi = ""

    76

    77

        declare_hmi = "\n".join(["%(name)s = None\n" % x + 

    78

                          "\n".join(["%(class)s.%(h)s = %(h)s"%

    79

                            dict(x,h=h) for h in x['handlers']])

    80

                                for x in hmi_frames])

    81

        global_hmi = ("global %s\n"%",".join(

    82

                         [x["name"] for x in hmi_frames]) 

    83

                      if len(hmi_frames) > 0 else "")

    84

        init_hmi = "\n".join(["""\

    85

def OnCloseFrame(evt):

    86

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

    87

    88

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

    89

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

    90

%(name)s.Show()

    91

""" % x for x in hmi_frames])

    92

        cleanup_hmi = "\n".join(

    93

            ["if %(name)s is not None: %(name)s.Destroy()" % x 

    94

                for x in hmi_frames])

    95

    96

        self.PreSectionsTexts = {

    97

            "globals":define_hmi,

    98

            "start":global_hmi,

    99

            "stop":global_hmi + cleanup_hmi

   100

        }

   101

        self.PostSectionsTexts = {

   102

            "globals":declare_hmi,

   103

            "start":init_hmi,

   104

        }

   105

   106

        return PythonFileCTNMixin.CTNGenerate_C(self, buildpath, locations)

   107

   108

    def _editWXGLADE(self):

   109

        wxg_filename = self._getWXGLADEpath()

   110

        open_wxglade = True

   111

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

   112

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

   113

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

   114

                                      _("Open wxGlade"),

   115

                                      wx.YES_NO|wx.ICON_QUESTION)

   116

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

   117

            dialog.Destroy()

   118

        if open_wxglade:

   119

            if not os.path.exists(wxg_filename):

   120

                hmi_name = self.BaseParams.getName()

   121

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

   122

    <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">

   123

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

   124

            <style>wxDEFAULT_FRAME_STYLE</style>

   125

            <title>frame_1</title>

   126

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

   127

                <orient>wxVERTICAL</orient>

   128

            <object class="sizerslot" />

   129

        </object>

   130

        </object>

   131

    </application>

   132

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

   133

            if wx.Platform == '__WXMSW__':

   134

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

   135

            self.launch_wxglade([wxg_filename])

   136