chiark / gitweb /
.gdbinit: Delete this obsolete file.
[catacomb-python] / pgen.c
1 /* -*-c-*-
2  *
3  * Prime number generation
4  *
5  * (c) 2005 Straylight/Edgeware
6  */
7
8 /*----- Licensing notice --------------------------------------------------*
9  *
10  * This file is part of the Python interface to Catacomb.
11  *
12  * Catacomb/Python is free software; you can redistribute it and/or modify
13  * it under the terms of the GNU General Public License as published by
14  * the Free Software Foundation; either version 2 of the License, or
15  * (at your option) any later version.
16  *
17  * Catacomb/Python is distributed in the hope that it will be useful,
18  * but WITHOUT ANY WARRANTY; without even the implied warranty of
19  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
20  * GNU General Public License for more details.
21  *
22  * You should have received a copy of the GNU General Public License
23  * along with Catacomb/Python; if not, write to the Free Software Foundation,
24  * Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
25  */
26
27 /*----- Header files ------------------------------------------------------*/
28
29 #include "catacomb-python.h"
30
31 /*----- Filters -----------------------------------------------------------*/
32
33 PyTypeObject *pfilt_pytype;
34
35 static PyObject *pfilt_pywrap(pfilt *f)
36 {
37   pfilt_pyobj *o = 0;
38   o = PyObject_New(pfilt_pyobj, pfilt_pytype);
39   o->f = *f;
40   o->st = pfilt_step(f, 0);
41   return ((PyObject *)o);
42 }
43
44 static PyObject *pfilt_pymake(PyTypeObject *ty, PyObject *xobj)
45 {
46   mp *x = 0;
47   pfilt_pyobj *o = 0;
48
49   if (PFILT_PYCHECK(xobj)) RETURN_OBJ(xobj);
50   if ((x = getmp(xobj)) == 0) goto end;
51   o = (pfilt_pyobj *)ty->tp_alloc(ty, 0);
52   o->st = pfilt_create(&o->f, x);
53 end:
54   mp_drop(x);
55   return ((PyObject *)o);
56 }
57
58 static PyObject *pfilt_pynew(PyTypeObject *ty, PyObject *arg, PyObject *kw)
59 {
60   static const char *const kwlist[] = { "x", 0 };
61   PyObject *xobj;
62
63   if (!PyArg_ParseTupleAndKeywords(arg, kw, "O:new", KWLIST, &xobj))
64     return (0);
65   return (pfilt_pymake(ty, xobj));
66 }
67
68 static void pfilt_pydealloc(PyObject *me)
69   { pfilt_destroy(PFILT_F(me)); FREEOBJ(me); }
70
71 static PyObject *pfmeth_step(PyObject *me, PyObject *arg)
72 {
73   mpw x;
74
75   if (!PyArg_ParseTuple(arg, "O&:step", convmpw, &x)) return (0);
76   PFILT_ST(me) = pfilt_step(PFILT_F(me), x);
77   RETURN_ME;
78 }
79
80 static PyObject *pfmeth_muladd(PyObject *me, PyObject *arg)
81 {
82   mpw m, a;
83   pfilt_pyobj *o = 0;
84
85   if (!PyArg_ParseTuple(arg, "O&O&:muladd", convmpw, &m, convmpw, &a))
86     return (0);
87   o = PyObject_New(pfilt_pyobj, pfilt_pytype);
88   o->st = pfilt_muladd(&o->f, PFILT_F(me), m, a);
89   return ((PyObject *)o);
90 }
91
92 static CONVFUNC(pfilt, pfilt *, PFILT_F)
93
94 static PyObject *pfmeth_jump(PyObject *me, PyObject *arg)
95 {
96   pfilt *f;
97
98   if (!PyArg_ParseTuple(arg, "O&:jump", convpfilt, &f)) return (0);
99   PFILT_ST(me) = pfilt_jump(PFILT_F(me), f);
100   RETURN_ME;
101 }
102
103 static PyObject *meth__PrimeFilter_smallfactor(PyObject *me, PyObject *arg)
104 {
105   mp *x = 0;
106   PyObject *rc = 0;
107
108   if (!PyArg_ParseTuple(arg, "OO&:smallfactor", &me, convmp, &x)) goto end;
109   rc = PyInt_FromLong(pfilt_smallfactor(x));
110 end:
111   mp_drop(x);
112   return (rc);
113 }
114
115 static int pfilt_pynonzerop(PyObject *me)
116   { return (PFILT_ST(me) != PGEN_FAIL); }
117
118 static PyObject *pfilt_pyint(PyObject *me)
119 {
120   long l;
121   PyObject *rc = 0;
122
123   if (!mp_tolong_checked(PFILT_F(me)->m, &l, 0)) rc = PyInt_FromLong(l);
124   else rc = mp_topylong(PFILT_F(me)->m);
125   return (rc);
126 }
127
128 static PyObject *pfilt_pylong(PyObject *me)
129   { return (mp_topylong(PFILT_F(me)->m)); }
130
131 static PyObject *pfget_x(PyObject *me, void *hunoz)
132   { return (mp_pywrap(MP_COPY(PFILT_F(me)->m))); }
133
134 static PyObject *pfget_status(PyObject *me, void *hunoz)
135   { return (PyInt_FromLong(PFILT_ST(me))); }
136
137 static PyGetSetDef pfilt_pygetset[] = {
138 #define GETSETNAME(op, name) pf##op##_##name
139   GET   (x,             "F.x -> current position of filter")
140   GET   (status,        "F.status -> primality status of filter")
141 #undef GETSETNAME
142   { 0 }
143 };
144
145 static PyMethodDef pfilt_pymethods[] = {
146 #define METHNAME(name) pfmeth_##name
147   METH  (step,          "F.step(N)")
148   METH  (muladd,        "F.muladd(M, A)")
149   METH  (jump,          "F.jump(FF)")
150 #undef METHNAME
151   { 0 }
152 };
153
154 static PyNumberMethods pfilt_pynumber = {
155   0,                                    /* @nb_add@ */
156   0,                                    /* @nb_subtract@ */
157   0,                                    /* @nb_multiply@ */
158   0,                                    /* @nb_divide@ */
159   0,                                    /* @nb_remainder@ */
160   0,                                    /* @nb_divmod@ */
161   0,                                    /* @nb_power@ */
162   0,                                    /* @nb_negative@ */
163   0,                                    /* @nb_positive@ */
164   0,                                    /* @nb_absolute@ */
165   pfilt_pynonzerop,                     /* @nb_nonzero@ */
166   0,                                    /* @nb_invert@ */
167   0,                                    /* @nb_lshift@ */
168   0,                                    /* @nb_rshift@ */
169   0,                                    /* @nb_and@ */
170   0,                                    /* @nb_xor@ */
171   0,                                    /* @nb_or@ */
172   0,                                    /* @nb_coerce@ */
173   pfilt_pyint,                          /* @nb_int@ */
174   pfilt_pylong,                         /* @nb_long@ */
175   0,                                    /* @nb_float@ */
176   0,                                    /* @nb_oct@ */
177   0,                                    /* @nb_hex@ */
178
179   0,                                    /* @nb_inplace_add@ */
180   0,                                    /* @nb_inplace_subtract@ */
181   0,                                    /* @nb_inplace_multiply@ */
182   0,                                    /* @nb_inplace_divide@ */
183   0,                                    /* @nb_inplace_remainder@ */
184   0,                                    /* @nb_inplace_power@ */
185   0,                                    /* @nb_inplace_lshift@ */
186   0,                                    /* @nb_inplace_rshift@ */
187   0,                                    /* @nb_inplace_and@ */
188   0,                                    /* @nb_inplace_xor@ */
189   0,                                    /* @nb_inplace_or@ */
190
191   0,                                    /* @nb_floor_divide@ */
192   0,                                    /* @nb_true_divide@ */
193   0,                                    /* @nb_inplace_floor_divide@ */
194   0,                                    /* @nb_inplace_true_divide@ */
195 };
196
197 static PyTypeObject pfilt_pytype_skel = {
198   PyObject_HEAD_INIT(0) 0,              /* Header */
199   "PrimeFilter",                        /* @tp_name@ */
200   sizeof(pfilt_pyobj),                  /* @tp_basicsize@ */
201   0,                                    /* @tp_itemsize@ */
202
203   pfilt_pydealloc,                      /* @tp_dealloc@ */
204   0,                                    /* @tp_print@ */
205   0,                                    /* @tp_getattr@ */
206   0,                                    /* @tp_setattr@ */
207   0,                                    /* @tp_compare@ */
208   0,                                    /* @tp_repr@ */
209   &pfilt_pynumber,                      /* @tp_as_number@ */
210   0,                                    /* @tp_as_sequence@ */
211   0,                                    /* @tp_as_mapping@ */
212   0,                                    /* @tp_hash@ */
213   0,                                    /* @tp_call@ */
214   0,                                    /* @tp_str@ */
215   0,                                    /* @tp_getattro@ */
216   0,                                    /* @tp_setattro@ */
217   0,                                    /* @tp_as_buffer@ */
218   Py_TPFLAGS_DEFAULT |                  /* @tp_flags@ */
219     Py_TPFLAGS_BASETYPE,
220
221   /* @tp_doc@ */
222 "PrimeFilter(X): small-primes filter.",
223
224   0,                                    /* @tp_traverse@ */
225   0,                                    /* @tp_clear@ */
226   0,                                    /* @tp_richcompare@ */
227   0,                                    /* @tp_weaklistoffset@ */
228   0,                                    /* @tp_iter@ */
229   0,                                    /* @tp_iternext@ */
230   pfilt_pymethods,                      /* @tp_methods@ */
231   0,                                    /* @tp_members@ */
232   pfilt_pygetset,                       /* @tp_getset@ */
233   0,                                    /* @tp_base@ */
234   0,                                    /* @tp_dict@ */
235   0,                                    /* @tp_descr_get@ */
236   0,                                    /* @tp_descr_set@ */
237   0,                                    /* @tp_dictoffset@ */
238   0,                                    /* @tp_init@ */
239   PyType_GenericAlloc,                  /* @tp_alloc@ */
240   pfilt_pynew,                          /* @tp_new@ */
241   0,                                    /* @tp_free@ */
242   0                                     /* @tp_is_gc@ */
243 };
244
245 /*----- Rabin-Miller testing ----------------------------------------------*/
246
247 typedef struct rabin_pyobj {
248   PyObject_HEAD
249   rabin r;
250 } rabin_pyobj;
251
252 static PyTypeObject *rabin_pytype;
253 #define RABIN_R(o) (&((rabin_pyobj *)(o))->r)
254
255 static PyObject *rabin_pynew(PyTypeObject *ty, PyObject *arg, PyObject *kw)
256 {
257   mp *x = 0;
258   rabin_pyobj *o = 0;
259   static const char *const kwlist[] = { "x", 0 };
260
261   if (!PyArg_ParseTupleAndKeywords(arg, kw, "O&:new", KWLIST, convmp, &x))
262     goto end;
263   if (!MP_POSP(x) || MP_EVENP(x)) VALERR("must be positive and odd");
264   o = (rabin_pyobj *)ty->tp_alloc(ty, 0);
265   rabin_create(&o->r, x);
266 end:
267   return ((PyObject *)o);
268 }
269
270 static void rabin_pydealloc(PyObject *me)
271 {
272   rabin_destroy(RABIN_R(me));
273   FREEOBJ(me);
274 }
275
276 static PyObject *rmeth_test(PyObject *me, PyObject *arg)
277 {
278   mp *w = 0;
279   PyObject *rc = 0;
280
281   if (!PyArg_ParseTuple(arg, "O&:test", convmp, &w)) goto end;
282   rc = PyInt_FromLong(rabin_test(RABIN_R(me), w));
283 end:
284   mp_drop(w);
285   return (rc);
286 }
287
288 static PyObject *rmeth_rtest(PyObject *me, PyObject *arg)
289 {
290   mp *w = 0;
291   PyObject *rc = 0;
292
293   if (!PyArg_ParseTuple(arg, "O&:rtest", convmp, &w)) goto end;
294   rc = PyInt_FromLong(rabin_rtest(RABIN_R(me), w));
295 end:
296   mp_drop(w);
297   return (rc);
298 }
299
300 static PyObject *rget_niters(PyObject *me, void *hunoz)
301   { return (PyInt_FromLong(rabin_iters(mp_bits(RABIN_R(me)->mm.m)))); }
302
303 static PyObject *rget_x(PyObject *me, void *hunoz)
304   { return (mp_pywrap(MP_COPY(RABIN_R(me)->mm.m))); }
305
306 static PyObject *meth__RabinMiller_iters(PyObject *me, PyObject *arg)
307 {
308   unsigned n;
309
310   if (!PyArg_ParseTuple(arg, "OO&:iters", &me, convuint, &n)) return (0);
311   return (PyInt_FromLong(rabin_iters(n)));
312 }
313
314 static PyGetSetDef rabin_pygetset[] = {
315 #define GETSETNAME(op, name) r##op##_##name
316   GET   (x,             "R.x -> number under test")
317   GET   (niters,        "R.niters -> suggested number of tests")
318 #undef GETSETNAME
319   { 0 }
320 };
321
322 static PyMethodDef rabin_pymethods[] = {
323 #define METHNAME(name) rmeth_##name
324   METH  (test,          "R.test(W) -> PGST")
325   METH  (rtest,         "R.rtest(W) -> PGST")
326 #undef METHNAME
327   { 0 }
328 };
329
330 static PyTypeObject rabin_pytype_skel = {
331   PyObject_HEAD_INIT(0) 0,              /* Header */
332   "RabinMiller",                        /* @tp_name@ */
333   sizeof(rabin_pyobj),                  /* @tp_basicsize@ */
334   0,                                    /* @tp_itemsize@ */
335
336   rabin_pydealloc,                      /* @tp_dealloc@ */
337   0,                                    /* @tp_print@ */
338   0,                                    /* @tp_getattr@ */
339   0,                                    /* @tp_setattr@ */
340   0,                                    /* @tp_compare@ */
341   0,                                    /* @tp_repr@ */
342   0,                                    /* @tp_as_number@ */
343   0,                                    /* @tp_as_sequence@ */
344   0,                                    /* @tp_as_mapping@ */
345   0,                                    /* @tp_hash@ */
346   0,                                    /* @tp_call@ */
347   0,                                    /* @tp_str@ */
348   0,                                    /* @tp_getattro@ */
349   0,                                    /* @tp_setattro@ */
350   0,                                    /* @tp_as_buffer@ */
351   Py_TPFLAGS_DEFAULT |                  /* @tp_flags@ */
352     Py_TPFLAGS_BASETYPE,
353
354   /* @tp_doc@ */
355 "RabinMiller(X): Rabin-Miller strong primality test.",
356
357   0,                                    /* @tp_traverse@ */
358   0,                                    /* @tp_clear@ */
359   0,                                    /* @tp_richcompare@ */
360   0,                                    /* @tp_weaklistoffset@ */
361   0,                                    /* @tp_iter@ */
362   0,                                    /* @tp_iternext@ */
363   rabin_pymethods,                      /* @tp_methods@ */
364   0,                                    /* @tp_members@ */
365   rabin_pygetset,                       /* @tp_getset@ */
366   0,                                    /* @tp_base@ */
367   0,                                    /* @tp_dict@ */
368   0,                                    /* @tp_descr_get@ */
369   0,                                    /* @tp_descr_set@ */
370   0,                                    /* @tp_dictoffset@ */
371   0,                                    /* @tp_init@ */
372   PyType_GenericAlloc,                  /* @tp_alloc@ */
373   rabin_pynew,                          /* @tp_new@ */
374   0,                                    /* @tp_free@ */
375   0                                     /* @tp_is_gc@ */
376 };
377
378 /*----- Events ------------------------------------------------------------*/
379
380 typedef struct pgevent_pyobj {
381   PyObject_HEAD
382   PyObject *r;
383   pgen_event *ev;
384 } pgevent_pyobj;
385
386 static PyTypeObject *pgevent_pytype;
387 #define PGEVENT_EV(o) (((pgevent_pyobj *)(o))->ev)
388
389 static PyObject *pgevent_pywrap(pgen_event *ev)
390 {
391   pgevent_pyobj *o = PyObject_New(pgevent_pyobj, pgevent_pytype);
392   o->ev = ev; o->r = 0;
393   return ((PyObject *)o);
394 }
395
396 static CONVFUNC(pgevent, pgen_event *, PGEVENT_EV)
397
398 static void pgevent_kill(PyObject *me)
399 {
400   pgevent_pyobj *ev = (pgevent_pyobj *)me;
401
402   ev->ev = 0;
403   if (ev->r) GRAND_R(ev->r) = 0;
404 }
405
406 static void pgevent_pydealloc(PyObject *me)
407 {
408   pgevent_pyobj *ev = (pgevent_pyobj *)me;
409   if (ev->r) Py_DECREF(ev->r);
410   FREEOBJ(me);
411 }
412
413 #define PGEVENT_CHECK(me) do {                                          \
414   if (!PGEVENT_EV(me)) {                                                \
415     PyErr_SetString(PyExc_ValueError, "event object is no longer valid"); \
416     return (0);                                                         \
417   }                                                                     \
418 } while (0)
419
420 static PyObject *peget_name(PyObject *me, void *hunoz)
421   { PGEVENT_CHECK(me); return (PyString_FromString(PGEVENT_EV(me)->name)); }
422
423 static PyObject *peget_x(PyObject *me, void *hunoz)
424   { PGEVENT_CHECK(me); return (mp_pywrap(MP_COPY(PGEVENT_EV(me)->m))); }
425
426 static PyObject *peget_steps(PyObject *me, void *hunoz)
427   { PGEVENT_CHECK(me); return (PyInt_FromLong(PGEVENT_EV(me)->steps)); }
428
429 static PyObject *peget_tests(PyObject *me, void *hunoz)
430   { PGEVENT_CHECK(me); return (PyInt_FromLong(PGEVENT_EV(me)->tests)); }
431
432 static PyObject *peget_rng(PyObject *me, void *hunoz)
433 {
434   pgevent_pyobj *ev = (pgevent_pyobj *)me;
435
436   PGEVENT_CHECK(me);
437   if (!ev->r) ev->r = grand_pywrap(ev->ev->r, 0);
438   Py_INCREF(ev->r); return ((PyObject *)ev->r);
439 }
440
441 static int peset_x(PyObject *me, PyObject *xobj, void *hunoz)
442 {
443   mp *x = 0;
444   pgen_event *ev = PGEVENT_EV(me);
445   int rc = -1;
446   if (!xobj) NIERR("__del__");
447   PGEVENT_CHECK(me);
448   if ((x = getmp(xobj)) == 0) goto end;
449   mp_drop(ev->m);
450   ev->m = MP_COPY(x);
451   rc = 0;
452 end:
453   mp_drop(x);
454   return (rc);
455 }
456
457 static PyGetSetDef pgevent_pygetset[] = {
458 #define GETSETNAME(op, name) pe##op##_##name
459   GET   (name,          "EV.name -> value being generated")
460   GETSET(x,             "EV.x -> value under test")
461   GET   (steps,         "EV.steps -> number of steps left")
462   GET   (tests,         "EV.tests -> tests before passing")
463   GET   (rng,           "EV.rng -> (noncrypto) random number generator")
464 #undef GETSETNAME
465   { 0 }
466 };
467
468 static PyTypeObject pgevent_pytype_skel = {
469   PyObject_HEAD_INIT(0) 0,              /* Header */
470   "PrimeGenEvent",                      /* @tp_name@ */
471   sizeof(pgevent_pyobj),                /* @tp_basicsize@ */
472   0,                                    /* @tp_itemsize@ */
473
474   pgevent_pydealloc,                    /* @tp_dealloc@ */
475   0,                                    /* @tp_print@ */
476   0,                                    /* @tp_getattr@ */
477   0,                                    /* @tp_setattr@ */
478   0,                                    /* @tp_compare@ */
479   0,                                    /* @tp_repr@ */
480   0,                                    /* @tp_as_number@ */
481   0,                                    /* @tp_as_sequence@ */
482   0,                                    /* @tp_as_mapping@ */
483   0,                                    /* @tp_hash@ */
484   0,                                    /* @tp_call@ */
485   0,                                    /* @tp_str@ */
486   0,                                    /* @tp_getattro@ */
487   0,                                    /* @tp_setattro@ */
488   0,                                    /* @tp_as_buffer@ */
489   Py_TPFLAGS_DEFAULT |                  /* @tp_flags@ */
490     Py_TPFLAGS_BASETYPE,
491
492   /* @tp_doc@ */
493 "Prime-generation event.",
494
495   0,                                    /* @tp_traverse@ */
496   0,                                    /* @tp_clear@ */
497   0,                                    /* @tp_richcompare@ */
498   0,                                    /* @tp_weaklistoffset@ */
499   0,                                    /* @tp_iter@ */
500   0,                                    /* @tp_iternext@ */
501   0,                                    /* @tp_methods@ */
502   0,                                    /* @tp_members@ */
503   pgevent_pygetset,                     /* @tp_getset@ */
504   0,                                    /* @tp_base@ */
505   0,                                    /* @tp_dict@ */
506   0,                                    /* @tp_descr_get@ */
507   0,                                    /* @tp_descr_set@ */
508   0,                                    /* @tp_dictoffset@ */
509   0,                                    /* @tp_init@ */
510   PyType_GenericAlloc,                  /* @tp_alloc@ */
511   abstract_pynew,                       /* @tp_new@ */
512   0,                                    /* @tp_free@ */
513   0                                     /* @tp_is_gc@ */
514 };
515
516 /*----- Event handlers ----------------------------------------------------*/
517
518 PyTypeObject *pgev_pytype;
519
520 typedef struct pgstep_pyobj {
521   PGEV_HEAD
522   pgen_filterctx f;
523 } pgstep_pyobj;
524
525 static PyTypeObject *pgstep_pytype;
526 #define PGSTEP_STEP(o) (((pgstep_pyobj *)(o))->f.step)
527
528 typedef struct pgjump_pyobj {
529   PGEV_HEAD
530   PyObject *fobj;
531   pgen_jumpctx j;
532 } pgjump_pyobj;
533
534 static PyTypeObject *pgjump_pytype;
535 #define PGJUMP_FOBJ(o) (((pgjump_pyobj *)(o))->fobj)
536 #define PGJUMP_J(o) (&((pgjump_pyobj *)(o))->j)
537
538 typedef struct pgtest_pyobj {
539   PGEV_HEAD
540   rabin r;
541 } pgtest_pyobj;
542
543 static PyTypeObject *pgtest_pytype;
544
545 static int pgev_python(int rq, pgen_event *ev, void *p)
546 {
547   pypgev *pg = p;
548   PyObject *pyev = 0;
549   PyObject *rc = 0;
550   int st = PGEN_ABORT;
551   long l;
552   static const char *const meth[] =
553     { "pg_abort", "pg_done", "pg_begin", "pg_try", "pg_fail", "pg_pass" };
554
555   rq++;
556   if (rq > N(meth)) SYSERR("event code out of range");
557   pyev = pgevent_pywrap(ev);
558   if ((rc = PyObject_CallMethod(pg->obj, (/*unconst*/ char *)meth[rq],
559                                 "(O)", pyev)) == 0)
560     goto end;
561   if (rc == Py_None)
562     st = PGEN_TRY;
563   else if ((l = PyInt_AsLong(rc)) == -1 && PyErr_Occurred())
564     goto end;
565   else if (l < PGEN_ABORT || l > PGEN_PASS)
566     VALERR("return code out of range");
567   else
568     st = l;
569 end:
570   if (PyErr_Occurred())
571     stash_exception(pg->exc, "exception from `pgen' handler");
572   if (pyev) {
573     pgevent_kill(pyev);
574     Py_DECREF(pyev);
575   }
576   Py_XDECREF(rc);
577   return (st);
578 }
579
580 static PyObject *pgev_pywrap(const pgev *pg)
581 {
582   pgev_pyobj *o;
583
584   o = PyObject_New(pgev_pyobj, pgev_pytype);
585   o->pg = *pg;
586   return ((PyObject *)o);
587 }
588
589 int convpgev(PyObject *o, void *p)
590 {
591   pypgev *pg = p;
592
593   if (PGEV_PYCHECK(o))
594     pg->ev = *PGEV_PG(o);
595   else {
596     pg->ev.proc = pgev_python;
597     pg->ev.ctx = pg;
598     pg->obj = o; Py_INCREF(o);
599   }
600   return (1);
601 }
602
603 void droppgev(pypgev *pg)
604 {
605   if (pg->ev.proc == pgev_python)
606     { assert(pg->ev.ctx == pg); Py_DECREF(pg->obj); }
607 }
608
609 static PyObject *pgmeth_common(PyObject *me, PyObject *arg, int rq)
610 {
611   pgen_event *ev;
612   pgev *pg = PGEV_PG(me);
613   PyObject *rc = 0;
614
615   if (!PyArg_ParseTuple(arg, "O&", convpgevent, &ev)) goto end;
616   rc = PyInt_FromLong(!pg->proc ? rq : pg->proc(rq, ev, pg->ctx));
617 end:
618   return (rc);
619 }
620
621 #define PGMETH(lc, uc)                                                  \
622   static PyObject *pgmeth_pg_##lc(PyObject *me, PyObject *arg)          \
623     { return pgmeth_common(me, arg, PGEN_##uc); }
624 PGMETH(abort,   ABORT)
625 PGMETH(done,    DONE)
626 PGMETH(begin,   BEGIN)
627 PGMETH(try,     TRY)
628 PGMETH(pass,    PASS)
629 PGMETH(fail,    FAIL)
630 #undef PGMETH
631
632 static PyObject *pgev_stdev(pgen_proc *proc)
633   { pgev pg; pg.proc = proc; pg.ctx = 0; return (pgev_pywrap(&pg)); }
634
635 static PyMethodDef pgev_pymethods[] = {
636 #define METHNAME(name) pgmeth_##name
637   METH  (pg_abort,      "E.pg_abort(EV) -> PGRC -- prime generation aborted")
638   METH  (pg_done,       "E.pg_done(EV) -> PGRC -- prime generation finished")
639   METH  (pg_begin,     "E.pg_begin(EV) -> PGRC -- commence stepping/testing")
640   METH  (pg_try,        "E.pg_try(EV) -> PGRC -- found new candidate")
641   METH  (pg_pass,       "E.pg_pass(EV) -> PGRC -- passed primality test")
642   METH  (pg_fail,       "E.pg_fail(EV) -> PGRC -- failed primality test")
643 #undef METHNAME
644   { 0 }
645 };
646
647 static PyTypeObject pgev_pytype_skel = {
648   PyObject_HEAD_INIT(0) 0,              /* Header */
649   "PrimeGenBuiltinHandler",             /* @tp_name@ */
650   sizeof(pgev_pyobj),                   /* @tp_basicsize@ */
651   0,                                    /* @tp_itemsize@ */
652
653   0,                                    /* @tp_dealloc@ */
654   0,                                    /* @tp_print@ */
655   0,                                    /* @tp_getattr@ */
656   0,                                    /* @tp_setattr@ */
657   0,                                    /* @tp_compare@ */
658   0,                                    /* @tp_repr@ */
659   0,                                    /* @tp_as_number@ */
660   0,                                    /* @tp_as_sequence@ */
661   0,                                    /* @tp_as_mapping@ */
662   0,                                    /* @tp_hash@ */
663   0,                                    /* @tp_call@ */
664   0,                                    /* @tp_str@ */
665   0,                                    /* @tp_getattro@ */
666   0,                                    /* @tp_setattro@ */
667   0,                                    /* @tp_as_buffer@ */
668   Py_TPFLAGS_DEFAULT |                  /* @tp_flags@ */
669     Py_TPFLAGS_BASETYPE,
670
671   /* @tp_doc@ */
672 "Built-in prime-generation event handler, base class.",
673
674   0,                                    /* @tp_traverse@ */
675   0,                                    /* @tp_clear@ */
676   0,                                    /* @tp_richcompare@ */
677   0,                                    /* @tp_weaklistoffset@ */
678   0,                                    /* @tp_iter@ */
679   0,                                    /* @tp_iternext@ */
680   pgev_pymethods,                       /* @tp_methods@ */
681   0,                                    /* @tp_members@ */
682   0,                                    /* @tp_getset@ */
683   0,                                    /* @tp_base@ */
684   0,                                    /* @tp_dict@ */
685   0,                                    /* @tp_descr_get@ */
686   0,                                    /* @tp_descr_set@ */
687   0,                                    /* @tp_dictoffset@ */
688   0,                                    /* @tp_init@ */
689   PyType_GenericAlloc,                  /* @tp_alloc@ */
690   abstract_pynew,                       /* @tp_new@ */
691   0,                                    /* @tp_free@ */
692   0                                     /* @tp_is_gc@ */
693 };
694
695 static PyObject *pgstep_pynew(PyTypeObject *ty, PyObject *arg, PyObject *kw)
696 {
697   mpw s;
698   pgstep_pyobj *rc = 0;
699   static const char *const kwlist[] = { "step", 0 };
700
701   if (!PyArg_ParseTupleAndKeywords(arg, kw, "O&:new", KWLIST, convmpw, &s))
702     goto end;
703   rc = (pgstep_pyobj *)ty->tp_alloc(ty, 0);
704   rc->f.step = s;
705   rc->pg.proc = pgen_filter;
706   rc->pg.ctx = &rc->f;
707 end:
708   return ((PyObject *)rc);
709 }
710
711 static PyObject *psget_step(PyObject *me, void *hunoz)
712   { return (PyInt_FromLong(PGSTEP_STEP(me))); }
713
714 static PyGetSetDef pgstep_pygetset[] = {
715 #define GETSETNAME(op, name) ps##op##_##name
716   GET   (step,          "S.step -> step size for the stepper")
717 #undef GETSETNAME
718   { 0 }
719 };
720
721 static PyTypeObject pgstep_pytype_skel = {
722   PyObject_HEAD_INIT(0) 0,              /* Header */
723   "PrimeGenStepper",                    /* @tp_name@ */
724   sizeof(pgstep_pyobj),                 /* @tp_basicsize@ */
725   0,                                    /* @tp_itemsize@ */
726
727   0,                                    /* @tp_dealloc@ */
728   0,                                    /* @tp_print@ */
729   0,                                    /* @tp_getattr@ */
730   0,                                    /* @tp_setattr@ */
731   0,                                    /* @tp_compare@ */
732   0,                                    /* @tp_repr@ */
733   0,                                    /* @tp_as_number@ */
734   0,                                    /* @tp_as_sequence@ */
735   0,                                    /* @tp_as_mapping@ */
736   0,                                    /* @tp_hash@ */
737   0,                                    /* @tp_call@ */
738   0,                                    /* @tp_str@ */
739   0,                                    /* @tp_getattro@ */
740   0,                                    /* @tp_setattro@ */
741   0,                                    /* @tp_as_buffer@ */
742   Py_TPFLAGS_DEFAULT |                  /* @tp_flags@ */
743     Py_TPFLAGS_BASETYPE,
744
745   /* @tp_doc@ */
746 "PrimeGenStepper(STEP): simple stepper with small-factors filter.",
747
748   0,                                    /* @tp_traverse@ */
749   0,                                    /* @tp_clear@ */
750   0,                                    /* @tp_richcompare@ */
751   0,                                    /* @tp_weaklistoffset@ */
752   0,                                    /* @tp_iter@ */
753   0,                                    /* @tp_iternext@ */
754   0,                                    /* @tp_methods@ */
755   0,                                    /* @tp_members@ */
756   pgstep_pygetset,                      /* @tp_getset@ */
757   0,                                    /* @tp_base@ */
758   0,                                    /* @tp_dict@ */
759   0,                                    /* @tp_descr_get@ */
760   0,                                    /* @tp_descr_set@ */
761   0,                                    /* @tp_dictoffset@ */
762   0,                                    /* @tp_init@ */
763   PyType_GenericAlloc,                  /* @tp_alloc@ */
764   pgstep_pynew,                         /* @tp_new@ */
765   0,                                    /* @tp_free@ */
766   0                                     /* @tp_is_gc@ */
767 };
768
769 static PyObject *pgjump_pynew(PyTypeObject *ty, PyObject *arg, PyObject *kw)
770 {
771   PyObject *o, *fobj;
772   pgjump_pyobj *rc = 0;
773   static const char *const kwlist[] = { "jump", 0 };
774
775   if (!PyArg_ParseTupleAndKeywords(arg, kw, "O:new", KWLIST, &o) ||
776       (fobj = pfilt_pymake(pfilt_pytype, o)) == 0)
777     goto end;
778   rc = (pgjump_pyobj *)ty->tp_alloc(ty, 0);
779   rc->fobj = fobj;
780   rc->j.j = PFILT_F(fobj);
781   rc->pg.proc = pgen_jump;
782   rc->pg.ctx = &rc->j;
783 end:
784   return ((PyObject *)rc);
785 }
786
787 static void pgjump_pydealloc(PyObject *me)
788 {
789   Py_DECREF(PGJUMP_FOBJ(me));
790   FREEOBJ(me);
791 }
792
793 static PyObject *pjget_jump(PyObject *me, void *hunoz)
794   { RETURN_OBJ(PGJUMP_FOBJ(me)); }
795
796 static PyGetSetDef pgjump_pygetset[] = {
797 #define GETSETNAME(op, name) pj##op##_##name
798   GET   (jump,          "S.jump -> jump size for the stepper")
799 #undef GETSETNAME
800   { 0 }
801 };
802
803 static PyTypeObject pgjump_pytype_skel = {
804   PyObject_HEAD_INIT(0) 0,              /* Header */
805   "PrimeGenJumper",                     /* @tp_name@ */
806   sizeof(pgjump_pyobj),                 /* @tp_basicsize@ */
807   0,                                    /* @tp_itemsize@ */
808
809   pgjump_pydealloc,                     /* @tp_dealloc@ */
810   0,                                    /* @tp_print@ */
811   0,                                    /* @tp_getattr@ */
812   0,                                    /* @tp_setattr@ */
813   0,                                    /* @tp_compare@ */
814   0,                                    /* @tp_repr@ */
815   0,                                    /* @tp_as_number@ */
816   0,                                    /* @tp_as_sequence@ */
817   0,                                    /* @tp_as_mapping@ */
818   0,                                    /* @tp_hash@ */
819   0,                                    /* @tp_call@ */
820   0,                                    /* @tp_str@ */
821   0,                                    /* @tp_getattro@ */
822   0,                                    /* @tp_setattro@ */
823   0,                                    /* @tp_as_buffer@ */
824   Py_TPFLAGS_DEFAULT |                  /* @tp_flags@ */
825     Py_TPFLAGS_BASETYPE,
826
827   /* @tp_doc@ */
828 "PrimeGenJumper(JUMP): stepper for larger steps with small-factors filter.",
829
830   0,                                    /* @tp_traverse@ */
831   0,                                    /* @tp_clear@ */
832   0,                                    /* @tp_richcompare@ */
833   0,                                    /* @tp_weaklistoffset@ */
834   0,                                    /* @tp_iter@ */
835   0,                                    /* @tp_iternext@ */
836   0,                                    /* @tp_methods@ */
837   0,                                    /* @tp_members@ */
838   pgjump_pygetset,                      /* @tp_getset@ */
839   0,                                    /* @tp_base@ */
840   0,                                    /* @tp_dict@ */
841   0,                                    /* @tp_descr_get@ */
842   0,                                    /* @tp_descr_set@ */
843   0,                                    /* @tp_dictoffset@ */
844   0,                                    /* @tp_init@ */
845   PyType_GenericAlloc,                  /* @tp_alloc@ */
846   pgjump_pynew,                         /* @tp_new@ */
847   0,                                    /* @tp_free@ */
848   0                                     /* @tp_is_gc@ */
849 };
850
851 static PyObject *pgtest_pynew(PyTypeObject *ty, PyObject *arg, PyObject *kw)
852 {
853   pgtest_pyobj *rc = 0;
854   static const char *const kwlist[] = { 0 };
855
856   if (!PyArg_ParseTupleAndKeywords(arg, kw, ":new", KWLIST)) goto end;
857   rc = (pgtest_pyobj *)ty->tp_alloc(ty, 0);
858   rc->pg.proc = pgen_test;
859   rc->pg.ctx = &rc->r;
860 end:
861   return ((PyObject *)rc);
862 }
863
864 static PyTypeObject pgtest_pytype_skel = {
865   PyObject_HEAD_INIT(0) 0,              /* Header */
866   "PrimeGenTester",                     /* @tp_name@ */
867   sizeof(pgtest_pyobj),                 /* @tp_basicsize@ */
868   0,                                    /* @tp_itemsize@ */
869
870   0,                                    /* @tp_dealloc@ */
871   0,                                    /* @tp_print@ */
872   0,                                    /* @tp_getattr@ */
873   0,                                    /* @tp_setattr@ */
874   0,                                    /* @tp_compare@ */
875   0,                                    /* @tp_repr@ */
876   0,                                    /* @tp_as_number@ */
877   0,                                    /* @tp_as_sequence@ */
878   0,                                    /* @tp_as_mapping@ */
879   0,                                    /* @tp_hash@ */
880   0,                                    /* @tp_call@ */
881   0,                                    /* @tp_str@ */
882   0,                                    /* @tp_getattro@ */
883   0,                                    /* @tp_setattro@ */
884   0,                                    /* @tp_as_buffer@ */
885   Py_TPFLAGS_DEFAULT |                  /* @tp_flags@ */
886     Py_TPFLAGS_BASETYPE,
887
888   /* @tp_doc@ */
889 "PrimeGenTester(): Rabin-Miller tester.",
890
891   0,                                    /* @tp_traverse@ */
892   0,                                    /* @tp_clear@ */
893   0,                                    /* @tp_richcompare@ */
894   0,                                    /* @tp_weaklistoffset@ */
895   0,                                    /* @tp_iter@ */
896   0,                                    /* @tp_iternext@ */
897   0,                                    /* @tp_methods@ */
898   0,                                    /* @tp_members@ */
899   0,                                    /* @tp_getset@ */
900   0,                                    /* @tp_base@ */
901   0,                                    /* @tp_dict@ */
902   0,                                    /* @tp_descr_get@ */
903   0,                                    /* @tp_descr_set@ */
904   0,                                    /* @tp_dictoffset@ */
905   0,                                    /* @tp_init@ */
906   PyType_GenericAlloc,                  /* @tp_alloc@ */
907   pgtest_pynew,                         /* @tp_new@ */
908   0,                                    /* @tp_free@ */
909   0                                     /* @tp_is_gc@ */
910 };
911
912 /*----- Prime generation functions ----------------------------------------*/
913
914 void pgenerr(struct excinfo *exc)
915 {
916   if (exc->ty) RESTORE_EXCINFO(exc);
917   else PyErr_SetString(PyExc_ValueError, "prime generation failed");
918 }
919
920 static PyObject *meth_pgen(PyObject *me, PyObject *arg, PyObject *kw)
921 {
922   mp *x = 0;
923   mp *r = 0;
924   PyObject *rc = 0;
925   char *p = "p";
926   pgen_filterctx fc = { 2 };
927   rabin tc;
928   struct excinfo exc = EXCINFO_INIT;
929   pypgev step = { { 0 } }, test = { { 0 } }, evt = { { 0 } };
930   unsigned nsteps = 0, ntests = 0;
931   static const char *const kwlist[] =
932     { "start", "name", "stepper", "tester", "event", "nsteps", "ntests", 0 };
933
934   step.exc = &exc; step.ev.proc = pgen_filter; step.ev.ctx = &fc;
935   test.exc = &exc; test.ev.proc = pgen_test; test.ev.ctx = &tc;
936   evt.exc = &exc;
937   if (!PyArg_ParseTupleAndKeywords(arg, kw, "O&|sO&O&O&O&O&:pgen", KWLIST,
938                                    convmp, &x, &p, convpgev, &step,
939                                    convpgev, &test, convpgev, &evt,
940                                    convuint, &nsteps, convuint, &ntests))
941     goto end;
942   if (!ntests) ntests = rabin_iters(mp_bits(x));
943   if ((r = pgen(p, MP_NEW, x, evt.ev.proc, evt.ev.ctx,
944                 nsteps, step.ev.proc, step.ev.ctx,
945                 ntests, test.ev.proc, test.ev.ctx)) == 0)
946     PGENERR(&exc);
947   rc = mp_pywrap(r); r = 0;
948 end:
949   mp_drop(r); mp_drop(x);
950   droppgev(&step); droppgev(&test); droppgev(&evt);
951   return (rc);
952 }
953
954 static PyObject *meth_strongprime_setup(PyObject *me,
955                                         PyObject *arg, PyObject *kw)
956 {
957   mp *x = 0;
958   pfilt f;
959   grand *r = &rand_global;
960   unsigned nbits;
961   char *name = "p";
962   unsigned n = 0;
963   struct excinfo exc = EXCINFO_INIT;
964   pypgev evt = { { 0 } };
965   PyObject *rc = 0;
966   static const char *const kwlist[] =
967     { "nbits", "name", "event", "rng", "nsteps", 0 };
968
969   evt.exc = &exc;
970   if (!PyArg_ParseTupleAndKeywords(arg, kw, "O&|sO&O&O&", KWLIST,
971                                    convuint, &nbits, &name,
972                                    convpgev, &evt, convgrand, &r,
973                                    convuint, &n))
974     goto end;
975   if ((x = strongprime_setup(name, MP_NEW, &f, nbits,
976                              r, n, evt.ev.proc, evt.ev.ctx)) == 0)
977     PGENERR(&exc);
978   rc = Py_BuildValue("(NN)", mp_pywrap(x), pfilt_pywrap(&f));
979   x = 0;
980 end:
981   mp_drop(x);
982   droppgev(&evt);
983   return (rc);
984 }
985
986 static PyObject *meth_strongprime(PyObject *me, PyObject *arg, PyObject *kw)
987 {
988   mp *x = 0;
989   grand *r = &rand_global;
990   unsigned nbits;
991   char *name = "p";
992   unsigned n = 0;
993   struct excinfo exc = EXCINFO_INIT;
994   pypgev evt = { { 0 } };
995   PyObject *rc = 0;
996   static const char *const kwlist[] =
997     { "nbits", "name", "event", "rng", "nsteps", 0 };
998
999   evt.exc = &exc;
1000   if (!PyArg_ParseTupleAndKeywords(arg, kw, "O&|sO&O&O&", KWLIST,
1001                                    convuint, &nbits, &name,
1002                                    convpgev, &evt, convgrand, &r,
1003                                    convuint, &n))
1004     goto end;
1005   if ((x = strongprime(name, MP_NEW, nbits,
1006                        r, n, evt.ev.proc, evt.ev.ctx)) == 0)
1007     PGENERR(&exc);
1008   rc = mp_pywrap(x);
1009   x = 0;
1010 end:
1011   mp_drop(x);
1012   droppgev(&evt);
1013   return (rc);
1014 }
1015
1016 static PyObject *meth_limlee(PyObject *me, PyObject *arg, PyObject *kw)
1017 {
1018   char *p = "p";
1019   struct excinfo exc = EXCINFO_INIT;
1020   pypgev ie = { { 0 } }, oe = { { 0 } };
1021   unsigned ql, pl;
1022   grand *r = &rand_global;
1023   unsigned on = 0;
1024   size_t i, nf = 0;
1025   PyObject *rc = 0, *vec;
1026   static const char *const kwlist[] =
1027     { "pbits", "qbits", "name", "event", "ievent", "rng", "nsteps", 0 };
1028   mp *x = 0, **v = 0;
1029
1030   ie.exc = oe.exc = &exc;
1031   if (!PyArg_ParseTupleAndKeywords(arg, kw, "O&O&|sO&O&O&O&:limlee", KWLIST,
1032                                    convuint, &pl, convuint, &ql,
1033                                    &p, convpgev, &oe, convpgev, &ie,
1034                                    convgrand, &r, convuint, &on))
1035     goto end;
1036   if ((x = limlee(p, MP_NEW, MP_NEW, ql, pl, r, on,
1037                   oe.ev.proc, oe.ev.ctx, ie.ev.proc, ie.ev.ctx,
1038                   &nf, &v)) == 0)
1039     PGENERR(&exc);;
1040   vec = PyList_New(nf);
1041   for (i = 0; i < nf; i++)
1042     PyList_SetItem(vec, i, mp_pywrap(v[i]));
1043   xfree(v);
1044   rc = Py_BuildValue("(NN)", mp_pywrap(x), vec);
1045 end:
1046   droppgev(&oe); droppgev(&ie);
1047   return (rc);
1048 }
1049
1050 /*----- Global stuff ------------------------------------------------------*/
1051
1052 static PyMethodDef methods[] = {
1053 #define METHNAME(name) meth_##name
1054   METH  (_PrimeFilter_smallfactor,      "smallfactor(X) -> PGRC")
1055   METH  (_RabinMiller_iters,            "iters(NBITS) -> NITERS")
1056   KWMETH(pgen,                          "\
1057 pgen(START, [name = 'p'[, [stepper = PrimeGenStepper(2)],\n\
1058      [tester = PrimeGenTester()], [event = pgen_nullev],\n\
1059      [nsteps = 0], [ntests = RabinMiller.iters(START.nbits)]) -> P")
1060   KWMETH(strongprime_setup,             "\
1061 strongprime_setup(NBITS, [name = 'p'], [event = pgen_nullev],\n\
1062                   [rng = rand], [nsteps = 0]) -> (START, JUMP)")
1063   KWMETH(strongprime,                   "\
1064 strongprime(NBITS, [name = 'p'], [event = pgen_nullev],\n\
1065             [rng = rand], [nsteps = 0]) -> P")
1066   KWMETH(limlee,                        "\
1067 limlee(PBITS, QBITS, [name = 'p'], [event = pgen_nullev],\n\
1068        [ievent = pgen_nullev], [rng = rand], [nsteps = 0]) -> (P, [Q, ...])")
1069 #undef METHNAME
1070   { 0 }
1071 };
1072
1073 void pgen_pyinit(void)
1074 {
1075   INITTYPE(pfilt, root);
1076   INITTYPE(rabin, root);
1077   INITTYPE(pgevent, root);
1078   INITTYPE(pgev, root);
1079   INITTYPE(pgstep, pgev);
1080   INITTYPE(pgjump, pgev);
1081   INITTYPE(pgtest, pgev);
1082   addmethods(methods);
1083 }
1084
1085 static PyObject *obj;
1086
1087 void pgen_pyinsert(PyObject *mod)
1088 {
1089   INSERT("PrimeFilter", pfilt_pytype);
1090   INSERT("RabinMiller", rabin_pytype);
1091   INSERT("PrimeGenEvent", pgevent_pytype);
1092   INSERT("PrimeGenBuiltinHandler", pgev_pytype);
1093   INSERT("PrimeGenStepper", pgstep_pytype);
1094   INSERT("PrimeGenJumper", pgjump_pytype);
1095   INSERT("PrimeGenTester", pgtest_pytype);
1096   INSERT("pgen_nullev", obj = pgev_stdev(0));
1097   INSERT("pgen_stdev", pgev_stdev(pgen_ev));
1098   INSERT("pgen_spinev", pgev_stdev(pgen_evspin));
1099   INSERT("pgen_subev", pgev_stdev(pgen_subev));
1100 }
1101
1102 /*----- That's all, folks -------------------------------------------------*/