1

#!/usr/bin/env python

     2

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

     3

     4

# This file is part of Beremiz, a Integrated Development Environment for

     5

# programming IEC 61131-3 automates supporting plcopen standard and CanFestival.

     6

#

     7

# Copyright (c) 2016 Mario de Sousa (msousa@fe.up.pt)

     8

#

     9

# This program is free software: you can redistribute it and/or modify

    10

# it under the terms of the GNU General Public License as published by

    11

# the Free Software Foundation, either version 2 of the License, or

    12

# (at your option) any later version.

    13

#

    14

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

    15

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

    16

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

    17

# GNU General Public License for more details.

    18

#

    19

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

    20

# along with this program.  If not, see <http://www.gnu.org/licenses/>.

    21

#

    22

# This code is made available on the understanding that it will not be

    23

# used in safety-critical situations without a full and competent review.

    24

    25

    26

from __future__ import absolute_import

    27

import os

    28

from six.moves import xrange

    29

    30

from modbus.mb_utils import *

    31

from ConfigTreeNode import ConfigTreeNode

    32

from PLCControler import LOCATION_CONFNODE, LOCATION_VAR_MEMORY

    33

    34

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

    35

base_folder = os.path.join(base_folder, "..")

    36

ModbusPath = os.path.join(base_folder, "Modbus")

    37

    38

    39

#

    40

#

    41

#

    42

# C L I E N T    R E Q U E S T            #

    43

#

    44

#

    45

#

    46

    47

    48

