Feat/write point slice ptr (#585)

Co-authored-by: Arya Tabaie <[email protected]>
Consensys · Dec 21, 2024 · f08f759 · f08f759
1 parent 97d69dc
commit f08f759
Show file tree

Hide file tree

Showing 9 changed files with 70 additions and 9 deletions.
diff --git a/ecc/bls12-377/marshal.go b/ecc/bls12-377/marshal.go
diff --git a/ecc/bls12-381/marshal.go b/ecc/bls12-381/marshal.go
diff --git a/ecc/bls24-315/marshal.go b/ecc/bls24-315/marshal.go
diff --git a/ecc/bls24-317/marshal.go b/ecc/bls24-317/marshal.go
diff --git a/ecc/bn254/marshal.go b/ecc/bn254/marshal.go
diff --git a/ecc/bw6-633/marshal.go b/ecc/bw6-633/marshal.go
diff --git a/ecc/bw6-761/marshal.go b/ecc/bw6-761/marshal.go
diff --git a/ecc/utils.go b/ecc/utils.go
@@ -178,7 +178,8 @@ func getVector(l *Lattice, a, b *big.Int) [2]big.Int {
 	return res
 }
 
-// NextPowerOfTwo returns the next power of 2 of n
+// NextPowerOfTwo(n) = 2^⌈log₂(n)⌉
+// or 2ᵏ where 2ᵏ⁻¹ < n ≤ 2ᵏ
 func NextPowerOfTwo(n uint64) uint64 {
 	c := bits.OnesCount64(n)
 	if c == 0 {

diff --git a/internal/generator/ecc/template/marshal.go.tmpl b/internal/generator/ecc/template/marshal.go.tmpl
@@ -464,7 +464,7 @@ func NewEncoder(w io.Writer, options ...func(*Encoder)) *Encoder {
 
 
 // Encode writes the binary encoding of v to the stream
-// type must be uint64, *fr.Element, *fp.Element, *G1Affine, *G2Affine, []G1Affine or []G2Affine
+// type must be uint64, *fr.Element, *fp.Element, *G1Affine, *G2Affine, []G1Affine, []G2Affine, *[]G1Affine or *[]G2Affine
 func (enc *Encoder) Encode(v interface{}) (err error) {
 	if enc.raw {
 		return enc.encodeRaw(v)
@@ -645,6 +645,8 @@ func (enc *Encoder) encode{{- $.Raw}}(v interface{}) (err error) {
 			}
 		}
 		return
+	case *[]G1Affine:
+		return enc.encode{{- $.Raw}}(*t)
 	case []G1Affine:
 		// write slice length
 		err = binary.Write(enc.w, binary.BigEndian, uint32(len(t)))
@@ -664,6 +666,8 @@ func (enc *Encoder) encode{{- $.Raw}}(v interface{}) (err error) {
 			}
 		}
 		return nil
+	case *[]G2Affine:
+		return enc.encode{{- $.Raw}}(*t)
 	case []G2Affine:
 		// write slice length
 		err = binary.Write(enc.w, binary.BigEndian, uint32(len(t)))