1

import os, sys

     2

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

     3

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

     4

     5

from DEFControler import *

     6

from defeditor import *

     7

from FBD_Objects import *

     8

     9

class _EditorFramePlug(EditorFrame):

    10

    """

    11

    This Class add IEC specific features to the SVGUI DEFEditor :

    12

        - FDB preview

    13

        - FBD begin drag 

    14

    """

    15

    def __init__(self,controller):

    17

        self.FbdWindow = wx.Panel(name='fbdwindow',parent=self.EditorPanel,

    18

                                       pos=wx.Point(300, 355),size=wx.Size(240, 240),

    19

                                       style=wx.TAB_TRAVERSAL|wx.SIMPLE_BORDER)

    20

        self.FbdWindow.SetBackgroundColour(wxColour(255,255,255))

    21

        self.FbdWindow.Bind(wx.EVT_LEFT_DOWN, self.OnFbdClick)

    22

        wx.EVT_PAINT(self.FbdWindow,self.OnPaintFBD)

    23

    24

        self.FbdData = None

    25

        self.RefreshProjectTree()

    26

        if (controller.SvgFilepath):

    27

            self.OpenSVGFile(controller.filepath)

    28

            self.mySVGctrl.Refresh()

    29

        self.OnNewFile()

    30

        self.RefreshFBD()

    31

    32

    def SetFbdDragData(self,selected_type):

    33

        self.FbdBlock = FBD_Block(parent=self.FbdWindow,type=selected_type,name='')

    34

        name = self.Controler.GetSelectedElementName()

    35

        self.FbdData = str((selected_type,"functionBlock", name))

    36

    37

    def RefreshFBD(self):

    38

        dc = wx.ClientDC(self.FbdWindow)

    39

        dc.Clear()

    40

        if self.Controler.HasOpenedProject():

    41

            selected_type = self.Controler.GetSelectedElementType()

    42

            if selected_type:

    43

                self.SetFbdDragData(selected_type)

    44

                self.FbdBlock = FBD_Block(parent=self.FbdWindow,type=selected_type,name='')

    45

                width,height = self.FbdBlock.GetMinSize()

    46

                self.FbdBlock.SetSize(width,height)

    47

                clientsize = self.FbdWindow.GetClientSize()

    48

                x = (clientsize.width - width) / 2

    49

                y = (clientsize.height - height) / 2

    50

                self.FbdBlock.SetPosition(x, y)

    51

                self.FbdBlock.Draw(dc)

    52

    53

    def OnPaintFBD(self,event):

    54

        self.RefreshFBD()

    55

        event.Skip()

    56

    57

    def OnFbdClick(self,event):

    58

        if self.FbdData:

    59

            data = wx.TextDataObject(self.FbdData)

    60

            DropSrc = wx.DropSource(self.FbdWindow)

    61

            DropSrc.SetData(data)

    62

            DropSrc.DoDragDrop()

    63

    64

    def OnProjectTreeItemSelected(self,event):

    65

        EditorFrame.OnProjectTreeItemSelected(self,event)

    66

        self.RefreshFBD()

    67

    68

    def OnNew(self,event):

    69

        EditorFrame.OnNew(self,event)

    70

        self.RefreshFBD()

    71

    72

    def OnOpen(self,event):

    73

        EditorFrame.OnOpen(self,event)

    74

        self.RefreshFBD()

    75

    76

    def OnGenerate(self,event):

    77

        self.SaveProject()

    78

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

    79

        event.Skip()    

    80

    81

    def OnClose(self, event):

    83

        event.Skip()

    84

    85

"""

    86

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

    87

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

    88

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

    89

"""

    90

TYPECONVERSION = {"BOOL" : "X", "UINT" : "W","REAL" : "D","STRING" : "B"}

    91

CTYPECONVERSION = {"BOOL" : "bool", "UINT" : "unsigned int", "STRING" : "char*", "REAL" : "float"}

    92

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

    93

