1

#!/usr/bin/env python

     2

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

     4

# See COPYING file for copyrights details.

     5

     6

from __future__ import absolute_import

     8

     9

def class_extraction(value):

    10

    class_type = {

    11

        "configuration": ITEM_CONFIGURATION,

    12

        "resource": ITEM_RESOURCE,

    13

        "action": ITEM_ACTION,

    14

        "transition": ITEM_TRANSITION,

    15

        "program": ITEM_PROGRAM}.get(value)

    16

    if class_type is not None:

    17

        return class_type

    18

    19

    pou_type = POU_TYPES.get(value)

    20

    if pou_type is not None:

    21

        return pou_type

    22

    23

    var_type = VAR_CLASS_INFOS.get(value)

    24

    if var_type is not None:

    25

        return var_type[1]

    26

    27

    return None

    28

    29

    30

class _VariablesTreeItemInfos(object):

    31

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

    32

    33

    def __init__(self, *args):

    34

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

    35

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

    36

    37

    def copy(self):

    38

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

    39

    40

    41

class VariablesTreeInfosFactory(object):

    42

    43

    def __init__(self):

    44

        self.Root = None

    45

    46

    def GetRoot(self):

    47

        return self.Root

    48

    49

    def SetRoot(self, context, *args):

    50

        self.Root = _VariablesTreeItemInfos(

    51

            *([''] + _translate_args(

    52

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

    53

                args) + [[]]))

    54

    55

    def AddVariable(self, context, *args):

    56

        if self.Root is not None:

    57

            self.Root.variables.append(_VariablesTreeItemInfos(

    58

                *(_translate_args(

    59

                    [_StringValue, class_extraction, _StringValue] +

    60

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

    61

    62

    63

    66

    def __init__(self, controller):

    72

    79