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

   120

        return True

   121

   122

    def GetElementIdFromName(self, name):

   123

        element = self.GetElementByName(name)

   124

        if element is not None:

   125

            return element.getid()

   126

        return None

   127

   128

    _View = None

   129

    def _OpenView(self, logger):

   130

        if not self._View:

   131

            def _onclose():

   132

                self._View = None

   133

            def _onsave():

   134

                self.GetPlugRoot().SaveProject()

   135

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

   136

            self._View._onclose = _onclose

   137

            self._View._onsave = _onsave

   138

            self._View.Show()

   139

   140

    def _ImportSVG(self, logger):

   141

        if not self._View:

   142

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

   143

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

   144

                svgpath = dialog.GetPath()

   145

                if os.path.isfile(svgpath):

   146

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

   147

                else:

   148

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

   149

            dialog.Destroy()

   150

   151

    def _ImportXML(self, logger):

   152

        if not self._View:

   153

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

   154

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

   155

                xmlpath = dialog.GetPath()

   156

                if os.path.isfile(xmlpath):

   157

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

   158

                else:

   159

                    logger.write_error("No such XML file: %s\n"%xmlpath)

   160

            dialog.Destroy()

   161

   162

    PluginMethods = [

   163

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

   164

         "name" : "HMI Editor",

   165

         "tooltip" : "HMI Editor",

   166

         "method" : "_OpenView"},

   167

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

   168

         "name" : "Import SVG",

   169

         "tooltip" : "Import SVG",

   170

         "method" : "_ImportSVG"},

   171

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

   172

         "name" : "Import XML",

   173

         "tooltip" : "Import XML",

   174

         "method" : "_ImportXML"},

   175

    ]

   176

   177

    def OnPlugSave(self):

   178

        self.SaveXMLFile()

   179

        return True

   180

   181

    def PlugGenerate_C(self, buildpath, locations, logger):

   182

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

   183

   184

        doc = SVGDocument(self.GetSVGFilePath())

   185

        root_element = doc.GetRootElement()

   186

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

   187

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

   188

   198

        generator.GenerateProgram(window_size, buildpath, progname)

   199

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

   200

   201

        if wx.Platform == '__WXMSW__':

   202

            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"

   203

            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"

   204

        else:

   205

            status, result, err_result = ProcessLogger(logger, "wx-config --cxxflags", no_stdout=True).spin()

   206

            if status:

   207

                logger.write_error("Unable to get wx cxxflags\n")

   208

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

   209

   210

            status, result, err_result = ProcessLogger(logger, "wx-config --libs", no_stdout=True).spin()

   211

            if status:

   212

                logger.write_error("Unable to get wx libs\n")

   213

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

   214

   215

        return [(Gen_C_file, cxx_flags)],libs,True

   216

   217

    def BlockTypesFactory(self):

   218

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

   219

            name = block.getinstanceName()

   220

            block_id = self.GetElementIdFromName(name)

   221

            if block_id == None:

   222

                raise ValueError, "No corresponding block found"

   223

            type = block.gettypeName()

   224

            block_infos = GetBlockType(type)

   225

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

   226

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

   227

                generator.ComputedBlocks[block] = True

   228

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

   229

                    connections = variable.connectionPointIn.getconnections()

   230

                    if connections and len(connections) == 1:

   231

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

   232

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

   233

                        generator.Program += ("  %s := %s;\n"%(parameter, generator.ExtractModifier(variable, value)))

   234

            if link:

   235

                connectionPoint = link.getposition()[-1]

   236

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

   237

                    blockPointx, blockPointy = variable.connectionPointOut.getrelPositionXY()

   238

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

   239

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

   240

                raise ValueError, "No output variable found"

   241

            else:

   242

                return None

   243

   244

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

   245

            block_id = self.GetElementIdFromName(name)

   246

            if block_id == None:

   247

                raise ValueError, "No corresponding block found"

   248

            block_infos = GetBlockType(type)

   249

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

   250

            variables = []

   251

            if block is not None:

   252

                input_variables = block.inputVariables.getvariable()

   253

                output_variables = block.outputVariables.getvariable()

   254

            else:

   255

                input_variables = None

   256

                output_variables = None

   257

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

   258

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

   259

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

   260

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

   261

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

   262

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

   263

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

   264

            return variables

   265

   266

        return [{"name" : "SVGUI function blocks", "list" :

   267

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

   268

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

   269

                    "outputs" : [],

   270

                    "comment" : "SVGUI Container",

   271

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

   272

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

   273

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

   274

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

   275

                    "comment" : "SVGUI Button",

   276

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

   277

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

   278

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

   279

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

   280

                    "comment" : "SVGUI Text Control",

   281

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

   282

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

   283

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

   284

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

   285

                    "comment" : "SVGUI ScrollBar",

   286

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

   287

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

   288

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

   289

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

   290

                    "comment" : "SVGUI Notebook",

   291

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

   292

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

   293

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

   294

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

   295

                    "comment" : "SVGUI Rotating Control",

   296

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

   297

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

   298

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

   299

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

   300

                    "comment" : "SVGUI Transform",

   301

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

   302

               ]}

   303

        ]

   304

   305

   306

   307

