*** empty log message ***
[bpt/guile.git] / libguile / ramap.c
CommitLineData
950cc72b 1/* Copyright (C) 1996, 1998, 2000 Free Software Foundation, Inc.
0f2d19dd
JB
2 *
3 * This program is free software; you can redistribute it and/or modify
4 * it under the terms of the GNU General Public License as published by
5 * the Free Software Foundation; either version 2, or (at your option)
6 * any later version.
7 *
8 * This program is distributed in the hope that it will be useful,
9 * but WITHOUT ANY WARRANTY; without even the implied warranty of
10 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
11 * GNU General Public License for more details.
12 *
13 * You should have received a copy of the GNU General Public License
14 * along with this software; see the file COPYING. If not, write to
82892bed
JB
15 * the Free Software Foundation, Inc., 59 Temple Place, Suite 330,
16 * Boston, MA 02111-1307 USA
0f2d19dd
JB
17 *
18 * As a special exception, the Free Software Foundation gives permission
19 * for additional uses of the text contained in its release of GUILE.
20 *
21 * The exception is that, if you link the GUILE library with other files
22 * to produce an executable, this does not by itself cause the
23 * resulting executable to be covered by the GNU General Public License.
24 * Your use of that executable is in no way restricted on account of
25 * linking the GUILE library code into it.
26 *
27 * This exception does not however invalidate any other reasons why
28 * the executable file might be covered by the GNU General Public License.
29 *
30 * This exception applies only to the code released by the
31 * Free Software Foundation under the name GUILE. If you copy
32 * code from other Free Software Foundation releases into a copy of
33 * GUILE, as the General Public License permits, the exception does
34 * not apply to the code that you add in this way. To avoid misleading
35 * anyone as to the status of such modified files, you must delete
36 * this exception notice from them.
37 *
38 * If you write modifications of your own for GUILE, it is your choice
39 * whether to permit this exception to apply to your modifications.
82892bed 40 * If you do not wish that, delete this exception notice. */
1bbd0b84
GB
41
42/* Software engineering face-lift by Greg J. Badros, 11-Dec-1999,
43 gjb@cs.washington.edu, http://www.cs.washington.edu/homes/gjb */
44
c209c88e
GB
45/*
46 HWN:FIXME::
47 Someone should rename this to arraymap.c; that would reflect the
48 contents better. */
0f2d19dd
JB
49\f
50
51
52\f
53
54#include <stdio.h>
55#include "_scm.h"
20e6290e
JB
56#include "unif.h"
57#include "smob.h"
58#include "chars.h"
59#include "eq.h"
60#include "eval.h"
61#include "feature.h"
ba11fd4c 62#include "root.h"
003d1fd0 63#include "vectors.h"
0f2d19dd 64
b6791b2e 65#include "validate.h"
20e6290e 66#include "ramap.h"
0f2d19dd
JB
67\f
68
0f2d19dd
JB
69typedef struct
70{
71 char *name;
72 SCM sproc;
73 int (*vproc) ();
74} ra_iproc;
75
ad310508
MD
76
77/* These tables are a kluge that will not scale well when more
78 * vectorized subrs are added. It is tempting to steal some bits from
79 * the SCM_CAR of all subrs (like those selected by SCM_SMOBNUM) to hold an
80 * offset into a table of vectorized subrs.
81 */
82
83static ra_iproc ra_rpsubrs[] =
84{
85 {"=", SCM_UNDEFINED, scm_ra_eqp},
86 {"<", SCM_UNDEFINED, scm_ra_lessp},
87 {"<=", SCM_UNDEFINED, scm_ra_leqp},
88 {">", SCM_UNDEFINED, scm_ra_grp},
89 {">=", SCM_UNDEFINED, scm_ra_greqp},
90 {0, 0, 0}
91};
92
93static ra_iproc ra_asubrs[] =
94{
95 {"+", SCM_UNDEFINED, scm_ra_sum},
96 {"-", SCM_UNDEFINED, scm_ra_difference},
97 {"*", SCM_UNDEFINED, scm_ra_product},
98 {"/", SCM_UNDEFINED, scm_ra_divide},
99 {0, 0, 0}
100};
101
0f2d19dd 102
0f2d19dd
JB
103
104/* Fast, recycling scm_vector ref */
105#define RVREF(ra, i, e) (e = scm_cvref(ra, i, e))
106
107/* #define RVREF(ra, i, e) (scm_cvref(ra, i, SCM_UNDEFINED)) to turn off */
108
109/* IVDEP means "ignore scm_vector dependencies", meaning we guarantee that
110 elements of scm_vector operands are not aliased */
111#ifdef _UNICOS
112#define IVDEP(test, line) if (test) {_Pragma("ivdep"); line} else {line}
113#else
114#define IVDEP(test, line) line
115#endif
116
117\f
118
119/* inds must be a uvect or ivect, no check. */
120
1cc91f1b 121
c209c88e
GB
122
123/*
124 Yes, this is really ugly, but it prevents multiple code
125 */
126#define BINARY_ELTS_CODE(OPERATOR, type) \
127do { type *v0 = (type*)SCM_VELTS (ra0);\
128 type *v1 = (type*)SCM_VELTS (ra1);\
129 IVDEP (ra0 != ra1, \
130 for (; n-- > 0; i0 += inc0, i1 += inc1) \
131 v0[i0] OPERATOR v1[i1];) \
132 break; \
133} while (0)
134
135/* This macro is used for all but binary division and
136 multiplication of complex numbers -- see the expanded
137 version in the functions later in this file */
138#define BINARY_PAIR_ELTS_CODE(OPERATOR, type) \
139do { type (*v0)[2] = (type (*)[2]) SCM_VELTS (ra0);\
140 type (*v1)[2] = (type (*)[2]) SCM_VELTS (ra1);\
141 IVDEP (ra0 != ra1, \
142 for (; n-- > 0; i0 += inc0, i1 += inc1) {\
143 v0[i0][0] OPERATOR v1[i1][0]; \
144 v0[i0][1] OPERATOR v1[i1][1]; \
145 }) \
146 break; \
147} while (0)
148
149#define UNARY_ELTS_CODE(OPERATOR, type) \
150 do { type *v0 = (type *) SCM_VELTS (ra0);\
151 for (; n-- > 0; i0 += inc0) \
152 v0[i0] OPERATOR v0[i0];\
153 break;\
154 } while (0)
155
156
157/* This macro is used for all but unary divison
158 of complex numbers -- see the expanded version in the
159 function later in this file. */
160#define UNARY_PAIR_ELTS_CODE(OPERATOR, type) \
161 do { type (*v0)[2] = (type (*)[2]) SCM_VELTS (ra0);\
162 for (; n-- > 0; i0 += inc0) {\
163 v0[i0][0] OPERATOR v0[i0][0];\
164 v0[i0][1] OPERATOR v0[i0][1];\
165 }\
166 break;\
167 } while (0)
168
0f2d19dd 169static scm_sizet
1bbd0b84 170cind (SCM ra, SCM inds)
0f2d19dd
JB
171{
172 scm_sizet i;
173 int k;
c209c88e 174 long *ve = (long*) SCM_VELTS (inds);
0f2d19dd
JB
175 if (!SCM_ARRAYP (ra))
176 return *ve;
177 i = SCM_ARRAY_BASE (ra);
178 for (k = 0; k < SCM_ARRAY_NDIM (ra); k++)
179 i += (ve[k] - SCM_ARRAY_DIMS (ra)[k].lbnd) * SCM_ARRAY_DIMS (ra)[k].inc;
180 return i;
181}
182
183
184/* Checker for scm_array mapping functions:
185 return values: 4 --> shapes, increments, and bases are the same;
186 3 --> shapes and increments are the same;
187 2 --> shapes are the same;
188 1 --> ras are at least as big as ra0;
189 0 --> no match.
190 */
1cc91f1b 191
0f2d19dd 192int
6e8d25a6 193scm_ra_matchp (SCM ra0, SCM ras)
0f2d19dd
JB
194{
195 SCM ra1;
196 scm_array_dim dims;
197 scm_array_dim *s0 = &dims;
198 scm_array_dim *s1;
199 scm_sizet bas0 = 0;
200 int i, ndim = 1;
201 int exact = 2 /* 4 */ ; /* Don't care about values >2 (yet?) */
ff467021 202 if (SCM_IMP (ra0)) return 0;
0f2d19dd 203 switch (SCM_TYP7 (ra0))
c209c88e
GB
204 {
205 default:
206 return 0;
207 case scm_tc7_vector:
208 case scm_tc7_wvect:
209 case scm_tc7_string:
210 case scm_tc7_byvect:
211 case scm_tc7_bvect:
212 case scm_tc7_uvect:
213 case scm_tc7_ivect:
214 case scm_tc7_svect:
5c11cc9d 215#ifdef HAVE_LONG_LONGS
c209c88e 216 case scm_tc7_llvect:
05f92e0f 217#endif
c209c88e
GB
218 case scm_tc7_fvect:
219 case scm_tc7_dvect:
220 case scm_tc7_cvect:
221 s0->lbnd = 0;
222 s0->inc = 1;
223 s0->ubnd = (long) SCM_LENGTH (ra0) - 1;
224 break;
225 case scm_tc7_smob:
226 if (!SCM_ARRAYP (ra0))
227 return 0;
228 ndim = SCM_ARRAY_NDIM (ra0);
229 s0 = SCM_ARRAY_DIMS (ra0);
230 bas0 = SCM_ARRAY_BASE (ra0);
231 break;
232 }
368cf54d 233 while (SCM_NIMP (ras))
c209c88e
GB
234 {
235 ra1 = SCM_CAR (ras);
236 if (SCM_IMP (ra1))
237 return 0;
238 switch SCM_TYP7
239 (ra1)
240 {
241 default:
242 return 0;
243 case scm_tc7_vector:
244 case scm_tc7_wvect:
245 case scm_tc7_string:
246 case scm_tc7_byvect:
247 case scm_tc7_bvect:
248 case scm_tc7_uvect:
249 case scm_tc7_ivect:
250 case scm_tc7_svect:
5c11cc9d 251#ifdef HAVE_LONG_LONGS
c209c88e 252 case scm_tc7_llvect:
05f92e0f 253#endif
c209c88e
GB
254 case scm_tc7_fvect:
255 case scm_tc7_dvect:
256 case scm_tc7_cvect:
257 if (1 != ndim)
258 return 0;
259 switch (exact)
260 {
261 case 4:
262 if (0 != bas0)
0f2d19dd 263 exact = 3;
c209c88e
GB
264 case 3:
265 if (1 != s0->inc)
266 exact = 2;
267 case 2:
268 if ((0 == s0->lbnd) && (s0->ubnd == SCM_LENGTH (ra1) - 1))
269 break;
270 exact = 1;
271 case 1:
272 if (s0->lbnd < 0 || s0->ubnd >= SCM_LENGTH (ra1))
273 return 0;
0f2d19dd 274 }
c209c88e
GB
275 break;
276 case scm_tc7_smob:
277 if (!SCM_ARRAYP (ra1) || ndim != SCM_ARRAY_NDIM (ra1))
278 return 0;
279 s1 = SCM_ARRAY_DIMS (ra1);
280 if (bas0 != SCM_ARRAY_BASE (ra1))
281 exact = 3;
282 for (i = 0; i < ndim; i++)
283 switch (exact)
284 {
285 case 4:
286 case 3:
287 if (s0[i].inc != s1[i].inc)
288 exact = 2;
289 case 2:
290 if (s0[i].lbnd == s1[i].lbnd && s0[i].ubnd == s1[i].ubnd)
291 break;
292 exact = 1;
293 default:
294 if (s0[i].lbnd < s1[i].lbnd || s0[i].ubnd > s1[i].ubnd)
295 return (s0[i].lbnd <= s0[i].ubnd ? 0 : 1);
296 }
297 break;
298 }
299 ras = SCM_CDR (ras);
300 }
0f2d19dd
JB
301 return exact;
302}
303
1bbd0b84
GB
304/* array mapper: apply cproc to each dimension of the given arrays?.
305 int (*cproc) (); procedure to call on unrolled arrays?
5c11cc9d 306 cproc (dest, source list) or
1bbd0b84
GB
307 cproc (dest, data, source list).
308 SCM data; data to give to cproc or unbound.
309 SCM ra0; destination array.
310 SCM lra; list of source arrays.
311 const char *what; caller, for error reporting. */
312int
313scm_ramapc (int (*cproc)(), SCM data, SCM ra0, SCM lra, const char *what)
0f2d19dd
JB
314{
315 SCM inds, z;
316 SCM vra0, ra1, vra1;
317 SCM lvra, *plvra;
318 long *vinds;
319 int k, kmax;
320 switch (scm_ra_matchp (ra0, lra))
321 {
322 default:
323 case 0:
5c11cc9d 324 scm_wta (ra0, "array shape mismatch", what);
0f2d19dd
JB
325 case 2:
326 case 3:
327 case 4: /* Try unrolling arrays */
328 kmax = (SCM_ARRAYP (ra0) ? SCM_ARRAY_NDIM (ra0) - 1 : 0);
329 if (kmax < 0)
330 goto gencase;
331 vra0 = scm_array_contents (ra0, SCM_UNDEFINED);
ff467021 332 if (SCM_IMP (vra0)) goto gencase;
0f2d19dd
JB
333 if (!SCM_ARRAYP (vra0))
334 {
335 vra1 = scm_make_ra (1);
336 SCM_ARRAY_BASE (vra1) = 0;
337 SCM_ARRAY_DIMS (vra1)->lbnd = 0;
338 SCM_ARRAY_DIMS (vra1)->ubnd = SCM_LENGTH (vra0) - 1;
339 SCM_ARRAY_DIMS (vra1)->inc = 1;
340 SCM_ARRAY_V (vra1) = vra0;
341 vra0 = vra1;
342 }
343 lvra = SCM_EOL;
344 plvra = &lvra;
345 for (z = lra; SCM_NIMP (z); z = SCM_CDR (z))
346 {
347 ra1 = SCM_CAR (z);
348 vra1 = scm_make_ra (1);
349 SCM_ARRAY_DIMS (vra1)->lbnd = SCM_ARRAY_DIMS (vra0)->lbnd;
350 SCM_ARRAY_DIMS (vra1)->ubnd = SCM_ARRAY_DIMS (vra0)->ubnd;
351 if (!SCM_ARRAYP (ra1))
352 {
353 SCM_ARRAY_BASE (vra1) = 0;
354 SCM_ARRAY_DIMS (vra1)->inc = 1;
355 SCM_ARRAY_V (vra1) = ra1;
356 }
357 else if (!SCM_ARRAY_CONTP (ra1))
358 goto gencase;
359 else
360 {
361 SCM_ARRAY_BASE (vra1) = SCM_ARRAY_BASE (ra1);
362 SCM_ARRAY_DIMS (vra1)->inc = SCM_ARRAY_DIMS (ra1)[kmax].inc;
363 SCM_ARRAY_V (vra1) = SCM_ARRAY_V (ra1);
364 }
365 *plvra = scm_cons (vra1, SCM_EOL);
25d8012c 366 plvra = SCM_CDRLOC (*plvra);
0f2d19dd
JB
367 }
368 return (SCM_UNBNDP (data) ? cproc(vra0, lvra) : cproc(vra0, data, lvra));
369 case 1:
370 gencase: /* Have to loop over all dimensions. */
c209c88e
GB
371 vra0 = scm_make_ra (1);
372 if (SCM_ARRAYP (ra0))
373 {
374 kmax = SCM_ARRAY_NDIM (ra0) - 1;
375 if (kmax < 0)
0f2d19dd 376 {
c209c88e
GB
377 SCM_ARRAY_DIMS (vra0)->lbnd = 0;
378 SCM_ARRAY_DIMS (vra0)->ubnd = 0;
379 SCM_ARRAY_DIMS (vra0)->inc = 1;
0f2d19dd 380 }
c209c88e
GB
381 else
382 {
383 SCM_ARRAY_DIMS (vra0)->lbnd = SCM_ARRAY_DIMS (ra0)[kmax].lbnd;
384 SCM_ARRAY_DIMS (vra0)->ubnd = SCM_ARRAY_DIMS (ra0)[kmax].ubnd;
385 SCM_ARRAY_DIMS (vra0)->inc = SCM_ARRAY_DIMS (ra0)[kmax].inc;
386 }
387 SCM_ARRAY_BASE (vra0) = SCM_ARRAY_BASE (ra0);
388 SCM_ARRAY_V (vra0) = SCM_ARRAY_V (ra0);
389 }
390 else
391 {
392 kmax = 0;
393 SCM_ARRAY_DIMS (vra0)->lbnd = 0;
394 SCM_ARRAY_DIMS (vra0)->ubnd = SCM_LENGTH (ra0) - 1;
395 SCM_ARRAY_DIMS (vra0)->inc = 1;
396 SCM_ARRAY_BASE (vra0) = 0;
397 SCM_ARRAY_V (vra0) = ra0;
398 ra0 = vra0;
399 }
400 lvra = SCM_EOL;
401 plvra = &lvra;
402 for (z = lra; SCM_NIMP (z); z = SCM_CDR (z))
403 {
404 ra1 = SCM_CAR (z);
405 vra1 = scm_make_ra (1);
406 SCM_ARRAY_DIMS (vra1)->lbnd = SCM_ARRAY_DIMS (vra0)->lbnd;
407 SCM_ARRAY_DIMS (vra1)->ubnd = SCM_ARRAY_DIMS (vra0)->ubnd;
408 if (SCM_ARRAYP (ra1))
409 {
410 if (kmax >= 0)
411 SCM_ARRAY_DIMS (vra1)->inc = SCM_ARRAY_DIMS (ra1)[kmax].inc;
412 SCM_ARRAY_V (vra1) = SCM_ARRAY_V (ra1);
413 }
414 else
415 {
416 SCM_ARRAY_DIMS (vra1)->inc = 1;
417 SCM_ARRAY_V (vra1) = ra1;
418 }
419 *plvra = scm_cons (vra1, SCM_EOL);
420 plvra = SCM_CDRLOC (*plvra);
421 }
422 inds = scm_make_uve (SCM_ARRAY_NDIM (ra0), SCM_MAKINUM (-1L));
423 vinds = (long *) SCM_VELTS (inds);
424 for (k = 0; k <= kmax; k++)
425 vinds[k] = SCM_ARRAY_DIMS (ra0)[k].lbnd;
426 k = kmax;
427 do
428 {
429 if (k == kmax)
430 {
431 SCM y = lra;
432 SCM_ARRAY_BASE (vra0) = cind (ra0, inds);
433 for (z = lvra; SCM_NIMP (z); z = SCM_CDR (z), y = SCM_CDR (y))
434 SCM_ARRAY_BASE (SCM_CAR (z)) = cind (SCM_CAR (y), inds);
435 if (0 == (SCM_UNBNDP (data) ? cproc(vra0, lvra) : cproc(vra0, data, lvra)))
436 return 0;
437 k--;
438 continue;
439 }
440 if (vinds[k] < SCM_ARRAY_DIMS (ra0)[k].ubnd)
441 {
442 vinds[k]++;
443 k++;
444 continue;
445 }
446 vinds[k] = SCM_ARRAY_DIMS (ra0)[k].lbnd - 1;
447 k--;
448 }
449 while (k >= 0);
450 return 1;
0f2d19dd
JB
451 }
452}
453
454
3b3b36dd 455SCM_DEFINE (scm_array_fill_x, "array-fill!", 2, 0, 0,
c209c88e 456 (SCM ra, SCM fill),
b380b885
MD
457 "Stores @var{fill} in every element of @var{array}. The value returned\n"
458 "is unspecified.")
1bbd0b84 459#define FUNC_NAME s_scm_array_fill_x
ad310508 460{
c209c88e 461 scm_ramapc (scm_array_fill_int, fill, ra, SCM_EOL, FUNC_NAME);
ad310508
MD
462 return SCM_UNSPECIFIED;
463}
1bbd0b84 464#undef FUNC_NAME
ad310508 465
5c11cc9d
GH
466/* to be used as cproc in scm_ramapc to fill an array dimension with
467 "fill". */
0f2d19dd 468int
1bbd0b84
GB
469scm_array_fill_int (SCM ra, SCM fill, SCM ignore)
470#define FUNC_NAME s_scm_array_fill_x
0f2d19dd 471{
5c11cc9d
GH
472 scm_sizet i;
473 scm_sizet n = SCM_ARRAY_DIMS (ra)->ubnd - SCM_ARRAY_DIMS (ra)->lbnd + 1;
0f2d19dd
JB
474 long inc = SCM_ARRAY_DIMS (ra)->inc;
475 scm_sizet base = SCM_ARRAY_BASE (ra);
5c11cc9d 476
0f2d19dd 477 ra = SCM_ARRAY_V (ra);
5c11cc9d
GH
478 switch SCM_TYP7 (ra)
479 {
480 default:
481 for (i = base; n--; i += inc)
482 scm_array_set_x (ra, fill, SCM_MAKINUM (i));
483 break;
484 case scm_tc7_vector:
485 case scm_tc7_wvect:
486 for (i = base; n--; i += inc)
487 SCM_VELTS (ra)[i] = fill;
488 break;
489 case scm_tc7_string:
7866a09b 490 SCM_ASRTGO (SCM_CHARP (fill), badarg2);
5c11cc9d 491 for (i = base; n--; i += inc)
7866a09b 492 SCM_CHARS (ra)[i] = SCM_CHAR (fill);
5c11cc9d
GH
493 break;
494 case scm_tc7_byvect:
7866a09b
GB
495 if (SCM_CHARP (fill))
496 fill = SCM_MAKINUM ((char) SCM_CHAR (fill));
5c11cc9d
GH
497 SCM_ASRTGO (SCM_INUMP (fill)
498 && -128 <= SCM_INUM (fill) && SCM_INUM (fill) < 128,
499 badarg2);
500 for (i = base; n--; i += inc)
501 SCM_CHARS (ra)[i] = SCM_INUM (fill);
502 break;
503 case scm_tc7_bvect:
1bbd0b84 504 { /* scope */
5c11cc9d
GH
505 long *ve = (long *) SCM_VELTS (ra);
506 if (1 == inc && (n >= SCM_LONG_BIT || n == SCM_LENGTH (ra)))
507 {
508 i = base / SCM_LONG_BIT;
509 if (SCM_BOOL_F == fill)
510 {
511 if (base % SCM_LONG_BIT) /* leading partial word */
512 ve[i++] &= ~(~0L << (base % SCM_LONG_BIT));
513 for (; i < (base + n) / SCM_LONG_BIT; i++)
514 ve[i] = 0L;
515 if ((base + n) % SCM_LONG_BIT) /* trailing partial word */
516 ve[i] &= (~0L << ((base + n) % SCM_LONG_BIT));
517 }
518 else if (SCM_BOOL_T == fill)
519 {
520 if (base % SCM_LONG_BIT)
521 ve[i++] |= ~0L << (base % SCM_LONG_BIT);
522 for (; i < (base + n) / SCM_LONG_BIT; i++)
523 ve[i] = ~0L;
524 if ((base + n) % SCM_LONG_BIT)
525 ve[i] |= ~(~0L << ((base + n) % SCM_LONG_BIT));
526 }
527 else
1bbd0b84 528 badarg2:SCM_WTA (2,fill);
5c11cc9d
GH
529 }
530 else
531 {
532 if (SCM_BOOL_F == fill)
533 for (i = base; n--; i += inc)
534 ve[i / SCM_LONG_BIT] &= ~(1L << (i % SCM_LONG_BIT));
535 else if (SCM_BOOL_T == fill)
536 for (i = base; n--; i += inc)
537 ve[i / SCM_LONG_BIT] |= (1L << (i % SCM_LONG_BIT));
538 else
539 goto badarg2;
540 }
541 break;
542 }
543 case scm_tc7_uvect:
1bbd0b84
GB
544 { /* scope */
545 unsigned long f = SCM_NUM2ULONG (2,fill);
5c11cc9d
GH
546 unsigned long *ve = (long *) SCM_VELTS (ra);
547
0f2d19dd 548 for (i = base; n--; i += inc)
5c11cc9d 549 ve[i] = f;
0f2d19dd 550 break;
5c11cc9d
GH
551 }
552 case scm_tc7_ivect:
1bbd0b84
GB
553 { /* scope */
554 long f = SCM_NUM2LONG (2,fill);
5c11cc9d
GH
555 long *ve = (long *) SCM_VELTS (ra);
556
0f2d19dd 557 for (i = base; n--; i += inc)
5c11cc9d 558 ve[i] = f;
0f2d19dd 559 break;
5c11cc9d
GH
560 }
561 case scm_tc7_svect:
562 SCM_ASRTGO (SCM_INUMP (fill), badarg2);
1bbd0b84 563 { /* scope */
5c11cc9d
GH
564 short f = SCM_INUM (fill);
565 short *ve = (short *) SCM_VELTS (ra);
566
567 if (f != SCM_INUM (fill))
1bbd0b84 568 SCM_OUT_OF_RANGE (2, fill);
0f2d19dd 569 for (i = base; n--; i += inc)
5c11cc9d 570 ve[i] = f;
0f2d19dd 571 break;
5c11cc9d
GH
572 }
573#ifdef HAVE_LONG_LONGS
574 case scm_tc7_llvect:
1bbd0b84
GB
575 { /* scope */
576 long long f = SCM_NUM2LONG_LONG (2,fill);
5c11cc9d
GH
577 long long *ve = (long long *) SCM_VELTS (ra);
578
b1d24656 579 for (i = base; n--; i += inc)
5c11cc9d 580 ve[i] = f;
b1d24656 581 break;
5c11cc9d 582 }
05f92e0f 583#endif
5c11cc9d 584 case scm_tc7_fvect:
1bbd0b84 585 { /* scope */
5c11cc9d 586 float f, *ve = (float *) SCM_VELTS (ra);
0c95b57d 587 SCM_ASRTGO (SCM_REALP (fill), badarg2);
5c11cc9d
GH
588 f = SCM_REALPART (fill);
589 for (i = base; n--; i += inc)
590 ve[i] = f;
591 break;
592 }
5c11cc9d 593 case scm_tc7_dvect:
1bbd0b84 594 { /* scope */
5c11cc9d 595 double f, *ve = (double *) SCM_VELTS (ra);
0c95b57d 596 SCM_ASRTGO (SCM_REALP (fill), badarg2);
5c11cc9d
GH
597 f = SCM_REALPART (fill);
598 for (i = base; n--; i += inc)
599 ve[i] = f;
600 break;
601 }
602 case scm_tc7_cvect:
1bbd0b84 603 { /* scope */
5c11cc9d
GH
604 double fr, fi;
605 double (*ve)[2] = (double (*)[2]) SCM_VELTS (ra);
0c95b57d 606 SCM_ASRTGO (SCM_INEXP (fill), badarg2);
5c11cc9d
GH
607 fr = SCM_REALPART (fill);
608 fi = (SCM_CPLXP (fill) ? SCM_IMAG (fill) : 0.0);
609 for (i = base; n--; i += inc)
610 {
611 ve[i][0] = fr;
612 ve[i][1] = fi;
613 }
614 break;
0f2d19dd 615 }
5c11cc9d 616 }
0f2d19dd
JB
617 return 1;
618}
1bbd0b84 619#undef FUNC_NAME
0f2d19dd 620
0f2d19dd 621
c209c88e 622
0f2d19dd 623static int
1bbd0b84 624racp (SCM src, SCM dst)
0f2d19dd
JB
625{
626 long n = (SCM_ARRAY_DIMS (src)->ubnd - SCM_ARRAY_DIMS (src)->lbnd + 1);
627 long inc_d, inc_s = SCM_ARRAY_DIMS (src)->inc;
628 scm_sizet i_d, i_s = SCM_ARRAY_BASE (src);
629 dst = SCM_CAR (dst);
630 inc_d = SCM_ARRAY_DIMS (dst)->inc;
631 i_d = SCM_ARRAY_BASE (dst);
632 src = SCM_ARRAY_V (src);
633 dst = SCM_ARRAY_V (dst);
c209c88e
GB
634
635
636 /* untested optimization: don't copy if we're we. This allows the
637 ugly UNICOS macros (IVDEP) to go .
638 */
639
640 if (src == dst)
641 return 1 ;
642
0f2d19dd
JB
643 switch SCM_TYP7
644 (dst)
c209c88e
GB
645 {
646 default:
647 gencase:
648 case scm_tc7_vector:
649 case scm_tc7_wvect:
95f5b0f5 650
c209c88e
GB
651 for (; n-- > 0; i_s += inc_s, i_d += inc_d)
652 scm_array_set_x (dst, scm_cvref (src, i_s, SCM_UNDEFINED), SCM_MAKINUM (i_d));
653 break;
654 case scm_tc7_string:
655 case scm_tc7_byvect:
656 if (scm_tc7_string != SCM_TYP7 (dst))
657 goto gencase;
658 for (; n-- > 0; i_s += inc_s, i_d += inc_d)
659 SCM_CHARS (dst)[i_d] = SCM_CHARS (src)[i_s];
660 break;
661 case scm_tc7_bvect:
662 if (scm_tc7_bvect != SCM_TYP7 (dst))
663 goto gencase;
664 if (1 == inc_d && 1 == inc_s && i_s % SCM_LONG_BIT == i_d % SCM_LONG_BIT && n >= SCM_LONG_BIT)
665 {
666 long *sv = (long *) SCM_VELTS (src);
667 long *dv = (long *) SCM_VELTS (dst);
668 sv += i_s / SCM_LONG_BIT;
669 dv += i_d / SCM_LONG_BIT;
670 if (i_s % SCM_LONG_BIT)
671 { /* leading partial word */
672 *dv = (*dv & ~(~0L << (i_s % SCM_LONG_BIT))) | (*sv & (~0L << (i_s % SCM_LONG_BIT)));
673 dv++;
674 sv++;
675 n -= SCM_LONG_BIT - (i_s % SCM_LONG_BIT);
676 }
677 for (; n >= SCM_LONG_BIT; n -= SCM_LONG_BIT, sv++, dv++)
678 * dv = *sv;
679 if (n) /* trailing partial word */
680 *dv = (*dv & (~0L << n)) | (*sv & ~(~0L << n));
681 }
682 else
683 {
684 for (; n-- > 0; i_s += inc_s, i_d += inc_d)
685 if (SCM_BITVEC_REF(src, i_s))
686 SCM_BITVEC_SET(dst, i_d);
687 else
688 SCM_BITVEC_CLR(dst, i_d);
689 }
690 break;
691 case scm_tc7_uvect:
692 if (scm_tc7_uvect != SCM_TYP7 (src))
693 goto gencase;
694 else
695 {
696 long *d = (long *) SCM_VELTS (dst), *s = (long *) SCM_VELTS (src);
697 for (; n-- > 0; i_s += inc_s, i_d += inc_d)
698 d[i_d] = s[i_s];
699 break;
700 }
701 case scm_tc7_ivect:
702 if (scm_tc7_uvect != SCM_TYP7 (src) && scm_tc7_ivect != SCM_TYP7 (src))
703 goto gencase;
704 else
705 {
706 long *d = (long *) SCM_VELTS (dst), *s = (long *) SCM_VELTS (src);
707 for (; n-- > 0; i_s += inc_s, i_d += inc_d)
708 d[i_d] = s[i_s];
709 break;
710 }
c209c88e
GB
711 case scm_tc7_fvect:
712 {
713 float *d = (float *) SCM_VELTS (dst);
714 float *s = (float *) SCM_VELTS (src);
715 switch SCM_TYP7
716 (src)
0f2d19dd 717 {
c209c88e
GB
718 default:
719 goto gencase;
720 case scm_tc7_ivect:
721 case scm_tc7_uvect:
0f2d19dd 722 for (; n-- > 0; i_s += inc_s, i_d += inc_d)
c209c88e
GB
723 d[i_d] = ((long *) s)[i_s];
724 break;
725 case scm_tc7_fvect:
726 for (; n-- > 0; i_s += inc_s, i_d += inc_d)
727 d[i_d] = s[i_s];
728 break;
729 case scm_tc7_dvect:
730 for (; n-- > 0; i_s += inc_s, i_d += inc_d)
731 d[i_d] = ((double *) s)[i_s];
732 break;
0f2d19dd
JB
733 }
734 break;
c209c88e 735 }
c209c88e
GB
736 case scm_tc7_dvect:
737 {
738 double *d = (double *) SCM_VELTS (dst);
739 double *s = (double *) SCM_VELTS (src);
740 switch SCM_TYP7
741 (src)
0f2d19dd 742 {
c209c88e
GB
743 default:
744 goto gencase;
745 case scm_tc7_ivect:
746 case scm_tc7_uvect:
747 for (; n-- > 0; i_s += inc_s, i_d += inc_d)
748 d[i_d] = ((long *) s)[i_s];
749 break;
750 case scm_tc7_fvect:
751 for (; n-- > 0; i_s += inc_s, i_d += inc_d)
752 d[i_d] = ((float *) s)[i_s];
753 break;
754 case scm_tc7_dvect:
755 for (; n-- > 0; i_s += inc_s, i_d += inc_d)
756 d[i_d] = s[i_s];
0f2d19dd
JB
757 break;
758 }
c209c88e
GB
759 break;
760 }
761 case scm_tc7_cvect:
762 {
763 double (*d)[2] = (double (*)[2]) SCM_VELTS (dst);
764 double (*s)[2] = (double (*)[2]) SCM_VELTS (src);
765 switch SCM_TYP7
766 (src)
0f2d19dd 767 {
c209c88e
GB
768 default:
769 goto gencase;
770 case scm_tc7_ivect:
771 case scm_tc7_uvect:
772 for (; n-- > 0; i_s += inc_s, i_d += inc_d)
773 {
774 d[i_d][0] = ((long *) s)[i_s];
775 d[i_d][1] = 0.0;
776 }
0f2d19dd 777 break;
c209c88e
GB
778 case scm_tc7_fvect:
779 for (; n-- > 0; i_s += inc_s, i_d += inc_d)
0f2d19dd 780 {
c209c88e
GB
781 d[i_d][0] = ((float *) s)[i_s];
782 d[i_d][1] = 0.0;
0f2d19dd 783 }
c209c88e
GB
784 break;
785 case scm_tc7_dvect:
786 for (; n-- > 0; i_s += inc_s, i_d += inc_d)
0f2d19dd 787 {
c209c88e
GB
788 d[i_d][0] = ((double *) s)[i_s];
789 d[i_d][1] = 0.0;
0f2d19dd 790 }
c209c88e
GB
791 break;
792 case scm_tc7_cvect:
793 for (; n-- > 0; i_s += inc_s, i_d += inc_d)
0f2d19dd 794 {
c209c88e
GB
795 d[i_d][0] = s[i_s][0];
796 d[i_d][1] = s[i_s][1];
0f2d19dd 797 }
c209c88e
GB
798 }
799 break;
0f2d19dd 800 }
c209c88e 801 }
0f2d19dd
JB
802 return 1;
803}
804
805
dba2ab49 806/* This name is obsolete. Will go away in release 1.5. */
1bbd0b84
GB
807SCM_REGISTER_PROC(s_serial_array_copy_x, "serial-array-copy!", 2, 0, 0, scm_array_copy_x);
808SCM_REGISTER_PROC(s_array_copy_in_order_x, "array-copy-in-order!", 2, 0, 0, scm_array_copy_x);
1cc91f1b 809
1bbd0b84 810
3b3b36dd 811SCM_DEFINE (scm_array_copy_x, "array-copy!", 2, 0, 0,
c209c88e 812 (SCM src, SCM dst),
b380b885
MD
813 "Copies every element from vector or array @var{source} to the\n"
814 "corresponding element of @var{destination}. @var{destination} must have\n"
815 "the same rank as @var{source}, and be at least as large in each\n"
816 "dimension. The order is unspecified.")
1bbd0b84 817#define FUNC_NAME s_scm_array_copy_x
0f2d19dd 818{
c209c88e 819 scm_ramapc (racp, SCM_UNDEFINED, src, scm_cons (dst, SCM_EOL), FUNC_NAME);
0f2d19dd
JB
820 return SCM_UNSPECIFIED;
821}
1bbd0b84 822#undef FUNC_NAME
0f2d19dd
JB
823
824/* Functions callable by ARRAY-MAP! */
825
1cc91f1b 826
0f2d19dd 827int
1bbd0b84 828scm_ra_eqp (SCM ra0, SCM ras)
0f2d19dd
JB
829{
830 SCM ra1 = SCM_CAR (ras), ra2 = SCM_CAR (SCM_CDR (ras));
831 long n = SCM_ARRAY_DIMS (ra0)->ubnd - SCM_ARRAY_DIMS (ra0)->lbnd + 1;
832 scm_sizet i0 = SCM_ARRAY_BASE (ra0), i1 = SCM_ARRAY_BASE (ra1), i2 = SCM_ARRAY_BASE (ra2);
833 long inc0 = SCM_ARRAY_DIMS (ra0)->inc;
834 long inc1 = SCM_ARRAY_DIMS (ra1)->inc;
835 long inc2 = SCM_ARRAY_DIMS (ra1)->inc;
836 ra0 = SCM_ARRAY_V (ra0);
837 ra1 = SCM_ARRAY_V (ra1);
838 ra2 = SCM_ARRAY_V (ra2);
839 switch (SCM_TYP7 (ra1) == SCM_TYP7 (ra2) ? SCM_TYP7 (ra1) : 0)
840 {
841 default:
842 {
843 SCM e1 = SCM_UNDEFINED, e2 = SCM_UNDEFINED;
844 for (; n-- > 0; i0 += inc0, i1 += inc1, i2 += inc2)
c209c88e
GB
845 if (SCM_BITVEC_REF (ra0, i0))
846 if (SCM_FALSEP(scm_eq_p (RVREF (ra1, i1, e1), RVREF (ra2, i2, e2))))
847 SCM_BITVEC_CLR (ra0, i0);
0f2d19dd
JB
848 break;
849 }
850 case scm_tc7_uvect:
851 case scm_tc7_ivect:
852 for (; n-- > 0; i0 += inc0, i1 += inc1, i2 += inc2)
c209c88e
GB
853 if (SCM_BITVEC_REF (ra0, i0))
854 if (SCM_VELTS (ra1)[i1] != SCM_VELTS (ra2)[i2])
855 SCM_BITVEC_CLR (ra0, i0);
0f2d19dd 856 break;
0f2d19dd
JB
857 case scm_tc7_fvect:
858 for (; n-- > 0; i0 += inc0, i1 += inc1, i2 += inc2)
c209c88e
GB
859 if (SCM_BITVEC_REF (ra0, i0))
860 if (((float *) SCM_VELTS (ra1))[i1] != ((float *) SCM_VELTS (ra2))[i2])
861 SCM_BITVEC_CLR (ra0, i0);
0f2d19dd 862 break;
0f2d19dd
JB
863 case scm_tc7_dvect:
864 for (; n-- > 0; i0 += inc0, i1 += inc1, i2 += inc2)
c209c88e
GB
865 if (SCM_BITVEC_REF (ra0, i0))
866 if (((double *) SCM_VELTS (ra1))[i1] != ((double *) SCM_VELTS (ra2))[i2])
867 SCM_BITVEC_CLR (ra0, i0);
0f2d19dd
JB
868 break;
869 case scm_tc7_cvect:
870 for (; n-- > 0; i0 += inc0, i1 += inc1, i2 += inc2)
c209c88e
GB
871 if (SCM_BITVEC_REF (ra0, i0))
872 if (((double *) SCM_VELTS (ra1))[2 * i1] != ((double *) SCM_VELTS (ra2))[2 * i2] ||
873 ((double *) SCM_VELTS (ra1))[2 * i1 + 1] != ((double *) SCM_VELTS (ra2))[2 * i2 + 1])
874 SCM_BITVEC_CLR (ra0, i0);
0f2d19dd 875 break;
0f2d19dd
JB
876 }
877 return 1;
878}
879
880/* opt 0 means <, nonzero means >= */
1cc91f1b 881
0f2d19dd 882static int
1bbd0b84 883ra_compare (SCM ra0,SCM ra1,SCM ra2,int opt)
0f2d19dd
JB
884{
885 long n = SCM_ARRAY_DIMS (ra0)->ubnd - SCM_ARRAY_DIMS (ra0)->lbnd + 1;
886 scm_sizet i0 = SCM_ARRAY_BASE (ra0), i1 = SCM_ARRAY_BASE (ra1), i2 = SCM_ARRAY_BASE (ra2);
887 long inc0 = SCM_ARRAY_DIMS (ra0)->inc;
888 long inc1 = SCM_ARRAY_DIMS (ra1)->inc;
889 long inc2 = SCM_ARRAY_DIMS (ra1)->inc;
890 ra0 = SCM_ARRAY_V (ra0);
891 ra1 = SCM_ARRAY_V (ra1);
892 ra2 = SCM_ARRAY_V (ra2);
893 switch (SCM_TYP7 (ra1) == SCM_TYP7 (ra2) ? SCM_TYP7 (ra1) : 0)
894 {
895 default:
896 {
897 SCM e1 = SCM_UNDEFINED, e2 = SCM_UNDEFINED;
898 for (; n-- > 0; i0 += inc0, i1 += inc1, i2 += inc2)
c209c88e
GB
899 if (SCM_BITVEC_REF (ra0, i0))
900 if (opt ?
901 SCM_NFALSEP (scm_less_p (RVREF (ra1, i1, e1), RVREF (ra2, i2, e2))) :
902 SCM_FALSEP (scm_less_p (RVREF (ra1, i1, e1), RVREF (ra2, i2, e2))))
903 SCM_BITVEC_CLR (ra0, i0);
0f2d19dd
JB
904 break;
905 }
906 case scm_tc7_uvect:
907 case scm_tc7_ivect:
908 for (; n-- > 0; i0 += inc0, i1 += inc1, i2 += inc2)
909 {
c209c88e
GB
910 if (SCM_BITVEC_REF (ra0, i0))
911 if (opt ?
912 SCM_VELTS (ra1)[i1] < SCM_VELTS (ra2)[i2] :
913 SCM_VELTS (ra1)[i1] >= SCM_VELTS (ra2)[i2])
914 SCM_BITVEC_CLR (ra0, i0);
0f2d19dd
JB
915 }
916 break;
0f2d19dd
JB
917 case scm_tc7_fvect:
918 for (; n-- > 0; i0 += inc0, i1 += inc1, i2 += inc2)
c209c88e
GB
919 if (SCM_BITVEC_REF(ra0, i0))
920 if (opt ?
921 ((float *) SCM_VELTS (ra1))[i1] < ((float *) SCM_VELTS (ra2))[i2] :
922 ((float *) SCM_VELTS (ra1))[i1] >= ((float *) SCM_VELTS (ra2))[i2])
923 SCM_BITVEC_CLR (ra0, i0);
0f2d19dd 924 break;
0f2d19dd
JB
925 case scm_tc7_dvect:
926 for (; n-- > 0; i0 += inc0, i1 += inc1, i2 += inc2)
c209c88e
GB
927 if (SCM_BITVEC_REF (ra0, i0))
928 if (opt ?
929 ((double *) SCM_VELTS (ra1))[i1] < ((double *) SCM_VELTS (ra2))[i2] :
930 ((double *) SCM_VELTS (ra1))[i1] >= ((double *) SCM_VELTS (ra2))[i2])
931 SCM_BITVEC_CLR (ra0, i0);
0f2d19dd 932 break;
0f2d19dd
JB
933 }
934 return 1;
935}
936
937
1cc91f1b 938
0f2d19dd 939int
1bbd0b84 940scm_ra_lessp (SCM ra0, SCM ras)
0f2d19dd
JB
941{
942 return ra_compare (ra0, SCM_CAR (ras), SCM_CAR (SCM_CDR (ras)), 0);
943}
944
1cc91f1b 945
0f2d19dd 946int
1bbd0b84 947scm_ra_leqp (SCM ra0, SCM ras)
0f2d19dd
JB
948{
949 return ra_compare (ra0, SCM_CAR (SCM_CDR (ras)), SCM_CAR (ras), 1);
950}
951
1cc91f1b 952
0f2d19dd 953int
1bbd0b84 954scm_ra_grp (SCM ra0, SCM ras)
0f2d19dd
JB
955{
956 return ra_compare (ra0, SCM_CAR (SCM_CDR (ras)), SCM_CAR (ras), 0);
957}
958
1cc91f1b 959
0f2d19dd 960int
1bbd0b84 961scm_ra_greqp (SCM ra0, SCM ras)
0f2d19dd
JB
962{
963 return ra_compare (ra0, SCM_CAR (ras), SCM_CAR (SCM_CDR (ras)), 1);
964}
965
966
0f2d19dd 967int
1bbd0b84 968scm_ra_sum (SCM ra0, SCM ras)
0f2d19dd
JB
969{
970 long n = SCM_ARRAY_DIMS (ra0)->ubnd - SCM_ARRAY_DIMS (ra0)->lbnd + 1;
971 scm_sizet i0 = SCM_ARRAY_BASE (ra0);
972 long inc0 = SCM_ARRAY_DIMS (ra0)->inc;
973 ra0 = SCM_ARRAY_V (ra0);
ff467021 974 if (SCM_NNULLP(ras))
c209c88e
GB
975 {
976 SCM ra1 = SCM_CAR (ras);
977 scm_sizet i1 = SCM_ARRAY_BASE (ra1);
978 long inc1 = SCM_ARRAY_DIMS (ra1)->inc;
979 ra1 = SCM_ARRAY_V (ra1);
980 switch (SCM_TYP7 (ra0) == SCM_TYP7 (ra1) ? SCM_TYP7 (ra0) : 0)
981 {
982 default:
0f2d19dd 983 {
c209c88e
GB
984 SCM e0 = SCM_UNDEFINED, e1 = SCM_UNDEFINED;
985 for (; n-- > 0; i0 += inc0, i1 += inc1)
986 scm_array_set_x (ra0, scm_sum (RVREF (ra0, i0, e0), RVREF (ra1, i1, e1)),
987 SCM_MAKINUM (i0));
988 break;
989 }
990 case scm_tc7_uvect:
991 case scm_tc7_ivect:
992 BINARY_ELTS_CODE( +=, long);
c209c88e
GB
993 case scm_tc7_fvect:
994 BINARY_ELTS_CODE( +=, float);
c209c88e
GB
995 case scm_tc7_dvect:
996 BINARY_ELTS_CODE( +=, double);
997 case scm_tc7_cvect:
998 BINARY_PAIR_ELTS_CODE( +=, double);
c209c88e
GB
999 }
1000 }
0f2d19dd
JB
1001 return 1;
1002}
1003
1004
1cc91f1b 1005
0f2d19dd 1006int
1bbd0b84 1007scm_ra_difference (SCM ra0, SCM ras)
0f2d19dd
JB
1008{
1009 long n = SCM_ARRAY_DIMS (ra0)->ubnd - SCM_ARRAY_DIMS (ra0)->lbnd + 1;
1010 scm_sizet i0 = SCM_ARRAY_BASE (ra0);
1011 long inc0 = SCM_ARRAY_DIMS (ra0)->inc;
1012 ra0 = SCM_ARRAY_V (ra0);
ff467021 1013 if (SCM_NULLP (ras))
c209c88e
GB
1014 {
1015 switch (SCM_TYP7 (ra0))
1016 {
1017 default:
1018 {
1019 SCM e0 = SCM_UNDEFINED;
1020 for (; n-- > 0; i0 += inc0)
1021 scm_array_set_x (ra0,
1022 scm_difference (RVREF (ra0, i0, e0), SCM_UNDEFINED),
1023 SCM_MAKINUM (i0));
1024 break;
1025 }
c209c88e
GB
1026 case scm_tc7_fvect:
1027 UNARY_ELTS_CODE( = -, float);
c209c88e
GB
1028 case scm_tc7_dvect:
1029 UNARY_ELTS_CODE( = -, double);
1030 case scm_tc7_cvect:
1031 UNARY_PAIR_ELTS_CODE( = -, double);
c209c88e
GB
1032 }
1033 }
0f2d19dd
JB
1034 else
1035 {
1036 SCM ra1 = SCM_CAR (ras);
1037 scm_sizet i1 = SCM_ARRAY_BASE (ra1);
1038 long inc1 = SCM_ARRAY_DIMS (ra1)->inc;
1039 ra1 = SCM_ARRAY_V (ra1);
1040 switch (SCM_TYP7 (ra0) == SCM_TYP7 (ra1) ? SCM_TYP7 (ra0) : 0)
1041 {
1042 default:
1043 {
1044 SCM e0 = SCM_UNDEFINED, e1 = SCM_UNDEFINED;
1045 for (; n-- > 0; i0 += inc0, i1 += inc1)
1046 scm_array_set_x (ra0, scm_difference (RVREF (ra0, i0, e0), RVREF (ra1, i1, e1)), SCM_MAKINUM (i0));
1047 break;
1048 }
0f2d19dd 1049 case scm_tc7_fvect:
c209c88e 1050 BINARY_ELTS_CODE( -=, float);
0f2d19dd 1051 case scm_tc7_dvect:
c209c88e 1052 BINARY_ELTS_CODE( -=, double);
0f2d19dd 1053 case scm_tc7_cvect:
c209c88e 1054 BINARY_PAIR_ELTS_CODE( -=, double);
0f2d19dd
JB
1055 }
1056 }
1057 return 1;
1058}
1059
1060
1cc91f1b 1061
0f2d19dd 1062int
1bbd0b84 1063scm_ra_product (SCM ra0, SCM ras)
0f2d19dd
JB
1064{
1065 long n = SCM_ARRAY_DIMS (ra0)->ubnd - SCM_ARRAY_DIMS (ra0)->lbnd + 1;
1066 scm_sizet i0 = SCM_ARRAY_BASE (ra0);
1067 long inc0 = SCM_ARRAY_DIMS (ra0)->inc;
1068 ra0 = SCM_ARRAY_V (ra0);
ff467021 1069 if (SCM_NNULLP (ras))
c209c88e
GB
1070 {
1071 SCM ra1 = SCM_CAR (ras);
1072 scm_sizet i1 = SCM_ARRAY_BASE (ra1);
1073 long inc1 = SCM_ARRAY_DIMS (ra1)->inc;
1074 ra1 = SCM_ARRAY_V (ra1);
1075 switch (SCM_TYP7 (ra0) == SCM_TYP7 (ra1) ? SCM_TYP7 (ra0) : 0)
1076 {
1077 default:
0f2d19dd 1078 {
c209c88e
GB
1079 SCM e0 = SCM_UNDEFINED, e1 = SCM_UNDEFINED;
1080 for (; n-- > 0; i0 += inc0, i1 += inc1)
1081 scm_array_set_x (ra0, scm_product (RVREF (ra0, i0, e0), RVREF (ra1, i1, e1)),
1082 SCM_MAKINUM (i0));
1083 break;
1084 }
1085 case scm_tc7_uvect:
1086 case scm_tc7_ivect:
1087 BINARY_ELTS_CODE( *=, long);
c209c88e
GB
1088 case scm_tc7_fvect:
1089 BINARY_ELTS_CODE( *=, float);
c209c88e
GB
1090 case scm_tc7_dvect:
1091 BINARY_ELTS_CODE( *=, double);
1092 case scm_tc7_cvect:
1093 {
1094 double (*v0)[2] = (double (*)[2]) SCM_VELTS (ra0);
1095 register double r;
1096 double (*v1)[2] = (double (*)[2]) SCM_VELTS (ra1);
1097 IVDEP (ra0 != ra1,
1098 for (; n-- > 0; i0 += inc0, i1 += inc1)
1099 {
1100 r = v0[i0][0] * v1[i1][0] - v0[i0][1] * v1[i1][1];
1101 v0[i0][1] = v0[i0][0] * v1[i1][1] + v0[i0][1] * v1[i1][0];
1102 v0[i0][0] = r;
1103 }
1104 );
1105 break;
0f2d19dd 1106 }
c209c88e
GB
1107 }
1108 }
0f2d19dd
JB
1109 return 1;
1110}
1111
1cc91f1b 1112
0f2d19dd 1113int
1bbd0b84 1114scm_ra_divide (SCM ra0, SCM ras)
0f2d19dd
JB
1115{
1116 long n = SCM_ARRAY_DIMS (ra0)->ubnd - SCM_ARRAY_DIMS (ra0)->lbnd + 1;
1117 scm_sizet i0 = SCM_ARRAY_BASE (ra0);
1118 long inc0 = SCM_ARRAY_DIMS (ra0)->inc;
1119 ra0 = SCM_ARRAY_V (ra0);
ff467021 1120 if (SCM_NULLP (ras))
c209c88e
GB
1121 {
1122 switch (SCM_TYP7 (ra0))
1123 {
1124 default:
1125 {
1126 SCM e0 = SCM_UNDEFINED;
1127 for (; n-- > 0; i0 += inc0)
1128 scm_array_set_x (ra0, scm_divide (RVREF (ra0, i0, e0), SCM_UNDEFINED), SCM_MAKINUM (i0));
1129 break;
1130 }
c209c88e
GB
1131 case scm_tc7_fvect:
1132 UNARY_ELTS_CODE( = 1.0 / , float);
c209c88e
GB
1133 case scm_tc7_dvect:
1134 UNARY_ELTS_CODE( = 1.0 / , double);
1135 case scm_tc7_cvect:
1136 {
1137 register double d;
1138 double (*v0)[2] = (double (*)[2]) SCM_VELTS (ra0);
1139 for (; n-- > 0; i0 += inc0)
0f2d19dd 1140 {
c209c88e
GB
1141 d = v0[i0][0] * v0[i0][0] + v0[i0][1] * v0[i0][1];
1142 v0[i0][0] /= d;
1143 v0[i0][1] /= -d;
0f2d19dd 1144 }
c209c88e
GB
1145 break;
1146 }
c209c88e
GB
1147 }
1148 }
0f2d19dd
JB
1149 else
1150 {
1151 SCM ra1 = SCM_CAR (ras);
1152 scm_sizet i1 = SCM_ARRAY_BASE (ra1);
1153 long inc1 = SCM_ARRAY_DIMS (ra1)->inc;
1154 ra1 = SCM_ARRAY_V (ra1);
1155 switch (SCM_TYP7 (ra0) == SCM_TYP7 (ra1) ? SCM_TYP7 (ra0) : 0)
1156 {
1157 default:
1158 {
1159 SCM e0 = SCM_UNDEFINED, e1 = SCM_UNDEFINED;
1160 for (; n-- > 0; i0 += inc0, i1 += inc1)
1161 scm_array_set_x (ra0, scm_divide (RVREF (ra0, i0, e0), RVREF (ra1, i1, e1)), SCM_MAKINUM (i0));
1162 break;
1163 }
0f2d19dd 1164 case scm_tc7_fvect:
c209c88e 1165 BINARY_ELTS_CODE( /=, float);
0f2d19dd 1166 case scm_tc7_dvect:
c209c88e 1167 BINARY_ELTS_CODE( /=, double);
0f2d19dd
JB
1168 case scm_tc7_cvect:
1169 {
1170 register double d, r;
1171 double (*v0)[2] = (double (*)[2]) SCM_VELTS (ra0);
1172 double (*v1)[2] = (double (*)[2]) SCM_VELTS (ra1);
1173 IVDEP (ra0 != ra1,
1174 for (; n-- > 0; i0 += inc0, i1 += inc1)
c209c88e
GB
1175 {
1176 d = v1[i1][0] * v1[i1][0] + v1[i1][1] * v1[i1][1];
1177 r = (v0[i0][0] * v1[i1][0] + v0[i0][1] * v1[i1][1]) / d;
1178 v0[i0][1] = (v0[i0][1] * v1[i1][0] - v0[i0][0] * v1[i1][1]) / d;
1179 v0[i0][0] = r;
1180 }
0f2d19dd
JB
1181 )
1182 break;
1183 }
0f2d19dd
JB
1184 }
1185 }
1186 return 1;
1187}
1188
1cc91f1b 1189
0f2d19dd 1190int
1bbd0b84 1191scm_array_identity (SCM dst, SCM src)
0f2d19dd
JB
1192{
1193 return racp (SCM_CAR (src), scm_cons (dst, SCM_EOL));
1194}
1195
1196
1cc91f1b 1197
0f2d19dd 1198static int
1bbd0b84 1199ramap (SCM ra0,SCM proc,SCM ras)
0f2d19dd
JB
1200{
1201 long i = SCM_ARRAY_DIMS (ra0)->lbnd;
1202 long inc = SCM_ARRAY_DIMS (ra0)->inc;
1203 long n = SCM_ARRAY_DIMS (ra0)->ubnd;
1204 long base = SCM_ARRAY_BASE (ra0) - i * inc;
1205 ra0 = SCM_ARRAY_V (ra0);
ff467021 1206 if (SCM_NULLP (ras))
c209c88e
GB
1207 for (; i <= n; i++)
1208 scm_array_set_x (ra0, scm_apply (proc, SCM_EOL, SCM_EOL), SCM_MAKINUM (i * inc + base));
0f2d19dd
JB
1209 else
1210 {
1211 SCM ra1 = SCM_CAR (ras);
1212 SCM args, *ve = &ras;
1213 scm_sizet k, i1 = SCM_ARRAY_BASE (ra1);
1214 long inc1 = SCM_ARRAY_DIMS (ra1)->inc;
1215 ra1 = SCM_ARRAY_V (ra1);
1216 ras = SCM_CDR (ras);
ff467021 1217 if (SCM_NULLP(ras))
c209c88e 1218 ras = scm_nullvect;
0f2d19dd
JB
1219 else
1220 {
1221 ras = scm_vector (ras);
1222 ve = SCM_VELTS (ras);
1223 }
1224 for (; i <= n; i++, i1 += inc1)
1225 {
1226 args = SCM_EOL;
1227 for (k = SCM_LENGTH (ras); k--;)
1228 args = scm_cons (scm_uniform_vector_ref (ve[k], SCM_MAKINUM (i)), args);
1229 args = scm_cons (scm_cvref (ra1, i1, SCM_UNDEFINED), args);
1230 scm_array_set_x (ra0, scm_apply (proc, args, SCM_EOL), SCM_MAKINUM (i * inc + base));
1231 }
1232 }
1233 return 1;
1234}
1235
1cc91f1b 1236
0f2d19dd 1237static int
1bbd0b84 1238ramap_cxr (SCM ra0,SCM proc,SCM ras)
0f2d19dd
JB
1239{
1240 SCM ra1 = SCM_CAR (ras);
1241 SCM e1 = SCM_UNDEFINED;
1242 scm_sizet i0 = SCM_ARRAY_BASE (ra0), i1 = SCM_ARRAY_BASE (ra1);
1243 long inc0 = SCM_ARRAY_DIMS (ra0)->inc, inc1 = SCM_ARRAY_DIMS (ra1)->inc;
1244 long n = SCM_ARRAY_DIMS (ra0)->ubnd - SCM_ARRAY_DIMS (ra1)->lbnd + 1;
1245 ra0 = SCM_ARRAY_V (ra0);
1246 ra1 = SCM_ARRAY_V (ra1);
ff467021 1247 switch (SCM_TYP7 (ra0))
c209c88e
GB
1248 {
1249 default:
1250 gencase:
1251 for (; n-- > 0; i0 += inc0, i1 += inc1)
1252 scm_array_set_x (ra0, scm_apply (proc, RVREF (ra1, i1, e1), scm_listofnull), SCM_MAKINUM (i0));
1253 break;
c209c88e
GB
1254 case scm_tc7_fvect:
1255 {
1256 float *dst = (float *) SCM_VELTS (ra0);
1257 switch (SCM_TYP7 (ra1))
1258 {
1259 default:
1260 goto gencase;
1261 case scm_tc7_fvect:
1262 for (; n-- > 0; i0 += inc0, i1 += inc1)
1263 dst[i0] = SCM_DSUBRF (proc) ((double) ((float *) SCM_VELTS (ra1))[i1]);
1264 break;
1265 case scm_tc7_uvect:
1266 case scm_tc7_ivect:
1267 for (; n-- > 0; i0 += inc0, i1 += inc1)
f1267706 1268 dst[i0] = SCM_DSUBRF (proc) (SCM_UNPACK (SCM_VELTS (ra1)[i1]));
c209c88e
GB
1269 break;
1270 }
1271 break;
1272 }
c209c88e
GB
1273 case scm_tc7_dvect:
1274 {
1275 double *dst = (double *) SCM_VELTS (ra0);
1276 switch (SCM_TYP7 (ra1))
1277 {
1278 default:
1279 goto gencase;
1280 case scm_tc7_dvect:
1281 for (; n-- > 0; i0 += inc0, i1 += inc1)
1282 dst[i0] = SCM_DSUBRF (proc) (((double *) SCM_VELTS (ra1))[i1]);
1283 break;
1284 case scm_tc7_uvect:
1285 case scm_tc7_ivect:
1286 for (; n-- > 0; i0 += inc0, i1 += inc1)
f1267706 1287 dst[i0] = SCM_DSUBRF (proc) (SCM_UNPACK (SCM_VELTS (ra1)[i1]));
c209c88e
GB
1288 break;
1289 }
1290 break;
0f2d19dd 1291 }
c209c88e 1292 }
0f2d19dd
JB
1293 return 1;
1294}
1295
1296
1cc91f1b 1297
0f2d19dd 1298static int
1bbd0b84 1299ramap_rp (SCM ra0,SCM proc,SCM ras)
0f2d19dd
JB
1300{
1301 SCM ra1 = SCM_CAR (ras), ra2 = SCM_CAR (SCM_CDR (ras));
1302 SCM e1 = SCM_UNDEFINED, e2 = SCM_UNDEFINED;
1303 long n = SCM_ARRAY_DIMS (ra0)->ubnd - SCM_ARRAY_DIMS (ra0)->lbnd + 1;
1304 scm_sizet i0 = SCM_ARRAY_BASE (ra0), i1 = SCM_ARRAY_BASE (ra1), i2 = SCM_ARRAY_BASE (ra2);
1305 long inc0 = SCM_ARRAY_DIMS (ra0)->inc;
1306 long inc1 = SCM_ARRAY_DIMS (ra1)->inc;
1307 long inc2 = SCM_ARRAY_DIMS (ra1)->inc;
1308 ra0 = SCM_ARRAY_V (ra0);
1309 ra1 = SCM_ARRAY_V (ra1);
1310 ra2 = SCM_ARRAY_V (ra2);
1311 switch (SCM_TYP7 (ra1) == SCM_TYP7 (ra2) ? SCM_TYP7 (ra1) : 0)
1312 {
1313 default:
1314 for (; n-- > 0; i0 += inc0, i1 += inc1, i2 += inc2)
c209c88e
GB
1315 if (SCM_BITVEC_REF (ra0, i0))
1316 if (SCM_FALSEP (SCM_SUBRF (proc) (RVREF (ra1, i1, e1), RVREF (ra2, i2, e2))))
1317 SCM_BITVEC_CLR (ra0, i0);
0f2d19dd
JB
1318 break;
1319 case scm_tc7_uvect:
1320 case scm_tc7_ivect:
1321 for (; n-- > 0; i0 += inc0, i1 += inc1, i2 += inc2)
c209c88e
GB
1322 if (SCM_BITVEC_REF (ra0, i0))
1323 {
1324 if (SCM_FALSEP (SCM_SUBRF (proc) (SCM_MAKINUM (SCM_VELTS (ra1)[i1]),
1325 SCM_MAKINUM (SCM_VELTS (ra2)[i2]))))
1326 SCM_BITVEC_CLR (ra0, i0);
1327 }
0f2d19dd 1328 break;
0f2d19dd
JB
1329 case scm_tc7_fvect:
1330 {
950cc72b 1331 SCM a1 = scm_make_real (1.0), a2 = scm_make_real (1.0);
0f2d19dd 1332 for (; n-- > 0; i0 += inc0, i1 += inc1, i2 += inc2)
c209c88e
GB
1333 if (SCM_BITVEC_REF (ra0, i0))
1334 {
950cc72b
MD
1335 SCM_REAL_VALUE (a1) = ((float *) SCM_VELTS (ra1))[i1];
1336 SCM_REAL_VALUE (a2) = ((float *) SCM_VELTS (ra2))[i2];
c209c88e
GB
1337 if (SCM_FALSEP (SCM_SUBRF (proc) (a1, a2)))
1338 SCM_BITVEC_CLR (ra0, i0);
1339 }
0f2d19dd
JB
1340 break;
1341 }
0f2d19dd
JB
1342 case scm_tc7_dvect:
1343 {
1344 SCM a1 = scm_makdbl (1.0 / 3.0, 0.0), a2 = scm_makdbl (1.0 / 3.0, 0.0);
1345 for (; n-- > 0; i0 += inc0, i1 += inc1, i2 += inc2)
c209c88e
GB
1346 if (SCM_BITVEC_REF (ra0, i0))
1347 {
1348 SCM_REAL (a1) = ((double *) SCM_VELTS (ra1))[i1];
1349 SCM_REAL (a2) = ((double *) SCM_VELTS (ra2))[i2];
1350 if (SCM_FALSEP (SCM_SUBRF (proc) (a1, a2)))
1351 SCM_BITVEC_CLR (ra0, i0);
1352 }
0f2d19dd
JB
1353 break;
1354 }
1355 case scm_tc7_cvect:
1356 {
1357 SCM a1 = scm_makdbl (1.0, 1.0), a2 = scm_makdbl (1.0, 1.0);
1358 for (; n-- > 0; i0 += inc0, i1 += inc1, i2 += inc2)
c209c88e
GB
1359 if (SCM_BITVEC_REF (ra0, i0))
1360 {
950cc72b
MD
1361 SCM_COMPLEX_REAL (a1) = ((double *) SCM_VELTS (ra1))[2 * i1];
1362 SCM_COMPLEX_IMAG (a1) = ((double *) SCM_VELTS (ra1))[2 * i1 + 1];
1363 SCM_COMPLEX_REAL (a2) = ((double *) SCM_VELTS (ra2))[2 * i2];
1364 SCM_COMPLEX_IMAG (a2) = ((double *) SCM_VELTS (ra2))[2 * i2 + 1];
c209c88e
GB
1365 if (SCM_FALSEP (SCM_SUBRF (proc) (a1, a2)))
1366 SCM_BITVEC_CLR (ra0, i0);
1367 }
0f2d19dd
JB
1368 break;
1369 }
0f2d19dd
JB
1370 }
1371 return 1;
1372}
1373
1374
1cc91f1b 1375
0f2d19dd 1376static int
1bbd0b84 1377ramap_1 (SCM ra0,SCM proc,SCM ras)
0f2d19dd
JB
1378{
1379 SCM ra1 = SCM_CAR (ras);
1380 SCM e1 = SCM_UNDEFINED;
1381 long n = SCM_ARRAY_DIMS (ra0)->ubnd - SCM_ARRAY_DIMS (ra0)->lbnd + 1;
1382 scm_sizet i0 = SCM_ARRAY_BASE (ra0), i1 = SCM_ARRAY_BASE (ra1);
1383 long inc0 = SCM_ARRAY_DIMS (ra0)->inc, inc1 = SCM_ARRAY_DIMS (ra1)->inc;
1384 ra0 = SCM_ARRAY_V (ra0);
1385 ra1 = SCM_ARRAY_V (ra1);
95f5b0f5 1386 if (scm_tc7_vector == SCM_TYP7 (ra0) || scm_tc7_wvect == SCM_TYP7 (ra0))
0f2d19dd
JB
1387 for (; n-- > 0; i0 += inc0, i1 += inc1)
1388 scm_array_set_x (ra0, SCM_SUBRF (proc) (scm_cvref (ra1, i1, SCM_UNDEFINED)), SCM_MAKINUM (i0));
1389 else
1390 for (; n-- > 0; i0 += inc0, i1 += inc1)
1391 scm_array_set_x (ra0, SCM_SUBRF (proc) (RVREF (ra1, i1, e1)), SCM_MAKINUM (i0));
1392 return 1;
1393}
1394
1395
1cc91f1b 1396
0f2d19dd 1397static int
1bbd0b84 1398ramap_2o (SCM ra0,SCM proc,SCM ras)
0f2d19dd
JB
1399{
1400 SCM ra1 = SCM_CAR (ras);
1401 SCM e1 = SCM_UNDEFINED;
1402 long n = SCM_ARRAY_DIMS (ra0)->ubnd - SCM_ARRAY_DIMS (ra0)->lbnd + 1;
1403 scm_sizet i0 = SCM_ARRAY_BASE (ra0), i1 = SCM_ARRAY_BASE (ra1);
1404 long inc0 = SCM_ARRAY_DIMS (ra0)->inc, inc1 = SCM_ARRAY_DIMS (ra1)->inc;
1405 ra0 = SCM_ARRAY_V (ra0);
1406 ra1 = SCM_ARRAY_V (ra1);
1407 ras = SCM_CDR (ras);
ff467021 1408 if (SCM_NULLP (ras))
c209c88e
GB
1409 {
1410 if (scm_tc7_vector == SCM_TYP7 (ra0)
1411 || scm_tc7_wvect == SCM_TYP7 (ra0))
95f5b0f5 1412
c209c88e
GB
1413 for (; n-- > 0; i0 += inc0, i1 += inc1)
1414 scm_array_set_x (ra0, SCM_SUBRF (proc) (scm_cvref (ra1, i1, SCM_UNDEFINED), SCM_UNDEFINED),
1415 SCM_MAKINUM (i0));
1416 else
1417 for (; n-- > 0; i0 += inc0, i1 += inc1)
1418 scm_array_set_x (ra0, SCM_SUBRF (proc) (RVREF (ra1, i1, e1), SCM_UNDEFINED),
1419 SCM_MAKINUM (i0));
1420 }
0f2d19dd
JB
1421 else
1422 {
1423 SCM ra2 = SCM_CAR (ras);
1424 SCM e2 = SCM_UNDEFINED;
1425 scm_sizet i2 = SCM_ARRAY_BASE (ra2);
1426 long inc2 = SCM_ARRAY_DIMS (ra2)->inc;
1427 ra2 = SCM_ARRAY_V (ra2);
95f5b0f5 1428 if (scm_tc7_vector == SCM_TYP7 (ra0) || scm_tc7_wvect == SCM_TYP7 (ra0))
0f2d19dd
JB
1429 for (; n-- > 0; i0 += inc0, i1 += inc1, i2 += inc2)
1430 scm_array_set_x (ra0,
c209c88e
GB
1431 SCM_SUBRF (proc) (scm_cvref (ra1, i1, SCM_UNDEFINED), scm_cvref (ra2, i2, SCM_UNDEFINED)),
1432 SCM_MAKINUM (i0));
0f2d19dd
JB
1433 else
1434 for (; n-- > 0; i0 += inc0, i1 += inc1, i2 += inc2)
1435 scm_array_set_x (ra0,
c209c88e
GB
1436 SCM_SUBRF (proc) (RVREF (ra1, i1, e1), RVREF (ra2, i2, e2)),
1437 SCM_MAKINUM (i0));
0f2d19dd
JB
1438 }
1439 return 1;
1440}
1441
1442
1cc91f1b 1443
0f2d19dd 1444static int
1bbd0b84 1445ramap_a (SCM ra0,SCM proc,SCM ras)
0f2d19dd
JB
1446{
1447 SCM e0 = SCM_UNDEFINED, e1 = SCM_UNDEFINED;
1448 long n = SCM_ARRAY_DIMS (ra0)->ubnd - SCM_ARRAY_DIMS (ra0)->lbnd + 1;
1449 scm_sizet i0 = SCM_ARRAY_BASE (ra0);
1450 long inc0 = SCM_ARRAY_DIMS (ra0)->inc;
1451 ra0 = SCM_ARRAY_V (ra0);
ff467021 1452 if (SCM_NULLP (ras))
c209c88e
GB
1453 for (; n-- > 0; i0 += inc0)
1454 scm_array_set_x (ra0, SCM_SUBRF (proc) (RVREF (ra0, i0, e0), SCM_UNDEFINED), SCM_MAKINUM (i0));
0f2d19dd
JB
1455 else
1456 {
1457 SCM ra1 = SCM_CAR (ras);
1458 scm_sizet i1 = SCM_ARRAY_BASE (ra1);
1459 long inc1 = SCM_ARRAY_DIMS (ra1)->inc;
1460 ra1 = SCM_ARRAY_V (ra1);
1461 for (; n-- > 0; i0 += inc0, i1 += inc1)
1462 scm_array_set_x (ra0, SCM_SUBRF (proc) (RVREF (ra0, i0, e0), RVREF (ra1, i1, e1)),
c209c88e 1463 SCM_MAKINUM (i0));
0f2d19dd
JB
1464 }
1465 return 1;
1466}
1467
dba2ab49 1468/* This name is obsolete. Will go away in release 1.5. */
1bbd0b84
GB
1469SCM_REGISTER_PROC(s_serial_array_map_x, "serial-array-map!", 2, 0, 1, scm_array_map_x);
1470SCM_REGISTER_PROC(s_array_map_in_order_x, "array-map-in-order!", 2, 0, 1, scm_array_map_x);
1cc91f1b 1471
1bbd0b84 1472
3b3b36dd 1473SCM_DEFINE (scm_array_map_x, "array-map!", 2, 0, 1,
c209c88e 1474 (SCM ra0, SCM proc, SCM lra),
b380b885
MD
1475 "@var{array1}, @dots{} must have the same number of dimensions as\n"
1476 "@var{array0} and have a range for each index which includes the range\n"
1477 "for the corresponding index in @var{array0}. @var{proc} is applied to\n"
1478 "each tuple of elements of @var{array1} @dots{} and the result is stored\n"
1479 "as the corresponding element in @var{array0}. The value returned is\n"
1480 "unspecified. The order of application is unspecified.")
1bbd0b84 1481#define FUNC_NAME s_scm_array_map_x
0f2d19dd 1482{
3b3b36dd 1483 SCM_VALIDATE_PROC (2,proc);
0f2d19dd 1484 switch (SCM_TYP7 (proc))
c209c88e
GB
1485 {
1486 default:
1487 gencase:
1488 scm_ramapc (ramap, proc, ra0, lra, FUNC_NAME);
1489 return SCM_UNSPECIFIED;
1490 case scm_tc7_subr_1:
1491 scm_ramapc (ramap_1, proc, ra0, lra, FUNC_NAME);
1492 return SCM_UNSPECIFIED;
1493 case scm_tc7_subr_2:
1494 case scm_tc7_subr_2o:
1495 scm_ramapc (ramap_2o, proc, ra0, lra, FUNC_NAME);
1496 return SCM_UNSPECIFIED;
1497 case scm_tc7_cxr:
1498 if (!SCM_SUBRF (proc))
1499 goto gencase;
1500 scm_ramapc (ramap_cxr, proc, ra0, lra, FUNC_NAME);
1501 return SCM_UNSPECIFIED;
1502 case scm_tc7_rpsubr:
0f2d19dd 1503 {
c209c88e
GB
1504 ra_iproc *p;
1505 if (SCM_FALSEP (scm_array_p (ra0, SCM_BOOL_T)))
0f2d19dd 1506 goto gencase;
c209c88e
GB
1507 scm_array_fill_x (ra0, SCM_BOOL_T);
1508 for (p = ra_rpsubrs; p->name; p++)
1509 if (proc == p->sproc)
1510 {
1511 while (SCM_NNULLP (lra) && SCM_NNULLP (SCM_CDR (lra)))
1512 {
1513 scm_ramapc (p->vproc, SCM_UNDEFINED, ra0, lra, FUNC_NAME);
1514 lra = SCM_CDR (lra);
1515 }
1516 return SCM_UNSPECIFIED;
1517 }
1518 while (SCM_NNULLP (lra) && SCM_NNULLP (SCM_CDR (lra)))
1519 {
1520 scm_ramapc (ramap_rp, proc, ra0, lra, FUNC_NAME);
1521 lra = SCM_CDR (lra);
1522 }
0f2d19dd 1523 return SCM_UNSPECIFIED;
c209c88e
GB
1524 }
1525 case scm_tc7_asubr:
1526 if (SCM_NULLP (lra))
1527 {
1528 SCM prot, fill = SCM_SUBRF (proc) (SCM_UNDEFINED, SCM_UNDEFINED);
1529 if (SCM_INUMP(fill))
1530 {
1531 prot = scm_array_prototype (ra0);
1532 if (SCM_INEXP (prot))
1533 fill = scm_makdbl ((double) SCM_INUM (fill), 0.0);
1534 }
1535
1536 scm_array_fill_x (ra0, fill);
1537 }
1538 else
0f2d19dd 1539 {
c209c88e
GB
1540 SCM tail, ra1 = SCM_CAR (lra);
1541 SCM v0 = (SCM_ARRAYP (ra0) ? SCM_ARRAY_V (ra0) : ra0);
0f2d19dd 1542 ra_iproc *p;
c209c88e
GB
1543 /* Check to see if order might matter.
1544 This might be an argument for a separate
1545 SERIAL-ARRAY-MAP! */
1546 if (v0 == ra1 || (SCM_ARRAYP (ra1) && v0 == SCM_ARRAY_V (ra1)))
1547 if (ra0 != ra1 || (SCM_ARRAYP(ra0) && !SCM_ARRAY_CONTP(ra0)))
1548 goto gencase;
1549 for (tail = SCM_CDR (lra); SCM_NNULLP (tail); tail = SCM_CDR (tail))
1550 {
1551 ra1 = SCM_CAR (tail);
1552 if (v0 == ra1 || (SCM_ARRAYP (ra1) && v0 == SCM_ARRAY_V (ra1)))
1553 goto gencase;
1554 }
1555 for (p = ra_asubrs; p->name; p++)
0f2d19dd
JB
1556 if (proc == p->sproc)
1557 {
c209c88e
GB
1558 if (ra0 != SCM_CAR (lra))
1559 scm_ramapc (scm_array_identity, SCM_UNDEFINED, ra0, scm_cons (SCM_CAR (lra), SCM_EOL), FUNC_NAME);
1560 lra = SCM_CDR (lra);
1561 while (1)
0f2d19dd 1562 {
c209c88e
GB
1563 scm_ramapc (p->vproc, SCM_UNDEFINED, ra0, lra, FUNC_NAME);
1564 if (SCM_IMP (lra) || SCM_IMP (SCM_CDR (lra)))
1565 return SCM_UNSPECIFIED;
0f2d19dd
JB
1566 lra = SCM_CDR (lra);
1567 }
0f2d19dd 1568 }
c209c88e
GB
1569 scm_ramapc (ramap_2o, proc, ra0, lra, FUNC_NAME);
1570 lra = SCM_CDR (lra);
1571 if (SCM_NIMP (lra))
1572 for (lra = SCM_CDR (lra); SCM_NIMP (lra); lra = SCM_CDR (lra))
1573 scm_ramapc (ramap_a, proc, ra0, lra, FUNC_NAME);
0f2d19dd 1574 }
c209c88e
GB
1575 return SCM_UNSPECIFIED;
1576 }
0f2d19dd 1577}
1bbd0b84 1578#undef FUNC_NAME
0f2d19dd 1579
1cc91f1b 1580
0f2d19dd 1581static int
1bbd0b84 1582rafe (SCM ra0,SCM proc,SCM ras)
0f2d19dd
JB
1583{
1584 long i = SCM_ARRAY_DIMS (ra0)->lbnd;
1585 scm_sizet i0 = SCM_ARRAY_BASE (ra0);
1586 long inc0 = SCM_ARRAY_DIMS (ra0)->inc;
1587 long n = SCM_ARRAY_DIMS (ra0)->ubnd;
1588 ra0 = SCM_ARRAY_V (ra0);
ff467021 1589 if (SCM_NULLP (ras))
c209c88e
GB
1590 for (; i <= n; i++, i0 += inc0)
1591 scm_apply (proc, scm_cvref (ra0, i0, SCM_UNDEFINED), scm_listofnull);
0f2d19dd
JB
1592 else
1593 {
1594 SCM ra1 = SCM_CAR (ras);
1595 SCM args, *ve = &ras;
1596 scm_sizet k, i1 = SCM_ARRAY_BASE (ra1);
1597 long inc1 = SCM_ARRAY_DIMS (ra1)->inc;
1598 ra1 = SCM_ARRAY_V (ra1);
1599 ras = SCM_CDR (ras);
ff467021 1600 if (SCM_NULLP(ras))
c209c88e 1601 ras = scm_nullvect;
0f2d19dd
JB
1602 else
1603 {
1604 ras = scm_vector (ras);
1605 ve = SCM_VELTS (ras);
1606 }
1607 for (; i <= n; i++, i0 += inc0, i1 += inc1)
1608 {
1609 args = SCM_EOL;
1610 for (k = SCM_LENGTH (ras); k--;)
1611 args = scm_cons (scm_uniform_vector_ref (ve[k], SCM_MAKINUM (i)), args);
1612 args = scm_cons2 (scm_cvref (ra0, i0, SCM_UNDEFINED), scm_cvref (ra1, i1, SCM_UNDEFINED), args);
1613 scm_apply (proc, args, SCM_EOL);
1614 }
1615 }
1616 return 1;
1617}
1618
1619
3b3b36dd 1620SCM_DEFINE (scm_array_for_each, "array-for-each", 2, 0, 1,
c209c88e 1621 (SCM proc, SCM ra0, SCM lra),
b380b885
MD
1622 "@var{proc} is applied to each tuple of elements of @var{array0} @dots{}\n"
1623 "in row-major order. The value returned is unspecified.")
1bbd0b84 1624#define FUNC_NAME s_scm_array_for_each
0f2d19dd 1625{
3b3b36dd 1626 SCM_VALIDATE_PROC (1,proc);
c209c88e 1627 scm_ramapc (rafe, proc, ra0, lra, FUNC_NAME);
0f2d19dd
JB
1628 return SCM_UNSPECIFIED;
1629}
1bbd0b84 1630#undef FUNC_NAME
0f2d19dd 1631
3b3b36dd 1632SCM_DEFINE (scm_array_index_map_x, "array-index-map!", 2, 0, 0,
c209c88e 1633 (SCM ra, SCM proc),
b380b885
MD
1634 "applies @var{proc} to the indices of each element of @var{array} in\n"
1635 "turn, storing the result in the corresponding element. The value\n"
1636 "returned and the order of application are unspecified.\n\n"
1637 "One can implement @var{array-indexes} as\n"
1638 "@example\n"
1639 "(define (array-indexes array)\n"
1640 " (let ((ra (apply make-array #f (array-shape array))))\n"
1641 " (array-index-map! ra (lambda x x))\n"
1642 " ra))\n"
1643 "@end example\n"
1644 "Another example:\n"
1645 "@example\n"
1646 "(define (apl:index-generator n)\n"
1647 " (let ((v (make-uniform-vector n 1)))\n"
1648 " (array-index-map! v (lambda (i) i))\n"
1649 " v))\n"
1650 "@end example")
1bbd0b84 1651#define FUNC_NAME s_scm_array_index_map_x
0f2d19dd
JB
1652{
1653 scm_sizet i;
6b5a304f 1654 SCM_VALIDATE_NIM (1,ra);
3b3b36dd 1655 SCM_VALIDATE_PROC (2,proc);
e42c09cc
RS
1656 switch (SCM_TYP7(ra))
1657 {
1658 default:
1bbd0b84 1659 badarg:SCM_WTA (1,ra);
e42c09cc
RS
1660 case scm_tc7_vector:
1661 case scm_tc7_wvect:
0f2d19dd 1662 {
e42c09cc 1663 SCM *ve = SCM_VELTS (ra);
0f2d19dd 1664 for (i = 0; i < SCM_LENGTH (ra); i++)
e42c09cc
RS
1665 ve[i] = scm_apply (proc, SCM_MAKINUM (i), scm_listofnull);
1666 return SCM_UNSPECIFIED;
1667 }
1668 case scm_tc7_string:
1669 case scm_tc7_byvect:
1670 case scm_tc7_bvect:
1671 case scm_tc7_uvect:
1672 case scm_tc7_ivect:
05f92e0f 1673 case scm_tc7_svect:
5c11cc9d 1674#ifdef HAVE_LONG_LONGS
05f92e0f
JB
1675 case scm_tc7_llvect:
1676#endif
e42c09cc
RS
1677 case scm_tc7_fvect:
1678 case scm_tc7_dvect:
1679 case scm_tc7_cvect:
1680 for (i = 0; i < SCM_LENGTH (ra); i++)
1681 scm_array_set_x (ra, scm_apply (proc, SCM_MAKINUM (i), scm_listofnull),
1682 SCM_MAKINUM (i));
1683 return SCM_UNSPECIFIED;
1684 case scm_tc7_smob:
1685 SCM_ASRTGO (SCM_ARRAYP (ra), badarg);
1686 {
1687 SCM args = SCM_EOL;
1688 SCM inds = scm_make_uve (SCM_ARRAY_NDIM (ra), SCM_MAKINUM (-1L));
c209c88e 1689 long *vinds = (long *) SCM_VELTS (inds);
e42c09cc
RS
1690 int j, k, kmax = SCM_ARRAY_NDIM (ra) - 1;
1691 if (kmax < 0)
1692 return scm_array_set_x (ra, scm_apply(proc, SCM_EOL, SCM_EOL),
1693 SCM_EOL);
1694 for (k = 0; k <= kmax; k++)
1695 vinds[k] = SCM_ARRAY_DIMS (ra)[k].lbnd;
1696 k = kmax;
1697 do
1698 {
1699 if (k == kmax)
1700 {
1701 vinds[k] = SCM_ARRAY_DIMS (ra)[k].lbnd;
1702 i = cind (ra, inds);
1703 for (; vinds[k] <= SCM_ARRAY_DIMS (ra)[k].ubnd; vinds[k]++)
1704 {
1705 for (j = kmax + 1, args = SCM_EOL; j--;)
1706 args = scm_cons (SCM_MAKINUM (vinds[j]), args);
1707 scm_array_set_x (SCM_ARRAY_V (ra),
1708 scm_apply (proc, args, SCM_EOL),
1709 SCM_MAKINUM (i));
1710 i += SCM_ARRAY_DIMS (ra)[k].inc;
1711 }
1712 k--;
1713 continue;
1714 }
1715 if (vinds[k] < SCM_ARRAY_DIMS (ra)[k].ubnd)
1716 {
1717 vinds[k]++;
1718 k++;
1719 continue;
1720 }
1721 vinds[k] = SCM_ARRAY_DIMS (ra)[k].lbnd - 1;
1722 k--;
1723 }
1724 while (k >= 0);
0f2d19dd 1725 return SCM_UNSPECIFIED;
0f2d19dd 1726 }
e42c09cc 1727 }
0f2d19dd 1728}
1bbd0b84 1729#undef FUNC_NAME
0f2d19dd 1730
1cc91f1b 1731
0f2d19dd 1732static int
1bbd0b84 1733raeql_1 (SCM ra0,SCM as_equal,SCM ra1)
0f2d19dd
JB
1734{
1735 SCM e0 = SCM_UNDEFINED, e1 = SCM_UNDEFINED;
1736 scm_sizet i0 = 0, i1 = 0;
1737 long inc0 = 1, inc1 = 1;
1738 scm_sizet n = SCM_LENGTH (ra0);
1739 ra1 = SCM_CAR (ra1);
ff467021 1740 if (SCM_ARRAYP(ra0))
c209c88e
GB
1741 {
1742 n = SCM_ARRAY_DIMS (ra0)->ubnd - SCM_ARRAY_DIMS (ra0)->lbnd + 1;
1743 i0 = SCM_ARRAY_BASE (ra0);
1744 inc0 = SCM_ARRAY_DIMS (ra0)->inc;
1745 ra0 = SCM_ARRAY_V (ra0);
1746 }
ff467021 1747 if (SCM_ARRAYP (ra1))
c209c88e
GB
1748 {
1749 i1 = SCM_ARRAY_BASE (ra1);
1750 inc1 = SCM_ARRAY_DIMS (ra1)->inc;
1751 ra1 = SCM_ARRAY_V (ra1);
1752 }
1753 switch (SCM_TYP7 (ra0))
1754 {
1755 case scm_tc7_vector:
1756 case scm_tc7_wvect:
1757 default:
1758 for (; n--; i0 += inc0, i1 += inc1)
1759 {
1760 if (SCM_FALSEP (as_equal))
1761 {
1762 if (SCM_FALSEP (scm_array_equal_p (RVREF (ra0, i0, e0), RVREF (ra1, i1, e1))))
1763 return 0;
1764 }
1765 else if (SCM_FALSEP (scm_equal_p (RVREF (ra0, i0, e0), RVREF (ra1, i1, e1))))
1766 return 0;
1767 }
1768 return 1;
1769 case scm_tc7_string:
1770 case scm_tc7_byvect:
0f2d19dd 1771 {
c209c88e
GB
1772 char *v0 = SCM_CHARS (ra0) + i0;
1773 char *v1 = SCM_CHARS (ra1) + i1;
1774 for (; n--; v0 += inc0, v1 += inc1)
1775 if (*v0 != *v1)
1776 return 0;
1777 return 1;
0f2d19dd 1778 }
c209c88e
GB
1779 case scm_tc7_bvect:
1780 for (; n--; i0 += inc0, i1 += inc1)
1781 if (SCM_BITVEC_REF (ra0, i0) != SCM_BITVEC_REF (ra1, i1))
1782 return 0;
1783 return 1;
1784 case scm_tc7_uvect:
1785 case scm_tc7_ivect:
0f2d19dd 1786 {
c209c88e
GB
1787 long *v0 = (long *) SCM_VELTS (ra0) + i0;
1788 long *v1 = (long *) SCM_VELTS (ra1) + i1;
1789 for (; n--; v0 += inc0, v1 += inc1)
1790 if (*v0 != *v1)
1791 return 0;
0f2d19dd 1792 return 1;
c209c88e
GB
1793 }
1794 case scm_tc7_svect:
1795 {
1796 short *v0 = (short *) SCM_VELTS (ra0) + i0;
1797 short *v1 = (short *) SCM_VELTS (ra1) + i1;
1798 for (; n--; v0 += inc0, v1 += inc1)
1799 if (*v0 != *v1)
0f2d19dd
JB
1800 return 0;
1801 return 1;
c209c88e 1802 }
5c11cc9d 1803#ifdef HAVE_LONG_LONGS
c209c88e
GB
1804 case scm_tc7_llvect:
1805 {
1806 long long *v0 = (long long *) SCM_VELTS (ra0) + i0;
1807 long long *v1 = (long long *) SCM_VELTS (ra1) + i1;
1808 for (; n--; v0 += inc0, v1 += inc1)
1809 if (*v0 != *v1)
1810 return 0;
1811 return 1;
1812 }
05f92e0f 1813#endif
c209c88e
GB
1814 case scm_tc7_fvect:
1815 {
1816 float *v0 = (float *) SCM_VELTS (ra0) + i0;
1817 float *v1 = (float *) SCM_VELTS (ra1) + i1;
1818 for (; n--; v0 += inc0, v1 += inc1)
1819 if (*v0 != *v1)
1820 return 0;
1821 return 1;
1822 }
c209c88e
GB
1823 case scm_tc7_dvect:
1824 {
1825 double *v0 = (double *) SCM_VELTS (ra0) + i0;
1826 double *v1 = (double *) SCM_VELTS (ra1) + i1;
1827 for (; n--; v0 += inc0, v1 += inc1)
1828 if (*v0 != *v1)
1829 return 0;
1830 return 1;
1831 }
1832 case scm_tc7_cvect:
1833 {
1834 double (*v0)[2] = (double (*)[2]) SCM_VELTS (ra0) + i0;
1835 double (*v1)[2] = (double (*)[2]) SCM_VELTS (ra1) + i1;
1836 for (; n--; v0 += inc0, v1 += inc1)
1837 {
1838 if ((*v0)[0] != (*v1)[0])
0f2d19dd 1839 return 0;
c209c88e
GB
1840 if ((*v0)[1] != (*v1)[1])
1841 return 0;
1842 }
1843 return 1;
0f2d19dd 1844 }
c209c88e 1845 }
0f2d19dd
JB
1846}
1847
1848
1cc91f1b 1849
0f2d19dd 1850static int
1bbd0b84 1851raeql (SCM ra0,SCM as_equal,SCM ra1)
0f2d19dd
JB
1852{
1853 SCM v0 = ra0, v1 = ra1;
1854 scm_array_dim dim0, dim1;
1855 scm_array_dim *s0 = &dim0, *s1 = &dim1;
1856 scm_sizet bas0 = 0, bas1 = 0;
1857 int k, unroll = 1, vlen = 1, ndim = 1;
ff467021 1858 if (SCM_ARRAYP (ra0))
c209c88e
GB
1859 {
1860 ndim = SCM_ARRAY_NDIM (ra0);
1861 s0 = SCM_ARRAY_DIMS (ra0);
1862 bas0 = SCM_ARRAY_BASE (ra0);
1863 v0 = SCM_ARRAY_V (ra0);
1864 }
0f2d19dd
JB
1865 else
1866 {
1867 s0->inc = 1;
1868 s0->lbnd = 0;
1869 s0->ubnd = SCM_LENGTH (v0) - 1;
1870 unroll = 0;
1871 }
ff467021 1872 if (SCM_ARRAYP (ra1))
c209c88e
GB
1873 {
1874 if (ndim != SCM_ARRAY_NDIM (ra1))
1875 return 0;
1876 s1 = SCM_ARRAY_DIMS (ra1);
1877 bas1 = SCM_ARRAY_BASE (ra1);
1878 v1 = SCM_ARRAY_V (ra1);
1879 }
0f2d19dd
JB
1880 else
1881 {
c209c88e
GB
1882 /*
1883 Huh ? Schizophrenic return type. --hwn
1884 */
0f2d19dd 1885 if (1 != ndim)
c209c88e 1886 return 0;
0f2d19dd
JB
1887 s1->inc = 1;
1888 s1->lbnd = 0;
1889 s1->ubnd = SCM_LENGTH (v1) - 1;
1890 unroll = 0;
1891 }
1892 if (SCM_TYP7 (v0) != SCM_TYP7 (v1))
1893 return 0;
1894 for (k = ndim; k--;)
1895 {
1896 if (s0[k].lbnd != s1[k].lbnd || s0[k].ubnd != s1[k].ubnd)
1897 return 0;
1898 if (unroll)
1899 {
1900 unroll = (s0[k].inc == s1[k].inc);
1901 vlen *= s0[k].ubnd - s1[k].lbnd + 1;
1902 }
1903 }
1904 if (unroll && bas0 == bas1 && v0 == v1)
c209c88e 1905 return 1;
0f2d19dd
JB
1906 return scm_ramapc (raeql_1, as_equal, ra0, scm_cons (ra1, SCM_EOL), "");
1907}
1908
1cc91f1b 1909
0f2d19dd 1910SCM
1bbd0b84 1911scm_raequal (SCM ra0, SCM ra1)
0f2d19dd 1912{
156dcb09 1913 return SCM_BOOL(raeql (ra0, SCM_BOOL_T, ra1));
0f2d19dd
JB
1914}
1915
4079f87e 1916#if 0
c3ee7520
GB
1917/* GJB:FIXME:: Why not use SCM_DEFINE1 for array-equal? */
1918SCM_DEFINE1 (scm_array_equal_p, "array-equal?", scm_tc7_rpsubr,
c209c88e 1919 (SCM ra0, SCM ra1),
b380b885
MD
1920 "Returns @code{#t} iff all arguments are arrays with the same shape, the\n"
1921 "same type, and have corresponding elements which are either\n"
1922 "@code{equal?} or @code{array-equal?}. This function differs from\n"
1923 "@code{equal?} in that a one dimensional shared array may be\n"
1924 "@var{array-equal?} but not @var{equal?} to a vector or uniform vector.")
4079f87e 1925#define FUNC_NAME s_scm_array_equal_p
0f981281
GB
1926{
1927}
4079f87e
GB
1928#undef FUNC_NAME
1929#endif
1930
0f2d19dd
JB
1931static char s_array_equal_p[] = "array-equal?";
1932
1cc91f1b 1933
0f2d19dd 1934SCM
1bbd0b84 1935scm_array_equal_p (SCM ra0, SCM ra1)
0f2d19dd
JB
1936{
1937 if (SCM_IMP (ra0) || SCM_IMP (ra1))
c209c88e 1938 callequal:return scm_equal_p (ra0, ra1);
ff467021 1939 switch (SCM_TYP7(ra0))
c209c88e
GB
1940 {
1941 default:
1942 goto callequal;
1943 case scm_tc7_bvect:
1944 case scm_tc7_string:
1945 case scm_tc7_byvect:
1946 case scm_tc7_uvect:
1947 case scm_tc7_ivect:
1948 case scm_tc7_fvect:
1949 case scm_tc7_dvect:
1950 case scm_tc7_cvect:
1951 case scm_tc7_vector:
1952 case scm_tc7_wvect:
1953 break;
1954 case scm_tc7_smob:
1955 if (!SCM_ARRAYP (ra0))
0f2d19dd 1956 goto callequal;
c209c88e 1957 }
ff467021 1958 switch (SCM_TYP7 (ra1))
c209c88e
GB
1959 {
1960 default:
1961 goto callequal;
1962 case scm_tc7_bvect:
1963 case scm_tc7_string:
1964 case scm_tc7_byvect:
1965 case scm_tc7_uvect:
1966 case scm_tc7_ivect:
1967 case scm_tc7_fvect:
1968 case scm_tc7_dvect:
1969 case scm_tc7_cvect:
1970 case scm_tc7_vector:
1971 case scm_tc7_wvect:
1972 break;
1973 case scm_tc7_smob:
1974 if (!SCM_ARRAYP (ra1))
0f2d19dd 1975 goto callequal;
c209c88e 1976 }
156dcb09 1977 return SCM_BOOL(raeql (ra0, SCM_BOOL_F, ra1));
0f2d19dd
JB
1978}
1979
1980
1981
1cc91f1b 1982static void
1bbd0b84 1983init_raprocs (ra_iproc *subra)
0f2d19dd
JB
1984{
1985 for (; subra->name; subra++)
1986 subra->sproc = SCM_CDR (scm_intern (subra->name, strlen (subra->name)));
1987}
1988
1cc91f1b 1989
0f2d19dd
JB
1990void
1991scm_init_ramap ()
0f2d19dd
JB
1992{
1993 init_raprocs (ra_rpsubrs);
1994 init_raprocs (ra_asubrs);
1995 scm_make_subr (s_array_equal_p, scm_tc7_rpsubr, scm_array_equal_p);
1996 scm_smobs[0x0ff & (scm_tc16_array >> 8)].equalp = scm_raequal;
1997#include "ramap.x"
1bbd0b84 1998 scm_add_feature (s_scm_array_for_each);
0f2d19dd 1999}