1

import os, shutil, sys

     2

base_folder = os.path.split(sys.path[0])[0]

     3

sys.path.append(os.path.join(base_folder, "wxsvg", "SVGUIEditor"))

     4

sys.path.append(os.path.join(base_folder, "plcopeneditor", "graphics"))

     5

     6

import wx

     7

     8

from SVGUIGenerator import *

     9

from SVGUIControler import *

    10

from SVGUIEditor import *

    11

from FBD_Objects import *

    12

    13

from wxPopen import ProcessLogger

    14

from wx.wxsvg import SVGDocument

    15

    16

[ID_SVGUIEDITORFBDPANEL, 

    17

] = [wx.NewId() for _init_ctrls in range(1)]

    18

    19

SVGUIFB_Types = {ITEM_CONTAINER : "Container",

    20

                 ITEM_BUTTON : "Button", 

    21

                 ITEM_TEXT : "TextCtrl", 

    22

                 ITEM_SCROLLBAR : "ScrollBar", 

    23

                 ITEM_ROTATING : "RotatingCtrl", 

    24

                 ITEM_NOTEBOOK : "NoteBook", 

    25

                 ITEM_TRANSFORM : "Transform"}

    26

    27

class _SVGUIEditor(SVGUIEditor):

    28

    """

    29

    This Class add IEC specific features to the SVGUIEditor :

    30

        - FDB preview

    31

        - FBD begin drag 

    32

    """

    33

    34

    def _init_coll_EditorGridSizer_Items(self, parent):

    35

        SVGUIEditor._init_coll_EditorGridSizer_Items(self, parent)

    36

        parent.AddWindow(self.FBDPanel, 0, border=0, flag=wx.GROW)

    37

    38

    def _init_ctrls(self, prnt):

    39

        SVGUIEditor._init_ctrls(self, prnt, False)

    40

    41

        self.FBDPanel = wx.Panel(id=ID_SVGUIEDITORFBDPANEL, 

    42

                  name='FBDPanel', parent=self.EditorPanel, pos=wx.Point(0, 0),

    43

                  size=wx.Size(0, 0), style=wx.TAB_TRAVERSAL|wx.SIMPLE_BORDER)

    44

        self.FBDPanel.SetBackgroundColour(wx.WHITE)

    45

        self.FBDPanel.Bind(wx.EVT_LEFT_DOWN, self.OnFBDPanelClick)

    46

        self.FBDPanel.Bind(wx.EVT_PAINT, self.OnPaintFBDPanel)

    47

    48

        setattr(self.FBDPanel, "GetScaling", lambda: None) 

    49

    50

        self._init_sizers()

    51

    52

    def __init__(self, parent, controler = None, fileOpen = None):

    53

        SVGUIEditor.__init__(self, parent, controler, fileOpen)

    54

    55

        self.FBDBlock = None

    56

    57

    def RefreshView(self, select_id = None):

    58

        SVGUIEditor.RefreshView(self, select_id)

    59

        self.FBDPanel.Refresh()

    60

    61

    def OnPaintFBDPanel(self,event):

    62

        dc = wx.ClientDC(self.FBDPanel)

    63

        dc.Clear()

    64

        selected = self.GetSelected()

    65

        if selected is not None:

    66

            selected_type = self.Controler.GetElementType(selected)

    67

            if selected_type is not None:

    68

                self.FBDBlock = FBD_Block(parent=self.FBDPanel,type=SVGUIFB_Types[selected_type],name=self.Controler.GetElementName(selected))

    69

                width, height = self.FBDBlock.GetMinSize()

    70

                self.FBDBlock.SetSize(width,height)

    71

                clientsize = self.FBDPanel.GetClientSize()

    72

                x = (clientsize.width - width) / 2

    73

                y = (clientsize.height - height) / 2

    74

                self.FBDBlock.SetPosition(x, y)

    75

                self.FBDBlock.Draw(dc)

    76

        else:

    77

            self.FBDBlock = None

    78

        event.Skip()

    79

    80

    def OnFBDPanelClick(self, event):

    81

        if self.FBDBlock:

    82

            data = wx.TextDataObject(str((self.FBDBlock.GetType(), "functionBlock", self.FBDBlock.GetName())))

    83

            DropSrc = wx.DropSource(self.FBDPanel)

    84

            DropSrc.SetData(data)

    85

            DropSrc.DoDragDrop()

    86

        event.Skip()

    87

    88

    def OnInterfaceTreeItemSelected(self, event):

    89

        self.FBDPanel.Refresh()

    90

        SVGUIEditor.OnInterfaceTreeItemSelected(self, event)

    91

    92

    def OnGenerate(self,event):

    93

        self.SaveProject()

    94

        self.Controler.PlugGenerate_C(sys.path[0],(0,0,4,5),None)

    95

        event.Skip()    

    96

    97

