1
|
1 // The constants for the tree class. -*- C++ -*- |
|
2 /* |
|
3 |
|
4 Copyright (C) 1992, 1993 John W. Eaton |
|
5 |
|
6 This file is part of Octave. |
|
7 |
|
8 Octave is free software; you can redistribute it and/or modify it |
|
9 under the terms of the GNU General Public License as published by the |
|
10 Free Software Foundation; either version 2, or (at your option) any |
|
11 later version. |
|
12 |
|
13 Octave is distributed in the hope that it will be useful, but WITHOUT |
|
14 ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or |
|
15 FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License |
|
16 for more details. |
|
17 |
|
18 You should have received a copy of the GNU General Public License |
|
19 along with Octave; see the file COPYING. If not, write to the Free |
|
20 Software Foundation, 675 Mass Ave, Cambridge, MA 02139, USA. |
|
21 |
|
22 */ |
|
23 |
240
|
24 #ifdef HAVE_CONFIG_H |
|
25 #include "config.h" |
1
|
26 #endif |
|
27 |
|
28 #include <ctype.h> |
|
29 #include <string.h> |
|
30 #include <iostream.h> |
|
31 #include <strstream.h> |
|
32 #include <math.h> |
|
33 |
|
34 #include "variables.h" |
|
35 #include "error.h" |
|
36 #include "gripes.h" |
|
37 #include "user-prefs.h" |
|
38 #include "utils.h" |
|
39 #include "pager.h" |
|
40 #include "mappers.h" |
|
41 #include "pr-output.h" |
|
42 #include "tree-const.h" |
|
43 #include "arith-ops.h" |
|
44 |
|
45 // A couple of handy helper functions. |
|
46 |
|
47 static int |
|
48 any_element_is_negative (const Matrix& a) |
|
49 { |
|
50 int nr = a.rows (); |
|
51 int nc = a.columns (); |
|
52 for (int j = 0; j < nc; j++) |
|
53 for (int i = 0; i < nr; i++) |
|
54 if (a.elem (i, j) < 0.0) |
|
55 return 1; |
|
56 return 0; |
|
57 } |
|
58 |
|
59 static int |
|
60 any_element_is_complex (const ComplexMatrix& a) |
|
61 { |
|
62 int nr = a.rows (); |
|
63 int nc = a.columns (); |
|
64 for (int j = 0; j < nc; j++) |
|
65 for (int i = 0; i < nr; i++) |
|
66 if (imag (a.elem (i, j)) != 0.0) |
|
67 return 1; |
|
68 return 0; |
|
69 } |
|
70 |
|
71 // Now, the classes. |
|
72 |
|
73 /* |
|
74 * The real representation of constants. |
|
75 */ |
|
76 tree_constant_rep::tree_constant_rep (void) |
|
77 { |
|
78 type_tag = unknown_constant; |
|
79 } |
|
80 |
|
81 tree_constant_rep::tree_constant_rep (double d) |
|
82 { |
|
83 scalar = d; |
|
84 type_tag = scalar_constant; |
|
85 } |
|
86 |
161
|
87 tree_constant_rep::tree_constant_rep (const Matrix& m) |
1
|
88 { |
|
89 if (m.rows () == 1 && m.columns () == 1) |
|
90 { |
|
91 scalar = m.elem (0, 0); |
|
92 type_tag = scalar_constant; |
|
93 } |
|
94 else |
|
95 { |
|
96 matrix = new Matrix (m); |
|
97 type_tag = matrix_constant; |
|
98 } |
|
99 } |
|
100 |
161
|
101 tree_constant_rep::tree_constant_rep (const DiagMatrix& d) |
1
|
102 { |
|
103 if (d.rows () == 1 && d.columns () == 1) |
|
104 { |
|
105 scalar = d.elem (0, 0); |
|
106 type_tag = scalar_constant; |
|
107 } |
|
108 else |
|
109 { |
|
110 matrix = new Matrix (d); |
|
111 type_tag = matrix_constant; |
|
112 } |
|
113 } |
|
114 |
161
|
115 tree_constant_rep::tree_constant_rep (const RowVector& v) |
1
|
116 { |
|
117 int len = v.capacity (); |
|
118 if (len == 1) |
|
119 { |
|
120 scalar = v.elem (0); |
|
121 type_tag = scalar_constant; |
|
122 } |
|
123 else |
|
124 { |
|
125 if (user_pref.prefer_column_vectors) |
|
126 { |
|
127 Matrix m (len, 1); |
|
128 for (int i = 0; i < len; i++) |
|
129 m.elem (i, 0) = v.elem (i); |
|
130 matrix = new Matrix (m); |
|
131 type_tag = matrix_constant; |
|
132 } |
|
133 else |
|
134 { |
|
135 Matrix m (1, len); |
|
136 for (int i = 0; i < len; i++) |
|
137 m.elem (0, i) = v.elem (i); |
|
138 matrix = new Matrix (m); |
|
139 type_tag = matrix_constant; |
|
140 } |
|
141 } |
|
142 } |
|
143 |
161
|
144 tree_constant_rep::tree_constant_rep (const RowVector& v, int prefer_column_vector) |
1
|
145 { |
|
146 int len = v.capacity (); |
|
147 if (len == 1) |
|
148 { |
|
149 scalar = v.elem (0); |
|
150 type_tag = scalar_constant; |
|
151 } |
|
152 else |
|
153 { |
|
154 if (prefer_column_vector) |
|
155 { |
|
156 Matrix m (len, 1); |
|
157 for (int i = 0; i < len; i++) |
|
158 m.elem (i, 0) = v.elem (i); |
|
159 matrix = new Matrix (m); |
|
160 type_tag = matrix_constant; |
|
161 } |
|
162 else |
|
163 { |
|
164 Matrix m (1, len); |
|
165 for (int i = 0; i < len; i++) |
|
166 m.elem (0, i) = v.elem (i); |
|
167 matrix = new Matrix (m); |
|
168 type_tag = matrix_constant; |
|
169 } |
|
170 } |
|
171 } |
|
172 |
161
|
173 tree_constant_rep::tree_constant_rep (const ColumnVector& v) |
1
|
174 { |
|
175 int len = v.capacity (); |
|
176 if (len == 1) |
|
177 { |
|
178 scalar = v.elem (0); |
|
179 type_tag = scalar_constant; |
|
180 } |
|
181 else |
|
182 { |
|
183 if (user_pref.prefer_column_vectors) |
|
184 { |
|
185 Matrix m (len, 1); |
|
186 for (int i = 0; i < len; i++) |
|
187 m.elem (i, 0) = v.elem (i); |
|
188 matrix = new Matrix (m); |
|
189 type_tag = matrix_constant; |
|
190 } |
|
191 else |
|
192 { |
|
193 Matrix m (1, len); |
|
194 for (int i = 0; i < len; i++) |
|
195 m.elem (0, i) = v.elem (i); |
|
196 matrix = new Matrix (m); |
|
197 type_tag = matrix_constant; |
|
198 } |
|
199 } |
|
200 } |
|
201 |
161
|
202 tree_constant_rep::tree_constant_rep (const ColumnVector& v, |
1
|
203 int prefer_column_vector) |
|
204 { |
|
205 int len = v.capacity (); |
|
206 if (len == 1) |
|
207 { |
|
208 scalar = v.elem (0); |
|
209 type_tag = scalar_constant; |
|
210 } |
|
211 else |
|
212 { |
|
213 if (prefer_column_vector) |
|
214 { |
|
215 Matrix m (len, 1); |
|
216 for (int i = 0; i < len; i++) |
|
217 m.elem (i, 0) = v.elem (i); |
|
218 matrix = new Matrix (m); |
|
219 type_tag = matrix_constant; |
|
220 } |
|
221 else |
|
222 { |
|
223 Matrix m (1, len); |
|
224 for (int i = 0; i < len; i++) |
|
225 m.elem (0, i) = v.elem (i); |
|
226 matrix = new Matrix (m); |
|
227 type_tag = matrix_constant; |
|
228 } |
|
229 } |
|
230 } |
|
231 |
161
|
232 tree_constant_rep::tree_constant_rep (const Complex& c) |
1
|
233 { |
|
234 complex_scalar = new Complex (c); |
|
235 type_tag = complex_scalar_constant; |
|
236 } |
|
237 |
161
|
238 tree_constant_rep::tree_constant_rep (const ComplexRowVector& v) |
1
|
239 { |
|
240 int len = v.capacity (); |
|
241 if (len == 1) |
|
242 { |
|
243 complex_scalar = new Complex (v.elem (0)); |
|
244 type_tag = complex_scalar_constant; |
|
245 } |
|
246 else |
|
247 { |
|
248 if (user_pref.prefer_column_vectors) |
|
249 { |
|
250 ComplexMatrix m (len, 1); |
|
251 for (int i = 0; i < len; i++) |
|
252 m.elem (i, 0) = v.elem (i); |
|
253 complex_matrix = new ComplexMatrix (m); |
|
254 type_tag = complex_matrix_constant; |
|
255 } |
|
256 else |
|
257 { |
|
258 ComplexMatrix m (1, len); |
|
259 for (int i = 0; i < len; i++) |
|
260 m.elem (0, i) = v.elem (i); |
|
261 complex_matrix = new ComplexMatrix (m); |
|
262 type_tag = complex_matrix_constant; |
|
263 } |
|
264 } |
|
265 } |
|
266 |
161
|
267 tree_constant_rep::tree_constant_rep (const ComplexMatrix& m) |
1
|
268 { |
|
269 if (m.rows () == 1 && m.columns () == 1) |
|
270 { |
|
271 complex_scalar = new Complex (m.elem (0, 0)); |
|
272 type_tag = complex_scalar_constant; |
|
273 } |
|
274 else |
|
275 { |
|
276 complex_matrix = new ComplexMatrix (m); |
|
277 type_tag = complex_matrix_constant; |
|
278 } |
|
279 } |
|
280 |
161
|
281 tree_constant_rep::tree_constant_rep (const ComplexDiagMatrix& d) |
1
|
282 { |
|
283 if (d.rows () == 1 && d.columns () == 1) |
|
284 { |
|
285 complex_scalar = new Complex (d.elem (0, 0)); |
|
286 type_tag = complex_scalar_constant; |
|
287 } |
|
288 else |
|
289 { |
|
290 complex_matrix = new ComplexMatrix (d); |
|
291 type_tag = complex_matrix_constant; |
|
292 } |
|
293 } |
|
294 |
161
|
295 tree_constant_rep::tree_constant_rep (const ComplexRowVector& v, |
1
|
296 int prefer_column_vector) |
|
297 { |
|
298 int len = v.capacity (); |
|
299 if (len == 1) |
|
300 { |
|
301 complex_scalar = new Complex (v.elem (0)); |
|
302 type_tag = complex_scalar_constant; |
|
303 } |
|
304 else |
|
305 { |
|
306 if (prefer_column_vector) |
|
307 { |
|
308 ComplexMatrix m (len, 1); |
|
309 for (int i = 0; i < len; i++) |
|
310 m.elem (i, 0) = v.elem (i); |
|
311 complex_matrix = new ComplexMatrix (m); |
|
312 type_tag = complex_matrix_constant; |
|
313 } |
|
314 else |
|
315 { |
|
316 ComplexMatrix m (1, len); |
|
317 for (int i = 0; i < len; i++) |
|
318 m.elem (0, i) = v.elem (i); |
|
319 complex_matrix = new ComplexMatrix (m); |
|
320 type_tag = complex_matrix_constant; |
|
321 } |
|
322 } |
|
323 } |
|
324 |
161
|
325 tree_constant_rep::tree_constant_rep (const ComplexColumnVector& v) |
1
|
326 { |
|
327 int len = v.capacity (); |
|
328 if (len == 1) |
|
329 { |
|
330 complex_scalar = new Complex (v.elem (0)); |
|
331 type_tag = complex_scalar_constant; |
|
332 } |
|
333 else |
|
334 { |
|
335 if (user_pref.prefer_column_vectors) |
|
336 { |
|
337 ComplexMatrix m (len, 1); |
|
338 for (int i = 0; i < len; i++) |
|
339 m.elem (i, 0) = v.elem (i); |
|
340 complex_matrix = new ComplexMatrix (m); |
|
341 type_tag = complex_matrix_constant; |
|
342 } |
|
343 else |
|
344 { |
|
345 ComplexMatrix m (1, len); |
|
346 for (int i = 0; i < len; i++) |
|
347 m.elem (0, i) = v.elem (i); |
|
348 complex_matrix = new ComplexMatrix (m); |
|
349 type_tag = complex_matrix_constant; |
|
350 } |
|
351 } |
|
352 } |
|
353 |
161
|
354 tree_constant_rep::tree_constant_rep (const ComplexColumnVector& v, |
1
|
355 int prefer_column_vector) |
|
356 { |
|
357 int len = v.capacity (); |
|
358 if (len == 1) |
|
359 { |
|
360 complex_scalar = new Complex (v.elem (0)); |
|
361 type_tag = complex_scalar_constant; |
|
362 } |
|
363 else |
|
364 { |
|
365 if (prefer_column_vector) |
|
366 { |
|
367 ComplexMatrix m (len, 1); |
|
368 for (int i = 0; i < len; i++) |
|
369 m.elem (i, 0) = v.elem (i); |
|
370 complex_matrix = new ComplexMatrix (m); |
|
371 type_tag = complex_matrix_constant; |
|
372 } |
|
373 else |
|
374 { |
|
375 ComplexMatrix m (1, len); |
|
376 for (int i = 0; i < len; i++) |
|
377 m.elem (0, i) = v.elem (i); |
|
378 complex_matrix = new ComplexMatrix (m); |
|
379 type_tag = complex_matrix_constant; |
|
380 } |
|
381 } |
|
382 } |
|
383 |
|
384 tree_constant_rep::tree_constant_rep (const char *s) |
|
385 { |
|
386 string = strsave (s); |
|
387 type_tag = string_constant; |
|
388 } |
|
389 |
|
390 tree_constant_rep::tree_constant_rep (double b, double l, double i) |
|
391 { |
|
392 range = new Range (b, l, i); |
|
393 int nel = range->nelem (); |
|
394 if (nel < 0) |
|
395 { |
191
|
396 delete range; |
|
397 type_tag = unknown_constant; |
1
|
398 if (nel == -1) |
240
|
399 ::error ("number of elements in range exceeds INT_MAX"); |
1
|
400 else |
240
|
401 ::error ("invalid range"); |
1
|
402 } |
|
403 else if (nel > 1) |
|
404 type_tag = range_constant; |
|
405 else |
|
406 { |
|
407 delete range; |
|
408 if (nel == 1) |
|
409 { |
|
410 scalar = b; |
|
411 type_tag = scalar_constant; |
|
412 } |
|
413 else if (nel == 0) |
|
414 { |
|
415 matrix = new Matrix (); |
|
416 type_tag = matrix_constant; |
|
417 } |
|
418 else |
|
419 panic_impossible (); |
|
420 } |
|
421 } |
|
422 |
161
|
423 tree_constant_rep::tree_constant_rep (const Range& r) |
1
|
424 { |
|
425 if (r.nelem () > 1) |
|
426 { |
|
427 range = new Range (r); |
|
428 type_tag = range_constant; |
|
429 } |
|
430 else if (r.nelem () == 1) |
|
431 { |
|
432 scalar = r.base (); |
|
433 type_tag = scalar_constant; |
|
434 } |
|
435 else if (r.nelem () == 0) |
|
436 { |
|
437 matrix = new Matrix (); |
|
438 type_tag = matrix_constant; |
|
439 } |
|
440 else |
|
441 panic_impossible (); |
|
442 } |
|
443 |
|
444 tree_constant_rep::tree_constant_rep (tree_constant_rep::constant_type t) |
|
445 { |
|
446 assert (t == magic_colon); |
|
447 |
|
448 type_tag = magic_colon; |
|
449 } |
|
450 |
161
|
451 tree_constant_rep::tree_constant_rep (const tree_constant_rep& t) |
1
|
452 { |
|
453 type_tag = t.type_tag; |
|
454 |
|
455 switch (t.type_tag) |
|
456 { |
|
457 case unknown_constant: |
|
458 break; |
|
459 case scalar_constant: |
|
460 scalar = t.scalar; |
|
461 break; |
|
462 case matrix_constant: |
|
463 matrix = new Matrix (*(t.matrix)); |
|
464 break; |
|
465 case string_constant: |
|
466 string = strsave (t.string); |
|
467 break; |
|
468 case complex_matrix_constant: |
|
469 complex_matrix = new ComplexMatrix (*(t.complex_matrix)); |
|
470 break; |
|
471 case complex_scalar_constant: |
|
472 complex_scalar = new Complex (*(t.complex_scalar)); |
|
473 break; |
|
474 case range_constant: |
|
475 range = new Range (*(t.range)); |
|
476 break; |
|
477 case magic_colon: |
|
478 break; |
|
479 default: |
|
480 panic_impossible (); |
|
481 break; |
|
482 } |
|
483 } |
|
484 |
|
485 tree_constant_rep::~tree_constant_rep (void) |
|
486 { |
|
487 switch (type_tag) |
|
488 { |
|
489 case unknown_constant: |
|
490 break; |
|
491 case scalar_constant: |
|
492 break; |
|
493 case matrix_constant: |
|
494 delete matrix; |
|
495 break; |
|
496 case complex_scalar_constant: |
|
497 delete complex_scalar; |
|
498 break; |
|
499 case complex_matrix_constant: |
|
500 delete complex_matrix; |
|
501 break; |
|
502 case string_constant: |
|
503 delete [] string; |
|
504 break; |
|
505 case range_constant: |
|
506 delete range; |
|
507 break; |
|
508 case magic_colon: |
|
509 break; |
|
510 default: |
|
511 panic_impossible (); |
|
512 break; |
|
513 } |
|
514 } |
|
515 |
|
516 #if defined (MDEBUG) |
|
517 void * |
|
518 tree_constant_rep::operator new (size_t size) |
|
519 { |
|
520 tree_constant_rep *p = ::new tree_constant_rep; |
|
521 cerr << "tree_constant_rep::new(): " << p << "\n"; |
|
522 return p; |
|
523 } |
|
524 |
|
525 void |
|
526 tree_constant_rep::operator delete (void *p, size_t size) |
|
527 { |
|
528 cerr << "tree_constant_rep::delete(): " << p << "\n"; |
|
529 ::delete p; |
|
530 } |
|
531 #endif |
|
532 |
|
533 void |
|
534 tree_constant_rep::resize (int i, int j) |
|
535 { |
|
536 switch (type_tag) |
|
537 { |
|
538 case matrix_constant: |
|
539 matrix->resize (i, j); |
|
540 break; |
|
541 case complex_matrix_constant: |
|
542 complex_matrix->resize (i, j); |
|
543 break; |
|
544 default: |
|
545 panic_impossible (); |
|
546 break; |
|
547 } |
|
548 } |
|
549 |
|
550 void |
|
551 tree_constant_rep::resize (int i, int j, double val) |
|
552 { |
|
553 switch (type_tag) |
|
554 { |
|
555 case matrix_constant: |
|
556 matrix->resize (i, j, val); |
|
557 break; |
|
558 case complex_matrix_constant: |
|
559 complex_matrix->resize (i, j, val); |
|
560 break; |
|
561 default: |
|
562 panic_impossible (); |
|
563 break; |
|
564 } |
|
565 } |
|
566 |
|
567 void |
|
568 tree_constant_rep::maybe_resize (int i, int j) |
|
569 { |
|
570 int nr = rows (); |
|
571 int nc = columns (); |
|
572 |
|
573 i++; |
|
574 j++; |
|
575 |
|
576 assert (i > 0 && j > 0); |
|
577 |
|
578 if (i > nr || j > nc) |
|
579 { |
|
580 if (user_pref.resize_on_range_error) |
|
581 resize (MAX (i, nr), MAX (j, nc), 0.0); |
|
582 else |
|
583 { |
|
584 if (i > nr) |
240
|
585 ::error ("row index = %d exceeds max row dimension = %d", i, nr); |
191
|
586 |
1
|
587 if (j > nc) |
240
|
588 ::error ("column index = %d exceeds max column dimension = %d", |
|
589 j, nc); |
1
|
590 } |
|
591 } |
|
592 } |
|
593 |
|
594 void |
|
595 tree_constant_rep::maybe_resize (int i, force_orient f_orient = no_orient) |
|
596 { |
|
597 int nr = rows (); |
|
598 int nc = columns (); |
|
599 |
|
600 i++; |
|
601 |
|
602 assert (i > 0 && (nr <= 1 || nc <= 1)); |
|
603 |
|
604 if (nr <= 1 && nc <= 1 && i >= 1) |
|
605 { |
|
606 if (user_pref.resize_on_range_error) |
|
607 { |
|
608 if (f_orient == row_orient) |
|
609 resize (1, i, 0.0); |
|
610 else if (f_orient == column_orient) |
|
611 resize (i, 1, 0.0); |
|
612 else if (user_pref.prefer_column_vectors) |
|
613 resize (i, 1, 0.0); |
|
614 else |
|
615 resize (1, i, 0.0); |
|
616 } |
|
617 else |
240
|
618 ::error ("matrix index = %d exceeds max dimension = %d", i, nc); |
1
|
619 } |
|
620 else if (nr == 1 && i > nc) |
|
621 { |
|
622 if (user_pref.resize_on_range_error) |
|
623 resize (1, i, 0.0); |
|
624 else |
240
|
625 ::error ("matrix index = %d exceeds max dimension = %d", i, nc); |
1
|
626 } |
|
627 else if (nc == 1 && i > nr) |
|
628 { |
|
629 if (user_pref.resize_on_range_error) |
|
630 resize (i, 1, 0.0); |
|
631 else |
240
|
632 ::error ("matrix index = %d exceeds max dimension = ", i, nc); |
1
|
633 } |
|
634 } |
|
635 |
|
636 double |
164
|
637 tree_constant_rep::to_scalar (void) const |
1
|
638 { |
|
639 tree_constant tmp = make_numeric (); |
|
640 |
|
641 double retval = 0.0; |
|
642 |
|
643 switch (tmp.const_type ()) |
|
644 { |
|
645 case tree_constant_rep::scalar_constant: |
|
646 case tree_constant_rep::complex_scalar_constant: |
|
647 retval = tmp.double_value (); |
|
648 break; |
|
649 case tree_constant_rep::matrix_constant: |
|
650 if (user_pref.do_fortran_indexing) |
|
651 { |
|
652 Matrix m = tmp.matrix_value (); |
|
653 retval = m (0, 0); |
|
654 } |
|
655 break; |
|
656 case tree_constant_rep::complex_matrix_constant: |
|
657 if (user_pref.do_fortran_indexing) |
|
658 { |
|
659 int flag = user_pref.ok_to_lose_imaginary_part; |
|
660 if (flag == -1) |
|
661 warning ("implicit conversion of complex value to real value"); |
|
662 |
|
663 if (flag != 0) |
|
664 { |
|
665 ComplexMatrix m = tmp.complex_matrix_value (); |
240
|
666 return ::real (m (0, 0)); |
1
|
667 } |
|
668 else |
|
669 jump_to_top_level (); |
|
670 } |
|
671 else |
|
672 { |
240
|
673 ::error ("complex matrix used in invalid context"); |
1
|
674 jump_to_top_level (); |
|
675 } |
|
676 break; |
|
677 default: |
|
678 break; |
|
679 } |
|
680 return retval; |
|
681 } |
|
682 |
|
683 ColumnVector |
164
|
684 tree_constant_rep::to_vector (void) const |
1
|
685 { |
|
686 tree_constant tmp = make_numeric (); |
|
687 |
|
688 ColumnVector retval; |
|
689 |
|
690 switch (tmp.const_type ()) |
|
691 { |
|
692 case tree_constant_rep::scalar_constant: |
|
693 case tree_constant_rep::complex_scalar_constant: |
|
694 retval.resize (1); |
|
695 retval.elem (0) = tmp.double_value (); |
|
696 break; |
|
697 case tree_constant_rep::complex_matrix_constant: |
|
698 case tree_constant_rep::matrix_constant: |
|
699 { |
|
700 Matrix m = tmp.matrix_value (); |
|
701 int nr = m.rows (); |
|
702 int nc = m.columns (); |
|
703 if (nr == 1) |
|
704 { |
|
705 retval.resize (nc); |
|
706 for (int i = 0; i < nc; i++) |
|
707 retval.elem (i) = m (0, i); |
|
708 } |
|
709 else if (nc == 1) |
|
710 { |
|
711 retval.resize (nr); |
|
712 for (int i = 0; i < nr; i++) |
|
713 retval.elem (i) = m.elem (i, 0); |
|
714 } |
|
715 } |
|
716 break; |
|
717 default: |
|
718 panic_impossible (); |
|
719 break; |
|
720 } |
|
721 return retval; |
|
722 } |
|
723 |
|
724 Matrix |
164
|
725 tree_constant_rep::to_matrix (void) const |
1
|
726 { |
|
727 tree_constant tmp = make_numeric (); |
|
728 |
|
729 Matrix retval; |
|
730 |
|
731 switch (tmp.const_type ()) |
|
732 { |
|
733 case tree_constant_rep::scalar_constant: |
|
734 retval.resize (1, 1); |
|
735 retval.elem (0, 0) = tmp.double_value (); |
|
736 break; |
|
737 case tree_constant_rep::matrix_constant: |
|
738 retval = tmp.matrix_value (); |
|
739 break; |
|
740 default: |
|
741 break; |
|
742 } |
|
743 return retval; |
|
744 } |
|
745 |
|
746 tree_constant_rep::constant_type |
|
747 tree_constant_rep::force_numeric (int force_str_conv = 0) |
|
748 { |
|
749 switch (type_tag) |
|
750 { |
|
751 case scalar_constant: |
|
752 case matrix_constant: |
|
753 case complex_scalar_constant: |
|
754 case complex_matrix_constant: |
|
755 break; |
|
756 case string_constant: |
|
757 { |
|
758 if (! force_str_conv && ! user_pref.implicit_str_to_num_ok) |
|
759 { |
240
|
760 ::error ("failed to convert `%s' to a numeric type --", string); |
|
761 ::error ("default conversion turned off"); |
1
|
762 // Abort! |
|
763 jump_to_top_level (); |
|
764 } |
|
765 |
|
766 int len = strlen (string); |
|
767 if (len > 1) |
|
768 { |
|
769 type_tag = matrix_constant; |
|
770 Matrix *tm = new Matrix (1, len); |
|
771 for (int i = 0; i < len; i++) |
|
772 tm->elem (0, i) = toascii ((int) string[i]); |
|
773 matrix = tm; |
|
774 } |
|
775 else if (len == 1) |
|
776 { |
|
777 type_tag = scalar_constant; |
|
778 scalar = toascii ((int) string[0]); |
|
779 } |
240
|
780 else if (len == 0) |
|
781 { |
|
782 type_tag = matrix_constant; |
|
783 matrix = new Matrix (0, 0); |
|
784 } |
|
785 else |
|
786 panic_impossible (); |
1
|
787 } |
|
788 break; |
|
789 case range_constant: |
|
790 { |
|
791 int len = range->nelem (); |
|
792 if (len > 1) |
|
793 { |
|
794 type_tag = matrix_constant; |
|
795 Matrix *tm = new Matrix (1, len); |
|
796 double b = range->base (); |
|
797 double increment = range->inc (); |
|
798 for (int i = 0; i < len; i++) |
|
799 tm->elem (0, i) = b + i * increment; |
|
800 matrix = tm; |
|
801 } |
|
802 else if (len == 1) |
|
803 { |
|
804 type_tag = scalar_constant; |
|
805 scalar = range->base (); |
|
806 } |
|
807 } |
|
808 break; |
|
809 case magic_colon: |
|
810 default: |
|
811 panic_impossible (); |
|
812 break; |
|
813 } |
|
814 return type_tag; |
|
815 } |
|
816 |
|
817 tree_constant |
164
|
818 tree_constant_rep::make_numeric (int force_str_conv = 0) const |
1
|
819 { |
|
820 tree_constant retval; |
|
821 switch (type_tag) |
|
822 { |
|
823 case scalar_constant: |
|
824 retval = tree_constant (scalar); |
|
825 break; |
|
826 case matrix_constant: |
|
827 retval = tree_constant (*matrix); |
|
828 break; |
|
829 case complex_scalar_constant: |
|
830 retval = tree_constant (*complex_scalar); |
|
831 break; |
|
832 case complex_matrix_constant: |
|
833 retval = tree_constant (*complex_matrix); |
|
834 break; |
|
835 case string_constant: |
|
836 retval = tree_constant (string); |
|
837 retval.force_numeric (force_str_conv); |
|
838 break; |
|
839 case range_constant: |
|
840 retval = tree_constant (*range); |
|
841 retval.force_numeric (force_str_conv); |
|
842 break; |
|
843 case magic_colon: |
|
844 default: |
|
845 panic_impossible (); |
|
846 break; |
|
847 } |
|
848 return retval; |
|
849 } |
|
850 |
|
851 tree_constant |
|
852 do_binary_op (tree_constant& a, tree_constant& b, tree::expression_type t) |
|
853 { |
143
|
854 tree_constant ans; |
|
855 |
1
|
856 int first_empty = (a.rows () == 0 || a.columns () == 0); |
|
857 int second_empty = (b.rows () == 0 || b.columns () == 0); |
|
858 |
|
859 if (first_empty || second_empty) |
|
860 { |
|
861 int flag = user_pref.propagate_empty_matrices; |
|
862 if (flag < 0) |
|
863 warning ("binary operation on empty matrix"); |
|
864 else if (flag == 0) |
|
865 { |
240
|
866 ::error ("invalid binary operation on empty matrix"); |
143
|
867 return ans; |
1
|
868 } |
|
869 } |
|
870 |
|
871 tree_constant tmp_a = a.make_numeric (); |
|
872 tree_constant tmp_b = b.make_numeric (); |
|
873 |
|
874 tree_constant_rep::constant_type a_type = tmp_a.const_type (); |
|
875 tree_constant_rep::constant_type b_type = tmp_b.const_type (); |
|
876 |
|
877 double d1, d2; |
|
878 Matrix m1, m2; |
|
879 Complex c1, c2; |
|
880 ComplexMatrix cm1, cm2; |
|
881 |
|
882 switch (a_type) |
|
883 { |
|
884 case tree_constant_rep::scalar_constant: |
|
885 d1 = tmp_a.double_value (); |
|
886 switch (b_type) |
|
887 { |
|
888 case tree_constant_rep::scalar_constant: |
|
889 d2 = tmp_b.double_value (); |
|
890 ans = do_binary_op (d1, d2, t); |
|
891 break; |
|
892 case tree_constant_rep::matrix_constant: |
|
893 m2 = tmp_b.matrix_value (); |
|
894 ans = do_binary_op (d1, m2, t); |
|
895 break; |
|
896 case tree_constant_rep::complex_scalar_constant: |
|
897 c2 = tmp_b.complex_value (); |
|
898 ans = do_binary_op (d1, c2, t); |
|
899 break; |
|
900 case tree_constant_rep::complex_matrix_constant: |
|
901 cm2 = tmp_b.complex_matrix_value (); |
|
902 ans = do_binary_op (d1, cm2, t); |
|
903 break; |
|
904 case tree_constant_rep::magic_colon: |
|
905 default: |
|
906 panic_impossible (); |
|
907 break; |
|
908 } |
|
909 break; |
|
910 case tree_constant_rep::matrix_constant: |
|
911 m1 = tmp_a.matrix_value (); |
|
912 switch (b_type) |
|
913 { |
|
914 case tree_constant_rep::scalar_constant: |
|
915 d2 = tmp_b.double_value (); |
|
916 ans = do_binary_op (m1, d2, t); |
|
917 break; |
|
918 case tree_constant_rep::matrix_constant: |
|
919 m2 = tmp_b.matrix_value (); |
|
920 ans = do_binary_op (m1, m2, t); |
|
921 break; |
|
922 case tree_constant_rep::complex_scalar_constant: |
|
923 c2 = tmp_b.complex_value (); |
|
924 ans = do_binary_op (m1, c2, t); |
|
925 break; |
|
926 case tree_constant_rep::complex_matrix_constant: |
|
927 cm2 = tmp_b.complex_matrix_value (); |
|
928 ans = do_binary_op (m1, cm2, t); |
|
929 break; |
|
930 case tree_constant_rep::magic_colon: |
|
931 default: |
|
932 panic_impossible (); |
|
933 break; |
|
934 } |
|
935 break; |
|
936 case tree_constant_rep::complex_scalar_constant: |
|
937 c1 = tmp_a.complex_value (); |
|
938 switch (b_type) |
|
939 { |
|
940 case tree_constant_rep::scalar_constant: |
|
941 d2 = tmp_b.double_value (); |
|
942 ans = do_binary_op (c1, d2, t); |
|
943 break; |
|
944 case tree_constant_rep::matrix_constant: |
|
945 m2 = tmp_b.matrix_value (); |
|
946 ans = do_binary_op (c1, m2, t); |
|
947 break; |
|
948 case tree_constant_rep::complex_scalar_constant: |
|
949 c2 = tmp_b.complex_value (); |
|
950 ans = do_binary_op (c1, c2, t); |
|
951 break; |
|
952 case tree_constant_rep::complex_matrix_constant: |
|
953 cm2 = tmp_b.complex_matrix_value (); |
|
954 ans = do_binary_op (c1, cm2, t); |
|
955 break; |
|
956 case tree_constant_rep::magic_colon: |
|
957 default: |
|
958 panic_impossible (); |
|
959 break; |
|
960 } |
|
961 break; |
|
962 case tree_constant_rep::complex_matrix_constant: |
|
963 cm1 = tmp_a.complex_matrix_value (); |
|
964 switch (b_type) |
|
965 { |
|
966 case tree_constant_rep::scalar_constant: |
|
967 d2 = tmp_b.double_value (); |
|
968 ans = do_binary_op (cm1, d2, t); |
|
969 break; |
|
970 case tree_constant_rep::matrix_constant: |
|
971 m2 = tmp_b.matrix_value (); |
|
972 ans = do_binary_op (cm1, m2, t); |
|
973 break; |
|
974 case tree_constant_rep::complex_scalar_constant: |
|
975 c2 = tmp_b.complex_value (); |
|
976 ans = do_binary_op (cm1, c2, t); |
|
977 break; |
|
978 case tree_constant_rep::complex_matrix_constant: |
|
979 cm2 = tmp_b.complex_matrix_value (); |
|
980 ans = do_binary_op (cm1, cm2, t); |
|
981 break; |
|
982 case tree_constant_rep::magic_colon: |
|
983 default: |
|
984 panic_impossible (); |
|
985 break; |
|
986 } |
|
987 break; |
|
988 case tree_constant_rep::magic_colon: |
|
989 default: |
|
990 panic_impossible (); |
|
991 break; |
|
992 } |
|
993 return ans; |
|
994 } |
|
995 |
|
996 tree_constant |
|
997 do_unary_op (tree_constant& a, tree::expression_type t) |
|
998 { |
143
|
999 tree_constant ans; |
|
1000 |
1
|
1001 if (a.rows () == 0 || a.columns () == 0) |
|
1002 { |
|
1003 int flag = user_pref.propagate_empty_matrices; |
|
1004 if (flag < 0) |
|
1005 warning ("unary operation on empty matrix"); |
|
1006 else if (flag == 0) |
|
1007 { |
240
|
1008 ::error ("invalid unary operation on empty matrix"); |
143
|
1009 return ans; |
1
|
1010 } |
|
1011 } |
|
1012 |
|
1013 tree_constant tmp_a = a.make_numeric (); |
|
1014 |
|
1015 switch (tmp_a.const_type ()) |
|
1016 { |
|
1017 case tree_constant_rep::scalar_constant: |
|
1018 ans = do_unary_op (tmp_a.double_value (), t); |
|
1019 break; |
|
1020 case tree_constant_rep::matrix_constant: |
|
1021 { |
|
1022 Matrix m = tmp_a.matrix_value (); |
|
1023 ans = do_unary_op (m, t); |
|
1024 } |
|
1025 break; |
|
1026 case tree_constant_rep::complex_scalar_constant: |
|
1027 ans = do_unary_op (tmp_a.complex_value (), t); |
|
1028 break; |
|
1029 case tree_constant_rep::complex_matrix_constant: |
|
1030 { |
|
1031 ComplexMatrix m = tmp_a.complex_matrix_value (); |
|
1032 ans = do_unary_op (m, t); |
|
1033 } |
|
1034 break; |
|
1035 case tree_constant_rep::magic_colon: |
|
1036 default: |
|
1037 panic_impossible (); |
|
1038 break; |
|
1039 } |
|
1040 return ans; |
|
1041 } |
|
1042 |
|
1043 void |
|
1044 tree_constant_rep::bump_value (tree::expression_type etype) |
|
1045 { |
|
1046 switch (etype) |
|
1047 { |
|
1048 case tree::increment: |
|
1049 switch (type_tag) |
|
1050 { |
|
1051 case scalar_constant: |
|
1052 scalar++; |
|
1053 break; |
|
1054 case matrix_constant: |
|
1055 *matrix = *matrix + 1.0; |
|
1056 break; |
|
1057 case complex_scalar_constant: |
|
1058 *complex_scalar = *complex_scalar + 1.0; |
|
1059 break; |
|
1060 case complex_matrix_constant: |
|
1061 *complex_matrix = *complex_matrix + 1.0; |
|
1062 break; |
|
1063 case string_constant: |
240
|
1064 ::error ("string++ and ++string not implemented yet, ok?"); |
1
|
1065 break; |
|
1066 case range_constant: |
|
1067 range->set_base (range->base () + 1.0); |
|
1068 range->set_limit (range->limit () + 1.0); |
|
1069 break; |
|
1070 case magic_colon: |
|
1071 default: |
|
1072 panic_impossible (); |
|
1073 break; |
|
1074 } |
|
1075 break; |
|
1076 case tree::decrement: |
|
1077 switch (type_tag) |
|
1078 { |
|
1079 case scalar_constant: |
|
1080 scalar--; |
|
1081 break; |
|
1082 case matrix_constant: |
|
1083 *matrix = *matrix - 1.0; |
|
1084 break; |
|
1085 case string_constant: |
240
|
1086 ::error ("string-- and -- string not implemented yet, ok?"); |
1
|
1087 break; |
|
1088 case range_constant: |
|
1089 range->set_base (range->base () - 1.0); |
|
1090 range->set_limit (range->limit () - 1.0); |
|
1091 break; |
|
1092 case magic_colon: |
|
1093 default: |
|
1094 panic_impossible (); |
|
1095 break; |
|
1096 } |
|
1097 break; |
|
1098 default: |
|
1099 panic_impossible (); |
|
1100 break; |
|
1101 } |
|
1102 } |
|
1103 |
|
1104 void |
|
1105 tree_constant_rep::eval (int print) |
|
1106 { |
143
|
1107 if (error_state) |
|
1108 return; |
|
1109 |
1
|
1110 switch (type_tag) |
|
1111 { |
|
1112 case complex_scalar_constant: |
240
|
1113 if (::imag (*complex_scalar) == 0.0) |
1
|
1114 { |
240
|
1115 double d = ::real (*complex_scalar); |
1
|
1116 delete complex_scalar; |
|
1117 scalar = d; |
|
1118 type_tag = scalar_constant; |
|
1119 } |
|
1120 break; |
|
1121 case complex_matrix_constant: |
|
1122 if (! any_element_is_complex (*complex_matrix)) |
|
1123 { |
240
|
1124 Matrix *m = new Matrix (::real (*complex_matrix)); |
1
|
1125 delete complex_matrix; |
|
1126 matrix = m; |
|
1127 type_tag = matrix_constant; |
|
1128 } |
|
1129 break; |
|
1130 case scalar_constant: |
|
1131 case matrix_constant: |
|
1132 case string_constant: |
|
1133 case range_constant: |
|
1134 case magic_colon: |
|
1135 break; |
|
1136 default: |
|
1137 panic_impossible (); |
|
1138 break; |
|
1139 } |
|
1140 |
145
|
1141 // Avoid calling rows() and columns() for things like magic_colon. |
143
|
1142 |
145
|
1143 int nr = 1; |
|
1144 int nc = 1; |
|
1145 if (type_tag == matrix_constant |
|
1146 || type_tag == complex_matrix_constant |
|
1147 || type_tag == range_constant) |
143
|
1148 { |
145
|
1149 nr = rows (); |
|
1150 nc = columns (); |
|
1151 } |
|
1152 |
|
1153 switch (type_tag) |
|
1154 { |
|
1155 case matrix_constant: |
|
1156 if (nr == 1 && nc == 1) |
143
|
1157 { |
145
|
1158 double d = matrix->elem (0, 0); |
|
1159 delete matrix; |
|
1160 scalar = d; |
|
1161 type_tag = scalar_constant; |
143
|
1162 } |
145
|
1163 break; |
|
1164 case complex_matrix_constant: |
|
1165 if (nr == 1 && nc == 1) |
|
1166 { |
|
1167 Complex c = complex_matrix->elem (0, 0); |
|
1168 delete complex_matrix; |
|
1169 complex_scalar = new Complex (c); |
|
1170 type_tag = complex_scalar_constant; |
|
1171 } |
|
1172 break; |
|
1173 case range_constant: |
|
1174 if (nr == 1 && nc == 1) |
|
1175 { |
|
1176 double d = range->base (); |
|
1177 delete range; |
|
1178 scalar = d; |
|
1179 type_tag = scalar_constant; |
|
1180 } |
|
1181 break; |
|
1182 default: |
|
1183 break; |
143
|
1184 } |
|
1185 |
1
|
1186 if (print) |
|
1187 { |
|
1188 ostrstream output_buf; |
|
1189 switch (type_tag) |
|
1190 { |
|
1191 case scalar_constant: |
|
1192 octave_print_internal (output_buf, scalar); |
|
1193 break; |
|
1194 case matrix_constant: |
|
1195 if (nr == 0 || nc == 0) |
|
1196 { |
|
1197 output_buf << "[]"; |
|
1198 if (user_pref.print_empty_dimensions) |
|
1199 output_buf << "(" << nr << "x" << nc << ")"; |
|
1200 output_buf << "\n"; |
|
1201 } |
|
1202 else |
|
1203 octave_print_internal (output_buf, *matrix); |
|
1204 break; |
|
1205 case complex_scalar_constant: |
|
1206 octave_print_internal (output_buf, *complex_scalar); |
|
1207 break; |
|
1208 case complex_matrix_constant: |
|
1209 if (nr == 0 || nc == 0) |
|
1210 { |
|
1211 output_buf << "[]"; |
|
1212 if (user_pref.print_empty_dimensions) |
|
1213 output_buf << "(" << nr << "x" << nc << ")"; |
|
1214 output_buf << "\n"; |
|
1215 } |
|
1216 else |
|
1217 octave_print_internal (output_buf, *complex_matrix); |
|
1218 break; |
|
1219 case string_constant: |
|
1220 output_buf << string << "\n"; |
|
1221 break; |
|
1222 case range_constant: |
|
1223 octave_print_internal (output_buf, *range); |
|
1224 break; |
|
1225 case magic_colon: |
|
1226 default: |
|
1227 panic_impossible (); |
|
1228 break; |
|
1229 } |
|
1230 |
|
1231 output_buf << ends; |
|
1232 maybe_page_output (output_buf); |
|
1233 } |
|
1234 } |
|
1235 |
|
1236 tree_constant * |
164
|
1237 tree_constant_rep::eval (const tree_constant *args, int nargin, int nargout, |
1
|
1238 int print) |
|
1239 { |
143
|
1240 if (error_state) |
|
1241 return NULL_TREE_CONST; |
|
1242 |
1
|
1243 tree_constant *retval = new tree_constant [2]; |
|
1244 switch (type_tag) |
|
1245 { |
|
1246 case complex_scalar_constant: |
|
1247 case scalar_constant: |
|
1248 retval[0] = do_scalar_index (args, nargin); |
|
1249 break; |
|
1250 case complex_matrix_constant: |
|
1251 case matrix_constant: |
|
1252 retval[0] = do_matrix_index (args, nargin); |
|
1253 break; |
|
1254 case string_constant: |
|
1255 gripe_string_invalid (); |
|
1256 // retval[0] = do_string_index (args, nargin); |
|
1257 break; |
|
1258 case magic_colon: |
|
1259 case range_constant: |
|
1260 // This isn\'t great, but it\'s easier than implementing a lot of |
|
1261 // range indexing functions. |
|
1262 force_numeric (); |
|
1263 assert (type_tag != magic_colon && type_tag != range_constant); |
|
1264 return eval (args, nargin, nargout, print); |
|
1265 break; |
|
1266 default: |
|
1267 panic_impossible (); |
|
1268 break; |
|
1269 } |
|
1270 |
|
1271 if (retval[0].is_defined ()) |
|
1272 retval[0].eval (print); |
|
1273 return retval; |
|
1274 } |
|
1275 |
|
1276 int |
276
|
1277 tree_constant_rep::save (ostream& os, int mark_as_global, int precision) |
1
|
1278 { |
|
1279 switch (type_tag) |
|
1280 { |
|
1281 case scalar_constant: |
|
1282 case matrix_constant: |
|
1283 case complex_scalar_constant: |
|
1284 case complex_matrix_constant: |
|
1285 case string_constant: |
|
1286 case range_constant: |
|
1287 if (mark_as_global) |
|
1288 os << "# type: global "; |
|
1289 else |
|
1290 os << "# type: "; |
|
1291 break; |
|
1292 case magic_colon: |
|
1293 default: |
|
1294 break; |
|
1295 } |
|
1296 |
276
|
1297 long old_precision = os.precision (); |
|
1298 os.precision (precision); |
|
1299 |
1
|
1300 switch (type_tag) |
|
1301 { |
|
1302 case scalar_constant: |
|
1303 os << "scalar\n" |
|
1304 << scalar << "\n"; |
|
1305 break; |
|
1306 case matrix_constant: |
|
1307 os << "matrix\n" |
|
1308 << "# rows: " << rows () << "\n" |
|
1309 << "# columns: " << columns () << "\n" |
|
1310 << *matrix ; |
|
1311 break; |
|
1312 case complex_scalar_constant: |
|
1313 os << "complex scalar\n" |
|
1314 << *complex_scalar << "\n"; |
|
1315 break; |
|
1316 case complex_matrix_constant: |
|
1317 os << "complex matrix\n" |
|
1318 << "# rows: " << rows () << "\n" |
|
1319 << "# columns: " << columns () << "\n" |
|
1320 << *complex_matrix ; |
|
1321 break; |
|
1322 case string_constant: |
|
1323 os << "string\n" |
|
1324 << "# length: " << strlen (string) << "\n" |
|
1325 << string << "\n"; |
|
1326 break; |
|
1327 case range_constant: |
|
1328 { |
|
1329 os << "range\n" |
|
1330 << "# base, limit, increment\n" |
|
1331 << range->base () << " " |
|
1332 << range->limit () << " " |
|
1333 << range->inc () << "\n"; |
|
1334 } |
|
1335 break; |
|
1336 case magic_colon: |
|
1337 default: |
|
1338 panic_impossible (); |
|
1339 break; |
|
1340 } |
276
|
1341 |
|
1342 os.precision (old_precision); |
|
1343 |
1
|
1344 // Really want to return 1 only if write is successful. |
|
1345 return 1; |
|
1346 } |
|
1347 |
|
1348 int |
|
1349 tree_constant_rep::save_three_d (ostream& os, int parametric) |
|
1350 { |
|
1351 int nr = rows (); |
|
1352 int nc = columns (); |
|
1353 |
|
1354 switch (type_tag) |
|
1355 { |
|
1356 case matrix_constant: |
|
1357 os << "# 3D data...\n" |
|
1358 << "# type: matrix\n" |
|
1359 << "# total rows: " << nr << "\n" |
|
1360 << "# total columns: " << nc << "\n"; |
|
1361 |
|
1362 if (parametric) |
|
1363 { |
|
1364 int extras = nc % 3; |
|
1365 if (extras) |
|
1366 warning ("ignoring last %d columns", extras); |
|
1367 |
|
1368 for (int i = 0; i < nc-extras; i += 3) |
|
1369 { |
|
1370 os << matrix->extract (0, i, nr-1, i+2); |
|
1371 if (i+3 < nc-extras) |
|
1372 os << "\n"; |
|
1373 } |
|
1374 } |
|
1375 else |
|
1376 { |
|
1377 for (int i = 0; i < nc; i++) |
|
1378 { |
|
1379 os << matrix->extract (0, i, nr-1, i); |
|
1380 if (i+1 < nc) |
|
1381 os << "\n"; |
|
1382 } |
|
1383 } |
|
1384 break; |
|
1385 default: |
240
|
1386 ::error ("for now, I can only save real matrices in 3D format"); |
1
|
1387 return 0; |
|
1388 break; |
|
1389 } |
|
1390 // Really want to return 1 only if write is successful. |
|
1391 return 1; |
|
1392 } |
|
1393 |
|
1394 int |
|
1395 tree_constant_rep::load (istream& is) |
|
1396 { |
|
1397 int is_global = 0; |
|
1398 |
|
1399 type_tag = unknown_constant; |
|
1400 |
|
1401 // Look for type keyword |
279
|
1402 |
|
1403 char *tag = extract_keyword (is, "type"); |
|
1404 |
|
1405 if (tag != (char *) NULL && *tag != '\0') |
1
|
1406 { |
279
|
1407 char *ptr = strchr (tag, ' '); |
|
1408 if (ptr != (char *) NULL) |
1
|
1409 { |
279
|
1410 *ptr = '\0'; |
|
1411 is_global = (strncmp (tag, "global", 6) == 0); |
|
1412 *ptr = ' '; |
|
1413 ptr++; |
1
|
1414 } |
|
1415 else |
279
|
1416 ptr = &tag[0]; |
|
1417 |
|
1418 if (strncmp (ptr, "scalar", 6) == 0) |
|
1419 type_tag = load (is, scalar_constant); |
|
1420 else if (strncmp (ptr, "matrix", 6) == 0) |
|
1421 type_tag = load (is, matrix_constant); |
|
1422 else if (strncmp (ptr, "complex scalar", 14) == 0) |
|
1423 type_tag = load (is, complex_scalar_constant); |
|
1424 else if (strncmp (ptr, "complex matrix", 14) == 0) |
|
1425 type_tag = load (is, complex_matrix_constant); |
|
1426 else if (strncmp (ptr, "string", 6) == 0) |
|
1427 type_tag = load (is, string_constant); |
|
1428 else if (strncmp (ptr, "range", 5) == 0) |
|
1429 type_tag = load (is, range_constant); |
|
1430 else |
|
1431 ::error ("unknown constant type `%s'", tag); |
1
|
1432 } |
279
|
1433 else |
|
1434 ::error ("failed to extract keyword specifying value type"); |
1
|
1435 |
|
1436 return is_global; |
|
1437 } |
|
1438 |
|
1439 tree_constant_rep::constant_type |
|
1440 tree_constant_rep::load (istream& is, tree_constant_rep::constant_type t) |
|
1441 { |
|
1442 tree_constant_rep::constant_type status = unknown_constant; |
|
1443 |
|
1444 switch (t) |
|
1445 { |
|
1446 case scalar_constant: |
|
1447 is >> scalar; |
|
1448 if (is) |
|
1449 status = scalar_constant; |
|
1450 else |
240
|
1451 ::error ("failed to load scalar constant"); |
1
|
1452 break; |
|
1453 case matrix_constant: |
|
1454 { |
|
1455 int nr = 0, nc = 0; |
|
1456 |
|
1457 if (extract_keyword (is, "rows", nr) && nr > 0 |
|
1458 && extract_keyword (is, "columns", nc) && nc > 0) |
|
1459 { |
|
1460 matrix = new Matrix (nr, nc); |
|
1461 is >> *matrix; |
|
1462 if (is) |
|
1463 status = matrix_constant; |
|
1464 else |
240
|
1465 ::error ("failed to load matrix constant"); |
1
|
1466 } |
|
1467 else |
240
|
1468 ::error ("failed to extract number of rows and columns"); |
1
|
1469 } |
|
1470 break; |
|
1471 case complex_scalar_constant: |
|
1472 is >> *complex_scalar; |
|
1473 if (is) |
|
1474 status = complex_scalar_constant; |
|
1475 else |
240
|
1476 ::error ("failed to load complex scalar constant"); |
1
|
1477 break; |
|
1478 case complex_matrix_constant: |
|
1479 { |
|
1480 int nr = 0, nc = 0; |
|
1481 |
|
1482 if (extract_keyword (is, "rows", nr) && nr > 0 |
|
1483 && extract_keyword (is, "columns", nc) && nc > 0) |
|
1484 { |
|
1485 complex_matrix = new ComplexMatrix (nr, nc); |
|
1486 is >> *complex_matrix; |
|
1487 if (is) |
|
1488 status = complex_matrix_constant; |
|
1489 else |
240
|
1490 ::error ("failed to load complex matrix constant"); |
1
|
1491 } |
|
1492 else |
240
|
1493 ::error ("failed to extract number of rows and columns"); |
1
|
1494 } |
|
1495 break; |
|
1496 case string_constant: |
|
1497 { |
|
1498 int len; |
|
1499 if (extract_keyword (is, "length", len) && len > 0) |
|
1500 { |
|
1501 string = new char [len+1]; |
|
1502 is.get (string, len+1, EOF); |
|
1503 if (is) |
|
1504 status = string_constant; |
|
1505 else |
240
|
1506 ::error ("failed to load string constant"); |
1
|
1507 } |
|
1508 else |
240
|
1509 ::error ("failed to extract string length"); |
1
|
1510 } |
|
1511 break; |
|
1512 case range_constant: |
|
1513 skip_comments (is); |
|
1514 range = new Range (); |
|
1515 is >> *range; |
|
1516 if (is) |
|
1517 status = range_constant; |
|
1518 else |
240
|
1519 ::error ("failed to load range constant"); |
1
|
1520 break; |
|
1521 default: |
|
1522 panic_impossible (); |
|
1523 break; |
|
1524 } |
|
1525 return status; |
|
1526 } |
|
1527 |
|
1528 double |
164
|
1529 tree_constant_rep::double_value (void) const |
1
|
1530 { |
105
|
1531 switch (type_tag) |
1
|
1532 { |
105
|
1533 case scalar_constant: |
|
1534 return scalar; |
|
1535 case complex_scalar_constant: |
|
1536 { |
|
1537 int flag = user_pref.ok_to_lose_imaginary_part; |
|
1538 if (flag == -1) |
|
1539 warning ("implicit conversion of complex value to real value"); |
1
|
1540 |
105
|
1541 if (flag != 0) |
240
|
1542 return ::real (*complex_scalar); |
105
|
1543 |
240
|
1544 ::error ("implicit conversion of complex value to real value"); |
|
1545 ::error ("not allowed"); |
105
|
1546 jump_to_top_level (); |
|
1547 } |
|
1548 default: |
|
1549 panic_impossible (); |
|
1550 break; |
1
|
1551 } |
|
1552 } |
|
1553 |
|
1554 Matrix |
164
|
1555 tree_constant_rep::matrix_value (void) const |
1
|
1556 { |
105
|
1557 switch (type_tag) |
1
|
1558 { |
206
|
1559 case scalar_constant: |
240
|
1560 return Matrix (1, 1, scalar); |
105
|
1561 case matrix_constant: |
|
1562 return *matrix; |
206
|
1563 case complex_scalar_constant: |
105
|
1564 case complex_matrix_constant: |
|
1565 { |
|
1566 int flag = user_pref.ok_to_lose_imaginary_part; |
|
1567 if (flag == -1) |
|
1568 warning ("implicit conversion of complex matrix to real matrix"); |
1
|
1569 |
105
|
1570 if (flag != 0) |
206
|
1571 { |
|
1572 if (type_tag == complex_scalar_constant) |
240
|
1573 return Matrix (1, 1, ::real (*complex_scalar)); |
206
|
1574 else if (type_tag == complex_matrix_constant) |
240
|
1575 return ::real (*complex_matrix); |
206
|
1576 else |
|
1577 panic_impossible (); |
|
1578 } |
105
|
1579 else |
240
|
1580 { |
|
1581 ::error ("implicit conversion of complex matrix to real matrix"); |
|
1582 ::error ("not allowed"); |
|
1583 } |
105
|
1584 jump_to_top_level (); |
|
1585 } |
|
1586 default: |
|
1587 panic_impossible (); |
|
1588 break; |
1
|
1589 } |
|
1590 } |
|
1591 |
|
1592 Complex |
164
|
1593 tree_constant_rep::complex_value (void) const |
1
|
1594 { |
101
|
1595 switch (type_tag) |
|
1596 { |
|
1597 case complex_scalar_constant: |
105
|
1598 return *complex_scalar; |
101
|
1599 case scalar_constant: |
105
|
1600 return Complex (scalar); |
101
|
1601 default: |
|
1602 panic_impossible (); |
|
1603 break; |
|
1604 } |
1
|
1605 } |
|
1606 |
|
1607 ComplexMatrix |
164
|
1608 tree_constant_rep::complex_matrix_value (void) const |
1
|
1609 { |
105
|
1610 switch (type_tag) |
101
|
1611 { |
|
1612 case scalar_constant: |
|
1613 { |
240
|
1614 return ComplexMatrix (1, 1, Complex (scalar)); |
101
|
1615 } |
|
1616 case complex_scalar_constant: |
105
|
1617 { |
240
|
1618 return ComplexMatrix (1, 1, *complex_scalar); |
105
|
1619 } |
101
|
1620 case matrix_constant: |
|
1621 { |
105
|
1622 return ComplexMatrix (*matrix); |
101
|
1623 } |
|
1624 case complex_matrix_constant: |
105
|
1625 return *complex_matrix; |
101
|
1626 break; |
|
1627 default: |
|
1628 panic_impossible (); |
|
1629 break; |
|
1630 } |
1
|
1631 } |
|
1632 |
|
1633 char * |
164
|
1634 tree_constant_rep::string_value (void) const |
1
|
1635 { |
|
1636 assert (type_tag == string_constant); |
|
1637 return string; |
|
1638 } |
|
1639 |
|
1640 Range |
164
|
1641 tree_constant_rep::range_value (void) const |
1
|
1642 { |
|
1643 assert (type_tag == range_constant); |
|
1644 return *range; |
|
1645 } |
|
1646 |
|
1647 int |
164
|
1648 tree_constant_rep::rows (void) const |
1
|
1649 { |
|
1650 int retval = -1; |
|
1651 switch (type_tag) |
|
1652 { |
|
1653 case scalar_constant: |
|
1654 case complex_scalar_constant: |
|
1655 case string_constant: |
|
1656 case range_constant: |
|
1657 retval = 1; |
|
1658 break; |
|
1659 case matrix_constant: |
|
1660 retval = matrix->rows (); |
|
1661 break; |
|
1662 case complex_matrix_constant: |
|
1663 retval = complex_matrix->rows (); |
|
1664 break; |
|
1665 case magic_colon: |
240
|
1666 ::error ("invalid use of colon operator"); |
1
|
1667 break; |
|
1668 case unknown_constant: |
|
1669 retval = 0; |
|
1670 break; |
|
1671 default: |
|
1672 panic_impossible (); |
|
1673 break; |
|
1674 } |
|
1675 return retval; |
|
1676 } |
|
1677 |
|
1678 int |
164
|
1679 tree_constant_rep::columns (void) const |
1
|
1680 { |
|
1681 int retval = -1; |
|
1682 switch (type_tag) |
|
1683 { |
|
1684 case scalar_constant: |
|
1685 case complex_scalar_constant: |
|
1686 retval = 1; |
|
1687 break; |
|
1688 case matrix_constant: |
|
1689 retval = matrix->columns (); |
|
1690 break; |
|
1691 case complex_matrix_constant: |
|
1692 retval = complex_matrix->columns (); |
|
1693 break; |
|
1694 case string_constant: |
|
1695 retval = strlen (string); |
|
1696 break; |
|
1697 case range_constant: |
|
1698 retval = range->nelem (); |
|
1699 break; |
|
1700 case magic_colon: |
240
|
1701 ::error ("invalid use of colon operator"); |
1
|
1702 break; |
|
1703 case unknown_constant: |
|
1704 retval = 0; |
|
1705 break; |
|
1706 default: |
|
1707 panic_impossible (); |
|
1708 break; |
|
1709 } |
|
1710 return retval; |
|
1711 } |
|
1712 |
|
1713 tree_constant |
164
|
1714 tree_constant_rep::all (void) const |
1
|
1715 { |
|
1716 if (type_tag == string_constant || type_tag == range_constant) |
|
1717 { |
|
1718 tree_constant tmp = make_numeric (); |
|
1719 return tmp.all (); |
|
1720 } |
|
1721 |
|
1722 tree_constant retval; |
|
1723 switch (type_tag) |
|
1724 { |
|
1725 case scalar_constant: |
|
1726 { |
|
1727 double status = (scalar != 0.0); |
|
1728 retval = tree_constant (status); |
|
1729 } |
|
1730 break; |
|
1731 case matrix_constant: |
|
1732 { |
|
1733 Matrix m = matrix->all (); |
|
1734 retval = tree_constant (m); |
|
1735 } |
|
1736 break; |
|
1737 case complex_scalar_constant: |
|
1738 { |
|
1739 double status = (*complex_scalar != 0.0); |
|
1740 retval = tree_constant (status); |
|
1741 } |
|
1742 break; |
|
1743 case complex_matrix_constant: |
|
1744 { |
|
1745 Matrix m = complex_matrix->all (); |
|
1746 retval = tree_constant (m); |
|
1747 } |
|
1748 break; |
|
1749 case string_constant: |
|
1750 case range_constant: |
|
1751 case magic_colon: |
|
1752 default: |
|
1753 panic_impossible (); |
|
1754 break; |
|
1755 } |
|
1756 return retval; |
|
1757 } |
|
1758 |
|
1759 tree_constant |
164
|
1760 tree_constant_rep::any (void) const |
1
|
1761 { |
|
1762 if (type_tag == string_constant || type_tag == range_constant) |
|
1763 { |
|
1764 tree_constant tmp = make_numeric (); |
|
1765 return tmp.any (); |
|
1766 } |
|
1767 |
|
1768 tree_constant retval; |
|
1769 switch (type_tag) |
|
1770 { |
|
1771 case scalar_constant: |
|
1772 { |
|
1773 double status = (scalar != 0.0); |
|
1774 retval = tree_constant (status); |
|
1775 } |
|
1776 break; |
|
1777 case matrix_constant: |
|
1778 { |
|
1779 Matrix m = matrix->any (); |
|
1780 retval = tree_constant (m); |
|
1781 } |
|
1782 break; |
|
1783 case complex_scalar_constant: |
|
1784 { |
|
1785 double status = (*complex_scalar != 0.0); |
|
1786 retval = tree_constant (status); |
|
1787 } |
|
1788 break; |
|
1789 case complex_matrix_constant: |
|
1790 { |
|
1791 Matrix m = complex_matrix->any (); |
|
1792 retval = tree_constant (m); |
|
1793 } |
|
1794 break; |
|
1795 case string_constant: |
|
1796 case range_constant: |
|
1797 case magic_colon: |
|
1798 default: |
|
1799 panic_impossible (); |
|
1800 break; |
|
1801 } |
|
1802 return retval; |
|
1803 } |
|
1804 |
|
1805 tree_constant |
164
|
1806 tree_constant_rep::isstr (void) const |
1
|
1807 { |
|
1808 double status = 0.0; |
164
|
1809 if (type_tag == string_constant) |
1
|
1810 status = 1.0; |
|
1811 tree_constant retval (status); |
|
1812 return retval; |
|
1813 } |
|
1814 |
|
1815 tree_constant |
|
1816 tree_constant_rep::convert_to_str (void) |
|
1817 { |
|
1818 tree_constant retval; |
|
1819 |
|
1820 switch (type_tag) |
|
1821 { |
|
1822 case complex_scalar_constant: |
|
1823 case scalar_constant: |
|
1824 { |
|
1825 double d = double_value (); |
|
1826 int i = NINT (d); |
|
1827 // Warn about out of range conversions? |
|
1828 char s[2]; |
|
1829 s[0] = (char) i; |
240
|
1830 s[1] = '\0'; |
1
|
1831 retval = tree_constant (s); |
|
1832 } |
|
1833 break; |
|
1834 case complex_matrix_constant: |
|
1835 case matrix_constant: |
|
1836 { |
|
1837 ColumnVector v = to_vector (); |
|
1838 int len = v.length (); |
|
1839 if (len == 0) |
240
|
1840 ::error ("can only convert vectors and scalars to strings"); |
1
|
1841 else |
|
1842 { |
|
1843 char *s = new char [len+1]; |
|
1844 s[len] = '\0'; |
|
1845 for (int i = 0; i < len; i++) |
|
1846 { |
|
1847 double d = v.elem (i); |
|
1848 int ival = NINT (d); |
|
1849 // Warn about out of range conversions? |
|
1850 s[i] = (char) ival; |
|
1851 } |
|
1852 retval = tree_constant (s); |
|
1853 delete [] s; |
|
1854 } |
|
1855 } |
|
1856 break; |
|
1857 case range_constant: |
|
1858 { |
|
1859 Range r = range_value (); |
|
1860 double b = r.base (); |
|
1861 double incr = r.inc (); |
|
1862 int nel = r.nelem (); |
|
1863 char *s = new char [nel+1]; |
|
1864 s[nel] = '\0'; |
|
1865 for (int i = 0; i < nel; i++) |
|
1866 { |
|
1867 double d = b + i * incr; |
|
1868 int ival = NINT (d); |
|
1869 // Warn about out of range conversions? |
|
1870 s[i] = (char) ival; |
|
1871 } |
|
1872 retval = tree_constant (s); |
|
1873 delete [] s; |
|
1874 } |
|
1875 break; |
|
1876 case string_constant: |
|
1877 retval = tree_constant (*this); |
|
1878 break; |
|
1879 case magic_colon: |
|
1880 default: |
|
1881 panic_impossible (); |
|
1882 break; |
|
1883 } |
|
1884 return retval; |
|
1885 } |
|
1886 |
|
1887 tree_constant |
164
|
1888 tree_constant_rep::cumprod (void) const |
1
|
1889 { |
|
1890 if (type_tag == string_constant || type_tag == range_constant) |
|
1891 { |
|
1892 tree_constant tmp = make_numeric (); |
|
1893 return tmp.cumprod (); |
|
1894 } |
|
1895 |
|
1896 tree_constant retval; |
|
1897 switch (type_tag) |
|
1898 { |
|
1899 case scalar_constant: |
|
1900 retval = tree_constant (scalar); |
|
1901 break; |
|
1902 case matrix_constant: |
|
1903 { |
|
1904 Matrix m = matrix->cumprod (); |
|
1905 retval = tree_constant (m); |
|
1906 } |
|
1907 break; |
|
1908 case complex_scalar_constant: |
|
1909 retval = tree_constant (*complex_scalar); |
|
1910 break; |
|
1911 case complex_matrix_constant: |
|
1912 { |
|
1913 ComplexMatrix m = complex_matrix->cumprod (); |
|
1914 retval = tree_constant (m); |
|
1915 } |
|
1916 break; |
|
1917 case string_constant: |
|
1918 case range_constant: |
|
1919 case magic_colon: |
|
1920 default: |
|
1921 panic_impossible (); |
|
1922 break; |
|
1923 } |
|
1924 return retval; |
|
1925 } |
|
1926 |
|
1927 tree_constant |
164
|
1928 tree_constant_rep::cumsum (void) const |
1
|
1929 { |
|
1930 if (type_tag == string_constant || type_tag == range_constant) |
|
1931 { |
|
1932 tree_constant tmp = make_numeric (); |
|
1933 return tmp.cumsum (); |
|
1934 } |
|
1935 |
|
1936 tree_constant retval; |
|
1937 switch (type_tag) |
|
1938 { |
|
1939 case scalar_constant: |
|
1940 retval = tree_constant (scalar); |
|
1941 break; |
|
1942 case matrix_constant: |
|
1943 { |
|
1944 Matrix m = matrix->cumsum (); |
|
1945 retval = tree_constant (m); |
|
1946 } |
|
1947 break; |
|
1948 case complex_scalar_constant: |
|
1949 retval = tree_constant (*complex_scalar); |
|
1950 break; |
|
1951 case complex_matrix_constant: |
|
1952 { |
|
1953 ComplexMatrix m = complex_matrix->cumsum (); |
|
1954 retval = tree_constant (m); |
|
1955 } |
|
1956 break; |
|
1957 case string_constant: |
|
1958 case range_constant: |
|
1959 case magic_colon: |
|
1960 default: |
|
1961 panic_impossible (); |
|
1962 break; |
|
1963 } |
|
1964 return retval; |
|
1965 } |
|
1966 |
|
1967 tree_constant |
164
|
1968 tree_constant_rep::prod (void) const |
1
|
1969 { |
|
1970 if (type_tag == string_constant || type_tag == range_constant) |
|
1971 { |
|
1972 tree_constant tmp = make_numeric (); |
|
1973 return tmp.prod (); |
|
1974 } |
|
1975 |
|
1976 tree_constant retval; |
|
1977 switch (type_tag) |
|
1978 { |
|
1979 case scalar_constant: |
|
1980 retval = tree_constant (scalar); |
|
1981 break; |
|
1982 case matrix_constant: |
|
1983 { |
|
1984 Matrix m = matrix->prod (); |
|
1985 retval = tree_constant (m); |
|
1986 } |
|
1987 break; |
|
1988 case complex_scalar_constant: |
|
1989 retval = tree_constant (*complex_scalar); |
|
1990 break; |
|
1991 case complex_matrix_constant: |
|
1992 { |
|
1993 ComplexMatrix m = complex_matrix->prod (); |
|
1994 retval = tree_constant (m); |
|
1995 } |
|
1996 break; |
|
1997 case string_constant: |
|
1998 case range_constant: |
|
1999 case magic_colon: |
|
2000 default: |
|
2001 panic_impossible (); |
|
2002 break; |
|
2003 } |
|
2004 return retval; |
|
2005 } |
|
2006 |
|
2007 tree_constant |
164
|
2008 tree_constant_rep::sum (void) const |
1
|
2009 { |
|
2010 if (type_tag == string_constant || type_tag == range_constant) |
|
2011 { |
|
2012 tree_constant tmp = make_numeric (); |
|
2013 return tmp.sum (); |
|
2014 } |
|
2015 |
|
2016 tree_constant retval; |
|
2017 switch (type_tag) |
|
2018 { |
|
2019 case scalar_constant: |
|
2020 retval = tree_constant (scalar); |
|
2021 break; |
|
2022 case matrix_constant: |
|
2023 { |
|
2024 Matrix m = matrix->sum (); |
|
2025 retval = tree_constant (m); |
|
2026 } |
|
2027 break; |
|
2028 case complex_scalar_constant: |
|
2029 retval = tree_constant (*complex_scalar); |
|
2030 break; |
|
2031 case complex_matrix_constant: |
|
2032 { |
|
2033 ComplexMatrix m = complex_matrix->sum (); |
|
2034 retval = tree_constant (m); |
|
2035 } |
|
2036 break; |
|
2037 case string_constant: |
|
2038 case range_constant: |
|
2039 case magic_colon: |
|
2040 default: |
|
2041 panic_impossible (); |
|
2042 break; |
|
2043 } |
|
2044 return retval; |
|
2045 } |
|
2046 |
|
2047 tree_constant |
164
|
2048 tree_constant_rep::sumsq (void) const |
1
|
2049 { |
|
2050 if (type_tag == string_constant || type_tag == range_constant) |
|
2051 { |
|
2052 tree_constant tmp = make_numeric (); |
|
2053 return tmp.sumsq (); |
|
2054 } |
|
2055 |
|
2056 tree_constant retval; |
|
2057 switch (type_tag) |
|
2058 { |
|
2059 case scalar_constant: |
|
2060 retval = tree_constant (scalar * scalar); |
|
2061 break; |
|
2062 case matrix_constant: |
|
2063 { |
|
2064 Matrix m = matrix->sumsq (); |
|
2065 retval = tree_constant (m); |
|
2066 } |
|
2067 break; |
|
2068 case complex_scalar_constant: |
|
2069 { |
|
2070 Complex c (*complex_scalar); |
|
2071 retval = tree_constant (c * c); |
|
2072 } |
|
2073 break; |
|
2074 case complex_matrix_constant: |
|
2075 { |
|
2076 ComplexMatrix m = complex_matrix->sumsq (); |
|
2077 retval = tree_constant (m); |
|
2078 } |
|
2079 break; |
|
2080 case string_constant: |
|
2081 case range_constant: |
|
2082 case magic_colon: |
|
2083 default: |
|
2084 panic_impossible (); |
|
2085 break; |
|
2086 } |
|
2087 return retval; |
|
2088 } |
|
2089 |
|
2090 static tree_constant |
164
|
2091 make_diag (const Matrix& v, int k) |
1
|
2092 { |
|
2093 int nr = v.rows (); |
|
2094 int nc = v.columns (); |
|
2095 assert (nc == 1 || nr == 1); |
|
2096 |
|
2097 tree_constant retval; |
|
2098 |
|
2099 int roff = 0; |
|
2100 int coff = 0; |
|
2101 if (k > 0) |
|
2102 { |
|
2103 roff = 0; |
|
2104 coff = k; |
|
2105 } |
|
2106 else if (k < 0) |
|
2107 { |
|
2108 roff = -k; |
|
2109 coff = 0; |
|
2110 } |
|
2111 |
|
2112 if (nr == 1) |
|
2113 { |
|
2114 int n = nc + ABS (k); |
|
2115 Matrix m (n, n, 0.0); |
|
2116 for (int i = 0; i < nc; i++) |
|
2117 m.elem (i+roff, i+coff) = v.elem (0, i); |
|
2118 retval = tree_constant (m); |
|
2119 } |
|
2120 else |
|
2121 { |
|
2122 int n = nr + ABS (k); |
|
2123 Matrix m (n, n, 0.0); |
|
2124 for (int i = 0; i < nr; i++) |
|
2125 m.elem (i+roff, i+coff) = v.elem (i, 0); |
|
2126 retval = tree_constant (m); |
|
2127 } |
|
2128 |
|
2129 return retval; |
|
2130 } |
|
2131 |
|
2132 static tree_constant |
164
|
2133 make_diag (const ComplexMatrix& v, int k) |
1
|
2134 { |
|
2135 int nr = v.rows (); |
|
2136 int nc = v.columns (); |
|
2137 assert (nc == 1 || nr == 1); |
|
2138 |
|
2139 tree_constant retval; |
|
2140 |
|
2141 int roff = 0; |
|
2142 int coff = 0; |
|
2143 if (k > 0) |
|
2144 { |
|
2145 roff = 0; |
|
2146 coff = k; |
|
2147 } |
|
2148 else if (k < 0) |
|
2149 { |
|
2150 roff = -k; |
|
2151 coff = 0; |
|
2152 } |
|
2153 |
|
2154 if (nr == 1) |
|
2155 { |
|
2156 int n = nc + ABS (k); |
|
2157 ComplexMatrix m (n, n, 0.0); |
|
2158 for (int i = 0; i < nc; i++) |
|
2159 m.elem (i+roff, i+coff) = v.elem (0, i); |
|
2160 retval = tree_constant (m); |
|
2161 } |
|
2162 else |
|
2163 { |
|
2164 int n = nr + ABS (k); |
|
2165 ComplexMatrix m (n, n, 0.0); |
|
2166 for (int i = 0; i < nr; i++) |
|
2167 m.elem (i+roff, i+coff) = v.elem (i, 0); |
|
2168 retval = tree_constant (m); |
|
2169 } |
|
2170 |
|
2171 return retval; |
|
2172 } |
|
2173 |
|
2174 tree_constant |
164
|
2175 tree_constant_rep::diag (void) const |
1
|
2176 { |
|
2177 if (type_tag == string_constant || type_tag == range_constant) |
|
2178 { |
|
2179 tree_constant tmp = make_numeric (); |
|
2180 return tmp.diag (); |
|
2181 } |
|
2182 |
|
2183 tree_constant retval; |
|
2184 switch (type_tag) |
|
2185 { |
|
2186 case scalar_constant: |
|
2187 retval = tree_constant (scalar); |
|
2188 break; |
|
2189 case matrix_constant: |
|
2190 { |
|
2191 int nr = rows (); |
|
2192 int nc = columns (); |
|
2193 if (nr == 1 || nc == 1) |
|
2194 retval = make_diag (matrix_value (), 0); |
|
2195 else |
|
2196 { |
|
2197 ColumnVector v = matrix->diag (); |
|
2198 if (v.capacity () > 0) |
|
2199 retval = tree_constant (v); |
|
2200 } |
|
2201 } |
|
2202 break; |
|
2203 case complex_scalar_constant: |
|
2204 retval = tree_constant (*complex_scalar); |
|
2205 break; |
|
2206 case complex_matrix_constant: |
|
2207 { |
|
2208 int nr = rows (); |
|
2209 int nc = columns (); |
|
2210 if (nr == 1 || nc == 1) |
|
2211 retval = make_diag (complex_matrix_value (), 0); |
|
2212 else |
|
2213 { |
|
2214 ComplexColumnVector v = complex_matrix->diag (); |
|
2215 if (v.capacity () > 0) |
|
2216 retval = tree_constant (v); |
|
2217 } |
|
2218 } |
|
2219 break; |
|
2220 case string_constant: |
|
2221 case range_constant: |
|
2222 case magic_colon: |
|
2223 default: |
|
2224 panic_impossible (); |
|
2225 break; |
|
2226 } |
|
2227 return retval; |
|
2228 } |
|
2229 |
|
2230 tree_constant |
164
|
2231 tree_constant_rep::diag (const tree_constant& a) const |
1
|
2232 { |
|
2233 if (type_tag == string_constant || type_tag == range_constant) |
|
2234 { |
|
2235 tree_constant tmp = make_numeric (); |
|
2236 return tmp.diag (a); |
|
2237 } |
|
2238 |
|
2239 tree_constant tmp_a = a.make_numeric (); |
|
2240 |
|
2241 tree_constant_rep::constant_type a_type = tmp_a.const_type (); |
|
2242 |
|
2243 tree_constant retval; |
|
2244 |
|
2245 switch (type_tag) |
|
2246 { |
|
2247 case scalar_constant: |
|
2248 if (a_type == scalar_constant) |
|
2249 { |
|
2250 int k = NINT (tmp_a.double_value ()); |
|
2251 int n = ABS (k) + 1; |
|
2252 if (k == 0) |
|
2253 retval = tree_constant (scalar); |
|
2254 else if (k > 0) |
|
2255 { |
|
2256 Matrix m (n, n, 0.0); |
|
2257 m.elem (0, k) = scalar; |
|
2258 retval = tree_constant (m); |
|
2259 } |
|
2260 else if (k < 0) |
|
2261 { |
|
2262 Matrix m (n, n, 0.0); |
|
2263 m.elem (-k, 0) = scalar; |
|
2264 retval = tree_constant (m); |
|
2265 } |
|
2266 } |
|
2267 break; |
|
2268 case matrix_constant: |
|
2269 if (a_type == scalar_constant) |
|
2270 { |
|
2271 int k = NINT (tmp_a.double_value ()); |
|
2272 int nr = rows (); |
|
2273 int nc = columns (); |
|
2274 if (nr == 1 || nc == 1) |
|
2275 retval = make_diag (matrix_value (), k); |
|
2276 else |
|
2277 { |
|
2278 ColumnVector d = matrix->diag (k); |
|
2279 retval = tree_constant (d); |
|
2280 } |
|
2281 } |
|
2282 else |
240
|
2283 ::error ("diag: invalid second argument"); |
1
|
2284 |
|
2285 break; |
|
2286 case complex_scalar_constant: |
|
2287 if (a_type == scalar_constant) |
|
2288 { |
|
2289 int k = NINT (tmp_a.double_value ()); |
|
2290 int n = ABS (k) + 1; |
|
2291 if (k == 0) |
|
2292 retval = tree_constant (*complex_scalar); |
|
2293 else if (k > 0) |
|
2294 { |
|
2295 ComplexMatrix m (n, n, 0.0); |
|
2296 m.elem (0, k) = *complex_scalar; |
|
2297 retval = tree_constant (m); |
|
2298 } |
|
2299 else if (k < 0) |
|
2300 { |
|
2301 ComplexMatrix m (n, n, 0.0); |
|
2302 m.elem (-k, 0) = *complex_scalar; |
|
2303 retval = tree_constant (m); |
|
2304 } |
|
2305 } |
|
2306 break; |
|
2307 case complex_matrix_constant: |
|
2308 if (a_type == scalar_constant) |
|
2309 { |
|
2310 int k = NINT (tmp_a.double_value ()); |
|
2311 int nr = rows (); |
|
2312 int nc = columns (); |
|
2313 if (nr == 1 || nc == 1) |
|
2314 retval = make_diag (complex_matrix_value (), k); |
|
2315 else |
|
2316 { |
|
2317 ComplexColumnVector d = complex_matrix->diag (k); |
|
2318 retval = tree_constant (d); |
|
2319 } |
|
2320 } |
|
2321 else |
240
|
2322 ::error ("diag: invalid second argument"); |
1
|
2323 |
|
2324 break; |
|
2325 case string_constant: |
|
2326 case range_constant: |
|
2327 case magic_colon: |
|
2328 default: |
|
2329 panic_impossible (); |
|
2330 break; |
|
2331 } |
|
2332 return retval; |
|
2333 } |
|
2334 |
|
2335 tree_constant |
164
|
2336 tree_constant_rep::mapper (Mapper_fcn& m_fcn, int print) const |
1
|
2337 { |
|
2338 tree_constant retval; |
|
2339 |
|
2340 if (type_tag == string_constant || type_tag == range_constant) |
|
2341 { |
|
2342 tree_constant tmp = make_numeric (); |
|
2343 return tmp.mapper (m_fcn, print); |
|
2344 } |
|
2345 |
|
2346 switch (type_tag) |
|
2347 { |
|
2348 case scalar_constant: |
|
2349 if (m_fcn.neg_arg_complex && scalar < 0.0) |
|
2350 { |
|
2351 if (m_fcn.c_c_mapper != NULL) |
|
2352 { |
|
2353 Complex c = m_fcn.c_c_mapper (Complex (scalar)); |
|
2354 retval = tree_constant (c); |
|
2355 } |
|
2356 else |
|
2357 panic_impossible (); |
|
2358 } |
|
2359 else |
|
2360 { |
|
2361 if (m_fcn.d_d_mapper != NULL) |
|
2362 { |
|
2363 double d = m_fcn.d_d_mapper (scalar); |
|
2364 retval = tree_constant (d); |
|
2365 } |
|
2366 else |
|
2367 panic_impossible (); |
|
2368 } |
|
2369 break; |
|
2370 case matrix_constant: |
|
2371 if (m_fcn.neg_arg_complex && any_element_is_negative (*matrix)) |
|
2372 { |
|
2373 if (m_fcn.c_c_mapper != NULL) |
|
2374 { |
|
2375 ComplexMatrix cm = map (m_fcn.c_c_mapper, |
|
2376 ComplexMatrix (*matrix)); |
|
2377 retval = tree_constant (cm); |
|
2378 } |
|
2379 else |
|
2380 panic_impossible (); |
|
2381 } |
|
2382 else |
|
2383 { |
|
2384 if (m_fcn.d_d_mapper != NULL) |
|
2385 { |
|
2386 Matrix m = map (m_fcn.d_d_mapper, *matrix); |
|
2387 retval = tree_constant (m); |
|
2388 } |
|
2389 else |
|
2390 panic_impossible (); |
|
2391 } |
|
2392 break; |
|
2393 case complex_scalar_constant: |
|
2394 if (m_fcn.d_c_mapper != NULL) |
|
2395 { |
|
2396 double d; |
|
2397 d = m_fcn.d_c_mapper (*complex_scalar); |
|
2398 retval = tree_constant (d); |
|
2399 } |
|
2400 else if (m_fcn.c_c_mapper != NULL) |
|
2401 { |
|
2402 Complex c; |
|
2403 c = m_fcn.c_c_mapper (*complex_scalar); |
|
2404 retval = tree_constant (c); |
|
2405 } |
|
2406 else |
|
2407 panic_impossible (); |
|
2408 break; |
|
2409 case complex_matrix_constant: |
|
2410 if (m_fcn.d_c_mapper != NULL) |
|
2411 { |
|
2412 Matrix m; |
|
2413 m = map (m_fcn.d_c_mapper, *complex_matrix); |
|
2414 retval = tree_constant (m); |
|
2415 } |
|
2416 else if (m_fcn.c_c_mapper != NULL) |
|
2417 { |
|
2418 ComplexMatrix cm; |
|
2419 cm = map (m_fcn.c_c_mapper, *complex_matrix); |
|
2420 retval = tree_constant (cm); |
|
2421 } |
|
2422 else |
|
2423 panic_impossible (); |
|
2424 break; |
|
2425 case string_constant: |
|
2426 case range_constant: |
|
2427 case magic_colon: |
|
2428 default: |
|
2429 panic_impossible (); |
|
2430 break; |
|
2431 } |
|
2432 |
|
2433 if (retval.is_defined ()) |
|
2434 return retval.eval (print); |
|
2435 else |
|
2436 return retval; |
|
2437 } |
|
2438 |
|
2439 tree_constant::~tree_constant (void) |
|
2440 { |
|
2441 #if defined (MDEBUG) |
|
2442 cerr << "~tree_constant: rep: " << rep |
|
2443 << " rep->count: " << rep->count << "\n"; |
|
2444 #endif |
|
2445 |
|
2446 if (--rep->count <= 0) |
|
2447 { |
|
2448 delete rep; |
|
2449 rep = (tree_constant_rep *) NULL; |
|
2450 } |
|
2451 } |
|
2452 |
259
|
2453 tree_constant |
|
2454 tree_constant::eval (int argc, char **argv, int print) |
|
2455 { |
|
2456 ::error ("first element of word-list command is a constant"); |
|
2457 return tree_constant (); |
|
2458 } |
|
2459 |
1
|
2460 #if defined (MDEBUG) |
|
2461 void * |
|
2462 tree_constant::operator new (size_t size) |
|
2463 { |
|
2464 tree_constant *p = ::new tree_constant; |
|
2465 cerr << "tree_constant::new(): " << p << "\n"; |
|
2466 return p; |
|
2467 } |
|
2468 |
|
2469 void |
|
2470 tree_constant::operator delete (void *p, size_t size) |
|
2471 { |
|
2472 cerr << "tree_constant::delete(): " << p << "\n"; |
|
2473 ::delete p; |
|
2474 } |
|
2475 #endif |
|
2476 |
|
2477 /* |
96
|
2478 * Construct return vector of empty matrices. Return empty matrices |
|
2479 * and/or gripe when appropriate. |
|
2480 */ |
|
2481 tree_constant * |
164
|
2482 vector_of_empties (int nargout, const char *fcn_name) |
96
|
2483 { |
|
2484 tree_constant *retval = NULL_TREE_CONST; |
|
2485 |
|
2486 // Got an empty argument, check if should gripe/return empty values. |
|
2487 |
|
2488 int flag = user_pref.propagate_empty_matrices; |
|
2489 if (flag != 0) |
|
2490 { |
|
2491 if (flag < 0) |
|
2492 gripe_empty_arg (fcn_name, 0); |
|
2493 |
|
2494 Matrix m; |
|
2495 retval = new tree_constant [nargout+1]; |
|
2496 for (int i = 0; i < nargout; i++) |
|
2497 retval[i] = tree_constant (m); |
|
2498 } |
|
2499 else |
|
2500 gripe_empty_arg (fcn_name, 1); |
|
2501 |
|
2502 return retval; |
|
2503 } |
|
2504 |
|
2505 /* |
1
|
2506 ;;; Local Variables: *** |
|
2507 ;;; mode: C++ *** |
|
2508 ;;; page-delimiter: "^/\\*" *** |
|
2509 ;;; End: *** |
|
2510 */ |