1

import os, sys

     2

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

     3

CanFestivalPath = os.path.join(base_folder, "CanFestival-3")

     4

sys.path.append(os.path.join(CanFestivalPath, "objdictgen"))

     5

     6

from nodelist import NodeList

     7

from nodemanager import NodeManager

     8

import config_utils, gen_cfile, eds_utils

     9

from networkedit import networkedit

    10

from objdictedit import objdictedit

    11

import canfestival_config as local_canfestival_config

    12

from ConfigTreeNode import ConfigTreeNode

    13

from commondialogs import CreateNodeDialog

    14

import wx

    15

    16

from SlaveEditor import SlaveEditor

    17

from NetworkEditor import NetworkEditor

    18

    19

from gnosis.xml.pickle import *

    20

from gnosis.xml.pickle.util import setParanoia

    21

setParanoia(0)

    22

    23

if wx.Platform == '__WXMSW__':

    24

    DEFAULT_SETTINGS = {

    25

        "CAN_Driver": "can_tcp_win32",

    26

        "CAN_Device": "127.0.0.1",

    27

        "CAN_Baudrate": "125K",

    28

        "Slave_NodeId": 2,

    29

        "Master_NodeId": 1,

    30

    }

    31

else:

    32

    DEFAULT_SETTINGS = {

    33

        "CAN_Driver": "../CanFestival-3/drivers/can_socket/libcanfestival_can_socket.so",

    34

        "CAN_Device": "vcan0",

    35

        "CAN_Baudrate": "125K",

    36

        "Slave_NodeId": 2,

    37

        "Master_NodeId": 1,

    38

    }

    39

    40

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

    41

#                    SLAVE

    42

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

    43

    44