class _SVGUICGenerator(SVGUICGenerator):

   308

   309

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

   310

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

   311

   312

        self.CurrentLocation = current_location

   313

   314

    def GenerateProgramHeadersPublicVars(self):

   315

        text = """    void OnPlcOutEvent(wxEvent& event);

   316

   317

    void Retrieve();

   318

    void Publish();

   319

    void Initialize();

   320

"""

   321

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

   322

        return text

   323

   324

    def GenerateIECVars(self):

   325

        text = ""

   326

        for element in self.Elements:

   327

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

   328

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

   329

        text +="\n"

   330

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

   331

        #Declaration des variables

   332

        for element in self.Elements:

   333

            block_infos = GetBlockType(SVGUIFB_Types[GetElementType(element)])

   334

            block_id = element.getid()

   335

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

   336

                element_c_type = CTYPECONVERSION[input[1]]

   337

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

   338

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

   339

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

   340

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

   341

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

   342

                element_c_type = CTYPECONVERSION[output[1]]

   343

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

   344

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

   345

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

   346

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

   347

            text +="\n"

   348

        return text

   349

   350

    def GenerateGlobalVarsAndFuncs(self, size):

   351

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

   352

#ifdef __WXMSW__

   353

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

   354

#define THREAD_RETURN_TYPE DWORD WINAPI

   355

#define STATE_TYPE long int

   356

#else

   357

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

   358

#define THREAD_RETURN_TYPE void*

   359

#define STATE_TYPE volatile int

   360

#endif

   361

   362

"""

   363

   364

        text += self.GenerateIECVars()

   365

   366

        text += """IMPLEMENT_APP_NO_MAIN(SVGViewApp);

   367

IMPLEMENT_WX_THEME_SUPPORT;

   368

SVGViewApp *myapp = NULL;

   369

wxSemaphore MyInitSem;

   370

   371

#ifdef __WXMSW__

   372

HANDLE wxMainLoop;

   373

DWORD wxMainLoopId;

   374

#else

   375

pthread_t wxMainLoop;

   376

#endif

   377

   378

"""

   379

   380

        text += """int myargc = 0;

   381

char** myargv = NULL;

   382

   383

#define UNCHANGED 1

   384

#define PLC_BUSY 2

   385

#define CHANGED 3

   386

#define GUI_BUSY 4

   387

#ifdef __WXMSW__

   388

#else

   389

#endif

   390

   391

bool refresh = false;

   392

bool refreshing = false;

   393

   394

THREAD_RETURN_TYPE InitWxEntry(void* args)

   395

{

   396

  wxEntry(myargc,myargv);

   397

  return 0;

   398

}

   399

   400

"""

   401

   402

        text += """bool SVGViewApp::OnInit()

   403

{

   404

  #ifndef __WXMSW__

   405

    setlocale(LC_NUMERIC, "C");

   406

  #endif

   407

"""

   408

   409

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

   410

  frame->Show();

   411

  myapp = this;

   412

"""%size

   413

        text += """  return true;

   414

}

   415

   416

extern "C" {

   417

   418

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

   419

{

   420

  myargc = argc;

   421

  myargv = argv;

   422

#ifdef __WXMSW__

   423

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

   424

#else

   425

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

   426

#endif

   427

  MyInitSem.Wait();

   428

  return 0;

   429

}

   430

   431

void __cleanup_%(location)s()

   432

{

   433

}

   434

   435

void __retrieve_%(location)s()

   436

{

   437

  if(myapp){

   438

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

   439

  }        

   440

}

   441

   442

void __publish_%(location)s()

   443

{

   444

  if(myapp){

   445

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

   446

  }

   447

}

   448

   449

}

   450

   451

IEC_STRING wxStringToIEC_STRING(wxString s)

   452

{

   453

  IEC_STRING res = {0,""};

   454

  int i;

   455

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

   456

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

   457

  res.len = i;

   458

  return res;

   459

}

   460

   461

"""%{"location" : "_".join(map(str, self.CurrentLocation))}

   462

   463

        return text

   464

   465

    def GenerateProgramEventTable(self):

   466

        text = """BEGIN_DECLARE_EVENT_TYPES()

   467

DECLARE_LOCAL_EVENT_TYPE( EVT_PLC, wxNewEventType() )

   468

END_DECLARE_EVENT_TYPES()

   469

   470

DEFINE_LOCAL_EVENT_TYPE( EVT_PLC )

   471

   472

"""     

   473

        #Event Table Declaration

   474

        text += "BEGIN_EVENT_TABLE(Program, SVGUIWindow)\n"

   475

        for element in self.Elements:

   476

            element_type = GetElementType(element)

   477

            element_name = element.getname()

   478

            if element_type == ITEM_BUTTON:

   479

                text += "  EVT_BUTTON (SVGUIID(\"%s\"), Program::On%sClick)\n"%(element_name, element_name)

   480

            elif element_type in [ITEM_SCROLLBAR, ITEM_ROTATING, ITEM_TRANSFORM]:

   481

                text += "  EVT_COMMAND_SCROLL_THUMBTRACK (SVGUIID(\"%s\"), Program::On%sChanging)\n"%(element_name, element_name)

   482

            elif element_type == ITEM_NOTEBOOK:

   483

                text += "  EVT_NOTEBOOK_PAGE_CHANGED (SVGUIID(\"%s\"), Program::On%sTabChanged)\n"%(element_name, element_name)

   484

        text += "  EVT_CUSTOM(EVT_PLC, wxID_ANY, Program::OnPlcOutEvent)\n"

   485

        text += "END_EVENT_TABLE()\n\n"

   486

        return text

   487

   488