1

#!/usr/bin/env python

     2

# -*- coding: utf-8 -*-

     3

     4

#This file is part of CanFestival, a library implementing CanOpen Stack. 

     5

#

     6

#Copyright (C): Edouard TISSERANT, Francis DUPIN and Laurent BESSARD

     7

#

     8

#See COPYING file for copyrights details.

     9

#

    10

#This library is free software; you can redistribute it and/or

    11

#modify it under the terms of the GNU Lesser General Public

    12

#License as published by the Free Software Foundation; either

    13

#version 2.1 of the License, or (at your option) any later version.

    14

#

    15

#This library is distributed in the hope that it will be useful,

    16

#but WITHOUT ANY WARRANTY; without even the implied warranty of

    17

#MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU

    18

#Lesser General Public License for more details.

    19

#

    20

#You should have received a copy of the GNU Lesser General Public

    21

#License along with this library; if not, write to the Free Software

    22

#Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

    23

    24

import wx

    25

import wx.grid

    26

    27

from types import *

    28

import os, re, platform, sys, time, traceback, getopt

    29

    30

__version__ = "$Revision$"

    31

    32

from nodelist import *

    33

from nodemanager import *

    34

from subindextable import *

    35

from commondialogs import *

    36

from doc_index.DS301_index import *

    37

    38

ScriptDirectory = os.path.split(os.path.realpath(__file__))[0]

    39

    40

try:

    41

    import wx.html

    42

    43

    EVT_HTML_URL_CLICK = wx.NewId()

    44

    45

    class HtmlWindowUrlClick(wx.PyEvent):

    46

        def __init__(self, linkinfo):

    47

            wx.PyEvent.__init__(self)

    48

            self.SetEventType(EVT_HTML_URL_CLICK)

    49

            self.linkinfo = (linkinfo.GetHref(), linkinfo.GetTarget())

    50

    51

    class UrlClickHtmlWindow(wx.html.HtmlWindow):

    52

        """ HTML window that generates and OnLinkClicked event.

    53

    54

        Use this to avoid having to override HTMLWindow

    55

        """

    56

        def OnLinkClicked(self, linkinfo):

    57

            wx.PostEvent(self, HtmlWindowUrlClick(linkinfo))

    58

    59

        def Bind(self, event, handler, source=None, id=wx.ID_ANY, id2=wx.ID_ANY):

    60

            if event == HtmlWindowUrlClick:

    61

                self.Connect(-1, -1, EVT_HTML_URL_CLICK, handler)

    62

            else:

    63

                wx.html.HtmlWindow.Bind(event, handler, source=source, id=id, id2=id2)

    64

    65

#-------------------------------------------------------------------------------

    66

#                                Html Frame

    67

#-------------------------------------------------------------------------------

    68

    69

    [ID_HTMLFRAME, ID_HTMLFRAMEHTMLCONTENT] = [wx.NewId() for _init_ctrls in range(2)]

    70

    71

    class HtmlFrame(wx.Frame):

    72

        def _init_ctrls(self, prnt):

    73

            # generated method, don't edit

    74

            wx.Frame.__init__(self, id=ID_HTMLFRAME, name='HtmlFrame',

    75

                  parent=prnt, pos=wx.Point(320, 231), size=wx.Size(853, 616),

    76

                  style=wx.DEFAULT_FRAME_STYLE, title='')

    77

            self.Bind(wx.EVT_CLOSE, self.OnCloseFrame)

    78

    79

            self.HtmlContent = UrlClickHtmlWindow(id=ID_HTMLFRAMEHTMLCONTENT,

    80

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

    81

                  size=wx.Size(-1, -1), style=wx.html.HW_SCROLLBAR_AUTO|wx.html.HW_NO_SELECTION)

    82

            self.HtmlContent.Bind(HtmlWindowUrlClick, self.OnLinkClick)

    83

    84

        def __init__(self, parent, opened):

    85

            self._init_ctrls(parent)

    86

            self.HtmlFrameOpened = opened

    87

    88

        def SetHtmlCode(self, htmlcode):

    89

            self.HtmlContent.SetPage(htmlcode)

    90

    91

        def SetHtmlPage(self, htmlpage):

    92

            self.HtmlContent.LoadPage(htmlpage)

    93

    94

        def OnCloseFrame(self, event):

    95

            self.HtmlFrameOpened.remove(self.GetTitle())

    96

            event.Skip()

    97

    98

        def OnLinkClick(self, event):

    99

            url = event.linkinfo[0]

   100

            try:

   101

                import webbrowser

   102

            except ImportError:

   103

                wx.MessageBox('Please point your browser at: %s' % url)

   104

            else:

   105

                webbrowser.open(url)

   106

   107

    Html_Window = True

   108

