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

"""

    38

Define in which order var types must be displayed

    39

"""

    40

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

    41

    42

"""

    43

Define which action qualifier must be associated with a duration 

    44

"""

    45

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

    46

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

    47

    48

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

    49

    50

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

    51

if cls:

    52

    def updateElementName(self, old_name, new_name):

    53

        index = self.text.find(old_name)

    54

        while index != -1:

    55

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

    56

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

    57

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

    58

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

    59

            else:

    60

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

    61

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

    62

    setattr(cls, "updateElementName", updateElementName)

    63

    64

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

    65

if cls:

    66

    cls.singleLineAttributes = False

    67

    cls.EnumeratedDataTypeValues = {}

    68

    cls.CustomDataTypeRange = {}

    69

    cls.CustomTypeHierarchy = {}

    70

    cls.ElementUsingTree = {}

    71

    cls.CustomBlockTypes = []

    72

    73

    def setname(self, name):

    74

        self.contentHeader.setname(name)

    75

    setattr(cls, "setname", setname)

    76

    77

    def getname(self):

    78

        return self.contentHeader.getname()

    79

    setattr(cls, "getname", getname)

    80

    81

    def getfileHeader(self):

    82

        fileheader = {}

    83

        fileheader["companyName"] = self.fileHeader.getcompanyName()

    84

        if self.fileHeader.getcompanyURL():

    85

            fileheader["companyURL"] = self.fileHeader.getcompanyURL()

    86

        fileheader["productName"] = self.fileHeader.getproductName()

    87

        fileheader["productVersion"] = self.fileHeader.getproductVersion()

    88

        if self.fileHeader.getproductRelease():

    89

            fileheader["productRelease"] = self.fileHeader.getproductRelease()

    90

        fileheader["creationDateTime"] = self.fileHeader.getcreationDateTime()

    91

        if self.fileHeader.getcontentDescription():

    92

            fileheader["contentDescription"] = self.fileHeader.getcontentDescription()

    93

        return fileheader

    94

    setattr(cls, "getfileHeader", getfileHeader)

    95

    96

    def setfileHeader(self, fileheader):

    97

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

    98

        if "companyURL" in fileheader:

    99

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

   100

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

   101

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

   102

        if "productRelease" in fileheader:

   103

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

   104

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

   105

        if "contentDescription" in fileheader:

   106

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

   107

    setattr(cls, "setfileHeader", setfileHeader)

   108

   109

    def getcontentHeader(self):

   110

        contentheader = {}

   111

        contentheader["projectName"] = self.contentHeader.getname()

   112

        if self.contentHeader.getversion():

   113

            contentheader["projectVersion"] = self.contentHeader.getversion()

   114

        if self.contentHeader.getmodificationDateTime():

   115

            contentheader["modificationDateTime"] = self.contentHeader.getmodificationDateTime()

   116

        if self.contentHeader.getorganization():

   117

            contentheader["organization"] = self.contentHeader.getorganization()

   118

        if self.contentHeader.getauthor():

   119

            contentheader["authorName"] = self.contentHeader.getauthor()

   120

        if self.contentHeader.getlanguage():

   121

            contentheader["language"] = self.contentHeader.getlanguage()

   122

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

   123

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

   124

        return contentheader

   125

    setattr(cls, "getcontentHeader", getcontentHeader)

   126

   127

    def setcontentHeader(self, contentheader):

   128

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

   129

        if "projectVersion" in contentheader:

   130

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

   131

        if "modificationDateTime" in contentheader:

   132

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

   133

        if "organization" in contentheader:

   134

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

   135

        if "authorName" in contentheader:

   136

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

   137

        if "language" in contentheader:

   138

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

   139

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

   140

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

   141

    setattr(cls, "setcontentHeader", setcontentHeader)

   142

   143

    def getdataTypes(self):

   144

        return self.types.getdataTypeElements()

   145

    setattr(cls, "getdataTypes", getdataTypes)

   146

   147

    def getdataType(self, name):

   148

        return self.types.getdataTypeElement(name)

   149

    setattr(cls, "getdataType", getdataType)

   150

   151

    def appenddataType(self, name):

   152

        if name in self.CustomTypeHierarchy:

   153

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

   154

        self.types.appenddataTypeElement(name)

   155

        self.AddCustomDataType(self.getdataType(name))

   156

    setattr(cls, "appenddataType", appenddataType)

   157

   158

    def insertdataType(self, index, datatype):

   159

        self.types.insertdataTypeElement(index, datatype)

   160

        self.AddCustomDataType(datatype)

   161

    setattr(cls, "insertdataType", insertdataType)

   162

   163

    def removedataType(self, name):

   164

        self.types.removedataTypeElement(name)

   165

        self.RefreshDataTypeHierarchy()

   166

        self.RefreshElementUsingTree()

   167

    setattr(cls, "removedataType", removedataType)

   168

   169

    def getpous(self):

   170

        return self.types.getpouElements()

   171

    setattr(cls, "getpous", getpous)

   172

   173

    def getpou(self, name):

   174

        return self.types.getpouElement(name)

   175

    setattr(cls, "getpou", getpou)

   176

   177

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

   178

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

   179

        self.AddCustomBlockType(self.getpou(name))

   180

    setattr(cls, "appendpou", appendpou)

   181

   182

    def insertpou(self, index, pou):

   183

        self.types.insertpouElement(index, pou)

   184

        self.AddCustomBlockType(pou)

   185

    setattr(cls, "insertpou", insertpou)

   186

   187

    def removepou(self, name):

   188

        self.types.removepouElement(name)

   189

        self.RefreshCustomBlockTypes()

   190

        self.RefreshElementUsingTree()

   191

    setattr(cls, "removepou", removepou)

   192

   193

    def getconfigurations(self):

   194

        configurations = self.instances.configurations.getconfiguration()

   195

        if configurations:

   196

            return configurations

   197

        return []

   198

    setattr(cls, "getconfigurations", getconfigurations)

   199

   200

    def getconfiguration(self, name):

   201

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

   202

            if configuration.getname() == name:

   203

                return configuration

   204

        return None

   205

    setattr(cls, "getconfiguration", getconfiguration)

   206

   207

    def addconfiguration(self, name):

   208

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

   209

            if configuration.getname() == name:

   210

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

   211

        new_configuration = PLCOpenClasses["configurations_configuration"]()

   212

        new_configuration.setname(name)

   213

        self.instances.configurations.appendconfiguration(new_configuration)

   214

    setattr(cls, "addconfiguration", addconfiguration)    

   215

   216

    def removeconfiguration(self, name):

   217

        found = False

   218

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

   219

            if configuration.getname() == name:

   220

                self.instances.configurations.removeconfiguration(idx)

   221

                found = True

   222

                break

   223

        if not found:

   224

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

   225

    setattr(cls, "removeconfiguration", removeconfiguration)

   226

   227

    def getconfigurationResource(self, config_name, name):

   228

        configuration = self.getconfiguration(config_name)

   229

        if configuration:

   230

            for resource in configuration.getresource():

   231

                if resource.getname() == name:

   232

                    return resource

   233

        return None

   234

    setattr(cls, "getconfigurationResource", getconfigurationResource)

   235

   236

    def addconfigurationResource(self, config_name, name):

   237

        configuration = self.getconfiguration(config_name)

   238

        if configuration:

   239

            for resource in configuration.getresource():

   240

                if resource.getname() == name:

   241

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

   242

            new_resource = PLCOpenClasses["configuration_resource"]()

   243

            new_resource.setname(name)

   244

            configuration.appendresource(new_resource)

   245

    setattr(cls, "addconfigurationResource", addconfigurationResource)

   246

   247

    def removeconfigurationResource(self, config_name, name):

   248

        configuration = self.getconfiguration(config_name)

   249

        if configuration:

   250

            found = False

   251

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

   252

                if resource.getname() == name:

   253

                    configuration.removeresource(idx)

   254

                    found = True

   255

                    break

   256

            if not found:

   257

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

   258

    setattr(cls, "removeconfigurationResource", removeconfigurationResource)

   259

   260

    def updateElementName(self, old_name, new_name):

   261

        for pou in self.types.getpouElements():

   262

            pou.updateElementName(old_name, new_name)

   263

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

   264

            configuration.updateElementName(old_name, new_name)

   265

    setattr(cls, "updateElementName", updateElementName)

   266

   267

    def RefreshDataTypeHierarchy(self):

   268

        self.EnumeratedDataTypeValues = {}

   269

        self.CustomDataTypeRange = {}

   270

        self.CustomTypeHierarchy = {}

   271

        for datatype in self.getdataTypes():

   272

            self.AddCustomDataType(datatype)

   273

    setattr(cls, "RefreshDataTypeHierarchy", RefreshDataTypeHierarchy)

   274

   275

    def AddCustomDataType(self, datatype):

   276

        name = datatype.getname()

   277

        basetype_content = datatype.getbaseType().getcontent()

   278

        if basetype_content["value"] is None:

   279

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

   280

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

   281

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

   282

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

   283

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

   284

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

   285

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

   286

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

   287

            self.CustomDataTypeRange[name] = range

   288

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

   289

            if base_type["value"] is None:

   290

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

   291

            else:

   292

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

   293

        else:

   294

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

   295

                values = []

   296

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

   297

                    values.append(value.getname())

   298

                self.EnumeratedDataTypeValues[name] = values

   299

            self.CustomTypeHierarchy[name] = "ANY_DERIVED"

   300

    setattr(cls, "AddCustomDataType", AddCustomDataType)

   301

   302

    # Update Block types with user-defined pou added

   303

    def RefreshCustomBlockTypes(self):

   304

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

   305

        self.CustomBlockTypes = []

   306

        for pou in self.getpous():

   307

            self.AddCustomBlockType(pou)

   308

    setattr(cls, "RefreshCustomBlockTypes", RefreshCustomBlockTypes)

   309

   310

    def AddCustomBlockType(self, pou): 

   311

        pou_name = pou.getname()

   312

        pou_type = pou.getpouType()

   313

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

   314

                       "inputs" : [], "outputs" : [], "comment" : "",

   315

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

   316

        if pou.getinterface():

   317

            return_type = pou.interface.getreturnType()

   318

            if return_type:

   319

                var_type = return_type.getcontent()

   320

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

   321

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

   322

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

   323

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

   324

                else:

   325

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

   326

            for type, varlist in pou.getvars():

   327

                if type == "InOut":

   328

                    for var in varlist.getvariable():

   329

                        var_type = var.type.getcontent()

   330

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

   331

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

   332

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

   333

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

   334

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

   335

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

   336

                        else:

   337

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

   338

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

   339

                elif type == "Input":

   340

                    for var in varlist.getvariable():

   341

                        var_type = var.type.getcontent()

   342

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

   343

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

   344

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

   345

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

   346

                        else:

   347

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

   348

                elif type == "Output":

   349

                    for var in varlist.getvariable():

   350

                        var_type = var.type.getcontent()

   351

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

   352

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

   353

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

   354

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

   355

                        else:

   356

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

   357

        if pou.getbodyType() in ["FBD","LD","SFC"]:

   358

            for instance in pou.getinstances():

   359

                if isinstance(instance, PLCOpenClasses.get("commonObjects_comment", None)):

   360

                    block_infos["comment"] = instance.getcontentText()

   361

        self.CustomBlockTypes.append(block_infos)

   362

    setattr(cls, "AddCustomBlockType", AddCustomBlockType)

   363

   364

    def RefreshElementUsingTree(self):

   365

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

   366

        self.ElementUsingTree = {}

   367

        pous = self.getpous()

   368

        datatypes = self.getdataTypes()

   369

        # Reference all the user-defined elementu names and initialize the tree of 

   370

        # user-defined elemnt cross-use

   371

        elementnames = [datatype.getname() for datatype in datatypes] + \

   372

                       [pou.getname() for pou in pous]

   373

        for name in elementnames:

   374

            self.ElementUsingTree[name] = []

   375

        # Analyze each datatype

   376

        for datatype in datatypes:

   377

            name = datatype.getname()

   378

            basetype_content = datatype.baseType.getcontent()

   379

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

   380

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

   381

                if typename in elementnames and name not in self.ElementUsingTree[typename]:

   382

                    self.ElementUsingTree[typename].append(name)

   383

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

   384

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

   385

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

   386

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

   387

                    if typename in elementnames and name not in self.ElementUsingTree[typename]:

   388

                        self.ElementUsingTree[typename].append(name)

   389

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

   390

                for element in basetype_content["value"].getvariable():

   391

                    type_content = element.type.getcontent()

   392

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

   393

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

   394

                        if typename in elementnames and name not in self.ElementUsingTree[typename]:

   395

                            self.ElementUsingTree[typename].append(name)

   396

        # Analyze each pou

   397

        for pou in pous:

   398

            name = pou.getname()

   399

            if pou.interface:

   400

                # Extract variables from every varLists

   401

                for type, varlist in pou.getvars():

   402

                    for var in varlist.getvariable():

   403

                        vartype_content = var.gettype().getcontent()

   404

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

   405

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

   406

                            if typename in elementnames and name not in self.ElementUsingTree[typename]:

   407

                                self.ElementUsingTree[typename].append(name)

   408

    setattr(cls, "RefreshElementUsingTree", RefreshElementUsingTree)

   409

   410

    def GetParentType(self, type):

   411

        if type in self.CustomTypeHierarchy:

   412

            return self.CustomTypeHierarchy[type]

   413

        elif type in TypeHierarchy:

   414

            return TypeHierarchy[type]

   415

        return None

   416

    setattr(cls, "GetParentType", GetParentType)

   417

   418

    def GetBaseType(self, type):

   419

        parent_type = self.GetParentType(type)

   420

        if parent_type is not None:

   421

            if parent_type.startswith("ANY"):

   422

                return type

   423

            else:

   424

                return self.GetBaseType(parent_type)

   425

        return None

   426

    setattr(cls, "GetBaseType", GetBaseType)

   427

   428

    def GetSubrangeBaseTypes(self, exclude):

   429

        derived = []

   430

        for type in self.CustomTypeHierarchy.keys():

   431

            for base_type in DataTypeRange.keys():

   432

                if self.IsOfType(type, base_type) and not self.IsOfType(type, exclude):

   433

                    derived.append(type)

   434

                    break

   435

        return DataTypeRange.keys() + derived

   436

    setattr(cls, "GetSubrangeBaseTypes", GetSubrangeBaseTypes)

   437

   438

    """

   439

    returns true if the given data type is the same that "reference" meta-type or one of its types.

   440

    """

   441

    def IsOfType(self, type, reference):

   442

        if reference is None:

   443

            return True

   444

        elif type == reference:

   445

            return True

   446

        else:

   447

            parent_type = self.GetParentType(type)

   448

            if parent_type is not None:

   449

                return self.IsOfType(parent_type, reference)

   450

        return False

   451

    setattr(cls, "IsOfType", IsOfType)

   452

   453

    # Return if pou given by name is used by another pou

   454

    def ElementIsUsed(self, name):

   455

        if name in self.ElementUsingTree:

   456

            return len(self.ElementUsingTree[name]) > 0

   457

        return False

   458

    setattr(cls, "ElementIsUsed", ElementIsUsed)

   459

   460

    def DataTypeIsDerived(self, name):

   461

        return name in self.CustomTypeHierarchy.values()

   462

    setattr(cls, "DataTypeIsDerived", DataTypeIsDerived)

   463

   464

    # Return if pou given by name is directly or undirectly used by the reference pou

   465

    def ElementIsUsedBy(self, name, reference):

   466

        if name in self.ElementUsingTree:

   467

            list = self.ElementUsingTree[name]

   468

            # Test if pou is directly used by reference

   469

            if reference in list:

   470

                return True

   471

            else:

   472

                # Test if pou is undirectly used by reference, by testing if pous 

   473

                # that directly use pou is directly or undirectly used by reference

   474

                used = False

   475

                for element in list:

   476

                    used |= self.ElementIsUsedBy(element, reference)

   477

                return used

   478

        return False

   479

    setattr(cls, "ElementIsUsedBy", ElementIsUsedBy)

   480

   481

    def GetDataTypeRange(self, type):

   482

        if type in self.CustomDataTypeRange:

   483

            return self.CustomDataTypeRange[type]

   484

        elif type in DataTypeRange:

   485

            return DataTypeRange[type]

   486

        else:

   487

            parent_type = self.GetParentType(type)

   488

            if parent_type is not None:

   489

                return self.GetDataTypeRange(parent_type)

   490

        return None

   491

    setattr(cls, "GetDataTypeRange", GetDataTypeRange)

   492

   493