1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151

use ark_ec::{AffineRepr, CurveGroup, Group};
use ark_ff::UniformRand;
use paste::paste;
use rand::rngs::StdRng;

macro_rules! impl_projective {
    ($name: ident, $GroupProjective: ty, $CamlG: ty, $CamlScalarField: ty, $BaseField: ty, $CamlBaseField: ty, $Projective: ty) => {

        paste! {
            #[ocaml_gen::func]
            #[ocaml::func]
            pub fn [<caml_ $name:snake _one>]() -> $GroupProjective {
                $Projective::generator().into()
            }

            #[ocaml_gen::func]
            #[ocaml::func]
            pub fn [<caml_ $name:snake _add>](
                x: ocaml::Pointer<$GroupProjective>,
                y: ocaml::Pointer<$GroupProjective>,
            ) -> $GroupProjective {
                x.as_ref() + y.as_ref()
            }

            #[ocaml_gen::func]
            #[ocaml::func]
            pub fn [<caml_ $name:snake _sub>](
                x: ocaml::Pointer<$GroupProjective>,
                y: ocaml::Pointer<$GroupProjective>,
            ) -> $GroupProjective {
                x.as_ref() - y.as_ref()
            }

            #[ocaml_gen::func]
            #[ocaml::func]
            pub fn [<caml_ $name:snake _negate>](
                x: ocaml::Pointer<$GroupProjective>,
            ) -> $GroupProjective {
                -(*x.as_ref())
            }

            #[ocaml_gen::func]
            #[ocaml::func]
            pub fn [<caml_ $name:snake _double>](
                x: ocaml::Pointer<$GroupProjective>,
            ) -> $GroupProjective {
                x.as_ref().double().into()
            }

            #[ocaml_gen::func]
            #[ocaml::func]
            pub fn [<caml_ $name:snake _scale>](
                x: ocaml::Pointer<$GroupProjective>,
                y: $CamlScalarField,
            ) -> $GroupProjective {
                let y: ark_ff::BigInteger256 = y.0.into();
                x.as_ref().mul_bigint(&y).into()
            }

            #[ocaml_gen::func]
            #[ocaml::func]
            pub fn [<caml_ $name:snake _random>]() -> $GroupProjective {
                let rng = &mut rand::rngs::OsRng;
                let proj: $Projective = UniformRand::rand(rng);
                proj.into()
            }

            #[ocaml_gen::func]
            #[ocaml::func]
            pub fn [<caml_ $name:snake _rng>](i: ocaml::Int) -> $GroupProjective {
                // We only care about entropy here, so we force a conversion i32
                // -> u32.
                let i: u64 = (i as u32).into();
                let mut rng: StdRng = rand::SeedableRng::seed_from_u64(i);
                let proj: $Projective = UniformRand::rand(&mut rng);
                proj.into()
            }

            #[ocaml_gen::func]
            #[ocaml::func]
            pub extern "C" fn [<caml_ $name:snake _endo_base>]() -> $CamlBaseField {
                let (endo_q, _endo_r) = poly_commitment::ipa::endos::<GAffine>();
                endo_q.into()
            }

            #[ocaml_gen::func]
            #[ocaml::func]
            pub extern "C" fn [<caml_ $name:snake _endo_scalar>]() -> $CamlScalarField {
                let (_endo_q, endo_r) = poly_commitment::ipa::endos::<GAffine>();
                endo_r.into()
            }

            #[ocaml_gen::func]
            #[ocaml::func]
            pub fn [<caml_ $name:snake _to_affine>](x: ocaml::Pointer<$GroupProjective>) -> $CamlG {
                x.as_ref().into_affine().into()
            }

            #[ocaml_gen::func]
            #[ocaml::func]
            pub fn [<caml_ $name:snake _of_affine>](x: $CamlG) -> $GroupProjective {
                Into::<GAffine>::into(x).into_group().into()
            }

            #[ocaml_gen::func]
            #[ocaml::func]
            pub fn [<caml_ $name:snake _of_affine_coordinates>](x: $CamlBaseField, y: $CamlBaseField) -> $GroupProjective {
                let res = $Projective::new_unchecked(x.into(), y.into(), <$BaseField as ark_ff::One>::one());
                res.into()
            }

            #[ocaml_gen::func]
            #[ocaml::func]
            pub fn [<caml_ $name:snake _affine_deep_copy>](x: $CamlG) -> $CamlG {
                x
            }
        }
    }
}

pub mod pallas {
    use super::*;
    use crate::arkworks::{CamlFp, CamlFq, CamlGPallas, CamlGroupProjectivePallas};
    use mina_curves::pasta::{curves::pallas::ProjectivePallas, Fp, Pallas as GAffine};

    impl_projective!(
        pallas,
        CamlGroupProjectivePallas,
        CamlGPallas,
        CamlFq,
        Fp,
        CamlFp,
        ProjectivePallas
    );
}

pub mod vesta {
    use super::*;
    use crate::arkworks::{CamlFp, CamlFq, CamlGVesta, CamlGroupProjectiveVesta};
    use mina_curves::pasta::{curves::vesta::ProjectiveVesta, Fq, Vesta as GAffine};

    impl_projective!(
        vesta,
        CamlGroupProjectiveVesta,
        CamlGVesta,
        CamlFp,
        Fq,
        CamlFq,
        ProjectiveVesta
    );
}