laurent@367: import wx
Laurent@1061: import os, sys, shutil
laurent@367: from xml.dom import minidom
laurent@367: 
Edouard@728: from py_ext import PythonFileCTNMixin
laurent@367: 
Edouard@728: class WxGladeHMI(PythonFileCTNMixin):
laurent@367: 
Edouard@717:     ConfNodeMethods = [
Edouard@734:         {"bitmap" : "editWXGLADE",
laurent@367:          "name" : _("WXGLADE GUI"),
laurent@367:          "tooltip" : _("Edit a WxWidgets GUI with WXGlade"),
laurent@367:          "method" : "_editWXGLADE"},
laurent@367:     ]
laurent@367: 
Laurent@1163:     def GetIconName(self):
Laurent@1163:         return "wxGlade"
Laurent@1163: 
Edouard@728:     def ConfNodePath(self):
Edouard@728:         return os.path.join(os.path.dirname(__file__))
Edouard@728: 
Laurent@1061:     def _getWXGLADEpath(self, project_path=None):
Laurent@1061:         if project_path is None:
Laurent@1061:             project_path = self.CTNPath()
Laurent@1061:         # define name for wxGlade gui file
Laurent@1061:         return os.path.join(project_path, "hmi.wxg")
laurent@367: 
laurent@367:     def launch_wxglade(self, options, wait=False):
laurent@367:         from wxglade import __file__ as fileName
laurent@367:         path = os.path.dirname(fileName)
laurent@367:         glade = os.path.join(path, 'wxglade.py')
laurent@367:         if wx.Platform == '__WXMSW__':
laurent@367:             glade = "\"%s\""%glade
laurent@367:         mode = {False:os.P_NOWAIT, True:os.P_WAIT}[wait]
laurent@367:         os.spawnv(mode, sys.executable, ["\"%s\""%sys.executable] + [glade] + options)
laurent@367: 
Laurent@1061:     def OnCTNSave(self, from_project_path=None):
Laurent@1061:         if from_project_path is not None:
Laurent@1061:             shutil.copyfile(self._getWXGLADEpath(from_project_path),
Laurent@1061:                             self._getWXGLADEpath())
Laurent@1061:         return PythonFileCTNMixin.OnCTNSave(self, from_project_path)
laurent@367: 
Edouard@718:     def CTNGenerate_C(self, buildpath, locations):
laurent@367:         
Edouard@1155:         hmi_frames = []
laurent@367:         
laurent@367:         wxgfile_path=self._getWXGLADEpath()
laurent@367:         if os.path.exists(wxgfile_path):
laurent@367:             wxgfile = open(wxgfile_path, 'r')
laurent@367:             wxgtree = minidom.parse(wxgfile)
laurent@367:             wxgfile.close()
laurent@367:             
laurent@367:             for node in wxgtree.childNodes[1].childNodes:
laurent@367:                 if node.nodeType == wxgtree.ELEMENT_NODE:
Edouard@1155:                     hmi_frames.append({
Edouard@1155:                         "name" : node.getAttribute("name"),
Edouard@1155:                         "class" : node.getAttribute("class"),
Edouard@1155:                         "handlers" : [
Edouard@1155:                             hnode.firstChild.data for hnode in 
Edouard@1155:                             node.getElementsByTagName("handler")]})
laurent@367:                     
laurent@367:             hmipyfile_path=os.path.join(self._getBuildPath(), "hmi.py")
laurent@367:             if wx.Platform == '__WXMSW__':
laurent@367:                 wxgfile_path = "\"%s\""%wxgfile_path
laurent@384:                 wxghmipyfile_path = "\"%s\""%hmipyfile_path
laurent@384:             else:
laurent@384:                 wxghmipyfile_path = hmipyfile_path
laurent@384:             self.launch_wxglade(['-o', wxghmipyfile_path, '-g', 'python', wxgfile_path], wait=True)
laurent@367:             
laurent@367:             hmipyfile = open(hmipyfile_path, 'r')
Edouard@1132:             define_hmi = hmipyfile.read().decode('utf-8')
laurent@367:             hmipyfile.close()
laurent@367:         
Laurent@1256:         else:
Laurent@1256:             define_hmi = ""
Laurent@1256:         
Edouard@1155:         declare_hmi = "\n".join(["%(name)s = None\n" % x + 
Edouard@1155:                           "\n".join(["%(class)s.%(h)s = %(h)s"%
Edouard@1155:                             dict(x,h=h) for h in x['handlers']])
Edouard@1155:                                 for x in hmi_frames])
Laurent@1256:         global_hmi = ("global %s\n"%",".join(
Laurent@1256:                          [x["name"] for x in hmi_frames]) 
Laurent@1256:                       if len(hmi_frames) > 0 else "")
Edouard@1155:         init_hmi = "\n".join(["""\
Edouard@1132: def OnCloseFrame(evt):
Edouard@1132:     wx.MessageBox(_("Please stop PLC to close"))
Edouard@1132: 
Edouard@1132: %(name)s = %(class)s(None)
Edouard@1132: %(name)s.Bind(wx.EVT_CLOSE, OnCloseFrame)
Edouard@1132: %(name)s.Show()
Edouard@1155: """ % x for x in hmi_frames])
Edouard@1155:         cleanup_hmi = "\n".join(
Edouard@1155:             ["if %(name)s is not None: %(name)s.Destroy()" % x 
Edouard@1155:                 for x in hmi_frames])
Laurent@1124:         
Edouard@1132:         self.PreSectionsTexts = {
Edouard@1155:             "globals":define_hmi,
Edouard@1132:             "start":global_hmi,
Laurent@1151:             "stop":global_hmi + cleanup_hmi
Edouard@1132:         }
Edouard@1132:         self.PostSectionsTexts = {
Edouard@1155:             "globals":declare_hmi,
Edouard@1132:             "start":init_hmi,
Edouard@1132:         }
Laurent@1124: 
Edouard@1132:         return PythonFileCTNMixin.CTNGenerate_C(self, buildpath, locations)
laurent@367: 
laurent@367:     def _editWXGLADE(self):
laurent@367:         wxg_filename = self._getWXGLADEpath()
greg@427:         open_wxglade = True
Edouard@718:         if not self.GetCTRoot().CheckProjectPathPerm():
Edouard@718:             dialog = wx.MessageDialog(self.GetCTRoot().AppFrame,
greg@427:                                       _("You don't have write permissions.\nOpen wxGlade anyway ?"),
greg@427:                                       _("Open wxGlade"),
greg@427:                                       wx.YES_NO|wx.ICON_QUESTION)
greg@427:             open_wxglade = dialog.ShowModal() == wx.ID_YES
greg@427:             dialog.Destroy()
greg@427:         if open_wxglade:
greg@427:             if not os.path.exists(wxg_filename):
greg@427:                 hmi_name = self.BaseParams.getName()
greg@427:                 open(wxg_filename,"w").write("""<?xml version="1.0"?>
greg@427:     <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">
greg@427:         <object class="%(class)s" name="%(name)s" base="EditFrame">
greg@427:             <style>wxDEFAULT_FRAME_STYLE</style>
greg@427:             <title>frame_1</title>
laurent@834:             <object class="wxBoxSizer" name="sizer_1" base="EditBoxSizer">
laurent@834:                 <orient>wxVERTICAL</orient>
laurent@834:             <object class="sizerslot" />
laurent@834:         </object>
greg@427:         </object>
greg@427:     </application>
greg@427:     """ % {"name": hmi_name, "class": "Class_%s" % hmi_name})
greg@427:             if wx.Platform == '__WXMSW__':
greg@427:                 wxg_filename = "\"%s\""%wxg_filename
greg@427:             self.launch_wxglade([wxg_filename])
Laurent@1061: