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
|
from typing import List, Tuple
from .formula import Formula
from .mod import Mod
from .point import Point
class Context(object):
intermediates: List[Tuple[str, Mod]]
actions: List[Tuple[Formula, Tuple[Point, ...]]]
def __init__(self):
self.intermediates = []
self.actions = []
def execute(self, formula: Formula, *points: Point, **params: Mod) -> Tuple[Point, ...]:
if len(points) != formula.num_inputs:
raise ValueError
self.actions.append((formula, tuple(points)))
coords = {}
for i, point in enumerate(points):
if point.coordinate_model != formula.coordinate_model:
raise ValueError
for coord, value in point.coords.items():
coords[coord + str(i + 1)] = value
locals = {**coords, **params}
for op in formula.code:
op_result = op(**locals)
self.intermediates.append((op.result, op_result))
locals[op.result] = op_result
result = []
for i in range(formula.num_outputs):
ind = str(i + formula.output_index)
resulting = {variable: locals[variable + ind]
for variable in formula.coordinate_model.variables
if variable + ind in locals}
result.append(Point(formula.coordinate_model, **resulting))
return tuple(result)
|
