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

class _VariableInfos(object):

    14

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

    15

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

    16

    17

    def __init__(self, *args):

    18

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

    19

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

    20

    21

    def copy(self):

    22

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

    23

    24

    25

class VariablesInfosFactory(object):

    26

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

    27

    28

    def __init__(self, variables):

    29

        self.Variables = variables

    30

        self.TreeStack = []

    31

        self.Type = None

    32

        self.Dimensions = None

    33

    34

    def SetType(self, context, *args):

    35

        self.Type = args[0][0]

    36

    37

    def GetType(self):

    38

        if len(self.Dimensions) > 0:

    39

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

    40

        return self.Type

    41

    42

    def GetTree(self):

    43

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

    44

    45

    def AddDimension(self, context, *args):

    46

        self.Dimensions.append(tuple(

    47

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

    48

    49

    def AddTree(self, context, *args):

    50

        self.TreeStack.append([])

    51

        self.Dimensions = []

    52

    53

    def AddVarToTree(self, context, *args):

    54

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

    55

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

    56

    57

    def AddVariable(self, context, *args):

    58

        self.Variables.append(_VariableInfos(*(

    59

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

    60

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

    61

    62

    63

class VariableInfoCollector(XSLTModelQuery):

    64

    def __init__(self, controller):

    65

        XSLTModelQuery.__init__(self,

    66

                                controller,

    67

                                "variables_infos.xslt",

    68

                                [(name, self.FactoryCaller(name)) 

    69

                                    for name in [

    70

                                        "SetType",

    71

                                        "AddDimension",

    72

                                        "AddTree",

    73

                                        "AddVarToTree",

    74

                                        "AddVariable"]])

    75

    76

    def FactoryCaller(self, funcname):

    77

        def CallFactory(*args):

    78

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

    79

        return CallFactory

    80

    81

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

    82

        self.factory = VariablesInfosFactory(variables)

    83

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

    84

        res = self.factory

    85

        self.factory = None

    86

        return res

    87