1

import os, shutil

     2

import cPickle

     3

from xml.dom import minidom

     4

     5

import wx

     6

     7

from xmlclass import *

     9

from ConfigEditor import ConfigEditor, ETHERCAT_VENDOR, ETHERCAT_GROUP, ETHERCAT_DEVICE

    10

    11

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

    12

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

    13

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

    14

    15

DATATYPECONVERSION = {"BOOL" : "BIT", "SINT" : "S8", "INT" : "S16", "DINT" : "S32", "LINT" : "S64",

    16

    "USINT" : "U8", "UINT" : "U16", "UDINT" : "U32", "ULINT" : "U64", 

    17

    "BYTE" : "U8", "WORD" : "U16", "DWORD" : "U32", "LWORD" : "U64"}

    18

    19

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

    20

#                 Ethercat MASTER

    21

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

    22

    23

EtherCATConfigClasses = GenerateClassesFromXSD(os.path.join(os.path.dirname(__file__), "EtherCATConfig.xsd")) 

    24

    25

def ExtractHexDecValue(value):

    26

    try:

    27

        return int(value)

    28

    except:

    29

        pass

    30

    try:

    31

        return int(value.replace("#", "0"), 16)

    32

    except:

    33

        raise "Invalid value for HexDecValue \"%s\"" % value

    34

    35

def GenerateHexDecValue(value, base=10):

    36

    if base == 10:

    37

        return str(value)

    38

    elif base == 16:

    39

        return "#x%.8x" % value

    40

    else:

    41

        raise "Not supported base"

    42

    43

cls = EtherCATConfigClasses.get("Config_Slave", None)

    44

if cls:

    45

    46

    def getType(self):

    47

        slave_info = self.getInfo()

    48

        return {"device_type": slave_info.getName(),

    49

                "vendor": GenerateHexDecValue(slave_info.getVendorId()),

    50

                "product_code": GenerateHexDecValue(slave_info.getProductCode(), 16),

    51

                "revision_number": GenerateHexDecValue(slave_info.getRevisionNo(), 16)}

    52

    setattr(cls, "getType", getType)

    53

    54

    def setType(self, type_infos):

    55

        slave_info = self.getInfo()

    56

        slave_info.setName(type_infos["device_type"])

    57

        slave_info.setVendorId(ExtractHexDecValue(type_infos["vendor"]))

    58

        slave_info.setProductCode(ExtractHexDecValue(type_infos["product_code"]))

    59

        slave_info.setRevisionNo(ExtractHexDecValue(type_infos["revision_number"]))

    60

    setattr(cls, "setType", setType)

    61

    62

cls = EtherCATConfigClasses.get("Slave_Info", None)

    63

if cls:

    64

    65

    def getSlavePosition(self):

    66

        return self.getPhysAddr(), self.getAutoIncAddr()

    67

    setattr(cls, "getSlavePosition", getSlavePosition)

    68

    69

    def setSlavePosition(self, alias, pos):

    70

        self.setPhysAddr(alias)

    71

        self.setAutoIncAddr(pos)

    72

    setattr(cls, "setSlavePosition", setSlavePosition)

    73

    74