class _SlaveCTN(NodeManager):

    45

    XSD = """<?xml version="1.0" encoding="ISO-8859-1" ?>

    46

    <xsd:schema xmlns:xsd="http://www.w3.org/2001/XMLSchema">

    47

      <xsd:element name="CanFestivalSlaveNode">

    48

        <xsd:complexType>

    49

          <xsd:attribute name="CAN_Device" type="xsd:string" use="optional" default="%(CAN_Device)s"/>

    50

          <xsd:attribute name="CAN_Baudrate" type="xsd:string" use="optional" default="%(CAN_Baudrate)s"/>

    51

          <xsd:attribute name="NodeId" type="xsd:string" use="optional" default="%(Slave_NodeId)d"/>

    52

          <xsd:attribute name="Sync_Align" type="xsd:integer" use="optional" default="0"/>

    53

          <xsd:attribute name="Sync_Align_Ratio" use="optional" default="50">

    54

            <xsd:simpleType>

    55

                <xsd:restriction base="xsd:integer">

    56

                    <xsd:minInclusive value="1"/>

    57

                    <xsd:maxInclusive value="99"/>

    58

                </xsd:restriction>

    59

            </xsd:simpleType>

    60

          </xsd:attribute>

    61

        </xsd:complexType>

    62

      </xsd:element>

    63

    </xsd:schema>

    64

    """ % DEFAULT_SETTINGS

    65

    66

    EditorType = SlaveEditor

    67

    IconPath = os.path.join(CanFestivalPath, "objdictgen", "networkedit.png")

    68

    69

    def __init__(self):

    70

        # TODO change netname when name change

    71

        NodeManager.__init__(self)

    72

        odfilepath = self.GetSlaveODPath()

    73

        if(os.path.isfile(odfilepath)):

    74

            self.OpenFileInCurrent(odfilepath)

    75

        else:

    76

            self.FilePath = ""

    77

            dialog = CreateNodeDialog(None, wx.OK)

    78

            dialog.Type.Enable(False)

    79

            dialog.GenSYNC.Enable(False)

    80

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

    81

                name, id, nodetype, description = dialog.GetValues()

    82

                profile, filepath = dialog.GetProfile()

    83

                NMT = dialog.GetNMTManagement()

    84

                options = dialog.GetOptions()

    85

                self.CreateNewNode(name,       # Name - will be changed at build time

    86

                                   id,         # NodeID - will be changed at build time

    87

                                   "slave",    # Type

    88

                                   description,# description 

    89

                                   profile,    # profile

    90

                                   filepath,   # prfile filepath

    91

                                   NMT,        # NMT

    92

                                   options)     # options

    93

            else:

    94

                self.CreateNewNode("SlaveNode",  # Name - will be changed at build time

    95

                                   0x00,         # NodeID - will be changed at build time

    96

                                   "slave",      # Type

    97

                                   "",           # description 

    98

                                   "None",       # profile

    99

                                   "", # prfile filepath

   100

                                   "heartbeat",  # NMT

   101

                                   [])           # options

   102

            dialog.Destroy()

   103

            self.OnCTNSave()

   104

   105

    def GetSlaveODPath(self):

   106

        return os.path.join(self.CTNPath(), 'slave.od')

   107

   108

    def GetCanDevice(self):

   109

        return self.CanFestivalSlaveNode.getCan_Device()

   110

   111

    def _OpenView(self):

   112

        ConfigTreeNode._OpenView(self)

   113

        if self._View is not None:

   114

            self._View.SetBusId(self.GetCurrentLocation())

   115

   116

    ConfNodeMethods = [

   117

        {"bitmap" : "NetworkEdit",

   118

         "name" : "Edit slave", 

   119

         "tooltip" : "Edit CanOpen slave with ObjdictEdit",

   120

         "method" : "_OpenView"},

   121

    ]

   122

   123

    def OnCTNClose(self):

   124

        if self._View:

   125

            self._View.Close()

   126

   127

    def CTNTestModified(self):

   128

        return self.ChangesToSave or self.OneFileHasChanged()

   129

   130

    def OnCTNSave(self):

   131

        return self.SaveCurrentInFile(self.GetSlaveODPath())

   132

   133

    def SetParamsAttribute(self, path, value):

   134

        result = ConfigTreeNode.SetParamsAttribute(self, path, value)

   135

   136

        # Filter IEC_Channel and Name, that have specific behavior

   137

        if path == "BaseParams.IEC_Channel" and self._View is not None:

   138

            self._View.SetBusId(self.GetCurrentLocation())

   139

   140

        return result

   141

   142

    def CTNGenerate_C(self, buildpath, locations):

   143

        """

   144

        Generate C code

   145

        @param current_location: Tupple containing confnode IEC location : %I0.0.4.5 => (0,0,4,5)

   146

        @param locations: List of complete variables locations \

   147

            [{"IEC_TYPE" : the IEC type (i.e. "INT", "STRING", ...)

   148

            "NAME" : name of the variable (generally "__IW0_1_2" style)

   149

            "DIR" : direction "Q","I" or "M"

   150

            "SIZE" : size "X", "B", "W", "D", "L"

   151

            "LOC" : tuple of interger for IEC location (0,1,2,...)

   152

            }, ...]

   153

        @return: [(C_file_name, CFLAGS),...] , LDFLAGS_TO_APPEND

   154

        """

   155

        current_location = self.GetCurrentLocation()

   156

        # define a unique name for the generated C file

   157

        prefix = "_".join(map(str, current_location))

   158

        Gen_OD_path = os.path.join(buildpath, "OD_%s.c"%prefix )

   159

        # Create a new copy of the model

   160

        slave = self.GetCurrentNodeCopy()

   161

        slave.SetNodeName("OD_%s"%prefix)

   162

        # allow access to local OD from Slave PLC

   163

        pointers = config_utils.LocalODPointers(locations, current_location, slave)

   164

        res = gen_cfile.GenerateFile(Gen_OD_path, slave, pointers)

   165

        if res :

   166

            raise Exception, res

   167

        res = eds_utils.GenerateEDSFile(os.path.join(buildpath, "Slave_%s.eds"%prefix), slave)

   168

        if res :

   169

            raise Exception, res

   170

        return [(Gen_OD_path,local_canfestival_config.getCFLAGS(CanFestivalPath))],"",False

   171

   172

    def LoadPrevious(self):

   173

        self.LoadCurrentPrevious()

   174

   175

    def LoadNext(self):

   176

        self.LoadCurrentNext()

   177

   178

    def GetBufferState(self):

   179

        return self.GetCurrentBufferState()

   180

   181

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

   182

