1

#!/usr/bin/env python

     2

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

     3

     4

# This file is part of Beremiz

     5

#

     6

# Copyright (C) 2011-2014: Laurent BESSARD, Edouard TISSERANT

     7

#                          RTES Lab : CRKim, JBLee, youcu

     8

#                          Higen Motor : Donggu Kang

     9

#

    10

# See COPYING file for copyrights details.

    11

    12

import os

    13

import shutil

    14

from lxml import etree

    15

    16

import wx

    17

import csv

    18

    19

from xmlclass import *

    20

    21

from ConfigTreeNode import ConfigTreeNode, XSDSchemaErrorMessage

    22

from PLCControler import UndoBuffer, LOCATION_CONFNODE, LOCATION_MODULE, LOCATION_GROUP, LOCATION_VAR_INPUT, LOCATION_VAR_OUTPUT, LOCATION_VAR_MEMORY

    23

    24

from EthercatSlave import ExtractHexDecValue, ExtractName

    25

from EthercatMaster import _EthercatCTN

    26

from ConfigEditor import LibraryEditor, ETHERCAT_VENDOR, ETHERCAT_GROUP, ETHERCAT_DEVICE

    27

    28

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

    29

    30

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

    31

#                 Ethercat ConfNode

    32

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

    33

    34

EtherCATInfoParser = GenerateParserFromXSD(os.path.join(os.path.dirname(__file__), "EtherCATInfo.xsd"))

    35

EtherCATInfo_XPath = lambda xpath: etree.XPath(xpath)

    36

    37

    38

def HexDecValue(context, *args):

    39

    return str(ExtractHexDecValue(args[0][0]))

    40

    41

    42

def EntryName(context, *args):

    43

    return ExtractName(args[0],

    44

        args[1][0] if len(args) > 1 else None)

    45

    46

ENTRY_INFOS_KEYS = [

    47

    ("Index", lambda x: "#x%4.4X" % int(x), "#x0000"),

    48

    ("SubIndex", str, "0"),

    49

    ("Name", str, ""),

    50

    ("Type", str, ""),

    51

    ("BitSize", int, 0),

    52

    ("Access", str, ""),

    53

    ("PDOMapping", str, ""),

    54

    ("PDO index", str, ""),

    55

    ("PDO name", str, ""),

    56

    ("PDO type", str, "")]

    57

    58

    59

class EntryListFactory:

    60

    61

    def __init__(self, entries):

    62

        self.Entries = entries

    63

    64

    def AddEntry(self, context, *args):

    65

        index, subindex = map(lambda x: int(x[0]), args[:2])

    66

        new_entry_infos = {

    67

            key: translate(arg[0]) if len(arg) > 0 else default

    68

            for (key, translate, default), arg

    69

            in zip(ENTRY_INFOS_KEYS, args)}

    70

    71

        if (index, subindex) != (0, 0):

    72

            entry_infos = self.Entries.get((index, subindex))

    73

            if entry_infos is not None:

    74

                for param in ["PDO index", "PDO name", "PDO type"]:

    75

                    value = new_entry_infos.get(param)

    76

                    if value is not None:

    77

                        entry_infos[param] = value

    78

            else:

    79

                self.Entries[(index, subindex)] = new_entry_infos

    80

    81

entries_list_xslt = etree.parse(

    82

    os.path.join(ScriptDirectory, "entries_list.xslt"))

    83

    84

cls = EtherCATInfoParser.GetElementClass("DeviceType")

    85

if cls:

    86

    87

    profile_numbers_xpath = EtherCATInfo_XPath("Profile/ProfileNo")

    88

    def GetProfileNumbers(self):

    89

        return [number.text for number in profile_numbers_xpath(self)]

    90

    setattr(cls, "GetProfileNumbers", GetProfileNumbers)

    91

    92

    def getCoE(self):

    93

        mailbox = self.getMailbox()

    94

        if mailbox is not None:

    95

            return mailbox.getCoE()

    96

        return None

    97

    setattr(cls, "getCoE", getCoE)

    98

    99

    def GetEntriesList(self, limits=None):

   100

        entries = {}

   101

   102

        factory = EntryListFactory(entries)

   103

   104

        entries_list_xslt_tree = etree.XSLT(

   105

            entries_list_xslt, extensions={

   106

                ("entries_list_ns", "AddEntry"): factory.AddEntry,

   107

                ("entries_list_ns", "HexDecValue"): HexDecValue,

   108

                ("entries_list_ns", "EntryName"): EntryName})

   109

        entries_list_xslt_tree(self, **dict(zip(

   110

            ["min_index", "max_index"],

   111

            map(lambda x: etree.XSLT.strparam(str(x)),

   112

                limits if limits is not None else [0x0000, 0xFFFF])

   113

            )))

   114

   115

        return entries

   116

    setattr(cls, "GetEntriesList", GetEntriesList)

   117

   118

    def GetSyncManagers(self):

   119

        sync_managers = []

   120

        for sync_manager in self.getSm():

   121

            sync_manager_infos = {}

   122

            for name, value in [("Name", sync_manager.getcontent()),

   123

                                ("Start Address", sync_manager.getStartAddress()),

   124

                                ("Default Size", sync_manager.getDefaultSize()),

   125

                                ("Control Byte", sync_manager.getControlByte()),

   126

                                ("Enable", sync_manager.getEnable())]:

   127

                if value is None:

   128

                    value = ""

   129

                sync_manager_infos[name] = value

   130

            sync_managers.append(sync_manager_infos)

   131

        return sync_managers

   132

    setattr(cls, "GetSyncManagers", GetSyncManagers)

   133

   134

   135

