*** empty log message ***
[bpt/guile.git] / libguile / unif.c
CommitLineData
7dc6e754 1/* Copyright (C) 1995,1996,1997,1998 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
0f2d19dd
JB
45\f
46
47#include <stdio.h>
48#include "_scm.h"
20e6290e
JB
49#include "chars.h"
50#include "eval.h"
ee149d03 51#include "fports.h"
20e6290e 52#include "smob.h"
20e6290e
JB
53#include "strop.h"
54#include "feature.h"
55
b6791b2e 56#include "validate.h"
20e6290e 57#include "unif.h"
95b88819 58#include "ramap.h"
0f2d19dd 59
3d8d56df
GH
60#ifdef HAVE_UNISTD_H
61#include <unistd.h>
62#endif
63
0f2d19dd
JB
64\f
65/* The set of uniform scm_vector types is:
66 * Vector of: Called:
67 * unsigned char string
68 * char byvect
69 * boolean bvect
a515d287
MD
70 * signed long ivect
71 * unsigned long uvect
0f2d19dd
JB
72 * float fvect
73 * double dvect
74 * complex double cvect
75 * short svect
5c11cc9d 76 * long long llvect
0f2d19dd
JB
77 */
78
79long scm_tc16_array;
80
afe5177e
GH
81/* return the size of an element in a uniform array or 0 if type not
82 found. */
83scm_sizet
84scm_uniform_element_size (SCM obj)
0f2d19dd 85{
afe5177e 86 scm_sizet result;
0f2d19dd 87
afe5177e 88 switch (SCM_TYP7 (obj))
0f2d19dd 89 {
0f2d19dd 90 case scm_tc7_bvect:
0f2d19dd
JB
91 case scm_tc7_uvect:
92 case scm_tc7_ivect:
afe5177e 93 result = sizeof (long);
0f2d19dd 94 break;
afe5177e 95
0f2d19dd 96 case scm_tc7_byvect:
afe5177e 97 result = sizeof (char);
0f2d19dd
JB
98 break;
99
100 case scm_tc7_svect:
afe5177e 101 result = sizeof (short);
0f2d19dd 102 break;
afe5177e 103
5c11cc9d 104#ifdef HAVE_LONG_LONGS
0f2d19dd 105 case scm_tc7_llvect:
afe5177e 106 result = sizeof (long_long);
0f2d19dd
JB
107 break;
108#endif
109
110#ifdef SCM_FLOATS
111#ifdef SCM_SINGLES
112 case scm_tc7_fvect:
afe5177e 113 result = sizeof (float);
0f2d19dd
JB
114 break;
115#endif
afe5177e 116
0f2d19dd 117 case scm_tc7_dvect:
afe5177e 118 result = sizeof (double);
0f2d19dd 119 break;
afe5177e 120
0f2d19dd 121 case scm_tc7_cvect:
afe5177e 122 result = 2 * sizeof (double);
0f2d19dd
JB
123 break;
124#endif
afe5177e
GH
125
126 default:
127 result = 0;
0f2d19dd 128 }
afe5177e 129 return result;
0f2d19dd
JB
130}
131
132
0f2d19dd
JB
133#ifdef SCM_FLOATS
134#ifdef SCM_SINGLES
135
1cc91f1b 136
0f2d19dd 137SCM
805df3e8 138scm_makflo (float x)
0f2d19dd
JB
139{
140 SCM z;
141 if (x == 0.0)
142 return scm_flo0;
143 SCM_NEWCELL (z);
144 SCM_DEFER_INTS;
898a256f 145 SCM_SETCAR (z, scm_tc_flo);
0f2d19dd
JB
146 SCM_FLO (z) = x;
147 SCM_ALLOW_INTS;
148 return z;
149}
150#endif
151#endif
152
1cc91f1b 153
0f2d19dd 154SCM
1bbd0b84 155scm_make_uve (long k, SCM prot)
0f2d19dd
JB
156{
157 SCM v;
158 long i, type;
159 if (SCM_BOOL_T == prot)
160 {
161 i = sizeof (long) * ((k + SCM_LONG_BIT - 1) / SCM_LONG_BIT);
162 type = scm_tc7_bvect;
163 }
7866a09b 164 else if (SCM_CHARP (prot) && (prot == SCM_MAKE_CHAR ('\0')))
0f2d19dd
JB
165 {
166 i = sizeof (char) * k;
167 type = scm_tc7_byvect;
168 }
7866a09b 169 else if (SCM_CHARP (prot))
0f2d19dd
JB
170 {
171 i = sizeof (char) * k;
172 type = scm_tc7_string;
173 }
174 else if (SCM_INUMP (prot))
175 {
176 i = sizeof (long) * k;
177 if (SCM_INUM (prot) > 0)
178 type = scm_tc7_uvect;
179 else
180 type = scm_tc7_ivect;
181 }
0c95b57d 182 else if (SCM_SYMBOLP (prot) && (1 == SCM_LENGTH (prot)))
0f2d19dd
JB
183 {
184 char s;
185
186 s = SCM_CHARS (prot)[0];
187 if (s == 's')
188 {
189 i = sizeof (short) * k;
190 type = scm_tc7_svect;
191 }
5c11cc9d 192#ifdef HAVE_LONG_LONGS
0f2d19dd
JB
193 else if (s == 'l')
194 {
195 i = sizeof (long_long) * k;
196 type = scm_tc7_llvect;
197 }
198#endif
199 else
200 {
a8741caa 201 return scm_make_vector (SCM_MAKINUM (k), SCM_UNDEFINED);
0f2d19dd
JB
202 }
203 }
204 else
205#ifdef SCM_FLOATS
206 if (SCM_IMP (prot) || !SCM_INEXP (prot))
207#endif
208 /* Huge non-unif vectors are NOT supported. */
5c11cc9d
GH
209 /* no special scm_vector */
210 return scm_make_vector (SCM_MAKINUM (k), SCM_UNDEFINED);
0f2d19dd
JB
211#ifdef SCM_FLOATS
212#ifdef SCM_SINGLES
213 else if (SCM_SINGP (prot))
214
215 {
216 i = sizeof (float) * k;
217 type = scm_tc7_fvect;
218 }
219#endif
220 else if (SCM_CPLXP (prot))
221 {
222 i = 2 * sizeof (double) * k;
223 type = scm_tc7_cvect;
224 }
225 else
226 {
227 i = sizeof (double) * k;
228 type = scm_tc7_dvect;
229 }
230#endif
231
232 SCM_NEWCELL (v);
233 SCM_DEFER_INTS;
5c11cc9d 234 SCM_SETCHARS (v, (char *) scm_must_malloc (i ? i : 1, "vector"));
0f2d19dd
JB
235 SCM_SETLENGTH (v, (k < SCM_LENGTH_MAX ? k : SCM_LENGTH_MAX), type);
236 SCM_ALLOW_INTS;
237 return v;
238}
239
3b3b36dd 240SCM_DEFINE (scm_uniform_vector_length, "uniform-vector-length", 1, 0, 0,
1bbd0b84 241 (SCM v),
b380b885 242 "Returns the number of elements in @var{uve}.")
1bbd0b84 243#define FUNC_NAME s_scm_uniform_vector_length
0f2d19dd
JB
244{
245 SCM_ASRTGO (SCM_NIMP (v), badarg1);
246 switch SCM_TYP7
247 (v)
248 {
249 default:
1bbd0b84 250 badarg1:SCM_WTA(1,v);
0f2d19dd
JB
251 case scm_tc7_bvect:
252 case scm_tc7_string:
253 case scm_tc7_byvect:
254 case scm_tc7_uvect:
255 case scm_tc7_ivect:
256 case scm_tc7_fvect:
257 case scm_tc7_dvect:
258 case scm_tc7_cvect:
259 case scm_tc7_vector:
95f5b0f5 260 case scm_tc7_wvect:
0f2d19dd 261 case scm_tc7_svect:
5c11cc9d 262#ifdef HAVE_LONG_LONGS
0f2d19dd
JB
263 case scm_tc7_llvect:
264#endif
265 return SCM_MAKINUM (SCM_LENGTH (v));
266 }
267}
1bbd0b84 268#undef FUNC_NAME
0f2d19dd 269
3b3b36dd 270SCM_DEFINE (scm_array_p, "array?", 1, 1, 0,
1bbd0b84 271 (SCM v, SCM prot),
b380b885
MD
272 "Returns @code{#t} if the @var{obj} is an array, and @code{#f} if not.\n\n"
273 "The @var{prototype} argument is used with uniform arrays and is described\n"
274 "elsewhere.")
1bbd0b84 275#define FUNC_NAME s_scm_array_p
0f2d19dd
JB
276{
277 int nprot;
278 int enclosed;
279 nprot = SCM_UNBNDP (prot);
280 enclosed = 0;
281 if (SCM_IMP (v))
282 return SCM_BOOL_F;
283loop:
284 switch (SCM_TYP7 (v))
285 {
286 case scm_tc7_smob:
287 if (!SCM_ARRAYP (v))
288 return SCM_BOOL_F;
289 if (nprot)
290 return SCM_BOOL_T;
291 if (enclosed++)
292 return SCM_BOOL_F;
293 v = SCM_ARRAY_V (v);
294 goto loop;
295 case scm_tc7_bvect:
1bbd0b84 296 return nprot || SCM_BOOL(SCM_BOOL_T==prot);
0f2d19dd 297 case scm_tc7_string:
7866a09b 298 return nprot || SCM_BOOL(SCM_CHARP(prot) && (prot != SCM_MAKE_CHAR('\0')));
0f2d19dd 299 case scm_tc7_byvect:
7866a09b 300 return nprot || SCM_BOOL(prot == SCM_MAKE_CHAR('\0'));
0f2d19dd 301 case scm_tc7_uvect:
1bbd0b84 302 return nprot || SCM_BOOL(SCM_INUMP(prot) && SCM_INUM(prot)>0);
0f2d19dd 303 case scm_tc7_ivect:
1bbd0b84 304 return nprot || SCM_BOOL(SCM_INUMP(prot) && SCM_INUM(prot)<=0);
0f2d19dd
JB
305 case scm_tc7_svect:
306 return ( nprot
368cf54d 307 || (SCM_SYMBOLP (prot)
0f2d19dd
JB
308 && (1 == SCM_LENGTH (prot))
309 && ('s' == SCM_CHARS (prot)[0])));
5c11cc9d 310#ifdef HAVE_LONG_LONGS
0f2d19dd
JB
311 case scm_tc7_llvect:
312 return ( nprot
368cf54d 313 || (SCM_SYMBOLP (prot)
0f2d19dd
JB
314 && (1 == SCM_LENGTH (prot))
315 && ('s' == SCM_CHARS (prot)[0])));
316#endif
317# ifdef SCM_FLOATS
318# ifdef SCM_SINGLES
319 case scm_tc7_fvect:
0c95b57d 320 return nprot || SCM_BOOL(SCM_SINGP(prot));
0f2d19dd
JB
321# endif
322 case scm_tc7_dvect:
0c95b57d 323 return nprot || SCM_BOOL(SCM_REALP(prot));
0f2d19dd 324 case scm_tc7_cvect:
0c95b57d 325 return nprot || SCM_BOOL(SCM_CPLXP(prot));
0f2d19dd
JB
326# endif
327 case scm_tc7_vector:
95f5b0f5 328 case scm_tc7_wvect:
1bbd0b84 329 return nprot || SCM_BOOL(SCM_NULLP(prot));
0f2d19dd
JB
330 default:;
331 }
332 return SCM_BOOL_F;
333}
1bbd0b84 334#undef FUNC_NAME
0f2d19dd
JB
335
336
3b3b36dd 337SCM_DEFINE (scm_array_rank, "array-rank", 1, 0, 0,
1bbd0b84 338 (SCM ra),
b380b885
MD
339 "Returns the number of dimensions of @var{obj}. If @var{obj} is not an\n"
340 "array, @code{0} is returned.")
1bbd0b84 341#define FUNC_NAME s_scm_array_rank
0f2d19dd
JB
342{
343 if (SCM_IMP (ra))
1bbd0b84 344 return SCM_INUM0;
0f2d19dd
JB
345 switch (SCM_TYP7 (ra))
346 {
347 default:
348 return SCM_INUM0;
349 case scm_tc7_string:
350 case scm_tc7_vector:
95f5b0f5 351 case scm_tc7_wvect:
0f2d19dd
JB
352 case scm_tc7_byvect:
353 case scm_tc7_uvect:
354 case scm_tc7_ivect:
355 case scm_tc7_fvect:
356 case scm_tc7_cvect:
357 case scm_tc7_dvect:
5c11cc9d 358#ifdef HAVE_LONG_LONGS
0f2d19dd
JB
359 case scm_tc7_llvect:
360#endif
361 case scm_tc7_svect:
362 return SCM_MAKINUM (1L);
363 case scm_tc7_smob:
364 if (SCM_ARRAYP (ra))
365 return SCM_MAKINUM (SCM_ARRAY_NDIM (ra));
366 return SCM_INUM0;
367 }
368}
1bbd0b84 369#undef FUNC_NAME
0f2d19dd
JB
370
371
3b3b36dd 372SCM_DEFINE (scm_array_dimensions, "array-dimensions", 1, 0, 0,
1bbd0b84 373 (SCM ra),
b380b885
MD
374 "@code{Array-dimensions} is similar to @code{array-shape} but replaces\n"
375 "elements with a @code{0} minimum with one greater than the maximum. So:\n"
376 "@example\n"
377 "(array-dimensions (make-array 'foo '(-1 3) 5)) @result{} ((-1 3) 5)\n"
378 "@end example")
1bbd0b84 379#define FUNC_NAME s_scm_array_dimensions
0f2d19dd
JB
380{
381 SCM res = SCM_EOL;
382 scm_sizet k;
383 scm_array_dim *s;
384 if (SCM_IMP (ra))
1bbd0b84 385 return SCM_BOOL_F;
0f2d19dd
JB
386 switch (SCM_TYP7 (ra))
387 {
388 default:
389 return SCM_BOOL_F;
390 case scm_tc7_string:
391 case scm_tc7_vector:
95f5b0f5 392 case scm_tc7_wvect:
0f2d19dd
JB
393 case scm_tc7_bvect:
394 case scm_tc7_byvect:
395 case scm_tc7_uvect:
396 case scm_tc7_ivect:
397 case scm_tc7_fvect:
398 case scm_tc7_cvect:
399 case scm_tc7_dvect:
400 case scm_tc7_svect:
5c11cc9d 401#ifdef HAVE_LONG_LONGS
0f2d19dd
JB
402 case scm_tc7_llvect:
403#endif
404 return scm_cons (SCM_MAKINUM (SCM_LENGTH (ra)), SCM_EOL);
405 case scm_tc7_smob:
406 if (!SCM_ARRAYP (ra))
407 return SCM_BOOL_F;
408 k = SCM_ARRAY_NDIM (ra);
409 s = SCM_ARRAY_DIMS (ra);
410 while (k--)
411 res = scm_cons (s[k].lbnd ? scm_cons2 (SCM_MAKINUM (s[k].lbnd), SCM_MAKINUM (s[k].ubnd), SCM_EOL) :
412 SCM_MAKINUM (1 + (s[k].ubnd))
413 , res);
414 return res;
415 }
416}
1bbd0b84 417#undef FUNC_NAME
0f2d19dd
JB
418
419
420static char s_bad_ind[] = "Bad scm_array index";
421
1cc91f1b 422
0f2d19dd 423long
1bbd0b84 424scm_aind (SCM ra, SCM args, const char *what)
0f2d19dd
JB
425{
426 SCM ind;
427 register long j;
428 register scm_sizet pos = SCM_ARRAY_BASE (ra);
429 register scm_sizet k = SCM_ARRAY_NDIM (ra);
430 scm_array_dim *s = SCM_ARRAY_DIMS (ra);
431 if (SCM_INUMP (args))
0f2d19dd 432 {
f5bf2977 433 SCM_ASSERT (1 == k, scm_makfrom0str (what), SCM_WNA, NULL);
0f2d19dd
JB
434 return pos + (SCM_INUM (args) - s->lbnd) * (s->inc);
435 }
436 while (k && SCM_NIMP (args))
437 {
438 ind = SCM_CAR (args);
439 args = SCM_CDR (args);
440 SCM_ASSERT (SCM_INUMP (ind), ind, s_bad_ind, what);
441 j = SCM_INUM (ind);
442 SCM_ASSERT (j >= (s->lbnd) && j <= (s->ubnd), ind, SCM_OUTOFRANGE, what);
443 pos += (j - s->lbnd) * (s->inc);
444 k--;
445 s++;
446 }
f5bf2977
GH
447 SCM_ASSERT (0 == k && SCM_NULLP (args), scm_makfrom0str (what), SCM_WNA,
448 NULL);
0f2d19dd
JB
449 return pos;
450}
451
452
1cc91f1b 453
0f2d19dd 454SCM
1bbd0b84 455scm_make_ra (int ndim)
0f2d19dd
JB
456{
457 SCM ra;
458 SCM_NEWCELL (ra);
459 SCM_DEFER_INTS;
23a62151
MD
460 SCM_NEWSMOB(ra, ((long) ndim << 17) + scm_tc16_array,
461 scm_must_malloc ((long) (sizeof (scm_array) + ndim * sizeof (scm_array_dim)),
0f2d19dd 462 "array"));
0f2d19dd
JB
463 SCM_ARRAY_V (ra) = scm_nullvect;
464 SCM_ALLOW_INTS;
465 return ra;
466}
467
468static char s_bad_spec[] = "Bad scm_array dimension";
469/* Increments will still need to be set. */
470
1cc91f1b 471
0f2d19dd 472SCM
1bbd0b84 473scm_shap2ra (SCM args, const char *what)
0f2d19dd
JB
474{
475 scm_array_dim *s;
476 SCM ra, spec, sp;
477 int ndim = scm_ilength (args);
478 SCM_ASSERT (0 <= ndim, args, s_bad_spec, what);
479 ra = scm_make_ra (ndim);
480 SCM_ARRAY_BASE (ra) = 0;
481 s = SCM_ARRAY_DIMS (ra);
482 for (; SCM_NIMP (args); s++, args = SCM_CDR (args))
483 {
484 spec = SCM_CAR (args);
485 if (SCM_IMP (spec))
486
487 {
20a54673
GH
488 SCM_ASSERT (SCM_INUMP (spec) && SCM_INUM (spec) >= 0, spec,
489 s_bad_spec, what);
0f2d19dd
JB
490 s->lbnd = 0;
491 s->ubnd = SCM_INUM (spec) - 1;
492 s->inc = 1;
493 }
494 else
495 {
20a54673
GH
496 SCM_ASSERT (SCM_CONSP (spec) && SCM_INUMP (SCM_CAR (spec)), spec,
497 s_bad_spec, what);
0f2d19dd
JB
498 s->lbnd = SCM_INUM (SCM_CAR (spec));
499 sp = SCM_CDR (spec);
0c95b57d 500 SCM_ASSERT (SCM_CONSP (sp)
20a54673
GH
501 && SCM_INUMP (SCM_CAR (sp)) && SCM_NULLP (SCM_CDR (sp)),
502 spec, s_bad_spec, what);
0f2d19dd
JB
503 s->ubnd = SCM_INUM (SCM_CAR (sp));
504 s->inc = 1;
505 }
506 }
507 return ra;
508}
509
3b3b36dd 510SCM_DEFINE (scm_dimensions_to_uniform_array, "dimensions->uniform-array", 2, 1, 0,
1bbd0b84 511 (SCM dims, SCM prot, SCM fill),
b380b885
MD
512 "@deffnx primitive make-uniform-vector length prototype [fill]\n"
513 "Creates and returns a uniform array or vector of type corresponding to\n"
514 "@var{prototype} with dimensions @var{dims} or length @var{length}. If\n"
515 "@var{fill} is supplied, it's used to fill the array, otherwise \n"
516 "@var{prototype} is used.")
1bbd0b84 517#define FUNC_NAME s_scm_dimensions_to_uniform_array
0f2d19dd
JB
518{
519 scm_sizet k, vlen = 1;
520 long rlen = 1;
521 scm_array_dim *s;
522 SCM ra;
523 if (SCM_INUMP (dims))
cda139a7 524 {
0f2d19dd
JB
525 if (SCM_INUM (dims) < SCM_LENGTH_MAX)
526 {
5c11cc9d
GH
527 SCM answer = scm_make_uve (SCM_INUM (dims), prot);
528
529 if (!SCM_UNBNDP (fill))
530 scm_array_fill_x (answer, fill);
0c95b57d 531 else if (SCM_SYMBOLP (prot))
0f2d19dd
JB
532 scm_array_fill_x (answer, SCM_MAKINUM (0));
533 else
534 scm_array_fill_x (answer, prot);
535 return answer;
536 }
537 else
538 dims = scm_cons (dims, SCM_EOL);
cda139a7 539 }
0c95b57d
GB
540 SCM_ASSERT (SCM_NULLP (dims) || SCM_CONSP (dims),
541 dims, SCM_ARG1, FUNC_NAME);
1bbd0b84 542 ra = scm_shap2ra (dims, FUNC_NAME);
898a256f 543 SCM_SETOR_CAR (ra, SCM_ARRAY_CONTIGUOUS);
0f2d19dd
JB
544 s = SCM_ARRAY_DIMS (ra);
545 k = SCM_ARRAY_NDIM (ra);
546 while (k--)
547 {
548 s[k].inc = (rlen > 0 ? rlen : 0);
549 rlen = (s[k].ubnd - s[k].lbnd + 1) * s[k].inc;
550 vlen *= (s[k].ubnd - s[k].lbnd + 1);
551 }
552 if (rlen < SCM_LENGTH_MAX)
553 SCM_ARRAY_V (ra) = scm_make_uve ((rlen > 0 ? rlen : 0L), prot);
554 else
555 {
556 scm_sizet bit;
557 switch (SCM_TYP7 (scm_make_uve (0L, prot)))
558 {
559 default:
560 bit = SCM_LONG_BIT;
561 break;
562 case scm_tc7_bvect:
563 bit = 1;
564 break;
565 case scm_tc7_string:
566 bit = SCM_CHAR_BIT;
567 break;
568 case scm_tc7_fvect:
569 bit = sizeof (float) * SCM_CHAR_BIT / sizeof (char);
570 break;
571 case scm_tc7_dvect:
572 bit = sizeof (double) * SCM_CHAR_BIT / sizeof (char);
573 break;
574 case scm_tc7_cvect:
575 bit = 2 * sizeof (double) * SCM_CHAR_BIT / sizeof (char);
576 break;
577 }
578 SCM_ARRAY_BASE (ra) = (SCM_LONG_BIT + bit - 1) / bit;
579 rlen += SCM_ARRAY_BASE (ra);
580 SCM_ARRAY_V (ra) = scm_make_uve (rlen, prot);
581 *((long *) SCM_VELTS (SCM_ARRAY_V (ra))) = rlen;
582 }
5c11cc9d 583 if (!SCM_UNBNDP (fill))
0f2d19dd 584 {
5c11cc9d 585 scm_array_fill_x (ra, fill);
0f2d19dd 586 }
0c95b57d 587 else if (SCM_SYMBOLP (prot))
0f2d19dd
JB
588 scm_array_fill_x (ra, SCM_MAKINUM (0));
589 else
590 scm_array_fill_x (ra, prot);
591 if (1 == SCM_ARRAY_NDIM (ra) && 0 == SCM_ARRAY_BASE (ra))
592 if (s->ubnd < s->lbnd || (0 == s->lbnd && 1 == s->inc))
593 return SCM_ARRAY_V (ra);
594 return ra;
595}
1bbd0b84 596#undef FUNC_NAME
0f2d19dd 597
1cc91f1b 598
0f2d19dd 599void
1bbd0b84 600scm_ra_set_contp (SCM ra)
0f2d19dd
JB
601{
602 scm_sizet k = SCM_ARRAY_NDIM (ra);
0f2d19dd 603 if (k)
0f2d19dd 604 {
fe0c6dae
JB
605 long inc = SCM_ARRAY_DIMS (ra)[k - 1].inc;
606 while (k--)
0f2d19dd 607 {
fe0c6dae
JB
608 if (inc != SCM_ARRAY_DIMS (ra)[k].inc)
609 {
898a256f 610 SCM_SETAND_CAR (ra, ~SCM_ARRAY_CONTIGUOUS);
fe0c6dae
JB
611 return;
612 }
613 inc *= (SCM_ARRAY_DIMS (ra)[k].ubnd
614 - SCM_ARRAY_DIMS (ra)[k].lbnd + 1);
0f2d19dd 615 }
0f2d19dd 616 }
898a256f 617 SCM_SETOR_CAR (ra, SCM_ARRAY_CONTIGUOUS);
0f2d19dd
JB
618}
619
620
3b3b36dd 621SCM_DEFINE (scm_make_shared_array, "make-shared-array", 2, 0, 1,
1bbd0b84 622 (SCM oldra, SCM mapfunc, SCM dims),
b380b885
MD
623 "@code{make-shared-array} can be used to create shared subarrays of other\n"
624 "arrays. The @var{mapper} is a function that translates coordinates in\n"
625 "the new array into coordinates in the old array. A @var{mapper} must be\n"
626 "linear, and its range must stay within the bounds of the old array, but\n"
627 "it can be otherwise arbitrary. A simple example:\n"
628 "@example\n"
629 "(define fred (make-array #f 8 8))\n"
630 "(define freds-diagonal\n"
631 " (make-shared-array fred (lambda (i) (list i i)) 8))\n"
632 "(array-set! freds-diagonal 'foo 3)\n"
633 "(array-ref fred 3 3) @result{} foo\n"
634 "(define freds-center\n"
635 " (make-shared-array fred (lambda (i j) (list (+ 3 i) (+ 3 j))) 2 2))\n"
636 "(array-ref freds-center 0 0) @result{} foo\n"
637 "@end example")
1bbd0b84 638#define FUNC_NAME s_scm_make_shared_array
0f2d19dd
JB
639{
640 SCM ra;
641 SCM inds, indptr;
642 SCM imap;
643 scm_sizet i, k;
644 long old_min, new_min, old_max, new_max;
645 scm_array_dim *s;
3b3b36dd
GB
646 SCM_VALIDATE_ARRAY (1,oldra);
647 SCM_VALIDATE_PROC (2,mapfunc);
1bbd0b84 648 ra = scm_shap2ra (dims, FUNC_NAME);
0f2d19dd
JB
649 if (SCM_ARRAYP (oldra))
650 {
651 SCM_ARRAY_V (ra) = SCM_ARRAY_V (oldra);
652 old_min = old_max = SCM_ARRAY_BASE (oldra);
653 s = SCM_ARRAY_DIMS (oldra);
654 k = SCM_ARRAY_NDIM (oldra);
655 while (k--)
656 {
657 if (s[k].inc > 0)
658 old_max += (s[k].ubnd - s[k].lbnd) * s[k].inc;
659 else
660 old_min += (s[k].ubnd - s[k].lbnd) * s[k].inc;
661 }
662 }
663 else
664 {
665 SCM_ARRAY_V (ra) = oldra;
666 old_min = 0;
667 old_max = (long) SCM_LENGTH (oldra) - 1;
668 }
669 inds = SCM_EOL;
670 s = SCM_ARRAY_DIMS (ra);
671 for (k = 0; k < SCM_ARRAY_NDIM (ra); k++)
672 {
673 inds = scm_cons (SCM_MAKINUM (s[k].lbnd), inds);
674 if (s[k].ubnd < s[k].lbnd)
675 {
676 if (1 == SCM_ARRAY_NDIM (ra))
677 ra = scm_make_uve (0L, scm_array_prototype (ra));
678 else
679 SCM_ARRAY_V (ra) = scm_make_uve (0L, scm_array_prototype (ra));
680 return ra;
681 }
682 }
92396c0a 683 imap = scm_apply (mapfunc, scm_reverse (inds), SCM_EOL);
0f2d19dd 684 if (SCM_ARRAYP (oldra))
1bbd0b84 685 i = (scm_sizet) scm_aind (oldra, imap, FUNC_NAME);
0f2d19dd
JB
686 else
687 {
688 if (SCM_NINUMP (imap))
689
690 {
691 SCM_ASSERT (1 == scm_ilength (imap) && SCM_INUMP (SCM_CAR (imap)),
1bbd0b84 692 imap, s_bad_ind, FUNC_NAME);
0f2d19dd
JB
693 imap = SCM_CAR (imap);
694 }
695 i = SCM_INUM (imap);
696 }
697 SCM_ARRAY_BASE (ra) = new_min = new_max = i;
698 indptr = inds;
699 k = SCM_ARRAY_NDIM (ra);
700 while (k--)
701 {
702 if (s[k].ubnd > s[k].lbnd)
703 {
898a256f 704 SCM_SETCAR (indptr, SCM_MAKINUM (SCM_INUM (SCM_CAR (indptr)) + 1));
0f2d19dd
JB
705 imap = scm_apply (mapfunc, scm_reverse (inds), SCM_EOL);
706 if (SCM_ARRAYP (oldra))
707
1bbd0b84 708 s[k].inc = scm_aind (oldra, imap, FUNC_NAME) - i;
0f2d19dd
JB
709 else
710 {
711 if (SCM_NINUMP (imap))
712
713 {
714 SCM_ASSERT (1 == scm_ilength (imap) && SCM_INUMP (SCM_CAR (imap)),
1bbd0b84 715 imap, s_bad_ind, FUNC_NAME);
0f2d19dd
JB
716 imap = SCM_CAR (imap);
717 }
718 s[k].inc = (long) SCM_INUM (imap) - i;
719 }
720 i += s[k].inc;
721 if (s[k].inc > 0)
722 new_max += (s[k].ubnd - s[k].lbnd) * s[k].inc;
723 else
724 new_min += (s[k].ubnd - s[k].lbnd) * s[k].inc;
725 }
726 else
727 s[k].inc = new_max - new_min + 1; /* contiguous by default */
728 indptr = SCM_CDR (indptr);
729 }
730 SCM_ASSERT (old_min <= new_min && old_max >= new_max, SCM_UNDEFINED,
1bbd0b84 731 "mapping out of range", FUNC_NAME);
0f2d19dd
JB
732 if (1 == SCM_ARRAY_NDIM (ra) && 0 == SCM_ARRAY_BASE (ra))
733 {
734 if (1 == s->inc && 0 == s->lbnd
735 && SCM_LENGTH (SCM_ARRAY_V (ra)) == 1 + s->ubnd)
736 return SCM_ARRAY_V (ra);
737 if (s->ubnd < s->lbnd)
738 return scm_make_uve (0L, scm_array_prototype (ra));
739 }
740 scm_ra_set_contp (ra);
741 return ra;
742}
1bbd0b84 743#undef FUNC_NAME
0f2d19dd
JB
744
745
746/* args are RA . DIMS */
3b3b36dd 747SCM_DEFINE (scm_transpose_array, "transpose-array", 0, 0, 1,
1bbd0b84 748 (SCM args),
b380b885
MD
749 "Returns an array sharing contents with @var{array}, but with dimensions\n"
750 "arranged in a different order. There must be one @var{dim} argument for\n"
751 "each dimension of @var{array}. @var{dim0}, @var{dim1}, @dots{} should\n"
752 "be integers between 0 and the rank of the array to be returned. Each\n"
753 "integer in that range must appear at least once in the argument list.\n\n"
754 "The values of @var{dim0}, @var{dim1}, @dots{} correspond to dimensions\n"
755 "in the array to be returned, their positions in the argument list to\n"
756 "dimensions of @var{array}. Several @var{dim}s may have the same value,\n"
757 "in which case the returned array will have smaller rank than\n"
758 "@var{array}.\n\n"
759 "examples:\n"
760 "@example\n"
761 "(transpose-array '#2((a b) (c d)) 1 0) @result{} #2((a c) (b d))\n"
762 "(transpose-array '#2((a b) (c d)) 0 0) @result{} #1(a d)\n"
763 "(transpose-array '#3(((a b c) (d e f)) ((1 2 3) (4 5 6))) 1 1 0) @result{}\n"
764 " #2((a 4) (b 5) (c 6))\n"
765 "@end example")
1bbd0b84 766#define FUNC_NAME s_scm_transpose_array
0f2d19dd
JB
767{
768 SCM ra, res, vargs, *ve = &vargs;
769 scm_array_dim *s, *r;
770 int ndim, i, k;
1bbd0b84 771 SCM_ASSERT (SCM_NNULLP (args), scm_makfrom0str (FUNC_NAME),
f5bf2977 772 SCM_WNA, NULL);
0f2d19dd 773 ra = SCM_CAR (args);
1bbd0b84 774 SCM_ASSERT (SCM_NIMP (ra), ra, SCM_ARG1, FUNC_NAME);
0f2d19dd 775 args = SCM_CDR (args);
f5bf2977 776 switch (SCM_TYP7 (ra))
0f2d19dd
JB
777 {
778 default:
1bbd0b84 779 badarg:SCM_WTA (1,ra);
0f2d19dd
JB
780 case scm_tc7_bvect:
781 case scm_tc7_string:
782 case scm_tc7_byvect:
783 case scm_tc7_uvect:
784 case scm_tc7_ivect:
785 case scm_tc7_fvect:
786 case scm_tc7_dvect:
787 case scm_tc7_cvect:
788 case scm_tc7_svect:
5c11cc9d 789#ifdef HAVE_LONG_LONGS
0f2d19dd
JB
790 case scm_tc7_llvect:
791#endif
f5bf2977 792 SCM_ASSERT (SCM_NIMP (args) && SCM_NULLP (SCM_CDR (args)),
1bbd0b84 793 scm_makfrom0str (FUNC_NAME), SCM_WNA, NULL);
f5bf2977 794 SCM_ASSERT (SCM_INUMP (SCM_CAR (args)), SCM_CAR (args), SCM_ARG2,
1bbd0b84 795 FUNC_NAME);
f5bf2977 796 SCM_ASSERT (SCM_INUM0 == SCM_CAR (args), SCM_CAR (args), SCM_OUTOFRANGE,
1bbd0b84 797 FUNC_NAME);
0f2d19dd
JB
798 return ra;
799 case scm_tc7_smob:
800 SCM_ASRTGO (SCM_ARRAYP (ra), badarg);
801 vargs = scm_vector (args);
f5bf2977 802 SCM_ASSERT (SCM_LENGTH (vargs) == SCM_ARRAY_NDIM (ra),
1bbd0b84 803 scm_makfrom0str (FUNC_NAME), SCM_WNA, NULL);
f5bf2977 804 ve = SCM_VELTS (vargs);
0f2d19dd
JB
805 ndim = 0;
806 for (k = 0; k < SCM_ARRAY_NDIM (ra); k++)
807 {
20a54673 808 SCM_ASSERT (SCM_INUMP (ve[k]), ve[k], (SCM_ARG2 + k),
1bbd0b84 809 FUNC_NAME);
0f2d19dd 810 i = SCM_INUM (ve[k]);
20a54673 811 SCM_ASSERT (i >= 0 && i < SCM_ARRAY_NDIM (ra), ve[k],
1bbd0b84 812 SCM_OUTOFRANGE, FUNC_NAME);
0f2d19dd
JB
813 if (ndim < i)
814 ndim = i;
815 }
816 ndim++;
817 res = scm_make_ra (ndim);
818 SCM_ARRAY_V (res) = SCM_ARRAY_V (ra);
819 SCM_ARRAY_BASE (res) = SCM_ARRAY_BASE (ra);
820 for (k = ndim; k--;)
821 {
822 SCM_ARRAY_DIMS (res)[k].lbnd = 0;
823 SCM_ARRAY_DIMS (res)[k].ubnd = -1;
824 }
825 for (k = SCM_ARRAY_NDIM (ra); k--;)
826 {
827 i = SCM_INUM (ve[k]);
828 s = &(SCM_ARRAY_DIMS (ra)[k]);
829 r = &(SCM_ARRAY_DIMS (res)[i]);
830 if (r->ubnd < r->lbnd)
831 {
832 r->lbnd = s->lbnd;
833 r->ubnd = s->ubnd;
834 r->inc = s->inc;
835 ndim--;
836 }
837 else
838 {
839 if (r->ubnd > s->ubnd)
840 r->ubnd = s->ubnd;
841 if (r->lbnd < s->lbnd)
842 {
843 SCM_ARRAY_BASE (res) += (s->lbnd - r->lbnd) * r->inc;
844 r->lbnd = s->lbnd;
845 }
846 r->inc += s->inc;
847 }
848 }
1bbd0b84 849 SCM_ASSERT (ndim <= 0, args, "bad argument list", FUNC_NAME);
0f2d19dd
JB
850 scm_ra_set_contp (res);
851 return res;
852 }
853}
1bbd0b84 854#undef FUNC_NAME
0f2d19dd
JB
855
856/* args are RA . AXES */
3b3b36dd 857SCM_DEFINE (scm_enclose_array, "enclose-array", 0, 0, 1,
1bbd0b84 858 (SCM axes),
b380b885
MD
859 "@var{dim0}, @var{dim1} @dots{} should be nonnegative integers less than\n"
860 "the rank of @var{array}. @var{enclose-array} returns an array\n"
861 "resembling an array of shared arrays. The dimensions of each shared\n"
862 "array are the same as the @var{dim}th dimensions of the original array,\n"
863 "the dimensions of the outer array are the same as those of the original\n"
864 "array that did not match a @var{dim}.\n\n"
865 "An enclosed array is not a general Scheme array. Its elements may not\n"
866 "be set using @code{array-set!}. Two references to the same element of\n"
867 "an enclosed array will be @code{equal?} but will not in general be\n"
868 "@code{eq?}. The value returned by @var{array-prototype} when given an\n"
869 "enclosed array is unspecified.\n\n"
870 "examples:\n"
871 "@example\n"
872 "(enclose-array '#3(((a b c) (d e f)) ((1 2 3) (4 5 6))) 1) @result{}\n"
873 " #<enclosed-array (#1(a d) #1(b e) #1(c f)) (#1(1 4) #1(2 5) #1(3 6))>\n\n"
874 "(enclose-array '#3(((a b c) (d e f)) ((1 2 3) (4 5 6))) 1 0) @result{}\n"
875 " #<enclosed-array #2((a 1) (d 4)) #2((b 2) (e 5)) #2((c 3) (f 6))>\n"
876 "@end example")
1bbd0b84 877#define FUNC_NAME s_scm_enclose_array
0f2d19dd
JB
878{
879 SCM axv, ra, res, ra_inr;
880 scm_array_dim vdim, *s = &vdim;
881 int ndim, j, k, ninr, noutr;
1bbd0b84 882 SCM_ASSERT (SCM_NIMP (axes), scm_makfrom0str (FUNC_NAME), SCM_WNA,
f5bf2977 883 NULL);
0f2d19dd
JB
884 ra = SCM_CAR (axes);
885 axes = SCM_CDR (axes);
886 if (SCM_NULLP (axes))
0f2d19dd
JB
887 axes = scm_cons ((SCM_ARRAYP (ra) ? SCM_MAKINUM (SCM_ARRAY_NDIM (ra) - 1) : SCM_INUM0), SCM_EOL);
888 ninr = scm_ilength (axes);
889 ra_inr = scm_make_ra (ninr);
890 SCM_ASRTGO (SCM_NIMP (ra), badarg1);
891 switch SCM_TYP7
892 (ra)
893 {
894 default:
1bbd0b84 895 badarg1:SCM_WTA (1,ra);
0f2d19dd
JB
896 case scm_tc7_string:
897 case scm_tc7_bvect:
898 case scm_tc7_byvect:
899 case scm_tc7_uvect:
900 case scm_tc7_ivect:
901 case scm_tc7_fvect:
902 case scm_tc7_dvect:
903 case scm_tc7_cvect:
904 case scm_tc7_vector:
95f5b0f5 905 case scm_tc7_wvect:
0f2d19dd 906 case scm_tc7_svect:
5c11cc9d 907#ifdef HAVE_LONG_LONGS
0f2d19dd
JB
908 case scm_tc7_llvect:
909#endif
910 s->lbnd = 0;
911 s->ubnd = SCM_LENGTH (ra) - 1;
912 s->inc = 1;
913 SCM_ARRAY_V (ra_inr) = ra;
914 SCM_ARRAY_BASE (ra_inr) = 0;
915 ndim = 1;
916 break;
917 case scm_tc7_smob:
918 SCM_ASRTGO (SCM_ARRAYP (ra), badarg1);
919 s = SCM_ARRAY_DIMS (ra);
920 SCM_ARRAY_V (ra_inr) = SCM_ARRAY_V (ra);
921 SCM_ARRAY_BASE (ra_inr) = SCM_ARRAY_BASE (ra);
922 ndim = SCM_ARRAY_NDIM (ra);
923 break;
924 }
925 noutr = ndim - ninr;
7866a09b 926 axv = scm_make_string (SCM_MAKINUM (ndim), SCM_MAKE_CHAR (0));
1bbd0b84 927 SCM_ASSERT (0 <= noutr && 0 <= ninr, scm_makfrom0str (FUNC_NAME),
f5bf2977 928 SCM_WNA, NULL);
0f2d19dd
JB
929 res = scm_make_ra (noutr);
930 SCM_ARRAY_BASE (res) = SCM_ARRAY_BASE (ra_inr);
931 SCM_ARRAY_V (res) = ra_inr;
932 for (k = 0; k < ninr; k++, axes = SCM_CDR (axes))
933 {
1bbd0b84 934 SCM_ASSERT (SCM_INUMP (SCM_CAR (axes)), SCM_CAR (axes), "bad axis", FUNC_NAME);
0f2d19dd
JB
935 j = SCM_INUM (SCM_CAR (axes));
936 SCM_ARRAY_DIMS (ra_inr)[k].lbnd = s[j].lbnd;
937 SCM_ARRAY_DIMS (ra_inr)[k].ubnd = s[j].ubnd;
938 SCM_ARRAY_DIMS (ra_inr)[k].inc = s[j].inc;
939 SCM_CHARS (axv)[j] = 1;
940 }
941 for (j = 0, k = 0; k < noutr; k++, j++)
942 {
943 while (SCM_CHARS (axv)[j])
944 j++;
945 SCM_ARRAY_DIMS (res)[k].lbnd = s[j].lbnd;
946 SCM_ARRAY_DIMS (res)[k].ubnd = s[j].ubnd;
947 SCM_ARRAY_DIMS (res)[k].inc = s[j].inc;
948 }
949 scm_ra_set_contp (ra_inr);
950 scm_ra_set_contp (res);
951 return res;
952}
1bbd0b84 953#undef FUNC_NAME
0f2d19dd
JB
954
955
956
3b3b36dd 957SCM_DEFINE (scm_array_in_bounds_p, "array-in-bounds?", 0, 0, 1,
1bbd0b84 958 (SCM args),
b380b885 959 "Returns @code{#t} if its arguments would be acceptable to array-ref.")
1bbd0b84 960#define FUNC_NAME s_scm_array_in_bounds_p
0f2d19dd
JB
961{
962 SCM v, ind = SCM_EOL;
963 long pos = 0;
964 register scm_sizet k;
965 register long j;
966 scm_array_dim *s;
1bbd0b84 967 SCM_ASSERT (SCM_NIMP (args), scm_makfrom0str (FUNC_NAME),
f5bf2977 968 SCM_WNA, NULL);
0f2d19dd
JB
969 v = SCM_CAR (args);
970 args = SCM_CDR (args);
971 SCM_ASRTGO (SCM_NIMP (v), badarg1);
972 if (SCM_NIMP (args))
973
974 {
975 ind = SCM_CAR (args);
976 args = SCM_CDR (args);
1bbd0b84 977 SCM_ASSERT (SCM_INUMP (ind), ind, SCM_ARG2, FUNC_NAME);
0f2d19dd
JB
978 pos = SCM_INUM (ind);
979 }
980tail:
981 switch SCM_TYP7
982 (v)
983 {
984 default:
1bbd0b84
GB
985 badarg1:SCM_WTA (1,v);
986 wna: scm_wrong_num_args (scm_makfrom0str (FUNC_NAME));
0f2d19dd
JB
987 case scm_tc7_smob:
988 k = SCM_ARRAY_NDIM (v);
989 s = SCM_ARRAY_DIMS (v);
990 pos = SCM_ARRAY_BASE (v);
991 if (!k)
992 {
993 SCM_ASRTGO (SCM_NULLP (ind), wna);
994 ind = SCM_INUM0;
995 }
996 else
997 while (!0)
998 {
999 j = SCM_INUM (ind);
1000 if (!(j >= (s->lbnd) && j <= (s->ubnd)))
1001 {
1002 SCM_ASRTGO (--k == scm_ilength (args), wna);
1003 return SCM_BOOL_F;
1004 }
1005 pos += (j - s->lbnd) * (s->inc);
1006 if (!(--k && SCM_NIMP (args)))
1007 break;
1008 ind = SCM_CAR (args);
1009 args = SCM_CDR (args);
1010 s++;
1bbd0b84 1011 SCM_ASSERT (SCM_INUMP (ind), ind, s_bad_ind, FUNC_NAME);
0f2d19dd
JB
1012 }
1013 SCM_ASRTGO (0 == k, wna);
1014 v = SCM_ARRAY_V (v);
1015 goto tail;
1016 case scm_tc7_bvect:
1017 case scm_tc7_string:
1018 case scm_tc7_byvect:
1019 case scm_tc7_uvect:
1020 case scm_tc7_ivect:
1021 case scm_tc7_fvect:
1022 case scm_tc7_dvect:
1023 case scm_tc7_cvect:
1024 case scm_tc7_svect:
5c11cc9d 1025#ifdef HAVE_LONG_LONGS
0f2d19dd
JB
1026 case scm_tc7_llvect:
1027#endif
1028 case scm_tc7_vector:
95f5b0f5 1029 case scm_tc7_wvect:
0f2d19dd 1030 SCM_ASRTGO (SCM_NULLP (args) && SCM_INUMP (ind), wna);
1bbd0b84 1031 return SCM_BOOL(pos >= 0 && pos < SCM_LENGTH (v));
0f2d19dd
JB
1032 }
1033}
1bbd0b84 1034#undef FUNC_NAME
0f2d19dd
JB
1035
1036
1bbd0b84 1037SCM_REGISTER_PROC(s_array_ref, "array-ref", 1, 0, 1, scm_uniform_vector_ref);
1cc91f1b 1038
1bbd0b84 1039
3b3b36dd 1040SCM_DEFINE (scm_uniform_vector_ref, "uniform-vector-ref", 2, 0, 0,
1bbd0b84 1041 (SCM v, SCM args),
b380b885 1042 "Returns the element at the @code{(index1, index2)} element in @var{array}.")
1bbd0b84 1043#define FUNC_NAME s_scm_uniform_vector_ref
0f2d19dd
JB
1044{
1045 long pos;
0f2d19dd 1046
35de7ebe 1047 if (SCM_IMP (v))
0f2d19dd
JB
1048 {
1049 SCM_ASRTGO (SCM_NULLP (args), badarg);
1050 return v;
1051 }
1052 else if (SCM_ARRAYP (v))
0f2d19dd 1053 {
1bbd0b84 1054 pos = scm_aind (v, args, FUNC_NAME);
0f2d19dd
JB
1055 v = SCM_ARRAY_V (v);
1056 }
1057 else
1058 {
1059 if (SCM_NIMP (args))
1060
1061 {
1bbd0b84 1062 SCM_ASSERT (SCM_CONSP (args) && SCM_INUMP (SCM_CAR (args)), args, SCM_ARG2, FUNC_NAME);
0f2d19dd
JB
1063 pos = SCM_INUM (SCM_CAR (args));
1064 SCM_ASRTGO (SCM_NULLP (SCM_CDR (args)), wna);
1065 }
1066 else
1067 {
3b3b36dd 1068 SCM_VALIDATE_INUM (2,args);
0f2d19dd
JB
1069 pos = SCM_INUM (args);
1070 }
1071 SCM_ASRTGO (pos >= 0 && pos < SCM_LENGTH (v), outrng);
1072 }
1073 switch SCM_TYP7
1074 (v)
1075 {
1076 default:
1077 if (SCM_NULLP (args))
1078 return v;
35de7ebe 1079 badarg:
1bbd0b84 1080 SCM_WTA (1,v);
35de7ebe 1081 abort ();
1bbd0b84
GB
1082 outrng:scm_out_of_range (FUNC_NAME, SCM_MAKINUM (pos));
1083 wna: scm_wrong_num_args (SCM_FUNC_NAME);
0f2d19dd
JB
1084 case scm_tc7_smob:
1085 { /* enclosed */
1086 int k = SCM_ARRAY_NDIM (v);
1087 SCM res = scm_make_ra (k);
1088 SCM_ARRAY_V (res) = SCM_ARRAY_V (v);
1089 SCM_ARRAY_BASE (res) = pos;
1090 while (k--)
1091 {
1092 SCM_ARRAY_DIMS (res)[k].lbnd = SCM_ARRAY_DIMS (v)[k].lbnd;
1093 SCM_ARRAY_DIMS (res)[k].ubnd = SCM_ARRAY_DIMS (v)[k].ubnd;
1094 SCM_ARRAY_DIMS (res)[k].inc = SCM_ARRAY_DIMS (v)[k].inc;
1095 }
1096 return res;
1097 }
1098 case scm_tc7_bvect:
1099 if (SCM_VELTS (v)[pos / SCM_LONG_BIT] & (1L << (pos % SCM_LONG_BIT)))
1100 return SCM_BOOL_T;
1101 else
1102 return SCM_BOOL_F;
1103 case scm_tc7_string:
7866a09b 1104 return SCM_MAKE_CHAR (SCM_UCHARS (v)[pos]);
0f2d19dd
JB
1105 case scm_tc7_byvect:
1106 return SCM_MAKINUM (((char *)SCM_CHARS (v))[pos]);
1107# ifdef SCM_INUMS_ONLY
1108 case scm_tc7_uvect:
1109 case scm_tc7_ivect:
1110 return SCM_MAKINUM (SCM_VELTS (v)[pos]);
1111# else
1112 case scm_tc7_uvect:
1113 return scm_ulong2num(SCM_VELTS(v)[pos]);
1114 case scm_tc7_ivect:
1115 return scm_long2num(SCM_VELTS(v)[pos]);
1116# endif
1117
1118 case scm_tc7_svect:
1119 return SCM_MAKINUM (((short *) SCM_CDR (v))[pos]);
5c11cc9d 1120#ifdef HAVE_LONG_LONGS
0f2d19dd
JB
1121 case scm_tc7_llvect:
1122 return scm_long_long2num (((long_long *) SCM_CDR (v))[pos]);
1123#endif
1124
1125#ifdef SCM_FLOATS
1126#ifdef SCM_SINGLES
1127 case scm_tc7_fvect:
1128 return scm_makflo (((float *) SCM_CDR (v))[pos]);
1129#endif
1130 case scm_tc7_dvect:
1131 return scm_makdbl (((double *) SCM_CDR (v))[pos], 0.0);
1132 case scm_tc7_cvect:
1133 return scm_makdbl (((double *) SCM_CDR (v))[2 * pos],
1134 ((double *) SCM_CDR (v))[2 * pos + 1]);
1135#endif
1136 case scm_tc7_vector:
95f5b0f5 1137 case scm_tc7_wvect:
0f2d19dd
JB
1138 return SCM_VELTS (v)[pos];
1139 }
1140}
1bbd0b84 1141#undef FUNC_NAME
0f2d19dd
JB
1142
1143/* Internal version of scm_uniform_vector_ref for uves that does no error checking and
1144 tries to recycle conses. (Make *sure* you want them recycled.) */
1cc91f1b 1145
0f2d19dd 1146SCM
6e8d25a6 1147scm_cvref (SCM v, scm_sizet pos, SCM last)
0f2d19dd 1148{
5c11cc9d 1149 switch SCM_TYP7 (v)
0f2d19dd
JB
1150 {
1151 default:
1152 scm_wta (v, (char *) SCM_ARG1, "PROGRAMMING ERROR: scm_cvref");
1153 case scm_tc7_bvect:
1154 if (SCM_VELTS (v)[pos / SCM_LONG_BIT] & (1L << (pos % SCM_LONG_BIT)))
1155 return SCM_BOOL_T;
1156 else
1157 return SCM_BOOL_F;
1158 case scm_tc7_string:
7866a09b 1159 return SCM_MAKE_CHAR (SCM_UCHARS (v)[pos]);
0f2d19dd
JB
1160 case scm_tc7_byvect:
1161 return SCM_MAKINUM (((char *)SCM_CHARS (v))[pos]);
1162# ifdef SCM_INUMS_ONLY
1163 case scm_tc7_uvect:
1164 case scm_tc7_ivect:
1165 return SCM_MAKINUM (SCM_VELTS (v)[pos]);
1166# else
1167 case scm_tc7_uvect:
1168 return scm_ulong2num(SCM_VELTS(v)[pos]);
1169 case scm_tc7_ivect:
1170 return scm_long2num(SCM_VELTS(v)[pos]);
1171# endif
1172 case scm_tc7_svect:
1173 return SCM_MAKINUM (((short *) SCM_CDR (v))[pos]);
5c11cc9d 1174#ifdef HAVE_LONG_LONGS
0f2d19dd
JB
1175 case scm_tc7_llvect:
1176 return scm_long_long2num (((long_long *) SCM_CDR (v))[pos]);
1177#endif
1178#ifdef SCM_FLOATS
1179#ifdef SCM_SINGLES
1180 case scm_tc7_fvect:
1181 if (SCM_NIMP (last) && (last != scm_flo0) && (scm_tc_flo == SCM_CAR (last)))
1182 {
1183 SCM_FLO (last) = ((float *) SCM_CDR (v))[pos];
1184 return last;
1185 }
1186 return scm_makflo (((float *) SCM_CDR (v))[pos]);
1187#endif
1188 case scm_tc7_dvect:
1189#ifdef SCM_SINGLES
1190 if (SCM_NIMP (last) && scm_tc_dblr == SCM_CAR (last))
1191#else
1192 if (SCM_NIMP (last) && (last != scm_flo0) && (scm_tc_dblr == SCM_CAR (last)))
1193#endif
1194 {
1195 SCM_REAL (last) = ((double *) SCM_CDR (v))[pos];
1196 return last;
1197 }
1198 return scm_makdbl (((double *) SCM_CDR (v))[pos], 0.0);
1199 case scm_tc7_cvect:
1200 if (SCM_NIMP (last) && scm_tc_dblc == SCM_CAR (last))
1201 {
1202 SCM_REAL (last) = ((double *) SCM_CDR (v))[2 * pos];
1203 SCM_IMAG (last) = ((double *) SCM_CDR (v))[2 * pos + 1];
1204 return last;
1205 }
1206 return scm_makdbl (((double *) SCM_CDR (v))[2 * pos],
1207 ((double *) SCM_CDR (v))[2 * pos + 1]);
1208#endif
1209 case scm_tc7_vector:
95f5b0f5 1210 case scm_tc7_wvect:
0f2d19dd
JB
1211 return SCM_VELTS (v)[pos];
1212 case scm_tc7_smob:
1213 { /* enclosed scm_array */
1214 int k = SCM_ARRAY_NDIM (v);
1215 SCM res = scm_make_ra (k);
1216 SCM_ARRAY_V (res) = SCM_ARRAY_V (v);
1217 SCM_ARRAY_BASE (res) = pos;
1218 while (k--)
1219 {
1220 SCM_ARRAY_DIMS (res)[k].ubnd = SCM_ARRAY_DIMS (v)[k].ubnd;
1221 SCM_ARRAY_DIMS (res)[k].lbnd = SCM_ARRAY_DIMS (v)[k].lbnd;
1222 SCM_ARRAY_DIMS (res)[k].inc = SCM_ARRAY_DIMS (v)[k].inc;
1223 }
1224 return res;
1225 }
1226 }
1227}
1228
1bbd0b84
GB
1229SCM_REGISTER_PROC(s_uniform_array_set1_x, "uniform-array-set1!", 3, 0, 0, scm_array_set_x);
1230
1cc91f1b 1231
0aa0871f
GH
1232/* Note that args may be a list or an immediate object, depending which
1233 PROC is used (and it's called from C too). */
3b3b36dd 1234SCM_DEFINE (scm_array_set_x, "array-set!", 2, 0, 1,
1bbd0b84 1235 (SCM v, SCM obj, SCM args),
b380b885
MD
1236 "Sets the element at the @code{(index1, index2)} element in @var{array} to\n"
1237 "@var{new-value}. The value returned by array-set! is unspecified.")
1bbd0b84 1238#define FUNC_NAME s_scm_array_set_x
0f2d19dd 1239{
f3667f52 1240 long pos = 0;
0f2d19dd
JB
1241 SCM_ASRTGO (SCM_NIMP (v), badarg1);
1242 if (SCM_ARRAYP (v))
0f2d19dd 1243 {
1bbd0b84 1244 pos = scm_aind (v, args, FUNC_NAME);
0f2d19dd
JB
1245 v = SCM_ARRAY_V (v);
1246 }
1247 else
1248 {
1249 if (SCM_NIMP (args))
0f2d19dd 1250 {
0aa0871f 1251 SCM_ASSERT (SCM_CONSP(args) && SCM_INUMP (SCM_CAR (args)), args,
1bbd0b84 1252 SCM_ARG3, FUNC_NAME);
0f2d19dd 1253 SCM_ASRTGO (SCM_NULLP (SCM_CDR (args)), wna);
0aa0871f 1254 pos = SCM_INUM (SCM_CAR (args));
0f2d19dd
JB
1255 }
1256 else
1257 {
3b3b36dd 1258 SCM_VALIDATE_INUM_COPY (3,args,pos);
0f2d19dd
JB
1259 }
1260 SCM_ASRTGO (pos >= 0 && pos < SCM_LENGTH (v), outrng);
1261 }
1262 switch (SCM_TYP7 (v))
1263 {
35de7ebe 1264 default: badarg1:
1bbd0b84 1265 SCM_WTA (1,v);
35de7ebe 1266 abort ();
1bbd0b84
GB
1267 outrng:scm_out_of_range (FUNC_NAME, SCM_MAKINUM (pos));
1268 wna: scm_wrong_num_args (SCM_FUNC_NAME);
0f2d19dd
JB
1269 case scm_tc7_smob: /* enclosed */
1270 goto badarg1;
1271 case scm_tc7_bvect:
1272 if (SCM_BOOL_F == obj)
1273 SCM_VELTS (v)[pos / SCM_LONG_BIT] &= ~(1L << (pos % SCM_LONG_BIT));
1274 else if (SCM_BOOL_T == obj)
1275 SCM_VELTS (v)[pos / SCM_LONG_BIT] |= (1L << (pos % SCM_LONG_BIT));
1276 else
1bbd0b84 1277 badobj:SCM_WTA (2,obj);
0f2d19dd
JB
1278 break;
1279 case scm_tc7_string:
7866a09b
GB
1280 SCM_ASRTGO (SCM_CHARP (obj), badobj);
1281 SCM_UCHARS (v)[pos] = SCM_CHAR (obj);
0f2d19dd
JB
1282 break;
1283 case scm_tc7_byvect:
7866a09b
GB
1284 if (SCM_CHARP (obj))
1285 obj = SCM_MAKINUM ((char) SCM_CHAR (obj));
0aa0871f 1286 SCM_ASRTGO (SCM_INUMP (obj), badobj);
0f2d19dd
JB
1287 ((char *)SCM_CHARS (v))[pos] = SCM_INUM (obj);
1288 break;
1289# ifdef SCM_INUMS_ONLY
1290 case scm_tc7_uvect:
1bbd0b84
GB
1291 SCM_ASRTGO (SCM_INUM (obj) >= 0, badobj);
1292 /* fall through */
0f2d19dd 1293 case scm_tc7_ivect:
1bbd0b84 1294 SCM_ASRTGO(SCM_INUMP(obj), badobj); SCM_VELTS(v)[pos] = SCM_INUM(obj); break;
0f2d19dd 1295# else
1bbd0b84
GB
1296 case scm_tc7_uvect:
1297 SCM_VELTS(v)[pos] = scm_num2ulong(obj, (char *)SCM_ARG2, FUNC_NAME); break;
1298 case scm_tc7_ivect:
1299 SCM_VELTS(v)[pos] = scm_num2long(obj, (char *)SCM_ARG2, FUNC_NAME); break;
0f2d19dd 1300# endif
0f2d19dd 1301 case scm_tc7_svect:
0aa0871f 1302 SCM_ASRTGO (SCM_INUMP (obj), badobj);
0f2d19dd
JB
1303 ((short *) SCM_CDR (v))[pos] = SCM_INUM (obj);
1304 break;
5c11cc9d 1305#ifdef HAVE_LONG_LONGS
0f2d19dd 1306 case scm_tc7_llvect:
1bbd0b84 1307 ((long_long *) SCM_CDR (v))[pos] = scm_num2long_long (obj, (char *)SCM_ARG2, FUNC_NAME);
0f2d19dd
JB
1308 break;
1309#endif
1310
1311
1312#ifdef SCM_FLOATS
1313#ifdef SCM_SINGLES
1314 case scm_tc7_fvect:
1bbd0b84 1315 ((float *) SCM_CDR (v))[pos] = (float)scm_num2dbl(obj, FUNC_NAME); break;
0f2d19dd
JB
1316 break;
1317#endif
1318 case scm_tc7_dvect:
1bbd0b84 1319 ((double *) SCM_CDR (v))[pos] = scm_num2dbl(obj, FUNC_NAME); break;
0f2d19dd
JB
1320 break;
1321 case scm_tc7_cvect:
0c95b57d 1322 SCM_ASRTGO (SCM_INEXP (obj), badobj);
0f2d19dd
JB
1323 ((double *) SCM_CDR (v))[2 * pos] = SCM_REALPART (obj);
1324 ((double *) SCM_CDR (v))[2 * pos + 1] = SCM_CPLXP (obj) ? SCM_IMAG (obj) : 0.0;
1325 break;
1326#endif
1327 case scm_tc7_vector:
95f5b0f5 1328 case scm_tc7_wvect:
0f2d19dd
JB
1329 SCM_VELTS (v)[pos] = obj;
1330 break;
1331 }
1332 return SCM_UNSPECIFIED;
1333}
1bbd0b84 1334#undef FUNC_NAME
0f2d19dd 1335
1d7bdb25
GH
1336/* attempts to unroll an array into a one-dimensional array.
1337 returns the unrolled array or #f if it can't be done. */
1bbd0b84 1338 /* if strict is not SCM_UNDEFINED, return #f if returned array
1d7bdb25 1339 wouldn't have contiguous elements. */
3b3b36dd 1340SCM_DEFINE (scm_array_contents, "array-contents", 1, 1, 0,
1bbd0b84 1341 (SCM ra, SCM strict),
b380b885
MD
1342 "@deffnx primitive array-contents array strict\n"
1343 "If @var{array} may be @dfn{unrolled} into a one dimensional shared array\n"
1344 "without changing their order (last subscript changing fastest), then\n"
1345 "@code{array-contents} returns that shared array, otherwise it returns\n"
1346 "@code{#f}. All arrays made by @var{make-array} and\n"
1347 "@var{make-uniform-array} may be unrolled, some arrays made by\n"
1348 "@var{make-shared-array} may not be.\n\n"
1349 "If the optional argument @var{strict} is provided, a shared array will\n"
1350 "be returned only if its elements are stored internally contiguous in\n"
1351 "memory.")
1bbd0b84 1352#define FUNC_NAME s_scm_array_contents
0f2d19dd
JB
1353{
1354 SCM sra;
1355 if (SCM_IMP (ra))
f3667f52 1356 return SCM_BOOL_F;
5c11cc9d 1357 switch SCM_TYP7 (ra)
0f2d19dd
JB
1358 {
1359 default:
1360 return SCM_BOOL_F;
1361 case scm_tc7_vector:
95f5b0f5 1362 case scm_tc7_wvect:
0f2d19dd
JB
1363 case scm_tc7_string:
1364 case scm_tc7_bvect:
1365 case scm_tc7_byvect:
1366 case scm_tc7_uvect:
1367 case scm_tc7_ivect:
1368 case scm_tc7_fvect:
1369 case scm_tc7_dvect:
1370 case scm_tc7_cvect:
1371 case scm_tc7_svect:
5c11cc9d 1372#ifdef HAVE_LONG_LONGS
0f2d19dd
JB
1373 case scm_tc7_llvect:
1374#endif
1375 return ra;
1376 case scm_tc7_smob:
1377 {
1378 scm_sizet k, ndim = SCM_ARRAY_NDIM (ra), len = 1;
1379 if (!SCM_ARRAYP (ra) || !SCM_ARRAY_CONTP (ra))
1380 return SCM_BOOL_F;
1381 for (k = 0; k < ndim; k++)
1382 len *= SCM_ARRAY_DIMS (ra)[k].ubnd - SCM_ARRAY_DIMS (ra)[k].lbnd + 1;
1383 if (!SCM_UNBNDP (strict))
1384 {
0f2d19dd
JB
1385 if (ndim && (1 != SCM_ARRAY_DIMS (ra)[ndim - 1].inc))
1386 return SCM_BOOL_F;
1387 if (scm_tc7_bvect == SCM_TYP7 (SCM_ARRAY_V (ra)))
1388 {
1389 if (len != SCM_LENGTH (SCM_ARRAY_V (ra)) ||
1390 SCM_ARRAY_BASE (ra) % SCM_LONG_BIT ||
1391 len % SCM_LONG_BIT)
1392 return SCM_BOOL_F;
1393 }
1394 }
1395 if ((len == SCM_LENGTH (SCM_ARRAY_V (ra))) && 0 == SCM_ARRAY_BASE (ra) && SCM_ARRAY_DIMS (ra)->inc)
1396 return SCM_ARRAY_V (ra);
1397 sra = scm_make_ra (1);
1398 SCM_ARRAY_DIMS (sra)->lbnd = 0;
1399 SCM_ARRAY_DIMS (sra)->ubnd = len - 1;
1400 SCM_ARRAY_V (sra) = SCM_ARRAY_V (ra);
1401 SCM_ARRAY_BASE (sra) = SCM_ARRAY_BASE (ra);
1402 SCM_ARRAY_DIMS (sra)->inc = (ndim ? SCM_ARRAY_DIMS (ra)[ndim - 1].inc : 1);
1403 return sra;
1404 }
1405 }
1406}
1bbd0b84 1407#undef FUNC_NAME
0f2d19dd 1408
1cc91f1b 1409
0f2d19dd 1410SCM
6e8d25a6 1411scm_ra2contig (SCM ra, int copy)
0f2d19dd
JB
1412{
1413 SCM ret;
1414 long inc = 1;
1415 scm_sizet k, len = 1;
1416 for (k = SCM_ARRAY_NDIM (ra); k--;)
1417 len *= SCM_ARRAY_DIMS (ra)[k].ubnd - SCM_ARRAY_DIMS (ra)[k].lbnd + 1;
1418 k = SCM_ARRAY_NDIM (ra);
1419 if (SCM_ARRAY_CONTP (ra) && ((0 == k) || (1 == SCM_ARRAY_DIMS (ra)[k - 1].inc)))
1420 {
1421 if (scm_tc7_bvect != SCM_TYP7 (ra))
1422 return ra;
1423 if ((len == SCM_LENGTH (SCM_ARRAY_V (ra)) &&
1424 0 == SCM_ARRAY_BASE (ra) % SCM_LONG_BIT &&
1425 0 == len % SCM_LONG_BIT))
1426 return ra;
1427 }
1428 ret = scm_make_ra (k);
1429 SCM_ARRAY_BASE (ret) = 0;
1430 while (k--)
1431 {
1432 SCM_ARRAY_DIMS (ret)[k].lbnd = SCM_ARRAY_DIMS (ra)[k].lbnd;
1433 SCM_ARRAY_DIMS (ret)[k].ubnd = SCM_ARRAY_DIMS (ra)[k].ubnd;
1434 SCM_ARRAY_DIMS (ret)[k].inc = inc;
1435 inc *= SCM_ARRAY_DIMS (ra)[k].ubnd - SCM_ARRAY_DIMS (ra)[k].lbnd + 1;
1436 }
1437 SCM_ARRAY_V (ret) = scm_make_uve ((inc - 1), scm_array_prototype (ra));
1438 if (copy)
1439 scm_array_copy_x (ra, ret);
1440 return ret;
1441}
1442
1443
1444
3b3b36dd 1445SCM_DEFINE (scm_uniform_array_read_x, "uniform-array-read!", 1, 3, 0,
1bbd0b84 1446 (SCM ra, SCM port_or_fd, SCM start, SCM end),
b380b885
MD
1447 "@deffnx primitive uniform-vector-read! uve [port-or-fdes] [start] [end]\n"
1448 "Attempts to read all elements of @var{ura}, in lexicographic order, as\n"
1449 "binary objects from @var{port-or-fdes}.\n"
1450 "If an end of file is encountered during\n"
1451 "uniform-array-read! the objects up to that point only are put into @var{ura}\n"
1452 "(starting at the beginning) and the remainder of the array is\n"
1453 "unchanged.\n\n"
1454 "The optional arguments @var{start} and @var{end} allow\n"
1455 "a specified region of a vector (or linearized array) to be read,\n"
1456 "leaving the remainder of the vector unchanged.\n\n"
1457 "@code{uniform-array-read!} returns the number of objects read.\n"
1458 "@var{port-or-fdes} may be omitted, in which case it defaults to the value\n"
1459 "returned by @code{(current-input-port)}.")
1bbd0b84 1460#define FUNC_NAME s_scm_uniform_array_read_x
0f2d19dd 1461{
35de7ebe 1462 SCM cra = SCM_UNDEFINED, v = ra;
3d8d56df 1463 long sz, vlen, ans;
1146b6cd
GH
1464 long cstart = 0;
1465 long cend;
1466 long offset = 0;
35de7ebe 1467
0f2d19dd 1468 SCM_ASRTGO (SCM_NIMP (v), badarg1);
3d8d56df
GH
1469 if (SCM_UNBNDP (port_or_fd))
1470 port_or_fd = scm_cur_inp;
1471 else
1472 SCM_ASSERT (SCM_INUMP (port_or_fd)
0c95b57d 1473 || (SCM_OPINPORTP (port_or_fd)),
1bbd0b84 1474 port_or_fd, SCM_ARG2, FUNC_NAME);
3d8d56df 1475 vlen = SCM_LENGTH (v);
35de7ebe 1476
0f2d19dd 1477loop:
35de7ebe 1478 switch SCM_TYP7 (v)
0f2d19dd
JB
1479 {
1480 default:
5d2d2ffc 1481 badarg1:SCM_WTA (SCM_ARG1,v);
0f2d19dd
JB
1482 case scm_tc7_smob:
1483 SCM_ASRTGO (SCM_ARRAYP (v), badarg1);
1484 cra = scm_ra2contig (ra, 0);
1146b6cd 1485 cstart += SCM_ARRAY_BASE (cra);
3d8d56df 1486 vlen = SCM_ARRAY_DIMS (cra)->inc *
0f2d19dd
JB
1487 (SCM_ARRAY_DIMS (cra)->ubnd - SCM_ARRAY_DIMS (cra)->lbnd + 1);
1488 v = SCM_ARRAY_V (cra);
1489 goto loop;
1490 case scm_tc7_string:
1491 case scm_tc7_byvect:
1492 sz = sizeof (char);
1493 break;
1494 case scm_tc7_bvect:
3d8d56df 1495 vlen = (vlen + SCM_LONG_BIT - 1) / SCM_LONG_BIT;
1146b6cd 1496 cstart /= SCM_LONG_BIT;
0f2d19dd
JB
1497 case scm_tc7_uvect:
1498 case scm_tc7_ivect:
1499 sz = sizeof (long);
1500 break;
1501 case scm_tc7_svect:
1502 sz = sizeof (short);
1503 break;
5c11cc9d 1504#ifdef HAVE_LONG_LONGS
0f2d19dd
JB
1505 case scm_tc7_llvect:
1506 sz = sizeof (long_long);
1507 break;
1508#endif
1509#ifdef SCM_FLOATS
1510#ifdef SCM_SINGLES
1511 case scm_tc7_fvect:
1512 sz = sizeof (float);
1513 break;
1514#endif
1515 case scm_tc7_dvect:
1516 sz = sizeof (double);
1517 break;
1518 case scm_tc7_cvect:
1519 sz = 2 * sizeof (double);
1520 break;
1521#endif
1522 }
3d8d56df 1523
1146b6cd
GH
1524 cend = vlen;
1525 if (!SCM_UNBNDP (start))
3d8d56df 1526 {
1146b6cd 1527 offset =
4638e087 1528 SCM_NUM2LONG (3, start);
35de7ebe 1529
1146b6cd 1530 if (offset < 0 || offset >= cend)
1bbd0b84 1531 scm_out_of_range (FUNC_NAME, start);
1146b6cd
GH
1532
1533 if (!SCM_UNBNDP (end))
1534 {
1535 long tend =
4638e087 1536 SCM_NUM2LONG (4, end);
3d8d56df 1537
1146b6cd 1538 if (tend <= offset || tend > cend)
1bbd0b84 1539 scm_out_of_range (FUNC_NAME, end);
1146b6cd
GH
1540 cend = tend;
1541 }
0f2d19dd 1542 }
35de7ebe 1543
3d8d56df
GH
1544 if (SCM_NIMP (port_or_fd))
1545 {
6c951427
GH
1546 scm_port *pt = SCM_PTAB_ENTRY (port_or_fd);
1547 int remaining = (cend - offset) * sz;
1548 char *dest = SCM_CHARS (v) + (cstart + offset) * sz;
1549
1550 if (pt->rw_active == SCM_PORT_WRITE)
affc96b5 1551 scm_flush (port_or_fd);
6c951427
GH
1552
1553 ans = cend - offset;
1554 while (remaining > 0)
3d8d56df 1555 {
6c951427
GH
1556 if (pt->read_pos < pt->read_end)
1557 {
1558 int to_copy = min (pt->read_end - pt->read_pos,
1559 remaining);
1560
1561 memcpy (dest, pt->read_pos, to_copy);
1562 pt->read_pos += to_copy;
1563 remaining -= to_copy;
1564 dest += to_copy;
1565 }
1566 else
1567 {
affc96b5 1568 if (scm_fill_input (port_or_fd) == EOF)
6c951427
GH
1569 {
1570 if (remaining % sz != 0)
1571 {
5d2d2ffc 1572 SCM_MISC_ERROR ("unexpected EOF", SCM_EOL);
6c951427
GH
1573 }
1574 ans -= remaining / sz;
1575 break;
1576 }
6c951427 1577 }
3d8d56df 1578 }
6c951427
GH
1579
1580 if (pt->rw_random)
1581 pt->rw_active = SCM_PORT_READ;
3d8d56df
GH
1582 }
1583 else /* file descriptor. */
1584 {
1585 SCM_SYSCALL (ans = read (SCM_INUM (port_or_fd),
1146b6cd
GH
1586 SCM_CHARS (v) + (cstart + offset) * sz,
1587 (scm_sizet) (sz * (cend - offset))));
3d8d56df 1588 if (ans == -1)
1bbd0b84 1589 SCM_SYSERROR;
3d8d56df 1590 }
0f2d19dd
JB
1591 if (SCM_TYP7 (v) == scm_tc7_bvect)
1592 ans *= SCM_LONG_BIT;
35de7ebe 1593
0f2d19dd
JB
1594 if (v != ra && cra != ra)
1595 scm_array_copy_x (cra, ra);
35de7ebe 1596
0f2d19dd
JB
1597 return SCM_MAKINUM (ans);
1598}
1bbd0b84 1599#undef FUNC_NAME
0f2d19dd 1600
3b3b36dd 1601SCM_DEFINE (scm_uniform_array_write, "uniform-array-write", 1, 3, 0,
1bbd0b84 1602 (SCM v, SCM port_or_fd, SCM start, SCM end),
b380b885
MD
1603 "@deffnx primitive uniform-vector-write uve [port-or-fdes] [start] [end]\n"
1604 "Writes all elements of @var{ura} as binary objects to\n"
1605 "@var{port-or-fdes}.\n\n"
1606 "The optional arguments @var{start}\n"
1607 "and @var{end} allow\n"
1608 "a specified region of a vector (or linearized array) to be written.\n\n"
1609 "The number of objects actually written is returned. \n"
1610 "@var{port-or-fdes} may be\n"
1611 "omitted, in which case it defaults to the value returned by\n"
1612 "@code{(current-output-port)}.")
1bbd0b84 1613#define FUNC_NAME s_scm_uniform_array_write
0f2d19dd 1614{
3d8d56df 1615 long sz, vlen, ans;
1146b6cd
GH
1616 long offset = 0;
1617 long cstart = 0;
1618 long cend;
3d8d56df 1619
78446828
MV
1620 port_or_fd = SCM_COERCE_OUTPORT (port_or_fd);
1621
0f2d19dd 1622 SCM_ASRTGO (SCM_NIMP (v), badarg1);
3d8d56df
GH
1623 if (SCM_UNBNDP (port_or_fd))
1624 port_or_fd = scm_cur_outp;
1625 else
1626 SCM_ASSERT (SCM_INUMP (port_or_fd)
0c95b57d 1627 || (SCM_OPOUTPORTP (port_or_fd)),
1bbd0b84 1628 port_or_fd, SCM_ARG2, FUNC_NAME);
3d8d56df
GH
1629 vlen = SCM_LENGTH (v);
1630
0f2d19dd 1631loop:
3d8d56df 1632 switch SCM_TYP7 (v)
0f2d19dd
JB
1633 {
1634 default:
4638e087 1635 badarg1:SCM_WTA (1, v);
0f2d19dd
JB
1636 case scm_tc7_smob:
1637 SCM_ASRTGO (SCM_ARRAYP (v), badarg1);
1638 v = scm_ra2contig (v, 1);
1146b6cd 1639 cstart = SCM_ARRAY_BASE (v);
3d8d56df
GH
1640 vlen = SCM_ARRAY_DIMS (v)->inc
1641 * (SCM_ARRAY_DIMS (v)->ubnd - SCM_ARRAY_DIMS (v)->lbnd + 1);
0f2d19dd
JB
1642 v = SCM_ARRAY_V (v);
1643 goto loop;
0f2d19dd 1644 case scm_tc7_string:
3d8d56df 1645 case scm_tc7_byvect:
0f2d19dd
JB
1646 sz = sizeof (char);
1647 break;
1648 case scm_tc7_bvect:
3d8d56df 1649 vlen = (vlen + SCM_LONG_BIT - 1) / SCM_LONG_BIT;
1146b6cd 1650 cstart /= SCM_LONG_BIT;
0f2d19dd
JB
1651 case scm_tc7_uvect:
1652 case scm_tc7_ivect:
1653 sz = sizeof (long);
1654 break;
1655 case scm_tc7_svect:
1656 sz = sizeof (short);
1657 break;
5c11cc9d 1658#ifdef HAVE_LONG_LONGS
0f2d19dd
JB
1659 case scm_tc7_llvect:
1660 sz = sizeof (long_long);
1661 break;
1662#endif
1663#ifdef SCM_FLOATS
1664#ifdef SCM_SINGLES
1665 case scm_tc7_fvect:
1666 sz = sizeof (float);
1667 break;
1668#endif
1669 case scm_tc7_dvect:
1670 sz = sizeof (double);
1671 break;
1672 case scm_tc7_cvect:
1673 sz = 2 * sizeof (double);
1674 break;
1675#endif
1676 }
3d8d56df 1677
1146b6cd
GH
1678 cend = vlen;
1679 if (!SCM_UNBNDP (start))
3d8d56df 1680 {
1146b6cd 1681 offset =
4638e087 1682 SCM_NUM2LONG (3, start);
3d8d56df 1683
1146b6cd 1684 if (offset < 0 || offset >= cend)
1bbd0b84 1685 scm_out_of_range (FUNC_NAME, start);
1146b6cd
GH
1686
1687 if (!SCM_UNBNDP (end))
1688 {
1689 long tend =
4638e087 1690 SCM_NUM2LONG (4, end);
3d8d56df 1691
1146b6cd 1692 if (tend <= offset || tend > cend)
1bbd0b84 1693 scm_out_of_range (FUNC_NAME, end);
1146b6cd
GH
1694 cend = tend;
1695 }
3d8d56df
GH
1696 }
1697
1698 if (SCM_NIMP (port_or_fd))
1699 {
6c951427 1700 char *source = SCM_CHARS (v) + (cstart + offset) * sz;
6c951427
GH
1701
1702 ans = cend - offset;
265e6a4d 1703 scm_lfwrite (source, ans * sz, port_or_fd);
3d8d56df
GH
1704 }
1705 else /* file descriptor. */
1706 {
1707 SCM_SYSCALL (ans = write (SCM_INUM (port_or_fd),
1146b6cd
GH
1708 SCM_CHARS (v) + (cstart + offset) * sz,
1709 (scm_sizet) (sz * (cend - offset))));
3d8d56df 1710 if (ans == -1)
1bbd0b84 1711 SCM_SYSERROR;
3d8d56df 1712 }
0f2d19dd
JB
1713 if (SCM_TYP7 (v) == scm_tc7_bvect)
1714 ans *= SCM_LONG_BIT;
3d8d56df 1715
0f2d19dd
JB
1716 return SCM_MAKINUM (ans);
1717}
1bbd0b84 1718#undef FUNC_NAME
0f2d19dd
JB
1719
1720
1721static char cnt_tab[16] =
1722{0, 1, 1, 2, 1, 2, 2, 3, 1, 2, 2, 3, 2, 3, 3, 4};
1723
3b3b36dd 1724SCM_DEFINE (scm_bit_count, "bit-count", 2, 0, 0,
1bbd0b84 1725 (SCM item, SCM seq),
b380b885 1726 "Returns the number occurrences of @var{bool} in @var{bv}.")
1bbd0b84 1727#define FUNC_NAME s_scm_bit_count
0f2d19dd
JB
1728{
1729 long i;
1730 register unsigned long cnt = 0, w;
3b3b36dd 1731 SCM_VALIDATE_INUM (2,seq);
5c11cc9d 1732 switch SCM_TYP7 (seq)
0f2d19dd
JB
1733 {
1734 default:
1bbd0b84 1735 SCM_WTA (2,seq);
0f2d19dd
JB
1736 case scm_tc7_bvect:
1737 if (0 == SCM_LENGTH (seq))
1738 return SCM_INUM0;
1739 i = (SCM_LENGTH (seq) - 1) / SCM_LONG_BIT;
1740 w = SCM_VELTS (seq)[i];
1741 if (SCM_FALSEP (item))
1742 w = ~w;
1743 w <<= SCM_LONG_BIT - 1 - ((SCM_LENGTH (seq) - 1) % SCM_LONG_BIT);
1744 while (!0)
1745 {
1746 for (; w; w >>= 4)
1747 cnt += cnt_tab[w & 0x0f];
1748 if (0 == i--)
1749 return SCM_MAKINUM (cnt);
1750 w = SCM_VELTS (seq)[i];
1751 if (SCM_FALSEP (item))
1752 w = ~w;
1753 }
1754 }
1755}
1bbd0b84 1756#undef FUNC_NAME
0f2d19dd
JB
1757
1758
3b3b36dd 1759SCM_DEFINE (scm_bit_position, "bit-position", 3, 0, 0,
1bbd0b84 1760 (SCM item, SCM v, SCM k),
b380b885
MD
1761 "Returns the minimum index of an occurrence of @var{bool} in @var{bv}\n"
1762 "which is at least @var{k}. If no @var{bool} occurs within the specified\n"
1763 "range @code{#f} is returned.")
1bbd0b84 1764#define FUNC_NAME s_scm_bit_position
0f2d19dd 1765{
1bbd0b84 1766 long i, lenw, xbits, pos;
0f2d19dd 1767 register unsigned long w;
6b5a304f 1768 SCM_VALIDATE_NIM (2,v);
3b3b36dd 1769 SCM_VALIDATE_INUM_COPY (3,k,pos);
0f2d19dd 1770 SCM_ASSERT ((pos <= SCM_LENGTH (v)) && (pos >= 0),
1bbd0b84 1771 k, SCM_OUTOFRANGE, FUNC_NAME);
0f2d19dd
JB
1772 if (pos == SCM_LENGTH (v))
1773 return SCM_BOOL_F;
5c11cc9d 1774 switch SCM_TYP7 (v)
0f2d19dd
JB
1775 {
1776 default:
1bbd0b84 1777 SCM_WTA (2,v);
0f2d19dd
JB
1778 case scm_tc7_bvect:
1779 if (0 == SCM_LENGTH (v))
1780 return SCM_MAKINUM (-1L);
1781 lenw = (SCM_LENGTH (v) - 1) / SCM_LONG_BIT; /* watch for part words */
1782 i = pos / SCM_LONG_BIT;
1783 w = SCM_VELTS (v)[i];
1784 if (SCM_FALSEP (item))
1785 w = ~w;
1786 xbits = (pos % SCM_LONG_BIT);
1787 pos -= xbits;
1788 w = ((w >> xbits) << xbits);
1789 xbits = SCM_LONG_BIT - 1 - (SCM_LENGTH (v) - 1) % SCM_LONG_BIT;
1790 while (!0)
1791 {
1792 if (w && (i == lenw))
1793 w = ((w << xbits) >> xbits);
1794 if (w)
1795 while (w)
1796 switch (w & 0x0f)
1797 {
1798 default:
1799 return SCM_MAKINUM (pos);
1800 case 2:
1801 case 6:
1802 case 10:
1803 case 14:
1804 return SCM_MAKINUM (pos + 1);
1805 case 4:
1806 case 12:
1807 return SCM_MAKINUM (pos + 2);
1808 case 8:
1809 return SCM_MAKINUM (pos + 3);
1810 case 0:
1811 pos += 4;
1812 w >>= 4;
1813 }
1814 if (++i > lenw)
1815 break;
1816 pos += SCM_LONG_BIT;
1817 w = SCM_VELTS (v)[i];
1818 if (SCM_FALSEP (item))
1819 w = ~w;
1820 }
1821 return SCM_BOOL_F;
1822 }
1823}
1bbd0b84 1824#undef FUNC_NAME
0f2d19dd
JB
1825
1826
3b3b36dd 1827SCM_DEFINE (scm_bit_set_star_x, "bit-set*!", 3, 0, 0,
1bbd0b84 1828 (SCM v, SCM kv, SCM obj),
b380b885
MD
1829 "If uve is a bit-vector @var{bv} and uve must be of the same length. If\n"
1830 "@var{bool} is @code{#t}, uve is OR'ed into @var{bv}; If @var{bool} is @code{#f}, the\n"
1831 "inversion of uve is AND'ed into @var{bv}.\n\n"
1832 "If uve is a unsigned integer vector all the elements of uve must be\n"
1833 "between 0 and the @code{LENGTH} of @var{bv}. The bits of @var{bv}\n"
1834 "corresponding to the indexes in uve are set to @var{bool}.\n\n"
1835 "The return value is unspecified.")
1bbd0b84 1836#define FUNC_NAME s_scm_bit_set_star_x
0f2d19dd
JB
1837{
1838 register long i, k, vlen;
1839 SCM_ASRTGO (SCM_NIMP (v), badarg1);
1840 SCM_ASRTGO (SCM_NIMP (kv), badarg2);
5c11cc9d 1841 switch SCM_TYP7 (kv)
0f2d19dd
JB
1842 {
1843 default:
1bbd0b84 1844 badarg2:SCM_WTA (2,kv);
0f2d19dd 1845 case scm_tc7_uvect:
5c11cc9d 1846 switch SCM_TYP7 (v)
0f2d19dd
JB
1847 {
1848 default:
1bbd0b84 1849 badarg1:SCM_WTA (1,v);
0f2d19dd
JB
1850 case scm_tc7_bvect:
1851 vlen = SCM_LENGTH (v);
1852 if (SCM_BOOL_F == obj)
1853 for (i = SCM_LENGTH (kv); i;)
1854 {
1855 k = SCM_VELTS (kv)[--i];
1bbd0b84 1856 SCM_ASSERT ((k < vlen), SCM_MAKINUM (k), SCM_OUTOFRANGE, FUNC_NAME);
0f2d19dd
JB
1857 SCM_VELTS (v)[k / SCM_LONG_BIT] &= ~(1L << (k % SCM_LONG_BIT));
1858 }
1859 else if (SCM_BOOL_T == obj)
1860 for (i = SCM_LENGTH (kv); i;)
1861 {
1862 k = SCM_VELTS (kv)[--i];
1bbd0b84 1863 SCM_ASSERT ((k < vlen), SCM_MAKINUM (k), SCM_OUTOFRANGE, FUNC_NAME);
0f2d19dd
JB
1864 SCM_VELTS (v)[k / SCM_LONG_BIT] |= (1L << (k % SCM_LONG_BIT));
1865 }
1866 else
1bbd0b84 1867 badarg3:SCM_WTA (3,obj);
0f2d19dd
JB
1868 }
1869 break;
1870 case scm_tc7_bvect:
1871 SCM_ASRTGO (SCM_TYP7 (v) == scm_tc7_bvect && SCM_LENGTH (v) == SCM_LENGTH (kv), badarg1);
1872 if (SCM_BOOL_F == obj)
1873 for (k = (SCM_LENGTH (v) + SCM_LONG_BIT - 1) / SCM_LONG_BIT; k--;)
1874 SCM_VELTS (v)[k] &= ~(SCM_VELTS (kv)[k]);
1875 else if (SCM_BOOL_T == obj)
1876 for (k = (SCM_LENGTH (v) + SCM_LONG_BIT - 1) / SCM_LONG_BIT; k--;)
1877 SCM_VELTS (v)[k] |= SCM_VELTS (kv)[k];
1878 else
1879 goto badarg3;
1880 break;
1881 }
1882 return SCM_UNSPECIFIED;
1883}
1bbd0b84 1884#undef FUNC_NAME
0f2d19dd
JB
1885
1886
3b3b36dd 1887SCM_DEFINE (scm_bit_count_star, "bit-count*", 3, 0, 0,
1bbd0b84 1888 (SCM v, SCM kv, SCM obj),
b380b885
MD
1889 "Returns\n"
1890 "@example\n"
1891 "(bit-count (bit-set*! (if bool bv (bit-invert! bv)) uve #t) #t).\n"
1892 "@end example\n"
1893 "@var{bv} is not modified.")
1bbd0b84 1894#define FUNC_NAME s_scm_bit_count_star
0f2d19dd
JB
1895{
1896 register long i, vlen, count = 0;
1897 register unsigned long k;
1898 SCM_ASRTGO (SCM_NIMP (v), badarg1);
1899 SCM_ASRTGO (SCM_NIMP (kv), badarg2);
5c11cc9d 1900 switch SCM_TYP7 (kv)
0f2d19dd
JB
1901 {
1902 default:
1bbd0b84 1903 badarg2:SCM_WTA (2,kv);
0f2d19dd
JB
1904 case scm_tc7_uvect:
1905 switch SCM_TYP7
1906 (v)
1907 {
1908 default:
1bbd0b84 1909 badarg1:SCM_WTA (1,v);
0f2d19dd
JB
1910 case scm_tc7_bvect:
1911 vlen = SCM_LENGTH (v);
1912 if (SCM_BOOL_F == obj)
1913 for (i = SCM_LENGTH (kv); i;)
1914 {
1915 k = SCM_VELTS (kv)[--i];
1bbd0b84 1916 SCM_ASSERT ((k < vlen), SCM_MAKINUM (k), SCM_OUTOFRANGE, FUNC_NAME);
0f2d19dd
JB
1917 if (!(SCM_VELTS (v)[k / SCM_LONG_BIT] & (1L << (k % SCM_LONG_BIT))))
1918 count++;
1919 }
1920 else if (SCM_BOOL_T == obj)
1921 for (i = SCM_LENGTH (kv); i;)
1922 {
1923 k = SCM_VELTS (kv)[--i];
1bbd0b84 1924 SCM_ASSERT ((k < vlen), SCM_MAKINUM (k), SCM_OUTOFRANGE, FUNC_NAME);
0f2d19dd
JB
1925 if (SCM_VELTS (v)[k / SCM_LONG_BIT] & (1L << (k % SCM_LONG_BIT)))
1926 count++;
1927 }
1928 else
1bbd0b84 1929 badarg3:SCM_WTA (3,obj);
0f2d19dd
JB
1930 }
1931 break;
1932 case scm_tc7_bvect:
1933 SCM_ASRTGO (SCM_TYP7 (v) == scm_tc7_bvect && SCM_LENGTH (v) == SCM_LENGTH (kv), badarg1);
1934 if (0 == SCM_LENGTH (v))
1935 return SCM_INUM0;
1936 SCM_ASRTGO (SCM_BOOL_T == obj || SCM_BOOL_F == obj, badarg3);
1937 obj = (SCM_BOOL_T == obj);
1938 i = (SCM_LENGTH (v) - 1) / SCM_LONG_BIT;
1939 k = SCM_VELTS (kv)[i] & (obj ? SCM_VELTS (v)[i] : ~SCM_VELTS (v)[i]);
1940 k <<= SCM_LONG_BIT - 1 - ((SCM_LENGTH (v) - 1) % SCM_LONG_BIT);
1941 while (!0)
1942 {
1943 for (; k; k >>= 4)
1944 count += cnt_tab[k & 0x0f];
1945 if (0 == i--)
1946 return SCM_MAKINUM (count);
1947 k = SCM_VELTS (kv)[i] & (obj ? SCM_VELTS (v)[i] : ~SCM_VELTS (v)[i]);
1948 }
1949 }
1950 return SCM_MAKINUM (count);
1951}
1bbd0b84 1952#undef FUNC_NAME
0f2d19dd
JB
1953
1954
3b3b36dd 1955SCM_DEFINE (scm_bit_invert_x, "bit-invert!", 1, 0, 0,
1bbd0b84 1956 (SCM v),
b380b885 1957 "Modifies @var{bv} by replacing each element with its negation.")
1bbd0b84 1958#define FUNC_NAME s_scm_bit_invert_x
0f2d19dd
JB
1959{
1960 register long k;
1961 SCM_ASRTGO (SCM_NIMP (v), badarg1);
1962 k = SCM_LENGTH (v);
1963 switch SCM_TYP7
1964 (v)
1965 {
1966 case scm_tc7_bvect:
1967 for (k = (k + SCM_LONG_BIT - 1) / SCM_LONG_BIT; k--;)
1968 SCM_VELTS (v)[k] = ~SCM_VELTS (v)[k];
1969 break;
1970 default:
1bbd0b84 1971 badarg1:SCM_WTA (1,v);
0f2d19dd
JB
1972 }
1973 return SCM_UNSPECIFIED;
1974}
1bbd0b84 1975#undef FUNC_NAME
0f2d19dd
JB
1976
1977
0f2d19dd 1978SCM
1bbd0b84 1979scm_istr2bve (char *str, long len)
0f2d19dd
JB
1980{
1981 SCM v = scm_make_uve (len, SCM_BOOL_T);
1982 long *data = (long *) SCM_VELTS (v);
1983 register unsigned long mask;
1984 register long k;
1985 register long j;
1986 for (k = 0; k < (len + SCM_LONG_BIT - 1) / SCM_LONG_BIT; k++)
1987 {
1988 data[k] = 0L;
1989 j = len - k * SCM_LONG_BIT;
1990 if (j > SCM_LONG_BIT)
1991 j = SCM_LONG_BIT;
1992 for (mask = 1L; j--; mask <<= 1)
1993 switch (*str++)
1994 {
1995 case '0':
1996 break;
1997 case '1':
1998 data[k] |= mask;
1999 break;
2000 default:
2001 return SCM_BOOL_F;
2002 }
2003 }
2004 return v;
2005}
2006
2007
1cc91f1b 2008
0f2d19dd 2009static SCM
1bbd0b84 2010ra2l (SCM ra,scm_sizet base,scm_sizet k)
0f2d19dd
JB
2011{
2012 register SCM res = SCM_EOL;
2013 register long inc = SCM_ARRAY_DIMS (ra)[k].inc;
2014 register scm_sizet i;
2015 if (SCM_ARRAY_DIMS (ra)[k].ubnd < SCM_ARRAY_DIMS (ra)[k].lbnd)
2016 return SCM_EOL;
2017 i = base + (1 + SCM_ARRAY_DIMS (ra)[k].ubnd - SCM_ARRAY_DIMS (ra)[k].lbnd) * inc;
2018 if (k < SCM_ARRAY_NDIM (ra) - 1)
2019 {
2020 do
2021 {
2022 i -= inc;
2023 res = scm_cons (ra2l (ra, i, k + 1), res);
2024 }
2025 while (i != base);
2026 }
2027 else
2028 do
2029 {
2030 i -= inc;
2031 res = scm_cons (scm_uniform_vector_ref (SCM_ARRAY_V (ra), SCM_MAKINUM (i)), res);
2032 }
2033 while (i != base);
2034 return res;
2035}
2036
2037
3b3b36dd 2038SCM_DEFINE (scm_array_to_list, "array->list", 1, 0, 0,
1bbd0b84 2039 (SCM v),
b380b885 2040 "Returns a list consisting of all the elements, in order, of @var{array}.")
1bbd0b84 2041#define FUNC_NAME s_scm_array_to_list
0f2d19dd
JB
2042{
2043 SCM res = SCM_EOL;
2044 register long k;
2045 SCM_ASRTGO (SCM_NIMP (v), badarg1);
2046 switch SCM_TYP7
2047 (v)
2048 {
2049 default:
1bbd0b84 2050 badarg1:SCM_WTA (1,v);
0f2d19dd
JB
2051 case scm_tc7_smob:
2052 SCM_ASRTGO (SCM_ARRAYP (v), badarg1);
2053 return ra2l (v, SCM_ARRAY_BASE (v), 0);
2054 case scm_tc7_vector:
95f5b0f5 2055 case scm_tc7_wvect:
0f2d19dd
JB
2056 return scm_vector_to_list (v);
2057 case scm_tc7_string:
2058 return scm_string_to_list (v);
2059 case scm_tc7_bvect:
2060 {
2061 long *data = (long *) SCM_VELTS (v);
2062 register unsigned long mask;
2063 for (k = (SCM_LENGTH (v) - 1) / SCM_LONG_BIT; k > 0; k--)
cdbadcac 2064 for (mask = 1UL << (SCM_LONG_BIT - 1); mask; mask >>= 1)
156dcb09 2065 res = scm_cons (SCM_BOOL(((long *) data)[k] & mask), res);
0f2d19dd 2066 for (mask = 1L << ((SCM_LENGTH (v) % SCM_LONG_BIT) - 1); mask; mask >>= 1)
156dcb09 2067 res = scm_cons (SCM_BOOL(((long *) data)[k] & mask), res);
0f2d19dd
JB
2068 return res;
2069 }
2070# ifdef SCM_INUMS_ONLY
2071 case scm_tc7_uvect:
2072 case scm_tc7_ivect:
2073 {
2074 long *data = (long *) SCM_VELTS (v);
2075 for (k = SCM_LENGTH (v) - 1; k >= 0; k--)
2076 res = scm_cons (SCM_MAKINUM (data[k]), res);
2077 return res;
2078 }
2079# else
2080 case scm_tc7_uvect: {
2081 long *data = (long *)SCM_VELTS(v);
2082 for (k = SCM_LENGTH(v) - 1; k >= 0; k--)
2083 res = scm_cons(scm_ulong2num(data[k]), res);
2084 return res;
2085 }
2086 case scm_tc7_ivect: {
2087 long *data = (long *)SCM_VELTS(v);
2088 for (k = SCM_LENGTH(v) - 1; k >= 0; k--)
2089 res = scm_cons(scm_long2num(data[k]), res);
2090 return res;
2091 }
2092# endif
2093 case scm_tc7_svect: {
2094 short *data;
2095 data = (short *)SCM_VELTS(v);
2096 for (k = SCM_LENGTH(v) - 1; k >= 0; k--)
2097 res = scm_cons(SCM_MAKINUM (data[k]), res);
2098 return res;
2099 }
5c11cc9d 2100#ifdef HAVE_LONG_LONGS
0f2d19dd
JB
2101 case scm_tc7_llvect: {
2102 long_long *data;
2103 data = (long_long *)SCM_VELTS(v);
2104 for (k = SCM_LENGTH(v) - 1; k >= 0; k--)
2105 res = scm_cons(scm_long_long2num(data[k]), res);
2106 return res;
2107 }
2108#endif
2109
2110
2111#ifdef SCM_FLOATS
2112#ifdef SCM_SINGLES
2113 case scm_tc7_fvect:
2114 {
2115 float *data = (float *) SCM_VELTS (v);
2116 for (k = SCM_LENGTH (v) - 1; k >= 0; k--)
2117 res = scm_cons (scm_makflo (data[k]), res);
2118 return res;
2119 }
2120#endif /*SCM_SINGLES*/
2121 case scm_tc7_dvect:
2122 {
2123 double *data = (double *) SCM_VELTS (v);
2124 for (k = SCM_LENGTH (v) - 1; k >= 0; k--)
2125 res = scm_cons (scm_makdbl (data[k], 0.0), res);
2126 return res;
2127 }
2128 case scm_tc7_cvect:
2129 {
2130 double (*data)[2] = (double (*)[2]) SCM_VELTS (v);
2131 for (k = SCM_LENGTH (v) - 1; k >= 0; k--)
2132 res = scm_cons (scm_makdbl (data[k][0], data[k][1]), res);
2133 return res;
2134 }
2135#endif /*SCM_FLOATS*/
2136 }
2137}
1bbd0b84 2138#undef FUNC_NAME
0f2d19dd
JB
2139
2140
20a54673 2141static char s_bad_ralst[] = "Bad scm_array contents list";
1cc91f1b 2142
1bbd0b84 2143static int l2ra(SCM lst, SCM ra, scm_sizet base, scm_sizet k);
1cc91f1b 2144
3b3b36dd 2145SCM_DEFINE (scm_list_to_uniform_array, "list->uniform-array", 3, 0, 0,
1bbd0b84 2146 (SCM ndim, SCM prot, SCM lst),
b380b885
MD
2147 "@deffnx procedure list->uniform-vector prot lst\n"
2148 "Returns a uniform array of the type indicated by prototype @var{prot}\n"
2149 "with elements the same as those of @var{lst}. Elements must be of the\n"
2150 "appropriate type, no coercions are done.")
1bbd0b84 2151#define FUNC_NAME s_scm_list_to_uniform_array
0f2d19dd
JB
2152{
2153 SCM shp = SCM_EOL;
2154 SCM row = lst;
2155 SCM ra;
2156 scm_sizet k;
2157 long n;
3b3b36dd 2158 SCM_VALIDATE_INUM_COPY (1,ndim,k);
0f2d19dd
JB
2159 while (k--)
2160 {
2161 n = scm_ilength (row);
1bbd0b84 2162 SCM_ASSERT (n >= 0, lst, SCM_ARG3, FUNC_NAME);
0f2d19dd
JB
2163 shp = scm_cons (SCM_MAKINUM (n), shp);
2164 if (SCM_NIMP (row))
2165 row = SCM_CAR (row);
2166 }
d12feca3
GH
2167 ra = scm_dimensions_to_uniform_array (scm_reverse (shp), prot,
2168 SCM_UNDEFINED);
0f2d19dd
JB
2169 if (SCM_NULLP (shp))
2170
2171 {
2172 SCM_ASRTGO (1 == scm_ilength (lst), badlst);
2173 scm_array_set_x (ra, SCM_CAR (lst), SCM_EOL);
2174 return ra;
2175 }
2176 if (!SCM_ARRAYP (ra))
2177 {
2178 for (k = 0; k < SCM_LENGTH (ra); k++, lst = SCM_CDR (lst))
2179 scm_array_set_x (ra, SCM_CAR (lst), SCM_MAKINUM (k));
2180 return ra;
2181 }
2182 if (l2ra (lst, ra, SCM_ARRAY_BASE (ra), 0))
2183 return ra;
2184 else
1bbd0b84 2185 badlst:scm_wta (lst, s_bad_ralst, FUNC_NAME);
0f2d19dd
JB
2186 return SCM_BOOL_F;
2187}
1bbd0b84 2188#undef FUNC_NAME
0f2d19dd 2189
0f2d19dd 2190static int
1bbd0b84 2191l2ra (SCM lst, SCM ra, scm_sizet base, scm_sizet k)
0f2d19dd
JB
2192{
2193 register long inc = SCM_ARRAY_DIMS (ra)[k].inc;
2194 register long n = (1 + SCM_ARRAY_DIMS (ra)[k].ubnd - SCM_ARRAY_DIMS (ra)[k].lbnd);
2195 int ok = 1;
2196 if (n <= 0)
2197 return (SCM_EOL == lst);
2198 if (k < SCM_ARRAY_NDIM (ra) - 1)
2199 {
2200 while (n--)
2201 {
2202 if (SCM_IMP (lst) || SCM_NCONSP (lst))
2203 return 0;
2204 ok = ok && l2ra (SCM_CAR (lst), ra, base, k + 1);
2205 base += inc;
2206 lst = SCM_CDR (lst);
2207 }
2208 if (SCM_NNULLP (lst))
2209 return 0;
2210 }
2211 else
2212 {
2213 while (n--)
2214 {
2215 if (SCM_IMP (lst) || SCM_NCONSP (lst))
2216 return 0;
2217 ok = ok && scm_array_set_x (SCM_ARRAY_V (ra), SCM_CAR (lst), SCM_MAKINUM (base));
2218 base += inc;
2219 lst = SCM_CDR (lst);
2220 }
2221 if (SCM_NNULLP (lst))
2222 return 0;
2223 }
2224 return ok;
2225}
2226
1cc91f1b 2227
0f2d19dd 2228static void
1bbd0b84 2229rapr1 (SCM ra,scm_sizet j,scm_sizet k,SCM port,scm_print_state *pstate)
0f2d19dd
JB
2230{
2231 long inc = 1;
2232 long n = SCM_LENGTH (ra);
2233 int enclosed = 0;
2234tail:
5c11cc9d 2235 switch SCM_TYP7 (ra)
0f2d19dd
JB
2236 {
2237 case scm_tc7_smob:
2238 if (enclosed++)
2239 {
2240 SCM_ARRAY_BASE (ra) = j;
2241 if (n-- > 0)
9882ea19 2242 scm_iprin1 (ra, port, pstate);
0f2d19dd
JB
2243 for (j += inc; n-- > 0; j += inc)
2244 {
b7f3516f 2245 scm_putc (' ', port);
0f2d19dd 2246 SCM_ARRAY_BASE (ra) = j;
9882ea19 2247 scm_iprin1 (ra, port, pstate);
0f2d19dd
JB
2248 }
2249 break;
2250 }
2251 if (k + 1 < SCM_ARRAY_NDIM (ra))
2252 {
2253 long i;
2254 inc = SCM_ARRAY_DIMS (ra)[k].inc;
2255 for (i = SCM_ARRAY_DIMS (ra)[k].lbnd; i < SCM_ARRAY_DIMS (ra)[k].ubnd; i++)
2256 {
b7f3516f 2257 scm_putc ('(', port);
9882ea19 2258 rapr1 (ra, j, k + 1, port, pstate);
b7f3516f 2259 scm_puts (") ", port);
0f2d19dd
JB
2260 j += inc;
2261 }
2262 if (i == SCM_ARRAY_DIMS (ra)[k].ubnd)
2263 { /* could be zero size. */
b7f3516f 2264 scm_putc ('(', port);
9882ea19 2265 rapr1 (ra, j, k + 1, port, pstate);
b7f3516f 2266 scm_putc (')', port);
0f2d19dd
JB
2267 }
2268 break;
2269 }
2270 if SCM_ARRAY_NDIM
2271 (ra)
2272 { /* Could be zero-dimensional */
2273 inc = SCM_ARRAY_DIMS (ra)[k].inc;
2274 n = (SCM_ARRAY_DIMS (ra)[k].ubnd - SCM_ARRAY_DIMS (ra)[k].lbnd + 1);
2275 }
2276 else
2277 n = 1;
2278 ra = SCM_ARRAY_V (ra);
2279 goto tail;
2280 default:
5c11cc9d 2281 /* scm_tc7_bvect and scm_tc7_llvect only? */
0f2d19dd 2282 if (n-- > 0)
9882ea19 2283 scm_iprin1 (scm_uniform_vector_ref (ra, SCM_MAKINUM (j)), port, pstate);
0f2d19dd
JB
2284 for (j += inc; n-- > 0; j += inc)
2285 {
b7f3516f 2286 scm_putc (' ', port);
9882ea19 2287 scm_iprin1 (scm_cvref (ra, j, SCM_UNDEFINED), port, pstate);
0f2d19dd
JB
2288 }
2289 break;
2290 case scm_tc7_string:
2291 if (n-- > 0)
7866a09b 2292 scm_iprin1 (SCM_MAKE_CHAR (SCM_UCHARS (ra)[j]), port, pstate);
9882ea19 2293 if (SCM_WRITINGP (pstate))
0f2d19dd
JB
2294 for (j += inc; n-- > 0; j += inc)
2295 {
b7f3516f 2296 scm_putc (' ', port);
7866a09b 2297 scm_iprin1 (SCM_MAKE_CHAR (SCM_UCHARS (ra)[j]), port, pstate);
0f2d19dd
JB
2298 }
2299 else
2300 for (j += inc; n-- > 0; j += inc)
b7f3516f 2301 scm_putc (SCM_CHARS (ra)[j], port);
0f2d19dd
JB
2302 break;
2303 case scm_tc7_byvect:
2304 if (n-- > 0)
2305 scm_intprint (((char *)SCM_CDR (ra))[j], 10, port);
2306 for (j += inc; n-- > 0; j += inc)
2307 {
b7f3516f 2308 scm_putc (' ', port);
0f2d19dd
JB
2309 scm_intprint (((char *)SCM_CDR (ra))[j], 10, port);
2310 }
2311 break;
2312
2313 case scm_tc7_uvect:
5c11cc9d
GH
2314 {
2315 char str[11];
2316
2317 if (n-- > 0)
2318 {
2319 /* intprint can't handle >= 2^31. */
2320 sprintf (str, "%lu", (unsigned long) SCM_VELTS (ra)[j]);
2321 scm_puts (str, port);
2322 }
2323 for (j += inc; n-- > 0; j += inc)
2324 {
2325 scm_putc (' ', port);
2326 sprintf (str, "%lu", (unsigned long) SCM_VELTS (ra)[j]);
2327 scm_puts (str, port);
2328 }
2329 }
0f2d19dd
JB
2330 case scm_tc7_ivect:
2331 if (n-- > 0)
2332 scm_intprint (SCM_VELTS (ra)[j], 10, port);
2333 for (j += inc; n-- > 0; j += inc)
2334 {
b7f3516f 2335 scm_putc (' ', port);
0f2d19dd
JB
2336 scm_intprint (SCM_VELTS (ra)[j], 10, port);
2337 }
2338 break;
2339
2340 case scm_tc7_svect:
2341 if (n-- > 0)
2342 scm_intprint (((short *)SCM_CDR (ra))[j], 10, port);
2343 for (j += inc; n-- > 0; j += inc)
2344 {
b7f3516f 2345 scm_putc (' ', port);
0f2d19dd
JB
2346 scm_intprint (((short *)SCM_CDR (ra))[j], 10, port);
2347 }
2348 break;
2349
2350#ifdef SCM_FLOATS
2351#ifdef SCM_SINGLES
2352 case scm_tc7_fvect:
2353 if (n-- > 0)
2354 {
2355 SCM z = scm_makflo (1.0);
2356 SCM_FLO (z) = ((float *) SCM_VELTS (ra))[j];
9882ea19 2357 scm_floprint (z, port, pstate);
0f2d19dd
JB
2358 for (j += inc; n-- > 0; j += inc)
2359 {
b7f3516f 2360 scm_putc (' ', port);
0f2d19dd 2361 SCM_FLO (z) = ((float *) SCM_VELTS (ra))[j];
9882ea19 2362 scm_floprint (z, port, pstate);
0f2d19dd
JB
2363 }
2364 }
2365 break;
2366#endif /*SCM_SINGLES*/
2367 case scm_tc7_dvect:
2368 if (n-- > 0)
2369 {
2370 SCM z = scm_makdbl (1.0 / 3.0, 0.0);
2371 SCM_REAL (z) = ((double *) SCM_VELTS (ra))[j];
9882ea19 2372 scm_floprint (z, port, pstate);
0f2d19dd
JB
2373 for (j += inc; n-- > 0; j += inc)
2374 {
b7f3516f 2375 scm_putc (' ', port);
0f2d19dd 2376 SCM_REAL (z) = ((double *) SCM_VELTS (ra))[j];
9882ea19 2377 scm_floprint (z, port, pstate);
0f2d19dd
JB
2378 }
2379 }
2380 break;
2381 case scm_tc7_cvect:
2382 if (n-- > 0)
2383 {
2384 SCM cz = scm_makdbl (0.0, 1.0), z = scm_makdbl (1.0 / 3.0, 0.0);
2385 SCM_REAL (z) = SCM_REAL (cz) = (((double *) SCM_VELTS (ra))[2 * j]);
2386 SCM_IMAG (cz) = ((double *) SCM_VELTS (ra))[2 * j + 1];
9882ea19 2387 scm_floprint ((0.0 == SCM_IMAG (cz) ? z : cz), port, pstate);
0f2d19dd
JB
2388 for (j += inc; n-- > 0; j += inc)
2389 {
b7f3516f 2390 scm_putc (' ', port);
0f2d19dd
JB
2391 SCM_REAL (z) = SCM_REAL (cz) = ((double *) SCM_VELTS (ra))[2 * j];
2392 SCM_IMAG (cz) = ((double *) SCM_VELTS (ra))[2 * j + 1];
9882ea19 2393 scm_floprint ((0.0 == SCM_IMAG (cz) ? z : cz), port, pstate);
0f2d19dd
JB
2394 }
2395 }
2396 break;
2397#endif /*SCM_FLOATS*/
2398 }
2399}
2400
2401
1cc91f1b 2402
0f2d19dd 2403int
1bbd0b84 2404scm_raprin1 (SCM exp, SCM port, scm_print_state *pstate)
0f2d19dd
JB
2405{
2406 SCM v = exp;
2407 scm_sizet base = 0;
b7f3516f 2408 scm_putc ('#', port);
0f2d19dd 2409tail:
5c11cc9d 2410 switch SCM_TYP7 (v)
0f2d19dd
JB
2411 {
2412 case scm_tc7_smob:
2413 {
2414 long ndim = SCM_ARRAY_NDIM (v);
2415 base = SCM_ARRAY_BASE (v);
2416 v = SCM_ARRAY_V (v);
2417 if (SCM_ARRAYP (v))
2418
2419 {
b7f3516f 2420 scm_puts ("<enclosed-array ", port);
9882ea19 2421 rapr1 (exp, base, 0, port, pstate);
b7f3516f 2422 scm_putc ('>', port);
0f2d19dd
JB
2423 return 1;
2424 }
2425 else
2426 {
2427 scm_intprint (ndim, 10, port);
2428 goto tail;
2429 }
2430 }
2431 case scm_tc7_bvect:
2432 if (exp == v)
2433 { /* a uve, not an scm_array */
2434 register long i, j, w;
b7f3516f 2435 scm_putc ('*', port);
0f2d19dd
JB
2436 for (i = 0; i < (SCM_LENGTH (exp)) / SCM_LONG_BIT; i++)
2437 {
2438 w = SCM_VELTS (exp)[i];
2439 for (j = SCM_LONG_BIT; j; j--)
2440 {
b7f3516f 2441 scm_putc (w & 1 ? '1' : '0', port);
0f2d19dd
JB
2442 w >>= 1;
2443 }
2444 }
2445 j = SCM_LENGTH (exp) % SCM_LONG_BIT;
2446 if (j)
2447 {
2448 w = SCM_VELTS (exp)[SCM_LENGTH (exp) / SCM_LONG_BIT];
2449 for (; j; j--)
2450 {
b7f3516f 2451 scm_putc (w & 1 ? '1' : '0', port);
0f2d19dd
JB
2452 w >>= 1;
2453 }
2454 }
2455 return 1;
2456 }
2457 else
b7f3516f 2458 scm_putc ('b', port);
0f2d19dd
JB
2459 break;
2460 case scm_tc7_string:
b7f3516f 2461 scm_putc ('a', port);
0f2d19dd
JB
2462 break;
2463 case scm_tc7_byvect:
05c33d09 2464 scm_putc ('y', port);
0f2d19dd
JB
2465 break;
2466 case scm_tc7_uvect:
b7f3516f 2467 scm_putc ('u', port);
0f2d19dd
JB
2468 break;
2469 case scm_tc7_ivect:
b7f3516f 2470 scm_putc ('e', port);
0f2d19dd
JB
2471 break;
2472 case scm_tc7_svect:
05c33d09 2473 scm_putc ('h', port);
0f2d19dd 2474 break;
5c11cc9d 2475#ifdef HAVE_LONG_LONGS
0f2d19dd 2476 case scm_tc7_llvect:
5c11cc9d 2477 scm_putc ('l', port);
0f2d19dd
JB
2478 break;
2479#endif
2480#ifdef SCM_FLOATS
2481#ifdef SCM_SINGLES
2482 case scm_tc7_fvect:
b7f3516f 2483 scm_putc ('s', port);
0f2d19dd
JB
2484 break;
2485#endif /*SCM_SINGLES*/
2486 case scm_tc7_dvect:
b7f3516f 2487 scm_putc ('i', port);
0f2d19dd
JB
2488 break;
2489 case scm_tc7_cvect:
b7f3516f 2490 scm_putc ('c', port);
0f2d19dd
JB
2491 break;
2492#endif /*SCM_FLOATS*/
2493 }
b7f3516f 2494 scm_putc ('(', port);
9882ea19 2495 rapr1 (exp, base, 0, port, pstate);
b7f3516f 2496 scm_putc (')', port);
0f2d19dd
JB
2497 return 1;
2498}
2499
3b3b36dd 2500SCM_DEFINE (scm_array_prototype, "array-prototype", 1, 0, 0,
1bbd0b84 2501 (SCM ra),
b380b885
MD
2502 "Returns an object that would produce an array of the same type as\n"
2503 "@var{array}, if used as the @var{prototype} for\n"
2504 "@code{make-uniform-array}.")
1bbd0b84 2505#define FUNC_NAME s_scm_array_prototype
0f2d19dd
JB
2506{
2507 int enclosed = 0;
2508 SCM_ASRTGO (SCM_NIMP (ra), badarg);
2509loop:
2510 switch SCM_TYP7
2511 (ra)
2512 {
2513 default:
1bbd0b84 2514 badarg:SCM_WTA (1,ra);
0f2d19dd
JB
2515 case scm_tc7_smob:
2516 SCM_ASRTGO (SCM_ARRAYP (ra), badarg);
2517 if (enclosed++)
2518 return SCM_UNSPECIFIED;
2519 ra = SCM_ARRAY_V (ra);
2520 goto loop;
2521 case scm_tc7_vector:
95f5b0f5 2522 case scm_tc7_wvect:
0f2d19dd
JB
2523 return SCM_EOL;
2524 case scm_tc7_bvect:
2525 return SCM_BOOL_T;
2526 case scm_tc7_string:
7866a09b 2527 return SCM_MAKE_CHAR ('a');
0f2d19dd 2528 case scm_tc7_byvect:
7866a09b 2529 return SCM_MAKE_CHAR ('\0');
0f2d19dd
JB
2530 case scm_tc7_uvect:
2531 return SCM_MAKINUM (1L);
2532 case scm_tc7_ivect:
2533 return SCM_MAKINUM (-1L);
2534 case scm_tc7_svect:
2535 return SCM_CDR (scm_intern ("s", 1));
5c11cc9d 2536#ifdef HAVE_LONG_LONGS
0f2d19dd
JB
2537 case scm_tc7_llvect:
2538 return SCM_CDR (scm_intern ("l", 1));
2539#endif
2540#ifdef SCM_FLOATS
2541#ifdef SCM_SINGLES
2542 case scm_tc7_fvect:
2543 return scm_makflo (1.0);
2544#endif
2545 case scm_tc7_dvect:
2546 return scm_makdbl (1.0 / 3.0, 0.0);
2547 case scm_tc7_cvect:
2548 return scm_makdbl (0.0, 1.0);
2549#endif
2550 }
2551}
1bbd0b84 2552#undef FUNC_NAME
0f2d19dd 2553
1cc91f1b 2554
0f2d19dd 2555static SCM
1bbd0b84 2556markra (SCM ptr)
0f2d19dd 2557{
0f2d19dd
JB
2558 return SCM_ARRAY_V (ptr);
2559}
2560
1cc91f1b 2561
0f2d19dd 2562static scm_sizet
1bbd0b84 2563freera (SCM ptr)
0f2d19dd
JB
2564{
2565 scm_must_free (SCM_CHARS (ptr));
2566 return sizeof (scm_array) + SCM_ARRAY_NDIM (ptr) * sizeof (scm_array_dim);
2567}
2568
0f2d19dd
JB
2569void
2570scm_init_unif ()
0f2d19dd 2571{
23a62151
MD
2572 scm_tc16_array = scm_make_smob_type_mfpe ("array", 0,
2573 markra,
2574 freera,
2575 scm_raprin1,
2576 scm_array_equal_p);
0f2d19dd 2577 scm_add_feature ("array");
23a62151 2578#include "unif.x"
0f2d19dd 2579}