class _EthercatPlug:

    75

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

    76

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

    77

      <xsd:element name="EtherlabNode">

    78

        <xsd:complexType>

    79

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

    80

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

    81

        </xsd:complexType>

    82

      </xsd:element>

    83

    </xsd:schema>

    84

    """

    86

    87

    def __init__(self):

    88

        filepath = self.ConfigFileName()

    89

    90

        self.Config = EtherCATConfigClasses["EtherCATConfig"]()

    91

        if os.path.isfile(filepath):

    92

            xmlfile = open(filepath, 'r')

    93

            tree = minidom.parse(xmlfile)

    94

            xmlfile.close()

    95

    96

            for child in tree.childNodes:

    97

                if child.nodeType == tree.ELEMENT_NODE and child.nodeName == "EtherCATConfig":

    98

                    self.Config.loadXMLTree(child, ["xmlns:xsi", "xsi:noNamespaceSchemaLocation"])

    99

                    self.CreateConfigBuffer(True)

   100

        else:

   101

            self.CreateConfigBuffer(False)

   102

            self.OnPlugSave()

   103

   104

    def ExtractHexDecValue(self, value):

   105

        return ExtractHexDecValue(value)

   106

   107

    def GetSizeOfType(self, type):

   108

        return TYPECONVERSION.get(self.GetPlugRoot().GetBaseType(type), None)

   109

   110

    def ConfigFileName(self):

   111

        return os.path.join(self.PlugPath(), "config.xml")

   112

   113

    def GetFilename(self):

   114

        return self.MandatoryParams[1].getName()

   115

   116

    def GetSlaves(self):

   117

        slaves = []

   118

        for slave in self.Config.getConfig().getSlave():

   119

            slaves.append(slave.getInfo().getSlavePosition())

   120

        slaves.sort()

   121

        return slaves

   122

   123

    def GetSlave(self, slave_pos):

   124

        for slave in self.Config.getConfig().getSlave():

   125

            slave_info = slave.getInfo()

   126

            if slave_info.getSlavePosition() == slave_pos:

   127

                return slave

   128

        return None

   129

   130

    def AddSlave(self):

   131

        slaves = self.GetSlaves()

   132

        if len(slaves) > 0:

   133

            new_pos = (slaves[-1][0] + 1, 0)

   134

        else:

   135

            new_pos = (0, 0)

   136

        slave = EtherCATConfigClasses["Config_Slave"]()

   137

        slave_infos = slave.getInfo()

   138

        slave_infos.setName("undefined")

   139

        slave_infos.setSlavePosition(new_pos[0], new_pos[1])

   140

        self.Config.getConfig().appendSlave(slave)

   141

        self.BufferConfig()

   142

        return new_pos

   143

   144

    def RemoveSlave(self, slave_pos):

   145

        config = self.Config.getConfig()

   146

        for idx, slave in enumerate(config.getSlave()):

   147

            slave_infos = slave.getInfo()

   148

            if slave_infos.getSlavePosition() == slave_pos:

   149

                config.removeSlave(idx)

   150

                self.BufferConfig()

   151

                return True

   152

        return False

   153

   154

    def SetSlavePos(self, slave_pos, alias=None, position=None):

   155

        slave = self.GetSlave(slave_pos)

   156

        if slave is not None:

   157

            slave_info = slave.getInfo()

   158

            new_pos = slave_pos

   159

            if alias is not None:

   160

                new_pos = (alias, new_pos[1])

   161

            if position is not None:

   162

                new_pos = (new_pos[0], position)

   163

            if self.GetSlave(new_pos) is not None:

   164

                return _("Slave with position \"%d:%d\" already exists!" % new_pos)

   165

            slave_info.setSlavePosition(*new_pos)

   166

            self.BufferConfig()

   167

        return None

   168

   169

    def GetSlaveType(self, slave_pos):

   170

        slave = self.GetSlave(slave_pos)

   171

        if slave is not None:

   172

            return slave.getType()

   173

        return None

   174

   175

    def SetSlaveType(self, slave_pos, type_infos):

   176

        slave = self.GetSlave(slave_pos)

   177

        if slave is not None:

   178

            slave.setType(type_infos)

   179

            self.BufferConfig()

   180

        return None

   181

   182

    def GetSlaveInfos(self, slave_pos):

   183

        slave = self.GetSlave(slave_pos)

   184

        if slave is not None:

   185

            type_infos = slave.getType()

   186

            device = self.GetModuleInfos(type_infos)

   187

            if device is not None:

   188

                infos = type_infos.copy()

   189

                infos.update({"physics": device.getPhysics(),

   190

                              "variables": []})

   191

                for TxPdo in device.getTxPdo():

   192

                    ExtractPdoInfos(TxPdo, "Transmit", infos)

   193

                for RxPdo in device.getRxPdo():

   194

                    ExtractPdoInfos(RxPdo, "Receive", infos)

   195

                return infos

   196

        return None

   197

   198

    def GetModuleInfos(self, type_infos):

   199

        return self.PlugParent.GetModuleInfos(type_infos)

   200

   201

    def GetSlaveTypesLibrary(self):

   202

        return self.PlugParent.GetModulesLibrary()

   203

   215

   262

    PluginMethods = [

   263

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

   264

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

   265

         "tooltip" : _("Edit Config"),

   266

         "method" : "_OpenView"},

   267

    ]

   268

   269

    def PlugTestModified(self):

   270

        return self.ChangesToSave or not self.ConfigIsSaved()    

   271

   272

    def OnPlugSave(self):

   273

        filepath = self.ConfigFileName()

   274

   275

        text = "<?xml version=\"1.0\" encoding=\"UTF-8\" standalone=\"no\"?>\n"

   276

        extras = {"xmlns:xsi":"http://www.w3.org/2001/XMLSchema-instance",

   277

                  "xsi:noNamespaceSchemaLocation" : "EtherCATInfo.xsd"}

   278

        text += self.Config.generateXMLText("EtherCATConfig", 0, extras)

   279

   280

        xmlfile = open(filepath,"w")

   281

        xmlfile.write(text.encode("utf-8"))

   282

        xmlfile.close()

   283

   284

        self.ConfigBuffer.CurrentSaved()

   285

        return True

   286

   287

    def PlugGenerate_C(self, buildpath, locations):

   288

        """

   289

        Generate C code

   290

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

   291

        @param locations: List of complete variables locations \

   292

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

   293

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

   294

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

   295

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

   296

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

   297

            }, ...]

   298

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

   299

        """

   300

        current_location = self.GetCurrentLocation()

   301

        # define a unique name for the generated C file

   302

        location_str = "_".join(map(lambda x:str(x), current_location))

   303

   304

        Gen_Ethercatfile_path = os.path.join(buildpath, "ethercat_%s.c"%location_str)

   305

   306

        file_generator = _EthercatCFileGenerator(self, Gen_Ethercatfile_path)

   307

   308

        for location in locations:

   309

            loc = location["LOC"][len(current_location):]

   310

            file_generator.DeclareVariable(loc[:2], loc[2], loc[3], location["IEC_TYPE"], location["DIR"], location["NAME"])

   311

   312

        file_generator.GenerateCFile()

   313

   314

        return [(Gen_Ethercatfile_path, '"-I%s"'%os.path.abspath(self.GetPlugRoot().GetIECLibPath()))], "-lethercat -lrtdm", True

   315

   316

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

   317

#                      Current Buffering Management Functions

   318

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

   319

   320

    """

   321

    Return a copy of the config

   322

    """

   323

    def Copy(self, model):

   324

        return cPickle.loads(cPickle.dumps(model))

   325

   326

    def CreateConfigBuffer(self, saved):

   327

        self.ConfigBuffer = UndoBuffer(cPickle.dumps(self.Config), saved)

   328

   329

    def BufferConfig(self):

   330

        self.ConfigBuffer.Buffering(cPickle.dumps(self.Config))

   331

   332

    def ConfigIsSaved(self):

   333

        if self.ConfigBuffer is not None:

   334

            return self.ConfigBuffer.IsCurrentSaved()

   335

        else:

   336

            return True

   337

   338

    def LoadPrevious(self):

   339

        self.Config = cPickle.loads(self.ConfigBuffer.Previous())

   340

   341

    def LoadNext(self):

   342

        self.Config = cPickle.loads(self.ConfigBuffer.Next())

   343

   344

    def GetBufferState(self):

   345

        first = self.ConfigBuffer.IsFirst()

   346

        last = self.ConfigBuffer.IsLast()

   347

        return not first, not last

   348

   349

   350

SLAVE_PDOS_CONFIGURATION_DECLARATION = """

   351

