nim-groth16/groth16/math/domain.nim
Mark Spanbroek 0f2a84ec33 refactor: replace Fp etc shortcuts by full generic type
reason: compile error deep in constantine, because it will
try to determine the field, which it can't when we abbreviate it

Co-Authored-By: Dmitriy Ryajov <dryajov@gmail.com>
2025-07-02 14:50:27 +02:00

61 lines
1.9 KiB
Nim

#
# power-of-two sized multiplicative FFT domains in the scalar field
#
import constantine/math/arithmetic
import constantine/math/io/io_fields
#import constantine/math/io/io_bigints
import constantine/named/properties_fields
import groth16/bn128
import groth16/misc
#-------------------------------------------------------------------------------
type
Domain* = object
domainSize* : int # `N = 2^n`
logDomainSize* : int # `n = log2(N)`
domainGen* : Fr[BN254_Snarks] # `g`
invDomainGen* : Fr[BN254_Snarks] # `g^-1`
invDomainSize* : Fr[BN254_Snarks] # `1/n`
#-------------------------------------------------------------------------------
# the generator of the multiplicative subgroup with size `2^28`
const gen28 = fromHex( Fr[BN254_Snarks], "0x2a3c09f0a58a7e8500e0a7eb8ef62abc402d111e41112ed49bd61b6e725b19f0" )
func createDomain*(size: int): Domain =
let log2 = ceilingLog2(size)
assert( (1 shl log2) == size , "domain must have a power-of-two size" )
let expo : uint = 1'u shl (28 - log2)
let gen = smallPowFr(gen28, expo)
let halfSize = size div 2
let a = smallPowFr(gen, size )
let b = smallPowFr(gen, halfSize)
assert( bool(a == oneFr) , "domain generator sanity check /A" )
assert( not bool(b == oneFr) , "domain generator sanity check /B" )
return Domain( domainSize: size
, logDomainSize: log2
, domainGen: gen
, invDomainGen: invFr(gen)
, invDomainSize: invFr(intToFr(size))
)
#-------------------------------------------------------------------------------
func enumerateDomain*(D: Domain): seq[Fr[BN254_Snarks]] =
var xs = newSeq[Fr[BN254_Snarks]](D.domainSize)
var g = oneFr
for i in 0..<D.domainSize:
xs[i] = g
g *= D.domainGen
return xs
#-------------------------------------------------------------------------------