def GroupItemCompare(x, y):

   136

    if x["type"] == y["type"]:

   137

        if x["type"] == ETHERCAT_GROUP:

   138

            return cmp(x["order"], y["order"])

   139

        else:

   140

            return cmp(x["name"], y["name"])

   141

    elif x["type"] == ETHERCAT_GROUP:

   142

        return -1

   143

    return 1

   144

   145

   146

def SortGroupItems(group):

   147

    for item in group["children"]:

   148

        if item["type"] == ETHERCAT_GROUP:

   149

            SortGroupItems(item)

   150

    group["children"].sort(GroupItemCompare)

   151

   152

   153

class ModulesLibrary:

   154

   155

    MODULES_EXTRA_PARAMS = [

   156

        ("pdo_alignment", {

   157

            "column_label": _("PDO alignment"),

   158

            "column_size": 150,

   159

            "default": 8,

   160

            "description": _(

   161

"Minimal size in bits between 2 pdo entries")}),

   162

        ("max_pdo_size", {

   163

            "column_label": _("Max entries by PDO"),

   164

            "column_size": 150,

   165

            "default": 255,

   166

            "description": _(

   167

"""Maximal number of entries mapped in a PDO

   168

including empty entries used for PDO alignment""")}),

   169

        ("add_pdo", {

   170

            "column_label": _("Creating new PDO"),

   171

            "column_size": 150,

   172

            "default": 0,

   173

            "description": _(

   174

"""Adding a PDO not defined in default configuration

   175

for mapping needed location variables

   176

(1 if possible)""")})

   177

    ]

   178

   179

    def __init__(self, path, parent_library=None):

   180

        self.Path = path

   181

        if not os.path.exists(self.Path):

   182

            os.makedirs(self.Path)

   183

        self.ParentLibrary = parent_library

   184

   185

        if parent_library is not None:

   186

            self.LoadModules()

   187

        else:

   188

            self.Library = None

   189

        self.LoadModulesExtraParams()

   190

   191

    def GetPath(self):

   192

        return self.Path

   193

   194

    def GetModulesExtraParamsFilePath(self):

   195

        return os.path.join(self.Path, "modules_extra_params.cfg")

   196

   197

    groups_xpath = EtherCATInfo_XPath("Descriptions/Groups/Group")

   198

    devices_xpath = EtherCATInfo_XPath("Descriptions/Devices/Device")

   199

    def LoadModules(self):

   200

        self.Library = {}

   201

   202

        files = os.listdir(self.Path)

   203

        for file in files:

   204

            filepath = os.path.join(self.Path, file)

   205

            if os.path.isfile(filepath) and os.path.splitext(filepath)[-1] == ".xml":

   206

                self.modules_infos = None

   207

   208

                xmlfile = open(filepath, 'r')

   209

                try:

   210

                    self.modules_infos, error = EtherCATInfoParser.LoadXMLString(xmlfile.read())

   211

                    if error is not None:

   212

                        self.GetCTRoot().logger.write_warning(

   213

                            XSDSchemaErrorMessage % (filepath + error))

   214

                except Exception, exc:

   215

                    self.modules_infos, error = None, unicode(exc)

   216

                xmlfile.close()

   217

   218

                if self.modules_infos is not None:

   219

                    vendor = self.modules_infos.getVendor()

   220

   221

                    vendor_category = self.Library.setdefault(

   222

                        ExtractHexDecValue(vendor.getId()),

   223

                        {"name": ExtractName(vendor.getName(), _("Miscellaneous")),

   224

                         "groups": {}})

   225

   226

                    for group in self.groups_xpath(self.modules_infos):

   227

                        group_type = group.getType()

   228

   229

                        vendor_category["groups"].setdefault(group_type,

   230

                            {"name": ExtractName(group.getName(), group_type),

   231

                             "parent": group.getParentGroup(),

   232

                             "order": group.getSortOrder(),

   233

                             # "value": group.getcontent()["value"],

   234

                             "devices": []})

   235

   236

                    for device in self.devices_xpath(self.modules_infos):

   237

                        device_group = device.getGroupType()

   238

                        if not vendor_category["groups"].has_key(device_group):

   239

                            raise ValueError, "Not such group \"%\"" % device_group

   240

                        vendor_category["groups"][device_group]["devices"].append(

   241

                            (device.getType().getcontent(), device))

   242

   243

                else:

   244

   245

                    self.GetCTRoot().logger.write_error(

   246

                        _("Couldn't load %s XML file:\n%s") % (filepath, error))

   247

   248

        return self.Library

   249

   250

    def GetModulesLibrary(self, profile_filter=None):

   251

        if self.Library is None:

   252

            self.LoadModules()

   253

        library = []

   254

        for vendor_id, vendor in self.Library.iteritems():

   255

            groups = []

   256

            children_dict = {}

   257

            for group_type, group in vendor["groups"].iteritems():

   258

                group_infos = {"name": group["name"],

   259

                               "order": group["order"],

   260

                               "type": ETHERCAT_GROUP,

   261

                               "infos": None,

   262

                               "children": children_dict.setdefault(group_type, [])}

   263

                device_dict = {}

   264

                for device_type, device in group["devices"]:

   265

                    if profile_filter is None or profile_filter in device.GetProfileNumbers():

   266

                        product_code = device.getType().getProductCode()

   267

                        revision_number = device.getType().getRevisionNo()

   268

                        module_infos = {"device_type": device_type,

   269

                                        "vendor": vendor_id,

   270

                                        "product_code": product_code,

   271

                                        "revision_number": revision_number}

   272

                        module_infos.update(self.GetModuleExtraParams(vendor_id, product_code, revision_number))

   273

                        device_infos = {"name": ExtractName(device.getName()),

   274

                                        "type": ETHERCAT_DEVICE,

   275

                                        "infos": module_infos,

   276

                                        "children": []}

   277

                        group_infos["children"].append(device_infos)

   278

                        device_type_occurrences = device_dict.setdefault(device_type, [])

   279

                        device_type_occurrences.append(device_infos)

   280

                for device_type_occurrences in device_dict.itervalues():

   281

                    if len(device_type_occurrences) > 1:

   282

                        for occurrence in device_type_occurrences:

   283

                            occurrence["name"] += _(" (rev. %s)") % occurrence["infos"]["revision_number"]

   284

                if len(group_infos["children"]) > 0:

   285

                    if group["parent"] is not None:

   286

                        parent_children = children_dict.setdefault(group["parent"], [])

   287

                        parent_children.append(group_infos)

   288

                    else:

   289

                        groups.append(group_infos)

   290

            if len(groups) > 0:

   291

                library.append({"name": vendor["name"],

   292

                                "type": ETHERCAT_VENDOR,

   293

                                "infos": None,

   294

                                "children": groups})

   295

        library.sort(lambda x, y: cmp(x["name"], y["name"]))

   296

        return library

   297

   298

    def GetVendors(self):

   299

        return [(vendor_id, vendor["name"]) for vendor_id, vendor in self.Library.items()]

   300

   301

    def GetModuleInfos(self, module_infos):

   302

        vendor = ExtractHexDecValue(module_infos["vendor"])

   303

        vendor_infos = self.Library.get(vendor)

   304

        if vendor_infos is not None:

   305

            for group_name, group_infos in vendor_infos["groups"].iteritems():

   306

                for device_type, device_infos in group_infos["devices"]:

   307

                    product_code = ExtractHexDecValue(device_infos.getType().getProductCode())

   308

                    revision_number = ExtractHexDecValue(device_infos.getType().getRevisionNo())

   309

                    if (product_code == ExtractHexDecValue(module_infos["product_code"]) and

   310

                        revision_number == ExtractHexDecValue(module_infos["revision_number"])):

   311

                        self.cntdevice = device_infos

   312

                        self.cntdeviceType = device_type

   313

                        return device_infos, self.GetModuleExtraParams(vendor, product_code, revision_number)

   314

        return None, None

   315

   316

    def ImportModuleLibrary(self, filepath):

   317

        if os.path.isfile(filepath):

   318

            shutil.copy(filepath, self.Path)

   319

            self.LoadModules()

   320

            return True

   321

        return False

   322

   323

    def LoadModulesExtraParams(self):

   324

        self.ModulesExtraParams = {}

   325

   326

        csvfile_path = self.GetModulesExtraParamsFilePath()

   327

        if os.path.exists(csvfile_path):

   328

            csvfile = open(csvfile_path, "rb")

   329

            sample = csvfile.read(1024)

   330

            csvfile.seek(0)

   331

            dialect = csv.Sniffer().sniff(sample)

   332

            has_header = csv.Sniffer().has_header(sample)

   333

            reader = csv.reader(csvfile, dialect)

   334

            for row in reader:

   335

                if has_header:

   336

                    has_header = False

   337

                else:

   338

                    params_values = {}

   339

                    for (param, param_infos), value in zip(

   340

                        self.MODULES_EXTRA_PARAMS, row[3:]):

   341

                        if value != "":

   342

                            params_values[param] = int(value)

   343

                    self.ModulesExtraParams[

   344

                        tuple(map(int, row[:3]))] = params_values

   345

            csvfile.close()

   346

   347

    def SaveModulesExtraParams(self):

   348

        csvfile = open(self.GetModulesExtraParamsFilePath(), "wb")

   349

        extra_params = [param for param, params_infos in self.MODULES_EXTRA_PARAMS]

   350

        writer = csv.writer(csvfile, delimiter=';')

   351

        writer.writerow(['Vendor', 'product_code', 'revision_number'] + extra_params)

   352

        for (vendor, product_code, revision_number), module_extra_params in self.ModulesExtraParams.iteritems():

   353

            writer.writerow([vendor, product_code, revision_number] +

   354

                            [module_extra_params.get(param, '')

   355

                             for param in extra_params])

   356

        csvfile.close()

   357

   358

    def SetModuleExtraParam(self, vendor, product_code, revision_number, param, value):

   359

        vendor = ExtractHexDecValue(vendor)

   360

        product_code = ExtractHexDecValue(product_code)

   361

        revision_number = ExtractHexDecValue(revision_number)

   362

   363

        module_infos = (vendor, product_code, revision_number)

   364

        self.ModulesExtraParams.setdefault(module_infos, {})

   365

        self.ModulesExtraParams[module_infos][param] = value

   366

   367

        self.SaveModulesExtraParams()

   368

   369

    def GetModuleExtraParams(self, vendor, product_code, revision_number):

   370

        vendor = ExtractHexDecValue(vendor)

   371

        product_code = ExtractHexDecValue(product_code)

   372

        revision_number = ExtractHexDecValue(revision_number)

   373

   374

        if self.ParentLibrary is not None:

   375

            extra_params = self.ParentLibrary.GetModuleExtraParams(vendor, product_code, revision_number)

   376

        else:

   377

            extra_params = {}

   378

   379

        extra_params.update(self.ModulesExtraParams.get((vendor, product_code, revision_number), {}))

   380

   381

        for param, param_infos in self.MODULES_EXTRA_PARAMS:

   382

            extra_params.setdefault(param, param_infos["default"])

   383

   384

        return extra_params

   385

   386

