NNEquivalence/variable.py at master · phK3/NNEquivalence · 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

class Variable:

    # types are Real and Int just as in SMTLib
    def __init__(self, layer, row, netPrefix, prefix_name='x', type='Real'):
        self.layer = layer
        self.row = row
        self.prefix_name = prefix_name

        self.name = ''
        if not netPrefix == '':
            self.name += netPrefix + '_'

        self.name += prefix_name + '_' + str(layer) + '_' + str(row)
        self.type = type
        # lower and upper bounds
        self.hasLo = False
        self.hasHi = False
        # set to +/- 99999 as default
        # TODO: change them to useful values later
        self.lo = -99999
        self.hi = 99999

    def addPrefix(self, prefix):
        self.prefix_name = prefix + '_' + self.prefix_name
        self.name = prefix + '_' + self.name

    def setLo(self, lowerBound):
        self.lo = lowerBound
        self.hasLo = True
        return self.lo

    def setHi(self, upperBound):
        self.hi = upperBound
        self.hasHi = True
        return self.hi

    def __str__(self):
        return self.name

    def __repr__(self):
        return self.name