TYPECONVERSION = {"BOOL" : "X", "SINT" : "B", "INT" : "W", "DINT" : "D", "LINT" : "L",

    98

    "USINT" : "B", "UINT" : "W", "UDINT" : "D", "ULINT" : "L", "REAL" : "D", "LREAL" : "L",

    99

    "STRING" : "B", "BYTE" : "B", "WORD" : "W", "DWORD" : "D", "LWORD" : "L", "WSTRING" : "W"}

   100

   101

CTYPECONVERSION = {"BOOL" : "IEC_BOOL", "UINT" : "IEC_UINT", "STRING" : "IEC_STRING", "REAL" : "IEC_REAL"}

   102

CPRINTTYPECONVERSION = {"BOOL" : "d", "UINT" : "d", "STRING" : "s", "REAL" : "f"}

   103

   104

class RootClass(SVGUIControler):

   105

   106

    def __init__(self):

   107

        SVGUIControler.__init__(self)

   108

        filepath = os.path.join(self.PlugPath(), "gui.xml")

   109

   110

        if os.path.isfile(filepath):

   111

            svgfile = os.path.join(self.PlugPath(), "gui.svg")

   112

            if os.path.isfile(svgfile):

   113

                self.SvgFilepath = svgfile

   114

            self.OpenXMLFile(filepath)

   115

        else:

   116

            self.CreateNewInterface()

   117

            self.SetFilePath(filepath)

   118

   119

    def GetElementIdFromName(self, name):

   120

        element = self.GetElementByName(name)

   121

        if element is not None:

   122

            return element.getid()

   123

        return None

   124

   125

    _View = None

   127

        if not self._View:

   128

            def _onclose():

   129

                self._View = None

   130

            def _onsave():

   131

                self.GetPlugRoot().SaveProject()

   132

            self._View = _SVGUIEditor(self.GetPlugRoot().AppFrame, self)

   133

            self._View._onclose = _onclose

   134

            self._View._onsave = _onsave

   135

            self._View.Show()

   136

   138

        if not self._View:

   139

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

   140

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

   141

                svgpath = dialog.GetPath()

   142

                if os.path.isfile(svgpath):

   143

                    shutil.copy(svgpath, os.path.join(self.PlugPath(), "gui.svg"))

   144

                else:

   146

            dialog.Destroy()

   147

   149

        if not self._View:

   150

            dialog = wx.FileDialog(self.GetPlugRoot().AppFrame, "Choose a XML file", os.getcwd(), "",  "XML files (*.xml)|*.xml|All files|*.*", wx.OPEN)

   151

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

   152

                xmlpath = dialog.GetPath()

   153

                if os.path.isfile(xmlpath):

   154

                    shutil.copy(xmlpath, os.path.join(self.PlugPath(), "gui.xml"))

   155

                else:

   157

            dialog.Destroy()

   158

   159

    PluginMethods = [

   160

        {"bitmap" : os.path.join("images","HMIEditor"),

   161

         "name" : "HMI Editor",

   162

         "tooltip" : "HMI Editor",

   163

         "method" : "_OpenView"},

   164

        {"bitmap" : os.path.join("images","ImportSVG"),

   165

         "name" : "Import SVG",

   166

         "tooltip" : "Import SVG",

   167

         "method" : "_ImportSVG"},

   168

        {"bitmap" : os.path.join("images","ImportDEF"),

   169

         "name" : "Import XML",

   170

         "tooltip" : "Import XML",

   171

         "method" : "_ImportXML"},

   172

    ]

   173

   174

    def OnPlugSave(self):

   175

        self.SaveXMLFile(os.path.join(self.PlugPath(), "gui.xml"))

   176

        return True

   177

   179

        progname = "SVGUI_%s"%"_".join(map(str, self.GetCurrentLocation()))

   180

   181

        doc = SVGDocument(self.GetSVGFilePath())

   182

        root_element = doc.GetRootElement()

   183

        window_size = (int(float(root_element.GetAttribute("width"))),

   184

                       int(float(root_element.GetAttribute("height"))))

   185

   195

   196

        generator = _SVGUICGenerator(self, self.GetElementsByType(), 

   197

                                     os.path.split(self.GetSVGFilePath())[1], 

   198

                                     os.path.split(self.GetFilePath())[1], 

   199

                                     self.GetCurrentLocation())

   200

        generator.GenerateProgram(window_size, buildpath, progname)

   201

        Gen_C_file = os.path.join(buildpath, progname+".cpp" )

   202

   203

        if wx.Platform == '__WXMSW__':

   204

            cxx_flags = "-I..\\..\\wxPython-src-2.8.7.1\\bld\\lib\\wx\\include\\msw-unicode-release-2.8 -I..\\..\\wxPython-src-2.8.7.1\\include -I..\\..\\wxPython-src-2.8.7.1\\contrib\\include -I..\\..\\matiec\\lib -DWXUSINGDLL -D__WXMSW__ -mthreads"

   205

            libs = "\"..\\lib\\libwxsvg.a\" \"..\\lib\\libwxsvg_agg.a\" \"..\\lib\\libagg.a\" \"..\\lib\\libaggplatformwin32.a\" \"..\\lib\\libaggfontwin32tt.a\" -L..\\..\\wxPython-src-2.8.7.1\\bld\\lib -mno-cygwin -mwindows -mthreads  -mno-cygwin -mwindows -Wl,--subsystem,windows -mwindows -lwx_mswu_richtext-2.8 -lwx_mswu_aui-2.8 -lwx_mswu_xrc-2.8 -lwx_mswu_qa-2.8 -lwx_mswu_html-2.8 -lwx_mswu_adv-2.8 -lwx_mswu_core-2.8 -lwx_baseu_xml-2.8 -lwx_baseu_net-2.8 -lwx_baseu-2.8"

   206

        else:

   208

            if status:

   210

            cxx_flags = result.strip() + " -I../matiec/lib"

   211

   213

            if status:

   215

            libs = result.strip() + " -lwxsvg"

   216

   218

   219

    def BlockTypesFactory(self):

   220

   221

        SVGUIBlock_Types = []

   222

   223

        def GetSVGUIBlockType(type):

   224

            for category in SVGUIBlock_Types:

   225

                for blocktype in category["list"]:

   226

                    if blocktype["name"] == type:

   227

                        return blocktype

   228

        setattr(self, "GetSVGUIBlockType", GetSVGUIBlockType)

   229

   230

        def generate_svgui_block(generator, block, body, link, order=False):

   231

            name = block.getinstanceName()

   232

            block_id = self.GetElementIdFromName(name)

   233

            if block_id == None:

   234

                raise ValueError, "No corresponding block found"

   235

            type = block.gettypeName()

   236

            block_infos = GetSVGUIBlockType(type)

   237

            current_location = ".".join(map(str, self.GetCurrentLocation()))

   238

            if not generator.ComputedBlocks.get(block, False) and not order:

   239

                generator.ComputedBlocks[block] = True

   240

                for num, variable in enumerate(block.inputVariables.getvariable()):

   241

                    connections = variable.connectionPointIn.getconnections()

   242

                    input_info = (generator.TagName, "block", block.getlocalId(), "input", num)

   243

                    if connections and len(connections) == 1:

   244

                        parameter = "%sQ%s%s.%d.%d"%("%", TYPECONVERSION[block_infos["inputs"][num][1]], current_location, block_id, num+1)

   245

                        value = generator.ComputeFBDExpression(body, connections[0])

   246

                        generator.Program += [(generator.CurrentIndent, ()),

   247

                                              (parameter, input_info),

   248

                                              (" := ", ())]

   249

                        generator.Program += generator.ExtractModifier(variable, value, input_info)

   250

                        generator.Program += [(";\n", ())]

   251

            if link:

   252

                connectionPoint = link.getposition()[-1]

   253

                for num, variable in enumerate(block.outputVariables.getvariable()):

   254

                    blockPointx, blockPointy = variable.connectionPointOut.getrelPositionXY()

   255

                    output_info = (generator.TagName, "block", block.getlocalId(), "output", num)

   256

                    if block.getx() + blockPointx == connectionPoint.getx() and block.gety() + blockPointy == connectionPoint.gety():

   257

                        return [("%sI%s%s.%d.%d"%("%", TYPECONVERSION[block_infos["outputs"][num][1]], current_location, block_id, num+1), output_info)]

   258

                raise ValueError, "No output variable found"

   259

            else:

   260

                return None

   261

   262

        def initialise_block(type, name, block = None):

   263

            block_id = self.GetElementIdFromName(name)

   264

            if block_id == None:

   265

                raise ValueError, "No corresponding block found"

   266

            block_infos = GetSVGUIBlockType(type)

   267

            current_location = ".".join(map(str, self.GetCurrentLocation()))

   268

            variables = []

   269

            if block is not None:

   270

                input_variables = block.inputVariables.getvariable()

   271

                output_variables = block.outputVariables.getvariable()

   272

            else:

   273

                input_variables = None

   274

                output_variables = None

   275

            for num, (input_name, input_type, input_modifier) in enumerate(block_infos["inputs"]):

   276

                if input_variables is not None and num < len(input_variables):

   277

                    connections = input_variables[num].connectionPointIn.getconnections()

   278

                if input_variables is None or connections and len(connections) == 1:

   279

                    variables.append((input_type, None, "%sQ%s%s.%d.%d"%("%", TYPECONVERSION[input_type], current_location, block_id, num+1), None))

   280

            for num, (output_name, output_type, output_modifier) in enumerate(block_infos["outputs"]):

   281

                variables.append((output_type, None, "%sI%s%s.%d.%d"%("%", TYPECONVERSION[input_type], current_location, block_id, num+1), None))

   282

            return variables

   283

   284

        SVGUIBlock_Types.extend([{"name" : "SVGUI function blocks", "list" :

   285

                [{"name" : "Container", "type" : "functionBlock", "extensible" : False, 

   286

                    "inputs" : [("Show","BOOL","none"),("Enable","BOOL","none")], 

   287

                    "outputs" : [],

   288

                    "comment" : "SVGUI Container",

   289

                    "generate" : generate_svgui_block, "initialise" : initialise_block},

   290

                {"name" : "Button", "type" : "functionBlock", "extensible" : False, 

   291

                    "inputs" : [("Show","BOOL","none"),("Enable","BOOL","none"),("Value","BOOL","none")], 

   292

                    "outputs" : [("State","BOOL","none")],

   293

                    "comment" : "SVGUI Button",

   294

                    "generate" : generate_svgui_block, "initialise" : initialise_block},

   295

                {"name" : "TextCtrl", "type" : "functionBlock", "extensible" : False, 

   296

                    "inputs" : [("Show","BOOL","none"),("Enable","BOOL","none"),("SetText","STRING","none")], 

   297

                    "outputs" : [("Text","STRING","none")],

   298

                    "comment" : "SVGUI Text Control",

   299

                    "generate" : generate_svgui_block, "initialise" : initialise_block},

   300

                {"name" : "ScrollBar", "type" : "functionBlock", "extensible" : False, 

   301

                    "inputs" : [("Show","BOOL","none"),("Enable","BOOL","none"),("SetThumb","UINT","none"),("SetRange","UINT","none"),("SetPosition","UINT","none")], 

   302

                    "outputs" : [("Position","UINT","none")],

   303

                    "comment" : "SVGUI ScrollBar",

   304

                    "generate" : generate_svgui_block, "initialise" : initialise_block},

   305

                {"name" : "NoteBook", "type" : "functionBlock", "extensible" : False, 

   306

                    "inputs" : [("Show","BOOL","none"),("Enable","BOOL","none"),("SetSelected","BOOL","none")], 

   307

                    "outputs" : [("Selected","UINT","none")],

   308

                    "comment" : "SVGUI Notebook",

   309

                    "generate" : generate_svgui_block, "initialise" : initialise_block},

   310

                {"name" : "RotatingCtrl", "type" : "functionBlock", "extensible" : False, 

   311

                    "inputs" : [("Show","BOOL","none"),("Enable","BOOL","none"),("SetAngle","REAL","none")], 

   312

                    "outputs" : [("Angle","REAL","none")],

   313

                    "comment" : "SVGUI Rotating Control",

   314

                    "generate" : generate_svgui_block, "initialise" : initialise_block},

   315

                {"name" : "Transform", "type" : "functionBlock", "extensible" : False, 

   316

                    "inputs" : [("Show","BOOL","none"),("Enable","BOOL","none"),("SetX","REAL","none"),("SetY","REAL","none"),("SetXScale","REAL","none"),("SetYScale","REAL","none"),("SetAngle","REAL","none")], 

   317

                    "outputs" : [("X","REAL","none"),("Y","REAL","none")],

   318

                    "comment" : "SVGUI Transform",

   319

                    "generate" : generate_svgui_block, "initialise" : initialise_block},

   320

               ]}

   321

        ])

   322

   323

        return SVGUIBlock_Types

   324

   325

   326

