circom/c/buildasm/fr.c.ejs
c/buildasm/fr.c.ejsRaw
1#include "<%=name.toLowerCase()+".h"%>"
2#include <stdio.h>
3#include <stdlib.h>
4#include <gmp.h>
5#include <assert.h>
6
7mpz_t q;
8mpz_t zero;
9mpz_t one;
10mpz_t mask;
11size_t nBits;
12
13
14void <%=name%>_toMpz(mpz_t r, P<%=name%>Element pE) {
15 <%=name%>_toNormal(pE);
16 if (!(pE->type & <%=name%>_LONG)) {
17 mpz_set_si(r, pE->shortVal);
18 if (pE->shortVal<0) {
19 mpz_add(r, r, q);
20 }
21 } else {
22 mpz_import(r, <%=name%>_N64, -1, 8, -1, 0, (const void *)pE->longVal);
23 }
24}
25
26void <%=name%>_fromMpz(P<%=name%>Element pE, mpz_t v) {
27 if (mpz_fits_sint_p(v)) {
28 pE->type = <%=name%>_SHORT;
29 pE->shortVal = mpz_get_si(v);
30 } else {
31 pE->type = <%=name%>_LONG;
32 for (int i=0; i<<%=name%>_N64; i++) pE->longVal[i] = 0;
33 mpz_export((void *)(pE->longVal), NULL, -1, 8, -1, 0, v);
34 }
35}
36
37
38void <%=name%>_init() {
39 mpz_init(q);
40 mpz_import(q, <%=name%>_N64, -1, 8, -1, 0, (const void *)Fr_q.longVal);
41 mpz_init_set_ui(zero, 0);
42 mpz_init_set_ui(one, 1);
43 nBits = mpz_sizeinbase (q, 2);
44 mpz_init(mask);
45 mpz_mul_2exp(mask, one, nBits-1);
46 mpz_sub(mask, mask, one);
47
48}
49
50void <%=name%>_str2element(P<%=name%>Element pE, char const *s) {
51 mpz_t mr;
52 mpz_init_set_str(mr, s, 10);
53 <%=name%>_fromMpz(pE, mr);
54}
55
56char *<%=name%>_element2str(P<%=name%>Element pE) {
57 mpz_t r;
58 if (!(pE->type & <%=name%>_LONG)) {
59 if (pE->shortVal>=0) {
60 char *r = new char[32];
61 sprintf(r, "%d", pE->shortVal);
62 return r;
63 } else {
64 mpz_init_set_si(r, pE->shortVal);
65 mpz_add(r, r, q);
66 mpz_clear(q);
67 }
68 } else {
69 <%=name%>_toNormal(pE);
70 mpz_init(r);
71 mpz_import(r, <%=name%>_N64, -1, 8, -1, 0, (const void *)pE->longVal);
72 }
73 char *res = mpz_get_str (0, 10, r);
74 mpz_clear(r);
75 return res;
76}
77
78void <%=name%>_idiv(P<%=name%>Element r, P<%=name%>Element a, P<%=name%>Element b) {
79 mpz_t ma;
80 mpz_t mb;
81 mpz_t mr;
82 mpz_init(ma);
83 mpz_init(mb);
84 mpz_init(mr);
85
86 <%=name%>_toMpz(ma, a);
87 // char *s1 = mpz_get_str (0, 10, ma);
88 // printf("s1 %s\n", s1);
89 <%=name%>_toMpz(mb, b);
90 // char *s2 = mpz_get_str (0, 10, mb);
91 // printf("s2 %s\n", s2);
92 mpz_fdiv_q(mr, ma, mb);
93 // char *sr = mpz_get_str (0, 10, mr);
94 // printf("r %s\n", sr);
95 <%=name%>_fromMpz(r, mr);
96}
97
98void <%=name%>_mod(P<%=name%>Element r, P<%=name%>Element a, P<%=name%>Element b) {
99 mpz_t ma;
100 mpz_t mb;
101 mpz_t mr;
102 mpz_init(ma);
103 mpz_init(mb);
104 mpz_init(mr);
105
106 <%=name%>_toMpz(ma, a);
107 <%=name%>_toMpz(mb, b);
108 mpz_fdiv_r(mr, ma, mb);
109 <%=name%>_fromMpz(r, mr);
110}
111
112void <%=name%>_shl(P<%=name%>Element r, P<%=name%>Element a, P<%=name%>Element b) {
113 mpz_t ma;
114 mpz_t mb;
115 mpz_t mr;
116 mpz_init(ma);
117 mpz_init(mb);
118 mpz_init(mr);
119
120 <%=name%>_toMpz(ma, a);
121 <%=name%>_toMpz(mb, b);
122 if (mpz_cmp_ui(mb, nBits) >= 0) {
123 mpz_set(mr, zero);
124 } else {
125 mpz_mul_2exp(mr, ma, mpz_get_ui(mb));
126 mpz_and(mr, mr, mask);
127 }
128 <%=name%>_fromMpz(r, mr);
129}
130
131void <%=name%>_shr(P<%=name%>Element r, P<%=name%>Element a, P<%=name%>Element b) {
132 mpz_t ma;
133 mpz_t mb;
134 mpz_t mr;
135 mpz_init(ma);
136 mpz_init(mb);
137 mpz_init(mr);
138
139 <%=name%>_toMpz(ma, a);
140 <%=name%>_toMpz(mb, b);
141 if (mpz_cmp_ui(mb, nBits) >= 0) {
142 mpz_set(mr, zero);
143 } else {
144 mpz_tdiv_q_2exp(mr, ma, mpz_get_ui(mb));
145 mpz_and(mr, mr, mask);
146 }
147 <%=name%>_fromMpz(r, mr);
148}
149
150
151void <%=name%>_pow(P<%=name%>Element r, P<%=name%>Element a, P<%=name%>Element b) {
152 mpz_t ma;
153 mpz_t mb;
154 mpz_t mr;
155 mpz_init(ma);
156 mpz_init(mb);
157 mpz_init(mr);
158
159 <%=name%>_toMpz(ma, a);
160 <%=name%>_toMpz(mb, b);
161 mpz_powm(mr, ma, mb, q);
162 <%=name%>_fromMpz(r, mr);
163}
164
165void <%=name%>_inv(P<%=name%>Element r, P<%=name%>Element a) {
166 mpz_t ma;
167 mpz_t mr;
168 mpz_init(ma);
169 mpz_init(mr);
170
171 <%=name%>_toMpz(ma, a);
172 mpz_invert(mr, ma, q);
173 <%=name%>_fromMpz(r, mr);
174}
175
176void <%=name%>_div(P<%=name%>Element r, P<%=name%>Element a, P<%=name%>Element b) {
177 <%=name%>Element tmp;
178 <%=name%>_inv(&tmp, b);
179 <%=name%>_mul(r, a, &tmp);
180}
181
182void <%=name%>_fail() {
183 assert(false);
184}
185