class _RequestPlug(object):

    49

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

    50

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

    51

      <xsd:element name="ModbusRequest">

    52

        <xsd:complexType>

    53

          <xsd:attribute name="Function" type="xsd:string" use="optional" default="01 - Read Coils"/>

    54

          <xsd:attribute name="SlaveID" use="optional" default="1">

    55

            <xsd:simpleType>

    56

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

    57

                    <xsd:minInclusive value="0"/>

    58

                    <xsd:maxInclusive value="255"/>

    59

                </xsd:restriction>

    60

            </xsd:simpleType>

    61

          </xsd:attribute>

    62

          <xsd:attribute name="Nr_of_Channels" use="optional" default="1">

    63

            <xsd:simpleType>

    64

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

    65

                    <xsd:minInclusive value="1"/>

    66

                    <xsd:maxInclusive value="2000"/>

    67

                </xsd:restriction>

    68

            </xsd:simpleType>

    69

          </xsd:attribute>

    70

          <xsd:attribute name="Start_Address" use="optional" default="0">

    71

            <xsd:simpleType>

    72

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

    73

                    <xsd:minInclusive value="0"/>

    74

                    <xsd:maxInclusive value="65535"/>

    75

                </xsd:restriction>

    76

            </xsd:simpleType>

    77

          </xsd:attribute>

    78

          <xsd:attribute name="Timeout_in_ms" use="optional" default="10">

    79

            <xsd:simpleType>

    80

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

    81

                    <xsd:minInclusive value="1"/>

    82

                    <xsd:maxInclusive value="100000"/>

    83

                </xsd:restriction>

    84

            </xsd:simpleType>

    85

          </xsd:attribute>

    87

        </xsd:complexType>

    88

      </xsd:element>

    89

    </xsd:schema>

    90

    """

    91

    92

    def GetParamsAttributes(self, path=None):

    93

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

    94

        for element in infos:

    95

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

    96

                for child in element["children"]:

    97

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

    98

                        list = modbus_function_dict.keys()

    99

                        list.sort()

   100

                        child["type"] = list

   101

        return infos

   102

   103

    def GetVariableLocationTree(self):

   104

        current_location = self.GetCurrentLocation()

   105

        name = self.BaseParams.getName()

   106

        address = self.GetParamsAttributes()[0]["children"][3]["value"]

   107

        count = self.GetParamsAttributes()[0]["children"][2]["value"]

   108

        function = self.GetParamsAttributes()[0]["children"][0]["value"]

   109

        # 'BOOL' or 'WORD'

   110

        datatype = modbus_function_dict[function][3]

   111

        # 1 or 16

   112

        datasize = modbus_function_dict[function][4]

   113

        # 'Q' for coils and holding registers, 'I' for input discretes and input registers

   114

        # datazone = modbus_function_dict[function][5]

   115

        # 'X' for bits, 'W' for words

   116

        datatacc = modbus_function_dict[function][6]

   117

        # 'Coil', 'Holding Register', 'Input Discrete' or 'Input Register'

   118

        dataname = modbus_function_dict[function][7]

   132

        entries = []

   142

        for offset in range(address, address + count):

   143

            entries.append({

   144

                "name": dataname + " " + str(offset),

   145

                "type": LOCATION_VAR_MEMORY,

   146

                "size": datasize,

   147

                "IEC_type": datatype,

   148

                "var_name": "MB_" + "".join([w[0] for w in dataname.split()]) + "_" + str(offset),

   149

                "location": datatacc + ".".join([str(i) for i in current_location]) + "." + str(offset),

   150

                "description": "description",

   151

                "children": []})

   152

        return {"name": name,

   153

                "type": LOCATION_CONFNODE,

   154

                "location": ".".join([str(i) for i in current_location]) + ".x",

   155

                "children": entries}

   156

   157

    def CTNGenerate_C(self, buildpath, locations):

   158

        """

   159

        Generate C code

   160

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

   161

        @param locations: List of complete variables locations \

   162

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

   163

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

   164

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

   165

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

   166

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

   167

            }, ...]

   168

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

   169

        """

   170

        return [], "", False

   171

   172

   173

#

   174

#

   175

#

   176

# S E R V E R    M E M O R Y    A R E A       #

   177

#

   178

#

   179

#

   180

   181

# dictionary implementing:

   182

# key - string with the description we want in the request plugin GUI

   183

# list - (modbus function number, request type, max count value)

   184

modbus_memtype_dict = {

   185

    "01 - Coils":            ('1', 'rw_bits',  65536, "BOOL", 1, "Q", "X", "Coil"),

   186

    "02 - Input Discretes":  ('2', 'ro_bits',  65536, "BOOL", 1, "I", "X", "Input Discrete"),

   187

    "03 - Holding Registers": ('3', 'rw_words', 65536, "WORD", 16, "Q", "W", "Holding Register"),

   188

    "04 - Input Registers":  ('4', 'ro_words', 65536, "WORD", 16, "I", "W", "Input Register"),

   189

}

   190

   191

   192

class _MemoryAreaPlug(object):

   193

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

   194

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

   195

      <xsd:element name="MemoryArea">

   196

        <xsd:complexType>

   197

          <xsd:attribute name="MemoryAreaType" type="xsd:string" use="optional" default="01 - Coils"/>

   198

          <xsd:attribute name="Nr_of_Channels" use="optional" default="1">

   199

            <xsd:simpleType>

   200

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

   201

                    <xsd:minInclusive value="1"/>

   202

                    <xsd:maxInclusive value="65536"/>

   203

                </xsd:restriction>

   204

            </xsd:simpleType>

   205

          </xsd:attribute>

   206

          <xsd:attribute name="Start_Address" use="optional" default="0">

   207

            <xsd:simpleType>

   208

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

   209

                    <xsd:minInclusive value="0"/>

   210

                    <xsd:maxInclusive value="65535"/>

   211

                </xsd:restriction>

   212

            </xsd:simpleType>

   213

          </xsd:attribute>

   214

        </xsd:complexType>

   215

      </xsd:element>

   216

    </xsd:schema>

   217

    """

   218

   219

    def GetParamsAttributes(self, path=None):

   220

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

   221

        for element in infos:

   222

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

   223

                for child in element["children"]:

   224

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

   225

                        list = modbus_memtype_dict.keys()

   226

                        list.sort()

   227

                        child["type"] = list

   228

        return infos

   229

   230

    def GetVariableLocationTree(self):

   231

        current_location = self.GetCurrentLocation()

   232

        name = self.BaseParams.getName()

   233

        address = self.GetParamsAttributes()[0]["children"][2]["value"]

   234

        count = self.GetParamsAttributes()[0]["children"][1]["value"]

   235

        function = self.GetParamsAttributes()[0]["children"][0]["value"]

   236

        # 'BOOL' or 'WORD'

   237

        datatype = modbus_memtype_dict[function][3]

   238

        # 1 or 16

   239

        datasize = modbus_memtype_dict[function][4]

   240

        # 'Q' for coils and holding registers, 'I' for input discretes and input registers

   241

        # datazone = modbus_memtype_dict[function][5]

   242

        # 'X' for bits, 'W' for words

   243

        datatacc = modbus_memtype_dict[function][6]

   244

        # 'Coil', 'Holding Register', 'Input Discrete' or 'Input Register'

   245

        dataname = modbus_memtype_dict[function][7]

   246

        entries = []

   247

        for offset in range(address, address + count):

   248

            entries.append({

   249

                "name": dataname + " " + str(offset),

   250

                "type": LOCATION_VAR_MEMORY,

   251

                "size": datasize,

   252

                "IEC_type": datatype,

   253

                "var_name": "MB_" + "".join([w[0] for w in dataname.split()]) + "_" + str(offset),

   254

                "location": datatacc + ".".join([str(i) for i in current_location]) + "." + str(offset),

   255

                "description": "description",

   256

                "children": []})

   257

        return {"name": name,

   258

                "type": LOCATION_CONFNODE,

   259

                "location": ".".join([str(i) for i in current_location]) + ".x",

   260

                "children": entries}

   261

   262

    def CTNGenerate_C(self, buildpath, locations):

   263

        """

   264

        Generate C code

   265

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

   266

        @param locations: List of complete variables locations \

   267

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

   268

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

   269

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

   270

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

   271

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

   272

            }, ...]

   273

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

   274

        """

   275

        return [], "", False

   276

   277

   278

