Line data Source code
1 : /* Copyright (C) 2006 The PARI group.
2 :
3 : This file is part of the PARI package.
4 :
5 : PARI/GP is free software; you can redistribute it and/or modify it under the
6 : terms of the GNU General Public License as published by the Free Software
7 : Foundation; either version 2 of the License, or (at your option) any later
8 : version. It is distributed in the hope that it will be useful, but WITHOUT
9 : ANY WARRANTY WHATSOEVER.
10 :
11 : Check the License for details. You should have received a copy of it, along
12 : with the package; see the file 'COPYING'. If not, write to the Free Software
13 : Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. */
14 :
15 : #include "pari.h"
16 : #include "paripriv.h"
17 : #include "anal.h"
18 : #include "opcode.h"
19 :
20 : /********************************************************************/
21 : /* */
22 : /* break/next/return handling */
23 : /* */
24 : /********************************************************************/
25 :
26 : static THREAD long br_status, br_count;
27 : static THREAD GEN br_res;
28 :
29 : long
30 137599399 : loop_break(void)
31 : {
32 137599399 : switch(br_status)
33 : {
34 21 : case br_MULTINEXT :
35 21 : if (! --br_count) br_status = br_NEXT;
36 21 : return 1;
37 70302 : case br_BREAK : if (! --br_count) br_status = br_NONE; /* fall through */
38 78931 : case br_RETURN: return 1;
39 25125 : case br_NEXT: br_status = br_NONE; /* fall through */
40 : }
41 137520447 : return 0;
42 : }
43 :
44 : static void
45 98512 : reset_break(void)
46 : {
47 98512 : br_status = br_NONE;
48 98512 : if (br_res) { gunclone_deep(br_res); br_res = NULL; }
49 98512 : }
50 :
51 : GEN
52 40898 : return0(GEN x)
53 : {
54 40898 : GEN y = br_res;
55 40898 : br_res = (x && x != gnil)? gcloneref(x): NULL;
56 40898 : guncloneNULL_deep(y);
57 40898 : br_status = br_RETURN; return NULL;
58 : }
59 :
60 : GEN
61 25853 : next0(long n)
62 : {
63 25853 : if (n < 1) pari_err_DOMAIN("next", "n", "<", gen_1, stoi(n));
64 25846 : if (n == 1) br_status = br_NEXT;
65 : else
66 : {
67 14 : br_count = n-1;
68 14 : br_status = br_MULTINEXT;
69 : }
70 25846 : return NULL;
71 : }
72 :
73 : GEN
74 70358 : break0(long n)
75 : {
76 70358 : if (n < 1) pari_err_DOMAIN("break", "n", "<", gen_1, stoi(n));
77 70351 : br_count = n;
78 70351 : br_status = br_BREAK; return NULL;
79 : }
80 :
81 : /*******************************************************************/
82 : /* */
83 : /* VARIABLES */
84 : /* */
85 : /*******************************************************************/
86 :
87 : /* As a rule, ep->value is a clone (COPY). push_val and pop_val are private
88 : * functions for use in sumiter: we want a temporary ep->value, which is NOT
89 : * a clone (PUSH), to avoid unnecessary copies. */
90 :
91 : enum {PUSH_VAL = 0, COPY_VAL = 1, DEFAULT_VAL = 2, REF_VAL = 3};
92 :
93 : /* ep->args is the stack of old values (INITIAL if initial value, from
94 : * installep) */
95 : typedef struct var_cell {
96 : struct var_cell *prev; /* cell attached to previous value on stack */
97 : GEN value; /* last value (not including current one, in ep->value) */
98 : char flag; /* status of _current_ ep->value: PUSH or COPY ? */
99 : long valence; /* valence of entree* attached to 'value', to be restored
100 : * by pop_val */
101 : } var_cell;
102 : #define INITIAL NULL
103 :
104 : /* Push x on value stack attached to ep. */
105 : static void
106 19359 : new_val_cell(entree *ep, GEN x, char flag)
107 : {
108 19359 : var_cell *v = (var_cell*) pari_malloc(sizeof(var_cell));
109 19359 : v->value = (GEN)ep->value;
110 19359 : v->prev = (var_cell*) ep->pvalue;
111 19359 : v->flag = flag;
112 19359 : v->valence= ep->valence;
113 :
114 : /* beware: f(p) = Nv = 0
115 : * Nv = p; f(Nv) --> this call would destroy p [ isclone ] */
116 19359 : ep->value = (flag == COPY_VAL)? gclone(x):
117 0 : (x && isclone(x))? gcopy(x): x;
118 : /* Do this last. In case the clone is <C-C>'ed before completion ! */
119 19359 : ep->pvalue= (char*)v;
120 19359 : ep->valence=EpVAR;
121 19359 : }
122 :
123 : /* kill ep->value and replace by preceding one, poped from value stack */
124 : static void
125 18911 : pop_val(entree *ep)
126 : {
127 18911 : var_cell *v = (var_cell*) ep->pvalue;
128 18911 : if (v != INITIAL)
129 : {
130 18911 : GEN old_val = (GEN) ep->value; /* protect against SIGINT */
131 18911 : ep->value = v->value;
132 18911 : if (v->flag == COPY_VAL) gunclone_deep(old_val);
133 18911 : ep->pvalue = (char*) v->prev;
134 18911 : ep->valence=v->valence;
135 18911 : pari_free((void*)v);
136 : }
137 18911 : }
138 :
139 : void
140 36281 : freeep(entree *ep)
141 : {
142 36281 : if (EpSTATIC(ep)) return; /* gp function loaded at init time */
143 36281 : if (ep->help) {pari_free((void*)ep->help); ep->help=NULL;}
144 36281 : if (ep->code) {pari_free((void*)ep->code); ep->code=NULL;}
145 36281 : switch(EpVALENCE(ep))
146 : {
147 24237 : case EpVAR:
148 43092 : while (ep->pvalue!=INITIAL) pop_val(ep);
149 24237 : break;
150 28 : case EpALIAS:
151 28 : killblock((GEN)ep->value); ep->value=NULL; break;
152 : }
153 : }
154 :
155 : INLINE void
156 42 : pushvalue(entree *ep, GEN x) {
157 42 : new_val_cell(ep, x, COPY_VAL);
158 42 : }
159 :
160 : INLINE void
161 14 : zerovalue(entree *ep)
162 : {
163 14 : var_cell *v = (var_cell*) pari_malloc(sizeof(var_cell));
164 14 : v->value = (GEN)ep->value;
165 14 : v->prev = (var_cell*) ep->pvalue;
166 14 : v->flag = COPY_VAL;
167 14 : v->valence= ep->valence;
168 14 : ep->value = gen_0;
169 14 : ep->pvalue= (char*)v;
170 14 : ep->valence=EpVAR;
171 14 : }
172 :
173 : /* as above IF ep->value was PUSHed, or was created after block number 'loc'
174 : return 0 if not deleted, 1 otherwise [for recover()] */
175 : int
176 424973 : pop_val_if_newer(entree *ep, long loc)
177 : {
178 424973 : var_cell *v = (var_cell*) ep->pvalue;
179 :
180 424973 : if (v == INITIAL) return 0;
181 390088 : if (v->flag == COPY_VAL && !pop_entree_block(ep, loc)) return 0;
182 462 : ep->value = v->value;
183 462 : ep->pvalue= (char*) v->prev;
184 462 : ep->valence=v->valence;
185 462 : pari_free((void*)v); return 1;
186 : }
187 :
188 : /* set new value of ep directly to val (COPY), do not save last value unless
189 : * it's INITIAL. */
190 : void
191 29490346 : changevalue(entree *ep, GEN x)
192 : {
193 29490346 : var_cell *v = (var_cell*) ep->pvalue;
194 29490346 : if (v == INITIAL) new_val_cell(ep, x, COPY_VAL);
195 : else
196 : {
197 29471029 : GEN old_val = (GEN) ep->value; /* beware: gunclone_deep may destroy old x */
198 29471029 : ep->value = (void *) gclone(x);
199 29471029 : if (v->flag == COPY_VAL) gunclone_deep(old_val); else v->flag = COPY_VAL;
200 : }
201 29490346 : }
202 :
203 : INLINE GEN
204 746851 : copyvalue(entree *ep)
205 : {
206 746851 : var_cell *v = (var_cell*) ep->pvalue;
207 746851 : if (v && v->flag != COPY_VAL)
208 : {
209 0 : ep->value = (void*) gclone((GEN)ep->value);
210 0 : v->flag = COPY_VAL;
211 : }
212 746851 : return (GEN) ep->value;
213 : }
214 :
215 : INLINE void
216 0 : err_var(GEN x) { pari_err_TYPE("evaluator [variable name expected]", x); }
217 :
218 : enum chk_VALUE { chk_ERROR, chk_NOCREATE, chk_CREATE };
219 :
220 : INLINE void
221 123803434 : checkvalue(entree *ep, enum chk_VALUE flag)
222 : {
223 123803434 : if (mt_is_thread())
224 27 : pari_err(e_MISC,"mt: attempt to change exported variable '%s'",ep->name);
225 123803407 : if (ep->valence==EpNEW)
226 23676 : switch(flag)
227 : {
228 4675 : case chk_ERROR:
229 : /* Do nothing until we can report a meaningful error message
230 : The extra variable will be cleaned-up anyway */
231 : case chk_CREATE:
232 4675 : pari_var_create(ep);
233 4675 : ep->valence = EpVAR;
234 4675 : ep->value = initial_value(ep);
235 4675 : break;
236 19001 : case chk_NOCREATE:
237 19001 : break;
238 : }
239 123779731 : else if (ep->valence!=EpVAR)
240 0 : pari_err(e_MISC, "attempt to change built-in %s", ep->name);
241 123803407 : }
242 :
243 : INLINE GEN
244 23308355 : checkvalueptr(entree *ep)
245 : {
246 23308355 : checkvalue(ep, chk_NOCREATE);
247 23308355 : return ep->valence==EpNEW? gen_0: (GEN)ep->value;
248 : }
249 :
250 : /* make GP variables safe for set_avma(top) */
251 : static void
252 0 : lvar_make_safe(void)
253 : {
254 : long n;
255 : entree *ep;
256 0 : for (n = 0; n < functions_tblsz; n++)
257 0 : for (ep = functions_hash[n]; ep; ep = ep->next)
258 0 : if (EpVALENCE(ep) == EpVAR)
259 : { /* make sure ep->value is a COPY */
260 0 : var_cell *v = (var_cell*)ep->pvalue;
261 0 : if (v && v->flag == PUSH_VAL) {
262 0 : GEN x = (GEN)ep->value;
263 0 : if (x) changevalue(ep, (GEN)ep->value); else pop_val(ep);
264 : }
265 : }
266 0 : }
267 :
268 : static void
269 107074509 : check_array_index(long c, long l)
270 : {
271 107074509 : if (c < 1) pari_err_COMPONENT("", "<", gen_1, stoi(c));
272 107074502 : if (c >= l) pari_err_COMPONENT("", ">", stoi(l-1), stoi(c));
273 107074460 : }
274 :
275 : GEN*
276 0 : safegel(GEN x, long l)
277 : {
278 0 : if (!is_matvec_t(typ(x)))
279 0 : pari_err_TYPE("safegel",x);
280 0 : check_array_index(l, lg(x));
281 0 : return &(gel(x,l));
282 : }
283 :
284 : GEN*
285 0 : safelistel(GEN x, long l)
286 : {
287 : GEN d;
288 0 : if (typ(x)!=t_LIST || list_typ(x)!=t_LIST_RAW)
289 0 : pari_err_TYPE("safelistel",x);
290 0 : d = list_data(x);
291 0 : check_array_index(l, lg(d));
292 0 : return &(gel(d,l));
293 : }
294 :
295 : long*
296 0 : safeel(GEN x, long l)
297 : {
298 0 : if (typ(x)!=t_VECSMALL)
299 0 : pari_err_TYPE("safeel",x);
300 0 : check_array_index(l, lg(x));
301 0 : return &(x[l]);
302 : }
303 :
304 : GEN*
305 0 : safegcoeff(GEN x, long a, long b)
306 : {
307 0 : if (typ(x)!=t_MAT) pari_err_TYPE("safegcoeff", x);
308 0 : check_array_index(b, lg(x));
309 0 : check_array_index(a, lg(gel(x,b)));
310 0 : return &(gcoeff(x,a,b));
311 : }
312 :
313 : typedef struct matcomp
314 : {
315 : GEN *ptcell;
316 : GEN parent;
317 : int full_col, full_row;
318 : } matcomp;
319 :
320 : typedef struct gp_pointer
321 : {
322 : matcomp c;
323 : GEN x, ox;
324 : entree *ep;
325 : long vn;
326 : long sp;
327 : } gp_pointer;
328 :
329 : /* assign res at *pt in "simple array object" p and return it, or a copy.*/
330 : static void
331 9701769 : change_compo(matcomp *c, GEN res)
332 : {
333 9701769 : GEN p = c->parent, *pt = c->ptcell, po;
334 : long i, t;
335 :
336 9701769 : if (typ(p) == t_VECSMALL)
337 : {
338 35 : if (typ(res) != t_INT || is_bigint(res))
339 14 : pari_err_TYPE("t_VECSMALL assignment", res);
340 21 : *pt = (GEN)itos(res); return;
341 : }
342 9701734 : t = typ(res);
343 9701734 : if (c->full_row)
344 : {
345 204988 : if (t != t_VEC) pari_err_TYPE("matrix row assignment", res);
346 204967 : if (lg(res) != lg(p)) pari_err_DIM("matrix row assignment");
347 2105362 : for (i=1; i<lg(p); i++)
348 : {
349 1900416 : GEN p1 = gcoeff(p,c->full_row,i); /* Protect against SIGINT */
350 1900416 : gcoeff(p,c->full_row,i) = gclone(gel(res,i));
351 1900416 : if (isclone(p1)) gunclone_deep(p1);
352 : }
353 204946 : return;
354 : }
355 9496746 : if (c->full_col)
356 : {
357 355397 : if (t != t_COL) pari_err_TYPE("matrix col assignment", res);
358 355383 : if (lg(res) != lg(*pt)) pari_err_DIM("matrix col assignment");
359 : }
360 :
361 9496725 : po = *pt; /* Protect against SIGINT */
362 9496725 : *pt = gclone(res);
363 9496725 : gunclone_deep(po);
364 : }
365 :
366 : /***************************************************************************
367 : ** **
368 : ** Byte-code evaluator **
369 : ** **
370 : ***************************************************************************/
371 :
372 : struct var_lex
373 : {
374 : long flag;
375 : GEN value;
376 : };
377 :
378 : struct trace
379 : {
380 : long pc;
381 : GEN closure;
382 : };
383 :
384 : static THREAD long sp, rp, dbg_level;
385 : static THREAD long *st, *precs;
386 : static THREAD GEN *locks;
387 : static THREAD gp_pointer *ptrs;
388 : static THREAD entree **lvars;
389 : static THREAD struct var_lex *var;
390 : static THREAD struct trace *trace;
391 : static THREAD pari_stack s_st, s_ptrs, s_var, s_trace, s_prec;
392 : static THREAD pari_stack s_lvars, s_locks;
393 :
394 : static void
395 162363898 : changelex(long vn, GEN x)
396 : {
397 162363898 : struct var_lex *v=var+s_var.n+vn;
398 162363898 : GEN old_val = v->value;
399 162363898 : v->value = gclone(x);
400 162363898 : if (v->flag == COPY_VAL) gunclone_deep(old_val); else v->flag = COPY_VAL;
401 162363898 : }
402 :
403 : INLINE GEN
404 9791047 : copylex(long vn)
405 : {
406 9791047 : struct var_lex *v = var+s_var.n+vn;
407 9791047 : if (v->flag!=COPY_VAL && v->flag!=REF_VAL)
408 : {
409 52612 : v->value = gclone(v->value);
410 52612 : v->flag = COPY_VAL;
411 : }
412 9791047 : return v->value;
413 : }
414 :
415 : INLINE void
416 504 : setreflex(long vn)
417 : {
418 504 : struct var_lex *v = var+s_var.n+vn;
419 504 : v->flag = REF_VAL;
420 504 : }
421 :
422 : INLINE void
423 63505001 : pushlex(long vn, GEN x)
424 : {
425 63505001 : struct var_lex *v=var+s_var.n+vn;
426 63505001 : v->flag = PUSH_VAL;
427 63505001 : v->value = x;
428 63505001 : }
429 :
430 : INLINE void
431 186421312 : freelex(void)
432 : {
433 186421312 : struct var_lex *v=var+s_var.n-1;
434 186421312 : s_var.n--;
435 186421312 : if (v->flag == COPY_VAL) gunclone_deep(v->value);
436 186421312 : }
437 :
438 : INLINE void
439 306004734 : restore_vars(long nbmvar, long nblvar, long nblock)
440 : {
441 : long j;
442 486652616 : for(j=1; j<=nbmvar; j++) freelex();
443 306004927 : for(j=1; j<=nblvar; j++) { s_lvars.n--; pop_val(lvars[s_lvars.n]); }
444 306005340 : for(j=1; j<=nblock; j++) { s_locks.n--; gunclone_deep(locks[s_locks.n]); }
445 306004871 : }
446 :
447 : INLINE void
448 5662856 : restore_trace(long nbtrace)
449 : {
450 : long j;
451 11338462 : for(j=1; j<=nbtrace; j++)
452 : {
453 5675606 : GEN C = trace[s_trace.n-j].closure;
454 5675606 : clone_unlock(C);
455 : }
456 5662856 : s_trace.n -= nbtrace;
457 5662856 : }
458 :
459 : INLINE long
460 311495232 : trace_push(long pc, GEN C)
461 : {
462 : long tr;
463 311495232 : BLOCK_SIGINT_START
464 312290620 : tr = pari_stack_new(&s_trace);
465 312019917 : trace[tr].pc = pc;
466 312019917 : clone_lock(C);
467 311686020 : trace[tr].closure = C;
468 311686020 : BLOCK_SIGINT_END
469 312360507 : return tr;
470 : }
471 :
472 : void
473 5773846 : push_lex(GEN a, GEN C)
474 : {
475 5773846 : long vn=pari_stack_new(&s_var);
476 5773846 : struct var_lex *v=var+vn;
477 5773846 : v->flag = PUSH_VAL;
478 5773846 : v->value = a;
479 5773846 : if (C) (void) trace_push(-1, C);
480 5773847 : }
481 :
482 : GEN
483 90193369 : get_lex(long vn)
484 : {
485 90193369 : struct var_lex *v=var+s_var.n+vn;
486 90193369 : return v->value;
487 : }
488 :
489 : void
490 83678054 : set_lex(long vn, GEN x)
491 : {
492 83678054 : struct var_lex *v=var+s_var.n+vn;
493 83678054 : if (v->flag == COPY_VAL) { gunclone_deep(v->value); v->flag = PUSH_VAL; }
494 83678054 : v->value = x;
495 83678054 : }
496 :
497 : void
498 5606462 : pop_lex(long n)
499 : {
500 : long j;
501 11380003 : for(j=1; j<=n; j++)
502 5773541 : freelex();
503 5606462 : restore_trace(1);
504 5606462 : }
505 :
506 : static THREAD pari_stack s_relocs;
507 : static THREAD entree **relocs;
508 :
509 : void
510 374518 : pari_init_evaluator(void)
511 : {
512 374518 : sp=0;
513 374518 : pari_stack_init(&s_st,sizeof(*st),(void**)&st);
514 374501 : pari_stack_alloc(&s_st,32);
515 374569 : s_st.n=s_st.alloc;
516 374569 : rp=0;
517 374569 : pari_stack_init(&s_ptrs,sizeof(*ptrs),(void**)&ptrs);
518 374549 : pari_stack_alloc(&s_ptrs,16);
519 374578 : s_ptrs.n=s_ptrs.alloc;
520 374578 : pari_stack_init(&s_var,sizeof(*var),(void**)&var);
521 374567 : pari_stack_init(&s_lvars,sizeof(*lvars),(void**)&lvars);
522 374529 : pari_stack_init(&s_locks,sizeof(*locks),(void**)&locks);
523 374506 : pari_stack_init(&s_trace,sizeof(*trace),(void**)&trace);
524 374486 : br_res = NULL;
525 374486 : pari_stack_init(&s_relocs,sizeof(*relocs),(void**)&relocs);
526 374498 : pari_stack_init(&s_prec,sizeof(*precs),(void**)&precs);
527 374493 : }
528 : void
529 369205 : pari_close_evaluator(void)
530 : {
531 369205 : pari_stack_delete(&s_st);
532 373194 : pari_stack_delete(&s_ptrs);
533 373489 : pari_stack_delete(&s_var);
534 373865 : pari_stack_delete(&s_lvars);
535 373364 : pari_stack_delete(&s_trace);
536 373880 : pari_stack_delete(&s_relocs);
537 373440 : pari_stack_delete(&s_prec);
538 373526 : }
539 :
540 : static gp_pointer *
541 58686013 : new_ptr(void)
542 : {
543 58686013 : if (rp==s_ptrs.n-1)
544 : {
545 : long i;
546 0 : gp_pointer *old = ptrs;
547 0 : (void)pari_stack_new(&s_ptrs);
548 0 : if (old != ptrs)
549 0 : for(i=0; i<rp; i++)
550 : {
551 0 : gp_pointer *g = &ptrs[i];
552 0 : if(g->sp >= 0) gel(st,g->sp) = (GEN) &(g->x);
553 : }
554 : }
555 58686013 : return &ptrs[rp++];
556 : }
557 :
558 : void
559 482741 : push_localbitprec(long p)
560 : {
561 482741 : long n = pari_stack_new(&s_prec);
562 482890 : precs[n] = p;
563 482890 : }
564 : void
565 99053 : push_localprec(long p) { push_localbitprec(p); }
566 :
567 : void
568 99004 : pop_localprec(void) { s_prec.n--; }
569 :
570 : long
571 22782833 : get_localbitprec(void) { return s_prec.n? precs[s_prec.n-1]: precreal; }
572 :
573 : long
574 22380393 : get_localprec(void) { return nbits2prec(get_localbitprec()); }
575 :
576 : static void
577 11197 : checkprec(const char *f, long p, long M)
578 : {
579 11197 : if (p < 1) pari_err_DOMAIN(f, "p", "<", gen_1, stoi(p));
580 11183 : if (p > M) pari_err_DOMAIN(f, "p", ">", utoipos(M), utoi(p));
581 11171 : }
582 : static long
583 11290 : _prec(GEN p, const char *f)
584 : {
585 11290 : pari_sp av = avma;
586 11290 : if (typ(p) == t_INT) return itos(p);
587 35 : p = gceil(p);
588 35 : if (typ(p) != t_INT) pari_err_TYPE(f, p);
589 28 : return gc_long(av, itos(p));
590 : }
591 : void
592 7847 : localprec(GEN pp)
593 : {
594 7847 : long p = _prec(pp, "localprec");
595 7839 : checkprec("localprec", p, prec2ndec(LGBITS));
596 7826 : p = ndec2nbits(p); push_localbitprec(p);
597 7826 : }
598 : void
599 3359 : localbitprec(GEN pp)
600 : {
601 3359 : long p = _prec(pp, "localbitprec");
602 3358 : checkprec("localbitprec", p, (long)LGBITS);
603 3345 : push_localbitprec(p);
604 3345 : }
605 : long
606 14 : getlocalprec(long prec) { return prec2ndec(prec); }
607 : long
608 3416 : getlocalbitprec(long bit) { return bit; }
609 :
610 : static GEN
611 1246 : _precision0(GEN x)
612 : {
613 1246 : long a = gprecision(x);
614 1246 : return a? utoi(prec2ndec(a)): mkoo();
615 : }
616 : GEN
617 42 : precision0(GEN x, long n)
618 42 : { return n? gprec(x,n): _precision0(x); }
619 : static GEN
620 676 : _bitprecision0(GEN x)
621 : {
622 676 : long a = gprecision(x);
623 676 : return a? utoi(a): mkoo();
624 : }
625 : GEN
626 42 : bitprecision0(GEN x, long n)
627 : {
628 42 : if (n < 0)
629 0 : pari_err_DOMAIN("bitprecision", "bitprecision", "<", gen_0, stoi(n));
630 42 : if (n) {
631 42 : pari_sp av = avma;
632 42 : GEN y = gprec_w(x, nbits2prec(n));
633 42 : return gc_GEN(av, y);
634 : }
635 0 : return _bitprecision0(x);
636 : }
637 : GEN
638 1288 : precision00(GEN x, GEN n)
639 : {
640 1288 : if (!n) return _precision0(x);
641 42 : return precision0(x, _prec(n, "precision"));
642 : }
643 : GEN
644 718 : bitprecision00(GEN x, GEN n)
645 : {
646 718 : if (!n) return _bitprecision0(x);
647 42 : return bitprecision0(x, _prec(n, "bitprecision"));
648 : }
649 :
650 : INLINE GEN
651 75467384 : copyupto(GEN z, GEN t)
652 : {
653 75467384 : if (is_universal_constant(z) || (z>(GEN)pari_mainstack->bot && z<=t))
654 71083225 : return z;
655 : else
656 4383974 : return gcopy(z);
657 : }
658 :
659 : static void closure_eval(GEN C);
660 :
661 : INLINE GEN
662 41666 : get_and_reset_break(void)
663 : {
664 41666 : GEN z = br_res? gcopy(br_res): gnil;
665 41666 : reset_break(); return z;
666 : }
667 :
668 : INLINE GEN
669 51368597 : closure_return(GEN C)
670 : {
671 51368597 : pari_sp av = avma;
672 51368597 : closure_eval(C);
673 51342039 : if (br_status) { set_avma(av); return get_and_reset_break(); }
674 51300422 : return gc_upto(av, gel(st,--sp));
675 : }
676 :
677 : /* for the break_loop debugger. Not memory clean */
678 : GEN
679 175 : closure_evalbrk(GEN C, long *status)
680 : {
681 175 : closure_eval(C); *status = br_status;
682 140 : return br_status? get_and_reset_break(): gel(st,--sp);
683 : }
684 :
685 : INLINE long
686 1154759 : closure_varn(GEN x)
687 : {
688 1154759 : if (!x) return -1;
689 1154157 : if (!gequalX(x)) err_var(x);
690 1154157 : return varn(x);
691 : }
692 :
693 : INLINE void
694 93838440 : closure_castgen(GEN z, long mode)
695 : {
696 93838440 : switch (mode)
697 : {
698 93837551 : case Ggen:
699 93837551 : gel(st,sp++)=z;
700 93837551 : break;
701 889 : case Gsmall:
702 889 : st[sp++]=gtos(z);
703 889 : break;
704 0 : case Gusmall:
705 0 : st[sp++]=gtou(z);
706 0 : break;
707 0 : case Gvar:
708 0 : st[sp++]=closure_varn(z);
709 0 : break;
710 0 : case Gvoid:
711 0 : break;
712 0 : default:
713 0 : pari_err_BUG("closure_castgen, type unknown");
714 : }
715 93838440 : }
716 :
717 : INLINE void
718 5873 : closure_castlong(long z, long mode)
719 : {
720 5873 : switch (mode)
721 : {
722 0 : case Gsmall:
723 0 : st[sp++]=z;
724 0 : break;
725 0 : case Gusmall:
726 0 : if (z < 0)
727 0 : pari_err_TYPE("stou [integer >=0 expected]", stoi(z));
728 0 : st[sp++]=(ulong) z;
729 0 : break;
730 5866 : case Ggen:
731 5866 : gel(st,sp++)=stoi(z);
732 5866 : break;
733 0 : case Gvar:
734 0 : err_var(stoi(z));
735 7 : case Gvoid:
736 7 : break;
737 0 : default:
738 0 : pari_err_BUG("closure_castlong, type unknown");
739 : }
740 5873 : }
741 :
742 : const char *
743 13876 : closure_func_err(void)
744 : {
745 13876 : long fun=s_trace.n-1, pc;
746 : const char *code;
747 : GEN C, oper;
748 13876 : if (fun < 0 || trace[fun].pc < 0) return NULL;
749 13166 : pc = trace[fun].pc; C = trace[fun].closure;
750 13166 : code = closure_codestr(C); oper = closure_get_oper(C);
751 13166 : if (code[pc]==OCcallgen || code[pc]==OCcallgen2 ||
752 3675 : code[pc]==OCcallint || code[pc]==OCcalllong || code[pc]==OCcallvoid)
753 10212 : return ((entree*)oper[pc])->name;
754 2954 : return NULL;
755 : }
756 :
757 : /* return the next label for the call chain debugger closure_err(),
758 : * incorporating the name of the user of member function. Return NULL for an
759 : * anonymous (inline) closure. */
760 : static char *
761 245 : get_next_label(const char *s, int member, char **next_fun)
762 : {
763 245 : const char *v, *t = s+1;
764 : char *u, *next_label;
765 :
766 245 : if (!is_keyword_char(*s)) return NULL;
767 1036 : while (is_keyword_char(*t)) t++;
768 : /* e.g. (x->1/x)(0) instead of (x)->1/x */
769 224 : if (t[0] == '-' && t[1] == '>') return NULL;
770 217 : next_label = (char*)pari_malloc(t - s + 32);
771 217 : sprintf(next_label, "in %sfunction ", member? "member ": "");
772 217 : u = *next_fun = next_label + strlen(next_label);
773 217 : v = s;
774 1246 : while (v < t) *u++ = *v++;
775 217 : *u++ = 0; return next_label;
776 : }
777 :
778 : static const char *
779 21 : get_arg_name(GEN C, long i)
780 : {
781 21 : GEN d = closure_get_dbg(C), frpc = gel(d,2), fram = gel(d,3);
782 21 : long j, l = lg(frpc);
783 28 : for (j=1; j<l; j++)
784 28 : if (frpc[j]==1 && i<lg(gel(fram,j)))
785 21 : return ((entree*)mael(fram,j,i))->name;
786 0 : return "(unnamed)";
787 : }
788 :
789 : void
790 12990 : closure_err(long level)
791 : {
792 : GEN base;
793 12990 : const long lastfun = s_trace.n - 1 - level;
794 : char *next_label, *next_fun;
795 12990 : long i = maxss(0, lastfun - 19);
796 12990 : if (lastfun < 0) return; /*e.g. when called by gp_main_loop's simplify */
797 12969 : if (i > 0) while (lg(trace[i].closure)==6) i--;
798 12969 : if (lg(trace[i].closure)==6) return; /* Should never happen */
799 12969 : base = closure_get_text(trace[i].closure); /* gcc -Wall*/
800 12969 : next_label = pari_strdup(i == 0? "at top-level": "[...] at");
801 12969 : next_fun = next_label;
802 13647 : for (; i <= lastfun; i++)
803 : {
804 13647 : GEN C = trace[i].closure;
805 13647 : if (lg(C) >= 7) base=closure_get_text(C);
806 13647 : if ((i==lastfun || lg(trace[i+1].closure)>=7))
807 : {
808 13214 : GEN dbg = gel(closure_get_dbg(C),1);
809 : /* After a SIGINT, pc can be slightly off: ensure 0 <= pc < lg() */
810 13214 : long pc = minss(lg(dbg)-1, trace[i].pc>=0 ? trace[i].pc: 1);
811 13214 : long offset = pc? dbg[pc]: 0;
812 : int member;
813 : const char *s, *sbase;
814 13214 : if (typ(base)!=t_VEC) sbase = GSTR(base);
815 189 : else if (offset>=0) sbase = GSTR(gel(base,2));
816 21 : else { sbase = GSTR(gel(base,1)); offset += strlen(sbase); }
817 13214 : s = sbase + offset;
818 13214 : member = offset>0 && (s[-1] == '.');
819 : /* avoid "in function foo: foo" */
820 13214 : if (!next_fun || strcmp(next_fun, s)) {
821 13207 : print_errcontext(pariErr, next_label, s, sbase);
822 13207 : out_putc(pariErr, '\n');
823 : }
824 13214 : pari_free(next_label);
825 13214 : if (i == lastfun) break;
826 :
827 245 : next_label = get_next_label(s, member, &next_fun);
828 245 : if (!next_label) {
829 28 : next_label = pari_strdup("in anonymous function");
830 28 : next_fun = NULL;
831 : }
832 : }
833 : }
834 : }
835 :
836 : GEN
837 41 : pari_self(void)
838 : {
839 41 : long fun = s_trace.n - 1;
840 76 : if (fun > 0) while (lg(trace[fun].closure)==6) fun--;
841 41 : return fun >= 0 ? trace[fun].closure: NULL;
842 : }
843 :
844 : long
845 91 : closure_context(long start, long level)
846 : {
847 91 : const long lastfun = s_trace.n - 1 - level;
848 91 : long i, fun = lastfun;
849 91 : if (fun<0) return lastfun;
850 224 : while (fun>start && lg(trace[fun].closure)==6) fun--;
851 315 : for (i=fun; i <= lastfun; i++)
852 224 : push_frame(trace[i].closure, trace[i].pc,0);
853 126 : for ( ; i < s_trace.n; i++)
854 35 : push_frame(trace[i].closure, trace[i].pc,1);
855 91 : return s_trace.n-level;
856 : }
857 :
858 : INLINE void
859 3102476123 : st_alloc(long n)
860 : {
861 3102476123 : if (sp+n>s_st.n)
862 : {
863 70 : pari_stack_alloc(&s_st,n+16);
864 70 : s_st.n=s_st.alloc;
865 70 : if (DEBUGMEM>=2) pari_warn(warner,"doubling evaluator stack");
866 : }
867 3102476123 : }
868 :
869 : INLINE void
870 9906673 : ptr_proplock(gp_pointer *g, GEN C)
871 : {
872 9906673 : g->x = C;
873 9906673 : if (isclone(g->x))
874 : {
875 445046 : clone_unlock_deep(g->ox);
876 445046 : g->ox = g->x;
877 445046 : ++bl_refc(g->ox);
878 : }
879 9906673 : }
880 :
881 : static void
882 305899000 : closure_eval(GEN C)
883 : {
884 305899000 : const char *code=closure_codestr(C);
885 305897410 : GEN oper=closure_get_oper(C);
886 305887537 : GEN data=closure_get_data(C);
887 305885975 : long loper=lg(oper);
888 305885975 : long saved_sp=sp-closure_arity(C);
889 305881063 : long saved_rp=rp, saved_prec=s_prec.n;
890 305881063 : long j, nbmvar=0, nblvar=0, nblock=0;
891 : long pc, t;
892 : #ifdef STACK_CHECK
893 : GEN stackelt;
894 305881063 : if (PARI_stack_limit && (void*) &stackelt <= PARI_stack_limit)
895 0 : pari_err(e_MISC, "deep recursion");
896 : #endif
897 305881063 : t = trace_push(0, C);
898 306586900 : if (lg(C)==8)
899 : {
900 14669609 : GEN z=closure_get_frame(C);
901 14669346 : long l=lg(z)-1;
902 14669346 : pari_stack_alloc(&s_var,l);
903 14620261 : s_var.n+=l;
904 14620261 : nbmvar+=l;
905 51555638 : for(j=1;j<=l;j++)
906 : {
907 36935377 : var[s_var.n-j].flag=PUSH_VAL;
908 36935377 : var[s_var.n-j].value=gel(z,j);
909 : }
910 : }
911 :
912 3336454107 : for(pc=1;pc<loper;pc++)
913 : {
914 3030470651 : op_code opcode=(op_code) code[pc];
915 3030470651 : long operand=oper[pc];
916 3030470651 : if (sp<0) pari_err_BUG("closure_eval, stack underflow");
917 3030470651 : st_alloc(16);
918 3030315425 : trace[t].pc = pc;
919 : CHECK_CTRLC
920 3030315425 : switch(opcode)
921 : {
922 208387740 : case OCpushlong:
923 208387740 : st[sp++]=operand;
924 208387740 : break;
925 99989 : case OCpushgnil:
926 99989 : gel(st,sp++)=gnil;
927 99989 : break;
928 165717305 : case OCpushgen:
929 165717305 : gel(st,sp++)=gel(data,operand);
930 165717305 : break;
931 86696 : case OCpushreal:
932 86696 : gel(st,sp++)=strtor(GSTR(data[operand]),get_localprec());
933 86696 : break;
934 276074417 : case OCpushstoi:
935 276074417 : gel(st,sp++)=stoi(operand);
936 276074376 : break;
937 25977 : case OCpushvar:
938 : {
939 25977 : entree *ep = (entree *)operand;
940 25977 : gel(st,sp++)=pol_x(pari_var_create(ep));
941 25977 : break;
942 : }
943 93567613 : case OCpushdyn:
944 : {
945 93567613 : entree *ep = (entree *)operand;
946 93567613 : if (!mt_is_thread())
947 : {
948 93566114 : checkvalue(ep, chk_CREATE);
949 93566114 : gel(st,sp++)=(GEN)ep->value;
950 : } else
951 : {
952 1499 : GEN val = export_get(ep->name);
953 1499 : if (!val)
954 0 : pari_err(e_MISC,"mt: please use export(%s)", ep->name);
955 1499 : gel(st,sp++)=val;
956 : }
957 93567613 : break;
958 : }
959 634390160 : case OCpushlex:
960 634390160 : gel(st,sp++)=var[s_var.n+operand].value;
961 634390160 : break;
962 23308355 : case OCsimpleptrdyn:
963 : {
964 23308355 : gp_pointer *g = new_ptr();
965 23308355 : g->vn=0;
966 23308355 : g->ep = (entree*) operand;
967 23308355 : g->x = checkvalueptr(g->ep);
968 23308355 : g->ox = g->x; clone_lock(g->ox);
969 23308355 : g->sp = sp;
970 23308355 : gel(st,sp++) = (GEN)&(g->x);
971 23308355 : break;
972 : }
973 25675840 : case OCsimpleptrlex:
974 : {
975 25675840 : gp_pointer *g = new_ptr();
976 25675840 : g->vn=operand;
977 25675840 : g->ep=(entree *)0x1L;
978 25675840 : g->x = (GEN) var[s_var.n+operand].value;
979 25675840 : g->ox = g->x; clone_lock(g->ox);
980 25675840 : g->sp = sp;
981 25675840 : gel(st,sp++) = (GEN)&(g->x);
982 25675840 : break;
983 : }
984 5033 : case OCnewptrdyn:
985 : {
986 5033 : entree *ep = (entree *)operand;
987 5033 : gp_pointer *g = new_ptr();
988 : matcomp *C;
989 5033 : checkvalue(ep, chk_ERROR);
990 5033 : g->sp = -1;
991 5033 : g->x = copyvalue(ep);
992 5033 : g->ox = g->x; clone_lock(g->ox);
993 5033 : g->vn=0;
994 5033 : g->ep=NULL;
995 5033 : C=&g->c;
996 5033 : C->full_col = C->full_row = 0;
997 5033 : C->parent = (GEN) g->x;
998 5033 : C->ptcell = (GEN *) &g->x;
999 5033 : break;
1000 : }
1001 9696785 : case OCnewptrlex:
1002 : {
1003 9696785 : gp_pointer *g = new_ptr();
1004 : matcomp *C;
1005 9696785 : g->sp = -1;
1006 9696785 : g->x = copylex(operand);
1007 9696785 : g->ox = g->x; clone_lock(g->ox);
1008 9696785 : g->vn=0;
1009 9696785 : g->ep=NULL;
1010 9696785 : C=&g->c;
1011 9696785 : C->full_col = C->full_row = 0;
1012 9696785 : C->parent = (GEN) g->x;
1013 9696785 : C->ptcell = (GEN *) &(g->x);
1014 9696785 : break;
1015 : }
1016 559727 : case OCpushptr:
1017 : {
1018 559727 : gp_pointer *g = &ptrs[rp-1];
1019 559727 : g->sp = sp;
1020 559727 : gel(st,sp++) = (GEN)&(g->x);
1021 : }
1022 559727 : break;
1023 49543866 : case OCendptr:
1024 99087732 : for(j=0;j<operand;j++)
1025 : {
1026 49543866 : gp_pointer *g = &ptrs[--rp];
1027 49543866 : if (g->ep)
1028 : {
1029 48984139 : if (g->vn)
1030 25675840 : changelex(g->vn, g->x);
1031 : else
1032 23308299 : changevalue(g->ep, g->x);
1033 : }
1034 559727 : else change_compo(&(g->c), g->x);
1035 49543866 : clone_unlock_deep(g->ox);
1036 : }
1037 49543866 : break;
1038 6182040 : case OCstoredyn:
1039 : {
1040 6182040 : entree *ep = (entree *)operand;
1041 6182040 : checkvalue(ep, chk_NOCREATE);
1042 6182031 : changevalue(ep, gel(st,--sp));
1043 6182031 : break;
1044 : }
1045 136688058 : case OCstorelex:
1046 136688058 : changelex(operand,gel(st,--sp));
1047 136688058 : break;
1048 9142042 : case OCstoreptr:
1049 : {
1050 9142042 : gp_pointer *g = &ptrs[--rp];
1051 9142042 : change_compo(&(g->c), gel(st,--sp));
1052 9141965 : clone_unlock_deep(g->ox);
1053 9141965 : break;
1054 : }
1055 58444671 : case OCstackgen:
1056 : {
1057 58444671 : GEN z = gc_upto(st[sp-2],gel(st,sp-1));
1058 58444669 : gmael(st,sp-3,operand) = copyupto(z,gel(st,sp-2));
1059 58444673 : st[sp-2] = avma;
1060 58444673 : sp--;
1061 58444673 : break;
1062 : }
1063 22293692 : case OCprecreal:
1064 22293692 : st[sp++]=get_localprec();
1065 22293686 : break;
1066 29624 : case OCbitprecreal:
1067 29624 : st[sp++]=get_localbitprec();
1068 29624 : break;
1069 952 : case OCprecdl:
1070 952 : st[sp++]=precdl;
1071 952 : break;
1072 2982 : case OCavma:
1073 2982 : st[sp++]=avma;
1074 2982 : break;
1075 741818 : case OCcowvardyn:
1076 : {
1077 741818 : entree *ep = (entree *)operand;
1078 741818 : checkvalue(ep, chk_ERROR);
1079 741818 : (void)copyvalue(ep);
1080 741818 : break;
1081 : }
1082 93205 : case OCcowvarlex:
1083 93205 : (void)copylex(operand);
1084 93205 : break;
1085 504 : case OCsetref:
1086 504 : setreflex(operand);
1087 504 : break;
1088 483 : case OClock:
1089 : {
1090 483 : GEN v = gel(st,sp-1);
1091 483 : if (isclone(v))
1092 : {
1093 469 : long n = pari_stack_new(&s_locks);
1094 469 : locks[n] = v;
1095 469 : nblock++;
1096 469 : ++bl_refc(v);
1097 : }
1098 483 : break;
1099 : }
1100 0 : case OCevalmnem:
1101 : {
1102 0 : entree *ep = (entree*) operand;
1103 0 : const char *flags = ep->code;
1104 0 : flags = strchr(flags, '\n'); /* Skip to the following '\n' */
1105 0 : st[sp-1] = eval_mnemonic(gel(st,sp-1), flags+1);
1106 0 : break;
1107 : }
1108 20472440 : case OCstoi:
1109 20472440 : gel(st,sp-1)=stoi(st[sp-1]);
1110 20472258 : break;
1111 0 : case OCutoi:
1112 0 : gel(st,sp-1)=utoi(st[sp-1]);
1113 0 : break;
1114 72975899 : case OCitos:
1115 72975899 : st[sp+operand]=gtos(gel(st,sp+operand));
1116 72975866 : break;
1117 100663 : case OCitou:
1118 100663 : st[sp+operand]=gtou(gel(st,sp+operand));
1119 100664 : break;
1120 5428 : case OCtostr:
1121 : {
1122 5428 : GEN z = gel(st,sp+operand);
1123 5428 : st[sp+operand] = (long) (z ? GENtostr_unquoted(z): NULL);
1124 5428 : break;
1125 : }
1126 1154759 : case OCvarn:
1127 1154759 : st[sp+operand] = closure_varn(gel(st,sp+operand));
1128 1154759 : break;
1129 26261729 : case OCcopy:
1130 26261729 : gel(st,sp-1) = gcopy(gel(st,sp-1));
1131 26261750 : break;
1132 2982 : case OCgc:
1133 : {
1134 : pari_sp av;
1135 : GEN x;
1136 2982 : sp--;
1137 2982 : av = st[sp-1];
1138 2982 : x = gel(st,sp);
1139 2982 : if (isonstack(x))
1140 : {
1141 2982 : pari_sp av2 = (pari_sp)(x + lg(x));
1142 2982 : if ((long) (av - av2) > 1000000L)
1143 : {
1144 7 : if (DEBUGMEM>=2)
1145 0 : pari_warn(warnmem,"eval: recovering %ld bytes", av - av2);
1146 7 : x = gc_upto(av, x);
1147 : }
1148 0 : } else set_avma(av);
1149 2982 : gel(st,sp-1) = x;
1150 2982 : break;
1151 : }
1152 0 : case OCcopyifclone:
1153 0 : if (isclone(gel(st,sp-1)))
1154 0 : gel(st,sp-1) = gcopy(gel(st,sp-1));
1155 0 : break;
1156 92147531 : case OCcompo1:
1157 : {
1158 92147531 : GEN p=gel(st,sp-2);
1159 92147531 : long c=st[sp-1];
1160 92147531 : sp-=2;
1161 92147531 : switch(typ(p))
1162 : {
1163 92140660 : case t_VEC: case t_COL:
1164 92140660 : check_array_index(c, lg(p));
1165 92140658 : closure_castgen(gel(p,c),operand);
1166 92140660 : break;
1167 977 : case t_LIST:
1168 : {
1169 : long lx;
1170 977 : if (list_typ(p)!=t_LIST_RAW)
1171 0 : pari_err_TYPE("_[_] OCcompo1 [not a vector]", p);
1172 977 : p = list_data(p); lx = p? lg(p): 1;
1173 977 : check_array_index(c, lx);
1174 977 : closure_castgen(gel(p,c),operand);
1175 977 : break;
1176 : }
1177 5887 : case t_VECSMALL:
1178 5887 : check_array_index(c,lg(p));
1179 5873 : closure_castlong(p[c],operand);
1180 5873 : break;
1181 7 : default:
1182 7 : pari_err_TYPE("_[_] OCcompo1 [not a vector]", p);
1183 0 : break;
1184 : }
1185 92147510 : break;
1186 : }
1187 9425325 : case OCcompo1ptr:
1188 : {
1189 9425325 : long c=st[sp-1];
1190 : long lx;
1191 9425325 : gp_pointer *g = &ptrs[rp-1];
1192 9425325 : matcomp *C=&g->c;
1193 9425325 : GEN p = g->x;
1194 9425325 : sp--;
1195 9425325 : switch(typ(p))
1196 : {
1197 9425248 : case t_VEC: case t_COL:
1198 9425248 : check_array_index(c, lg(p));
1199 9425248 : C->ptcell = (GEN *) p+c;
1200 9425248 : ptr_proplock(g, *(C->ptcell));
1201 9425248 : break;
1202 42 : case t_VECSMALL:
1203 42 : check_array_index(c, lg(p));
1204 35 : C->ptcell = (GEN *) p+c;
1205 35 : g->x = stoi(p[c]);
1206 35 : break;
1207 28 : case t_LIST:
1208 28 : if (list_typ(p)!=t_LIST_RAW)
1209 0 : pari_err_TYPE("&_[_] OCcompo1 [not a vector]", p);
1210 28 : p = list_data(p); lx = p? lg(p): 1;
1211 28 : check_array_index(c,lx);
1212 28 : C->ptcell = (GEN *) p+c;
1213 28 : ptr_proplock(g, *(C->ptcell));
1214 28 : break;
1215 7 : default:
1216 7 : pari_err_TYPE("&_[_] OCcompo1ptr [not a vector]", p);
1217 : }
1218 9425311 : C->parent = p;
1219 9425311 : break;
1220 : }
1221 1696810 : case OCcompo2:
1222 : {
1223 1696810 : GEN p=gel(st,sp-3);
1224 1696810 : long c=st[sp-2];
1225 1696810 : long d=st[sp-1];
1226 1696810 : if (typ(p)!=t_MAT) pari_err_TYPE("_[_,_] OCcompo2 [not a matrix]", p);
1227 1696803 : check_array_index(d, lg(p));
1228 1696803 : check_array_index(c, lg(gel(p,d)));
1229 1696803 : sp-=3;
1230 1696803 : closure_castgen(gcoeff(p,c,d),operand);
1231 1696803 : break;
1232 : }
1233 126000 : case OCcompo2ptr:
1234 : {
1235 126000 : long c=st[sp-2];
1236 126000 : long d=st[sp-1];
1237 126000 : gp_pointer *g = &ptrs[rp-1];
1238 126000 : matcomp *C=&g->c;
1239 126000 : GEN p = g->x;
1240 126000 : sp-=2;
1241 126000 : if (typ(p)!=t_MAT)
1242 0 : pari_err_TYPE("&_[_,_] OCcompo2ptr [not a matrix]", p);
1243 126000 : check_array_index(d, lg(p));
1244 126000 : check_array_index(c, lg(gel(p,d)));
1245 126000 : C->ptcell = (GEN *) gel(p,d)+c;
1246 126000 : C->parent = p;
1247 126000 : ptr_proplock(g, *(C->ptcell));
1248 126000 : break;
1249 : }
1250 1022649 : case OCcompoC:
1251 : {
1252 1022649 : GEN p=gel(st,sp-2);
1253 1022649 : long c=st[sp-1];
1254 1022649 : if (typ(p)!=t_MAT)
1255 7 : pari_err_TYPE("_[,_] OCcompoC [not a matrix]", p);
1256 1022642 : check_array_index(c, lg(p));
1257 1022635 : sp--;
1258 1022635 : gel(st,sp-1) = gel(p,c);
1259 1022635 : break;
1260 : }
1261 355411 : case OCcompoCptr:
1262 : {
1263 355411 : long c=st[sp-1];
1264 355411 : gp_pointer *g = &ptrs[rp-1];
1265 355411 : matcomp *C=&g->c;
1266 355411 : GEN p = g->x;
1267 355411 : sp--;
1268 355411 : if (typ(p)!=t_MAT)
1269 7 : pari_err_TYPE("&_[,_] OCcompoCptr [not a matrix]", p);
1270 355404 : check_array_index(c, lg(p));
1271 355397 : C->ptcell = (GEN *) p+c;
1272 355397 : C->full_col = c;
1273 355397 : C->parent = p;
1274 355397 : ptr_proplock(g, *(C->ptcell));
1275 355397 : break;
1276 : }
1277 273028 : case OCcompoL:
1278 : {
1279 273028 : GEN p=gel(st,sp-2);
1280 273028 : long r=st[sp-1];
1281 273028 : sp--;
1282 273028 : if (typ(p)!=t_MAT)
1283 7 : pari_err_TYPE("_[_,] OCcompoL [not a matrix]", p);
1284 273021 : check_array_index(r,lg(p) == 1? 1: lgcols(p));
1285 273014 : gel(st,sp-1) = row(p,r);
1286 273014 : break;
1287 : }
1288 205002 : case OCcompoLptr:
1289 : {
1290 205002 : long r=st[sp-1];
1291 205002 : gp_pointer *g = &ptrs[rp-1];
1292 205002 : matcomp *C=&g->c;
1293 205002 : GEN p = g->x, p2;
1294 205002 : sp--;
1295 205002 : if (typ(p)!=t_MAT)
1296 7 : pari_err_TYPE("&_[_,] OCcompoLptr [not a matrix]", p);
1297 204995 : check_array_index(r,lg(p) == 1? 1: lgcols(p));
1298 204988 : p2 = rowcopy(p,r);
1299 204988 : C->full_row = r; /* record row number */
1300 204988 : C->ptcell = &p2;
1301 204988 : C->parent = p;
1302 204988 : g->x = p2;
1303 204988 : break;
1304 : }
1305 102942 : case OCdefaultarg:
1306 102942 : if (var[s_var.n+operand].flag==DEFAULT_VAL)
1307 : {
1308 3465 : GEN z = gel(st,sp-1);
1309 3465 : if (typ(z)==t_CLOSURE)
1310 : {
1311 1057 : pushlex(operand, closure_evalnobrk(z));
1312 1057 : copylex(operand);
1313 : }
1314 : else
1315 2408 : pushlex(operand, z);
1316 : }
1317 102942 : sp--;
1318 102942 : break;
1319 51 : case OClocalvar:
1320 : {
1321 : long n;
1322 51 : entree *ep = (entree *)operand;
1323 51 : checkvalue(ep, chk_NOCREATE);
1324 42 : n = pari_stack_new(&s_lvars);
1325 42 : lvars[n] = ep;
1326 42 : nblvar++;
1327 42 : pushvalue(ep,gel(st,--sp));
1328 42 : break;
1329 : }
1330 23 : case OClocalvar0:
1331 : {
1332 : long n;
1333 23 : entree *ep = (entree *)operand;
1334 23 : checkvalue(ep, chk_NOCREATE);
1335 14 : n = pari_stack_new(&s_lvars);
1336 14 : lvars[n] = ep;
1337 14 : nblvar++;
1338 14 : zerovalue(ep);
1339 12 : break;
1340 : }
1341 41 : case OCexportvar:
1342 : {
1343 41 : entree *ep = (entree *)operand;
1344 41 : mt_export_add(ep->name, gel(st,--sp));
1345 41 : break;
1346 : }
1347 6 : case OCunexportvar:
1348 : {
1349 6 : entree *ep = (entree *)operand;
1350 6 : mt_export_del(ep->name);
1351 6 : break;
1352 : }
1353 :
1354 : #define EVAL_f(f, type, resEQ) \
1355 : switch (ep->arity) \
1356 : { \
1357 : case 0: resEQ ((type (*)(void))f)(); break; \
1358 : case 1: sp--; resEQ ((type (*)(long))f)(st[sp]); break; \
1359 : case 2: sp-=2; resEQ((type (*)(long,long))f)(st[sp],st[sp+1]); break; \
1360 : case 3: sp-=3; resEQ((type (*)(long,long,long))f)(st[sp],st[sp+1],st[sp+2]); break; \
1361 : case 4: sp-=4; resEQ((type (*)(long,long,long,long))f)(st[sp],st[sp+1],st[sp+2],st[sp+3]); break; \
1362 : case 5: sp-=5; resEQ((type (*)(long,long,long,long,long))f)(st[sp],st[sp+1],st[sp+2],st[sp+3],st[sp+4]); break; \
1363 : case 6: sp-=6; resEQ((type (*)(long,long,long,long,long,long))f)(st[sp],st[sp+1],st[sp+2],st[sp+3],st[sp+4],st[sp+5]); break; \
1364 : case 7: sp-=7; resEQ((type (*)(long,long,long,long,long,long,long))f)(st[sp],st[sp+1],st[sp+2],st[sp+3],st[sp+4],st[sp+5],st[sp+6]); break; \
1365 : case 8: sp-=8; resEQ((type (*)(long,long,long,long,long,long,long,long))f)(st[sp],st[sp+1],st[sp+2],st[sp+3],st[sp+4],st[sp+5],st[sp+6],st[sp+7]); break; \
1366 : case 9: sp-=9; resEQ((type (*)(long,long,long,long,long,long,long,long,long))f)(st[sp],st[sp+1],st[sp+2],st[sp+3],st[sp+4],st[sp+5],st[sp+6],st[sp+7],st[sp+8]); break; \
1367 : case 10: sp-=10; resEQ((type (*)(long,long,long,long,long,long,long,long,long,long))f)(st[sp],st[sp+1],st[sp+2],st[sp+3],st[sp+4],st[sp+5],st[sp+6],st[sp+7],st[sp+8],st[sp+9]); break; \
1368 : case 11: sp-=11; resEQ((type (*)(long,long,long,long,long,long,long,long,long,long,long))f)(st[sp],st[sp+1],st[sp+2],st[sp+3],st[sp+4],st[sp+5],st[sp+6],st[sp+7],st[sp+8],st[sp+9],st[sp+10]); break; \
1369 : case 12: sp-=12; resEQ((type (*)(long,long,long,long,long,long,long,long,long,long,long,long))f)(st[sp],st[sp+1],st[sp+2],st[sp+3],st[sp+4],st[sp+5],st[sp+6],st[sp+7],st[sp+8],st[sp+9],st[sp+10],st[sp+11]); break; \
1370 : case 13: sp-=13; resEQ((type (*)(long,long,long,long,long,long,long,long,long,long,long,long,long))f)(st[sp],st[sp+1],st[sp+2],st[sp+3],st[sp+4],st[sp+5],st[sp+6],st[sp+7],st[sp+8],st[sp+9],st[sp+10],st[sp+11],st[sp+12]); break; \
1371 : case 14: sp-=14; resEQ((type (*)(long,long,long,long,long,long,long,long,long,long,long,long,long,long))f)(st[sp],st[sp+1],st[sp+2],st[sp+3],st[sp+4],st[sp+5],st[sp+6],st[sp+7],st[sp+8],st[sp+9],st[sp+10],st[sp+11],st[sp+12],st[sp+13]); break; \
1372 : case 15: sp-=15; resEQ((type (*)(long,long,long,long,long,long,long,long,long,long,long,long,long,long,long))f)(st[sp],st[sp+1],st[sp+2],st[sp+3],st[sp+4],st[sp+5],st[sp+6],st[sp+7],st[sp+8],st[sp+9],st[sp+10],st[sp+11],st[sp+12],st[sp+13],st[sp+14]); break; \
1373 : case 16: sp-=16; resEQ((type (*)(long,long,long,long,long,long,long,long,long,long,long,long,long,long,long,long))f)(st[sp],st[sp+1],st[sp+2],st[sp+3],st[sp+4],st[sp+5],st[sp+6],st[sp+7],st[sp+8],st[sp+9],st[sp+10],st[sp+11],st[sp+12],st[sp+13],st[sp+14],st[sp+15]); break; \
1374 : case 17: sp-=17; resEQ((type (*)(long,long,long,long,long,long,long,long,long,long,long,long,long,long,long,long,long))f)(st[sp],st[sp+1],st[sp+2],st[sp+3],st[sp+4],st[sp+5],st[sp+6],st[sp+7],st[sp+8],st[sp+9],st[sp+10],st[sp+11],st[sp+12],st[sp+13],st[sp+14],st[sp+15],st[sp+16]); break; \
1375 : case 18: sp-=18; resEQ((type (*)(long,long,long,long,long,long,long,long,long,long,long,long,long,long,long,long,long,long))f)(st[sp],st[sp+1],st[sp+2],st[sp+3],st[sp+4],st[sp+5],st[sp+6],st[sp+7],st[sp+8],st[sp+9],st[sp+10],st[sp+11],st[sp+12],st[sp+13],st[sp+14],st[sp+15],st[sp+16],st[sp+17]); break; \
1376 : case 19: sp-=19; resEQ((type (*)(long,long,long,long,long,long,long,long,long,long,long,long,long,long,long,long,long,long,long))f)(st[sp],st[sp+1],st[sp+2],st[sp+3],st[sp+4],st[sp+5],st[sp+6],st[sp+7],st[sp+8],st[sp+9],st[sp+10],st[sp+11],st[sp+12],st[sp+13],st[sp+14],st[sp+15],st[sp+16],st[sp+17],st[sp+18]); break; \
1377 : case 20: sp-=20; resEQ((type (*)(long,long,long,long,long,long,long,long,long,long,long,long,long,long,long,long,long,long,long,long))f)(st[sp],st[sp+1],st[sp+2],st[sp+3],st[sp+4],st[sp+5],st[sp+6],st[sp+7],st[sp+8],st[sp+9],st[sp+10],st[sp+11],st[sp+12],st[sp+13],st[sp+14],st[sp+15],st[sp+16],st[sp+17],st[sp+18],st[sp+19]); break; \
1378 : default: \
1379 : pari_err_IMPL("functions with more than 20 parameters");\
1380 : goto endeval; /*LCOV_EXCL_LINE*/ \
1381 : }
1382 :
1383 :
1384 107531550 : case OCcallgen:
1385 : {
1386 107531550 : entree *ep = (entree *)operand;
1387 : GEN res;
1388 : /* Macro Madness : evaluate function ep->value on arguments
1389 : * st[sp-ep->arity .. sp]. Set res = result. */
1390 107531550 : EVAL_f(ep->value, GEN, res=);
1391 107514743 : if (br_status) goto endeval;
1392 107377614 : gel(st,sp++)=res;
1393 107377614 : break;
1394 : }
1395 617247534 : case OCcallgen2: /*same for ep->arity = 2. Is this optimization worth it ?*/
1396 : {
1397 617247534 : entree *ep = (entree *)operand;
1398 : GEN res;
1399 617247534 : sp-=2;
1400 617247534 : res = ((GEN (*)(GEN,GEN))ep->value)(gel(st,sp),gel(st,sp+1));
1401 617299710 : if (br_status) goto endeval;
1402 617299682 : gel(st,sp++)=res;
1403 617299682 : break;
1404 : }
1405 18768209 : case OCcalllong:
1406 : {
1407 18768209 : entree *ep = (entree *)operand;
1408 : long res;
1409 18768209 : EVAL_f(ep->value, long, res=);
1410 18769466 : if (br_status) goto endeval;
1411 18769466 : st[sp++] = res;
1412 18769466 : break;
1413 : }
1414 1707229 : case OCcallint:
1415 : {
1416 1707229 : entree *ep = (entree *)operand;
1417 : long res;
1418 1707229 : EVAL_f(ep->value, int, res=);
1419 1707019 : if (br_status) goto endeval;
1420 1707019 : st[sp++] = res;
1421 1707019 : break;
1422 : }
1423 49783577 : case OCcallvoid:
1424 : {
1425 49783577 : entree *ep = (entree *)operand;
1426 49783577 : EVAL_f(ep->value, void,(void));
1427 49783055 : if (br_status) goto endeval;
1428 49624494 : break;
1429 : }
1430 : #undef EVAL_f
1431 :
1432 35464656 : case OCcalluser:
1433 : {
1434 35464656 : long n=operand;
1435 35464656 : GEN fun = gel(st,sp-1-n);
1436 : long arity, isvar;
1437 : GEN z;
1438 35464656 : if (typ(fun)!=t_CLOSURE) pari_err(e_NOTFUNC, fun);
1439 35461947 : isvar = closure_is_variadic(fun);
1440 35461950 : arity = closure_arity(fun);
1441 35461941 : if (!isvar || n < arity)
1442 : {
1443 35461871 : st_alloc(arity-n);
1444 35461855 : if (n>arity)
1445 0 : pari_err(e_MISC,"too many parameters in user-defined function call");
1446 35488368 : for (j=n+1;j<=arity;j++)
1447 26513 : gel(st,sp++)=0;
1448 35461855 : if (isvar) gel(st,sp-1) = cgetg(1,t_VEC);
1449 : }
1450 : else
1451 : {
1452 : GEN v;
1453 70 : long j, m = n-arity+1;
1454 70 : v = cgetg(m+1,t_VEC);
1455 70 : sp-=m;
1456 301 : for (j=1; j<=m; j++)
1457 231 : gel(v,j) = gel(st,sp+j-1)? gcopy(gel(st,sp+j-1)): gen_0;
1458 70 : gel(st,sp++)=v;
1459 : }
1460 35461925 : z = closure_return(fun);
1461 35457211 : if (br_status) goto endeval;
1462 35457211 : gel(st, sp-1) = z;
1463 35457211 : break;
1464 : }
1465 43021336 : case OCnewframe:
1466 43021336 : if (operand>0) nbmvar+=operand;
1467 13 : else operand=-operand;
1468 43021336 : pari_stack_alloc(&s_var,operand);
1469 43021336 : s_var.n+=operand;
1470 123223101 : for(j=1;j<=operand;j++)
1471 : {
1472 80201765 : var[s_var.n-j].flag=PUSH_VAL;
1473 80201765 : var[s_var.n-j].value=gen_0;
1474 : }
1475 43021336 : break;
1476 7130 : case OCsaveframe:
1477 : {
1478 7130 : GEN cl = (operand?gcopy:shallowcopy)(gel(st,sp-1));
1479 7130 : GEN f = gel(cl, 7);
1480 7130 : long j, l = lg(f);
1481 7130 : GEN v = cgetg(l, t_VEC);
1482 76757 : for (j = 1; j < l; j++)
1483 69627 : if (signe(gel(f,l-j))==0)
1484 : {
1485 10086 : GEN val = var[s_var.n-j].value;
1486 10086 : gel(v,j) = operand?gcopy(val):val;
1487 : } else
1488 59541 : gel(v,j) = gnil;
1489 7130 : gel(cl,7) = v;
1490 7130 : gel(st,sp-1) = cl;
1491 : }
1492 7130 : break;
1493 112 : case OCpackargs:
1494 : {
1495 112 : GEN def = cgetg(operand+1, t_VECSMALL);
1496 112 : GEN args = cgetg(operand+1, t_VEC);
1497 112 : pari_stack_alloc(&s_var,operand);
1498 112 : sp-=operand;
1499 238 : for (j=0;j<operand;j++)
1500 : {
1501 126 : if (gel(st,sp+j))
1502 : {
1503 126 : gel(args,j+1) = gel(st,sp+j);
1504 126 : uel(def ,j+1) = 1;
1505 : }
1506 : else
1507 : {
1508 0 : gel(args,j+1) = gen_0;
1509 0 : uel(def ,j+1) = 0;
1510 : }
1511 : }
1512 112 : gel(st, sp++) = args;
1513 112 : gel(st, sp++) = def;
1514 112 : break;
1515 : }
1516 36512188 : case OCgetargs:
1517 36512188 : pari_stack_alloc(&s_var,operand);
1518 36511281 : s_var.n+=operand;
1519 36511281 : nbmvar+=operand;
1520 36511281 : sp-=operand;
1521 100023761 : for (j=0;j<operand;j++)
1522 : {
1523 63512772 : if (gel(st,sp+j))
1524 63503861 : pushlex(j-operand,gel(st,sp+j));
1525 : else
1526 : {
1527 8911 : var[s_var.n+j-operand].flag=DEFAULT_VAL;
1528 8911 : var[s_var.n+j-operand].value=gen_0;
1529 : }
1530 : }
1531 36510989 : break;
1532 49 : case OCcheckuserargs:
1533 105 : for (j=0; j<operand; j++)
1534 77 : if (var[s_var.n-operand+j].flag==DEFAULT_VAL)
1535 21 : pari_err(e_MISC,"missing mandatory argument"
1536 : " '%s' in user function",get_arg_name(C,j+1));
1537 28 : break;
1538 13915091 : case OCcheckargs:
1539 62394269 : for (j=sp-1;operand;operand>>=1UL,j--)
1540 48479188 : if ((operand&1L) && gel(st,j)==NULL)
1541 0 : pari_err(e_MISC,"missing mandatory argument");
1542 13915081 : break;
1543 441 : case OCcheckargs0:
1544 882 : for (j=sp-1;operand;operand>>=1UL,j--)
1545 441 : if ((operand&1L) && gel(st,j))
1546 0 : pari_err(e_MISC,"argument type not implemented");
1547 441 : break;
1548 23413 : case OCdefaultlong:
1549 23413 : sp--;
1550 23413 : if (st[sp+operand])
1551 1050 : st[sp+operand]=gtos(gel(st,sp+operand));
1552 : else
1553 22363 : st[sp+operand]=st[sp];
1554 23413 : break;
1555 0 : case OCdefaultulong:
1556 0 : sp--;
1557 0 : if (st[sp+operand])
1558 0 : st[sp+operand]=gtou(gel(st,sp+operand));
1559 : else
1560 0 : st[sp+operand]=st[sp];
1561 0 : break;
1562 0 : case OCdefaultgen:
1563 0 : sp--;
1564 0 : if (!st[sp+operand])
1565 0 : st[sp+operand]=st[sp];
1566 0 : break;
1567 21735795 : case OCvec:
1568 21735795 : gel(st,sp++)=cgetg(operand,t_VEC);
1569 21735794 : st[sp++]=avma;
1570 21735794 : break;
1571 4984 : case OCcol:
1572 4984 : gel(st,sp++)=cgetg(operand,t_COL);
1573 4984 : st[sp++]=avma;
1574 4984 : break;
1575 55978 : case OCmat:
1576 : {
1577 : GEN z;
1578 55978 : long l=st[sp-1];
1579 55978 : z=cgetg(operand,t_MAT);
1580 186624 : for(j=1;j<operand;j++)
1581 130646 : gel(z,j) = cgetg(l,t_COL);
1582 55978 : gel(st,sp-1) = z;
1583 55978 : st[sp++]=avma;
1584 : }
1585 55978 : break;
1586 107175822 : case OCpop:
1587 107175822 : sp-=operand;
1588 107175822 : break;
1589 31396969 : case OCdup:
1590 : {
1591 31396969 : long i, s=st[sp-1];
1592 31396969 : st_alloc(operand);
1593 62804368 : for(i=1;i<=operand;i++)
1594 31407399 : st[sp++]=s;
1595 : }
1596 31396969 : break;
1597 : }
1598 : }
1599 : if (0)
1600 : {
1601 295718 : endeval:
1602 295718 : sp = saved_sp;
1603 295718 : for( ; rp>saved_rp ; )
1604 : {
1605 0 : gp_pointer *g = &ptrs[--rp];
1606 0 : clone_unlock_deep(g->ox);
1607 : }
1608 : }
1609 306279174 : s_prec.n = saved_prec;
1610 306279174 : s_trace.n--;
1611 306279174 : restore_vars(nbmvar, nblvar, nblock);
1612 305921678 : clone_unlock(C);
1613 305885122 : }
1614 :
1615 : GEN
1616 34051464 : closure_evalgen(GEN C)
1617 : {
1618 34051464 : pari_sp ltop=avma;
1619 34051464 : closure_eval(C);
1620 34008215 : if (br_status) return gc_NULL(ltop);
1621 34008153 : return gc_upto(ltop,gel(st,--sp));
1622 : }
1623 :
1624 : long
1625 1040701 : evalstate_get_trace(void)
1626 1040701 : { return s_trace.n; }
1627 :
1628 : void
1629 18 : evalstate_set_trace(long lvl)
1630 18 : { s_trace.n = lvl; }
1631 :
1632 : void
1633 1431409 : evalstate_save(struct pari_evalstate *state)
1634 : {
1635 1431409 : state->avma = avma;
1636 1431409 : state->sp = sp;
1637 1431409 : state->rp = rp;
1638 1431409 : state->prec = s_prec.n;
1639 1431409 : state->var = s_var.n;
1640 1431409 : state->lvars= s_lvars.n;
1641 1431409 : state->locks= s_locks.n;
1642 1431409 : state->trace= s_trace.n;
1643 1431409 : compilestate_save(&state->comp);
1644 1431409 : mtstate_save(&state->mt);
1645 1431409 : }
1646 :
1647 : void
1648 56394 : evalstate_restore(struct pari_evalstate *state)
1649 : {
1650 56394 : set_avma(state->avma);
1651 56394 : mtstate_restore(&state->mt);
1652 56394 : sp = state->sp;
1653 56394 : rp = state->rp;
1654 56394 : s_prec.n = state->prec;
1655 56394 : restore_vars(s_var.n-state->var, s_lvars.n-state->lvars,
1656 56394 : s_locks.n-state->locks);
1657 56394 : restore_trace(s_trace.n-state->trace);
1658 56394 : reset_break();
1659 56394 : compilestate_restore(&state->comp);
1660 56394 : }
1661 :
1662 : GEN
1663 43250 : evalstate_restore_err(struct pari_evalstate *state)
1664 : {
1665 43250 : GENbin* err = copy_bin(pari_err_last());
1666 43250 : evalstate_restore(state);
1667 43250 : return bin_copy(err);
1668 : }
1669 :
1670 : void
1671 452 : evalstate_reset(void)
1672 : {
1673 452 : mtstate_reset();
1674 452 : restore_vars(s_var.n, s_lvars.n, s_locks.n);
1675 452 : sp = rp = dbg_level = s_trace.n = 0;
1676 452 : reset_break();
1677 452 : compilestate_reset();
1678 452 : parsestate_reset();
1679 452 : set_avma(pari_mainstack->top);
1680 452 : }
1681 :
1682 : void
1683 0 : evalstate_clone(void)
1684 : {
1685 : long i;
1686 0 : for (i = 1; i<=s_var.n; i++) copylex(-i);
1687 0 : lvar_make_safe();
1688 0 : for (i = 0; i< s_trace.n; i++)
1689 : {
1690 0 : GEN C = trace[i].closure;
1691 0 : if (isonstack(C)) trace[i].closure = gclone(C);
1692 : }
1693 0 : }
1694 :
1695 : GEN
1696 76028462 : closure_evalnobrk(GEN C)
1697 : {
1698 76028462 : pari_sp ltop=avma;
1699 76028462 : closure_eval(C);
1700 76028441 : if (br_status) pari_err(e_MISC, "break not allowed here");
1701 76028434 : return gc_upto(ltop,gel(st,--sp));
1702 : }
1703 :
1704 : void
1705 144433902 : closure_evalvoid(GEN C)
1706 : {
1707 144433902 : pari_sp ltop=avma;
1708 144433902 : closure_eval(C);
1709 144504642 : set_avma(ltop);
1710 144487602 : }
1711 :
1712 : GEN
1713 936656 : closure_evalres(GEN C)
1714 : {
1715 936656 : return closure_return(C);
1716 : }
1717 :
1718 : INLINE GEN
1719 14970161 : closure_returnupto(GEN C)
1720 : {
1721 14970161 : pari_sp av=avma;
1722 14970161 : return copyupto(closure_return(C),(GEN)av);
1723 : }
1724 :
1725 : GEN
1726 12 : pareval_worker(GEN C)
1727 : {
1728 12 : return closure_callgenall(C, 0);
1729 : }
1730 :
1731 : GEN
1732 6 : pareval(GEN C)
1733 : {
1734 6 : pari_sp av = avma;
1735 6 : long l = lg(C), i;
1736 : GEN worker;
1737 6 : if (!is_vec_t(typ(C))) pari_err_TYPE("pareval",C);
1738 18 : for (i=1; i<l; i++)
1739 12 : if (typ(gel(C,i))!=t_CLOSURE)
1740 0 : pari_err_TYPE("pareval",gel(C,i));
1741 6 : worker = snm_closure(is_entry("_pareval_worker"), NULL);
1742 6 : return gc_upto(av, gen_parapply(worker, C));
1743 : }
1744 :
1745 : GEN
1746 624 : parvector_worker(GEN i, GEN C)
1747 : {
1748 624 : return closure_callgen1(C, i);
1749 : }
1750 :
1751 : GEN
1752 86 : parmatrix_worker(GEN i, GEN j, GEN C)
1753 : {
1754 86 : return closure_callgen2(C, i, j);
1755 : }
1756 :
1757 : GEN
1758 9590 : parfor_worker(GEN i, GEN C)
1759 : {
1760 9590 : retmkvec2(gcopy(i), closure_callgen1(C, i));
1761 : }
1762 :
1763 : GEN
1764 13 : parmatrix(long n, long m, GEN code)
1765 : {
1766 13 : long i, pending = 0, workid, nm = n*m;
1767 : GEN worker, a, M, done;
1768 : struct pari_mt pt;
1769 13 : if (m < 0) pari_err_DOMAIN("parmatrix", "nbcols", "<", gen_0, stoi(m));
1770 13 : if (n < 0) pari_err_DOMAIN("parmatrix", "nbrows", "<", gen_0, stoi(n));
1771 13 : worker = snm_closure(is_entry("_parmatrix_worker"), mkvec(code));
1772 13 : mt_queue_start_lim(&pt, worker, n);
1773 13 : a = mkvec2(cgetipos(3), cgetipos(3)); /* left on the stack */
1774 13 : M = cgetg(m+1, t_MAT);
1775 46 : for (i = 1; i <= m; i++)
1776 33 : gel(M,i) = cgetg(n+1, t_COL);
1777 104 : for (i = 0; i < nm || pending; i < nm ? i++: 0)
1778 : {
1779 91 : mael(a,1,2) = 1+(i%n);
1780 91 : mael(a,2,2) = 1+(i/n);
1781 91 : mt_queue_submit(&pt, i, i < nm ? a: NULL);
1782 91 : done = mt_queue_get(&pt, &workid, &pending);
1783 91 : if (done) gcoeff(M,1+(workid%n),1+(workid/n)) = done;
1784 : }
1785 13 : mt_queue_end(&pt);
1786 13 : return M;
1787 : }
1788 :
1789 : GEN
1790 31 : parvector(long n, GEN code)
1791 : {
1792 31 : long i, pending = 0, workid;
1793 : GEN worker, a, V, done;
1794 : struct pari_mt pt;
1795 31 : if (n < 0) pari_err_DOMAIN("parvector", "dimension", "<", gen_0, stoi(n));
1796 31 : worker = snm_closure(is_entry("_parvector_worker"), mkvec(code));
1797 31 : mt_queue_start_lim(&pt, worker, n);
1798 31 : a = mkvec(cgetipos(3)); /* left on the stack */
1799 31 : V = cgetg(n+1, t_VEC);
1800 615 : for (i=1; i<=n || pending; i<=n ? i++: 0)
1801 : {
1802 590 : mael(a,1,2) = i;
1803 590 : mt_queue_submit(&pt, i, i<=n? a: NULL);
1804 586 : done = mt_queue_get(&pt, &workid, &pending);
1805 584 : if (done) gel(V,workid) = done;
1806 : }
1807 25 : mt_queue_end(&pt);
1808 25 : return V;
1809 : }
1810 :
1811 : /* suitable for gc_upto */
1812 : GEN
1813 7493 : parsum_slice_worker(GEN a, GEN b, GEN m, GEN worker)
1814 : {
1815 7493 : pari_sp av = avma;
1816 7493 : GEN s = gen_0;
1817 134632 : while (gcmp(a,b)<=0)
1818 : {
1819 126702 : s = gadd(s, closure_callgen1(worker, a));
1820 126967 : a = addii(a, m);
1821 127139 : if (gc_needed(av,1))
1822 : {
1823 0 : if (DEBUGMEM>1) pari_warn(warnmem,"parsum_slice");
1824 0 : (void)gc_all(av,2,&a,&s);
1825 : }
1826 : }
1827 7428 : return gc_upto(av,s);
1828 : }
1829 :
1830 : GEN
1831 2062 : parsum(GEN a, GEN b, GEN code)
1832 : {
1833 2062 : pari_sp av = avma;
1834 : GEN worker, mG, v, s, N;
1835 : long r, m, pending;
1836 : struct pari_mt pt;
1837 : pari_sp av2;
1838 :
1839 2062 : if (typ(a) != t_INT) pari_err_TYPE("parsum",a);
1840 2062 : if (gcmp(b,a) < 0) return gen_0;
1841 2062 : b = gfloor(b);
1842 2062 : N = addiu(subii(b, a), 1);
1843 2062 : mG = sqrti(N);
1844 2062 : m = itou(mG);
1845 2062 : worker = snm_closure(is_entry("_parsum_slice_worker"), mkvec3(b,mG,code));
1846 2062 : mt_queue_start_lim(&pt, worker, m);
1847 2062 : s = gen_0; a = setloop(a); v = mkvec(a); pending = 0; av2 = avma;
1848 12087 : for (r = 1; r <= m || pending; r++)
1849 : {
1850 : long workid;
1851 : GEN done;
1852 10046 : mt_queue_submit(&pt, 0, r <= m? v: NULL);
1853 10028 : done = mt_queue_get(&pt, &workid, &pending);
1854 10025 : if (done)
1855 : {
1856 7518 : s = gadd(s, done);
1857 7518 : if (gc_needed(av2,1))
1858 : {
1859 0 : if (DEBUGMEM>1) pari_warn(warnmem,"parsum");
1860 0 : s = gc_upto(av2,s);
1861 : }
1862 : }
1863 10025 : a = incloop(a); gel(v,1) = a;
1864 : }
1865 2041 : mt_queue_end(&pt); return gc_upto(av, s);
1866 : }
1867 :
1868 : void
1869 346 : parfor(GEN a, GEN b, GEN code, void *E, long call(void*, GEN, GEN))
1870 : {
1871 346 : pari_sp av = avma, av2;
1872 346 : long running, pending = 0, lim;
1873 346 : long status = br_NONE;
1874 346 : GEN worker = snm_closure(is_entry("_parfor_worker"), mkvec(code));
1875 346 : GEN done, stop = NULL;
1876 : struct pari_mt pt;
1877 346 : if (typ(a) != t_INT) pari_err_TYPE("parfor",a);
1878 346 : if (b)
1879 : {
1880 346 : if (gcmp(b,a) < 0) return;
1881 346 : if (typ(b) == t_INFINITY)
1882 : {
1883 6 : if (inf_get_sign(b) < 0) return;
1884 6 : b = NULL;
1885 : }
1886 : else
1887 340 : b = gfloor(b);
1888 : }
1889 346 : lim = b ? itos_or_0(subii(addis(b,1),a)): 0;
1890 346 : mt_queue_start_lim(&pt, worker, lim);
1891 346 : a = mkvec(setloop(a));
1892 346 : av2 = avma;
1893 7497 : while ((running = (!stop && (!b || cmpii(gel(a,1),b) <= 0))) || pending)
1894 : {
1895 7157 : mt_queue_submit(&pt, 0, running ? a: NULL);
1896 7153 : done = mt_queue_get(&pt, NULL, &pending);
1897 7151 : if (call && done && (!stop || cmpii(gel(done,1),stop) < 0))
1898 5457 : if (call(E, gel(done,1), gel(done,2)))
1899 : {
1900 223 : status = br_status;
1901 223 : br_status = br_NONE;
1902 223 : stop = gc_INT(av2, gel(done,1));
1903 : }
1904 7151 : gel(a,1) = incloop(gel(a,1));
1905 7151 : if (!stop) set_avma(av2);
1906 : }
1907 340 : set_avma(av2);
1908 340 : mt_queue_end(&pt);
1909 340 : br_status = status;
1910 340 : set_avma(av);
1911 : }
1912 :
1913 : static void
1914 0 : parforiter_init(struct parfor_iter *T, GEN code)
1915 : {
1916 0 : T->pending = 0;
1917 0 : T->worker = snm_closure(is_entry("_parfor_worker"), mkvec(code));
1918 0 : mt_queue_start(&T->pt, T->worker);
1919 0 : }
1920 :
1921 : static GEN
1922 0 : parforiter_next(struct parfor_iter *T, GEN v)
1923 : {
1924 0 : mt_queue_submit(&T->pt, 0, v);
1925 0 : return mt_queue_get(&T->pt, NULL, &T->pending);
1926 : }
1927 :
1928 : static void
1929 0 : parforiter_stop(struct parfor_iter *T)
1930 : {
1931 0 : while (T->pending)
1932 : {
1933 0 : mt_queue_submit(&T->pt, 0, NULL);
1934 0 : (void) mt_queue_get(&T->pt, NULL, &T->pending);
1935 : }
1936 0 : mt_queue_end(&T->pt);
1937 0 : }
1938 :
1939 : void
1940 0 : parfor_init(parfor_t *T, GEN a, GEN b, GEN code)
1941 : {
1942 0 : if (typ(a) != t_INT) pari_err_TYPE("parfor",a);
1943 0 : T->b = b ? gfloor(b): NULL;
1944 0 : T->a = mkvec(setloop(a));
1945 0 : parforiter_init(&T->iter, code);
1946 0 : }
1947 :
1948 : GEN
1949 0 : parfor_next(parfor_t *T)
1950 : {
1951 : long running;
1952 0 : while ((running=((!T->b || cmpii(gel(T->a,1),T->b) <= 0))) || T->iter.pending)
1953 : {
1954 0 : GEN done = parforiter_next(&T->iter, running ? T->a: NULL);
1955 0 : gel(T->a,1) = incloop(gel(T->a,1));
1956 0 : if (done) return done;
1957 : }
1958 0 : mt_queue_end(&T->iter.pt);
1959 0 : return NULL;
1960 : }
1961 :
1962 : void
1963 0 : parfor_stop(parfor_t *T) { parforiter_stop(&T->iter); }
1964 :
1965 : static long
1966 8459 : gp_evalvoid2(void *E, GEN x, GEN y)
1967 : {
1968 8459 : GEN code =(GEN) E;
1969 8459 : push_lex(x, code);
1970 8459 : push_lex(y, NULL);
1971 8459 : closure_evalvoid(code);
1972 8459 : pop_lex(2);
1973 8459 : return loop_break();
1974 : }
1975 :
1976 : void
1977 346 : parfor0(GEN a, GEN b, GEN code, GEN code2)
1978 : {
1979 346 : parfor(a, b, code, (void*)code2, code2 ? gp_evalvoid2: NULL);
1980 340 : }
1981 :
1982 0 : static int negcmp(GEN x, GEN y) { return gcmp(y,x); }
1983 :
1984 : void
1985 39 : parforstep(GEN a, GEN b, GEN s, GEN code, void *E, long call(void*, GEN, GEN))
1986 : {
1987 39 : pari_sp av = avma, av2;
1988 39 : long running, pending = 0;
1989 39 : long status = br_NONE;
1990 39 : GEN worker = snm_closure(is_entry("_parfor_worker"), mkvec(code));
1991 39 : GEN done, stop = NULL;
1992 : struct pari_mt pt;
1993 : long i, ss;
1994 39 : GEN v = NULL, lim;
1995 : int (*cmp)(GEN,GEN);
1996 :
1997 39 : b = gcopy(b);
1998 39 : s = gcopy(s); av = avma;
1999 39 : switch(typ(s))
2000 : {
2001 13 : case t_VEC: case t_COL:
2002 : {
2003 13 : GEN vs = vecsum(s);
2004 13 : ss = gsigne(vs); v = s;
2005 0 : lim = typ(b)==t_INFINITY ? inf_get_sign(b)==ss
2006 0 : ? int2n(BITS_IN_LONG): gen_0
2007 13 : : gdiv(gmulgs(gadd(gsub(b,a),vs),lg(vs)-1),vs);
2008 13 : break;
2009 : }
2010 13 : case t_INTMOD:
2011 13 : if (typ(a) != t_INT) a = gceil(a);
2012 13 : a = addii(a, modii(subii(gel(s,2),a), gel(s,1)));
2013 13 : s = gel(s,1); /* FALL THROUGH */
2014 26 : default:
2015 26 : ss = gsigne(s);
2016 26 : lim = typ(b)==t_INFINITY ? inf_get_sign(b)==ss
2017 0 : ? int2n(BITS_IN_LONG): gen_0
2018 26 : : gdiv(gadd(gsub(b,a),s),s);
2019 : }
2020 39 : lim = ceil_safe(lim);
2021 39 : if (!ss || typ(lim)!=t_INT) pari_err_DOMAIN("parforstep","step","=",gen_0,s);
2022 39 : if (signe(lim)<=0) { set_avma(av); return; }
2023 39 : cmp = (ss > 0)? &gcmp: &negcmp;
2024 39 : i = 0;
2025 39 : a = mkvec(a);
2026 39 : mt_queue_start_lim(&pt, worker, itou_or_0(lim));
2027 39 : av2 = avma;
2028 2695 : while ((running = (!stop && (!b || cmp(gel(a,1),b) <= 0))) || pending)
2029 : {
2030 2656 : mt_queue_submit(&pt, 0, running ? a: NULL);
2031 2656 : done = mt_queue_get(&pt, NULL, &pending);
2032 2656 : if (call && done && (!stop || cmp(gel(done,1),stop) < 0))
2033 2521 : if (call(E, gel(done,1), gel(done,2)))
2034 : {
2035 0 : status = br_status;
2036 0 : br_status = br_NONE;
2037 0 : stop = gel(done,1);
2038 : }
2039 2656 : if (running)
2040 : {
2041 2521 : if (v)
2042 : {
2043 1637 : if (++i >= lg(v)) i = 1;
2044 1637 : s = gel(v,i);
2045 : }
2046 2521 : gel(a,1) = gadd(gel(a,1),s);
2047 2521 : if (!stop) gel(a,1) = gc_upto(av2, gel(a,1));
2048 : }
2049 : }
2050 39 : mt_queue_end(&pt);
2051 39 : br_status = status;
2052 39 : set_avma(av);
2053 : }
2054 :
2055 : void
2056 39 : parforstep0(GEN a, GEN b, GEN s, GEN code, GEN code2)
2057 : {
2058 39 : parforstep(a, b, s, code, (void*)code2, code2 ? gp_evalvoid2: NULL);
2059 39 : }
2060 :
2061 : void
2062 0 : parforstep_init(parforstep_t *T, GEN a, GEN b, GEN s, GEN code)
2063 : {
2064 : long ss;
2065 0 : if (typ(a) != t_INT) pari_err_TYPE("parfor",a);
2066 0 : switch(typ(s))
2067 : {
2068 0 : case t_VEC: case t_COL:
2069 0 : ss = gsigne(vecsum(s));
2070 0 : break;
2071 0 : case t_INTMOD:
2072 0 : if (typ(a) != t_INT) a = gceil(a);
2073 0 : a = addii(a, modii(subii(gel(s,2),a), gel(s,1)));
2074 0 : s = gel(s,1);
2075 0 : default: /* FALL THROUGH */
2076 0 : ss = gsigne(s);
2077 : }
2078 0 : T->cmp = (ss > 0)? &gcmp: &negcmp;
2079 0 : T->s = s;
2080 0 : T->i = 0;
2081 0 : T->b = b;
2082 0 : T->a = mkvec(a);
2083 0 : parforiter_init(&T->iter, code);
2084 0 : }
2085 :
2086 : GEN
2087 0 : parforstep_next(parforstep_t *T)
2088 : {
2089 : long running;
2090 0 : while ((running=((!T->b || T->cmp(gel(T->a,1),T->b) <= 0))) || T->iter.pending)
2091 : {
2092 0 : GEN done = parforiter_next(&T->iter, running ? T->a: NULL);
2093 0 : if (running)
2094 : {
2095 0 : if (is_vec_t(typ(T->s)))
2096 : {
2097 0 : if (++(T->i) >= lg(T->s)) T->i = 1;
2098 0 : gel(T->a,1) = gadd(gel(T->a,1), gel(T->s,T->i));
2099 : }
2100 0 : else gel(T->a,1) = gadd(gel(T->a,1), T->s);
2101 : }
2102 0 : if (done) return done;
2103 : }
2104 0 : mt_queue_end(&T->iter.pt);
2105 0 : return NULL;
2106 : }
2107 :
2108 : void
2109 0 : parforstep_stop(parforstep_t *T) { parforiter_stop(&T->iter); }
2110 :
2111 : void
2112 0 : parforprimestep_init(parforprime_t *T, GEN a, GEN b, GEN q, GEN code)
2113 : {
2114 0 : forprimestep_init(&T->forprime, a, b, q);
2115 0 : T->v = mkvec(gen_0);
2116 0 : parforiter_init(&T->iter, code);
2117 0 : }
2118 :
2119 : void
2120 0 : parforprime_init(parforprime_t *T, GEN a, GEN b, GEN code)
2121 0 : { parforprimestep_init(T, a, b, NULL, code); }
2122 :
2123 : GEN
2124 0 : parforprime_next(parforprime_t *T)
2125 : {
2126 : long running;
2127 0 : while ((running = !!forprime_next(&T->forprime)) || T->iter.pending)
2128 : {
2129 : GEN done;
2130 0 : gel(T->v, 1) = T->forprime.pp;
2131 0 : done = parforiter_next(&T->iter, running ? T->v: NULL);
2132 0 : if (done) return done;
2133 : }
2134 0 : mt_queue_end(&T->iter.pt);
2135 0 : return NULL;
2136 : }
2137 :
2138 : void
2139 0 : parforprime_stop(parforprime_t *T) { parforiter_stop(&T->iter); }
2140 :
2141 : void
2142 20 : parforprimestep(GEN a, GEN b, GEN q, GEN code, void *E, long call(void*, GEN, GEN))
2143 : {
2144 20 : pari_sp av = avma, av2;
2145 20 : long running, pending = 0;
2146 20 : long status = br_NONE;
2147 20 : GEN worker = snm_closure(is_entry("_parfor_worker"), mkvec(code));
2148 20 : GEN v, done, stop = NULL;
2149 : struct pari_mt pt;
2150 : forprime_t T;
2151 :
2152 20 : if (!forprimestep_init(&T, a,b,q)) { set_avma(av); return; }
2153 20 : mt_queue_start(&pt, worker);
2154 20 : v = mkvec(gen_0);
2155 20 : av2 = avma;
2156 172 : while ((running = (!stop && forprime_next(&T))) || pending)
2157 : {
2158 152 : gel(v, 1) = T.pp;
2159 152 : mt_queue_submit(&pt, 0, running ? v: NULL);
2160 152 : done = mt_queue_get(&pt, NULL, &pending);
2161 152 : if (call && done && (!stop || cmpii(gel(done,1),stop) < 0))
2162 125 : if (call(E, gel(done,1), gel(done,2)))
2163 : {
2164 0 : status = br_status;
2165 0 : br_status = br_NONE;
2166 0 : stop = gc_INT(av2, gel(done,1));
2167 : }
2168 152 : if (!stop) set_avma(av2);
2169 : }
2170 20 : set_avma(av2);
2171 20 : mt_queue_end(&pt);
2172 20 : br_status = status;
2173 20 : set_avma(av);
2174 : }
2175 :
2176 : void
2177 13 : parforprime(GEN a, GEN b, GEN code, void *E, long call(void*, GEN, GEN))
2178 : {
2179 13 : parforprimestep(a, b, NULL, code, E, call);
2180 13 : }
2181 :
2182 : void
2183 13 : parforprime0(GEN a, GEN b, GEN code, GEN code2)
2184 : {
2185 13 : parforprime(a, b, code, (void*)code2, code2? gp_evalvoid2: NULL);
2186 13 : }
2187 :
2188 : void
2189 7 : parforprimestep0(GEN a, GEN b, GEN q, GEN code, GEN code2)
2190 : {
2191 7 : parforprimestep(a, b, q, code, (void*)code2, code2? gp_evalvoid2: NULL);
2192 7 : }
2193 :
2194 : void
2195 0 : parforvec_init(parforvec_t *T, GEN x, GEN code, long flag)
2196 : {
2197 0 : forvec_init(&T->forvec, x, flag);
2198 0 : T->v = mkvec(gen_0); T->running = 1;
2199 0 : parforiter_init(&T->iter, code);
2200 0 : }
2201 :
2202 : GEN
2203 0 : parforvec_next(parforvec_t *T)
2204 : {
2205 0 : GEN v = NULL;
2206 0 : while ((T->running && (v = forvec_next(&T->forvec))) || T->iter.pending)
2207 : {
2208 : GEN done;
2209 0 : if (!v) T->running = 0;
2210 0 : else if (T->running) gel(T->v, 1) = v;
2211 0 : done = parforiter_next(&T->iter, v ? T->v: NULL);
2212 0 : if (done) return done;
2213 : }
2214 0 : mt_queue_end(&T->iter.pt);
2215 0 : return NULL;
2216 : }
2217 :
2218 : void
2219 0 : parforvec_stop(parforvec_t *T) { parforiter_stop(&T->iter); }
2220 :
2221 : void
2222 39 : parforvec(GEN x, GEN code, long flag, void *E, long call(void*, GEN, GEN))
2223 : {
2224 39 : pari_sp av = avma, av2;
2225 39 : long running, pending = 0;
2226 39 : long status = br_NONE;
2227 39 : GEN worker = snm_closure(is_entry("_parfor_worker"), mkvec(code));
2228 39 : GEN done, stop = NULL;
2229 : struct pari_mt pt;
2230 : forvec_t T;
2231 39 : GEN a, v = gen_0;
2232 :
2233 39 : if (!forvec_init(&T, x, flag)) { set_avma(av); return; }
2234 39 : mt_queue_start(&pt, worker);
2235 39 : a = mkvec(gen_0);
2236 39 : av2 = avma;
2237 415 : while ((running = (!stop && v && (v = forvec_next(&T)))) || pending)
2238 : {
2239 376 : gel(a, 1) = v;
2240 376 : mt_queue_submit(&pt, 0, running ? a: NULL);
2241 376 : done = mt_queue_get(&pt, NULL, &pending);
2242 376 : if (call && done && (!stop || lexcmp(gel(done,1),stop) < 0))
2243 300 : if (call(E, gel(done,1), gel(done,2)))
2244 : {
2245 0 : status = br_status;
2246 0 : br_status = br_NONE;
2247 0 : stop = gc_GEN(av2, gel(done,1));
2248 : }
2249 376 : if (!stop) set_avma(av2);
2250 : }
2251 39 : set_avma(av2);
2252 39 : mt_queue_end(&pt);
2253 39 : br_status = status;
2254 39 : set_avma(av);
2255 : }
2256 :
2257 : void
2258 39 : parforvec0(GEN x, GEN code, GEN code2, long flag)
2259 : {
2260 39 : parforvec(x, code, flag, (void*)code2, code2? gp_evalvoid2: NULL);
2261 39 : }
2262 :
2263 : void
2264 0 : parforeach_init(parforeach_t *T, GEN x, GEN code)
2265 : {
2266 0 : switch(typ(x))
2267 : {
2268 0 : case t_LIST:
2269 0 : x = list_data(x); /* FALL THROUGH */
2270 0 : if (!x) return;
2271 : case t_MAT: case t_VEC: case t_COL:
2272 0 : break;
2273 0 : default:
2274 0 : pari_err_TYPE("foreach",x);
2275 : return; /*LCOV_EXCL_LINE*/
2276 : }
2277 0 : T->x = x; T->i = 1; T->l = lg(x);
2278 0 : T->W = mkvec(gen_0);
2279 0 : T->iter.pending = 0;
2280 0 : T->iter.worker = snm_closure(is_entry("_parvector_worker"), mkvec(code));
2281 0 : mt_queue_start(&T->iter.pt, T->iter.worker);
2282 : }
2283 :
2284 : GEN
2285 0 : parforeach_next(parforeach_t *T)
2286 : {
2287 0 : while (T->i < T->l || T->iter.pending)
2288 : {
2289 : GEN done;
2290 : long workid;
2291 0 : if (T->i < T->l) gel(T->W,1) = gel(T->x, T->i);
2292 0 : mt_queue_submit(&T->iter.pt, T->i, T->i < T->l ? T->W: NULL);
2293 0 : T->i = minss(T->i+1, T->l);
2294 0 : done = mt_queue_get(&T->iter.pt, &workid, &T->iter.pending);
2295 0 : if (done) return mkvec2(gel(T->x,workid),done);
2296 : }
2297 0 : mt_queue_end(&T->iter.pt);
2298 0 : return NULL;
2299 : }
2300 :
2301 : void
2302 0 : parforeach_stop(parforeach_t *T) { parforiter_stop(&T->iter); }
2303 :
2304 : void
2305 7 : parforeach(GEN x, GEN code, void *E, long call(void*, GEN, GEN))
2306 : {
2307 7 : pari_sp av = avma, av2;
2308 7 : long pending = 0, n, i, stop = 0;
2309 7 : long status = br_NONE, workid;
2310 7 : GEN worker = snm_closure(is_entry("_parvector_worker"), mkvec(code));
2311 : GEN done, W;
2312 : struct pari_mt pt;
2313 7 : switch(typ(x))
2314 : {
2315 0 : case t_LIST:
2316 0 : x = list_data(x); /* FALL THROUGH */
2317 0 : if (!x) return;
2318 : case t_MAT: case t_VEC: case t_COL:
2319 7 : break;
2320 0 : default:
2321 0 : pari_err_TYPE("foreach",x);
2322 : return; /*LCOV_EXCL_LINE*/
2323 : }
2324 7 : clone_lock(x); n = lg(x)-1;
2325 7 : mt_queue_start_lim(&pt, worker, n);
2326 7 : W = cgetg(2, t_VEC);
2327 7 : av2 = avma;
2328 70 : for (i=1; i<=n || pending; i++)
2329 : {
2330 63 : if (!stop && i <= n) gel(W,1) = gel(x,i);
2331 63 : mt_queue_submit(&pt, i, !stop && i<=n? W: NULL);
2332 63 : done = mt_queue_get(&pt, &workid, &pending);
2333 63 : if (call && done && (!stop || workid < stop))
2334 56 : if (call(E, gel(x, workid), done))
2335 : {
2336 0 : status = br_status;
2337 0 : br_status = br_NONE;
2338 0 : stop = workid;
2339 : }
2340 63 : if (!stop) set_avma(av2);
2341 : }
2342 7 : set_avma(av2);
2343 7 : mt_queue_end(&pt);
2344 7 : clone_unlock_deep(x);
2345 7 : br_status = status;
2346 7 : set_avma(av);
2347 : }
2348 :
2349 : void
2350 7 : parforeach0(GEN x, GEN code, GEN code2)
2351 : {
2352 7 : parforeach(x, code, (void*)code2, code2? gp_evalvoid2: NULL);
2353 7 : }
2354 :
2355 : void
2356 0 : closure_callvoid1(GEN C, GEN x)
2357 : {
2358 0 : long i, ar = closure_arity(C);
2359 0 : gel(st,sp++) = x;
2360 0 : for(i=2; i <= ar; i++) gel(st,sp++) = NULL;
2361 0 : closure_evalvoid(C);
2362 0 : }
2363 :
2364 : GEN
2365 7 : closure_callgen0(GEN C)
2366 : {
2367 : GEN z;
2368 7 : long i, ar = closure_arity(C);
2369 7 : for(i=1; i<= ar; i++) gel(st,sp++) = NULL;
2370 7 : z = closure_returnupto(C);
2371 7 : return z;
2372 : }
2373 :
2374 : GEN
2375 182 : closure_callgen0prec(GEN C, long prec)
2376 : {
2377 : GEN z;
2378 182 : long i, ar = closure_arity(C);
2379 182 : for(i=1; i<= ar; i++) gel(st,sp++) = NULL;
2380 182 : push_localprec(prec);
2381 182 : z = closure_returnupto(C);
2382 182 : pop_localprec();
2383 182 : return z;
2384 : }
2385 :
2386 : GEN
2387 9619381 : closure_callgen1(GEN C, GEN x)
2388 : {
2389 9619381 : long i, ar = closure_arity(C);
2390 9619156 : gel(st,sp++) = x;
2391 9708875 : for(i=2; i<= ar; i++) gel(st,sp++) = NULL;
2392 9619156 : return closure_returnupto(C);
2393 : }
2394 :
2395 : GEN
2396 76681 : closure_callgen1prec(GEN C, GEN x, long prec)
2397 : {
2398 : GEN z;
2399 76681 : long i, ar = closure_arity(C);
2400 76681 : gel(st,sp++) = x;
2401 76695 : for(i=2; i<= ar; i++) gel(st,sp++) = NULL;
2402 76681 : push_localprec(prec);
2403 76681 : z = closure_returnupto(C);
2404 76681 : pop_localprec();
2405 76681 : return z;
2406 : }
2407 :
2408 : GEN
2409 67159 : closure_callgen2(GEN C, GEN x, GEN y)
2410 : {
2411 67159 : long i, ar = closure_arity(C);
2412 67159 : st_alloc(ar);
2413 67159 : gel(st,sp++) = x;
2414 67159 : gel(st,sp++) = y;
2415 67159 : for(i=3; i<=ar; i++) gel(st,sp++) = NULL;
2416 67159 : return closure_returnupto(C);
2417 : }
2418 :
2419 : GEN
2420 5206812 : closure_callgenvec(GEN C, GEN args)
2421 : {
2422 5206812 : long ar = closure_arity(C), isvar = closure_is_variadic(C);
2423 5206696 : long i, l = lg(args)-1;
2424 5206696 : st_alloc(ar);
2425 5206616 : if (l > ar)
2426 0 : pari_err(e_MISC,"too many parameters in user-defined function call");
2427 5206616 : if (closure_is_variadic(C) && l==ar && typ(gel(args,l))!=t_VEC)
2428 7 : pari_err_TYPE("call", gel(args,l));
2429 10457425 : for (i = 1; i <= l; i++) gel(st,sp++) = gel(args,i);
2430 5218502 : for( ; i <= ar; i++) gel(st,sp++) = NULL;
2431 5206487 : if (isvar && l<ar) gel(st,sp-1) = cgetg(1,t_VEC);
2432 5206487 : return closure_returnupto(C);
2433 : }
2434 :
2435 : GEN
2436 0 : closure_callgenvecprec(GEN C, GEN args, long prec)
2437 : {
2438 : GEN z;
2439 0 : push_localprec(prec);
2440 0 : z = closure_callgenvec(C, args);
2441 0 : pop_localprec();
2442 0 : return z;
2443 : }
2444 :
2445 : GEN
2446 336 : closure_callgenvecdef(GEN C, GEN args, GEN def)
2447 : {
2448 336 : long i, l = lg(args)-1, ar = closure_arity(C);
2449 336 : st_alloc(ar);
2450 336 : if (l > ar)
2451 0 : pari_err(e_MISC,"too many parameters in user-defined function call");
2452 336 : if (closure_is_variadic(C) && l==ar && typ(gel(args,l))!=t_VEC)
2453 0 : pari_err_TYPE("call", gel(args,l));
2454 700 : for (i = 1; i <= l; i++) gel(st,sp++) = def[i] ? gel(args,i): NULL;
2455 336 : for( ; i <= ar; i++) gel(st,sp++) = NULL;
2456 336 : return closure_returnupto(C);
2457 : }
2458 :
2459 : GEN
2460 336 : closure_callgenvecdefprec(GEN C, GEN args, GEN def, long prec)
2461 : {
2462 : GEN z;
2463 336 : push_localprec(prec);
2464 336 : z = closure_callgenvecdef(C, args, def);
2465 336 : pop_localprec();
2466 336 : return z;
2467 : }
2468 : GEN
2469 12 : closure_callgenall(GEN C, long n, ...)
2470 : {
2471 : va_list ap;
2472 12 : long i, ar = closure_arity(C);
2473 12 : va_start(ap,n);
2474 12 : if (n > ar)
2475 0 : pari_err(e_MISC,"too many parameters in user-defined function call");
2476 12 : st_alloc(ar);
2477 12 : for (i = 1; i <=n; i++) gel(st,sp++) = va_arg(ap, GEN);
2478 12 : for( ; i <=ar; i++) gel(st,sp++) = NULL;
2479 12 : va_end(ap);
2480 12 : return closure_returnupto(C);
2481 : }
2482 :
2483 : GEN
2484 39692084 : gp_eval(void *E, GEN x)
2485 : {
2486 39692084 : GEN code = (GEN)E;
2487 39692084 : set_lex(-1,x);
2488 39692084 : return closure_evalnobrk(code);
2489 : }
2490 :
2491 : GEN
2492 2052972 : gp_evalupto(void *E, GEN x)
2493 : {
2494 2052972 : pari_sp av = avma;
2495 2052972 : return copyupto(gp_eval(E,x), (GEN)av);
2496 : }
2497 :
2498 : GEN
2499 20734 : gp_evalprec(void *E, GEN x, long prec)
2500 : {
2501 : GEN z;
2502 20734 : push_localprec(prec);
2503 20734 : z = gp_eval(E, x);
2504 20734 : pop_localprec();
2505 20734 : return z;
2506 : }
2507 :
2508 : long
2509 26166231 : gp_evalbool(void *E, GEN x)
2510 26166231 : { pari_sp av = avma; return gc_long(av, !gequal0(gp_eval(E,x))); }
2511 :
2512 : long
2513 4146639 : gp_evalvoid(void *E, GEN x)
2514 : {
2515 4146639 : GEN code = (GEN)E;
2516 4146639 : set_lex(-1,x);
2517 4146639 : closure_evalvoid(code);
2518 4146639 : return loop_break();
2519 : }
2520 :
2521 : GEN
2522 112995 : gp_call(void *E, GEN x)
2523 : {
2524 112995 : GEN code = (GEN)E;
2525 112995 : return closure_callgen1(code, x);
2526 : }
2527 :
2528 : GEN
2529 23478 : gp_callprec(void *E, GEN x, long prec)
2530 : {
2531 23478 : GEN code = (GEN)E;
2532 23478 : return closure_callgen1prec(code, x, prec);
2533 : }
2534 :
2535 : GEN
2536 91 : gp_call2(void *E, GEN x, GEN y)
2537 : {
2538 91 : GEN code = (GEN)E;
2539 91 : return closure_callgen2(code, x, y);
2540 : }
2541 :
2542 : long
2543 872130 : gp_callbool(void *E, GEN x)
2544 : {
2545 872130 : pari_sp av = avma;
2546 872130 : GEN code = (GEN)E;
2547 872130 : return gc_long(av, !gequal0(closure_callgen1(code, x)));
2548 : }
2549 :
2550 : long
2551 0 : gp_callvoid(void *E, GEN x)
2552 : {
2553 0 : GEN code = (GEN)E;
2554 0 : closure_callvoid1(code, x);
2555 0 : return loop_break();
2556 : }
2557 :
2558 : INLINE const char *
2559 0 : disassemble_cast(long mode)
2560 : {
2561 0 : switch (mode)
2562 : {
2563 0 : case Gsmall:
2564 0 : return "small";
2565 0 : case Ggen:
2566 0 : return "gen";
2567 0 : case Gvar:
2568 0 : return "var";
2569 0 : case Gvoid:
2570 0 : return "void";
2571 0 : default:
2572 0 : return "unknown";
2573 : }
2574 : }
2575 :
2576 : void
2577 0 : closure_disassemble(GEN C)
2578 : {
2579 : const char * code;
2580 : GEN oper;
2581 : long i;
2582 0 : if (typ(C)!=t_CLOSURE) pari_err_TYPE("disassemble",C);
2583 0 : code=closure_codestr(C);
2584 0 : oper=closure_get_oper(C);
2585 0 : for(i=1;i<lg(oper);i++)
2586 : {
2587 0 : op_code opcode=(op_code) code[i];
2588 0 : long operand=oper[i];
2589 0 : pari_printf("%05ld\t",i);
2590 0 : switch(opcode)
2591 : {
2592 0 : case OCpushlong:
2593 0 : pari_printf("pushlong\t%ld\n",operand);
2594 0 : break;
2595 0 : case OCpushgnil:
2596 0 : pari_printf("pushgnil\n");
2597 0 : break;
2598 0 : case OCpushgen:
2599 0 : pari_printf("pushgen\t\t%ld\n",operand);
2600 0 : break;
2601 0 : case OCpushreal:
2602 0 : pari_printf("pushreal\t%ld\n",operand);
2603 0 : break;
2604 0 : case OCpushstoi:
2605 0 : pari_printf("pushstoi\t%ld\n",operand);
2606 0 : break;
2607 0 : case OCpushvar:
2608 : {
2609 0 : entree *ep = (entree *)operand;
2610 0 : pari_printf("pushvar\t%s\n",ep->name);
2611 0 : break;
2612 : }
2613 0 : case OCpushdyn:
2614 : {
2615 0 : entree *ep = (entree *)operand;
2616 0 : pari_printf("pushdyn\t\t%s\n",ep->name);
2617 0 : break;
2618 : }
2619 0 : case OCpushlex:
2620 0 : pari_printf("pushlex\t\t%ld\n",operand);
2621 0 : break;
2622 0 : case OCstoredyn:
2623 : {
2624 0 : entree *ep = (entree *)operand;
2625 0 : pari_printf("storedyn\t%s\n",ep->name);
2626 0 : break;
2627 : }
2628 0 : case OCstorelex:
2629 0 : pari_printf("storelex\t%ld\n",operand);
2630 0 : break;
2631 0 : case OCstoreptr:
2632 0 : pari_printf("storeptr\n");
2633 0 : break;
2634 0 : case OCsimpleptrdyn:
2635 : {
2636 0 : entree *ep = (entree *)operand;
2637 0 : pari_printf("simpleptrdyn\t%s\n",ep->name);
2638 0 : break;
2639 : }
2640 0 : case OCsimpleptrlex:
2641 0 : pari_printf("simpleptrlex\t%ld\n",operand);
2642 0 : break;
2643 0 : case OCnewptrdyn:
2644 : {
2645 0 : entree *ep = (entree *)operand;
2646 0 : pari_printf("newptrdyn\t%s\n",ep->name);
2647 0 : break;
2648 : }
2649 0 : case OCnewptrlex:
2650 0 : pari_printf("newptrlex\t%ld\n",operand);
2651 0 : break;
2652 0 : case OCpushptr:
2653 0 : pari_printf("pushptr\n");
2654 0 : break;
2655 0 : case OCstackgen:
2656 0 : pari_printf("stackgen\t%ld\n",operand);
2657 0 : break;
2658 0 : case OCendptr:
2659 0 : pari_printf("endptr\t\t%ld\n",operand);
2660 0 : break;
2661 0 : case OCprecreal:
2662 0 : pari_printf("precreal\n");
2663 0 : break;
2664 0 : case OCbitprecreal:
2665 0 : pari_printf("bitprecreal\n");
2666 0 : break;
2667 0 : case OCprecdl:
2668 0 : pari_printf("precdl\n");
2669 0 : break;
2670 0 : case OCstoi:
2671 0 : pari_printf("stoi\n");
2672 0 : break;
2673 0 : case OCutoi:
2674 0 : pari_printf("utoi\n");
2675 0 : break;
2676 0 : case OCitos:
2677 0 : pari_printf("itos\t\t%ld\n",operand);
2678 0 : break;
2679 0 : case OCitou:
2680 0 : pari_printf("itou\t\t%ld\n",operand);
2681 0 : break;
2682 0 : case OCtostr:
2683 0 : pari_printf("tostr\t\t%ld\n",operand);
2684 0 : break;
2685 0 : case OCvarn:
2686 0 : pari_printf("varn\t\t%ld\n",operand);
2687 0 : break;
2688 0 : case OCcopy:
2689 0 : pari_printf("copy\n");
2690 0 : break;
2691 0 : case OCcopyifclone:
2692 0 : pari_printf("copyifclone\n");
2693 0 : break;
2694 0 : case OCcompo1:
2695 0 : pari_printf("compo1\t\t%s\n",disassemble_cast(operand));
2696 0 : break;
2697 0 : case OCcompo1ptr:
2698 0 : pari_printf("compo1ptr\n");
2699 0 : break;
2700 0 : case OCcompo2:
2701 0 : pari_printf("compo2\t\t%s\n",disassemble_cast(operand));
2702 0 : break;
2703 0 : case OCcompo2ptr:
2704 0 : pari_printf("compo2ptr\n");
2705 0 : break;
2706 0 : case OCcompoC:
2707 0 : pari_printf("compoC\n");
2708 0 : break;
2709 0 : case OCcompoCptr:
2710 0 : pari_printf("compoCptr\n");
2711 0 : break;
2712 0 : case OCcompoL:
2713 0 : pari_printf("compoL\n");
2714 0 : break;
2715 0 : case OCcompoLptr:
2716 0 : pari_printf("compoLptr\n");
2717 0 : break;
2718 0 : case OCcheckargs:
2719 0 : pari_printf("checkargs\t0x%lx\n",operand);
2720 0 : break;
2721 0 : case OCcheckargs0:
2722 0 : pari_printf("checkargs0\t0x%lx\n",operand);
2723 0 : break;
2724 0 : case OCcheckuserargs:
2725 0 : pari_printf("checkuserargs\t%ld\n",operand);
2726 0 : break;
2727 0 : case OCdefaultlong:
2728 0 : pari_printf("defaultlong\t%ld\n",operand);
2729 0 : break;
2730 0 : case OCdefaultulong:
2731 0 : pari_printf("defaultulong\t%ld\n",operand);
2732 0 : break;
2733 0 : case OCdefaultgen:
2734 0 : pari_printf("defaultgen\t%ld\n",operand);
2735 0 : break;
2736 0 : case OCpackargs:
2737 0 : pari_printf("packargs\t%ld\n",operand);
2738 0 : break;
2739 0 : case OCgetargs:
2740 0 : pari_printf("getargs\t\t%ld\n",operand);
2741 0 : break;
2742 0 : case OCdefaultarg:
2743 0 : pari_printf("defaultarg\t%ld\n",operand);
2744 0 : break;
2745 0 : case OClocalvar:
2746 : {
2747 0 : entree *ep = (entree *)operand;
2748 0 : pari_printf("localvar\t%s\n",ep->name);
2749 0 : break;
2750 : }
2751 0 : case OClocalvar0:
2752 : {
2753 0 : entree *ep = (entree *)operand;
2754 0 : pari_printf("localvar0\t%s\n",ep->name);
2755 0 : break;
2756 : }
2757 0 : case OCexportvar:
2758 : {
2759 0 : entree *ep = (entree *)operand;
2760 0 : pari_printf("exportvar\t%s\n",ep->name);
2761 0 : break;
2762 : }
2763 0 : case OCunexportvar:
2764 : {
2765 0 : entree *ep = (entree *)operand;
2766 0 : pari_printf("unexportvar\t%s\n",ep->name);
2767 0 : break;
2768 : }
2769 0 : case OCcallgen:
2770 : {
2771 0 : entree *ep = (entree *)operand;
2772 0 : pari_printf("callgen\t\t%s\n",ep->name);
2773 0 : break;
2774 : }
2775 0 : case OCcallgen2:
2776 : {
2777 0 : entree *ep = (entree *)operand;
2778 0 : pari_printf("callgen2\t%s\n",ep->name);
2779 0 : break;
2780 : }
2781 0 : case OCcalllong:
2782 : {
2783 0 : entree *ep = (entree *)operand;
2784 0 : pari_printf("calllong\t%s\n",ep->name);
2785 0 : break;
2786 : }
2787 0 : case OCcallint:
2788 : {
2789 0 : entree *ep = (entree *)operand;
2790 0 : pari_printf("callint\t\t%s\n",ep->name);
2791 0 : break;
2792 : }
2793 0 : case OCcallvoid:
2794 : {
2795 0 : entree *ep = (entree *)operand;
2796 0 : pari_printf("callvoid\t%s\n",ep->name);
2797 0 : break;
2798 : }
2799 0 : case OCcalluser:
2800 0 : pari_printf("calluser\t%ld\n",operand);
2801 0 : break;
2802 0 : case OCvec:
2803 0 : pari_printf("vec\t\t%ld\n",operand);
2804 0 : break;
2805 0 : case OCcol:
2806 0 : pari_printf("col\t\t%ld\n",operand);
2807 0 : break;
2808 0 : case OCmat:
2809 0 : pari_printf("mat\t\t%ld\n",operand);
2810 0 : break;
2811 0 : case OCnewframe:
2812 0 : pari_printf("newframe\t%ld\n",operand);
2813 0 : break;
2814 0 : case OCsaveframe:
2815 0 : pari_printf("saveframe\t%ld\n", operand);
2816 0 : break;
2817 0 : case OCpop:
2818 0 : pari_printf("pop\t\t%ld\n",operand);
2819 0 : break;
2820 0 : case OCdup:
2821 0 : pari_printf("dup\t\t%ld\n",operand);
2822 0 : break;
2823 0 : case OCavma:
2824 0 : pari_printf("avma\n",operand);
2825 0 : break;
2826 0 : case OCgc:
2827 0 : pari_printf("gc\n",operand);
2828 0 : break;
2829 0 : case OCcowvardyn:
2830 : {
2831 0 : entree *ep = (entree *)operand;
2832 0 : pari_printf("cowvardyn\t%s\n",ep->name);
2833 0 : break;
2834 : }
2835 0 : case OCcowvarlex:
2836 0 : pari_printf("cowvarlex\t%ld\n",operand);
2837 0 : break;
2838 0 : case OCsetref:
2839 0 : pari_printf("setref\t\t%ld\n",operand);
2840 0 : break;
2841 0 : case OClock:
2842 0 : pari_printf("lock\t\t%ld\n",operand);
2843 0 : break;
2844 0 : case OCevalmnem:
2845 : {
2846 0 : entree *ep = (entree *)operand;
2847 0 : pari_printf("evalmnem\t%s\n",ep->name);
2848 0 : break;
2849 : }
2850 : }
2851 : }
2852 0 : }
2853 :
2854 : static int
2855 0 : opcode_need_relink(op_code opcode)
2856 : {
2857 0 : switch(opcode)
2858 : {
2859 0 : case OCpushlong:
2860 : case OCpushgen:
2861 : case OCpushgnil:
2862 : case OCpushreal:
2863 : case OCpushstoi:
2864 : case OCpushlex:
2865 : case OCstorelex:
2866 : case OCstoreptr:
2867 : case OCsimpleptrlex:
2868 : case OCnewptrlex:
2869 : case OCpushptr:
2870 : case OCstackgen:
2871 : case OCendptr:
2872 : case OCprecreal:
2873 : case OCbitprecreal:
2874 : case OCprecdl:
2875 : case OCstoi:
2876 : case OCutoi:
2877 : case OCitos:
2878 : case OCitou:
2879 : case OCtostr:
2880 : case OCvarn:
2881 : case OCcopy:
2882 : case OCcopyifclone:
2883 : case OCcompo1:
2884 : case OCcompo1ptr:
2885 : case OCcompo2:
2886 : case OCcompo2ptr:
2887 : case OCcompoC:
2888 : case OCcompoCptr:
2889 : case OCcompoL:
2890 : case OCcompoLptr:
2891 : case OCcheckargs:
2892 : case OCcheckargs0:
2893 : case OCcheckuserargs:
2894 : case OCpackargs:
2895 : case OCgetargs:
2896 : case OCdefaultarg:
2897 : case OCdefaultgen:
2898 : case OCdefaultlong:
2899 : case OCdefaultulong:
2900 : case OCcalluser:
2901 : case OCvec:
2902 : case OCcol:
2903 : case OCmat:
2904 : case OCnewframe:
2905 : case OCsaveframe:
2906 : case OCdup:
2907 : case OCpop:
2908 : case OCavma:
2909 : case OCgc:
2910 : case OCcowvarlex:
2911 : case OCsetref:
2912 : case OClock:
2913 0 : break;
2914 0 : case OCpushvar:
2915 : case OCpushdyn:
2916 : case OCstoredyn:
2917 : case OCsimpleptrdyn:
2918 : case OCnewptrdyn:
2919 : case OClocalvar:
2920 : case OClocalvar0:
2921 : case OCexportvar:
2922 : case OCunexportvar:
2923 : case OCcallgen:
2924 : case OCcallgen2:
2925 : case OCcalllong:
2926 : case OCcallint:
2927 : case OCcallvoid:
2928 : case OCcowvardyn:
2929 : case OCevalmnem:
2930 0 : return 1;
2931 : }
2932 0 : return 0;
2933 : }
2934 :
2935 : static void
2936 0 : closure_relink(GEN C, hashtable *table)
2937 : {
2938 0 : const char *code = closure_codestr(C);
2939 0 : GEN oper = closure_get_oper(C);
2940 0 : GEN fram = gel(closure_get_dbg(C),3);
2941 : long i, j;
2942 0 : for(i=1;i<lg(oper);i++)
2943 0 : if (oper[i] && opcode_need_relink((op_code)code[i]))
2944 0 : oper[i] = (long) hash_search(table,(void*) oper[i])->val;
2945 0 : for (i=1;i<lg(fram);i++)
2946 0 : for (j=1;j<lg(gel(fram,i));j++)
2947 0 : if (mael(fram,i,j))
2948 0 : mael(fram,i,j) = (long) hash_search(table,(void*) mael(fram,i,j))->val;
2949 0 : }
2950 :
2951 : void
2952 0 : gen_relink(GEN x, hashtable *table)
2953 : {
2954 0 : long i, lx, tx = typ(x);
2955 0 : switch(tx)
2956 : {
2957 0 : case t_CLOSURE:
2958 0 : closure_relink(x, table);
2959 0 : gen_relink(closure_get_data(x), table);
2960 0 : if (lg(x)==8) gen_relink(closure_get_frame(x), table);
2961 0 : break;
2962 0 : case t_LIST:
2963 0 : if (list_data(x)) gen_relink(list_data(x), table);
2964 0 : break;
2965 0 : case t_VEC: case t_COL: case t_MAT: case t_ERROR:
2966 0 : lx = lg(x);
2967 0 : for (i = lontyp[tx]; i < lx; i++) gen_relink(gel(x,i), table);
2968 : }
2969 0 : }
2970 :
2971 : static void
2972 0 : closure_unlink(GEN C)
2973 : {
2974 0 : const char *code = closure_codestr(C);
2975 0 : GEN oper = closure_get_oper(C);
2976 0 : GEN fram = gel(closure_get_dbg(C),3);
2977 : long i, j;
2978 0 : for(i=1;i<lg(oper);i++)
2979 0 : if (oper[i] && opcode_need_relink((op_code) code[i]))
2980 : {
2981 0 : long n = pari_stack_new(&s_relocs);
2982 0 : relocs[n] = (entree *) oper[i];
2983 : }
2984 0 : for (i=1;i<lg(fram);i++)
2985 0 : for (j=1;j<lg(gel(fram,i));j++)
2986 0 : if (mael(fram,i,j))
2987 : {
2988 0 : long n = pari_stack_new(&s_relocs);
2989 0 : relocs[n] = (entree *) mael(fram,i,j);
2990 : }
2991 0 : }
2992 :
2993 : static void
2994 16 : gen_unlink(GEN x)
2995 : {
2996 16 : long i, lx, tx = typ(x);
2997 16 : switch(tx)
2998 : {
2999 0 : case t_CLOSURE:
3000 0 : closure_unlink(x);
3001 0 : gen_unlink(closure_get_data(x));
3002 0 : if (lg(x)==8) gen_unlink(closure_get_frame(x));
3003 0 : break;
3004 4 : case t_LIST:
3005 4 : if (list_data(x)) gen_unlink(list_data(x));
3006 4 : break;
3007 0 : case t_VEC: case t_COL: case t_MAT: case t_ERROR:
3008 0 : lx = lg(x);
3009 0 : for (i = lontyp[tx]; i<lx; i++) gen_unlink(gel(x,i));
3010 : }
3011 16 : }
3012 :
3013 : GEN
3014 12 : copybin_unlink(GEN C)
3015 : {
3016 12 : long i, l , n, nold = s_relocs.n;
3017 : GEN v, w, V, res;
3018 12 : if (C)
3019 8 : gen_unlink(C);
3020 : else
3021 : { /* contents of all variables */
3022 4 : long v, maxv = pari_var_next();
3023 44 : for (v=0; v<maxv; v++)
3024 : {
3025 40 : entree *ep = varentries[v];
3026 40 : if (!ep || !ep->value) continue;
3027 8 : gen_unlink((GEN)ep->value);
3028 : }
3029 : }
3030 12 : n = s_relocs.n-nold;
3031 12 : v = cgetg(n+1, t_VECSMALL);
3032 12 : for(i=0; i<n; i++)
3033 0 : v[i+1] = (long) relocs[i];
3034 12 : s_relocs.n = nold;
3035 12 : w = vecsmall_uniq(v); l = lg(w);
3036 12 : res = cgetg(3,t_VEC);
3037 12 : V = cgetg(l, t_VEC);
3038 12 : for(i=1; i<l; i++)
3039 : {
3040 0 : entree *ep = (entree*) w[i];
3041 0 : gel(V,i) = strtoGENstr(ep->name);
3042 : }
3043 12 : gel(res,1) = vecsmall_copy(w);
3044 12 : gel(res,2) = V;
3045 12 : return res;
3046 : }
3047 :
3048 : /* e = t_VECSMALL of entree *ep [ addresses ],
3049 : * names = t_VEC of strtoGENstr(ep.names),
3050 : * Return hashtable : ep => is_entry(ep.name) */
3051 : hashtable *
3052 0 : hash_from_link(GEN e, GEN names, int use_stack)
3053 : {
3054 0 : long i, l = lg(e);
3055 0 : hashtable *h = hash_create_ulong(l-1, use_stack);
3056 0 : if (lg(names) != l) pari_err_DIM("hash_from_link");
3057 0 : for (i = 1; i < l; i++)
3058 : {
3059 0 : char *s = GSTR(gel(names,i));
3060 0 : hash_insert(h, (void*)e[i], (void*)fetch_entry(s));
3061 : }
3062 0 : return h;
3063 : }
3064 :
3065 : void
3066 0 : bincopy_relink(GEN C, GEN V)
3067 : {
3068 0 : pari_sp av = avma;
3069 0 : hashtable *table = hash_from_link(gel(V,1),gel(V,2),1);
3070 0 : gen_relink(C, table);
3071 0 : set_avma(av);
3072 0 : }
|