class _SVGUICGenerator(SVGUICGenerator):

   327

   328

    def __init__(self, controler, elements, svgfile, xmlfile, current_location):

   329

        SVGUICGenerator.__init__(self, elements, svgfile, xmlfile)

   330

   331

        self.CurrentLocation = current_location

   332

        self.Controler = controler

   333

   334

    def GenerateProgramHeadersPublicVars(self):

   337

   338

    void Retrieve();

   339

    void Publish();

   340

    void Initialize();

   341

"""

   342

#        text += "    void Print();\n"

   343

        return text

   344

   345

    def GenerateIECVars(self):

   346

        text = ""

   347

        for element in self.Elements:

   348

            text += "STATE_TYPE out_state_%d;\n"%element.getid()

   349

            text += "STATE_TYPE in_state_%d;\n"%element.getid()

   350

        text +="\n"

   351

        current_location = "_".join(map(str, self.CurrentLocation))

   352

        #Declaration des variables

   353

        for element in self.Elements:

   354

            block_infos = self.Controler.GetSVGUIBlockType(SVGUIFB_Types[GetElementType(element)])

   355

            block_id = element.getid()

   356

            for i, input in enumerate(block_infos["inputs"]):

   357

                element_c_type = CTYPECONVERSION[input[1]]

   358

                variable = "__Q%s%s_%d_%d"%(TYPECONVERSION[input[1]], current_location, block_id, i + 1)

   359

                text += "%s beremiz%s;\n"%(element_c_type, variable)

   360

                text += "%s* %s = &beremiz%s;\n"%(element_c_type, variable, variable)

   361

                text += "%s _copy%s;\n"%(element_c_type, variable)

   362

            for i, output in enumerate(block_infos["outputs"]):

   363

                element_c_type = CTYPECONVERSION[output[1]]

   364

                variable = "__I%s%s_%d_%d"%(TYPECONVERSION[output[1]], current_location, block_id, i + 1)

   365

                text += "%s beremiz%s;\n"%(element_c_type, variable)

   366

                text += "%s* %s = &beremiz%s;\n"%(element_c_type, variable, variable)

   367

                text += "%s _copy%s;\n"%(element_c_type, variable)

   368

            text +="\n"

   369

        return text

   370

   371

    def GenerateGlobalVarsAndFuncs(self, size):

   372

        text = """#include "iec_types.h"

   373