#

   279

#

   280

#

   281

# T C P    C L I E N T                 #

   282

#

   283

#

   284

#

   285

   286

class _ModbusTCPclientPlug(object):

   287

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

   288

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

   289

      <xsd:element name="ModbusTCPclient">

   290

        <xsd:complexType>

   291

          <xsd:attribute name="Remote_IP_Address" type="xsd:string" use="optional" default="localhost"/>

   292

          <xsd:attribute name="Remote_Port_Number" type="xsd:string" use="optional" default="502"/>

   293

          <xsd:attribute name="Invocation_Rate_in_ms" use="optional" default="100">

   294

            <xsd:simpleType>

   295

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

   297

                    <xsd:maxInclusive value="2147483647"/>

   298

                </xsd:restriction>

   299

            </xsd:simpleType>

   300

          </xsd:attribute>

   301

        </xsd:complexType>

   302

      </xsd:element>

   303

    </xsd:schema>

   304

    """

   305

    # NOTE: Max value of 2147483647 (i32_max) for Invocation_Rate_in_ms

   306

    # corresponds to aprox 25 days.

   307

    CTNChildrenTypes = [("ModbusRequest", _RequestPlug, "Request")]

   308

    # TODO: Replace with CTNType !!!

   309

    PlugType = "ModbusTCPclient"

   310

   311

    # Return the number of (modbus library) nodes this specific TCP client will need

   312

    #   return type: (tcp nodes, rtu nodes, ascii nodes)

   313

    def GetNodeCount(self):

   314

        return (1, 0, 0)

   315

   316

    def CTNGenerate_C(self, buildpath, locations):

   317

        """

   318

        Generate C code

   319

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

   320

        @param locations: List of complete variables locations \

   321

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

   322

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

   323

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

   324

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

   325

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

   326

            }, ...]

   327

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

   328

        """

   329

        return [], "", False

   330

   331

   332

#

   333

#

   334

#

   335

# T C P    S E R V E R                 #

   336

#

   337

#

   338

#

   339

   340

class _ModbusTCPserverPlug(object):

   341

    # NOTE: the Port number is a 'string' and not an 'integer'!

   342

    # This is because the underlying modbus library accepts strings

   343

    # (e.g.: well known port names!)

   344

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

   345

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

   346

      <xsd:element name="ModbusServerNode">

   347

        <xsd:complexType>

   348

          <xsd:attribute name="Local_IP_Address" type="xsd:string" use="optional"  default="#ANY#"/>

   349

          <xsd:attribute name="Local_Port_Number" type="xsd:string" use="optional" default="502"/>

   350

          <xsd:attribute name="SlaveID" use="optional" default="0">

   351

            <xsd:simpleType>

   352

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

   353

                    <xsd:minInclusive value="0"/>

   354

                    <xsd:maxInclusive value="255"/>

   355

                </xsd:restriction>

   356

            </xsd:simpleType>

   357

          </xsd:attribute>

   358

        </xsd:complexType>

   359

      </xsd:element>

   360

    </xsd:schema>

   361

    """

   362

    CTNChildrenTypes = [("MemoryArea", _MemoryAreaPlug, "Memory Area")]

   363

    # TODO: Replace with CTNType !!!

   364

    PlugType = "ModbusTCPserver"

   365

   366

    # Return the number of (modbus library) nodes this specific TCP server will need

   367

    #   return type: (tcp nodes, rtu nodes, ascii nodes)

   368

    def GetNodeCount(self):

   369

        return (1, 0, 0)

   370

   371

    # Return a list with a single tuple conatining the (location, port number)

   372

    #     location: location of this node in the configuration tree

   373

    #     port number: IP port used by this Modbus/IP server

   374

    def GetIPServerPortNumbers(self):

   375

        port = self.GetParamsAttributes()[0]["children"][1]["value"]

   376

        return [(self.GetCurrentLocation(), port)]

   377

   378

    def CTNGenerate_C(self, buildpath, locations):

   379

        """

   380

        Generate C code

   381

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

   382

        @param locations: List of complete variables locations \

   383

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

   384

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

   385

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

   386

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

   387

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

   388

            }, ...]

   389

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

   390

        """

   391

        return [], "", False

   392

   393

   394

