1

import os, shutil, sys

     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 SVGUIControler import *

     9

from SVGUIEditor import *

    12

from wxPopen import ProcessLogger

    13

    14

[ID_SVGUIEDITORFBDPANEL, 

    15

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

    16

    17

SVGUIFB_Types = {ITEM_CONTAINER : "Container",

    18

                 ITEM_BUTTON : "Button", 

    19

                 ITEM_TEXT : "TextCtrl", 

    20

                 ITEM_SCROLLBAR : "ScrollBar", 

    21

                 ITEM_ROTATING : "RotatingCtrl", 

    22

                 ITEM_NOTEBOOK : "NoteBook", 

    23

                 ITEM_TRANSFORM : "Transform"}

    24

    25

class _SVGUIEditor(SVGUIEditor):

    27

    This Class add IEC specific features to the SVGUIEditor :

    31

    32

    def _init_coll_EditorGridSizer_Items(self, parent):

    33

        SVGUIEditor._init_coll_EditorGridSizer_Items(self, parent)

    34

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

    35

    36

    def _init_ctrls(self, prnt):

    37

        SVGUIEditor._init_ctrls(self, prnt, False)

    38

    39

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

    40

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

    41

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

    42

        self.FBDPanel.SetBackgroundColour(wx.WHITE)

    43

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

    44

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

    45

    46

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

    47

    48

        self._init_sizers()

    49

    50

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

    51

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

    52

    53

        self.FBDBlock = None

    54

    55

    def RefreshView(self, select_id = None):

    56

        SVGUIEditor.RefreshView(self, select_id)

    57

        self.FBDPanel.Refresh()

    58

    59

    def OnPaintFBDPanel(self,event):

    60

        dc = wx.ClientDC(self.FBDPanel)

    62

        selected = self.GetSelected()

    63

        if selected is not None:

    64

            selected_type = self.Controler.GetElementType(selected)

    65

            if selected_type is not None:

    66

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

    67

                width, height = self.FBDBlock.GetMinSize()

    68

                self.FBDBlock.SetSize(width,height)

    69

                clientsize = self.FBDPanel.GetClientSize()

    72

                self.FBDBlock.SetPosition(x, y)

    73

                self.FBDBlock.Draw(dc)

    74

        else:

    75

            self.FBDBlock = None

    78

    def OnFBDPanelClick(self, event):

    79

        if self.FBDBlock:

    80

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

    81

            DropSrc = wx.DropSource(self.FBDPanel)

    84

        event.Skip()

    85

    86

    def OnInterfaceTreeItemSelected(self, event):

    87

        self.FBDPanel.Refresh()

    88

        SVGUIEditor.OnInterfaceTreeItemSelected(self, event)

    89

    98

   101

   102

class RootClass(SVGUIControler):

   105

        SVGUIControler.__init__(self)

   106

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

   114

            self.CreateNewInterface()

   117

    def GetElementIdFromName(self, name):

   118

        element = self.GetElementByName(name)

   119

        if element is not None:

   120

            return element.getid()

   121

        return None

   122

   130

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

   135

    def _ImportSVG(self, logger):

   136

        if not self._View:

   137

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

   138

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

   139

                svgpath = dialog.GetPath()

   140

                if os.path.isfile(svgpath):

   141

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

   142

                else:

   143

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

   144

            dialog.Destroy()

   145

   146

    def _ImportXML(self, logger):

   147

        if not self._View:

   148

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

   149

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

   150

                xmlpath = dialog.GetPath()

   151

                if os.path.isfile(xmlpath):

   152

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

   153

                else:

   154

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

   155

            dialog.Destroy()

   156

   165

         "method" : "_ImportSVG"},

   167

         "name" : "Import XML",

   168

         "tooltip" : "Import XML",

   169

         "method" : "_InportXML"},

   176

    def GenerateProgramHeadersPublicVars(self, elements):

   177

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

   178

   179

    void Retrieve();

   180

    void Publish();

   181

    void Initialize();

   182

"""

   183

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

   184

        return text

   185

   186

    def GenerateIECVars(self, elements):

   188

        for element in elements:

   189

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

   190

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

   192

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

   194

        for element in elements:

   195

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

   196

            block_id = element.getid()

   197

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

   199

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

   200

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

   201

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

   202

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

   204

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

   205

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

   206

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

   210

    def GenerateGlobalVarsAndFuncs(self, elements, size):

   211

        text = "#include \"iec_std_lib.h\"\n\n"

   212

   213

        text += self.GenerateIECVars(elements)

   214

   215

        text += """IMPLEMENT_APP_NO_MAIN(SVGViewApp);

   216

IMPLEMENT_WX_THEME_SUPPORT;

   217

SVGViewApp *myapp = NULL;

   218

pthread_t wxMainLoop;

   219

"""

   221

        text += """int myargc = 0;

   222

char** myargv = NULL;

   223

   224

#define UNCHANGED 1

   225

#define PLC_BUSY 2

   226

#define CHANGED 3

   227

#define GUI_BUSY 4

   228

   229

void* InitWxEntry(void* args)

   230