class RootClass(DEFControler):

    94

    96

        DEFControler.__init__(self)

    97

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

    98

    99

        if os.path.isfile(filepath):

   100

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

   101

            if os.path.isfile(svgfile):

   102

                self.SvgFilepath = svgfile

   103

            self.OpenXMLFile(filepath)

   104

        else:

   105

            self.CreateRootElement()

   106

            self.SetFilePath(filepath)

   107

   124

    def OnPlugSave(self):

   125

        self.SaveXMLFile()

   126

        return True

   127

   128

    def GenerateProgramHeadersPublicVars(self):

   129

        fct = ""

   130

        fct += "    void OnPlcOutEvent(wxEvent& event);\n\n"

   131

        fct += "    void IN_"+self.BusNumber+"();\n"

   132

        fct += "    void OUT_"+self.BusNumber+"();\n"

   133

        fct += "    void Initialize();\n"

   134

        fct += "    void SetNoChanges();\n"

   135

        fct += "    void Print();\n"

   136

        return fct

   137

   139

        text = ""

   140

        elementsTab = self.GetElementsTab()

   141

        for element in elementsTab:

   142

            infos = element.getElementAttributes()

   143

            for info in infos:

   144

                if info["name"] == "id":

   145

                    element_id = str(info["value"])

   148

        text +="\n"

   149

        #Declaration des variables

   150

        for element in elementsTab:

   151

            infos = element.getElementAttributes()

   152

            for info in infos:

   153

                if info["name"] == "id":

   154

                    element_id = str(info["value"])

   155

            type = element.GetElementInfos()["type"]

   156

            FbdBlock = self.GetBlockType(type)

   157

            element_num_patte = 1

   158

            for input in FbdBlock["inputs"]:

   159

                element_type = TYPECONVERSION[input[1]]

   160

                element_c_type = CTYPECONVERSION[input[1]]

   164

                element_num_patte +=1

   165

            element_num_patte = 1

   166

            for output in FbdBlock["outputs"]:

   167

                element_type = TYPECONVERSION[output[1]]

   168

                element_c_type = CTYPECONVERSION[output[1]]

   169

   173

                element_num_patte +=1

   174

            text +="\n"

   175

        return text

   176

   177

    def GenerateGlobalVarsAndFuncs(self):

   178

        text = ""

   183

        text += "IMPLEMENT_APP_NO_MAIN(SVGViewApp);\n"

   184

        text += "IMPLEMENT_WX_THEME_SUPPORT;\n"

   185

        text += "SVGViewApp *myapp;\n"

   186

        text += "pthread_t wxMainLoop,automate;\n"

   187

        text += "int myargc;\n"

   188

        text += "char** myargv;\n\n"

   189

   190

        text += "#define FREE_AND_NO_CHANGES 1 \n"

   191

        text += "#define PLC_BUSY 2 \n"

   192

        text += "#define FREE_AND_CHANGES 3 \n"

   193

        text += "#define PLC_OUT_BUSY 4 \n\n"

   194

   195

   196

        text += "void* InitWxEntry(void* args)\n{\n"

   197

        text += "  wxEntry(myargc,myargv);\n"

   198

        text += "  return args;\n"

   199

        text += "}\n\n"

   200

   201

        text += "void* SimulAutomate(void* args)\n{\n"

   202

        text += "  while(1){\n"

   203

        text += "    myapp->frame->m_svgCtrl->IN_"+self.BusNumber+"();\n"

   204

        text += "    //printf(\"AUTOMATE\\n\");\n"

   205

        text += "    myapp->frame->m_svgCtrl->OUT_"+self.BusNumber+"();\n"

   206

        text += "    sleep(1);\n"

   207

        text += "  }\n"

   208

        text += "  return args;\n"

   209

        text += "}\n\n"

   210

   211

        if (self.SVGUIRootElement):

   212

            width = self.SVGUIRootElement.GetBBox().GetWidth()

   213

            height = self.SVGUIRootElement.GetBBox().GetHeight()

   214

        else :

   215

            width = 250

   216

            height = 350

   217

        text += "bool SVGViewApp::OnInit()\n{\n"

   218

        text += "  #ifndef __WXMSW__\n"

   219

        text += "    setlocale(LC_NUMERIC, \"C\");\n"

   220

        text += "  #endif\n"

   221

        text += "  frame = new MainFrame(NULL, wxT(\"Program\"),wxDefaultPosition, wxSize((int)"+str(width)+", (int)"+str(height)+"));\n"

   222

        text += "  myapp = this;\n"

   223

        text += "  pthread_create(&automate, NULL, SimulAutomate, NULL);\n"

   224

        text += "  return true;\n"

   225

        text += "}\n\n"

   226

   227

        text += "int main(int argc, char** argv)\n{\n"

   228

        text += "  myargc = argc;\n"

   229

        text += "  myargv = argv;\n"

   230

        text += "  pthread_create(&wxMainLoop, NULL, InitWxEntry, NULL);\n"

   231

        text += "  pause();\n"

   232

        text += "}\n\n"

   233

   234

        return text

   235

   236

    def GenerateProgramEventTable(self):

   237

        evt = ""        

   238

        elementsTab = self.GetElementsTab()

   239

        #evt += "wxEVT_PLCOUT = wxNewEventType();\n\n";

   240

        evt += "BEGIN_DECLARE_EVENT_TYPES()\n"

   241

        evt += "DECLARE_LOCAL_EVENT_TYPE( EVT_PLC, wxNewEventType() )\n"

   242

        evt += "END_DECLARE_EVENT_TYPES()\n\n"

   243

   244

        evt += "DEFINE_LOCAL_EVENT_TYPE( EVT_PLC )\n\n"

   245

        #Event Table Declaration

   246

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

   247

        for element in elementsTab:

   248

            infos = element.getElementAttributes()

   249

            for info in infos:

   250

                if info["name"] == "id":

   251

                    element_id = str(info["value"])

   252

                if info["name"] == "name":

   253

                    element_name = str(info["value"])

   254

            type = element.GetElementInfos()["type"]

   255

            if type == "Button":

   256

                evt += "  EVT_BUTTON (SVGUIID(\""+element_id+"\"), Program::On"+element_name+"Click)\n"

   257

            elif type == "ScrollBar":

   258

                pass

   259

                #evt += "  EVT_LEFT_UP (Program::OnClick)\n"

   260

                #evt += "  EVT_COMMAND_SCROLL_THUMBTRACK (SVGUIID(\""+element_id+"\"), Program::On"+element_name+"Changed)\n"

   261

            elif type == "RotatingCtrl":

   262

                evt += "  EVT_COMMAND_SCROLL_THUMBTRACK (SVGUIID(\""+element_id+"\"), Program::On"+element_name+"Changed)\n"

   263

            elif type == "NoteBook":

   264

                evt += "  EVT_NOTEBOOK_PAGE_CHANGED (SVGUIID(\""+element_id+"\"), Program::On"+element_name+"TabChanged)\n"

   265

            elif type == "Container" or type == "Transform":

   266

                evt += "  EVT_PAINT(Program::On"+element_name+"Paint)\n"

   267

        evt += "  EVT_LEFT_UP (Program::OnClick)\n"

   268

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

   269

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

   270

        return evt

   271

   272

    def GenerateProgramInitFrame(self):

   273

        text = "MainFrame::MainFrame(wxWindow *parent, const wxString& title, const wxPoint& pos,const wxSize& size, long style): wxFrame(parent, wxID_ANY, title, pos, size, style)\n{\n"

   274

        text += "  m_svgCtrl = new Program(this);\n"

   275

        text += "  if (m_svgCtrl->LoadFiles(wxT(\""+self.SvgFilepath+"\"), wxT(\""+self.filepath+"\")))\n"

   276

        text += "  {\n"

   277

        text += "    Show(true);\n"

   278

        text += "    m_svgCtrl->SetFocus();\n"

   279

        text += "    m_svgCtrl->SetFitToFrame(true);\n"

   280

        text += "    m_svgCtrl->RefreshScale();\n"

   281

        text += "    m_svgCtrl->InitScrollBars();\n"

   282

        text += "    m_svgCtrl->Initialize();\n"

   283

        text += "    m_svgCtrl->Update();\n"

   284

        text += "    //m_svgCtrl->Print();\n"

   285

        text += "  }\n"

   286

        text += "  else\n"

   287

        text += "  {\n"

   288

        text += "    printf(\"Error while opening files\\n\");\n"

   289

        text += "    exit(0);\n"

   290

        text += "  }\n"

   291

        text += "}\n\n\n"

   292

        return text

   293

   294

    def GenerateProgramInitProgram(self):

   295

        elementsTab = self.GetElementsTab()

   296

        text = "Program::Program(wxWindow* parent):SVGUIWindow(parent)\n{\n"

   297

        for element in elementsTab:

   298

            infos = element.getElementAttributes()

   299

            for info in infos:

   300

                if info["name"] == "id":

   301

                    element_id = str(info["value"])

   302

            text += "    flag_"+element_id+" = true;\n"

   303

            text += "    step_"+element_id+" = FREE_AND_NO_CHANGES;\n"

   304

        text += "}\n\n"

   305

        return text

   306

   307

    def GenerateProgramEventFunctions(self):

   308

        fct=""

   309

        elementsTab = self.GetElementsTab()

   310

        for element in elementsTab:

   311

            infos = element.getElementAttributes()

   312

            for info in infos:

   313

                if info["name"] == "id":

   314

                    element_id = str(info["value"])

   315

                if info["name"] == "name":

   316

                    element_name = str(info["value"])

   317

            type = element.GetElementInfos()["type"]

   318

            FbdBlock = self.GetBlockType(type)

   319

            if type == "Button":

   320

                fct += "void Program::On"+element_name+"Click(wxCommandEvent& event)\n{\n"

   321

                fct += "  if (flag_"+element_id+")\n  {\n"

   322

                fct += "    flag_"+element_id+" = false;\n"

   323

                element_num_patte = 1

   324

                for output in FbdBlock["outputs"]:

   325

                    element_type = TYPECONVERSION[output[1]]

   327

                    element_num_patte +=1

   328

                fct += "    flag_"+element_id+" = true;\n"

   329

                fct += "  }\n"

   330

                fct += "  event.Skip();\n"

   331

                fct += "}\n\n"               

   332

   333

            elif type == "RotatingCtrl":

   334

                fct += "void Program::On"+element_name+"Changed(wxScrollEvent& event)\n{\n"

   335

                fct += "  SVGUIRotatingCtrl* rotating = (SVGUIRotatingCtrl*)GetElementById(wxT(\""+element_id+"\"));\n"

   336

                fct += "  rotating->SendScrollEvent(event);\n"

   337

                fct += "  double angle = rotating->GetAngle();\n"

   338

                fct += "  if (flag_"+element_id+")\n  {\n"

   339

                fct += "    flag_"+element_id+" = false;\n"

   340

                element_num_patte = 1

   341

                for output in FbdBlock["outputs"]:

   342

                    element_type = TYPECONVERSION[output[1]]

   343

   344

                    if element_num_patte == 1:

   345

                        value = "angle"

   346

                    elif element_num_patte == 2:

   347

                        value = "true"

   349

                    element_num_patte +=1

   350

                fct += "    flag_"+element_id+" = true;\n"

   351

                fct += "  }\n"

   352

                fct += "}\n\n"

   353

            elif type == "NoteBook":

   354

                fct += "void Program::On"+element_name+"TabChanged(wxNotebookEvent& event)\n{\n"

   355

                fct += "  SVGUINoteBook* notebook = (SVGUINoteBook*)GetElementById(wxT(\""+element_id+"\"));\n"

   356

                fct += "  notebook->SendNotebookEvent(event);\n"

   357

                fct += "  unsigned int selected = notebook->GetCurrentPage();\n"

   358

                fct += "  if (flag_"+element_id+")\n  {\n"

   359

                fct += "    flag_"+element_id+" = false;\n"

   360

                element_num_patte = 1

   361

                for output in FbdBlock["outputs"]:

   362

                    element_type = TYPECONVERSION[output[1]]

   363

   364

                    if element_num_patte == 1:

   365

                        value = "selected"

   366

                    elif element_num_patte == 2:

   367

                        value = "true"

   369

                    element_num_patte +=1

   370

                fct += "    flag_"+element_id+" = true;\n"

   371

                fct += "  }\n"

   372

                fct += "}\n\n"

   373

            elif type == "Transform":

   374

                fct += "void Program::On"+element_name+"Paint(wxPaintEvent& event)\n{\n"

   375

                fct += "  SVGUITransform* transform = (SVGUITransform*)GetElementById(wxT(\""+element_id+"\"));\n"

   376

                fct += "  if (flag_"+element_id+")\n  {\n"

   377

                fct += "    flag_"+element_id+" = false;\n"

   378

                element_num_patte = 1

   379

                for output in FbdBlock["outputs"]:                    

   380

                    if element_num_patte == 1:

   382

                        fct += "    {\n"

   385

                        fct += "    }\n"

   386

                    elif element_num_patte == 2:

   388

                        fct += "    {\n"

   391

                        fct += "    }\n"

   392

                    elif element_num_patte == 3:

   394

                        fct += "    {\n"

   397

                        fct += "    }\n"

   398

                    elif element_num_patte == 4:

   400

                        fct += "    {\n"

   403

                        fct += "    }\n"

   404

                    elif element_num_patte == 5:

   406

                        fct += "    {\n"

   409

                        fct += "    }\n"

   410

                    element_num_patte +=1

   411

                fct += "    flag_"+element_id+" = true;\n"

   412

                fct += "  }\n"

   413

                fct += "  event.Skip();\n"

   414

                fct += "}\n\n"

   415

            elif type == "Container":

   416

                fct += "void Program::On"+element_name+"Paint(wxPaintEvent& event)\n{\n"

   417

                fct += "  SVGUIContainer* container = (SVGUIContainer*)GetElementById(wxT(\""+element_id+"\"));\n"

   419

                fct += "  {\n"

   420

                fct += "    flag_"+element_id+" = false;\n"

   423

                fct += "    flag_"+element_id+" = true;\n"

   424

                fct += "  }\n"

   425

                fct += "  event.Skip();\n"

   426

                fct += "}\n\n"

   427

   428

        fct += "void Program::OnChar(wxKeyEvent& event)\n{\n"

   429

        fct += "  SVGUIContainer* container = GetSVGUIRootElement();\n"

   430

        fct += "  if (container->GetFocusedElementName() == wxT(\"TextCtrl\"))\n"

   431

        fct += "  {\n"

   432

        fct += "    wxString focusedId = container->GetFocusedElement();\n"

   433

        fct += "    SVGUITextCtrl* text = (SVGUITextCtrl*)GetElementById(container->GetFocusedElement());\n"

   434

        fct += "    text->SendKeyEvent(event);\n"

   435

        for element in elementsTab:

   436

            infos = element.getElementAttributes()

   437

            for info in infos:

   438

                if info["name"] == "id":

   439

                    element_id = str(info["value"])

   440

            type = element.GetElementInfos()["type"]

   441

            FbdBlock = self.GetBlockType(type)

   442

            if type == "TextCtrl":

   443

                fct += "    if (focusedId == wxT(\""+element_id+"\") && flag_"+element_id+")\n"

   444

                fct += "    {\n"

   445

                fct += "      flag_"+element_id+" = false;\n"

   448

                fct += "      flag_"+element_id+" = true;\n"

   449

                fct += "    }\n"

   450

        fct += "  }\n"

   451

        fct += "}\n"

   452

   453

   454

   455

        fct += "void Program::OnClick(wxMouseEvent& event)\n{\n"

   456

        fct += "  SVGUIContainer* container = GetSVGUIRootElement();\n"

   457

        fct += "  if (container->GetFocusedElementName() == wxT(\"ScrollBar\"))\n"

   458

        fct += "  {\n"

   459

        fct += "    wxString focusedId = container->GetFocusedElement();\n"

   460

        fct += "    SVGUIScrollBar* scrollbar = (SVGUIScrollBar*)GetElementById(focusedId);\n"

   461

        fct += "    scrollbar->SendMouseEvent(event);\n"

   462

        for element in elementsTab:

   463

            infos = element.getElementAttributes()

   464

            for info in infos:

   465

                if info["name"] == "id":

   466

                    element_id = str(info["value"])

   467

            type = element.GetElementInfos()["type"]

   468

            FbdBlock = self.GetBlockType(type)

   469

            if type == "ScrollBar":

   470

                fct += "    if (focusedId == wxT(\""+element_id+"\") && flag_"+element_id+")\n"

   471

                fct += "    {\n"

   472

                fct += "      flag_"+element_id+" = false;\n"

   473

                fct += "      unsigned int scrollPos = scrollbar->GetThumbPosition();\n"

   476

                fct += "      flag_"+element_id+" = true;\n"

   477

                fct += "    }\n"

   478

        fct += "  }\n"

   479

        fct += "  event.Skip();\n"

   480

        fct += "}\n"

   481

   482

   483

   484

   485

        fct += "void Program::OnPlcOutEvent(wxEvent& event)\n{\n"

   486

        fct += "  int old_state;\n"

   487

        for element in elementsTab:

   488

            infos = element.getElementAttributes()

   489

            for info in infos:

   490

                if info["name"] == "id":

   491

                    element_id = str(info["value"])

   492

            type = element.GetElementInfos()["type"]