#

   395

#

   396

#

   397

# R T U    C L I E N T                 #

   398

#

   399

#

   400

#

   401

   402

class _ModbusRTUclientPlug(object):

   403

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

   404

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

   405

      <xsd:element name="ModbusRTUclient">

   406

        <xsd:complexType>

   407

          <xsd:attribute name="Serial_Port" type="xsd:string"  use="optional" default="/dev/ttyS0"/>

   408

          <xsd:attribute name="Baud_Rate"   type="xsd:string"  use="optional" default="9600"/>

   409

          <xsd:attribute name="Parity"      type="xsd:string"  use="optional" default="even"/>

   410

          <xsd:attribute name="Stop_Bits"   type="xsd:string"  use="optional" default="1"/>

   411

          <xsd:attribute name="Invocation_Rate_in_ms" use="optional" default="100">

   412

            <xsd:simpleType>

   413

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

   415

                    <xsd:maxInclusive value="2147483647"/>

   416

                </xsd:restriction>

   417

            </xsd:simpleType>

   418

          </xsd:attribute>

   419

        </xsd:complexType>

   420

      </xsd:element>

   421

    </xsd:schema>

   422

    """

   423

    # NOTE: Max value of 2147483647 (i32_max) for Invocation_Rate_in_ms

   424

    # corresponds to aprox 25 days.

   425

    CTNChildrenTypes = [("ModbusRequest", _RequestPlug, "Request")]

   426

    # TODO: Replace with CTNType !!!

   427

    PlugType = "ModbusRTUclient"

   428

   429

    def GetParamsAttributes(self, path=None):

   430

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

   431

        for element in infos:

   432

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

   433

                for child in element["children"]:

   434

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

   435

                        child["type"] = modbus_serial_baudrate_list

   436

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

   437

                        child["type"] = modbus_serial_stopbits_list

   438

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

   439

                        child["type"] = modbus_serial_parity_dict.keys()

   440

        return infos

   441

   442

    # Return the number of (modbus library) nodes this specific RTU client will need

   443

    #   return type: (tcp nodes, rtu nodes, ascii nodes)

   444

    def GetNodeCount(self):

   445

        return (0, 1, 0)

   446

   447

    def CTNGenerate_C(self, buildpath, locations):

   448

        """

   449

        Generate C code

   450

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

   451

        @param locations: List of complete variables locations \

   452

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

   453

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

   454

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

   455

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

   456

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

   457

            }, ...]

   458

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

   459

        """

   460

        return [], "", False

   461

   462

   463

#

   464

#

   465

#

   466

# R T U    S L A V E                   #

   467

#

   468

#

   469

#

   470

   471

   472

class _ModbusRTUslavePlug(object):

   473

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

   474

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

   475

      <xsd:element name="ModbusRTUslave">

   476

        <xsd:complexType>

   477

          <xsd:attribute name="Serial_Port" type="xsd:string"  use="optional" default="/dev/ttyS0"/>

   478

          <xsd:attribute name="Baud_Rate"   type="xsd:string"  use="optional" default="9600"/>

   479

          <xsd:attribute name="Parity"      type="xsd:string"  use="optional" default="even"/>

   480

          <xsd:attribute name="Stop_Bits"   type="xsd:string"  use="optional" default="1"/>

   481

          <xsd:attribute name="SlaveID" use="optional" default="1">

   482

            <xsd:simpleType>

   483

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

   484

                    <xsd:minInclusive value="1"/>

   485

                    <xsd:maxInclusive value="255"/>

   486

                </xsd:restriction>

   487

            </xsd:simpleType>

   488

          </xsd:attribute>

   489

        </xsd:complexType>

   490

      </xsd:element>

   491

    </xsd:schema>

   492

    """

   493

    CTNChildrenTypes = [("MemoryArea", _MemoryAreaPlug, "Memory Area")]

   494

    # TODO: Replace with CTNType !!!

   495

    PlugType = "ModbusRTUslave"

   496

   497

    def GetParamsAttributes(self, path=None):

   498

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

   499

        for element in infos:

   500

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

   501

                for child in element["children"]:

   502

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

   503

                        child["type"] = modbus_serial_baudrate_list

   504

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

   505

                        child["type"] = modbus_serial_stopbits_list

   506

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

   507

                        child["type"] = modbus_serial_parity_dict.keys()

   508

        return infos

   509

   510

    # Return the number of (modbus library) nodes this specific RTU slave will need

   511

    #   return type: (tcp nodes, rtu nodes, ascii nodes)

   512

    def GetNodeCount(self):

   513

        return (0, 1, 0)