DomainSpecificLanguage/operators.py at main · Maleidner/DomainSpecificLanguage · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
from Datatypes import *

def _plus(arg1, arg2):
    var = arg1.value + arg2.value
    if arg1.timestamp == arg2.timestamp:
       return NumType(var, timestamp = arg1.timestamp)
    return NumType(var)

def _minus(arg1, arg2):
    var = arg1.value - arg2.value
    if arg1.timestamp == arg2.timestamp:
        return NumType(var, timestamp = arg1.timestamp)
    return NumType(var)

def _uminus(arg1):
    return NumType(-arg1.value)


def _divide(arg1, arg2):
    var = arg1.value / arg2.value
    if arg1.timestamp == arg2.timestamp:
         return NumType(var, timestamp = arg1.timestamp)
    return NumType(var)

def _times(arg1, arg2):
    var = arg1.value * arg2.value
    if arg1.timestamp == arg2.timestamp:
         return NumType(var, timestamp = arg1.timestamp)
    return NumType(var)

def _power(arg1, arg2):
    var = arg1.value ** arg2.value
    if arg1.timestamp == arg2.timestamp:
         return NumType(var, timestamp = arg1.timestamp)
    return NumType(var)

def _lessthan(arg1, arg2):
    correct = False
    if arg1.value < arg2.value:
        correct = True
    return BoolType(correct)

def _greaterthan(arg1, arg2):
    correct = False
    if arg1.value > arg2.value:
        correct = True
    return BoolType(correct)

def _lessthanorequal(arg1, arg2):
    correct = False
    if arg1.value <= arg2.value:
        correct = True
    return BoolType(correct)

def _greaterthan(arg1, arg2):
    correct = False
    if arg1.value >= arg2.value:
        correct = True
    return BoolType(correct)

def _equal(arg1, arg2):
    correct = False
    if arg1.value == arg2.value:
        correct = True
    return BoolType(correct)

def _seqto(arg1, arg2):
    l = []
    for i in range(int(arg1.value), int(arg2.value)+1):
        l.append(i)
    return l

def _maximum(arg1):
    if isinstance(arg1, ListType):
        tmp = []
        for i in arg1.value:
            tmp.append(i.value)
        result = max(tmp)
    else:
        result = arg1
    return NumType(result)


def _minimum(arg1):
    if isinstance(arg1, ListType):
        tmp = []
        for i in arg1.value:
            tmp.append(i.value)
        result = min(tmp)
    else:
        result = arg1
    return NumType(result)

def _first(arg1):
    if isinstance(arg1, ListType):
        result = arg1.value[0].value
    else:
        result = arg1.value
    return NumType(result)

def _last(arg1):
    if isinstance(arg1, ListType):
        length = len(arg1.value)
        result = arg1.value[length-1].value
    else:
        result = arg1.value
    return NumType(result)

def _sum(arg1):
    if isinstance(arg1, ListType):
        result = 0
        for i in arg1.value:
            result = result + i.value
    else:
        result = arg1.value
    return NumType(result)

def _count(arg1):
    if isinstance(arg1, ListType):
        result = len(arg1.value)
    else:
        result = 1
    return NumType(result)

def _timeof(arg1):
    return arg1.timestamp

def _iswithin(arg1, arg2, arg3):
    if arg1 > arg2 and arg1 < arg3:
        return BoolType(True)
    else:
        return BoolType(False)