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

from plcopen.types_enums import CLASS_TYPES, POU_TYPES, VAR_CLASS_INFOS

     9

    10

    11

def class_extraction(value):

    12

    class_type = CLASS_TYPES.get(value)

    13

    if class_type is not None:

    14

        return class_type

    15

    16

    pou_type = POU_TYPES.get(value)

    17

    if pou_type is not None:

    18

        return pou_type

    19

    20

    var_type = VAR_CLASS_INFOS.get(value)

    21

    if var_type is not None:

    22

        return var_type[1]

    23

    24

    return None

    25

    26

    27

class _VariablesTreeItemInfos(object):

    28

    __slots__ = ["name", "var_class", "type", "edit", "debug", "variables"]

    29

    30

    def __init__(self, *args):

    31

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

    32

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

    33

    34

    def copy(self):

    35

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

    36

    37

    38

class VariablesTreeInfosFactory(object):

    39

    40

    def __init__(self):

    41

        self.Root = None

    42

    43

    def GetRoot(self):

    44

        return self.Root

    45

    46

    def SetRoot(self, context, *args):

    47

        self.Root = _VariablesTreeItemInfos(

    48

            *([''] + _translate_args(

    49

                [class_extraction, _StringValue] + [_BoolValue] * 2,

    50

                args) + [[]]))

    51

    52

    def AddVariable(self, context, *args):

    53

        if self.Root is not None:

    54

            self.Root.variables.append(_VariablesTreeItemInfos(

    55

                *(_translate_args(

    56

                    [_StringValue, class_extraction, _StringValue] +

    57

                    [_BoolValue] * 2, args) + [[]])))

    58

    59

    60

class POUVariablesCollector(XSLTModelQuery):

    61

    def __init__(self, controller):

    62

        XSLTModelQuery.__init__(self,

    63

                                controller,

    64

                                "pou_variables.xslt",

    65

                                [(name, self.FactoryCaller(name))

    66

                                 for name in ["SetRoot", "AddVariable"]])

    67

    68

    def FactoryCaller(self, funcname):

    69

        def CallFactory(*args):

    70

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

    71

        return CallFactory

    72

    73

    def Collect(self, root, debug):

    74

        self.factory = VariablesTreeInfosFactory()

    75

        self._process_xslt(root, debug)

    76

        res = self.factory.GetRoot()

    77

        self.factory = None

    78

        return res