#                    MASTER

   183

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

   184

   185

class MiniNodeManager(NodeManager):

   186

   187

    def __init__(self, parent, filepath, fullname):

   188

        NodeManager.__init__(self)

   189

   190

        self.OpenFileInCurrent(filepath)

   191

   192

        self.Parent = parent

   193

        self.Fullname = fullname

   194

   195

    def OnCloseEditor(self, view):

   196

        self.Parent.OnCloseEditor(view)

   197

   198

    def CTNFullName(self):

   199

        return self.Fullname

   200

   201

    def GetBufferState(self):

   202

        return self.GetCurrentBufferState()

   203

   204

class _NodeListCTN(NodeList):

   205

    XSD = """<?xml version="1.0" encoding="ISO-8859-1" ?>

   206

    <xsd:schema xmlns:xsd="http://www.w3.org/2001/XMLSchema">

   207

      <xsd:element name="CanFestivalNode">

   208

        <xsd:complexType>

   209

          <xsd:attribute name="CAN_Device" type="xsd:string" use="optional" default="%(CAN_Device)s"/>

   210

          <xsd:attribute name="CAN_Baudrate" type="xsd:string" use="optional" default="%(CAN_Baudrate)s"/>

   211

          <xsd:attribute name="NodeId" type="xsd:string" use="optional" default="%(Master_NodeId)d"/>

   212

          <xsd:attribute name="Sync_TPDOs" type="xsd:boolean" use="optional" default="true"/>

   213

        </xsd:complexType>

   214

      </xsd:element>

   215

    </xsd:schema>

   216

    """ % DEFAULT_SETTINGS

   217

   218

    EditorType = NetworkEditor

   219

    IconPath = os.path.join(CanFestivalPath, "objdictgen", "networkedit.png")

   220

   221

    def __init__(self):

   222

        manager = NodeManager()

   223

        NodeList.__init__(self, manager)

   224

        self.LoadProject(self.CTNPath())

   225

        self.SetNetworkName(self.BaseParams.getName())

   226

   227

    def GetCanDevice(self):

   228

        return self.CanFestivalNode.getCan_Device()

   229

   230

    def SetParamsAttribute(self, path, value):

   231

        result = ConfigTreeNode.SetParamsAttribute(self, path, value)

   232

   233

        # Filter IEC_Channel and Name, that have specific behavior

   234

        if path == "BaseParams.IEC_Channel" and self._View is not None:

   235

            self._View.SetBusId(self.GetCurrentLocation())

   236

        elif path == "BaseParams.Name":

   237

            self.SetNetworkName(value)

   238

   239

        return result

   240

   241

    def _OpenView(self):

   242

        ConfigTreeNode._OpenView(self)

   243

        if self._View is not None:

   244

            self._View.SetBusId(self.GetCurrentLocation())

   245

   246

    _GeneratedView = None

   247

    def _ShowMasterGenerated(self):

   248

        if self._GeneratedView is None:

   249

            buildpath = self._getBuildPath()

   250

            # Eventually create build dir

   251

            if not os.path.exists(buildpath):

   252

                self.GetCTRoot().logger.write_error(_("Error: No PLC built\n"))

   253

                return

   254

   255

            masterpath = os.path.join(buildpath, "MasterGenerated.od")

   256

            if not os.path.exists(masterpath):

   257

                self.GetCTRoot().logger.write_error(_("Error: No Master generated\n"))

   258

                return

   259

   260

            app_frame = self.GetCTRoot().AppFrame

   261

   262

            manager = MiniNodeManager(self, masterpath, self.CTNFullName() + ".generated_master")

   263

            self._GeneratedView = SlaveEditor(app_frame.TabsOpened, manager, app_frame, False)

   264

   265

            app_frame.EditProjectElement(self._GeneratedView, "MasterGenerated")

   266

   267

    def _CloseGenerateView(self):

   268

        if self._GeneratedView is not None:

   269

            app_frame = self.GetCTRoot().AppFrame

   270

            if app_frame is not None:

   271

                app_frame.DeletePage(self._GeneratedView)

   272

   273

    ConfNodeMethods = [

   275

         "name" : _("Edit network"), 

   276

         "tooltip" : _("Edit CanOpen Network with NetworkEdit"),

   277

         "method" : "_OpenView"},

   279

         "name" : _("Show Master"), 

   280

         "tooltip" : _("Show Master generated by config_utils"),

   281

         "method" : "_ShowMasterGenerated"}

   282

    ]

   283

   284

    def OnCloseEditor(self, view):

   285

        ConfigTreeNode.OnCloseEditor(self, view)

   286

        if self._GeneratedView == view:

   287

            self._GeneratedView = None

   288

   289

    def OnCTNClose(self):

   290

        ConfigTreeNode.OnCTNClose(self)

   291

        self._CloseGenerateView()

   292

        return True

   293

   294

    def CTNTestModified(self):

   295

        return self.ChangesToSave or self.HasChanged()

   296

   297

    def OnCTNSave(self):

   298

        self.SetRoot(self.CTNPath())

   299

        return self.SaveProject() is None

   300

   301

    def CTNGenerate_C(self, buildpath, locations):

   302

        """

   303

        Generate C code

   304

        @param current_location: Tupple containing confnode IEC location : %I0.0.4.5 => (0,0,4,5)

   305

        @param locations: List of complete variables locations \

   306

            [{"IEC_TYPE" : the IEC type (i.e. "INT", "STRING", ...)

   307

            "NAME" : name of the variable (generally "__IW0_1_2" style)

   308

            "DIR" : direction "Q","I" or "M"

   309

            "SIZE" : size "X", "B", "W", "D", "L"

   310

            "LOC" : tuple of interger for IEC location (0,1,2,...)

   311

            }, ...]

   312

        @return: [(C_file_name, CFLAGS),...] , LDFLAGS_TO_APPEND

   313

        """

   314

        self._CloseGenerateView()

   315

        current_location = self.GetCurrentLocation()

   316

        # define a unique name for the generated C file

   317

        prefix = "_".join(map(str, current_location))

   318

        Gen_OD_path = os.path.join(buildpath, "OD_%s.c"%prefix )

   319

        # Create a new copy of the model with DCF loaded with PDO mappings for desired location

   320

        try:

   321

            master, pointers = config_utils.GenerateConciseDCF(locations, current_location, self, self.CanFestivalNode.getSync_TPDOs(),"OD_%s"%prefix)

   322

        except config_utils.PDOmappingException, e:

   323

            raise Exception, e.message

   324

        # Do generate C file.

   325

        res = gen_cfile.GenerateFile(Gen_OD_path, master, pointers)

   326

        if res :

   327

            raise Exception, res

   328

   329

        file = open(os.path.join(buildpath, "MasterGenerated.od"), "w")

   330

        dump(master, file)

   331

        file.close()

   332

   333

        return [(Gen_OD_path,local_canfestival_config.getCFLAGS(CanFestivalPath))],"",False

   334

   335

    def LoadPrevious(self):

   336

        self.Manager.LoadCurrentPrevious()

   337

   338

    def LoadNext(self):

   339

        self.Manager.LoadCurrentNext()

   340

   341

    def GetBufferState(self):

   342

        return self.Manager.GetCurrentBufferState()

   343

   344

