Commit | Line | Data |
a0d0e21e |
1 | /* av.c |
79072805 |
2 | * |
9607fc9c |
3 | * Copyright (c) 1991-1997, Larry Wall |
79072805 |
4 | * |
5 | * You may distribute under the terms of either the GNU General Public |
6 | * License or the Artistic License, as specified in the README file. |
7 | * |
a0d0e21e |
8 | */ |
9 | |
10 | /* |
11 | * "...for the Entwives desired order, and plenty, and peace (by which they |
12 | * meant that things should remain where they had set them)." --Treebeard |
79072805 |
13 | */ |
14 | |
15 | #include "EXTERN.h" |
16 | #include "perl.h" |
17 | |
fb73857a |
18 | void |
8ac85365 |
19 | av_reify(AV *av) |
a0d0e21e |
20 | { |
21 | I32 key; |
22 | SV* sv; |
fb73857a |
23 | |
93965878 |
24 | if (AvREAL(av)) |
25 | return; |
26 | #ifdef DEBUGGING |
27 | if (SvRMAGICAL(av) && mg_find((SV*)av,'P')) |
28 | warn("av_reify called on tied array"); |
29 | #endif |
a0d0e21e |
30 | key = AvMAX(av) + 1; |
93965878 |
31 | while (key > AvFILLp(av) + 1) |
a0d0e21e |
32 | AvARRAY(av)[--key] = &sv_undef; |
33 | while (key) { |
34 | sv = AvARRAY(av)[--key]; |
35 | assert(sv); |
11343788 |
36 | if (sv != &sv_undef) { |
37 | dTHR; |
a0d0e21e |
38 | (void)SvREFCNT_inc(sv); |
11343788 |
39 | } |
a0d0e21e |
40 | } |
29de640a |
41 | key = AvARRAY(av) - AvALLOC(av); |
42 | while (key) |
43 | AvALLOC(av)[--key] = &sv_undef; |
a0d0e21e |
44 | AvREAL_on(av); |
45 | } |
46 | |
47 | void |
8ac85365 |
48 | av_extend(AV *av, I32 key) |
a0d0e21e |
49 | { |
11343788 |
50 | dTHR; /* only necessary if we have to extend stack */ |
93965878 |
51 | MAGIC *mg; |
52 | if (SvRMAGICAL(av) && (mg = mg_find((SV*)av,'P'))) { |
53 | dSP; |
54 | ENTER; |
55 | SAVETMPS; |
56 | PUSHMARK(sp); |
57 | EXTEND(sp,2); |
58 | PUSHs(mg->mg_obj); |
59 | PUSHs(sv_2mortal(newSViv(key))); |
60 | PUTBACK; |
61 | perl_call_method("EXTEND", G_SCALAR|G_DISCARD); |
62 | FREETMPS; |
63 | LEAVE; |
64 | return; |
65 | } |
a0d0e21e |
66 | if (key > AvMAX(av)) { |
67 | SV** ary; |
68 | I32 tmp; |
69 | I32 newmax; |
70 | |
71 | if (AvALLOC(av) != AvARRAY(av)) { |
93965878 |
72 | ary = AvALLOC(av) + AvFILLp(av) + 1; |
a0d0e21e |
73 | tmp = AvARRAY(av) - AvALLOC(av); |
93965878 |
74 | Move(AvARRAY(av), AvALLOC(av), AvFILLp(av)+1, SV*); |
a0d0e21e |
75 | AvMAX(av) += tmp; |
76 | SvPVX(av) = (char*)AvALLOC(av); |
77 | if (AvREAL(av)) { |
78 | while (tmp) |
79 | ary[--tmp] = &sv_undef; |
80 | } |
81 | |
82 | if (key > AvMAX(av) - 10) { |
83 | newmax = key + AvMAX(av); |
84 | goto resize; |
85 | } |
86 | } |
87 | else { |
88 | if (AvALLOC(av)) { |
c07a80fd |
89 | #ifndef STRANGE_MALLOC |
4633a7c4 |
90 | U32 bytes; |
c07a80fd |
91 | #endif |
4633a7c4 |
92 | |
a0d0e21e |
93 | newmax = key + AvMAX(av) / 5; |
94 | resize: |
4633a7c4 |
95 | #ifdef STRANGE_MALLOC |
a0d0e21e |
96 | Renew(AvALLOC(av),newmax+1, SV*); |
4633a7c4 |
97 | #else |
98 | bytes = (newmax + 1) * sizeof(SV*); |
99 | #define MALLOC_OVERHEAD 16 |
100 | tmp = MALLOC_OVERHEAD; |
101 | while (tmp - MALLOC_OVERHEAD < bytes) |
102 | tmp += tmp; |
103 | tmp -= MALLOC_OVERHEAD; |
104 | tmp /= sizeof(SV*); |
105 | assert(tmp > newmax); |
106 | newmax = tmp - 1; |
107 | New(2,ary, newmax+1, SV*); |
108 | Copy(AvALLOC(av), ary, AvMAX(av)+1, SV*); |
fba3b22e |
109 | if (AvMAX(av) > 64) |
110 | offer_nice_chunk(AvALLOC(av), (AvMAX(av)+1) * sizeof(SV*)); |
4633a7c4 |
111 | else |
112 | Safefree(AvALLOC(av)); |
113 | AvALLOC(av) = ary; |
114 | #endif |
a0d0e21e |
115 | ary = AvALLOC(av) + AvMAX(av) + 1; |
116 | tmp = newmax - AvMAX(av); |
7d55f622 |
117 | if (av == curstack) { /* Oops, grew stack (via av_store()?) */ |
a0d0e21e |
118 | stack_sp = AvALLOC(av) + (stack_sp - stack_base); |
119 | stack_base = AvALLOC(av); |
120 | stack_max = stack_base + newmax; |
121 | } |
122 | } |
123 | else { |
124 | newmax = key < 4 ? 4 : key; |
125 | New(2,AvALLOC(av), newmax+1, SV*); |
126 | ary = AvALLOC(av) + 1; |
127 | tmp = newmax; |
128 | AvALLOC(av)[0] = &sv_undef; /* For the stacks */ |
129 | } |
130 | if (AvREAL(av)) { |
131 | while (tmp) |
132 | ary[--tmp] = &sv_undef; |
133 | } |
134 | |
135 | SvPVX(av) = (char*)AvALLOC(av); |
136 | AvMAX(av) = newmax; |
137 | } |
138 | } |
139 | } |
140 | |
79072805 |
141 | SV** |
8ac85365 |
142 | av_fetch(register AV *av, I32 key, I32 lval) |
79072805 |
143 | { |
144 | SV *sv; |
145 | |
a0d0e21e |
146 | if (!av) |
147 | return 0; |
148 | |
93965878 |
149 | if (key < 0) { |
150 | key += AvFILL(av) + 1; |
151 | if (key < 0) |
152 | return 0; |
153 | } |
154 | |
8990e307 |
155 | if (SvRMAGICAL(av)) { |
463ee0b2 |
156 | if (mg_find((SV*)av,'P')) { |
11343788 |
157 | dTHR; |
8990e307 |
158 | sv = sv_newmortal(); |
463ee0b2 |
159 | mg_copy((SV*)av, sv, 0, key); |
463ee0b2 |
160 | Sv = sv; |
161 | return &Sv; |
162 | } |
163 | } |
164 | |
93965878 |
165 | if (key > AvFILLp(av)) { |
a0d0e21e |
166 | if (!lval) |
167 | return 0; |
168 | if (AvREALISH(av)) |
169 | sv = NEWSV(5,0); |
170 | else |
171 | sv = sv_newmortal(); |
172 | return av_store(av,key,sv); |
79072805 |
173 | } |
a0d0e21e |
174 | if (AvARRAY(av)[key] == &sv_undef) { |
4dbf4341 |
175 | emptyness: |
79072805 |
176 | if (lval) { |
177 | sv = NEWSV(6,0); |
463ee0b2 |
178 | return av_store(av,key,sv); |
79072805 |
179 | } |
180 | return 0; |
181 | } |
4dbf4341 |
182 | else if (AvREIFY(av) |
183 | && (!AvARRAY(av)[key] /* eg. @_ could have freed elts */ |
184 | || SvTYPE(AvARRAY(av)[key]) == SVTYPEMASK)) { |
185 | AvARRAY(av)[key] = &sv_undef; /* 1/2 reify */ |
186 | goto emptyness; |
187 | } |
463ee0b2 |
188 | return &AvARRAY(av)[key]; |
79072805 |
189 | } |
190 | |
191 | SV** |
8ac85365 |
192 | av_store(register AV *av, I32 key, SV *val) |
79072805 |
193 | { |
79072805 |
194 | SV** ary; |
93965878 |
195 | U32 fill; |
196 | |
79072805 |
197 | |
a0d0e21e |
198 | if (!av) |
199 | return 0; |
43fcc5d2 |
200 | if (!val) |
201 | val = &sv_undef; |
463ee0b2 |
202 | |
a0d0e21e |
203 | if (key < 0) { |
204 | key += AvFILL(av) + 1; |
205 | if (key < 0) |
206 | return 0; |
79072805 |
207 | } |
93965878 |
208 | |
43fcc5d2 |
209 | if (SvREADONLY(av) && key >= AvFILL(av)) |
210 | croak(no_modify); |
93965878 |
211 | |
212 | if (SvRMAGICAL(av)) { |
213 | if (mg_find((SV*)av,'P')) { |
214 | if (val != &sv_undef) { |
215 | mg_copy((SV*)av, val, 0, key); |
216 | } |
217 | return 0; |
218 | } |
219 | } |
220 | |
49beac48 |
221 | if (!AvREAL(av) && AvREIFY(av)) |
a0d0e21e |
222 | av_reify(av); |
a0d0e21e |
223 | if (key > AvMAX(av)) |
224 | av_extend(av,key); |
463ee0b2 |
225 | ary = AvARRAY(av); |
93965878 |
226 | if (AvFILLp(av) < key) { |
a0d0e21e |
227 | if (!AvREAL(av)) { |
11343788 |
228 | dTHR; |
7d55f622 |
229 | if (av == curstack && key > stack_sp - stack_base) |
a0d0e21e |
230 | stack_sp = stack_base + key; /* XPUSH in disguise */ |
231 | do |
93965878 |
232 | ary[++AvFILLp(av)] = &sv_undef; |
233 | while (AvFILLp(av) < key); |
79072805 |
234 | } |
93965878 |
235 | AvFILLp(av) = key; |
79072805 |
236 | } |
a0d0e21e |
237 | else if (AvREAL(av)) |
238 | SvREFCNT_dec(ary[key]); |
79072805 |
239 | ary[key] = val; |
8990e307 |
240 | if (SvSMAGICAL(av)) { |
a0d0e21e |
241 | if (val != &sv_undef) { |
242 | MAGIC* mg = SvMAGIC(av); |
243 | sv_magic(val, (SV*)av, toLOWER(mg->mg_type), 0, key); |
244 | } |
463ee0b2 |
245 | mg_set((SV*)av); |
246 | } |
79072805 |
247 | return &ary[key]; |
248 | } |
249 | |
250 | AV * |
8ac85365 |
251 | newAV(void) |
79072805 |
252 | { |
463ee0b2 |
253 | register AV *av; |
79072805 |
254 | |
a0d0e21e |
255 | av = (AV*)NEWSV(3,0); |
256 | sv_upgrade((SV *)av, SVt_PVAV); |
463ee0b2 |
257 | AvREAL_on(av); |
258 | AvALLOC(av) = 0; |
259 | SvPVX(av) = 0; |
93965878 |
260 | AvMAX(av) = AvFILLp(av) = -1; |
463ee0b2 |
261 | return av; |
79072805 |
262 | } |
263 | |
264 | AV * |
8ac85365 |
265 | av_make(register I32 size, register SV **strp) |
79072805 |
266 | { |
463ee0b2 |
267 | register AV *av; |
79072805 |
268 | register I32 i; |
269 | register SV** ary; |
270 | |
a0d0e21e |
271 | av = (AV*)NEWSV(8,0); |
272 | sv_upgrade((SV *) av,SVt_PVAV); |
a0d0e21e |
273 | AvFLAGS(av) = AVf_REAL; |
573fa4ea |
274 | if (size) { /* `defined' was returning undef for size==0 anyway. */ |
275 | New(4,ary,size,SV*); |
276 | AvALLOC(av) = ary; |
277 | SvPVX(av) = (char*)ary; |
93965878 |
278 | AvFILLp(av) = size - 1; |
573fa4ea |
279 | AvMAX(av) = size - 1; |
280 | for (i = 0; i < size; i++) { |
281 | assert (*strp); |
282 | ary[i] = NEWSV(7,0); |
283 | sv_setsv(ary[i], *strp); |
284 | strp++; |
285 | } |
79072805 |
286 | } |
463ee0b2 |
287 | return av; |
79072805 |
288 | } |
289 | |
290 | AV * |
8ac85365 |
291 | av_fake(register I32 size, register SV **strp) |
79072805 |
292 | { |
463ee0b2 |
293 | register AV *av; |
79072805 |
294 | register SV** ary; |
295 | |
a0d0e21e |
296 | av = (AV*)NEWSV(9,0); |
297 | sv_upgrade((SV *)av, SVt_PVAV); |
79072805 |
298 | New(4,ary,size+1,SV*); |
463ee0b2 |
299 | AvALLOC(av) = ary; |
79072805 |
300 | Copy(strp,ary,size,SV*); |
a0d0e21e |
301 | AvFLAGS(av) = AVf_REIFY; |
463ee0b2 |
302 | SvPVX(av) = (char*)ary; |
93965878 |
303 | AvFILLp(av) = size - 1; |
463ee0b2 |
304 | AvMAX(av) = size - 1; |
79072805 |
305 | while (size--) { |
a0d0e21e |
306 | assert (*strp); |
307 | SvTEMP_off(*strp); |
79072805 |
308 | strp++; |
309 | } |
463ee0b2 |
310 | return av; |
79072805 |
311 | } |
312 | |
313 | void |
8ac85365 |
314 | av_clear(register AV *av) |
79072805 |
315 | { |
316 | register I32 key; |
a0d0e21e |
317 | SV** ary; |
79072805 |
318 | |
7d55f622 |
319 | #ifdef DEBUGGING |
320 | if (SvREFCNT(av) <= 0) { |
321 | warn("Attempt to clear deleted array"); |
322 | } |
323 | #endif |
a0d0e21e |
324 | if (!av || AvMAX(av) < 0) |
79072805 |
325 | return; |
326 | /*SUPPRESS 560*/ |
a0d0e21e |
327 | |
93965878 |
328 | /* Give any tie a chance to cleanup first */ |
329 | if (SvRMAGICAL(av)) |
330 | mg_clear((SV*)av); |
331 | |
a0d0e21e |
332 | if (AvREAL(av)) { |
333 | ary = AvARRAY(av); |
93965878 |
334 | key = AvFILLp(av) + 1; |
a0d0e21e |
335 | while (key) { |
336 | SvREFCNT_dec(ary[--key]); |
337 | ary[key] = &sv_undef; |
338 | } |
339 | } |
463ee0b2 |
340 | if (key = AvARRAY(av) - AvALLOC(av)) { |
341 | AvMAX(av) += key; |
a0d0e21e |
342 | SvPVX(av) = (char*)AvALLOC(av); |
79072805 |
343 | } |
93965878 |
344 | AvFILLp(av) = -1; |
fb73857a |
345 | |
79072805 |
346 | } |
347 | |
348 | void |
8ac85365 |
349 | av_undef(register AV *av) |
79072805 |
350 | { |
351 | register I32 key; |
352 | |
463ee0b2 |
353 | if (!av) |
79072805 |
354 | return; |
355 | /*SUPPRESS 560*/ |
93965878 |
356 | |
357 | /* Give any tie a chance to cleanup first */ |
358 | if (SvRMAGICAL(av) && mg_find((SV*)av,'P')) |
359 | av_fill(av, -1); /* mg_clear() ? */ |
360 | |
a0d0e21e |
361 | if (AvREAL(av)) { |
93965878 |
362 | key = AvFILLp(av) + 1; |
a0d0e21e |
363 | while (key) |
364 | SvREFCNT_dec(AvARRAY(av)[--key]); |
365 | } |
463ee0b2 |
366 | Safefree(AvALLOC(av)); |
1393e206 |
367 | AvARRAY(av) = 0; |
463ee0b2 |
368 | AvALLOC(av) = 0; |
369 | SvPVX(av) = 0; |
93965878 |
370 | AvMAX(av) = AvFILLp(av) = -1; |
748a9306 |
371 | if (AvARYLEN(av)) { |
372 | SvREFCNT_dec(AvARYLEN(av)); |
373 | AvARYLEN(av) = 0; |
374 | } |
79072805 |
375 | } |
376 | |
a0d0e21e |
377 | void |
8ac85365 |
378 | av_push(register AV *av, SV *val) |
93965878 |
379 | { |
380 | MAGIC *mg; |
a0d0e21e |
381 | if (!av) |
382 | return; |
93965878 |
383 | if (SvREADONLY(av)) |
384 | croak(no_modify); |
385 | |
386 | if (SvRMAGICAL(av) && (mg = mg_find((SV*)av,'P'))) { |
387 | dSP; |
388 | PUSHMARK(sp); |
389 | EXTEND(sp,2); |
390 | PUSHs(mg->mg_obj); |
391 | PUSHs(val); |
392 | PUTBACK; |
393 | perl_call_method("PUSH", G_SCALAR|G_DISCARD); |
394 | return; |
395 | } |
396 | av_store(av,AvFILLp(av)+1,val); |
79072805 |
397 | } |
398 | |
399 | SV * |
8ac85365 |
400 | av_pop(register AV *av) |
79072805 |
401 | { |
402 | SV *retval; |
93965878 |
403 | MAGIC* mg; |
79072805 |
404 | |
a0d0e21e |
405 | if (!av || AvFILL(av) < 0) |
406 | return &sv_undef; |
43fcc5d2 |
407 | if (SvREADONLY(av)) |
408 | croak(no_modify); |
93965878 |
409 | if (SvRMAGICAL(av) && (mg = mg_find((SV*)av,'P'))) { |
410 | dSP; |
411 | PUSHMARK(sp); |
412 | XPUSHs(mg->mg_obj); |
413 | PUTBACK; |
414 | if (perl_call_method("POP", G_SCALAR)) { |
415 | retval = newSVsv(*stack_sp--); |
416 | } else { |
417 | retval = &sv_undef; |
418 | } |
419 | return retval; |
420 | } |
421 | retval = AvARRAY(av)[AvFILLp(av)]; |
422 | AvARRAY(av)[AvFILLp(av)--] = &sv_undef; |
8990e307 |
423 | if (SvSMAGICAL(av)) |
463ee0b2 |
424 | mg_set((SV*)av); |
79072805 |
425 | return retval; |
426 | } |
427 | |
428 | void |
8ac85365 |
429 | av_unshift(register AV *av, register I32 num) |
79072805 |
430 | { |
431 | register I32 i; |
432 | register SV **sstr,**dstr; |
93965878 |
433 | MAGIC* mg; |
79072805 |
434 | |
a0d0e21e |
435 | if (!av || num <= 0) |
79072805 |
436 | return; |
43fcc5d2 |
437 | if (SvREADONLY(av)) |
438 | croak(no_modify); |
93965878 |
439 | |
440 | if (SvRMAGICAL(av) && (mg = mg_find((SV*)av,'P'))) { |
441 | dSP; |
442 | PUSHMARK(sp); |
443 | EXTEND(sp,1+num); |
444 | PUSHs(mg->mg_obj); |
445 | while (num-- > 0) { |
446 | PUSHs(&sv_undef); |
447 | } |
448 | PUTBACK; |
449 | perl_call_method("UNSHIFT", G_SCALAR|G_DISCARD); |
450 | return; |
451 | } |
452 | |
49beac48 |
453 | if (!AvREAL(av) && AvREIFY(av)) |
454 | av_reify(av); |
a0d0e21e |
455 | i = AvARRAY(av) - AvALLOC(av); |
456 | if (i) { |
457 | if (i > num) |
458 | i = num; |
459 | num -= i; |
460 | |
461 | AvMAX(av) += i; |
93965878 |
462 | AvFILLp(av) += i; |
a0d0e21e |
463 | SvPVX(av) = (char*)(AvARRAY(av) - i); |
464 | } |
465 | if (num) { |
93965878 |
466 | av_extend(av,AvFILLp(av)+num); |
467 | AvFILLp(av) += num; |
468 | dstr = AvARRAY(av) + AvFILLp(av); |
79072805 |
469 | sstr = dstr - num; |
470 | #ifdef BUGGY_MSC5 |
471 | # pragma loop_opt(off) /* don't loop-optimize the following code */ |
472 | #endif /* BUGGY_MSC5 */ |
93965878 |
473 | for (i = AvFILLp(av) - num; i >= 0; --i) { |
79072805 |
474 | *dstr-- = *sstr--; |
475 | #ifdef BUGGY_MSC5 |
476 | # pragma loop_opt() /* loop-optimization back to command-line setting */ |
477 | #endif /* BUGGY_MSC5 */ |
478 | } |
a0d0e21e |
479 | while (num) |
480 | AvARRAY(av)[--num] = &sv_undef; |
79072805 |
481 | } |
482 | } |
483 | |
484 | SV * |
8ac85365 |
485 | av_shift(register AV *av) |
79072805 |
486 | { |
487 | SV *retval; |
93965878 |
488 | MAGIC* mg; |
79072805 |
489 | |
a0d0e21e |
490 | if (!av || AvFILL(av) < 0) |
491 | return &sv_undef; |
43fcc5d2 |
492 | if (SvREADONLY(av)) |
493 | croak(no_modify); |
93965878 |
494 | if (SvRMAGICAL(av) && (mg = mg_find((SV*)av,'P'))) { |
495 | dSP; |
496 | PUSHMARK(sp); |
497 | XPUSHs(mg->mg_obj); |
498 | PUTBACK; |
499 | if (perl_call_method("SHIFT", G_SCALAR)) { |
500 | retval = newSVsv(*stack_sp--); |
501 | } else { |
502 | retval = &sv_undef; |
503 | } |
504 | return retval; |
505 | } |
463ee0b2 |
506 | retval = *AvARRAY(av); |
a0d0e21e |
507 | if (AvREAL(av)) |
508 | *AvARRAY(av) = &sv_undef; |
463ee0b2 |
509 | SvPVX(av) = (char*)(AvARRAY(av) + 1); |
510 | AvMAX(av)--; |
93965878 |
511 | AvFILLp(av)--; |
8990e307 |
512 | if (SvSMAGICAL(av)) |
463ee0b2 |
513 | mg_set((SV*)av); |
79072805 |
514 | return retval; |
515 | } |
516 | |
517 | I32 |
8ac85365 |
518 | av_len(register AV *av) |
79072805 |
519 | { |
463ee0b2 |
520 | return AvFILL(av); |
79072805 |
521 | } |
522 | |
523 | void |
8ac85365 |
524 | av_fill(register AV *av, I32 fill) |
79072805 |
525 | { |
93965878 |
526 | MAGIC *mg; |
a0d0e21e |
527 | if (!av) |
528 | croak("panic: null array"); |
79072805 |
529 | if (fill < 0) |
530 | fill = -1; |
93965878 |
531 | if (SvRMAGICAL(av) && (mg = mg_find((SV*)av,'P'))) { |
532 | dSP; |
533 | ENTER; |
534 | SAVETMPS; |
535 | PUSHMARK(sp); |
536 | EXTEND(sp,2); |
537 | PUSHs(mg->mg_obj); |
538 | PUSHs(sv_2mortal(newSViv(fill))); |
539 | PUTBACK; |
540 | perl_call_method("STORESIZE", G_SCALAR|G_DISCARD); |
541 | FREETMPS; |
542 | LEAVE; |
543 | return; |
544 | } |
463ee0b2 |
545 | if (fill <= AvMAX(av)) { |
93965878 |
546 | I32 key = AvFILLp(av); |
a0d0e21e |
547 | SV** ary = AvARRAY(av); |
548 | |
549 | if (AvREAL(av)) { |
550 | while (key > fill) { |
551 | SvREFCNT_dec(ary[key]); |
552 | ary[key--] = &sv_undef; |
553 | } |
554 | } |
555 | else { |
556 | while (key < fill) |
557 | ary[++key] = &sv_undef; |
558 | } |
559 | |
93965878 |
560 | AvFILLp(av) = fill; |
8990e307 |
561 | if (SvSMAGICAL(av)) |
463ee0b2 |
562 | mg_set((SV*)av); |
563 | } |
a0d0e21e |
564 | else |
565 | (void)av_store(av,fill,&sv_undef); |
79072805 |
566 | } |
c750a3ec |
567 | |
5d5aaa5e |
568 | |
569 | HV* |
570 | avhv_keys(AV *av) |
571 | { |
572 | SV **keysp; |
573 | HV *keys = Nullhv; |
574 | |
575 | keysp = av_fetch(av, 0, FALSE); |
576 | if (keysp) { |
d627ae4e |
577 | SV *sv = *keysp; |
578 | if (SvGMAGICAL(sv)) |
579 | mg_get(sv); |
580 | if (SvROK(sv)) { |
581 | sv = SvRV(sv); |
582 | if (SvTYPE(sv) == SVt_PVHV) |
583 | keys = (HV*)sv; |
5d5aaa5e |
584 | } |
585 | } |
586 | if (!keys) |
587 | croak("Can't coerce array into hash"); |
588 | return keys; |
589 | } |
590 | |
c750a3ec |
591 | SV** |
8ac85365 |
592 | avhv_fetch(AV *av, char *key, U32 klen, I32 lval) |
c750a3ec |
593 | { |
5d5aaa5e |
594 | SV **indsvp; |
595 | HV *keys = avhv_keys(av); |
c750a3ec |
596 | I32 ind; |
597 | |
5d5aaa5e |
598 | indsvp = hv_fetch(keys, key, klen, FALSE); |
c750a3ec |
599 | if (indsvp) { |
600 | ind = SvIV(*indsvp); |
601 | if (ind < 1) |
602 | croak("Bad index while coercing array into hash"); |
603 | } else { |
604 | if (!lval) |
605 | return 0; |
606 | |
607 | ind = AvFILL(av) + 1; |
5d5aaa5e |
608 | hv_store(keys, key, klen, newSViv(ind), 0); |
c750a3ec |
609 | } |
610 | return av_fetch(av, ind, lval); |
611 | } |
612 | |
613 | SV** |
8ac85365 |
614 | avhv_fetch_ent(AV *av, SV *keysv, I32 lval, U32 hash) |
97fcbf96 |
615 | { |
5d5aaa5e |
616 | SV **indsvp; |
617 | HV *keys = avhv_keys(av); |
97fcbf96 |
618 | HE *he; |
619 | I32 ind; |
620 | |
5d5aaa5e |
621 | he = hv_fetch_ent(keys, keysv, FALSE, hash); |
97fcbf96 |
622 | if (he) { |
623 | ind = SvIV(HeVAL(he)); |
624 | if (ind < 1) |
625 | croak("Bad index while coercing array into hash"); |
626 | } else { |
627 | if (!lval) |
628 | return 0; |
629 | |
630 | ind = AvFILL(av) + 1; |
5d5aaa5e |
631 | hv_store_ent(keys, keysv, newSViv(ind), 0); |
97fcbf96 |
632 | } |
633 | return av_fetch(av, ind, lval); |
634 | } |
635 | |
636 | SV** |
8ac85365 |
637 | avhv_store(AV *av, char *key, U32 klen, SV *val, U32 hash) |
c750a3ec |
638 | { |
5d5aaa5e |
639 | SV **indsvp; |
640 | HV *keys = avhv_keys(av); |
c750a3ec |
641 | I32 ind; |
642 | |
5d5aaa5e |
643 | indsvp = hv_fetch(keys, key, klen, FALSE); |
c750a3ec |
644 | if (indsvp) { |
645 | ind = SvIV(*indsvp); |
646 | if (ind < 1) |
647 | croak("Bad index while coercing array into hash"); |
648 | } else { |
649 | ind = AvFILL(av) + 1; |
5d5aaa5e |
650 | hv_store(keys, key, klen, newSViv(ind), hash); |
c750a3ec |
651 | } |
652 | return av_store(av, ind, val); |
653 | } |
654 | |
5bc6513d |
655 | SV** |
8ac85365 |
656 | avhv_store_ent(AV *av, SV *keysv, SV *val, U32 hash) |
5bc6513d |
657 | { |
5d5aaa5e |
658 | HV *keys = avhv_keys(av); |
5bc6513d |
659 | HE *he; |
660 | I32 ind; |
661 | |
5d5aaa5e |
662 | he = hv_fetch_ent(keys, keysv, FALSE, hash); |
5bc6513d |
663 | if (he) { |
664 | ind = SvIV(HeVAL(he)); |
665 | if (ind < 1) |
666 | croak("Bad index while coercing array into hash"); |
667 | } else { |
668 | ind = AvFILL(av) + 1; |
5d5aaa5e |
669 | hv_store_ent(keys, keysv, newSViv(ind), hash); |
5bc6513d |
670 | } |
671 | return av_store(av, ind, val); |
672 | } |
673 | |
c750a3ec |
674 | bool |
8ac85365 |
675 | avhv_exists_ent(AV *av, SV *keysv, U32 hash) |
97fcbf96 |
676 | { |
5d5aaa5e |
677 | HV *keys = avhv_keys(av); |
678 | return hv_exists_ent(keys, keysv, hash); |
97fcbf96 |
679 | } |
680 | |
681 | bool |
8ac85365 |
682 | avhv_exists(AV *av, char *key, U32 klen) |
c750a3ec |
683 | { |
5d5aaa5e |
684 | HV *keys = avhv_keys(av); |
685 | return hv_exists(keys, key, klen); |
c750a3ec |
686 | } |
687 | |
688 | /* avhv_delete leaks. Caller can re-index and compress if so desired. */ |
689 | SV * |
8ac85365 |
690 | avhv_delete(AV *av, char *key, U32 klen, I32 flags) |
c750a3ec |
691 | { |
5d5aaa5e |
692 | HV *keys = avhv_keys(av); |
c750a3ec |
693 | SV *sv; |
694 | SV **svp; |
695 | I32 ind; |
696 | |
5d5aaa5e |
697 | sv = hv_delete(keys, key, klen, 0); |
c750a3ec |
698 | if (!sv) |
699 | return Nullsv; |
700 | ind = SvIV(sv); |
701 | if (ind < 1) |
702 | croak("Bad index while coercing array into hash"); |
703 | svp = av_fetch(av, ind, FALSE); |
704 | if (!svp) |
705 | return Nullsv; |
706 | if (flags & G_DISCARD) { |
707 | sv = Nullsv; |
708 | SvREFCNT_dec(*svp); |
709 | } else { |
710 | sv = sv_2mortal(*svp); |
711 | } |
712 | *svp = &sv_undef; |
713 | return sv; |
714 | } |
715 | |
97fcbf96 |
716 | /* avhv_delete_ent leaks. Caller can re-index and compress if so desired. */ |
717 | SV * |
8ac85365 |
718 | avhv_delete_ent(AV *av, SV *keysv, I32 flags, U32 hash) |
97fcbf96 |
719 | { |
5d5aaa5e |
720 | HV *keys = avhv_keys(av); |
97fcbf96 |
721 | SV *sv; |
722 | SV **svp; |
723 | I32 ind; |
724 | |
5d5aaa5e |
725 | sv = hv_delete_ent(keys, keysv, 0, hash); |
97fcbf96 |
726 | if (!sv) |
727 | return Nullsv; |
728 | ind = SvIV(sv); |
729 | if (ind < 1) |
730 | croak("Bad index while coercing array into hash"); |
731 | svp = av_fetch(av, ind, FALSE); |
732 | if (!svp) |
733 | return Nullsv; |
734 | if (flags & G_DISCARD) { |
735 | sv = Nullsv; |
736 | SvREFCNT_dec(*svp); |
737 | } else { |
738 | sv = sv_2mortal(*svp); |
739 | } |
740 | *svp = &sv_undef; |
741 | return sv; |
742 | } |
743 | |
c750a3ec |
744 | I32 |
8ac85365 |
745 | avhv_iterinit(AV *av) |
c750a3ec |
746 | { |
5d5aaa5e |
747 | HV *keys = avhv_keys(av); |
748 | return hv_iterinit(keys); |
c750a3ec |
749 | } |
750 | |
751 | HE * |
8ac85365 |
752 | avhv_iternext(AV *av) |
c750a3ec |
753 | { |
5d5aaa5e |
754 | HV *keys = avhv_keys(av); |
755 | return hv_iternext(keys); |
c750a3ec |
756 | } |
757 | |
758 | SV * |
8ac85365 |
759 | avhv_iterval(AV *av, register HE *entry) |
c750a3ec |
760 | { |
5d5aaa5e |
761 | HV *keys = avhv_keys(av); |
c750a3ec |
762 | SV *sv; |
763 | I32 ind; |
764 | |
5d5aaa5e |
765 | sv = hv_iterval(keys, entry); |
c750a3ec |
766 | ind = SvIV(sv); |
767 | if (ind < 1) |
768 | croak("Bad index while coercing array into hash"); |
769 | return *av_fetch(av, ind, TRUE); |
770 | } |
771 | |
772 | SV * |
8ac85365 |
773 | avhv_iternextsv(AV *av, char **key, I32 *retlen) |
c750a3ec |
774 | { |
5d5aaa5e |
775 | HV *keys = avhv_keys(av); |
c750a3ec |
776 | HE *he; |
777 | SV *sv; |
778 | I32 ind; |
779 | |
5d5aaa5e |
780 | he = hv_iternext(keys); |
781 | if (!he) |
782 | return Nullsv; |
c750a3ec |
783 | *key = hv_iterkey(he, retlen); |
5d5aaa5e |
784 | sv = hv_iterval(keys, he); |
c750a3ec |
785 | ind = SvIV(sv); |
786 | if (ind < 1) |
787 | croak("Bad index while coercing array into hash"); |
788 | return *av_fetch(av, ind, TRUE); |
789 | } |