{

   231

  wxEntry(myargc,myargv);

   232

  return args;

   233

}

   234

   235

"""

   236

#        text += """void* SimulAutomate(void* args)

   237

#{

   238

#  while(1){

   239

#    myapp->frame->m_svgCtrl->IN_"+self.BusNumber+"();

   240

#    //printf(\"AUTOMATE\\n\");

   241

#    myapp->frame->m_svgCtrl->OUT_"+self.BusNumber+"();

   242

#    sleep(1);

   243

#  }

   244

#  return args;

   245

#}

   246

#

   247

#"""

   248

   249

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

   250

{

   251

  #ifndef __WXMSW__

   252

    setlocale(LC_NUMERIC, "C");

   253

  #endif

   254

"""

   255

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

   256

        text += """  frame = new MainFrame(NULL, wxT("Program"),wxDefaultPosition, wxDefaultSize);

   257

  myapp = this;

   258

"""

   260

        text += """  return true;

   261

}

   262

   263

extern "C" {

   264

   265

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

   266

{

   267

  myargc = argc;

   268

  myargv = argv;

   269

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

   270

}

   271

   272

void __cleanup_%(location)s()

   273

{

   274

}

   275

   276

void __retrieve_%(location)s()

   277

{

   278

  if(myapp){

   279

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

   280

  }        

   281

}

   282

   283

void __publish_%(location)s()

   284

{

   285

  if(myapp){

   286

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

   287

  }

   288

}

   289

   290

}

   291

   292

IEC_STRING wxStringToIEC_STRING(wxString s)

   293

{

   294

  STRING res = {0,""};

   295

  int i;

   296

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

   297

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

   298

  res.len = i;

   299

  return res;

   300

}

   301

   302

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

   306

    def GenerateProgramEventTable(self, elements):

   307

        text = ""        

   308

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

   309

   310

        text += """BEGIN_DECLARE_EVENT_TYPES()

   311

DECLARE_LOCAL_EVENT_TYPE( EVT_PLC, wxNewEventType() )

   312

END_DECLARE_EVENT_TYPES()

   314

DEFINE_LOCAL_EVENT_TYPE( EVT_PLC )

   315

   316

"""     

   318

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

   319

        for element in elements:

   320

            element_type = GetElementType(element)

   321

            element_name = element.getname()

   322

            if element_type == ITEM_BUTTON:

   323

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

   324

            elif element_type in [ITEM_SCROLLBAR, ITEM_ROTATING]:

   325

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

   326

            elif element_type == ITEM_NOTEBOOK:

   327

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

   328

##            elif element_type in [ITEM_CONTAINER, ITEM_TRANSFORM]:

   329

##                text += "  EVT_PAINT(Program::On%sPaint)\n"%element_name

   330

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

   331

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

   332

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

   335

    def GenerateProgramInitFrame(self, elements):

   336

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

   337

{

   338

  m_svgCtrl = new Program(this);

   339

  if (m_svgCtrl->LoadFiles(wxT("%s"), wxT("%s")))

   340

  {

   341

    Show(true);

   342

    m_svgCtrl->SetFocus();

   343

    m_svgCtrl->SetFitToFrame(true);

   344

    m_svgCtrl->RefreshScale();

   345

    m_svgCtrl->InitScrollBars();

   346

    m_svgCtrl->Initialize();

   347

    m_svgCtrl->Update();

   348

    //m_svgCtrl->Print();

   349

  }

   350

  else

   351

  {

   352

    printf("Error while opening files\\n");

   353

    exit(0);

   354

  }

   355

}

   356

   357

   358

"""%(self.GetSVGFilePath(), self.GetFilePath())

   359

   360

        return text

   361

   362

    def GenerateProgramInitProgram(self, elements):

   364

        for element in elements:

   365

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

   366

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

   370

    def GenerateProgramEventFunctions(self, elements):

   371

        text = ""

   372

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

   373

        for element in elements:

   374

            element_type = GetElementType(element)

   375

            element_state = "  in_state_%d = %s;\n"%(element.getid(), "%s")

   376

            element_name = element.getname()

   377

   378

            block_infos = GetBlockType(SVGUIFB_Types[element_type])

   379

            if element_type == ITEM_BUTTON:

   380

                text += """void Program::On%sClick(wxCommandEvent& event)

   381

