1package plonky2_verifier
 2
 3import (
 4	"fmt"
 5	. "gnark-plonky2-verifier/field"
 6	"regexp"
 7	"strconv"
 8)
 9
10var aritheticGateRegex = regexp.MustCompile("ArithmeticGate { num_ops: (?P<numOps>[0-9]+) }")
11
12func deserializeArithmeticGate(parameters map[string]string) gate {
13	// Has the format "ArithmeticGate { num_ops: 10 }"
14
15	numOps, hasNumOps := parameters["numOps"]
16	if !hasNumOps {
17		panic("no num_ops field in ArithmeticGate")
18	}
19
20	numOpsInt, err := strconv.Atoi(numOps)
21	if err != nil {
22		panic("Invalid num_ops field in ArithmeticGate")
23	}
24
25	return NewArithmeticGate(uint64(numOpsInt))
26}
27
28type ArithmeticGate struct {
29	numOps uint64
30}
31
32func NewArithmeticGate(numOps uint64) *ArithmeticGate {
33	return &ArithmeticGate{
34		numOps: numOps,
35	}
36}
37
38func (g *ArithmeticGate) Id() string {
39	return fmt.Sprintf("ArithmeticGate { num_ops: %d }", g.numOps)
40}
41
42func (g *ArithmeticGate) WireIthMultiplicand0(i uint64) uint64 {
43	return 4 * i
44}
45
46func (g *ArithmeticGate) WireIthMultiplicand1(i uint64) uint64 {
47	return 4*i + 1
48}
49
50func (g *ArithmeticGate) WireIthAddend(i uint64) uint64 {
51	return 4*i + 2
52}
53
54func (g *ArithmeticGate) WireIthOutput(i uint64) uint64 {
55	return 4*i + 3
56}
57
58func (g *ArithmeticGate) EvalUnfiltered(p *PlonkChip, vars EvaluationVars) []QuadraticExtension {
59	const0 := vars.localConstants[0]
60	const1 := vars.localConstants[1]
61
62	constraints := []QuadraticExtension{}
63	for i := uint64(0); i < g.numOps; i++ {
64		multiplicand0 := vars.localWires[g.WireIthMultiplicand0(i)]
65		multiplicand1 := vars.localWires[g.WireIthMultiplicand1(i)]
66		addend := vars.localWires[g.WireIthAddend(i)]
67		output := vars.localWires[g.WireIthOutput(i)]
68
69		computedOutput := p.qeAPI.AddExtension(
70			p.qeAPI.MulExtension(p.qeAPI.MulExtension(multiplicand0, multiplicand1), const0),
71			p.qeAPI.MulExtension(addend, const1),
72		)
73
74		constraints = append(constraints, p.qeAPI.SubExtension(output, computedOutput))
75	}
76
77	return constraints
78}