/* Slave %(slave)d, "%(device_type)s"

   352

 * Vendor ID:       0x%(vendor).8x

   353

 * Product code:    0x%(product_code).8x

   354

 * Revision number: 0x%(revision_number).8x

   355

 */

   356

   357

ec_pdo_entry_info_t slave_%(slave)d_pdo_entries[] = {

   358

%(pdos_entries_infos)s

   359

};

   360

   361

ec_pdo_info_t slave_%(slave)d_pdos[] = {

   362

%(pdos_infos)s

   363

};

   364

   365

ec_sync_info_t slave_%(slave)d_syncs[] = {

   366

%(pdos_sync_infos)s

   367

    {0xff}

   368

};

   369

"""

   370

   371

SLAVE_CONFIGURATION_TEMPLATE = """

   372

    if (!(slave%(slave)d = ecrt_master_slave_config(master, %(alias)d, %(position)d, 0x%(vendor).8x, 0x%(product_code).8x))) {

   373

        fprintf(stderr, "Failed to get slave %(device_type)s configuration at alias %(alias)d and position %(position)d.\\n");

   374

        return -1;

   375

    }

   376

   377

    if (ecrt_slave_config_pdos(slave%(slave)d, EC_END, slave_%(slave)d_syncs)) {

   378

        fprintf(stderr, "Failed to configure PDOs for slave %(device_type)s at alias %(alias)d and position %(position)d.\\n");

   379

        return -1;

   380

    }

   381