{

   382

  SVGUIButton* button = (SVGUIButton*)GetElementByName(wxT("%s"));\n"""%(element_name, element_name)

   383

                text += element_state%"GUI_BUSY"

   384

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

   386

                    if i == 0:

   387

                        value = "button->IsVisible()"

   388

                    elif i == 1:

   389

                        value = "button->GetToggle()"

   390

                    else:

   391

                        value = "0"

   392

                    text += "  _copy__I%s%s_%d_%d = %s;\n"%(TYPECONVERSION[output[1]], current_location, element.getid(), i + 1, value)

   393

                text += element_state%"CHANGED"

   394

                text += "  event.Skip();\n}\n\n"

   395

            elif element_type == ITEM_ROTATING:

   396

                text += """void Program::On%sChanged(wxScrollEvent& event)

   397

{

   398

  SVGUIRotatingCtrl* rotating = (SVGUIRotatingCtrl*)GetElementByName(wxT("%s"));

   399

  rotating->SendScrollEvent(event);

   400

  double angle = rotating->GetAngle();

   401

"""%(element_name, element_name)

   402

                text += element_state%"GUI_BUSY"

   403

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

   404

                    text += "  _copy__I%s%s_%d_%d = %s;\n"%(TYPECONVERSION[output[1]], current_location, element.getid(), i + 1, ["angle", "true"][value])

   405

                text += element_state%"CHANGED"

   406

                text += "  event.Skip();\n}\n\n"

   407

            elif element_type == ITEM_NOTEBOOK:

   408

                text += """void Program::On%sTabChanged(wxNotebookEvent& event)

   409

{

   410

  SVGUINoteBook* notebook = (SVGUINoteBook*)GetElementByName(wxT("%s"));

   411

  notebook->SendNotebookEvent(event);

   412

  unsigned int selected = notebook->GetCurrentPage();

   413

"""%(element_name, element_name)

   414

                text += element_state%"GUI_BUSY"

   415

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

   416

                    text += "  _copy__I%s%s_%d_%d = %s;\n"%(TYPECONVERSION[output[1]], current_location, element.getid(), i + 1, ["selected", "true"][value])

   417

                text += element_state%"CHANGED"

   418

                text += "  event.Skip();\n}\n\n"

   419

            elif element_type == ITEM_TRANSFORM:

   420

                text += """void Program::On%sPaint(wxPaintEvent& event)

   421

{

   422

  SVGUITransform* transform = (SVGUITransform*)GetElementByName(wxT("%s"));

   423

"""%(element_name, element_name)

   424

                text += element_state%"GUI_BUSY"

   425

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

   426

                    if i < 5:

   427

                        texts = {"location" : current_location, "id" : element.getid(), 

   428

                                 "pin" : i + 1, "param" : ["X", "Y", "XScale", "YScale", "Angle"][i]}

   429

   430

                        text += """  if (transform->Get%(param)s() != _copy__ID%(location)s_%(id)d_%(pin)d)

   431

  {

   432

    _copy__ID%(location)s_%(id)d_%(pin)d = transform->Get%(param)s();

   433

    _copy__IX%(location)s_%(id)d_6 = true;

   434

  }

   435

"""%texts

   436

                text += element_state%"CHANGED"

   437

                text += "  event.Skip();\n}\n\n"

   438

            elif element_type == ITEM_CONTAINER:

   439

                text += """void Program::On%sPaint(wxPaintEvent& event)

   440

{

   441

  SVGUIContainer* container = (SVGUIContainer*)GetElementByName(wxT("%s"));

   442

  bool isvisible = container->IsVisible();

   443

"""%(element_name, element_name)

   444

                text += element_state%"GUI_BUSY"

   445

                texts = {"location" : current_location, "id" : element.getid()}

   447

                text += """  if (isvisible != _copy__IX%(location)s_%(id)d_1)

   448

  {    

   449

    _copy__IX%(location)s_%(id)d_1 = container->IsVisible();

   450

    _copy__IX%(location)s_%(id)d_2 = true;

   451

"""%texts

   452

                text += "  %s  }\n  else {\n  %s  }\n"%(element_state%"CHANGED", element_state%"UNCHANGED")

   453

                text += "  event.Skip();\n}\n\n"

   454

   455

   456

        text += """void Program::OnChar(wxKeyEvent& event)

   457

{

   458

  SVGUIContainer* container = GetSVGUIRootElement();

   459

  if (container->GetFocusedElementName() == wxT("TextCtrl"))

   460

  {

   461

    wxString focusedId = container->GetFocusedElement();

   462

    SVGUITextCtrl* text = (SVGUITextCtrl*)GetElementById(focusedId);

   463

    text->SendKeyEvent(event);

   464

"""

   465

        for element in elements:

   466

            element_type = GetElementType(element)

   467

            if element_type == ITEM_TEXT:

   468

                texts = {"location" : current_location, "id" : element.getid()}

   469

   470

                text += """    if (focusedId == wxT("%(id)d"))

   471

    {

   472

      in_state_%(id)d = GUI_BUSY;

   473

      _copy__IB%(location)s_%(id)d_1 = wxStringToIEC_STRING(text->GetValue());

   474

      _copy__IX%(location)s_%(id)d_2 = true;

   475

      in_state_%(id)d = CHANGED;

   476

    }

   477

"""%texts

   478

   479

        text += "  }\n  event.Skip();\n}\n\n"

   480

   481

   482

        text += """void Program::OnClick(wxMouseEvent& event)

   483

{

   484

  SVGUIContainer* container = GetSVGUIRootElement();

   485

  if (container->GetFocusedElementName() == wxT("ScrollBar"))

   486

  {

   487

    wxString focusedId = container->GetFocusedElement();

   488

    SVGUIScrollBar* scrollbar = (SVGUIScrollBar*)GetElementById(focusedId);

   489

    scrollbar->SendMouseEvent(event);

   490

"""

   491

        for element in elements: