1

#!/usr/bin/env python

     2

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

     3

     4

#This file is part of PLCOpenEditor, a library implementing an IEC 61131-3 editor

     5

#based on the plcopen standard. 

     6

#

     7

#Copyright (C) 2007: Edouard TISSERANT and Laurent BESSARD

     8

#

     9

#See COPYING file for copyrights details.

    10

#

    11

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

    12

#modify it under the terms of the GNU General Public

    13

#License as published by the Free Software Foundation; either

    14

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

    15

#

    16

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

    17

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

    18

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

    19

#General Public License for more details.

    20

#

    21

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

    22

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

    23

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

    24

    25

from xmlclass import *

    26

from structures import *

    27

from types import *

    28

import os, re

    29

    30

"""

    31

Dictionary that makes the relation between var names in plcopen and displayed values

    32

"""

    33

VarTypes = {"Local" : "localVars", "Temp" : "tempVars", "Input" : "inputVars",

    34

            "Output" : "outputVars", "InOut" : "inOutVars", "External" : "externalVars",

    35

            "Global" : "globalVars", "Access" : "accessVars"}

    36

    37

searchResultVarTypes = {

    38

    "inputVars": "var_input",

    39

    "outputVars": "var_output",

    40

    "inOutVars": "var_inout"

    41

}

    42

    43

"""

    44

Define in which order var types must be displayed

    45

"""

    46

VarOrder = ["Local","Temp","Input","Output","InOut","External","Global","Access"]

    47

    48

"""

    49

Define which action qualifier must be associated with a duration 

    50

"""

    51

QualifierList = {"N" : False, "R" : False, "S" : False, "L" : True, "D" : True, 

    52

    "P" : False, "P0" : False, "P1" : False, "SD" : True, "DS" : True, "SL" : True}

    53

    54

    55

FILTER_ADDRESS_MODEL = "(%%[IQM](?:[XBWDL])?)(%s)((?:\.[0-9]+)*)" 

    56

    57

def update_address(address, address_model, new_leading):

    58

    result = address_model.match(address)

    59

    if result is None:

    60

        return address

    61

    groups = result.groups()

    62

    return groups[0] + new_leading + groups[2]

    63

    64

def _init_and_compare(function, v1, v2):

    65

    if v1 is None:

    66

        return v2

    67

    if v2 is not None:

    68

        return function(v1, v2)

    69

    return v1

    70

    71

"""

    72

Helper class for bounding_box calculation 

    73

"""

    74

class rect:

    75

    76

    def __init__(self, x=None, y=None, width=None, height=None):

    77

        self.x_min = x

    78

        self.x_max = None

    79

        self.y_min = y

    80

        self.y_max = None

    81

        if width is not None and x is not None:

    82

            self.x_max = x + width

    83

        if height is not None and y is not None:

    84

            self.y_max = y + height

    85

    86

    def update(self, x, y):

    87

        self.x_min = _init_and_compare(min, self.x_min, x)

    88

        self.x_max = _init_and_compare(max, self.x_max, x)

    89

        self.y_min = _init_and_compare(min, self.y_min, y)

    90

        self.y_max = _init_and_compare(max, self.y_max, y)

    91

    92

    def union(self, rect):

    93

        self.x_min = _init_and_compare(min, self.x_min, rect.x_min)

    94

        self.x_max = _init_and_compare(max, self.x_max, rect.x_max)

    95

        self.y_min = _init_and_compare(min, self.y_min, rect.y_min)

    96

        self.y_max = _init_and_compare(max, self.y_max, rect.y_max)

    97

    98

    def bounding_box(self):

    99

        width = height = None

   100

        if self.x_min is not None and self.x_max is not None:

   101

            width = self.x_max - self.x_min

   102

        if self.y_min is not None and self.y_max is not None:

   103

            height = self.y_max - self.y_min

   104

        return self.x_min, self.y_min, width, height

   105

   106

def TextLenInRowColumn(text):

   107

    if text == "":

   108

        return (0, 0)

   109

    lines = text.split("\n")

   110

    return len(lines) - 1, len(lines[-1])

   111

   112

def TestTextElement(text, criteria):

   113

    lines = text.splitlines()

   114

    if not criteria["case_sensitive"]:

   115

        text = text.upper()

   116

    test_result = []

   117

    result = criteria["pattern"].search(text)

   118

    while result is not None:

   119

        start = TextLenInRowColumn(text[:result.start()])

   120

        end = TextLenInRowColumn(text[:result.end() - 1])

   121

        test_result.append((start, end, "\n".join(lines[start[0]:end[0] + 1])))

   122

        result = criteria["pattern"].search(text, result.end())

   123

    return test_result

   124

   125

PLCOpenClasses = GenerateClassesFromXSD(os.path.join(os.path.split(__file__)[0], "tc6_xml_v201.xsd"))

   126

   127

ElementNameToClass = {}

   128

   129

cls = PLCOpenClasses.get("formattedText", None)

   130

if cls:

   131

    def updateElementName(self, old_name, new_name):

   132

        text = self.text

   133

        index = text.find(old_name)

   134

        while index != -1:

   135

            if index > 0 and (text[index - 1].isalnum() or text[index - 1] == "_"):

   136

                index = text.find(old_name, index + len(old_name))

   137

            elif index < len(text) - len(old_name) and (text[index + len(old_name)].isalnum() or text[index + len(old_name)] == "_"):

   138

                index = text.find(old_name, index + len(old_name))

   139

            else:

   140

                text = text[:index] + new_name + text[index + len(old_name):]

   141

                index = text.find(old_name, index + len(new_name))

   142

        self.text = text

   143

    setattr(cls, "updateElementName", updateElementName)

   144

   145

    def updateElementAddress(self, address_model, new_leading):

   146

        text = self.text

   147

        startpos = 0

   148

        result = address_model.search(text, startpos)

   149

        while result is not None:

   150

            groups = result.groups()

   151

            new_address = groups[0] + new_leading + groups[2]

   152

            text = text[:result.start()] + new_address + text[result.end():]

   153

            startpos = result.start() + len(new_address)

   154

            result = address_model.search(self.text, startpos)

   155

        self.text = text

   156

    setattr(cls, "updateElementAddress", updateElementAddress)

   157

   158

    def hasblock(self, block_type):

   159

        text = self.text.upper()

   160

        index = text.find(block_type.upper())

   161

        while index != -1:

   162

            if (not (index > 0 and (text[index - 1].isalnum() or text[index - 1] == "_")) and 

   163

                not (index < len(text) - len(block_type) and text[index + len(block_type)] != "(")):

   164

                return True

   165

            index = text.find(block_type.upper(), index + len(block_type))

   166

        return False

   167

    setattr(cls, "hasblock", hasblock)

   168

   169

    def Search(self, criteria, parent_infos):

   170

        return [(tuple(parent_infos),) + result for result in TestTextElement(self.gettext(), criteria)]

   171

    setattr(cls, "Search", Search)

   172

   173

cls = PLCOpenClasses.get("project", None)

   174

