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