class RootClass:

   345

    XSD = """<?xml version="1.0" encoding="ISO-8859-1" ?>

   346

    <xsd:schema xmlns:xsd="http://www.w3.org/2001/XMLSchema">

   347

      <xsd:element name="CanFestivalInstance">

   348

        <xsd:complexType>

   349

          <xsd:attribute name="CAN_Driver" type="xsd:string" use="optional" default="%(CAN_Driver)s"/>

   350

          <xsd:attribute name="Debug_mode" type="xsd:boolean" use="optional" default="false"/>

   351

        </xsd:complexType>

   352

      </xsd:element>

   353

    </xsd:schema>

   354

    """ % DEFAULT_SETTINGS

   355

   356

    CTNChildrenTypes = [("CanOpenNode",_NodeListCTN, "CanOpen Master"),

   357

                       ("CanOpenSlave",_SlaveCTN, "CanOpen Slave")]

   358

    def GetParamsAttributes(self, path = None):

   359

        infos = ConfigTreeNode.GetParamsAttributes(self, path = None)

   360

        for element in infos:

   361

            if element["name"] == "CanFestivalInstance":

   362

                for child in element["children"]:

   363

                    if child["name"] == "CAN_Driver":

   364

                        DLL_LIST= getattr(local_canfestival_config,"DLL_LIST",None)

   365

                        if DLL_LIST is not None:

   366

                            child["type"] = DLL_LIST  

   367

        return infos

   368

   369

    def GetCanDriver(self):

   370

        can_driver = self.CanFestivalInstance.getCAN_Driver()

   371

        if sys.platform == 'win32':

   372

            if self.CanFestivalInstance.getDebug_mode() and os.path.isfile(os.path.join("%s"%(can_driver + '_DEBUG.dll'))):

   373

                can_driver += '_DEBUG.dll'

   374

            else:

   375

                can_driver += '.dll'

   376

        return can_driver

   377

   378

    def CTNGenerate_C(self, buildpath, locations):

   379

   380

        format_dict = {"locstr" : "_".join(map(str,self.GetCurrentLocation())),

   381

                       "candriver" : self.GetCanDriver(),

   382

                       "nodes_includes" : "",

   383

                       "board_decls" : "",

   384

                       "nodes_init" : "",

   385

                       "nodes_open" : "",

   386

                       "nodes_stop" : "",

   387

                       "nodes_close" : "",

   388

                       "nodes_send_sync" : "",

   389

                       "nodes_proceed_sync" : "",

   390

                       "slavebootups" : "",

   391

                       "slavebootup_register" : "",

   392

                       "post_sync" : "",

   393

                       "post_sync_register" : "",

   394

                       }

   395

        for child in self.IECSortedChildren():

   396

            childlocstr = "_".join(map(str,child.GetCurrentLocation()))

   397

            nodename = "OD_%s" % childlocstr

   398

   399

            # Try to get Slave Node

   400

            child_data = getattr(child, "CanFestivalSlaveNode", None)

   401

            if child_data is None:

   402

                # Not a slave -> master

   403

                child_data = getattr(child, "CanFestivalNode")

   404

                # Apply sync setting

   405

                format_dict["nodes_init"] += 'NODE_MASTER_INIT(%s, %s)\n    '%(

   406

                       nodename,

   407

                       child_data.getNodeId())

   408

                if child_data.getSync_TPDOs():

   409

                    format_dict["nodes_send_sync"] += 'NODE_SEND_SYNC(%s)\n    '%(nodename)

   410

                    format_dict["nodes_proceed_sync"] += 'NODE_PROCEED_SYNC(%s)\n    '%(nodename)

   411

   412

                # initialize and declare node boot status variables for post_SlaveBootup lookup

   413

                SlaveIDs = child.GetSlaveIDs()

   414

                if len(SlaveIDs) == 0:

   415

                    # define post_SlaveBootup lookup functions

   416

                    format_dict["slavebootups"] += (

   417

                        "static void %s_post_SlaveBootup(CO_Data* d, UNS8 nodeId){}\n"%(nodename))

   418

                else:

   419

                    for id in SlaveIDs:

   420

                        format_dict["slavebootups"] += (

   421

                        "int %s_slave_%d_booted = 0;\n"%(nodename, id))

   422

                    # define post_SlaveBootup lookup functions

   423

                    format_dict["slavebootups"] += (

   424

                        "static void %s_post_SlaveBootup(CO_Data* d, UNS8 nodeId){\n"%(nodename)+

   425

                        "    switch(nodeId){\n")

   426

                    # one case per declared node, mark node as booted

   427

                    for id in SlaveIDs:

   428

                        format_dict["slavebootups"] += (

   429

                        "        case %d:\n"%(id)+

   430

                        "            %s_slave_%d_booted = 1;\n"%(nodename, id)+

   431

                        "            break;\n")

   432

                    format_dict["slavebootups"] += (

   433

                        "        default:\n"+

   434

                        "            break;\n"+

   435

                        "    }\n"+

   436

                        "    if( ")

   437

                    # expression to test if all declared nodes booted

   438

                    format_dict["slavebootups"] += " && ".join(["%s_slave_%d_booted"%(nodename, id) for id in SlaveIDs])

   439

                    format_dict["slavebootups"] += " )\n" + (

   440

                        "        Master_post_SlaveBootup(d,nodeId);\n"+

   441

                        "}\n")

   442

                # register previously declared func as post_SlaveBootup callback for that node

   443

                format_dict["slavebootup_register"] += (

   444

                    "%s_Data.post_SlaveBootup = %s_post_SlaveBootup;\n"%(nodename,nodename))

   445

            else:

   446

                # Slave node

   447

                align = child_data.getSync_Align()

   448

                align_ratio=child_data.getSync_Align_Ratio()

   449

                if align > 0:

   450

                    format_dict["post_sync"] += (

   451

                        "static int %s_CalCount = 0;\n"%(nodename)+

   452

                        "static void %s_post_sync(CO_Data* d){\n"%(nodename)+

   453

                        "    if(%s_CalCount < %d){\n"%(nodename, align)+

   454

                        "        %s_CalCount++;\n"%(nodename)+

   455

                        "        align_tick(-1);\n"+

   456

                        "    }else{\n"+

   457

                        "        align_tick(%d);\n"%(align_ratio)+

   458

                        "    }\n"+

   459

                        "}\n")

   460

                    format_dict["post_sync_register"] += (

   461

                        "%s_Data.post_sync = %s_post_sync;\n"%(nodename,nodename))

   462

                format_dict["nodes_init"] += 'NODE_SLAVE_INIT(%s, %s)\n    '%(

   463

                       nodename,

   464

                       child_data.getNodeId())

   465

   466

            # Include generated OD headers

   467

            format_dict["nodes_includes"] += '#include "%s.h"\n'%(nodename)

   468

            # Declare CAN channels according user filled config

   469

            format_dict["board_decls"] += 'BOARD_DECL(%s, "%s", "%s")\n'%(

   470

                   nodename,

   471

                   child.GetCanDevice(),

   472

                   child_data.getCAN_Baudrate())

   473

            format_dict["nodes_open"] += 'NODE_OPEN(%s)\n    '%(nodename)

   474

            format_dict["nodes_close"] += 'NODE_CLOSE(%s)\n    '%(nodename)

   475

            format_dict["nodes_stop"] += 'NODE_STOP(%s)\n    '%(nodename)

   476

   477

        filename = os.path.join(os.path.split(__file__)[0],"cf_runtime.c")

   478

        cf_main = open(filename).read() % format_dict

   479

        cf_main_path = os.path.join(buildpath, "CF_%(locstr)s.c"%format_dict)

   480

        f = open(cf_main_path,'w')

   481

        f.write(cf_main)

   482

        f.close()

   483

   484

        return [(cf_main_path, local_canfestival_config.getCFLAGS(CanFestivalPath))],local_canfestival_config.getLDFLAGS(CanFestivalPath), True