if cls:

   175

    cls.singleLineAttributes = False

   176

    cls.EnumeratedDataTypeValues = {}

   177

    cls.CustomDataTypeRange = {}

   178

    cls.CustomTypeHierarchy = {}

   179

    cls.ElementUsingTree = {}

   180

    cls.CustomBlockTypes = []

   181

   182

    def setname(self, name):

   183

        self.contentHeader.setname(name)

   184

    setattr(cls, "setname", setname)

   185

   186

    def getname(self):

   187

        return self.contentHeader.getname()

   188

    setattr(cls, "getname", getname)

   189

   190

    def getfileHeader(self):

   191

        fileheader = {}

   192

        for name, value in [("companyName", self.fileHeader.getcompanyName()),

   193

                            ("companyURL", self.fileHeader.getcompanyURL()),

   194

                            ("productName", self.fileHeader.getproductName()),

   195

                            ("productVersion", self.fileHeader.getproductVersion()),

   196

                            ("productRelease", self.fileHeader.getproductRelease()),

   197

                            ("creationDateTime", self.fileHeader.getcreationDateTime()),

   198

                            ("contentDescription", self.fileHeader.getcontentDescription())]:

   199

            if value is not None:

   200

                fileheader[name] = value

   201

            else:

   202

                fileheader[name] = ""

   203

        return fileheader

   204

    setattr(cls, "getfileHeader", getfileHeader)

   205

   206

    def setfileHeader(self, fileheader):

   207

        if fileheader.has_key("companyName"):

   208

            self.fileHeader.setcompanyName(fileheader["companyName"])

   209

        if fileheader.has_key("companyURL"):

   210

            self.fileHeader.setcompanyURL(fileheader["companyURL"])

   211

        if fileheader.has_key("productName"):

   212

            self.fileHeader.setproductName(fileheader["productName"])

   213

        if fileheader.has_key("productVersion"):

   214

            self.fileHeader.setproductVersion(fileheader["productVersion"])

   215

        if fileheader.has_key("productRelease"):

   216

            self.fileHeader.setproductRelease(fileheader["productRelease"])

   217

        if fileheader.has_key("creationDateTime"):

   218

            self.fileHeader.setcreationDateTime(fileheader["creationDateTime"])

   219

        if fileheader.has_key("contentDescription"):

   220

            self.fileHeader.setcontentDescription(fileheader["contentDescription"])

   221

    setattr(cls, "setfileHeader", setfileHeader)

   222

   223

    def getcontentHeader(self):

   224

        contentheader = {}

   225

        for name, value in [("projectName", self.contentHeader.getname()),

   226

                            ("projectVersion", self.contentHeader.getversion()),

   227

                            ("modificationDateTime", self.contentHeader.getmodificationDateTime()),

   228

                            ("organization", self.contentHeader.getorganization()),

   229

                            ("authorName", self.contentHeader.getauthor()),

   230

                            ("language", self.contentHeader.getlanguage())]:

   231

            if value is not None:

   232

                contentheader[name] = value

   233

            else:

   234

                contentheader[name] = ""

   235

        contentheader["pageSize"] = self.contentHeader.getpageSize()

   236

        contentheader["scaling"] = self.contentHeader.getscaling()

   237

        return contentheader

   238

    setattr(cls, "getcontentHeader", getcontentHeader)

   239

   240

    def setcontentHeader(self, contentheader):

   241

        if contentheader.has_key("projectName"):

   242

            self.contentHeader.setname(contentheader["projectName"])

   243

        if contentheader.has_key("projectVersion"):

   244

            self.contentHeader.setversion(contentheader["projectVersion"])

   245

        if contentheader.has_key("modificationDateTime"):

   246

            self.contentHeader.setmodificationDateTime(contentheader["modificationDateTime"])

   247

        if contentheader.has_key("organization"):

   248

            self.contentHeader.setorganization(contentheader["organization"])

   249

        if contentheader.has_key("authorName"):

   250

            self.contentHeader.setauthor(contentheader["authorName"])

   251

        if contentheader.has_key("language"):

   252

            self.contentHeader.setlanguage(contentheader["language"])

   253

        if contentheader.has_key("pageSize"):

   254

            self.contentHeader.setpageSize(*contentheader["pageSize"])

   255

        if contentheader.has_key("scaling"):

   256

            self.contentHeader.setscaling(contentheader["scaling"])

   257

    setattr(cls, "setcontentHeader", setcontentHeader)

   258

   259

    def getdataTypes(self):

   260

        return self.types.getdataTypeElements()

   261

    setattr(cls, "getdataTypes", getdataTypes)

   262

   263

    def getdataType(self, name):

   264

        return self.types.getdataTypeElement(name)

   265

    setattr(cls, "getdataType", getdataType)

   266

   267

    def appenddataType(self, name):

   268

        if self.CustomTypeHierarchy.has_key(name):

   269

            raise ValueError, "\"%s\" Data Type already exists !!!"%name

   270

        self.types.appenddataTypeElement(name)

   271

        self.AddCustomDataType(self.getdataType(name))

   272

    setattr(cls, "appenddataType", appenddataType)

   273

   274

    def insertdataType(self, index, datatype):

   275

        self.types.insertdataTypeElement(index, datatype)

   276

        self.AddCustomDataType(datatype)

   277

    setattr(cls, "insertdataType", insertdataType)

   278

   279

    def removedataType(self, name):

   280

        self.types.removedataTypeElement(name)

   281

        self.RefreshDataTypeHierarchy()

   282

        self.RefreshElementUsingTree()

   283

    setattr(cls, "removedataType", removedataType)

   284

   285

    def getpous(self):

   286

        return self.types.getpouElements()

   287

    setattr(cls, "getpous", getpous)

   288

   289

    def getpou(self, name):

   290

        return self.types.getpouElement(name)

   291

    setattr(cls, "getpou", getpou)

   292

   293

    def appendpou(self, name, pou_type, body_type):

   294

        self.types.appendpouElement(name, pou_type, body_type)

   295

        self.AddCustomBlockType(self.getpou(name))

   296

    setattr(cls, "appendpou", appendpou)

   297

   298

    def insertpou(self, index, pou):

   299

        self.types.insertpouElement(index, pou)

   300

        self.AddCustomBlockType(pou)

   301

    setattr(cls, "insertpou", insertpou)

   302

   303

    def removepou(self, name):

   304

        self.types.removepouElement(name)

   305

        self.RefreshCustomBlockTypes()

   306

        self.RefreshElementUsingTree()

   307

    setattr(cls, "removepou", removepou)

   308

   309

    def getconfigurations(self):

   310

        configurations = self.instances.configurations.getconfiguration()

   311

        if configurations:

   312

            return configurations

   313

        return []

   314

    setattr(cls, "getconfigurations", getconfigurations)

   315

   316

    def getconfiguration(self, name):

   317

        for configuration in self.instances.configurations.getconfiguration():

   318

            if configuration.getname() == name:

   319

                return configuration

   320

        return None

   321

    setattr(cls, "getconfiguration", getconfiguration)

   322

   323

    def addconfiguration(self, name):

   324

        for configuration in self.instances.configurations.getconfiguration():

   325

            if configuration.getname() == name:

   326

                raise ValueError, _("\"%s\" configuration already exists !!!")%name

   327

        new_configuration = PLCOpenClasses["configurations_configuration"]()

   328

        new_configuration.setname(name)

   329

        self.instances.configurations.appendconfiguration(new_configuration)

   330

    setattr(cls, "addconfiguration", addconfiguration)    

   331

   332

    def removeconfiguration(self, name):

   333

        found = False

   334

        for idx, configuration in enumerate(self.instances.configurations.getconfiguration()):

   335

            if configuration.getname() == name:

   336

                self.instances.configurations.removeconfiguration(idx)

   337

                found = True

   338

                break

   339

        if not found:

   340

            raise ValueError, ("\"%s\" configuration doesn't exist !!!")%name

   341

    setattr(cls, "removeconfiguration", removeconfiguration)

   342

   343

    def getconfigurationResource(self, config_name, name):

   344

        configuration = self.getconfiguration(config_name)

   345

        if configuration:

   346

            for resource in configuration.getresource():

   347

                if resource.getname() == name:

   348

                    return resource

   349

        return None

   350

    setattr(cls, "getconfigurationResource", getconfigurationResource)

   351

   352

    def addconfigurationResource(self, config_name, name):

   353

        configuration = self.getconfiguration(config_name)

   354

        if configuration:

   355

            for resource in configuration.getresource():

   356

                if resource.getname() == name:

   357

                    raise ValueError, _("\"%s\" resource already exists in \"%s\" configuration !!!")%(name, config_name)

   358

            new_resource = PLCOpenClasses["configuration_resource"]()

   359

            new_resource.setname(name)

   360

            configuration.appendresource(new_resource)

   361

    setattr(cls, "addconfigurationResource", addconfigurationResource)

   362

   363

    def removeconfigurationResource(self, config_name, name):

   364

        configuration = self.getconfiguration(config_name)

   365

        if configuration:

   366

            found = False

   367

            for idx, resource in enumerate(configuration.getresource()):

   368

                if resource.getname() == name:

   369

                    configuration.removeresource(idx)

   370

                    found = True

   371

                    break

   372

            if not found:

   373

                raise ValueError, _("\"%s\" resource doesn't exist in \"%s\" configuration !!!")%(name, config_name)

   374

    setattr(cls, "removeconfigurationResource", removeconfigurationResource)

   375

   376

    def updateElementName(self, old_name, new_name):

   377

        for datatype in self.types.getdataTypeElements():

   378

            datatype.updateElementName(old_name, new_name)

   379

        for pou in self.types.getpouElements():

   380

            pou.updateElementName(old_name, new_name)

   381

        for configuration in self.instances.configurations.getconfiguration():

   382

            configuration.updateElementName(old_name, new_name)

   383

    setattr(cls, "updateElementName", updateElementName)

   384

   385

    def updateElementAddress(self, old_leading, new_leading):

   386

        address_model = re.compile(FILTER_ADDRESS_MODEL % old_leading)

   387

        for pou in self.types.getpouElements():

   388

            pou.updateElementAddress(address_model, new_leading)

   389

        for configuration in self.instances.configurations.getconfiguration():

   390

            configuration.updateElementAddress(address_model, new_leading)

   391

    setattr(cls, "updateElementAddress", updateElementAddress)

   392

   393

    def removeVariableByAddress(self, address):

   394

        for pou in self.types.getpouElements():

   395

            pou.removeVariableByAddress(address)

   396

        for configuration in self.instances.configurations.getconfiguration():

   397

            configuration.removeVariableByAddress(address)

   398

    setattr(cls, "removeVariableByAddress", removeVariableByAddress)

   399

   400

    def removeVariableByFilter(self, leading):

   401

        address_model = re.compile(FILTER_ADDRESS_MODEL % leading)

   402

        for pou in self.types.getpouElements():

   403

            pou.removeVariableByFilter(address_model)

   404

        for configuration in self.instances.configurations.getconfiguration():

   405

            configuration.removeVariableByFilter(address_model)

   406

    setattr(cls, "removeVariableByFilter", removeVariableByFilter)

   407

   408

    def RefreshDataTypeHierarchy(self):

   409

        self.EnumeratedDataTypeValues = {}

   410

        self.CustomDataTypeRange = {}

   411

        self.CustomTypeHierarchy = {}

   412

        for datatype in self.getdataTypes():

   413

            self.AddCustomDataType(datatype)

   414

    setattr(cls, "RefreshDataTypeHierarchy", RefreshDataTypeHierarchy)

   415

   416

    def AddCustomDataType(self, datatype):

   417

        name = datatype.getname()

   418

        basetype_content = datatype.getbaseType().getcontent()

   419

        if basetype_content["value"] is None:

   420

            self.CustomTypeHierarchy[name] = basetype_content["name"]

   421

        elif basetype_content["name"] in ["string", "wstring"]:

   422

            self.CustomTypeHierarchy[name] = basetype_content["name"].upper()

   423

        elif basetype_content["name"] == "derived":

   424

            self.CustomTypeHierarchy[name] = basetype_content["value"].getname()

   425

        elif basetype_content["name"] in ["subrangeSigned", "subrangeUnsigned"]:

   426

            range = (basetype_content["value"].range.getlower(), 

   427

                     basetype_content["value"].range.getupper())

   428

            self.CustomDataTypeRange[name] = range

   429

            base_type = basetype_content["value"].baseType.getcontent()

   430

            if base_type["value"] is None:

   431

                self.CustomTypeHierarchy[name] = base_type["name"]

   432

            else:

   433

                self.CustomTypeHierarchy[name] = base_type["value"].getname()

   434

        else:

   435

            if basetype_content["name"] == "enum":

   436

                values = []

   437

                for value in basetype_content["value"].values.getvalue():

   438

                    values.append(value.getname())

   439

                self.EnumeratedDataTypeValues[name] = values

   440

            self.CustomTypeHierarchy[name] = "ANY_DERIVED"

   441

    setattr(cls, "AddCustomDataType", AddCustomDataType)

   442

   443

    # Update Block types with user-defined pou added

   444

    def RefreshCustomBlockTypes(self):

   445

        # Reset the tree of user-defined pou cross-use

   446

        self.CustomBlockTypes = []

   447

        for pou in self.getpous():

   448

            self.AddCustomBlockType(pou)

   449

    setattr(cls, "RefreshCustomBlockTypes", RefreshCustomBlockTypes)

   450

   451

    def AddCustomBlockType(self, pou): 

   452

        pou_name = pou.getname()

   453

        pou_type = pou.getpouType()

   454

        block_infos = {"name" : pou_name, "type" : pou_type, "extensible" : False,

   455

                       "inputs" : [], "outputs" : [], "comment" : pou.getdescription(),

   456

                       "generate" : generate_block, "initialise" : initialise_block}

   457

        if pou.getinterface():

   458

            return_type = pou.interface.getreturnType()

   459

            if return_type:

   460

                var_type = return_type.getcontent()

   461

                if var_type["name"] == "derived":

   462

                    block_infos["outputs"].append(("OUT", var_type["value"].getname(), "none"))

   463

                elif var_type["name"] in ["string", "wstring"]:

   464

                    block_infos["outputs"].append(("OUT", var_type["name"].upper(), "none"))

   465

                else:

   466

                    block_infos["outputs"].append(("OUT", var_type["name"], "none"))

   467

            for type, varlist in pou.getvars():

   468

                if type == "InOut":

   469

                    for var in varlist.getvariable():

   470

                        var_type = var.type.getcontent()

   471

                        if var_type["name"] == "derived":

   472

                            block_infos["inputs"].append((var.getname(), var_type["value"].getname(), "none"))

   473

                            block_infos["outputs"].append((var.getname(), var_type["value"].getname(), "none"))

   474

                        elif var_type["name"] in ["string", "wstring"]:

   475

                            block_infos["inputs"].append((var.getname(), var_type["name"].upper(), "none"))

   476

                            block_infos["outputs"].append((var.getname(), var_type["name"].upper(), "none"))

   477

                        else:

   478

                            block_infos["inputs"].append((var.getname(), var_type["name"], "none"))

   479

                            block_infos["outputs"].append((var.getname(), var_type["name"], "none"))

   480

                elif type == "Input":

   481

                    for var in varlist.getvariable():

   482

                        var_type = var.type.getcontent()

   483

                        if var_type["name"] == "derived":

   484

                            block_infos["inputs"].append((var.getname(), var_type["value"].getname(), "none"))

   485

                        elif var_type["name"] in ["string", "wstring"]:

   486

                            block_infos["inputs"].append((var.getname(), var_type["name"].upper(), "none"))

   487

                        else:

   488

                            block_infos["inputs"].append((var.getname(), var_type["name"], "none"))

   489

                elif type == "Output":

   490

                    for var in varlist.getvariable():

   491

                        var_type = var.type.getcontent()

   492

                        if var_type["name"] == "derived":

   493

                            block_infos["outputs"].append((var.getname(), var_type["value"].getname(), "none"))

   494

                        elif var_type["name"] in ["string", "wstring"]:

   495

                            block_infos["outputs"].append((var.getname(), var_type["name"].upper(), "none"))

   496

                        else:

   497

                            block_infos["outputs"].append((var.getname(), var_type["name"], "none"))    

   498

        block_infos["usage"] = "\n (%s) => (%s)" % (", ".join(["%s:%s" % (input[1], input[0]) for input in block_infos["inputs"]]),

   499

                                                    ", ".join(["%s:%s" % (output[1], output[0]) for output in block_infos["outputs"]]))

   500

        self.CustomBlockTypes.append(block_infos)

   501

    setattr(cls, "AddCustomBlockType", AddCustomBlockType)

   502

   511

    def RefreshElementUsingTree(self):

   512

        # Reset the tree of user-defined element cross-use

   513

        self.ElementUsingTree = {}

   514

        pous = self.getpous()

   515

        datatypes = self.getdataTypes()

   516

        # Analyze each datatype

   517

        for datatype in datatypes:

   518

            name = datatype.getname()

   519

            basetype_content = datatype.baseType.getcontent()

   520

            if basetype_content["name"] == "derived":

   521

                typename = basetype_content["value"].getname()

   522

                if name in self.ElementUsingTree[typename]:

   523

                    self.ElementUsingTree[typename].append(name)

   524

            elif basetype_content["name"] in ["subrangeSigned", "subrangeUnsigned", "array"]:

   525

                base_type = basetype_content["value"].baseType.getcontent()

   526

                if base_type["name"] == "derived":