1package plonky2_verifier
 2
 3import (
 4	"fmt"
 5	. "gnark-plonky2-verifier/field"
 6)
 7
 8type ArithmeticExtensionGate struct {
 9	numOps uint64
10}
11
12func NewArithmeticExtensionGate(numOps uint64) *ArithmeticExtensionGate {
13	return &ArithmeticExtensionGate{
14		numOps: numOps,
15	}
16}
17
18func (g *ArithmeticExtensionGate) Id() string {
19	return fmt.Sprintf("ArithmeticExtensionGate { num_ops: %d }", g.numOps)
20}
21
22func (g *ArithmeticExtensionGate) wiresIthMultiplicand0(i uint64) Range {
23	return Range{4 * D * i, 4*D*i + D}
24}
25
26func (g *ArithmeticExtensionGate) wiresIthMultiplicand1(i uint64) Range {
27	return Range{4*D*i + D, 4*D*i + 2*D}
28}
29
30func (g *ArithmeticExtensionGate) wiresIthAddend(i uint64) Range {
31	return Range{4*D*i + 2*D, 4*D*i + 3*D}
32}
33
34func (g *ArithmeticExtensionGate) wiresIthOutput(i uint64) Range {
35	return Range{4*D*i + 3*D, 4*D*i + 4*D}
36}
37
38func (g *ArithmeticExtensionGate) EvalUnfiltered(p *PlonkChip, vars EvaluationVars) []QuadraticExtension {
39	const0 := vars.localConstants[0]
40	const1 := vars.localConstants[1]
41
42	constraints := []QuadraticExtension{}
43	for i := uint64(0); i < g.numOps; i++ {
44		multiplicand0 := vars.GetLocalExtAlgebra(g.wiresIthMultiplicand0(i))
45		multiplicand1 := vars.GetLocalExtAlgebra(g.wiresIthMultiplicand1(i))
46		addend := vars.GetLocalExtAlgebra(g.wiresIthAddend(i))
47		output := vars.GetLocalExtAlgebra(g.wiresIthOutput(i))
48
49		mul := p.qeAPI.MulExtensionAlgebra(multiplicand0, multiplicand1)
50		scaled_mul := p.qeAPI.ScalarMulExtensionAlgebra(const0, mul)
51		computed_output := p.qeAPI.ScalarMulExtensionAlgebra(const1, addend)
52		computed_output = p.qeAPI.AddExtensionAlgebra(computed_output, scaled_mul)
53
54		diff := p.qeAPI.SubExtensionAlgebra(output, computed_output)
55		for j := 0; j < D; j++ {
56			constraints = append(constraints, diff[j])
57		}
58	}
59
60	return constraints
61}