use snarkvm_fields::{Field, PrimeField, Zero, field};
use snarkvm_utilities::{
BigInteger,
BitIteratorBE,
biginteger::{BigInteger256, BigInteger384},
};
use crate::{
AffineCurve,
ProjectiveCurve,
bls12_377::{Fq, Fq2, Fr, g1::Bls12_377G1Parameters},
traits::{ModelParameters, ShortWeierstrassParameters},
};
use std::ops::Neg;
#[derive(Copy, Clone, Debug, PartialEq, Eq, Hash)]
pub struct Bls12_377G2Parameters;
impl ModelParameters for Bls12_377G2Parameters {
type BaseField = Fq2;
type ScalarField = Fr;
}
impl ShortWeierstrassParameters for Bls12_377G2Parameters {
const AFFINE_GENERATOR_COEFFS: (Self::BaseField, Self::BaseField) = (G2_GENERATOR_X, G2_GENERATOR_Y);
const B1: Fr = field!(
Fr,
BigInteger256([12574070832645531618, 10005695704657941814, 1564543351912391449, 657300228442948690])
);
const B2: Fr = field!(
Fr,
BigInteger256([2417046298041509844, 11783911742408086824, 14689097366802547462, 270119112518072728])
);
const COFACTOR: &'static [u64] = &[
0x0000000000000001,
0x452217cc90000000,
0xa0f3622fba094800,
0xd693e8c36676bd09,
0x8c505634fae2e189,
0xfbb36b00e1dcc40c,
0xddd88d99a6f6a829,
0x26ba558ae9562a,
];
const COFACTOR_INV: Fr = field!(
Fr,
BigInteger256([15499857013495546999, 4613531467548868169, 14546778081091178013, 549402535258503313,])
);
const PHI: Fq2 = field!(
Fq2,
field!(Fq, BigInteger384([0, 0, 0, 0, 0, 0])),
field!(
Fq,
BigInteger384([
0x2c766f925a7b8727,
0x03d7f6b0253d58b5,
0x838ec0deec122131,
0xbd5eb3e9f658bb10,
0x6942bd126ed3e52e,
0x01673786dd04ed6a,
])
),
);
const R128: Fr = field!(
Fr,
BigInteger256([13717662654766427599, 14709524173037165000, 15342848074630952979, 736762107895475646])
);
const WEIERSTRASS_A: Fq2 = field!(Fq2, Bls12_377G1Parameters::WEIERSTRASS_A, Bls12_377G1Parameters::WEIERSTRASS_A,);
const WEIERSTRASS_B: Fq2 = field!(
Fq2,
field!(Fq, BigInteger384([0, 0, 0, 0, 0, 0])),
field!(
Fq,
BigInteger384([
9255502405446297221,
10229180150694123945,
9215585410771530959,
13357015519562362907,
5437107869987383107,
16259554076827459,
])
),
);
#[inline(always)]
fn mul_by_a(_: &Self::BaseField) -> Self::BaseField {
Self::BaseField::zero()
}
fn is_in_correct_subgroup_assuming_on_curve(
p: &crate::templates::short_weierstrass_jacobian::Affine<Self>,
) -> bool {
p.mul_bits(BitIteratorBE::new(Self::ScalarField::characteristic())).is_zero()
}
fn glv_endomorphism(
mut p: crate::templates::short_weierstrass_jacobian::Affine<Self>,
) -> crate::templates::short_weierstrass_jacobian::Affine<Self> {
p.x.mul_by_fp(&Self::PHI.c1);
p
}
fn mul_projective(
p: crate::templates::short_weierstrass_jacobian::Projective<Self>,
by: Self::ScalarField,
) -> crate::templates::short_weierstrass_jacobian::Projective<Self> {
type ScalarBigInt = <Fr as PrimeField>::BigInteger;
const GLV_WINDOW_SIZE: usize = 4;
const TABLE_SIZE: i64 = 1 << (GLV_WINDOW_SIZE + 1);
const HALF_TABLE_SIZE: i64 = 1 << (GLV_WINDOW_SIZE);
const MASK_FOR_MOD_TABLE_SIZE: u64 = (TABLE_SIZE as u64) - 1;
const L: usize = 1 << (GLV_WINDOW_SIZE - 1);
let decomposition = by.decompose(&Self::Q1, &Self::Q2, Self::B1, Self::B2, Self::R128, &Self::HALF_R);
let mut t_1 = Vec::with_capacity(L);
let double = crate::templates::short_weierstrass_jacobian::Affine::<Self>::from(p.double());
t_1.push(p);
for i in 1..L {
t_1.push(t_1[i - 1].add_mixed(&double));
}
let t_1 =
crate::templates::short_weierstrass_jacobian::Projective::<Self>::batch_normalization_into_affine(t_1);
let t_2 = t_1.iter().copied().map(Self::glv_endomorphism).collect::<Vec<_>>();
let mod_signed = |d| {
let d_mod_window_size = i64::try_from(d & MASK_FOR_MOD_TABLE_SIZE).unwrap();
if d_mod_window_size >= HALF_TABLE_SIZE { d_mod_window_size - TABLE_SIZE } else { d_mod_window_size }
};
let to_wnaf = |e: Self::ScalarField| -> Vec<i32> {
let mut naf = vec![];
let mut e = e.to_bigint();
while !e.is_zero() {
let next = if e.is_odd() {
let naf_sign = mod_signed(e.as_ref()[0]);
if naf_sign < 0 {
e.add_nocarry(&ScalarBigInt::from(-naf_sign as u64));
} else {
e.sub_noborrow(&ScalarBigInt::from(naf_sign as u64));
}
naf_sign.try_into().unwrap()
} else {
0
};
naf.push(next);
e.div2();
}
naf
};
let wnaf = |k1: Self::ScalarField, k2: Self::ScalarField, s1: bool, s2: bool| -> (Vec<i32>, Vec<i32>) {
let mut wnaf_1 = to_wnaf(k1);
let mut wnaf_2 = to_wnaf(k2);
if s1 {
wnaf_1.iter_mut().for_each(|e| *e = -*e);
}
if !s2 {
wnaf_2.iter_mut().for_each(|e| *e = -*e);
}
(wnaf_1, wnaf_2)
};
let naf_add = |table: &[crate::templates::short_weierstrass_jacobian::Affine<Self>],
naf: i32,
acc: &mut crate::templates::short_weierstrass_jacobian::Projective<Self>| {
if naf != 0 {
let mut p_1 = table[(naf.abs() >> 1) as usize];
if naf < 0 {
p_1 = p_1.neg();
}
acc.add_assign_mixed(&p_1);
}
};
let (naf_1, naf_2) = wnaf(decomposition.0, decomposition.1, decomposition.2, decomposition.3);
let max_len = naf_1.len().max(naf_2.len());
let mut acc = crate::templates::short_weierstrass_jacobian::Projective::<Self>::zero();
for i in (0..max_len).rev() {
if i < naf_1.len() {
naf_add(&t_1, naf_1[i], &mut acc)
}
if i < naf_2.len() {
naf_add(&t_2, naf_2[i], &mut acc)
}
if i != 0 {
acc.double_in_place();
}
}
acc
}
}
pub const G2_GENERATOR_X: Fq2 = field!(Fq2, G2_GENERATOR_X_C0, G2_GENERATOR_X_C1);
pub const G2_GENERATOR_Y: Fq2 = field!(Fq2, G2_GENERATOR_Y_C0, G2_GENERATOR_Y_C1);
pub const G2_GENERATOR_X_C0: Fq = field!(
Fq,
BigInteger384::new([
1394603105513884269,
11069732150289508451,
4261960060090787184,
13457254148541472797,
3177258746859163322,
82258727112085846
])
);
pub const G2_GENERATOR_X_C1: Fq = field!(
Fq,
BigInteger384::new([
12672065269715576738,
3451530808602826578,
9486610028138952262,
5031487885431614078,
9858745210421513581,
63301617551232910
])
);
pub const G2_GENERATOR_Y_C0: Fq = field!(
Fq,
BigInteger384::new([
1855632670224768760,
2989378521406112342,
9748867374972564648,
3204895972998458874,
16520689795595505429,
61918742406142643
])
);
pub const G2_GENERATOR_Y_C1: Fq = field!(
Fq,
BigInteger384::new([
1532128906028652860,
14539073382194201855,
10828918286556702479,
14664598863867299115,
483199896405477997,
73741830940675480
])
);