7 |
|
|
8 |
|
#include <stdio.h> |
9 |
|
#include <stdlib.h> |
10 |
+ |
#include <errno.h> |
11 |
|
#include "rtio.h" |
12 |
|
#include "resolu.h" |
13 |
|
#include "rmatrix.h" |
14 |
+ |
#include "platform.h" |
15 |
|
|
16 |
|
#define MAXCOMP 50 /* #components we support */ |
17 |
|
|
24 |
|
int op; /* '*' or '+' */ |
25 |
|
} ROPERAT; /* matrix operation */ |
26 |
|
|
25 |
– |
int outfmt = DTfromHeader; /* output format */ |
27 |
|
int verbose = 0; /* verbose reporting? */ |
28 |
|
|
29 |
|
static void |
30 |
|
op_default(ROPERAT *op) |
31 |
|
{ |
32 |
|
memset(op, 0, sizeof(ROPERAT)); |
33 |
< |
op->op = '*'; |
33 |
> |
op->op = '.'; |
34 |
|
} |
35 |
|
|
36 |
|
static RMATRIX * |
47 |
|
fputs(": cannot load matrix\n", stderr); |
48 |
|
return(NULL); |
49 |
|
} |
49 |
– |
if (op->transpose) { /* transpose matrix? */ |
50 |
– |
mtmp = rmx_transpose(mright); |
51 |
– |
if (mtmp == NULL) { |
52 |
– |
fputs(fname, stderr); |
53 |
– |
fputs(": transpose failed\n", stderr); |
54 |
– |
rmx_free(mright); |
55 |
– |
return(NULL); |
56 |
– |
} |
57 |
– |
if (verbose) { |
58 |
– |
fputs(fname, stderr); |
59 |
– |
fputs(": transposed rows and columns\n", stderr); |
60 |
– |
} |
61 |
– |
rmx_free(mright); |
62 |
– |
mright = mtmp; |
63 |
– |
} |
50 |
|
if (op->nsf > 0) { /* apply scalar(s) */ |
51 |
|
if (op->clen > 0) { |
52 |
|
fputs("Options -s and -c are exclusive\n", stderr); |
96 |
|
rmx_free(mright); |
97 |
|
mright = mtmp; |
98 |
|
} |
99 |
+ |
if (op->transpose) { /* transpose matrix? */ |
100 |
+ |
mtmp = rmx_transpose(mright); |
101 |
+ |
if (mtmp == NULL) { |
102 |
+ |
fputs(fname, stderr); |
103 |
+ |
fputs(": transpose failed\n", stderr); |
104 |
+ |
rmx_free(mright); |
105 |
+ |
return(NULL); |
106 |
+ |
} |
107 |
+ |
if (verbose) { |
108 |
+ |
fputs(fname, stderr); |
109 |
+ |
fputs(": transposed rows and columns\n", stderr); |
110 |
+ |
} |
111 |
+ |
rmx_free(mright); |
112 |
+ |
mright = mtmp; |
113 |
+ |
} |
114 |
|
if (mleft == NULL) /* just one matrix */ |
115 |
|
return(mright); |
116 |
< |
if (op->op == '*') { /* concatenate */ |
116 |
> |
if (op->op == '.') { /* concatenate */ |
117 |
|
RMATRIX *mres = rmx_multiply(mleft, mright); |
118 |
|
if (mres == NULL) { |
119 |
|
fputs(fname, stderr); |
144 |
|
fputs(": added in matrix\n", stderr); |
145 |
|
} |
146 |
|
rmx_free(mright); |
147 |
+ |
} else if ((op->op == '*') | (op->op == '/')) { |
148 |
+ |
const char * tnam = (op->op == '/') ? |
149 |
+ |
"division" : "multiplication"; |
150 |
+ |
errno = 0; |
151 |
+ |
if (!rmx_elemult(mleft, mright, (op->op == '/'))) { |
152 |
+ |
fprintf(stderr, "%s: element-wise %s failed\n", |
153 |
+ |
fname, tnam); |
154 |
+ |
rmx_free(mright); |
155 |
+ |
return(NULL); |
156 |
+ |
} |
157 |
+ |
if (errno) |
158 |
+ |
fprintf(stderr, |
159 |
+ |
"%s: warning - error during element-wise %s\n", |
160 |
+ |
fname, tnam); |
161 |
+ |
else if (verbose) |
162 |
+ |
fprintf(stderr, "%s: element-wise %s\n", fname, tnam); |
163 |
+ |
rmx_free(mright); |
164 |
|
} else { |
165 |
|
fprintf(stderr, "%s: unknown operation '%c'\n", fname, op->op); |
166 |
|
rmx_free(mright); |
183 |
|
int |
184 |
|
main(int argc, char *argv[]) |
185 |
|
{ |
186 |
+ |
int outfmt = DTfromHeader; |
187 |
|
RMATRIX *mres = NULL; |
188 |
|
ROPERAT op; |
189 |
|
int i; |
191 |
|
op_default(&op); |
192 |
|
/* get options and arguments */ |
193 |
|
for (i = 1; i < argc; i++) |
194 |
< |
if (argv[i][0] == '+' && !argv[i][1]) { |
195 |
< |
op.op = '+'; |
194 |
> |
if (argv[i][0] && !argv[i][1] && |
195 |
> |
strchr("+*/", argv[i][0]) != NULL) { |
196 |
> |
op.op = argv[i][0]; |
197 |
|
} else if (argv[i][0] != '-' || !argv[i][1]) { |
198 |
|
char *fname = NULL; /* load matrix */ |
199 |
|
if (argv[i][0] != '-') |
251 |
|
if (mres == NULL) /* check that we got something */ |
252 |
|
goto userr; |
253 |
|
/* write result to stdout */ |
234 |
– |
#ifdef getc_unlocked |
235 |
– |
flockfile(stdout); |
236 |
– |
#endif |
254 |
|
if (outfmt == DTfromHeader) |
255 |
|
outfmt = mres->dtype; |
239 |
– |
#ifdef _WIN32 |
256 |
|
if (outfmt != DTascii) |
257 |
< |
_setmode(fileno(stdout), _O_BINARY); |
242 |
< |
#endif |
257 |
> |
SET_FILE_BINARY(stdout); |
258 |
|
newheader("RADIANCE", stdout); |
259 |
|
printargs(argc, argv, stdout); |
260 |
|
if (!rmx_write(mres, outfmt, stdout)) { |
265 |
|
return(0); |
266 |
|
userr: |
267 |
|
fprintf(stderr, |
268 |
< |
"Usage: %s [-v][-f[adfc][-t][-s sf .. | -c ce ..] m1 [+] .. > mres\n", |
268 |
> |
"Usage: %s [-v][-f[adfc][-t][-s sf .. | -c ce ..] m1 [+*/] .. > mres\n", |
269 |
|
argv[0]); |
270 |
|
return(1); |
271 |
|
} |