1

#!/usr/bin/env python

     2

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

     3

# This file is part of Beremiz.

     4

# See COPYING file for copyrights details.

     5

     6

from __future__ import absolute_import

     7

from plcopen.XSLTModelQuery import XSLTModelQuery, _StringValue, _BoolValue, _translate_args

     8

     9

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

    10

#                 Helpers object for generating pou var list

    11

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

    12

    13

    14

class _VariableInfos(object):

    15

    __slots__ = ["Name", "Class", "Option", "Location", "InitialValue",

    16

                 "Edit", "Documentation", "Type", "Tree", "Number"]

    17

    18

    def __init__(self, *args):

    19

        for attr, value in zip(self.__slots__, args):

    20

            setattr(self, attr, value if value is not None else "")

    21

    22

    def copy(self):

    23

        return _VariableInfos(*[getattr(self, attr) for attr in self.__slots__])

    24

    25

    26

class VariablesInfosFactory(object):

    27

    """ Helpers object for generating pou var list """

    28

    29

    def __init__(self, variables):

    30

        self.Variables = variables

    31

        self.TreeStack = []

    32

        self.Type = None

    33

        self.Dimensions = None

    34

    35

    def SetType(self, context, *args):

    36

        self.Type = args[0][0]

    37

    38

    def GetType(self):

    39

        if len(self.Dimensions) > 0:

    40

            return ("array", self.Type, self.Dimensions)

    41

        return self.Type

    42

    43

    def GetTree(self):

    44

        return (self.TreeStack.pop(-1), self.Dimensions)

    45

    46

    def AddDimension(self, context, *args):

    47

        self.Dimensions.append(tuple(

    48

            _translate_args([_StringValue] * 2, args)))

    49

    50

    def AddTree(self, context, *args):

    51

        self.TreeStack.append([])

    52

        self.Dimensions = []

    53

    54

    def AddVarToTree(self, context, *args):

    55

        var = (args[0][0], self.Type, self.GetTree())

    56

        self.TreeStack[-1].append(var)

    57

    58

    def AddVariable(self, context, *args):

    59

        self.Variables.append(_VariableInfos(*(

    60

            _translate_args([_StringValue] * 5 + [_BoolValue] + [_StringValue], args) +

    61

            [self.GetType(), self.GetTree()])))

    62

    63

    64

class VariableInfoCollector(XSLTModelQuery):

    65

    def __init__(self, controller):

    66

        XSLTModelQuery.__init__(self,

    67

                                controller,

    68

                                "variables_infos.xslt",

    69

                                [(name, self.FactoryCaller(name))

    70

                                 for name in [

    71

                                     "SetType",

    72

                                     "AddDimension",

    73

                                     "AddTree",

    74

                                     "AddVarToTree",

    75

                                     "AddVariable"]])

    76

    77

    def FactoryCaller(self, funcname):

    78

        def CallFactory(*args):

    79

            return getattr(self.factory, funcname)(*args)

    80

        return CallFactory

    81

    82

    def Collect(self, root, debug, variables, tree):

    83

        self.factory = VariablesInfosFactory(variables)

    84

        self._process_xslt(root, debug, tree=str(tree))

    85

        res = self.factory

    86

        self.factory = None

    87

        return res