except:

   109

    Html_Window = False

   110

   111

   112

[ID_NETWORKEDIT, ID_NETWORKEDITNETWORKNODES, 

   113

 ID_NETWORKEDITHELPBAR,

   114

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

   115

   126

] = [wx.NewId() for _init_coll_AddMenu_Items in range(6)]

   127

   128

class networkedit(wx.Frame):

   130

        if self.ModeSolo:

   131

            parent.Append(menu=self.FileMenu, title='File')

   132

        parent.Append(menu=self.NetworkMenu, title='Network')

   133

        parent.Append(menu=self.EditMenu, title='Edit')

   134

        parent.Append(menu=self.AddMenu, title='Add')

   135

        parent.Append(menu=self.HelpMenu, title='Help')

   136

   169

    def _init_coll_EditMenu_Items(self, parent):

   171

              kind=wx.ITEM_NORMAL, text='Refresh\tCTRL+R')

   172

        parent.AppendSeparator()

   174

              kind=wx.ITEM_NORMAL, text='Undo\tCTRL+Z')

   176

              kind=wx.ITEM_NORMAL, text='Redo\tCTRL+Y')

   177

        parent.AppendSeparator()

   179

              kind=wx.ITEM_NORMAL, text='Node infos')

   181

              kind=wx.ITEM_NORMAL, text='DS-301 Profile')

   183

              kind=wx.ITEM_NORMAL, text='DS-302 Profile')

   185

              kind=wx.ITEM_NORMAL, text='Other Profile')

   191

        self.Bind(wx.EVT_MENU, self.OnCommunicationMenu,

   195

        self.Bind(wx.EVT_MENU, self.OnEditProfileMenu,

   198

    def _init_coll_AddMenu_Items(self, parent):

   200

              kind=wx.ITEM_NORMAL, text='SDO Server')

   202

              kind=wx.ITEM_NORMAL, text='SDO Client')

   204

              kind=wx.ITEM_NORMAL, text='PDO Transmit')

   206

              kind=wx.ITEM_NORMAL, text='PDO Receive')

   208

              kind=wx.ITEM_NORMAL, text='Map Variable')

   210

              kind=wx.ITEM_NORMAL, text='User Type')

   211

        self.Bind(wx.EVT_MENU, self.OnAddSDOServerMenu,

   213

        self.Bind(wx.EVT_MENU, self.OnAddSDOClientMenu,

   215

        self.Bind(wx.EVT_MENU, self.OnAddPDOTransmitMenu,

   217

        self.Bind(wx.EVT_MENU, self.OnAddPDOReceiveMenu,

   219

        self.Bind(wx.EVT_MENU, self.OnAddMapVariableMenu,

   221

        self.Bind(wx.EVT_MENU, self.OnAddUserTypeMenu,

   235

   236

    def _init_coll_HelpBar_Fields(self, parent):

   237

        parent.SetFieldsCount(3)

   238

   239

        parent.SetStatusText(number=0, text='')

   240

        parent.SetStatusText(number=1, text='')

   241

        parent.SetStatusText(number=2, text='')

   242

   243

        parent.SetStatusWidths([100, 110, -1])

   244

   245

    def _init_utils(self):

   248

   249

        if self.ModeSolo:

   250

            self.FileMenu = wx.Menu(title='')

   251

        self.NetworkMenu = wx.Menu(title='')

   252

        self.EditMenu = wx.Menu(title='')

   253

        self.AddMenu = wx.Menu(title='')

   254

        self.HelpMenu = wx.Menu(title='')

   255

   257

        if self.ModeSolo:

   258

            self._init_coll_FileMenu_Items(self.FileMenu)

   259

        self._init_coll_NetworkMenu_Items(self.NetworkMenu)

   260

        self._init_coll_EditMenu_Items(self.EditMenu)

   261

        self._init_coll_AddMenu_Items(self.AddMenu)

   262

        self._init_coll_HelpMenu_Items(self.HelpMenu)

   263

   264

    def _init_ctrls(self, prnt):

   265

        wx.Frame.__init__(self, id=ID_NETWORKEDIT, name='networkedit',

   266

              parent=prnt, pos=wx.Point(149, 178), size=wx.Size(1000, 700),

   267

              style=wx.DEFAULT_FRAME_STYLE, title='Networkedit')

   268

        self._init_utils()

   269

        self.SetClientSize(wx.Size(1000, 700))

   271

        self.Bind(wx.EVT_CLOSE, self.OnCloseFrame)

   272

        if not self.ModeSolo:

   275

            self.SetAcceleratorTable(accel)

   276

   277

        self.NetworkNodes = wx.Notebook(id=ID_NETWORKEDITNETWORKNODES,

   278

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

   279

              size=wx.Size(0, 0), style=wx.NB_LEFT)

   280

        self.NetworkNodes.Bind(wx.EVT_NOTEBOOK_PAGE_CHANGED,

   281

              self.OnNodeSelectedChanged, id=ID_NETWORKEDITNETWORKNODES)

   282

   283

        self.HelpBar = wx.StatusBar(id=ID_NETWORKEDITHELPBAR, name='HelpBar',

   284

              parent=self, style=wx.ST_SIZEGRIP)

   285

        self._init_coll_HelpBar_Fields(self.HelpBar)

   286

        self.SetStatusBar(self.HelpBar)

   287

   288

    def __init__(self, parent, nodelist = None, projectOpen = None):

   289

        self.ModeSolo = nodelist == None

   290

        self._init_ctrls(parent)

   291

        self.HtmlFrameOpened = []

   292

        self.BusId = None

   293

   294

        icon = wx.Icon(os.path.join(ScriptDirectory,"networkedit.ico"),wx.BITMAP_TYPE_ICO)

   295

        self.SetIcon(icon)

   296

   297

        if self.ModeSolo:

   298

            self.Manager = NodeManager()

   299

            if projectOpen:

   300

                self.NodeList = NodeList(self.Manager)

   301

                result = self.NodeList.LoadProject(projectOpen)

   302

                if not result:

   303

                    self.RefreshNetworkNodes()

   304

            else:

   305

                self.NodeList = None

   306

        else:

   307

            self.NodeList = nodelist

   308

            self.Manager = self.NodeList.GetManager()

   309

            self.NodeList.SetCurrentSelected(0)

   310

            self.RefreshNetworkNodes()

   311

            self.RefreshProfileMenu()

   312

            self.NetworkNodes.SetFocus()

   313

   314

        self.RefreshBufferState()

   315

        self.RefreshTitle()

   316

        self.RefreshMainMenu()

   317

   318

    def SetBusId(self, bus_id):

   319

        self.BusId = bus_id

   320

   321

    def GetBusId(self):

   322

        return self.BusId

   323

   324

    def GetCurrentNodeId(self):

   325

        selected = self.NetworkNodes.GetSelection()

   326

        # At init selected = -1

   327

        if selected > 0:

   328

            window = self.NetworkNodes.GetPage(selected)

   329

            return window.GetIndex()

   330

        else:

   331

            return 0

   332

   333

    def OnCloseFrame(self, event):

   334

        if not self.ModeSolo and getattr(self, "_onclose", None) != None:

   335

            self._onclose()

   336

        event.Skip()

   337

   338

    def OnChar(self, event):

   339

        if event.ControlDown() and event.GetKeyCode() == 83 and getattr(self, "_onsave", None) != None:

   340

            self._onsave()

   341

        event.Skip()

   342

   343

    def OnQuitMenu(self, event):

   344

        self.Close()

   345

        event.Skip()

   346

   347

    def OnAddSDOServerMenu(self, event):

   348

        self.Manager.AddSDOServerToCurrent()

   349

        self.RefreshBufferState()

   350

        self.RefreshCurrentIndexList()

   351

        event.Skip()

   352

   353

    def OnAddSDOClientMenu(self, event):

   354

        self.Manager.AddSDOClientToCurrent()

   355

        self.RefreshBufferState()

   356

        self.RefreshCurrentIndexList()

   357

        event.Skip()

   358

   359

    def OnAddPDOTransmitMenu(self, event):

   360

        self.Manager.AddPDOTransmitToCurrent()

   361

        self.RefreshBufferState()

   362

        self.RefreshCurrentIndexList()

   363

        event.Skip()

   364

   365

    def OnAddPDOReceiveMenu(self, event):

   366

        self.Manager.AddPDOReceiveToCurrent()

   367

        self.RefreshBufferState()

   368

        self.RefreshCurrentIndexList()

   369

        event.Skip()

   370

   371

    def OnAddMapVariableMenu(self, event):

   372

        self.AddMapVariable()

   373

        event.Skip()

   374

   375

    def OnAddUserTypeMenu(self, event):

   376

        self.AddUserType()

   377

        event.Skip()

   378

   379

    def OnNodeSelectedChanged(self, event):

   380

        selected = event.GetSelection()

   381

        # At init selected = -1

   382

        if selected >= 0:

   383

            window = self.NetworkNodes.GetPage(selected)

   384

            self.NodeList.SetCurrentSelected(window.GetIndex())

   385

        wx.CallAfter(self.RefreshMainMenu)

   386

        wx.CallAfter(self.RefreshStatusBar)

   387

        event.Skip()

   388

   389

#-------------------------------------------------------------------------------

   390

#                         Load and Save Funtions

   391

#-------------------------------------------------------------------------------

   392

   393

    def OnNewProjectMenu(self, event):

   394

        if self.NodeList:

   395

            defaultpath = os.path.dirname(self.NodeList.GetRoot())

   396

        else:

   397

            defaultpath = os.getcwd()

   398

        dialog = wx.DirDialog(self , "Choose a project", defaultpath, wx.DD_NEW_DIR_BUTTON)

   399

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

   400

            projectpath = dialog.GetPath()

   401

            if os.path.isdir(projectpath) and len(os.listdir(projectpath)) == 0:

   402

                manager = NodeManager()

   403

                nodelist = NodeList(manager)

   404

                result = nodelist.LoadProject(projectpath)

   405

                if not result:

   406

                    self.Manager = manager

   407

                    self.NodeList = nodelist

   408

                    self.NodeList.SetCurrentSelected(0)

   409

   410

                    self.RefreshNetworkNodes()

   411

                    self.RefreshBufferState()

   412

                    self.RefreshTitle()

   413

                    self.RefreshProfileMenu()

   414

                    self.RefreshMainMenu()

   415

                else:

   416

                    message = wx.MessageDialog(self, result, "ERROR", wx.OK|wx.ICON_ERROR)

   417

                    message.ShowModal()

   418

                    message.Destroy()

   419

        event.Skip()

   420

   421

    def OnOpenProjectMenu(self, event):

   422

        if self.NodeList:

   423

            defaultpath = os.path.dirname(self.NodeList.GetRoot())

   424

        else:

   425

            defaultpath = os.getcwd()

   426

        dialog = wx.DirDialog(self , "Choose a project", defaultpath, 0)

   427

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

   428

            projectpath = dialog.GetPath()

   429

            if os.path.isdir(projectpath):

   430

                manager = NodeManager()

   431

                nodelist = NodeList(manager)

   432

                result = nodelist.LoadProject(projectpath)

   433

                if not result:

   434

                    self.Manager = manager

   435

                    self.NodeList = nodelist

   436

                    self.NodeList.SetCurrentSelected(0)

   437

   438

                    self.RefreshNetworkNodes()

   439

                    self.RefreshBufferState()

   440

                    self.RefreshTitle()

   441

                    self.RefreshProfileMenu()

   442

                    self.RefreshMainMenu()

   443

                else:

   444

                    message = wx.MessageDialog(self, result, "Error", wx.OK|wx.ICON_ERROR)

   445

                    message.ShowModal()

   446

                    message.Destroy()

   447

        dialog.Destroy()

   448

        event.Skip()

   449

   450

    def OnSaveProjectMenu(self, event):

   451

        if not self.ModeSolo and getattr(self, "_onsave", None) != None:

   452

            self._onsave()

   453

        else:

   454

            result = self.NodeList.SaveProject()

   455

            if result:

   456

                message = wx.MessageDialog(self, result, "Error", wx.OK|wx.ICON_ERROR)

   457

                message.ShowModal()

   458

                message.Destroy()

   459

        event.Skip()

   460

   461

    def OnCloseProjectMenu(self, event):

   462

        if self.NodeList:

   463

            if self.NodeList.HasChanged():

   464

                dialog = wx.MessageDialog(self, "There are changes, do you want to save?",  "Close Project", wx.YES_NO|wx.CANCEL|wx.ICON_QUESTION)

   465

                answer = dialog.ShowModal()

   466

                dialog.Destroy()

   467

                if answer == wx.ID_YES:

   468

                    result = self.NodeList.SaveProject()

   469

                    if result:

   470

                        message = wx.MessageDialog(self, result, "Error", wx.OK|wx.ICON_ERROR)

   471

                        message.ShowModal()

   472

                        message.Destroy()

   473

                elif answer == wx.ID_NO:

   474

                    self.NodeList.ForceChanged(False)

   475

            if not self.NodeList.HasChanged():

   476

                self.Manager = None

   477

                self.NodeList = None

   478

                self.RefreshNetworkNodes()

   479

                self.RefreshTitle()

   480

                self.RefreshMainMenu()

   481

        event.Skip()

   482

   483

#-------------------------------------------------------------------------------

   484

#                             Slave Nodes Management

   485

#-------------------------------------------------------------------------------

   486

   487

    def OnAddSlaveMenu(self, event):

   488

        dialog = AddSlaveDialog(self)

   489

        dialog.SetNodeList(self.NodeList)

   490

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

   491

            values = dialog.GetValues()

   492

            result = self.NodeList.AddSlaveNode(values["slaveName"], values["slaveNodeID"], values["edsFile"])

   493

            if not result:

   494

                new_editingpanel = EditingPanel(self.NetworkNodes, self, self.NodeList, False)

   495

                new_editingpanel.SetIndex(values["slaveNodeID"])

   496

                idx = self.NodeList.GetOrderNumber(values["slaveNodeID"])

   497

                self.NetworkNodes.InsertPage(idx, new_editingpanel, "")

   498

                self.NodeList.SetCurrentSelected(idx)

   499

                self.NetworkNodes.SetSelection(idx)

   500

                self.RefreshBufferState()

   501

            else:

   502

                message = wx.MessageDialog(self, result, "Error", wx.OK|wx.ICON_ERROR)

   503

                message.ShowModal()

   504

                message.Destroy()

   505

        dialog.Destroy()