mirror of
https://github.com/logos-storage/nim-poseidon2.git
synced 2026-02-17 20:23:35 +00:00
72 lines
1.9 KiB
Nim
72 lines
1.9 KiB
Nim
|
|
import
|
|
constantine/math/arithmetic,
|
|
constantine/math/io/io_fields,
|
|
constantine/math/io/io_bigints,
|
|
constantine/named/algebras
|
|
|
|
#-------------------------------------------------------------------------------
|
|
|
|
type B* = BigInt[254]
|
|
type F* = Fr[BN254_Snarks]
|
|
type S* = (F,F,F)
|
|
|
|
#-------------------------------------------------------------------------------
|
|
|
|
func getZero*() : F =
|
|
var z : F
|
|
setZero(z)
|
|
return z
|
|
|
|
#-------------------------------------------------------------------------------
|
|
|
|
const zero* : F = getZero()
|
|
const one* : F = fromHex(F,"0x01") # note: `fromUint()` does not work at compile time
|
|
const two* : F = fromHex(F,"0x02")
|
|
|
|
const twoToThe64* : F = fromHex(F,"0x10000000000000000")
|
|
|
|
#-------------------------------------------------------------------------------
|
|
|
|
func hexToF*(s : string, endian: static Endianness = bigEndian) : F =
|
|
let bigint = B.fromHex(s, endian)
|
|
return F.fromBig(bigint)
|
|
|
|
func arrayFromHex*[N](
|
|
inp: array[N, string],
|
|
endian: static Endianness = bigEndian) : array[N, F] =
|
|
var tmp : array[N, F]
|
|
for i in low(inp)..high(inp):
|
|
tmp[i] = hexToF(inp[i], endian)
|
|
return tmp
|
|
|
|
#-------------------------------------------------------------------------------
|
|
|
|
func `+`*(x, y: F): F = ( var z: F = x ; z += y ; return z )
|
|
func `-`*(x, y: F): F = ( var z: F = x ; z -= y ; return z )
|
|
func `*`*(x, y: F): F = ( var z: F = x ; z *= y ; return z )
|
|
|
|
func `==`*(a, b: F): bool =
|
|
bool(arithmetic.`==`(a, b))
|
|
|
|
func sqr*(x : F): F =
|
|
var y = x
|
|
y.square()
|
|
return y
|
|
|
|
#-------------------------------------------------------------------------------
|
|
|
|
func fastPow*(base: F, expo: B): F =
|
|
var s : F = base
|
|
var a : F = one
|
|
var e : B = expo
|
|
for i in 0..<254:
|
|
if bool(isOdd(e)):
|
|
a *= s
|
|
s.square()
|
|
e.div2()
|
|
return a
|
|
|
|
#-------------------------------------------------------------------------------
|
|
|