#ifdef __WXMSW__

   374

#define COMPARE_AND_SWAP_VAL(Destination, comparand, exchange) InterlockedCompareExchange(Destination, exchange, comparand)

   375

#define THREAD_RETURN_TYPE DWORD WINAPI

   376

#define STATE_TYPE long int

   377

#else

   378

#define COMPARE_AND_SWAP_VAL(Destination, comparand, exchange) __sync_val_compare_and_swap(Destination, comparand, exchange)

   379

#define THREAD_RETURN_TYPE void*

   380

#define STATE_TYPE volatile int

   381

#endif

   382

   383

"""

   384

   385

        text += self.GenerateIECVars()

   386

   387

        text += """IMPLEMENT_APP_NO_MAIN(SVGViewApp);

   388

SVGViewApp *myapp = NULL;

   389

wxSemaphore MyInitSem;

   390

   391

#ifdef __WXMSW__

   392

HANDLE wxMainLoop;

   393

DWORD wxMainLoopId;

   394

#else

   395

pthread_t wxMainLoop;

   396

#endif

   397

   398

"""

   399

   400

        text += """int myargc = 0;

   401

char** myargv = NULL;

   402

   403

#define UNCHANGED 1

   404

#define PLC_BUSY 2

   405

#define CHANGED 3

   406

