summaryrefslogtreecommitdiff
path: root/gerber/am_eval.py
blob: 3a7e1ed4a589f032eafd3d1becb04604723756ac (plain)
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
#! /usr/bin/env python
# -*- coding: utf-8 -*-

# copyright 2014 Hamilton Kibbe <ham@hamiltonkib.be>
# copyright 2014 Paulo Henrique Silva <ph.silva@gmail.com>
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at

#     http://www.apache.org/licenses/LICENSE-2.0

# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
""" This module provides RS-274-X AM macro evaluation.
"""


class OpCode:
    PUSH = 1
    LOAD = 2
    STORE = 3
    ADD = 4
    SUB = 5
    MUL = 6
    DIV = 7
    PRIM = 8

    @staticmethod
    def str(opcode):
        if opcode == OpCode.PUSH:
            return "OPCODE_PUSH"
        elif opcode == OpCode.LOAD:
            return "OPCODE_LOAD"
        elif opcode == OpCode.STORE:
            return "OPCODE_STORE"
        elif opcode == OpCode.ADD:
            return "OPCODE_ADD"
        elif opcode == OpCode.SUB:
            return "OPCODE_SUB"
        elif opcode == OpCode.MUL:
            return "OPCODE_MUL"
        elif opcode == OpCode.DIV:
            return "OPCODE_DIV"
        elif opcode == OpCode.PRIM:
            return "OPCODE_PRIM"
        else:
            return "UNKNOWN"


def eval_macro(instructions, parameters={}):

    if not isinstance(parameters, type({})):
        p = {}
        for i, val in enumerate(parameters):
            p[i + 1] = val

        parameters = p

    stack = []

    def pop():
        return stack.pop()

    def push(op):
        stack.append(op)

    def top():
        return stack[-1]

    def empty():
        return len(stack) == 0

    for opcode, argument in instructions:
        if opcode == OpCode.PUSH:
            push(argument)

        elif opcode == OpCode.LOAD:
            push(parameters.get(argument, 0))

        elif opcode == OpCode.STORE:
            parameters[argument] = pop()

        elif opcode == OpCode.ADD:
            op1 = pop()
            op2 = pop()
            push(op2 + op1)

        elif opcode == OpCode.SUB:
            op1 = pop()
            op2 = pop()
            push(op2 - op2)

        elif opcode == OpCode.MUL:
            op1 = pop()
            op2 = pop()
            push(op2 * op1)

        elif opcode == OpCode.DIV:
            op1 = pop()
            op2 = pop()
            push(op2 / op1)

        elif opcode == OpCode.PRIM:
            yield "%d,%s" % (argument, ",".join([str(x) for x in stack]))
            stack = []