USERDATA_DIR = wx.StandardPaths.Get().GetUserDataDir()

   387

if wx.Platform != '__WXMSW__':

   388

    USERDATA_DIR += '_files'

   389

   390

ModulesDatabase = ModulesLibrary(

   391

    os.path.join(USERDATA_DIR, "ethercat_modules"))

   392

   393

   394

class RootClass:

   395

   396

    CTNChildrenTypes = [("EthercatNode", _EthercatCTN, "Ethercat Master")]

   397

    EditorType = LibraryEditor

   398

   399

    def __init__(self):

   400

        self.ModulesLibrary = None

   401

        self.LoadModulesLibrary()

   402

   403

    def GetIconName(self):

   404

        return "Ethercat"

   405

   406

    def GetModulesLibraryPath(self, project_path=None):

   407

        if project_path is None:

   408

            project_path = self.CTNPath()

   409

        return os.path.join(project_path, "modules")

   410

   411

    def OnCTNSave(self, from_project_path=None):

   412

        if from_project_path is not None:

   413

            shutil.copytree(self.GetModulesLibraryPath(from_project_path),

   414

                            self.GetModulesLibraryPath())

   415

        return True

   416

   417

    def CTNGenerate_C(self, buildpath, locations):

   418

        return [], "", False

   419

   420

    def LoadModulesLibrary(self):

   421

        if self.ModulesLibrary is None:

   422

            self.ModulesLibrary = ModulesLibrary(self.GetModulesLibraryPath(), ModulesDatabase)

   423

        else:

   424

            self.ModulesLibrary.LoadModulesLibrary()

   425

   426

    def GetModulesDatabaseInstance(self):

   427

        return ModulesDatabase

   428

   429

    def GetModulesLibraryInstance(self):

   430

        return self.ModulesLibrary

   431

   432

    def GetModulesLibrary(self, profile_filter=None):

   433

        return self.ModulesLibrary.GetModulesLibrary(profile_filter)

   434

   435

    def GetVendors(self):

   436

        return self.ModulesLibrary.GetVendors()

   437

   438

    def GetModuleInfos(self, module_infos):

   439

        return self.ModulesLibrary.GetModuleInfos(module_infos)