ark_ff/fields/models/
fp6_3over2.rs

1use super::cubic_extension::{CubicExtConfig, CubicExtField};
2use crate::fields::{
3    AddAssign, CyclotomicMultSubgroup, Fp2, Fp2Config, MulAssign, SqrtPrecomputation, SubAssign,
4};
5use core::marker::PhantomData;
6
7pub trait Fp6Config: 'static + Send + Sync + Copy {
8    type Fp2Config: Fp2Config;
9
10    const NONRESIDUE: Fp2<Self::Fp2Config>;
11
12    /// Determines the algorithm for computing square roots.
13    const SQRT_PRECOMP: Option<SqrtPrecomputation<Fp6<Self>>> = None;
14
15    /// Coefficients for the Frobenius automorphism.
16    const FROBENIUS_COEFF_FP6_C1: &'static [Fp2<Self::Fp2Config>];
17    const FROBENIUS_COEFF_FP6_C2: &'static [Fp2<Self::Fp2Config>];
18
19    #[inline(always)]
20    fn mul_fp2_by_nonresidue_in_place(fe: &mut Fp2<Self::Fp2Config>) -> &mut Fp2<Self::Fp2Config> {
21        *fe *= &Self::NONRESIDUE;
22        fe
23    }
24    #[inline(always)]
25    fn mul_fp2_by_nonresidue(mut fe: Fp2<Self::Fp2Config>) -> Fp2<Self::Fp2Config> {
26        Self::mul_fp2_by_nonresidue_in_place(&mut fe);
27        fe
28    }
29}
30
31pub struct Fp6ConfigWrapper<P: Fp6Config>(PhantomData<P>);
32
33impl<P: Fp6Config> CubicExtConfig for Fp6ConfigWrapper<P> {
34    type BasePrimeField = <P::Fp2Config as Fp2Config>::Fp;
35    type BaseField = Fp2<P::Fp2Config>;
36    type FrobCoeff = Fp2<P::Fp2Config>;
37
38    const SQRT_PRECOMP: Option<SqrtPrecomputation<CubicExtField<Self>>> = P::SQRT_PRECOMP;
39
40    const DEGREE_OVER_BASE_PRIME_FIELD: usize = 6;
41
42    const NONRESIDUE: Self::BaseField = P::NONRESIDUE;
43
44    const FROBENIUS_COEFF_C1: &'static [Self::FrobCoeff] = P::FROBENIUS_COEFF_FP6_C1;
45    const FROBENIUS_COEFF_C2: &'static [Self::FrobCoeff] = P::FROBENIUS_COEFF_FP6_C2;
46
47    #[inline(always)]
48    fn mul_base_field_by_nonresidue_in_place(fe: &mut Self::BaseField) -> &mut Self::BaseField {
49        P::mul_fp2_by_nonresidue_in_place(fe)
50    }
51
52    fn mul_base_field_by_frob_coeff(
53        c1: &mut Self::BaseField,
54        c2: &mut Self::BaseField,
55        power: usize,
56    ) {
57        *c1 *= &Self::FROBENIUS_COEFF_C1[power % Self::DEGREE_OVER_BASE_PRIME_FIELD];
58        *c2 *= &Self::FROBENIUS_COEFF_C2[power % Self::DEGREE_OVER_BASE_PRIME_FIELD];
59    }
60}
61
62pub type Fp6<P> = CubicExtField<Fp6ConfigWrapper<P>>;
63
64impl<P: Fp6Config> Fp6<P> {
65    pub fn mul_assign_by_fp2(&mut self, other: Fp2<P::Fp2Config>) {
66        self.c0 *= &other;
67        self.c1 *= &other;
68        self.c2 *= &other;
69    }
70
71    pub fn mul_by_fp(&mut self, element: &<P::Fp2Config as Fp2Config>::Fp) {
72        self.c0.mul_assign_by_fp(element);
73        self.c1.mul_assign_by_fp(element);
74        self.c2.mul_assign_by_fp(element);
75    }
76
77    pub fn mul_by_fp2(&mut self, element: &Fp2<P::Fp2Config>) {
78        self.c0.mul_assign(element);
79        self.c1.mul_assign(element);
80        self.c2.mul_assign(element);
81    }
82
83    pub fn mul_by_1(&mut self, c1: &Fp2<P::Fp2Config>) {
84        let mut b_b = self.c1;
85        b_b.mul_assign(c1);
86
87        let mut t1 = *c1;
88        {
89            let mut tmp = self.c1;
90            tmp.add_assign(&self.c2);
91
92            t1.mul_assign(&tmp);
93            t1.sub_assign(&b_b);
94            P::mul_fp2_by_nonresidue_in_place(&mut t1);
95        }
96
97        let mut t2 = *c1;
98        {
99            let mut tmp = self.c0;
100            tmp.add_assign(&self.c1);
101
102            t2.mul_assign(&tmp);
103            t2.sub_assign(&b_b);
104        }
105
106        self.c0 = t1;
107        self.c1 = t2;
108        self.c2 = b_b;
109    }
110
111    pub fn mul_by_01(&mut self, c0: &Fp2<P::Fp2Config>, c1: &Fp2<P::Fp2Config>) {
112        let mut a_a = self.c0;
113        let mut b_b = self.c1;
114        a_a.mul_assign(c0);
115        b_b.mul_assign(c1);
116
117        let mut t1 = *c1;
118        {
119            let mut tmp = self.c1;
120            tmp.add_assign(&self.c2);
121
122            t1.mul_assign(&tmp);
123            t1.sub_assign(&b_b);
124            P::mul_fp2_by_nonresidue_in_place(&mut t1);
125            t1.add_assign(&a_a);
126        }
127
128        let mut t3 = *c0;
129        {
130            let mut tmp = self.c0;
131            tmp.add_assign(&self.c2);
132
133            t3.mul_assign(&tmp);
134            t3.sub_assign(&a_a);
135            t3.add_assign(&b_b);
136        }
137
138        let mut t2 = *c0;
139        t2.add_assign(c1);
140        {
141            let mut tmp = self.c0;
142            tmp.add_assign(&self.c1);
143
144            t2.mul_assign(&tmp);
145            t2.sub_assign(&a_a);
146            t2.sub_assign(&b_b);
147        }
148
149        self.c0 = t1;
150        self.c1 = t2;
151        self.c2 = t3;
152    }
153}
154
155// We just use the default algorithms; there don't seem to be any faster ones.
156impl<P: Fp6Config> CyclotomicMultSubgroup for Fp6<P> {}
ark_ff/fields/models/fp6_3over2.rs

ark_ff/fields/models/
fp6_3over2.rs
