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 |
|
24 #ifdef __GNUG__ |
|
25 #pragma implementation |
|
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 { |
|
396 if (nel == -1) |
|
397 error ("number of elements in range exceeds INT_MAX"); |
|
398 else |
|
399 error ("invalid range"); |
|
400 |
|
401 jump_to_top_level (); |
|
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) |
|
585 message ((char *) NULL, |
|
586 "row index = %d exceeds max row dimension = %d", i, nr); |
|
587 if (j > nc) |
|
588 message ((char *) NULL, |
|
589 "column index = %d exceeds max column dimension = %d", |
|
590 j, nc); |
|
591 |
|
592 jump_to_top_level (); |
|
593 } |
|
594 } |
|
595 } |
|
596 |
|
597 void |
|
598 tree_constant_rep::maybe_resize (int i, force_orient f_orient = no_orient) |
|
599 { |
|
600 int nr = rows (); |
|
601 int nc = columns (); |
|
602 |
|
603 i++; |
|
604 |
|
605 assert (i > 0 && (nr <= 1 || nc <= 1)); |
|
606 |
|
607 if (nr <= 1 && nc <= 1 && i >= 1) |
|
608 { |
|
609 if (user_pref.resize_on_range_error) |
|
610 { |
|
611 if (f_orient == row_orient) |
|
612 resize (1, i, 0.0); |
|
613 else if (f_orient == column_orient) |
|
614 resize (i, 1, 0.0); |
|
615 else if (user_pref.prefer_column_vectors) |
|
616 resize (i, 1, 0.0); |
|
617 else |
|
618 resize (1, i, 0.0); |
|
619 } |
|
620 else |
|
621 { |
|
622 message ((char *) NULL, |
|
623 "matrix index = %d exceeds max dimension = %d", i, nc); |
|
624 jump_to_top_level (); |
|
625 } |
|
626 } |
|
627 else if (nr == 1 && i > nc) |
|
628 { |
|
629 if (user_pref.resize_on_range_error) |
|
630 resize (1, i, 0.0); |
|
631 else |
|
632 { |
|
633 message ((char *) NULL, |
|
634 "matrix index = %d exceeds max dimension = %d", i, nc); |
|
635 jump_to_top_level (); |
|
636 } |
|
637 } |
|
638 else if (nc == 1 && i > nr) |
|
639 { |
|
640 if (user_pref.resize_on_range_error) |
|
641 resize (i, 1, 0.0); |
|
642 else |
|
643 { |
|
644 message ((char *) NULL, |
|
645 "matrix index = %d exceeds max dimension = ", i, nc); |
|
646 jump_to_top_level (); |
|
647 } |
|
648 } |
|
649 } |
|
650 |
|
651 double |
164
|
652 tree_constant_rep::to_scalar (void) const |
1
|
653 { |
|
654 tree_constant tmp = make_numeric (); |
|
655 |
|
656 double retval = 0.0; |
|
657 |
|
658 switch (tmp.const_type ()) |
|
659 { |
|
660 case tree_constant_rep::scalar_constant: |
|
661 case tree_constant_rep::complex_scalar_constant: |
|
662 retval = tmp.double_value (); |
|
663 break; |
|
664 case tree_constant_rep::matrix_constant: |
|
665 if (user_pref.do_fortran_indexing) |
|
666 { |
|
667 Matrix m = tmp.matrix_value (); |
|
668 retval = m (0, 0); |
|
669 } |
|
670 break; |
|
671 case tree_constant_rep::complex_matrix_constant: |
|
672 if (user_pref.do_fortran_indexing) |
|
673 { |
|
674 int flag = user_pref.ok_to_lose_imaginary_part; |
|
675 if (flag == -1) |
|
676 warning ("implicit conversion of complex value to real value"); |
|
677 |
|
678 if (flag != 0) |
|
679 { |
|
680 ComplexMatrix m = tmp.complex_matrix_value (); |
|
681 return real (m (0, 0)); |
|
682 } |
|
683 else |
|
684 jump_to_top_level (); |
|
685 } |
|
686 else |
|
687 { |
|
688 error ("complex matrix used in invalid context"); |
|
689 jump_to_top_level (); |
|
690 } |
|
691 break; |
|
692 default: |
|
693 break; |
|
694 } |
|
695 return retval; |
|
696 } |
|
697 |
|
698 ColumnVector |
164
|
699 tree_constant_rep::to_vector (void) const |
1
|
700 { |
|
701 tree_constant tmp = make_numeric (); |
|
702 |
|
703 ColumnVector retval; |
|
704 |
|
705 switch (tmp.const_type ()) |
|
706 { |
|
707 case tree_constant_rep::scalar_constant: |
|
708 case tree_constant_rep::complex_scalar_constant: |
|
709 retval.resize (1); |
|
710 retval.elem (0) = tmp.double_value (); |
|
711 break; |
|
712 case tree_constant_rep::complex_matrix_constant: |
|
713 case tree_constant_rep::matrix_constant: |
|
714 { |
|
715 Matrix m = tmp.matrix_value (); |
|
716 int nr = m.rows (); |
|
717 int nc = m.columns (); |
|
718 if (nr == 1) |
|
719 { |
|
720 retval.resize (nc); |
|
721 for (int i = 0; i < nc; i++) |
|
722 retval.elem (i) = m (0, i); |
|
723 } |
|
724 else if (nc == 1) |
|
725 { |
|
726 retval.resize (nr); |
|
727 for (int i = 0; i < nr; i++) |
|
728 retval.elem (i) = m.elem (i, 0); |
|
729 } |
|
730 } |
|
731 break; |
|
732 default: |
|
733 panic_impossible (); |
|
734 break; |
|
735 } |
|
736 return retval; |
|
737 } |
|
738 |
|
739 Matrix |
164
|
740 tree_constant_rep::to_matrix (void) const |
1
|
741 { |
|
742 tree_constant tmp = make_numeric (); |
|
743 |
|
744 Matrix retval; |
|
745 |
|
746 switch (tmp.const_type ()) |
|
747 { |
|
748 case tree_constant_rep::scalar_constant: |
|
749 retval.resize (1, 1); |
|
750 retval.elem (0, 0) = tmp.double_value (); |
|
751 break; |
|
752 case tree_constant_rep::matrix_constant: |
|
753 retval = tmp.matrix_value (); |
|
754 break; |
|
755 default: |
|
756 break; |
|
757 } |
|
758 return retval; |
|
759 } |
|
760 |
|
761 tree_constant_rep::constant_type |
|
762 tree_constant_rep::force_numeric (int force_str_conv = 0) |
|
763 { |
|
764 switch (type_tag) |
|
765 { |
|
766 case scalar_constant: |
|
767 case matrix_constant: |
|
768 case complex_scalar_constant: |
|
769 case complex_matrix_constant: |
|
770 break; |
|
771 case string_constant: |
|
772 { |
|
773 if (! force_str_conv && ! user_pref.implicit_str_to_num_ok) |
|
774 { |
|
775 error ("failed to convert `%s' to a numeric type -- default\ |
|
776 conversion turned off", string); |
|
777 // Abort! |
|
778 jump_to_top_level (); |
|
779 } |
|
780 |
|
781 int len = strlen (string); |
|
782 if (len > 1) |
|
783 { |
|
784 type_tag = matrix_constant; |
|
785 Matrix *tm = new Matrix (1, len); |
|
786 for (int i = 0; i < len; i++) |
|
787 tm->elem (0, i) = toascii ((int) string[i]); |
|
788 matrix = tm; |
|
789 } |
|
790 else if (len == 1) |
|
791 { |
|
792 type_tag = scalar_constant; |
|
793 scalar = toascii ((int) string[0]); |
|
794 } |
|
795 } |
|
796 break; |
|
797 case range_constant: |
|
798 { |
|
799 int len = range->nelem (); |
|
800 if (len > 1) |
|
801 { |
|
802 type_tag = matrix_constant; |
|
803 Matrix *tm = new Matrix (1, len); |
|
804 double b = range->base (); |
|
805 double increment = range->inc (); |
|
806 for (int i = 0; i < len; i++) |
|
807 tm->elem (0, i) = b + i * increment; |
|
808 matrix = tm; |
|
809 } |
|
810 else if (len == 1) |
|
811 { |
|
812 type_tag = scalar_constant; |
|
813 scalar = range->base (); |
|
814 } |
|
815 } |
|
816 break; |
|
817 case magic_colon: |
|
818 default: |
|
819 panic_impossible (); |
|
820 break; |
|
821 } |
|
822 return type_tag; |
|
823 } |
|
824 |
|
825 tree_constant |
164
|
826 tree_constant_rep::make_numeric (int force_str_conv = 0) const |
1
|
827 { |
|
828 tree_constant retval; |
|
829 switch (type_tag) |
|
830 { |
|
831 case scalar_constant: |
|
832 retval = tree_constant (scalar); |
|
833 break; |
|
834 case matrix_constant: |
|
835 retval = tree_constant (*matrix); |
|
836 break; |
|
837 case complex_scalar_constant: |
|
838 retval = tree_constant (*complex_scalar); |
|
839 break; |
|
840 case complex_matrix_constant: |
|
841 retval = tree_constant (*complex_matrix); |
|
842 break; |
|
843 case string_constant: |
|
844 retval = tree_constant (string); |
|
845 retval.force_numeric (force_str_conv); |
|
846 break; |
|
847 case range_constant: |
|
848 retval = tree_constant (*range); |
|
849 retval.force_numeric (force_str_conv); |
|
850 break; |
|
851 case magic_colon: |
|
852 default: |
|
853 panic_impossible (); |
|
854 break; |
|
855 } |
|
856 return retval; |
|
857 } |
|
858 |
|
859 tree_constant |
|
860 do_binary_op (tree_constant& a, tree_constant& b, tree::expression_type t) |
|
861 { |
143
|
862 tree_constant ans; |
|
863 |
1
|
864 int first_empty = (a.rows () == 0 || a.columns () == 0); |
|
865 int second_empty = (b.rows () == 0 || b.columns () == 0); |
|
866 |
|
867 if (first_empty || second_empty) |
|
868 { |
|
869 int flag = user_pref.propagate_empty_matrices; |
|
870 if (flag < 0) |
|
871 warning ("binary operation on empty matrix"); |
|
872 else if (flag == 0) |
|
873 { |
|
874 error ("invalid binary operation on empty matrix"); |
143
|
875 return ans; |
1
|
876 } |
|
877 } |
|
878 |
|
879 tree_constant tmp_a = a.make_numeric (); |
|
880 tree_constant tmp_b = b.make_numeric (); |
|
881 |
|
882 tree_constant_rep::constant_type a_type = tmp_a.const_type (); |
|
883 tree_constant_rep::constant_type b_type = tmp_b.const_type (); |
|
884 |
|
885 double d1, d2; |
|
886 Matrix m1, m2; |
|
887 Complex c1, c2; |
|
888 ComplexMatrix cm1, cm2; |
|
889 |
|
890 switch (a_type) |
|
891 { |
|
892 case tree_constant_rep::scalar_constant: |
|
893 d1 = tmp_a.double_value (); |
|
894 switch (b_type) |
|
895 { |
|
896 case tree_constant_rep::scalar_constant: |
|
897 d2 = tmp_b.double_value (); |
|
898 ans = do_binary_op (d1, d2, t); |
|
899 break; |
|
900 case tree_constant_rep::matrix_constant: |
|
901 m2 = tmp_b.matrix_value (); |
|
902 ans = do_binary_op (d1, m2, t); |
|
903 break; |
|
904 case tree_constant_rep::complex_scalar_constant: |
|
905 c2 = tmp_b.complex_value (); |
|
906 ans = do_binary_op (d1, c2, t); |
|
907 break; |
|
908 case tree_constant_rep::complex_matrix_constant: |
|
909 cm2 = tmp_b.complex_matrix_value (); |
|
910 ans = do_binary_op (d1, cm2, t); |
|
911 break; |
|
912 case tree_constant_rep::magic_colon: |
|
913 default: |
|
914 panic_impossible (); |
|
915 break; |
|
916 } |
|
917 break; |
|
918 case tree_constant_rep::matrix_constant: |
|
919 m1 = tmp_a.matrix_value (); |
|
920 switch (b_type) |
|
921 { |
|
922 case tree_constant_rep::scalar_constant: |
|
923 d2 = tmp_b.double_value (); |
|
924 ans = do_binary_op (m1, d2, t); |
|
925 break; |
|
926 case tree_constant_rep::matrix_constant: |
|
927 m2 = tmp_b.matrix_value (); |
|
928 ans = do_binary_op (m1, m2, t); |
|
929 break; |
|
930 case tree_constant_rep::complex_scalar_constant: |
|
931 c2 = tmp_b.complex_value (); |
|
932 ans = do_binary_op (m1, c2, t); |
|
933 break; |
|
934 case tree_constant_rep::complex_matrix_constant: |
|
935 cm2 = tmp_b.complex_matrix_value (); |
|
936 ans = do_binary_op (m1, cm2, t); |
|
937 break; |
|
938 case tree_constant_rep::magic_colon: |
|
939 default: |
|
940 panic_impossible (); |
|
941 break; |
|
942 } |
|
943 break; |
|
944 case tree_constant_rep::complex_scalar_constant: |
|
945 c1 = tmp_a.complex_value (); |
|
946 switch (b_type) |
|
947 { |
|
948 case tree_constant_rep::scalar_constant: |
|
949 d2 = tmp_b.double_value (); |
|
950 ans = do_binary_op (c1, d2, t); |
|
951 break; |
|
952 case tree_constant_rep::matrix_constant: |
|
953 m2 = tmp_b.matrix_value (); |
|
954 ans = do_binary_op (c1, m2, t); |
|
955 break; |
|
956 case tree_constant_rep::complex_scalar_constant: |
|
957 c2 = tmp_b.complex_value (); |
|
958 ans = do_binary_op (c1, c2, t); |
|
959 break; |
|
960 case tree_constant_rep::complex_matrix_constant: |
|
961 cm2 = tmp_b.complex_matrix_value (); |
|
962 ans = do_binary_op (c1, cm2, t); |
|
963 break; |
|
964 case tree_constant_rep::magic_colon: |
|
965 default: |
|
966 panic_impossible (); |
|
967 break; |
|
968 } |
|
969 break; |
|
970 case tree_constant_rep::complex_matrix_constant: |
|
971 cm1 = tmp_a.complex_matrix_value (); |
|
972 switch (b_type) |
|
973 { |
|
974 case tree_constant_rep::scalar_constant: |
|
975 d2 = tmp_b.double_value (); |
|
976 ans = do_binary_op (cm1, d2, t); |
|
977 break; |
|
978 case tree_constant_rep::matrix_constant: |
|
979 m2 = tmp_b.matrix_value (); |
|
980 ans = do_binary_op (cm1, m2, t); |
|
981 break; |
|
982 case tree_constant_rep::complex_scalar_constant: |
|
983 c2 = tmp_b.complex_value (); |
|
984 ans = do_binary_op (cm1, c2, t); |
|
985 break; |
|
986 case tree_constant_rep::complex_matrix_constant: |
|
987 cm2 = tmp_b.complex_matrix_value (); |
|
988 ans = do_binary_op (cm1, cm2, t); |
|
989 break; |
|
990 case tree_constant_rep::magic_colon: |
|
991 default: |
|
992 panic_impossible (); |
|
993 break; |
|
994 } |
|
995 break; |
|
996 case tree_constant_rep::magic_colon: |
|
997 default: |
|
998 panic_impossible (); |
|
999 break; |
|
1000 } |
|
1001 return ans; |
|
1002 } |
|
1003 |
|
1004 tree_constant |
|
1005 do_unary_op (tree_constant& a, tree::expression_type t) |
|
1006 { |
143
|
1007 tree_constant ans; |
|
1008 |
1
|
1009 if (a.rows () == 0 || a.columns () == 0) |
|
1010 { |
|
1011 int flag = user_pref.propagate_empty_matrices; |
|
1012 if (flag < 0) |
|
1013 warning ("unary operation on empty matrix"); |
|
1014 else if (flag == 0) |
|
1015 { |
|
1016 error ("invalid unary operation on empty matrix"); |
143
|
1017 return ans; |
1
|
1018 } |
|
1019 } |
|
1020 |
|
1021 tree_constant tmp_a = a.make_numeric (); |
|
1022 |
|
1023 switch (tmp_a.const_type ()) |
|
1024 { |
|
1025 case tree_constant_rep::scalar_constant: |
|
1026 ans = do_unary_op (tmp_a.double_value (), t); |
|
1027 break; |
|
1028 case tree_constant_rep::matrix_constant: |
|
1029 { |
|
1030 Matrix m = tmp_a.matrix_value (); |
|
1031 ans = do_unary_op (m, t); |
|
1032 } |
|
1033 break; |
|
1034 case tree_constant_rep::complex_scalar_constant: |
|
1035 ans = do_unary_op (tmp_a.complex_value (), t); |
|
1036 break; |
|
1037 case tree_constant_rep::complex_matrix_constant: |
|
1038 { |
|
1039 ComplexMatrix m = tmp_a.complex_matrix_value (); |
|
1040 ans = do_unary_op (m, t); |
|
1041 } |
|
1042 break; |
|
1043 case tree_constant_rep::magic_colon: |
|
1044 default: |
|
1045 panic_impossible (); |
|
1046 break; |
|
1047 } |
|
1048 return ans; |
|
1049 } |
|
1050 |
|
1051 void |
|
1052 tree_constant_rep::bump_value (tree::expression_type etype) |
|
1053 { |
|
1054 switch (etype) |
|
1055 { |
|
1056 case tree::increment: |
|
1057 switch (type_tag) |
|
1058 { |
|
1059 case scalar_constant: |
|
1060 scalar++; |
|
1061 break; |
|
1062 case matrix_constant: |
|
1063 *matrix = *matrix + 1.0; |
|
1064 break; |
|
1065 case complex_scalar_constant: |
|
1066 *complex_scalar = *complex_scalar + 1.0; |
|
1067 break; |
|
1068 case complex_matrix_constant: |
|
1069 *complex_matrix = *complex_matrix + 1.0; |
|
1070 break; |
|
1071 case string_constant: |
|
1072 error ("string++ and ++string not implemented yet, ok?"); |
|
1073 break; |
|
1074 case range_constant: |
|
1075 range->set_base (range->base () + 1.0); |
|
1076 range->set_limit (range->limit () + 1.0); |
|
1077 break; |
|
1078 case magic_colon: |
|
1079 default: |
|
1080 panic_impossible (); |
|
1081 break; |
|
1082 } |
|
1083 break; |
|
1084 case tree::decrement: |
|
1085 switch (type_tag) |
|
1086 { |
|
1087 case scalar_constant: |
|
1088 scalar--; |
|
1089 break; |
|
1090 case matrix_constant: |
|
1091 *matrix = *matrix - 1.0; |
|
1092 break; |
|
1093 case string_constant: |
|
1094 error ("string-- and -- string not implemented yet, ok?"); |
|
1095 break; |
|
1096 case range_constant: |
|
1097 range->set_base (range->base () - 1.0); |
|
1098 range->set_limit (range->limit () - 1.0); |
|
1099 break; |
|
1100 case magic_colon: |
|
1101 default: |
|
1102 panic_impossible (); |
|
1103 break; |
|
1104 } |
|
1105 break; |
|
1106 default: |
|
1107 panic_impossible (); |
|
1108 break; |
|
1109 } |
|
1110 } |
|
1111 |
|
1112 void |
|
1113 tree_constant_rep::eval (int print) |
|
1114 { |
143
|
1115 if (error_state) |
|
1116 return; |
|
1117 |
1
|
1118 switch (type_tag) |
|
1119 { |
|
1120 case complex_scalar_constant: |
|
1121 if (imag (*complex_scalar) == 0.0) |
|
1122 { |
|
1123 double d = real (*complex_scalar); |
|
1124 delete complex_scalar; |
|
1125 scalar = d; |
|
1126 type_tag = scalar_constant; |
|
1127 } |
|
1128 break; |
|
1129 case complex_matrix_constant: |
|
1130 if (! any_element_is_complex (*complex_matrix)) |
|
1131 { |
|
1132 Matrix *m = new Matrix (real (*complex_matrix)); |
|
1133 delete complex_matrix; |
|
1134 matrix = m; |
|
1135 type_tag = matrix_constant; |
|
1136 } |
|
1137 break; |
|
1138 case scalar_constant: |
|
1139 case matrix_constant: |
|
1140 case string_constant: |
|
1141 case range_constant: |
|
1142 case magic_colon: |
|
1143 break; |
|
1144 default: |
|
1145 panic_impossible (); |
|
1146 break; |
|
1147 } |
|
1148 |
145
|
1149 // Avoid calling rows() and columns() for things like magic_colon. |
143
|
1150 |
145
|
1151 int nr = 1; |
|
1152 int nc = 1; |
|
1153 if (type_tag == matrix_constant |
|
1154 || type_tag == complex_matrix_constant |
|
1155 || type_tag == range_constant) |
143
|
1156 { |
145
|
1157 nr = rows (); |
|
1158 nc = columns (); |
|
1159 } |
|
1160 |
|
1161 switch (type_tag) |
|
1162 { |
|
1163 case matrix_constant: |
|
1164 if (nr == 1 && nc == 1) |
143
|
1165 { |
145
|
1166 double d = matrix->elem (0, 0); |
|
1167 delete matrix; |
|
1168 scalar = d; |
|
1169 type_tag = scalar_constant; |
143
|
1170 } |
145
|
1171 break; |
|
1172 case complex_matrix_constant: |
|
1173 if (nr == 1 && nc == 1) |
|
1174 { |
|
1175 Complex c = complex_matrix->elem (0, 0); |
|
1176 delete complex_matrix; |
|
1177 complex_scalar = new Complex (c); |
|
1178 type_tag = complex_scalar_constant; |
|
1179 } |
|
1180 break; |
|
1181 case range_constant: |
|
1182 if (nr == 1 && nc == 1) |
|
1183 { |
|
1184 double d = range->base (); |
|
1185 delete range; |
|
1186 scalar = d; |
|
1187 type_tag = scalar_constant; |
|
1188 } |
|
1189 break; |
|
1190 default: |
|
1191 break; |
143
|
1192 } |
|
1193 |
1
|
1194 if (print) |
|
1195 { |
|
1196 ostrstream output_buf; |
|
1197 switch (type_tag) |
|
1198 { |
|
1199 case scalar_constant: |
|
1200 octave_print_internal (output_buf, scalar); |
|
1201 break; |
|
1202 case matrix_constant: |
|
1203 if (nr == 0 || nc == 0) |
|
1204 { |
|
1205 output_buf << "[]"; |
|
1206 if (user_pref.print_empty_dimensions) |
|
1207 output_buf << "(" << nr << "x" << nc << ")"; |
|
1208 output_buf << "\n"; |
|
1209 } |
|
1210 else |
|
1211 octave_print_internal (output_buf, *matrix); |
|
1212 break; |
|
1213 case complex_scalar_constant: |
|
1214 octave_print_internal (output_buf, *complex_scalar); |
|
1215 break; |
|
1216 case complex_matrix_constant: |
|
1217 if (nr == 0 || nc == 0) |
|
1218 { |
|
1219 output_buf << "[]"; |
|
1220 if (user_pref.print_empty_dimensions) |
|
1221 output_buf << "(" << nr << "x" << nc << ")"; |
|
1222 output_buf << "\n"; |
|
1223 } |
|
1224 else |
|
1225 octave_print_internal (output_buf, *complex_matrix); |
|
1226 break; |
|
1227 case string_constant: |
|
1228 output_buf << string << "\n"; |
|
1229 break; |
|
1230 case range_constant: |
|
1231 octave_print_internal (output_buf, *range); |
|
1232 break; |
|
1233 case magic_colon: |
|
1234 default: |
|
1235 panic_impossible (); |
|
1236 break; |
|
1237 } |
|
1238 |
|
1239 output_buf << ends; |
|
1240 maybe_page_output (output_buf); |
|
1241 } |
|
1242 } |
|
1243 |
|
1244 tree_constant * |
164
|
1245 tree_constant_rep::eval (const tree_constant *args, int nargin, int nargout, |
1
|
1246 int print) |
|
1247 { |
143
|
1248 if (error_state) |
|
1249 return NULL_TREE_CONST; |
|
1250 |
1
|
1251 tree_constant *retval = new tree_constant [2]; |
|
1252 switch (type_tag) |
|
1253 { |
|
1254 case complex_scalar_constant: |
|
1255 case scalar_constant: |
|
1256 retval[0] = do_scalar_index (args, nargin); |
|
1257 break; |
|
1258 case complex_matrix_constant: |
|
1259 case matrix_constant: |
|
1260 retval[0] = do_matrix_index (args, nargin); |
|
1261 break; |
|
1262 case string_constant: |
|
1263 gripe_string_invalid (); |
|
1264 // retval[0] = do_string_index (args, nargin); |
|
1265 break; |
|
1266 case magic_colon: |
|
1267 case range_constant: |
|
1268 // This isn\'t great, but it\'s easier than implementing a lot of |
|
1269 // range indexing functions. |
|
1270 force_numeric (); |
|
1271 assert (type_tag != magic_colon && type_tag != range_constant); |
|
1272 return eval (args, nargin, nargout, print); |
|
1273 break; |
|
1274 default: |
|
1275 panic_impossible (); |
|
1276 break; |
|
1277 } |
|
1278 |
|
1279 if (retval[0].is_defined ()) |
|
1280 retval[0].eval (print); |
|
1281 return retval; |
|
1282 } |
|
1283 |
|
1284 int |
|
1285 tree_constant_rep::save (ostream& os, int mark_as_global) |
|
1286 { |
|
1287 switch (type_tag) |
|
1288 { |
|
1289 case scalar_constant: |
|
1290 case matrix_constant: |
|
1291 case complex_scalar_constant: |
|
1292 case complex_matrix_constant: |
|
1293 case string_constant: |
|
1294 case range_constant: |
|
1295 if (mark_as_global) |
|
1296 os << "# type: global "; |
|
1297 else |
|
1298 os << "# type: "; |
|
1299 break; |
|
1300 case magic_colon: |
|
1301 default: |
|
1302 break; |
|
1303 } |
|
1304 |
|
1305 switch (type_tag) |
|
1306 { |
|
1307 case scalar_constant: |
|
1308 os << "scalar\n" |
|
1309 << scalar << "\n"; |
|
1310 break; |
|
1311 case matrix_constant: |
|
1312 os << "matrix\n" |
|
1313 << "# rows: " << rows () << "\n" |
|
1314 << "# columns: " << columns () << "\n" |
|
1315 << *matrix ; |
|
1316 break; |
|
1317 case complex_scalar_constant: |
|
1318 os << "complex scalar\n" |
|
1319 << *complex_scalar << "\n"; |
|
1320 break; |
|
1321 case complex_matrix_constant: |
|
1322 os << "complex matrix\n" |
|
1323 << "# rows: " << rows () << "\n" |
|
1324 << "# columns: " << columns () << "\n" |
|
1325 << *complex_matrix ; |
|
1326 break; |
|
1327 case string_constant: |
|
1328 os << "string\n" |
|
1329 << "# length: " << strlen (string) << "\n" |
|
1330 << string << "\n"; |
|
1331 break; |
|
1332 case range_constant: |
|
1333 { |
|
1334 os << "range\n" |
|
1335 << "# base, limit, increment\n" |
|
1336 << range->base () << " " |
|
1337 << range->limit () << " " |
|
1338 << range->inc () << "\n"; |
|
1339 } |
|
1340 break; |
|
1341 case magic_colon: |
|
1342 default: |
|
1343 panic_impossible (); |
|
1344 break; |
|
1345 } |
|
1346 // Really want to return 1 only if write is successful. |
|
1347 return 1; |
|
1348 } |
|
1349 |
|
1350 int |
|
1351 tree_constant_rep::save_three_d (ostream& os, int parametric) |
|
1352 { |
|
1353 int nr = rows (); |
|
1354 int nc = columns (); |
|
1355 |
|
1356 switch (type_tag) |
|
1357 { |
|
1358 case matrix_constant: |
|
1359 os << "# 3D data...\n" |
|
1360 << "# type: matrix\n" |
|
1361 << "# total rows: " << nr << "\n" |
|
1362 << "# total columns: " << nc << "\n"; |
|
1363 |
|
1364 if (parametric) |
|
1365 { |
|
1366 int extras = nc % 3; |
|
1367 if (extras) |
|
1368 warning ("ignoring last %d columns", extras); |
|
1369 |
|
1370 for (int i = 0; i < nc-extras; i += 3) |
|
1371 { |
|
1372 os << matrix->extract (0, i, nr-1, i+2); |
|
1373 if (i+3 < nc-extras) |
|
1374 os << "\n"; |
|
1375 } |
|
1376 } |
|
1377 else |
|
1378 { |
|
1379 for (int i = 0; i < nc; i++) |
|
1380 { |
|
1381 os << matrix->extract (0, i, nr-1, i); |
|
1382 if (i+1 < nc) |
|
1383 os << "\n"; |
|
1384 } |
|
1385 } |
|
1386 break; |
|
1387 default: |
|
1388 error ("for now, I can only save real matrices in 3D format"); |
|
1389 return 0; |
|
1390 break; |
|
1391 } |
|
1392 // Really want to return 1 only if write is successful. |
|
1393 return 1; |
|
1394 } |
|
1395 |
|
1396 int |
|
1397 tree_constant_rep::load (istream& is) |
|
1398 { |
|
1399 int is_global = 0; |
|
1400 |
|
1401 type_tag = unknown_constant; |
|
1402 |
|
1403 // Look for type keyword |
|
1404 char tag [128]; |
|
1405 if (extract_keyword (is, "type", tag)) |
|
1406 { |
|
1407 if (tag != (char *) NULL && *tag != '\0') |
|
1408 { |
|
1409 char *ptr = strchr (tag, ' '); |
|
1410 if (ptr != (char *) NULL) |
|
1411 { |
|
1412 *ptr = '\0'; |
|
1413 is_global = (strncmp (tag, "global", 6) == 0); |
|
1414 *ptr = ' '; |
|
1415 ptr++; |
|
1416 } |
|
1417 else |
|
1418 ptr = &tag[0]; |
|
1419 |
|
1420 if (strncmp (ptr, "scalar", 6) == 0) |
|
1421 type_tag = load (is, scalar_constant); |
|
1422 else if (strncmp (ptr, "matrix", 6) == 0) |
|
1423 type_tag = load (is, matrix_constant); |
|
1424 else if (strncmp (ptr, "complex scalar", 14) == 0) |
|
1425 type_tag = load (is, complex_scalar_constant); |
|
1426 else if (strncmp (ptr, "complex matrix", 14) == 0) |
|
1427 type_tag = load (is, complex_matrix_constant); |
|
1428 else if (strncmp (ptr, "string", 6) == 0) |
|
1429 type_tag = load (is, string_constant); |
|
1430 else if (strncmp (ptr, "range", 5) == 0) |
|
1431 type_tag = load (is, range_constant); |
|
1432 else |
|
1433 error ("unknown constant type `%s'", tag); |
|
1434 } |
|
1435 else |
|
1436 error ("failed to extract keyword specifying value type"); |
|
1437 } |
|
1438 |
|
1439 return is_global; |
|
1440 } |
|
1441 |
|
1442 tree_constant_rep::constant_type |
|
1443 tree_constant_rep::load (istream& is, tree_constant_rep::constant_type t) |
|
1444 { |
|
1445 tree_constant_rep::constant_type status = unknown_constant; |
|
1446 |
|
1447 switch (t) |
|
1448 { |
|
1449 case scalar_constant: |
|
1450 is >> scalar; |
|
1451 if (is) |
|
1452 status = scalar_constant; |
|
1453 else |
|
1454 error ("failed to load scalar constant"); |
|
1455 break; |
|
1456 case matrix_constant: |
|
1457 { |
|
1458 int nr = 0, nc = 0; |
|
1459 |
|
1460 if (extract_keyword (is, "rows", nr) && nr > 0 |
|
1461 && extract_keyword (is, "columns", nc) && nc > 0) |
|
1462 { |
|
1463 matrix = new Matrix (nr, nc); |
|
1464 is >> *matrix; |
|
1465 if (is) |
|
1466 status = matrix_constant; |
|
1467 else |
|
1468 error ("failed to load matrix constant"); |
|
1469 } |
|
1470 else |
|
1471 error ("failed to extract number of rows and columns"); |
|
1472 } |
|
1473 break; |
|
1474 case complex_scalar_constant: |
|
1475 is >> *complex_scalar; |
|
1476 if (is) |
|
1477 status = complex_scalar_constant; |
|
1478 else |
|
1479 error ("failed to load complex scalar constant"); |
|
1480 break; |
|
1481 case complex_matrix_constant: |
|
1482 { |
|
1483 int nr = 0, nc = 0; |
|
1484 |
|
1485 if (extract_keyword (is, "rows", nr) && nr > 0 |
|
1486 && extract_keyword (is, "columns", nc) && nc > 0) |
|
1487 { |
|
1488 complex_matrix = new ComplexMatrix (nr, nc); |
|
1489 is >> *complex_matrix; |
|
1490 if (is) |
|
1491 status = complex_matrix_constant; |
|
1492 else |
|
1493 error ("failed to load complex matrix constant"); |
|
1494 } |
|
1495 else |
|
1496 error ("failed to extract number of rows and columns"); |
|
1497 } |
|
1498 break; |
|
1499 case string_constant: |
|
1500 { |
|
1501 int len; |
|
1502 if (extract_keyword (is, "length", len) && len > 0) |
|
1503 { |
|
1504 string = new char [len+1]; |
|
1505 is.get (string, len+1, EOF); |
|
1506 if (is) |
|
1507 status = string_constant; |
|
1508 else |
|
1509 error ("failed to load string constant"); |
|
1510 } |
|
1511 else |
|
1512 error ("failed to extract string length"); |
|
1513 } |
|
1514 break; |
|
1515 case range_constant: |
|
1516 skip_comments (is); |
|
1517 range = new Range (); |
|
1518 is >> *range; |
|
1519 if (is) |
|
1520 status = range_constant; |
|
1521 else |
|
1522 error ("failed to load range constant"); |
|
1523 break; |
|
1524 default: |
|
1525 panic_impossible (); |
|
1526 break; |
|
1527 } |
|
1528 return status; |
|
1529 } |
|
1530 |
|
1531 double |
164
|
1532 tree_constant_rep::double_value (void) const |
1
|
1533 { |
105
|
1534 switch (type_tag) |
1
|
1535 { |
105
|
1536 case scalar_constant: |
|
1537 return scalar; |
|
1538 case complex_scalar_constant: |
|
1539 { |
|
1540 int flag = user_pref.ok_to_lose_imaginary_part; |
|
1541 if (flag == -1) |
|
1542 warning ("implicit conversion of complex value to real value"); |
1
|
1543 |
105
|
1544 if (flag != 0) |
|
1545 return real (*complex_scalar); |
|
1546 |
|
1547 error ("implicit conversion of complex value to real value not allowed"); |
|
1548 jump_to_top_level (); |
|
1549 } |
|
1550 default: |
|
1551 panic_impossible (); |
|
1552 break; |
1
|
1553 } |
|
1554 } |
|
1555 |
|
1556 Matrix |
164
|
1557 tree_constant_rep::matrix_value (void) const |
1
|
1558 { |
105
|
1559 switch (type_tag) |
1
|
1560 { |
105
|
1561 case matrix_constant: |
|
1562 return *matrix; |
|
1563 case complex_matrix_constant: |
|
1564 { |
|
1565 int flag = user_pref.ok_to_lose_imaginary_part; |
|
1566 if (flag == -1) |
|
1567 warning ("implicit conversion of complex matrix to real matrix"); |
1
|
1568 |
105
|
1569 if (flag != 0) |
|
1570 return real (*complex_matrix); |
|
1571 else |
1
|
1572 error ("implicit conversion of complex matrix to real matrix not allowed"); |
105
|
1573 jump_to_top_level (); |
|
1574 } |
|
1575 default: |
|
1576 panic_impossible (); |
|
1577 break; |
1
|
1578 } |
|
1579 } |
|
1580 |
|
1581 Complex |
164
|
1582 tree_constant_rep::complex_value (void) const |
1
|
1583 { |
101
|
1584 switch (type_tag) |
|
1585 { |
|
1586 case complex_scalar_constant: |
105
|
1587 return *complex_scalar; |
101
|
1588 case scalar_constant: |
105
|
1589 return Complex (scalar); |
101
|
1590 default: |
|
1591 panic_impossible (); |
|
1592 break; |
|
1593 } |
1
|
1594 } |
|
1595 |
|
1596 ComplexMatrix |
164
|
1597 tree_constant_rep::complex_matrix_value (void) const |
1
|
1598 { |
105
|
1599 switch (type_tag) |
101
|
1600 { |
|
1601 case scalar_constant: |
|
1602 { |
105
|
1603 return ComplexMatrix (scalar); |
101
|
1604 } |
|
1605 case complex_scalar_constant: |
105
|
1606 { |
|
1607 return ComplexMatrix (*complex_scalar); |
|
1608 } |
101
|
1609 case matrix_constant: |
|
1610 { |
105
|
1611 return ComplexMatrix (*matrix); |
101
|
1612 } |
|
1613 case complex_matrix_constant: |
105
|
1614 return *complex_matrix; |
101
|
1615 break; |
|
1616 default: |
|
1617 panic_impossible (); |
|
1618 break; |
|
1619 } |
1
|
1620 } |
|
1621 |
|
1622 char * |
164
|
1623 tree_constant_rep::string_value (void) const |
1
|
1624 { |
|
1625 assert (type_tag == string_constant); |
|
1626 return string; |
|
1627 } |
|
1628 |
|
1629 Range |
164
|
1630 tree_constant_rep::range_value (void) const |
1
|
1631 { |
|
1632 assert (type_tag == range_constant); |
|
1633 return *range; |
|
1634 } |
|
1635 |
|
1636 int |
164
|
1637 tree_constant_rep::rows (void) const |
1
|
1638 { |
|
1639 int retval = -1; |
|
1640 switch (type_tag) |
|
1641 { |
|
1642 case scalar_constant: |
|
1643 case complex_scalar_constant: |
|
1644 case string_constant: |
|
1645 case range_constant: |
|
1646 retval = 1; |
|
1647 break; |
|
1648 case matrix_constant: |
|
1649 retval = matrix->rows (); |
|
1650 break; |
|
1651 case complex_matrix_constant: |
|
1652 retval = complex_matrix->rows (); |
|
1653 break; |
|
1654 case magic_colon: |
|
1655 error ("invalid use of colon operator"); |
|
1656 break; |
|
1657 case unknown_constant: |
|
1658 retval = 0; |
|
1659 break; |
|
1660 default: |
|
1661 panic_impossible (); |
|
1662 break; |
|
1663 } |
|
1664 return retval; |
|
1665 } |
|
1666 |
|
1667 int |
164
|
1668 tree_constant_rep::columns (void) const |
1
|
1669 { |
|
1670 int retval = -1; |
|
1671 switch (type_tag) |
|
1672 { |
|
1673 case scalar_constant: |
|
1674 case complex_scalar_constant: |
|
1675 retval = 1; |
|
1676 break; |
|
1677 case matrix_constant: |
|
1678 retval = matrix->columns (); |
|
1679 break; |
|
1680 case complex_matrix_constant: |
|
1681 retval = complex_matrix->columns (); |
|
1682 break; |
|
1683 case string_constant: |
|
1684 retval = strlen (string); |
|
1685 break; |
|
1686 case range_constant: |
|
1687 retval = range->nelem (); |
|
1688 break; |
|
1689 case magic_colon: |
|
1690 error ("invalid use of colon operator"); |
|
1691 break; |
|
1692 case unknown_constant: |
|
1693 retval = 0; |
|
1694 break; |
|
1695 default: |
|
1696 panic_impossible (); |
|
1697 break; |
|
1698 } |
|
1699 return retval; |
|
1700 } |
|
1701 |
|
1702 tree_constant |
164
|
1703 tree_constant_rep::all (void) const |
1
|
1704 { |
|
1705 if (type_tag == string_constant || type_tag == range_constant) |
|
1706 { |
|
1707 tree_constant tmp = make_numeric (); |
|
1708 return tmp.all (); |
|
1709 } |
|
1710 |
|
1711 tree_constant retval; |
|
1712 switch (type_tag) |
|
1713 { |
|
1714 case scalar_constant: |
|
1715 { |
|
1716 double status = (scalar != 0.0); |
|
1717 retval = tree_constant (status); |
|
1718 } |
|
1719 break; |
|
1720 case matrix_constant: |
|
1721 { |
|
1722 Matrix m = matrix->all (); |
|
1723 retval = tree_constant (m); |
|
1724 } |
|
1725 break; |
|
1726 case complex_scalar_constant: |
|
1727 { |
|
1728 double status = (*complex_scalar != 0.0); |
|
1729 retval = tree_constant (status); |
|
1730 } |
|
1731 break; |
|
1732 case complex_matrix_constant: |
|
1733 { |
|
1734 Matrix m = complex_matrix->all (); |
|
1735 retval = tree_constant (m); |
|
1736 } |
|
1737 break; |
|
1738 case string_constant: |
|
1739 case range_constant: |
|
1740 case magic_colon: |
|
1741 default: |
|
1742 panic_impossible (); |
|
1743 break; |
|
1744 } |
|
1745 return retval; |
|
1746 } |
|
1747 |
|
1748 tree_constant |
164
|
1749 tree_constant_rep::any (void) const |
1
|
1750 { |
|
1751 if (type_tag == string_constant || type_tag == range_constant) |
|
1752 { |
|
1753 tree_constant tmp = make_numeric (); |
|
1754 return tmp.any (); |
|
1755 } |
|
1756 |
|
1757 tree_constant retval; |
|
1758 switch (type_tag) |
|
1759 { |
|
1760 case scalar_constant: |
|
1761 { |
|
1762 double status = (scalar != 0.0); |
|
1763 retval = tree_constant (status); |
|
1764 } |
|
1765 break; |
|
1766 case matrix_constant: |
|
1767 { |
|
1768 Matrix m = matrix->any (); |
|
1769 retval = tree_constant (m); |
|
1770 } |
|
1771 break; |
|
1772 case complex_scalar_constant: |
|
1773 { |
|
1774 double status = (*complex_scalar != 0.0); |
|
1775 retval = tree_constant (status); |
|
1776 } |
|
1777 break; |
|
1778 case complex_matrix_constant: |
|
1779 { |
|
1780 Matrix m = complex_matrix->any (); |
|
1781 retval = tree_constant (m); |
|
1782 } |
|
1783 break; |
|
1784 case string_constant: |
|
1785 case range_constant: |
|
1786 case magic_colon: |
|
1787 default: |
|
1788 panic_impossible (); |
|
1789 break; |
|
1790 } |
|
1791 return retval; |
|
1792 } |
|
1793 |
|
1794 tree_constant |
164
|
1795 tree_constant_rep::isstr (void) const |
1
|
1796 { |
|
1797 double status = 0.0; |
164
|
1798 if (type_tag == string_constant) |
1
|
1799 status = 1.0; |
|
1800 tree_constant retval (status); |
|
1801 return retval; |
|
1802 } |
|
1803 |
|
1804 tree_constant |
|
1805 tree_constant_rep::convert_to_str (void) |
|
1806 { |
|
1807 tree_constant retval; |
|
1808 |
|
1809 switch (type_tag) |
|
1810 { |
|
1811 case complex_scalar_constant: |
|
1812 case scalar_constant: |
|
1813 { |
|
1814 double d = double_value (); |
|
1815 int i = NINT (d); |
|
1816 // Warn about out of range conversions? |
|
1817 char s[2]; |
|
1818 s[0] = (char) i; |
|
1819 retval = tree_constant (s); |
|
1820 } |
|
1821 break; |
|
1822 case complex_matrix_constant: |
|
1823 case matrix_constant: |
|
1824 { |
|
1825 ColumnVector v = to_vector (); |
|
1826 int len = v.length (); |
|
1827 if (len == 0) |
|
1828 error ("can only convert vectors and scalars to strings"); |
|
1829 else |
|
1830 { |
|
1831 char *s = new char [len+1]; |
|
1832 s[len] = '\0'; |
|
1833 for (int i = 0; i < len; i++) |
|
1834 { |
|
1835 double d = v.elem (i); |
|
1836 int ival = NINT (d); |
|
1837 // Warn about out of range conversions? |
|
1838 s[i] = (char) ival; |
|
1839 } |
|
1840 retval = tree_constant (s); |
|
1841 delete [] s; |
|
1842 } |
|
1843 } |
|
1844 break; |
|
1845 case range_constant: |
|
1846 { |
|
1847 Range r = range_value (); |
|
1848 double b = r.base (); |
|
1849 double incr = r.inc (); |
|
1850 int nel = r.nelem (); |
|
1851 char *s = new char [nel+1]; |
|
1852 s[nel] = '\0'; |
|
1853 for (int i = 0; i < nel; i++) |
|
1854 { |
|
1855 double d = b + i * incr; |
|
1856 int ival = NINT (d); |
|
1857 // Warn about out of range conversions? |
|
1858 s[i] = (char) ival; |
|
1859 } |
|
1860 retval = tree_constant (s); |
|
1861 delete [] s; |
|
1862 } |
|
1863 break; |
|
1864 case string_constant: |
|
1865 retval = tree_constant (*this); |
|
1866 break; |
|
1867 case magic_colon: |
|
1868 default: |
|
1869 panic_impossible (); |
|
1870 break; |
|
1871 } |
|
1872 return retval; |
|
1873 } |
|
1874 |
|
1875 tree_constant |
164
|
1876 tree_constant_rep::cumprod (void) const |
1
|
1877 { |
|
1878 if (type_tag == string_constant || type_tag == range_constant) |
|
1879 { |
|
1880 tree_constant tmp = make_numeric (); |
|
1881 return tmp.cumprod (); |
|
1882 } |
|
1883 |
|
1884 tree_constant retval; |
|
1885 switch (type_tag) |
|
1886 { |
|
1887 case scalar_constant: |
|
1888 retval = tree_constant (scalar); |
|
1889 break; |
|
1890 case matrix_constant: |
|
1891 { |
|
1892 Matrix m = matrix->cumprod (); |
|
1893 retval = tree_constant (m); |
|
1894 } |
|
1895 break; |
|
1896 case complex_scalar_constant: |
|
1897 retval = tree_constant (*complex_scalar); |
|
1898 break; |
|
1899 case complex_matrix_constant: |
|
1900 { |
|
1901 ComplexMatrix m = complex_matrix->cumprod (); |
|
1902 retval = tree_constant (m); |
|
1903 } |
|
1904 break; |
|
1905 case string_constant: |
|
1906 case range_constant: |
|
1907 case magic_colon: |
|
1908 default: |
|
1909 panic_impossible (); |
|
1910 break; |
|
1911 } |
|
1912 return retval; |
|
1913 } |
|
1914 |
|
1915 tree_constant |
164
|
1916 tree_constant_rep::cumsum (void) const |
1
|
1917 { |
|
1918 if (type_tag == string_constant || type_tag == range_constant) |
|
1919 { |
|
1920 tree_constant tmp = make_numeric (); |
|
1921 return tmp.cumsum (); |
|
1922 } |
|
1923 |
|
1924 tree_constant retval; |
|
1925 switch (type_tag) |
|
1926 { |
|
1927 case scalar_constant: |
|
1928 retval = tree_constant (scalar); |
|
1929 break; |
|
1930 case matrix_constant: |
|
1931 { |
|
1932 Matrix m = matrix->cumsum (); |
|
1933 retval = tree_constant (m); |
|
1934 } |
|
1935 break; |
|
1936 case complex_scalar_constant: |
|
1937 retval = tree_constant (*complex_scalar); |
|
1938 break; |
|
1939 case complex_matrix_constant: |
|
1940 { |
|
1941 ComplexMatrix m = complex_matrix->cumsum (); |
|
1942 retval = tree_constant (m); |
|
1943 } |
|
1944 break; |
|
1945 case string_constant: |
|
1946 case range_constant: |
|
1947 case magic_colon: |
|
1948 default: |
|
1949 panic_impossible (); |
|
1950 break; |
|
1951 } |
|
1952 return retval; |
|
1953 } |
|
1954 |
|
1955 tree_constant |
164
|
1956 tree_constant_rep::prod (void) const |
1
|
1957 { |
|
1958 if (type_tag == string_constant || type_tag == range_constant) |
|
1959 { |
|
1960 tree_constant tmp = make_numeric (); |
|
1961 return tmp.prod (); |
|
1962 } |
|
1963 |
|
1964 tree_constant retval; |
|
1965 switch (type_tag) |
|
1966 { |
|
1967 case scalar_constant: |
|
1968 retval = tree_constant (scalar); |
|
1969 break; |
|
1970 case matrix_constant: |
|
1971 { |
|
1972 Matrix m = matrix->prod (); |
|
1973 retval = tree_constant (m); |
|
1974 } |
|
1975 break; |
|
1976 case complex_scalar_constant: |
|
1977 retval = tree_constant (*complex_scalar); |
|
1978 break; |
|
1979 case complex_matrix_constant: |
|
1980 { |
|
1981 ComplexMatrix m = complex_matrix->prod (); |
|
1982 retval = tree_constant (m); |
|
1983 } |
|
1984 break; |
|
1985 case string_constant: |
|
1986 case range_constant: |
|
1987 case magic_colon: |
|
1988 default: |
|
1989 panic_impossible (); |
|
1990 break; |
|
1991 } |
|
1992 return retval; |
|
1993 } |
|
1994 |
|
1995 tree_constant |
164
|
1996 tree_constant_rep::sum (void) const |
1
|
1997 { |
|
1998 if (type_tag == string_constant || type_tag == range_constant) |
|
1999 { |
|
2000 tree_constant tmp = make_numeric (); |
|
2001 return tmp.sum (); |
|
2002 } |
|
2003 |
|
2004 tree_constant retval; |
|
2005 switch (type_tag) |
|
2006 { |
|
2007 case scalar_constant: |
|
2008 retval = tree_constant (scalar); |
|
2009 break; |
|
2010 case matrix_constant: |
|
2011 { |
|
2012 Matrix m = matrix->sum (); |
|
2013 retval = tree_constant (m); |
|
2014 } |
|
2015 break; |
|
2016 case complex_scalar_constant: |
|
2017 retval = tree_constant (*complex_scalar); |
|
2018 break; |
|
2019 case complex_matrix_constant: |
|
2020 { |
|
2021 ComplexMatrix m = complex_matrix->sum (); |
|
2022 retval = tree_constant (m); |
|
2023 } |
|
2024 break; |
|
2025 case string_constant: |
|
2026 case range_constant: |
|
2027 case magic_colon: |
|
2028 default: |
|
2029 panic_impossible (); |
|
2030 break; |
|
2031 } |
|
2032 return retval; |
|
2033 } |
|
2034 |
|
2035 tree_constant |
164
|
2036 tree_constant_rep::sumsq (void) const |
1
|
2037 { |
|
2038 if (type_tag == string_constant || type_tag == range_constant) |
|
2039 { |
|
2040 tree_constant tmp = make_numeric (); |
|
2041 return tmp.sumsq (); |
|
2042 } |
|
2043 |
|
2044 tree_constant retval; |
|
2045 switch (type_tag) |
|
2046 { |
|
2047 case scalar_constant: |
|
2048 retval = tree_constant (scalar * scalar); |
|
2049 break; |
|
2050 case matrix_constant: |
|
2051 { |
|
2052 Matrix m = matrix->sumsq (); |
|
2053 retval = tree_constant (m); |
|
2054 } |
|
2055 break; |
|
2056 case complex_scalar_constant: |
|
2057 { |
|
2058 Complex c (*complex_scalar); |
|
2059 retval = tree_constant (c * c); |
|
2060 } |
|
2061 break; |
|
2062 case complex_matrix_constant: |
|
2063 { |
|
2064 ComplexMatrix m = complex_matrix->sumsq (); |
|
2065 retval = tree_constant (m); |
|
2066 } |
|
2067 break; |
|
2068 case string_constant: |
|
2069 case range_constant: |
|
2070 case magic_colon: |
|
2071 default: |
|
2072 panic_impossible (); |
|
2073 break; |
|
2074 } |
|
2075 return retval; |
|
2076 } |
|
2077 |
|
2078 static tree_constant |
164
|
2079 make_diag (const Matrix& v, int k) |
1
|
2080 { |
|
2081 int nr = v.rows (); |
|
2082 int nc = v.columns (); |
|
2083 assert (nc == 1 || nr == 1); |
|
2084 |
|
2085 tree_constant retval; |
|
2086 |
|
2087 int roff = 0; |
|
2088 int coff = 0; |
|
2089 if (k > 0) |
|
2090 { |
|
2091 roff = 0; |
|
2092 coff = k; |
|
2093 } |
|
2094 else if (k < 0) |
|
2095 { |
|
2096 roff = -k; |
|
2097 coff = 0; |
|
2098 } |
|
2099 |
|
2100 if (nr == 1) |
|
2101 { |
|
2102 int n = nc + ABS (k); |
|
2103 Matrix m (n, n, 0.0); |
|
2104 for (int i = 0; i < nc; i++) |
|
2105 m.elem (i+roff, i+coff) = v.elem (0, i); |
|
2106 retval = tree_constant (m); |
|
2107 } |
|
2108 else |
|
2109 { |
|
2110 int n = nr + ABS (k); |
|
2111 Matrix m (n, n, 0.0); |
|
2112 for (int i = 0; i < nr; i++) |
|
2113 m.elem (i+roff, i+coff) = v.elem (i, 0); |
|
2114 retval = tree_constant (m); |
|
2115 } |
|
2116 |
|
2117 return retval; |
|
2118 } |
|
2119 |
|
2120 static tree_constant |
164
|
2121 make_diag (const ComplexMatrix& v, int k) |
1
|
2122 { |
|
2123 int nr = v.rows (); |
|
2124 int nc = v.columns (); |
|
2125 assert (nc == 1 || nr == 1); |
|
2126 |
|
2127 tree_constant retval; |
|
2128 |
|
2129 int roff = 0; |
|
2130 int coff = 0; |
|
2131 if (k > 0) |
|
2132 { |
|
2133 roff = 0; |
|
2134 coff = k; |
|
2135 } |
|
2136 else if (k < 0) |
|
2137 { |
|
2138 roff = -k; |
|
2139 coff = 0; |
|
2140 } |
|
2141 |
|
2142 if (nr == 1) |
|
2143 { |
|
2144 int n = nc + ABS (k); |
|
2145 ComplexMatrix m (n, n, 0.0); |
|
2146 for (int i = 0; i < nc; i++) |
|
2147 m.elem (i+roff, i+coff) = v.elem (0, i); |
|
2148 retval = tree_constant (m); |
|
2149 } |
|
2150 else |
|
2151 { |
|
2152 int n = nr + ABS (k); |
|
2153 ComplexMatrix m (n, n, 0.0); |
|
2154 for (int i = 0; i < nr; i++) |
|
2155 m.elem (i+roff, i+coff) = v.elem (i, 0); |
|
2156 retval = tree_constant (m); |
|
2157 } |
|
2158 |
|
2159 return retval; |
|
2160 } |
|
2161 |
|
2162 tree_constant |
164
|
2163 tree_constant_rep::diag (void) const |
1
|
2164 { |
|
2165 if (type_tag == string_constant || type_tag == range_constant) |
|
2166 { |
|
2167 tree_constant tmp = make_numeric (); |
|
2168 return tmp.diag (); |
|
2169 } |
|
2170 |
|
2171 tree_constant retval; |
|
2172 switch (type_tag) |
|
2173 { |
|
2174 case scalar_constant: |
|
2175 retval = tree_constant (scalar); |
|
2176 break; |
|
2177 case matrix_constant: |
|
2178 { |
|
2179 int nr = rows (); |
|
2180 int nc = columns (); |
|
2181 if (nr == 1 || nc == 1) |
|
2182 retval = make_diag (matrix_value (), 0); |
|
2183 else |
|
2184 { |
|
2185 ColumnVector v = matrix->diag (); |
|
2186 if (v.capacity () > 0) |
|
2187 retval = tree_constant (v); |
|
2188 } |
|
2189 } |
|
2190 break; |
|
2191 case complex_scalar_constant: |
|
2192 retval = tree_constant (*complex_scalar); |
|
2193 break; |
|
2194 case complex_matrix_constant: |
|
2195 { |
|
2196 int nr = rows (); |
|
2197 int nc = columns (); |
|
2198 if (nr == 1 || nc == 1) |
|
2199 retval = make_diag (complex_matrix_value (), 0); |
|
2200 else |
|
2201 { |
|
2202 ComplexColumnVector v = complex_matrix->diag (); |
|
2203 if (v.capacity () > 0) |
|
2204 retval = tree_constant (v); |
|
2205 } |
|
2206 } |
|
2207 break; |
|
2208 case string_constant: |
|
2209 case range_constant: |
|
2210 case magic_colon: |
|
2211 default: |
|
2212 panic_impossible (); |
|
2213 break; |
|
2214 } |
|
2215 return retval; |
|
2216 } |
|
2217 |
|
2218 tree_constant |
164
|
2219 tree_constant_rep::diag (const tree_constant& a) const |
1
|
2220 { |
|
2221 if (type_tag == string_constant || type_tag == range_constant) |
|
2222 { |
|
2223 tree_constant tmp = make_numeric (); |
|
2224 return tmp.diag (a); |
|
2225 } |
|
2226 |
|
2227 tree_constant tmp_a = a.make_numeric (); |
|
2228 |
|
2229 tree_constant_rep::constant_type a_type = tmp_a.const_type (); |
|
2230 |
|
2231 tree_constant retval; |
|
2232 |
|
2233 switch (type_tag) |
|
2234 { |
|
2235 case scalar_constant: |
|
2236 if (a_type == scalar_constant) |
|
2237 { |
|
2238 int k = NINT (tmp_a.double_value ()); |
|
2239 int n = ABS (k) + 1; |
|
2240 if (k == 0) |
|
2241 retval = tree_constant (scalar); |
|
2242 else if (k > 0) |
|
2243 { |
|
2244 Matrix m (n, n, 0.0); |
|
2245 m.elem (0, k) = scalar; |
|
2246 retval = tree_constant (m); |
|
2247 } |
|
2248 else if (k < 0) |
|
2249 { |
|
2250 Matrix m (n, n, 0.0); |
|
2251 m.elem (-k, 0) = scalar; |
|
2252 retval = tree_constant (m); |
|
2253 } |
|
2254 } |
|
2255 break; |
|
2256 case matrix_constant: |
|
2257 if (a_type == scalar_constant) |
|
2258 { |
|
2259 int k = NINT (tmp_a.double_value ()); |
|
2260 int nr = rows (); |
|
2261 int nc = columns (); |
|
2262 if (nr == 1 || nc == 1) |
|
2263 retval = make_diag (matrix_value (), k); |
|
2264 else |
|
2265 { |
|
2266 ColumnVector d = matrix->diag (k); |
|
2267 retval = tree_constant (d); |
|
2268 } |
|
2269 } |
|
2270 else |
|
2271 message ("diag", "invalid second argument"); |
|
2272 |
|
2273 break; |
|
2274 case complex_scalar_constant: |
|
2275 if (a_type == scalar_constant) |
|
2276 { |
|
2277 int k = NINT (tmp_a.double_value ()); |
|
2278 int n = ABS (k) + 1; |
|
2279 if (k == 0) |
|
2280 retval = tree_constant (*complex_scalar); |
|
2281 else if (k > 0) |
|
2282 { |
|
2283 ComplexMatrix m (n, n, 0.0); |
|
2284 m.elem (0, k) = *complex_scalar; |
|
2285 retval = tree_constant (m); |
|
2286 } |
|
2287 else if (k < 0) |
|
2288 { |
|
2289 ComplexMatrix m (n, n, 0.0); |
|
2290 m.elem (-k, 0) = *complex_scalar; |
|
2291 retval = tree_constant (m); |
|
2292 } |
|
2293 } |
|
2294 break; |
|
2295 case complex_matrix_constant: |
|
2296 if (a_type == scalar_constant) |
|
2297 { |
|
2298 int k = NINT (tmp_a.double_value ()); |
|
2299 int nr = rows (); |
|
2300 int nc = columns (); |
|
2301 if (nr == 1 || nc == 1) |
|
2302 retval = make_diag (complex_matrix_value (), k); |
|
2303 else |
|
2304 { |
|
2305 ComplexColumnVector d = complex_matrix->diag (k); |
|
2306 retval = tree_constant (d); |
|
2307 } |
|
2308 } |
|
2309 else |
|
2310 message ("diag", "invalid second argument"); |
|
2311 |
|
2312 break; |
|
2313 case string_constant: |
|
2314 case range_constant: |
|
2315 case magic_colon: |
|
2316 default: |
|
2317 panic_impossible (); |
|
2318 break; |
|
2319 } |
|
2320 return retval; |
|
2321 } |
|
2322 |
|
2323 void |
|
2324 tree_constant_rep::print_if_string (ostream& os, int warn) |
|
2325 { |
|
2326 if (type_tag == string_constant) |
|
2327 os << string << "\n"; |
|
2328 else if (warn) |
|
2329 warning ("expecting string, found numeric constant"); |
|
2330 } |
|
2331 |
|
2332 tree_constant |
164
|
2333 tree_constant_rep::mapper (Mapper_fcn& m_fcn, int print) const |
1
|
2334 { |
|
2335 tree_constant retval; |
|
2336 |
|
2337 if (type_tag == string_constant || type_tag == range_constant) |
|
2338 { |
|
2339 tree_constant tmp = make_numeric (); |
|
2340 return tmp.mapper (m_fcn, print); |
|
2341 } |
|
2342 |
|
2343 switch (type_tag) |
|
2344 { |
|
2345 case scalar_constant: |
|
2346 if (m_fcn.neg_arg_complex && scalar < 0.0) |
|
2347 { |
|
2348 if (m_fcn.c_c_mapper != NULL) |
|
2349 { |
|
2350 Complex c = m_fcn.c_c_mapper (Complex (scalar)); |
|
2351 retval = tree_constant (c); |
|
2352 } |
|
2353 else |
|
2354 panic_impossible (); |
|
2355 } |
|
2356 else |
|
2357 { |
|
2358 if (m_fcn.d_d_mapper != NULL) |
|
2359 { |
|
2360 double d = m_fcn.d_d_mapper (scalar); |
|
2361 retval = tree_constant (d); |
|
2362 } |
|
2363 else |
|
2364 panic_impossible (); |
|
2365 } |
|
2366 break; |
|
2367 case matrix_constant: |
|
2368 if (m_fcn.neg_arg_complex && any_element_is_negative (*matrix)) |
|
2369 { |
|
2370 if (m_fcn.c_c_mapper != NULL) |
|
2371 { |
|
2372 ComplexMatrix cm = map (m_fcn.c_c_mapper, |
|
2373 ComplexMatrix (*matrix)); |
|
2374 retval = tree_constant (cm); |
|
2375 } |
|
2376 else |
|
2377 panic_impossible (); |
|
2378 } |
|
2379 else |
|
2380 { |
|
2381 if (m_fcn.d_d_mapper != NULL) |
|
2382 { |
|
2383 Matrix m = map (m_fcn.d_d_mapper, *matrix); |
|
2384 retval = tree_constant (m); |
|
2385 } |
|
2386 else |
|
2387 panic_impossible (); |
|
2388 } |
|
2389 break; |
|
2390 case complex_scalar_constant: |
|
2391 if (m_fcn.d_c_mapper != NULL) |
|
2392 { |
|
2393 double d; |
|
2394 d = m_fcn.d_c_mapper (*complex_scalar); |
|
2395 retval = tree_constant (d); |
|
2396 } |
|
2397 else if (m_fcn.c_c_mapper != NULL) |
|
2398 { |
|
2399 Complex c; |
|
2400 c = m_fcn.c_c_mapper (*complex_scalar); |
|
2401 retval = tree_constant (c); |
|
2402 } |
|
2403 else |
|
2404 panic_impossible (); |
|
2405 break; |
|
2406 case complex_matrix_constant: |
|
2407 if (m_fcn.d_c_mapper != NULL) |
|
2408 { |
|
2409 Matrix m; |
|
2410 m = map (m_fcn.d_c_mapper, *complex_matrix); |
|
2411 retval = tree_constant (m); |
|
2412 } |
|
2413 else if (m_fcn.c_c_mapper != NULL) |
|
2414 { |
|
2415 ComplexMatrix cm; |
|
2416 cm = map (m_fcn.c_c_mapper, *complex_matrix); |
|
2417 retval = tree_constant (cm); |
|
2418 } |
|
2419 else |
|
2420 panic_impossible (); |
|
2421 break; |
|
2422 case string_constant: |
|
2423 case range_constant: |
|
2424 case magic_colon: |
|
2425 default: |
|
2426 panic_impossible (); |
|
2427 break; |
|
2428 } |
|
2429 |
|
2430 if (retval.is_defined ()) |
|
2431 return retval.eval (print); |
|
2432 else |
|
2433 return retval; |
|
2434 } |
|
2435 |
|
2436 tree_constant::~tree_constant (void) |
|
2437 { |
|
2438 #if defined (MDEBUG) |
|
2439 cerr << "~tree_constant: rep: " << rep |
|
2440 << " rep->count: " << rep->count << "\n"; |
|
2441 #endif |
|
2442 |
|
2443 if (--rep->count <= 0) |
|
2444 { |
|
2445 delete rep; |
|
2446 rep = (tree_constant_rep *) NULL; |
|
2447 } |
|
2448 } |
|
2449 |
|
2450 #if defined (MDEBUG) |
|
2451 void * |
|
2452 tree_constant::operator new (size_t size) |
|
2453 { |
|
2454 tree_constant *p = ::new tree_constant; |
|
2455 cerr << "tree_constant::new(): " << p << "\n"; |
|
2456 return p; |
|
2457 } |
|
2458 |
|
2459 void |
|
2460 tree_constant::operator delete (void *p, size_t size) |
|
2461 { |
|
2462 cerr << "tree_constant::delete(): " << p << "\n"; |
|
2463 ::delete p; |
|
2464 } |
|
2465 #endif |
|
2466 |
|
2467 /* |
96
|
2468 * Construct return vector of empty matrices. Return empty matrices |
|
2469 * and/or gripe when appropriate. |
|
2470 */ |
|
2471 tree_constant * |
164
|
2472 vector_of_empties (int nargout, const char *fcn_name) |
96
|
2473 { |
|
2474 tree_constant *retval = NULL_TREE_CONST; |
|
2475 |
|
2476 // Got an empty argument, check if should gripe/return empty values. |
|
2477 |
|
2478 int flag = user_pref.propagate_empty_matrices; |
|
2479 if (flag != 0) |
|
2480 { |
|
2481 if (flag < 0) |
|
2482 gripe_empty_arg (fcn_name, 0); |
|
2483 |
|
2484 Matrix m; |
|
2485 retval = new tree_constant [nargout+1]; |
|
2486 for (int i = 0; i < nargout; i++) |
|
2487 retval[i] = tree_constant (m); |
|
2488 } |
|
2489 else |
|
2490 gripe_empty_arg (fcn_name, 1); |
|
2491 |
|
2492 return retval; |
|
2493 } |
|
2494 |
|
2495 /* |
1
|
2496 ;;; Local Variables: *** |
|
2497 ;;; mode: C++ *** |
|
2498 ;;; page-delimiter: "^/\\*" *** |
|
2499 ;;; End: *** |
|
2500 */ |