#define GUI_BUSY 4

   407

#ifdef __WXMSW__

   408

#else

   409

#endif

   410

   411

bool refresh = false;

   412

bool refreshing = false;

   413

   414

THREAD_RETURN_TYPE InitWxEntry(void* args)

   415

{

   416

  wxEntry(myargc,myargv);

   418

  return 0;

   419

}

   420

   421

"""

   422

   425

{

   426

  #ifndef __WXMSW__

   427

    setlocale(LC_NUMERIC, "C");

   428

  #endif

   429

"""

   430

   431

        text += """  frame = new MainFrame(NULL, wxT("Program"),wxDefaultPosition, wxSize(%d, %d));

   432

  frame->Show();

   433

  myapp = this;

   434

"""%size

   435

        text += """  return true;

   436

}

   437

   438

extern "C" {

   439

   440

int __init_%(location)s(int argc, char** argv)

   441

{

   442

  myargc = argc;

   443

  myargv = argv;

   444

#ifdef __WXMSW__

   445

  wxMainLoop = CreateThread(NULL, 0, InitWxEntry, 0, 0, &wxMainLoopId);

   446

#else

   447

  pthread_create(&wxMainLoop, NULL, InitWxEntry, NULL);

   448

#endif

   449

  MyInitSem.Wait();

   450

  return 0;

   451

}

   452

   453

void __cleanup_%(location)s()

   454

{

   461

}

   462

   463

void __retrieve_%(location)s()

   464

{

   465

  if(myapp){

   466

    myapp->frame->m_svgCtrl->Retrieve();

   467

  }        

   468

}

   469

   470

void __publish_%(location)s()

   471

{

   472

  if(myapp){

   473

    myapp->frame->m_svgCtrl->Publish();

   474

  }

   475

}

   476

   477

}

   478

   479

IEC_STRING wxStringToIEC_STRING(wxString s)

   480

{

   481

  IEC_STRING res = {0,""};

   482

  int i;

   483

  for(i = 0; i<s.Length() && i<STR_MAX_LEN; i++)

   484

    res.body[i] = s.GetChar(i);