SolidPython/solid/greedy_scad_interface/scad_variable.py at 67e83fe9a2cf5b0b45171307f736a72a7eb2e33c · SolidCode/SolidPython · 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
import math

sin = lambda x: ScadValue(f'sin({x})') if isinstance(x, ScadValue) else math.sin(x)
cos = lambda x: ScadValue(f'cos({x})') if isinstance(x, ScadValue) else math.cos(x)
tan = lambda x: ScadValue(f'tan({x})') if isinstance(x, ScadValue) else math.tan(x)
asin = lambda x: ScadValue(f'asin({x})') if isinstance(x, ScadValue) else math.asin(x)
acos = lambda x: ScadValue(f'acos({x})') if isinstance(x, ScadValue) else math.acos(x)
atan = lambda x: ScadValue(f'atan({x})') if isinstance(x, ScadValue) else math.atan(x)
sqrt = lambda x: ScadValue(f'sqrt({x})') if isinstance(x, ScadValue) else math.sqrt(x)
not_ = lambda x: ScadValue(f'!{x}')

class ScadValue:
    def __init__(self, value):
        self.value = value

    def __repr__(self):
        return f'{self.value}'

    def __operator_base__(self, op, other):
        return ScadValue(f'({self} {op} {other})')

    def __unary_operator_base__(self, op):
        return ScadValue(f'({op}{self})')

    def __illegal_operator__(self):
       raise Exception("You can't compare a ScadValue with something else, " +\
                       "because we don't know the customized value at " +\
                       "SolidPythons runtime because it might get customized " +\
                       "at OpenSCAD runtime.")

    #basic operators +, -, *, /, %, **
    def __add__(self, other): return self.__operator_base__("+", other)
    def __sub__(self, other): return self.__operator_base__("-", other)
    def __mul__(self, other): return self.__operator_base__("*", other)
    def __mod__(self, other): return self.__operator_base__("%", other)
    def __pow__(self, other): return self.__operator_base__("^", other)
    def __truediv__(self, other): return self.__operator_base__("/", other)

    def __radd__(self, other): return self.__operator_base__("+", other)
    def __rsub__(self, other): return self.__operator_base__("-", other)
    def __rmul__(self, other): return self.__operator_base__("*", other)
    def __rmod__(self, other): return self.__operator_base__("%", other)
    def __rpow__(self, other): return self.__operator_base__("^", other)
    def __rtruediv__(self, other): return self.__operator_base__("/", other)

    #unary operators
    def __neg__(self): return self.__unary_operator_base__("-")

    #other operators
    def __abs__(self): return ScadValue(f'abs({self})')

    #"illegal" operators
    def __eq__(self, other): return self.__illegal_operator__()
    def __ne__(self, other): return self.__illegal_operator__()
    def __le__(self, other): return self.__illegal_operator__()
    def __ge__(self, other): return self.__illegal_operator__()
    def __lt__(self, other): return self.__illegal_operator__()
    def __gt__(self, other): return self.__illegal_operator__()

    #do not allow to evaluate to bool
    def __bool__(self):
        raise Exception("You can't use scad variables as truth statement because " +\
                        "we don't know the value of a customized variable at " +\
                        "SolidPython runtime.")

class ScadVariable(ScadValue):
    registered_variables = {}

    def __init__(self, name, default_value, options_str='', label='', tab=''):
        super().__init__(name)

        if name in self.registered_variables.keys():
            raise ValueError("Multiple instances of ScadVariable with the same name.")

        def_str = self.get_definition(name, default_value, options_str, label, tab)
        self.registered_variables.update({name : def_str})

    def get_definition(self, name, default_value, options_str, label, tab):
        from ..solidpython import py2openscad

        tab = tab and f'/* [{tab}] */\n'
        label = label and f'//{label}\n'
        options_str = options_str and f' //{options_str}'
        default_value = py2openscad(default_value)

        return f'{tab}{label}{name} = {default_value};{options_str}'