"""

   382

   383

class _EthercatCFileGenerator:

   384

   385

    def __init__(self, controler, filepath):

   386

        self.Controler = controler

   387

        self.FilePath = filepath

   388

   389

        self.UsedVariables = {}

   390

   391

    def __del__(self):

   392

        self.Controler = None

   393

   394

    def DeclareVariable(self, slave_identifier, index, subindex, iec_type, dir, name):

   395

        slave_variables = self.UsedVariables.setdefault(slave_identifier, {})

   396

   397

        entry_infos = slave_variables.get((index, subindex), None)

   398

        if entry_infos is None:

   399

            slave_variables[(index, subindex)] = (iec_type, dir, name)

   400

        elif entry_infos != (iec_type, dir, name):

   401

            raise ValueError, _("Definition conflict for location \"%s\"") % name 

   402

   403

    def GenerateCFile(self):

   404

   405

        current_location = self.Controler.GetCurrentLocation()

   406

        # define a unique name for the generated C file

   407

        location_str = "_".join(map(lambda x:str(x), current_location))

   408

   409

        plc_etherlab_filepath = os.path.join(os.path.split(__file__)[0], "plc_etherlab.c")

   410

        plc_etherlab_file = open(plc_etherlab_filepath, 'r')

   411

        plc_etherlab_code = plc_etherlab_file.read()

   412

        plc_etherlab_file.close()

   413

   414

        str_completion = {

   415

            "location": location_str,

   416

            "configure_pdos": int(self.Controler.EtherlabNode.getConfigurePDOs()),

   417

            "master_number": self.Controler.EtherlabNode.getMasterNumber(),

   418

            "located_variables_declaration": [],

   419

            "used_pdo_entry_offset_variables_declaration": [],

   420

            "used_pdo_entry_configuration": [],

   421

            "pdos_configuration_declaration": "",

   422

            "slaves_declaration": "",

   423

            "slaves_configuration": "",

   424

            "retrieve_variables": [],

   425

            "publish_variables": [],

   426

        }

   427

   428

        for slave_idx, slave_pos in enumerate(self.Controler.GetSlaves()):

   429

   430

            slave = self.Controler.GetSlave(slave_pos)

   431

            if slave is not None:

   432

                type_infos = slave.getType()

   433

   434

                device = self.Controler.GetModuleInfos(type_infos)

   436

   437

                    pdos = device.getTxPdo() + device.getRxPdo()

   438

                    if len(pdos) > 0:

   439

                        slave_variables = self.UsedVariables.get(slave_pos, {})

   440

   441

                        for element in ["vendor", "product_code", "revision_number"]:

   442

                            type_infos[element] = ExtractHexDecValue(type_infos[element])

   443

                        type_infos.update(dict(zip(["slave", "alias", "position"], (slave_idx,) + slave_pos)))

   444

   445

                        str_completion["slaves_declaration"] += "static ec_slave_config_t *slave%(slave)d = NULL;\n" % type_infos

   446

                        str_completion["slaves_configuration"] += SLAVE_CONFIGURATION_TEMPLATE % type_infos

   447

   448

                        pdos_infos = {

   449

                            "pdos_entries_infos": [],

   450

                            "pdos_infos": [],

   451

                            "pdos_sync_infos": [], 

   452

                        }

   453

                        pdos_infos.update(type_infos)

   454

   455

                        sync_managers = []

   456

                        for sync_manager_idx, sync_manager in enumerate(device.getSm()):

   457

                            sync_manager_infos = {

   458

                                "index": sync_manager_idx, 

   459

                                "slave": slave_idx,

   460

                                "pdos_number": 0,

   461

                            }

   462

   463

                            sync_manager_control_byte = ExtractHexDecValue(sync_manager.getControlByte())

   464

                            sync_manager_direction = sync_manager_control_byte & 0x0c

   465

                            sync_manager_watchdog = sync_manager_control_byte & 0x40

   466

                            if sync_manager_direction:

   467

                                sync_manager_infos["sync_manager_type"] = "EC_DIR_OUTPUT"

   468

                            else:

   469

                                sync_manager_infos["sync_manager_type"] = "EC_DIR_INPUT"

   470

                            if sync_manager_watchdog:

   471

                                sync_manager_infos["watchdog"] = "EC_WD_ENABLE"

   472

                            else:

   473

                                sync_manager_infos["watchdog"] = "EC_WD_DISABLE"

   474

   475

                            sync_managers.append(sync_manager_infos)

   476

   477

                        entry_offset = 0

   478

                        for pdo in pdos:

   479

                            entries = pdo.getEntry()

   480

   483

   485

                                index = ExtractHexDecValue(entry.getIndex().getcontent())

   486

                                subindex = ExtractHexDecValue(entry.getSubIndex())

   487

                                entry_infos = {

   488

                                    "index": index,

   489

                                    "subindex": subindex,

   490

                                    "name": ExtractName(entry.getName()),

   491

                                    "bitlen": entry.getBitLen(),

   492

                                }

   493

                                entry_infos.update(type_infos)

   495

   496

                                entry_declaration = slave_variables.get((index, subindex), None)

   497

                                if entry_declaration is not None:

   499

                                    entry_infos.update(dict(zip(["var_type", "dir", "var_name"], entry_declaration)))

   500

   501

                                    if entry_infos["var_type"] != entry.getDataType().getcontent():

   502

                                        raise ValueError, _("Wrong type for location \"%s\"!") % entry_infos["var_name"]

   503

   504

                                    data_type = DATATYPECONVERSION.get(entry_infos["var_type"], None)

   505

                                    if data_type is None:

   506

                                        raise ValueError, _("Type of location \"%s\" not yet supported!") % entry_infos["var_name"]

   507

   508

                                    if (entry_infos["dir"] == "I" and sync_managers[pdo.getSm()]["sync_manager_type"] != "EC_DIR_INPUT" or 

   509

                                        entry_infos["dir"] == "Q" and sync_managers[pdo.getSm()]["sync_manager_type"] != "EC_DIR_OUTPUT"):

   510

                                        raise ValueError, _("Wrong direction for location \"%s\"!") % entry_infos["var_name"]

   511

   516

                                    if data_type == "BIT":

   520