| 5 |  | * General matrix operations. | 
| 6 |  | */ | 
| 7 |  |  | 
| 8 | – | #include <stdio.h> | 
| 8 |  | #include <stdlib.h> | 
| 9 | < | #include <string.h> | 
| 10 | < | #include <fcntl.h> | 
| 9 | > | #include <errno.h> | 
| 10 | > | #include "rtio.h" | 
| 11 | > | #include "platform.h" | 
| 12 |  | #include "resolu.h" | 
| 13 | + | #include "paths.h" | 
| 14 |  | #include "rmatrix.h" | 
| 15 | + | #if !defined(_WIN32) && !defined(_WIN64) | 
| 16 | + | #include <sys/mman.h> | 
| 17 | + | #endif | 
| 18 |  |  | 
| 19 | < | static char     rmx_mismatch_warn[] = "WARNING: data type mismatch\n"; | 
| 19 | > | static const char       rmx_mismatch_warn[] = "WARNING: data type mismatch\n"; | 
| 20 |  |  | 
| 21 | < | /* Allocate a nr x nc matrix with n components */ | 
| 21 | > | #define array_size(rm)  (sizeof(double)*(rm)->nrows*(rm)->ncols*(rm)->ncomp) | 
| 22 | > | #define mapped_size(rm) ((char *)(rm)->mtx + array_size(rm) - (char *)(rm)->mapped) | 
| 23 | > |  | 
| 24 | > | /* Initialize a RMATRIX struct but don't allocate array space */ | 
| 25 |  | RMATRIX * | 
| 26 | < | rmx_alloc(int nr, int nc, int n) | 
| 26 | > | rmx_new(int nr, int nc, int n) | 
| 27 |  | { | 
| 28 |  | RMATRIX *dnew; | 
| 29 |  |  | 
| 30 | < | if ((nr <= 0) | (nc <= 0) | (n <= 0)) | 
| 30 | > | if (n <= 0) | 
| 31 |  | return(NULL); | 
| 32 | < | dnew = (RMATRIX *)malloc(sizeof(RMATRIX)-sizeof(dnew->mtx) + | 
| 33 | < | sizeof(dnew->mtx[0])*(n*nr*nc)); | 
| 34 | < | if (dnew == NULL) | 
| 35 | < | return(NULL); | 
| 36 | < | dnew->nrows = nr; dnew->ncols = nc; dnew->ncomp = n; | 
| 37 | < | dnew->dtype = DTdouble; | 
| 38 | < | dnew->info = NULL; | 
| 32 | > |  | 
| 33 | > | dnew = (RMATRIX *)calloc(1, sizeof(RMATRIX)); | 
| 34 | > | if (dnew) { | 
| 35 | > | dnew->dtype = DTdouble; | 
| 36 | > | dnew->nrows = nr; | 
| 37 | > | dnew->ncols = nc; | 
| 38 | > | dnew->ncomp = n; | 
| 39 | > | setcolor(dnew->cexp, 1.f, 1.f, 1.f); | 
| 40 | > | memcpy(dnew->wlpart, WLPART, sizeof(dnew->wlpart)); | 
| 41 | > | } | 
| 42 |  | return(dnew); | 
| 43 |  | } | 
| 44 |  |  | 
| 45 | < | /* Free a RMATRIX array */ | 
| 45 | > | /* Prepare a RMATRIX for writing (allocate array if needed) */ | 
| 46 | > | int | 
| 47 | > | rmx_prepare(RMATRIX *rm) | 
| 48 | > | { | 
| 49 | > | if (!rm) return(0); | 
| 50 | > | if (rm->mtx) | 
| 51 | > | return(1); | 
| 52 | > | if ((rm->nrows <= 0) | (rm->ncols <= 0) | (rm->ncomp <= 0)) | 
| 53 | > | return(0); | 
| 54 | > | rm->mtx = (double *)malloc(array_size(rm)); | 
| 55 | > | return(rm->mtx != NULL); | 
| 56 | > | } | 
| 57 | > |  | 
| 58 | > | /* Call rmx_new() and rmx_prepare() */ | 
| 59 | > | RMATRIX * | 
| 60 | > | rmx_alloc(int nr, int nc, int n) | 
| 61 | > | { | 
| 62 | > | RMATRIX *dnew = rmx_new(nr, nc, n); | 
| 63 | > |  | 
| 64 | > | if (dnew && !rmx_prepare(dnew)) { | 
| 65 | > | rmx_free(dnew); | 
| 66 | > | dnew = NULL; | 
| 67 | > | } | 
| 68 | > | return(dnew); | 
| 69 | > | } | 
| 70 | > |  | 
| 71 | > | /* Clear state by freeing info and matrix data */ | 
| 72 |  | void | 
| 73 | < | rmx_free(RMATRIX *rm) | 
| 73 | > | rmx_reset(RMATRIX *rm) | 
| 74 |  | { | 
| 75 |  | if (!rm) return; | 
| 76 | < | if (rm->info) | 
| 76 | > | if (rm->info) { | 
| 77 |  | free(rm->info); | 
| 78 | + | rm->info = NULL; | 
| 79 | + | } | 
| 80 | + | if (rm->mtx) { | 
| 81 | + | #ifdef MAP_FILE | 
| 82 | + | if (rm->mapped) { | 
| 83 | + | munmap(rm->mapped, mapped_size(rm)); | 
| 84 | + | rm->mapped = NULL; | 
| 85 | + | } else | 
| 86 | + | #endif | 
| 87 | + | free(rm->mtx); | 
| 88 | + | rm->mtx = NULL; | 
| 89 | + | } | 
| 90 | + | } | 
| 91 | + |  | 
| 92 | + | /* Free an RMATRIX struct and data */ | 
| 93 | + | void | 
| 94 | + | rmx_free(RMATRIX *rm) | 
| 95 | + | { | 
| 96 | + | if (!rm) return; | 
| 97 | + | rmx_reset(rm); | 
| 98 |  | free(rm); | 
| 99 |  | } | 
| 100 |  |  | 
| 102 |  | int | 
| 103 |  | rmx_newtype(int dtyp1, int dtyp2) | 
| 104 |  | { | 
| 105 | < | if ((dtyp1==DTxyze) | (dtyp1==DTrgbe) && dtyp1 != dtyp2) | 
| 105 | > | if ((dtyp1==DTxyze) | (dtyp1==DTrgbe) | (dtyp1==DTspec) | | 
| 106 | > | (dtyp2==DTxyze) | (dtyp2==DTrgbe) | (dtyp2==DTspec) | 
| 107 | > | && dtyp1 != dtyp2) | 
| 108 |  | return(0); | 
| 109 |  | if (dtyp1 < dtyp2) | 
| 110 |  | return(dtyp1); | 
| 115 |  | int | 
| 116 |  | rmx_addinfo(RMATRIX *rm, const char *info) | 
| 117 |  | { | 
| 118 | < | if (!info || !*info) | 
| 118 | > | int     oldlen = 0; | 
| 119 | > |  | 
| 120 | > | if (!rm || !info || !*info) | 
| 121 |  | return(0); | 
| 122 | < | if (!rm->info) | 
| 122 | > | if (!rm->info) { | 
| 123 |  | rm->info = (char *)malloc(strlen(info)+1); | 
| 124 | < | else | 
| 124 | > | if (rm->info) rm->info[0] = '\0'; | 
| 125 | > | } else { | 
| 126 | > | oldlen = strlen(rm->info); | 
| 127 |  | rm->info = (char *)realloc(rm->info, | 
| 128 | < | strlen(rm->info)+strlen(info)+1); | 
| 128 | > | oldlen+strlen(info)+1); | 
| 129 | > | } | 
| 130 |  | if (!rm->info) | 
| 131 |  | return(0); | 
| 132 | < | strcat(rm->info, info); | 
| 132 | > | strcpy(rm->info+oldlen, info); | 
| 133 |  | return(1); | 
| 134 |  | } | 
| 135 |  |  | 
| 137 |  | get_dminfo(char *s, void *p) | 
| 138 |  | { | 
| 139 |  | RMATRIX *ip = (RMATRIX *)p; | 
| 140 | < | char    fmt[64]; | 
| 140 | > | char    fmt[MAXFMTLEN]; | 
| 141 |  | int     i; | 
| 142 |  |  | 
| 143 | < | if (headidval(fmt, s)) | 
| 143 | > | if (headidval(NULL, s)) | 
| 144 |  | return(0); | 
| 145 | < | if (!strncmp(s, "NCOMP=", 6)) { | 
| 146 | < | ip->ncomp = atoi(s+6); | 
| 145 | > | if (isncomp(s)) { | 
| 146 | > | ip->ncomp = ncompval(s); | 
| 147 |  | return(0); | 
| 148 |  | } | 
| 149 |  | if (!strncmp(s, "NROWS=", 6)) { | 
| 154 |  | ip->ncols = atoi(s+6); | 
| 155 |  | return(0); | 
| 156 |  | } | 
| 157 | + | if ((i = isbigendian(s)) >= 0) { | 
| 158 | + | ip->swapin = (nativebigendian() != i); | 
| 159 | + | return(0); | 
| 160 | + | } | 
| 161 | + | if (isexpos(s)) { | 
| 162 | + | float   f = exposval(s); | 
| 163 | + | scalecolor(ip->cexp, f); | 
| 164 | + | return(0); | 
| 165 | + | } | 
| 166 | + | if (iscolcor(s)) { | 
| 167 | + | COLOR   ctmp; | 
| 168 | + | colcorval(ctmp, s); | 
| 169 | + | multcolor(ip->cexp, ctmp); | 
| 170 | + | return(0); | 
| 171 | + | } | 
| 172 | + | if (iswlsplit(s)) { | 
| 173 | + | wlsplitval(ip->wlpart, s); | 
| 174 | + | return(0); | 
| 175 | + | } | 
| 176 |  | if (!formatval(fmt, s)) { | 
| 177 |  | rmx_addinfo(ip, s); | 
| 178 |  | return(0); | 
| 179 | < | } | 
| 179 | > | }                       /* else check format */ | 
| 180 |  | for (i = 1; i < DTend; i++) | 
| 181 |  | if (!strcmp(fmt, cm_fmt_id[i])) { | 
| 182 |  | ip->dtype = i; | 
| 186 |  | } | 
| 187 |  |  | 
| 188 |  | static int | 
| 189 | < | rmx_load_ascii(RMATRIX *rm, FILE *fp) | 
| 189 | > | rmx_load_ascii(double *drp, const RMATRIX *rm, FILE *fp) | 
| 190 |  | { | 
| 191 | < | int     i, j, k; | 
| 192 | < | #ifdef _WIN32 | 
| 193 | < | _setmode(fileno(fp), _O_TEXT); | 
| 194 | < | #endif | 
| 195 | < | for (i = 0; i < rm->nrows; i++) | 
| 196 | < | for (j = 0; j < rm->ncols; j++) | 
| 115 | < | for (k = 0; k < rm->ncomp; k++) | 
| 116 | < | if (fscanf(fp, "%lf", &rmx_lval(rm,i,j,k)) != 1) | 
| 117 | < | return(0); | 
| 191 | > | int     j, k; | 
| 192 | > |  | 
| 193 | > | for (j = 0; j < rm->ncols; j++) | 
| 194 | > | for (k = rm->ncomp; k-- > 0; ) | 
| 195 | > | if (fscanf(fp, "%lf", drp++) != 1) | 
| 196 | > | return(0); | 
| 197 |  | return(1); | 
| 198 |  | } | 
| 199 |  |  | 
| 200 |  | static int | 
| 201 | < | rmx_load_float(RMATRIX *rm, FILE *fp) | 
| 201 | > | rmx_load_float(double *drp, const RMATRIX *rm, FILE *fp) | 
| 202 |  | { | 
| 203 | < | int     i, j, k; | 
| 203 | > | int     j, k; | 
| 204 |  | float   val[100]; | 
| 205 |  |  | 
| 206 |  | if (rm->ncomp > 100) { | 
| 207 |  | fputs("Unsupported # components in rmx_load_float()\n", stderr); | 
| 208 |  | exit(1); | 
| 209 |  | } | 
| 210 | < | for (i = 0; i < rm->nrows; i++) | 
| 211 | < | for (j = 0; j < rm->ncols; j++) { | 
| 212 | < | if (fread(val, sizeof(val[0]), rm->ncomp, fp) != rm->ncomp) | 
| 213 | < | return(0); | 
| 214 | < | for (k = rm->ncomp; k--; ) | 
| 215 | < | rmx_lval(rm,i,j,k) = val[k]; | 
| 216 | < | } | 
| 210 | > | for (j = 0; j < rm->ncols; j++) { | 
| 211 | > | if (getbinary(val, sizeof(val[0]), rm->ncomp, fp) != rm->ncomp) | 
| 212 | > | return(0); | 
| 213 | > | if (rm->swapin) | 
| 214 | > | swap32((char *)val, rm->ncomp); | 
| 215 | > | for (k = 0; k < rm->ncomp; k++) | 
| 216 | > | *drp++ = val[k]; | 
| 217 | > | } | 
| 218 |  | return(1); | 
| 219 |  | } | 
| 220 |  |  | 
| 221 |  | static int | 
| 222 | < | rmx_load_double(RMATRIX *rm, FILE *fp) | 
| 222 | > | rmx_load_double(double *drp, const RMATRIX *rm, FILE *fp) | 
| 223 |  | { | 
| 224 | < | int     i, j, k; | 
| 225 | < | double  val[100]; | 
| 224 | > | if (getbinary(drp, sizeof(*drp)*rm->ncomp, rm->ncols, fp) != rm->ncols) | 
| 225 | > | return(0); | 
| 226 | > | if (rm->swapin) | 
| 227 | > | swap64((char *)drp, rm->ncols*rm->ncomp); | 
| 228 | > | return(1); | 
| 229 | > | } | 
| 230 |  |  | 
| 231 | < | if (rm->ncomp > 100) { | 
| 232 | < | fputs("Unsupported # components in rmx_load_double()\n", stderr); | 
| 233 | < | exit(1); | 
| 231 | > | static int | 
| 232 | > | rmx_load_rgbe(double *drp, const RMATRIX *rm, FILE *fp) | 
| 233 | > | { | 
| 234 | > | COLR    *scan; | 
| 235 | > | COLOR   col; | 
| 236 | > | int     j; | 
| 237 | > |  | 
| 238 | > | if (rm->ncomp != 3) | 
| 239 | > | return(0); | 
| 240 | > | scan = (COLR *)tempbuffer(sizeof(COLR)*rm->ncols); | 
| 241 | > | if (!scan) | 
| 242 | > | return(0); | 
| 243 | > | if (freadcolrs(scan, rm->ncols, fp) < 0) | 
| 244 | > | return(0); | 
| 245 | > | for (j = 0; j < rm->ncols; j++) { | 
| 246 | > | colr_color(col, scan[j]); | 
| 247 | > | *drp++ = colval(col,RED); | 
| 248 | > | *drp++ = colval(col,GRN); | 
| 249 | > | *drp++ = colval(col,BLU); | 
| 250 |  | } | 
| 151 | – | for (i = 0; i < rm->nrows; i++) | 
| 152 | – | for (j = 0; j < rm->ncols; j++) { | 
| 153 | – | if (fread(val, sizeof(val[0]), rm->ncomp, fp) != rm->ncomp) | 
| 154 | – | return(0); | 
| 155 | – | for (k = rm->ncomp; k--; ) | 
| 156 | – | rmx_lval(rm,i,j,k) = val[k]; | 
| 157 | – | } | 
| 251 |  | return(1); | 
| 252 |  | } | 
| 253 |  |  | 
| 254 |  | static int | 
| 255 | < | rmx_load_rgbe(RMATRIX *rm, FILE *fp) | 
| 255 | > | rmx_load_spec(double *drp, const RMATRIX *rm, FILE *fp) | 
| 256 |  | { | 
| 257 | < | COLOR   *scan = (COLOR *)malloc(sizeof(COLOR)*rm->ncols); | 
| 258 | < | int     i, j; | 
| 257 | > | uby8    *scan; | 
| 258 | > | SCOLOR  scol; | 
| 259 | > | int     j, k; | 
| 260 |  |  | 
| 261 | < | if (scan == NULL) | 
| 261 | > | if ((rm->ncomp < 3) | (rm->ncomp > MAXCSAMP)) | 
| 262 |  | return(0); | 
| 263 | < | for (i = 0; i < rm->nrows; i++) { | 
| 264 | < | if (freadscan(scan, rm->ncols, fp) < 0) { | 
| 171 | < | free(scan); | 
| 263 | > | scan = (uby8 *)tempbuffer((rm->ncomp+1)*rm->ncols); | 
| 264 | > | if (!scan) | 
| 265 |  | return(0); | 
| 266 | < | } | 
| 267 | < | for (j = rm->ncols; j--; ) { | 
| 268 | < | rmx_lval(rm,i,j,0) = colval(scan[j],RED); | 
| 269 | < | rmx_lval(rm,i,j,1) = colval(scan[j],GRN); | 
| 270 | < | rmx_lval(rm,i,j,2) = colval(scan[j],BLU); | 
| 271 | < | } | 
| 266 | > | if (freadscolrs(scan, rm->ncomp, rm->ncols, fp) < 0) | 
| 267 | > | return(0); | 
| 268 | > | for (j = 0; j < rm->ncols; j++) { | 
| 269 | > | scolr2scolor(scol, scan+j*(rm->ncomp+1), rm->ncomp); | 
| 270 | > | for (k = 0; k < rm->ncomp; k++) | 
| 271 | > | *drp++ = scol[k]; | 
| 272 |  | } | 
| 180 | – | free(scan); | 
| 273 |  | return(1); | 
| 274 |  | } | 
| 275 |  |  | 
| 276 | + | /* Read matrix header from input stream (cannot be XML) */ | 
| 277 | + | int | 
| 278 | + | rmx_load_header(RMATRIX *rm, FILE *fp) | 
| 279 | + | { | 
| 280 | + | if (!rm | !fp) | 
| 281 | + | return(0); | 
| 282 | + | rmx_reset(rm);                          /* clear state */ | 
| 283 | + | if (rm->nrows | rm->ncols | !rm->dtype) { | 
| 284 | + | rm->nrows = rm->ncols = 0; | 
| 285 | + | rm->ncomp = 3; | 
| 286 | + | setcolor(rm->cexp, 1.f, 1.f, 1.f); | 
| 287 | + | memcpy(rm->wlpart, WLPART, sizeof(rm->wlpart)); | 
| 288 | + | rm->swapin = 0; | 
| 289 | + | } | 
| 290 | + | rm->dtype = DTascii;                    /* assumed w/o FORMAT */ | 
| 291 | + | if (getheader(fp, get_dminfo, rm) < 0) { | 
| 292 | + | fputs("Unrecognized matrix format\n", stderr); | 
| 293 | + | return(0); | 
| 294 | + | } | 
| 295 | + | if ((rm->dtype == DTrgbe) | (rm->dtype == DTxyze) && | 
| 296 | + | rm->ncomp != 3) | 
| 297 | + | return(0); | 
| 298 | + | if (rm->ncols <= 0 &&                   /* resolution string? */ | 
| 299 | + | !fscnresolu(&rm->ncols, &rm->nrows, fp)) | 
| 300 | + | return(0); | 
| 301 | + | if (rm->dtype == DTascii)               /* set file type (WINDOWS) */ | 
| 302 | + | SET_FILE_TEXT(fp); | 
| 303 | + | else | 
| 304 | + | SET_FILE_BINARY(fp); | 
| 305 | + | return(1); | 
| 306 | + | } | 
| 307 | + |  | 
| 308 | + | /* Load next row as double (cannot be XML) */ | 
| 309 | + | int | 
| 310 | + | rmx_load_row(double *drp, const RMATRIX *rm, FILE *fp) | 
| 311 | + | { | 
| 312 | + | switch (rm->dtype) { | 
| 313 | + | case DTascii: | 
| 314 | + | return(rmx_load_ascii(drp, rm, fp)); | 
| 315 | + | case DTfloat: | 
| 316 | + | return(rmx_load_float(drp, rm, fp)); | 
| 317 | + | case DTdouble: | 
| 318 | + | return(rmx_load_double(drp, rm, fp)); | 
| 319 | + | case DTrgbe: | 
| 320 | + | case DTxyze: | 
| 321 | + | return(rmx_load_rgbe(drp, rm, fp)); | 
| 322 | + | case DTspec: | 
| 323 | + | return(rmx_load_spec(drp, rm, fp)); | 
| 324 | + | default: | 
| 325 | + | fputs("Unsupported data type in rmx_load_row()\n", stderr); | 
| 326 | + | } | 
| 327 | + | return(0); | 
| 328 | + | } | 
| 329 | + |  | 
| 330 | + | /* Allocate & load post-header data from stream given type set in rm->dtype */ | 
| 331 | + | int | 
| 332 | + | rmx_load_data(RMATRIX *rm, FILE *fp) | 
| 333 | + | { | 
| 334 | + | int     i; | 
| 335 | + | #ifdef MAP_FILE | 
| 336 | + | long    pos;            /* map memory for file > 1MB if possible */ | 
| 337 | + | if ((rm->dtype == DTdouble) & !rm->swapin && array_size(rm) >= 1L<<20 && | 
| 338 | + | (pos = ftell(fp)) >= 0 && !(pos % sizeof(double))) { | 
| 339 | + | rm->mapped = mmap(NULL, array_size(rm)+pos, PROT_READ|PROT_WRITE, | 
| 340 | + | MAP_PRIVATE, fileno(fp), 0); | 
| 341 | + | if (rm->mapped != MAP_FAILED) { | 
| 342 | + | rm->mtx = (double *)rm->mapped + pos/sizeof(double); | 
| 343 | + | return(1); | 
| 344 | + | }               /* else fall back on reading into memory */ | 
| 345 | + | rm->mapped = NULL; | 
| 346 | + | } | 
| 347 | + | #endif | 
| 348 | + | if (!rmx_prepare(rm)) { /* need in-core matrix array */ | 
| 349 | + | fprintf(stderr, "Cannot allocate %g MByte matrix array\n", | 
| 350 | + | (1./(1L<<20))*(double)array_size(rm)); | 
| 351 | + | return(0); | 
| 352 | + | } | 
| 353 | + | for (i = 0; i < rm->nrows; i++) | 
| 354 | + | if (!rmx_load_row(rmx_lval(rm,i,0), rm, fp)) | 
| 355 | + | return(0); | 
| 356 | + | return(1); | 
| 357 | + | } | 
| 358 | + |  | 
| 359 |  | /* Load matrix from supported file type */ | 
| 360 |  | RMATRIX * | 
| 361 | < | rmx_load(const char *fname) | 
| 361 | > | rmx_load(const char *inspec, RMPref rmp) | 
| 362 |  | { | 
| 363 | < | FILE            *fp = stdin; | 
| 189 | < | RMATRIX         dinfo; | 
| 363 | > | FILE            *fp; | 
| 364 |  | RMATRIX         *dnew; | 
| 365 | + | int             ok; | 
| 366 |  |  | 
| 367 | < | if (fname == NULL) {                    /* reading from stdin? */ | 
| 368 | < | fname = "<stdin>"; | 
| 369 | < | #ifdef _WIN32 | 
| 370 | < | _setmode(fileno(stdin), _O_BINARY); | 
| 371 | < | #endif | 
| 372 | < | } else { | 
| 373 | < | const char      *sp = fname;    /* check suffix */ | 
| 367 | > | if (!inspec) | 
| 368 | > | inspec = stdin_name; | 
| 369 | > | else if (!*inspec) | 
| 370 | > | return(NULL); | 
| 371 | > | if (inspec == stdin_name)               /* reading from stdin? */ | 
| 372 | > | fp = stdin; | 
| 373 | > | else if (inspec[0] == '!') | 
| 374 | > | fp = popen(inspec+1, "r"); | 
| 375 | > | else if (rmp != RMPnone) { | 
| 376 | > | const char      *sp = inspec;   /* check suffix */ | 
| 377 |  | while (*sp) | 
| 378 |  | ++sp; | 
| 379 | < | while (sp > fname && sp[-1] != '.') | 
| 379 | > | while (sp > inspec && sp[-1] != '.') | 
| 380 |  | --sp; | 
| 381 |  | if (!strcasecmp(sp, "XML")) {   /* assume it's a BSDF */ | 
| 382 | < | CMATRIX *cm = cm_loadBTDF((char *)fname); | 
| 383 | < | if (cm == NULL) | 
| 382 | > | CMATRIX *cm = rmp==RMPtrans ? cm_loadBTDF(inspec) : | 
| 383 | > | cm_loadBRDF(inspec, rmp==RMPreflB) ; | 
| 384 | > | if (!cm) | 
| 385 |  | return(NULL); | 
| 386 |  | dnew = rmx_from_cmatrix(cm); | 
| 387 |  | cm_free(cm); | 
| 388 | < | return(dnew); | 
| 389 | < | } | 
| 390 | < | /* else open it ourselves */ | 
| 391 | < | if ((fp = fopen(fname, "rb")) == NULL) | 
| 213 | < | return(NULL); | 
| 388 | > | dnew->dtype = DTascii; | 
| 389 | > | return(dnew);           /* return here */ | 
| 390 | > | }                               /* else open it ourselves */ | 
| 391 | > | fp = fopen(inspec, "r"); | 
| 392 |  | } | 
| 393 | + | if (!fp) | 
| 394 | + | return(NULL); | 
| 395 |  | #ifdef getc_unlocked | 
| 396 |  | flockfile(fp); | 
| 397 |  | #endif | 
| 398 | < | dinfo.nrows = dinfo.ncols = dinfo.ncomp = 0; | 
| 399 | < | dinfo.dtype = DTascii;                  /* assumed w/o FORMAT */ | 
| 400 | < | dinfo.info = NULL; | 
| 401 | < | if (getheader(fp, get_dminfo, &dinfo) < 0) { | 
| 402 | < | fclose(fp); | 
| 398 | > | SET_FILE_BINARY(fp);                    /* load header info */ | 
| 399 | > | if (!rmx_load_header(dnew = rmx_new(0,0,3), fp)) { | 
| 400 | > | fprintf(stderr, "Bad header in: %s\n", inspec); | 
| 401 | > | if (inspec[0] == '!') pclose(fp); | 
| 402 | > | else fclose(fp); | 
| 403 | > | rmx_free(dnew); | 
| 404 |  | return(NULL); | 
| 405 |  | } | 
| 406 | < | if ((dinfo.nrows <= 0) | (dinfo.ncols <= 0)) { | 
| 407 | < | if (!fscnresolu(&dinfo.ncols, &dinfo.nrows, fp)) { | 
| 406 | > | ok = rmx_load_data(dnew, fp);           /* allocate & load data */ | 
| 407 | > |  | 
| 408 | > | if (fp != stdin) {                      /* close input stream */ | 
| 409 | > | if (inspec[0] == '!') | 
| 410 | > | pclose(fp); | 
| 411 | > | else | 
| 412 |  | fclose(fp); | 
| 228 | – | return(NULL); | 
| 229 | – | } | 
| 230 | – | if (dinfo.ncomp <= 0) | 
| 231 | – | dinfo.ncomp = 3; | 
| 232 | – | else if ((dinfo.dtype == DTrgbe) | (dinfo.dtype == DTxyze) && | 
| 233 | – | dinfo.ncomp != 3) { | 
| 234 | – | fclose(fp); | 
| 235 | – | return(NULL); | 
| 236 | – | } | 
| 413 |  | } | 
| 414 | < | dnew = rmx_alloc(dinfo.nrows, dinfo.ncols, dinfo.ncomp); | 
| 415 | < | if (dnew == NULL) { | 
| 416 | < | fclose(fp); | 
| 414 | > | #ifdef getc_unlocked | 
| 415 | > | else | 
| 416 | > | funlockfile(fp); | 
| 417 | > | #endif | 
| 418 | > | if (!ok) {                              /* load failure? */ | 
| 419 | > | fprintf(stderr, "Error loading data from: %s\n", inspec); | 
| 420 | > | rmx_free(dnew); | 
| 421 |  | return(NULL); | 
| 422 |  | } | 
| 423 | < | dnew->info = dinfo.info; | 
| 424 | < | switch (dinfo.dtype) { | 
| 425 | < | case DTascii: | 
| 426 | < | if (!rmx_load_ascii(dnew, fp)) | 
| 427 | < | goto loaderr; | 
| 428 | < | break; | 
| 429 | < | case DTfloat: | 
| 430 | < | if (!rmx_load_float(dnew, fp)) | 
| 431 | < | goto loaderr; | 
| 432 | < | dnew->dtype = DTfloat; | 
| 433 | < | break; | 
| 434 | < | case DTdouble: | 
| 435 | < | if (!rmx_load_double(dnew, fp)) | 
| 436 | < | goto loaderr; | 
| 437 | < | dnew->dtype = DTdouble; | 
| 438 | < | break; | 
| 439 | < | case DTrgbe: | 
| 440 | < | case DTxyze: | 
| 261 | < | if (!rmx_load_rgbe(dnew, fp)) | 
| 262 | < | goto loaderr; | 
| 263 | < | dnew->dtype = dinfo.dtype; | 
| 264 | < | break; | 
| 265 | < | default: | 
| 266 | < | goto loaderr; | 
| 423 | > | /* undo exposure? */ | 
| 424 | > | if ((dnew->cexp[0] != 1.f) | | 
| 425 | > | (dnew->cexp[1] != 1.f) | (dnew->cexp[2] != 1.f)) { | 
| 426 | > | double  cmlt[MAXCSAMP]; | 
| 427 | > | int     i; | 
| 428 | > | cmlt[0] = 1./dnew->cexp[0]; | 
| 429 | > | cmlt[1] = 1./dnew->cexp[1]; | 
| 430 | > | cmlt[2] = 1./dnew->cexp[2]; | 
| 431 | > | if (dnew->ncomp > MAXCSAMP) { | 
| 432 | > | fprintf(stderr, "Excess spectral components in: %s\n", | 
| 433 | > | inspec); | 
| 434 | > | rmx_free(dnew); | 
| 435 | > | return(NULL); | 
| 436 | > | } | 
| 437 | > | for (i = dnew->ncomp; i-- > 3; ) | 
| 438 | > | cmlt[i] = cmlt[1]; | 
| 439 | > | rmx_scale(dnew, cmlt); | 
| 440 | > | setcolor(dnew->cexp, 1.f, 1.f, 1.f); | 
| 441 |  | } | 
| 268 | – | if (fp != stdin) | 
| 269 | – | fclose(fp); | 
| 442 |  | return(dnew); | 
| 271 | – | loaderr:                                        /* should report error? */ | 
| 272 | – | fclose(fp); | 
| 273 | – | rmx_free(dnew); | 
| 274 | – | return(NULL); | 
| 443 |  | } | 
| 444 |  |  | 
| 445 |  | static int | 
| 446 | < | rmx_write_ascii(const RMATRIX *rm, FILE *fp) | 
| 446 | > | rmx_write_ascii(const double *dp, int nc, int len, FILE *fp) | 
| 447 |  | { | 
| 448 | < | int     i, j, k; | 
| 449 | < | #ifdef _WIN32 | 
| 450 | < | _setmode(fileno(fp), _O_TEXT); | 
| 451 | < | #endif | 
| 284 | < | for (i = 0; i < rm->nrows; i++) { | 
| 285 | < | for (j = 0; j < rm->ncols; j++) { | 
| 286 | < | for (k = 0; k < rm->ncomp; k++) | 
| 287 | < | fprintf(fp, " %.15e", rmx_lval(rm,i,j,k)); | 
| 448 | > | while (len-- > 0) { | 
| 449 | > | int     k = nc; | 
| 450 | > | while (nc-- > 0) | 
| 451 | > | fprintf(fp, " %.7e", *dp++); | 
| 452 |  | fputc('\t', fp); | 
| 289 | – | } | 
| 290 | – | fputc('\n', fp); | 
| 453 |  | } | 
| 454 | < | return(1); | 
| 454 | > | return(fputc('\n', fp) != EOF); | 
| 455 |  | } | 
| 456 |  |  | 
| 457 |  | static int | 
| 458 | < | rmx_write_float(const RMATRIX *rm, FILE *fp) | 
| 458 | > | rmx_write_float(const double *dp, int len, FILE *fp) | 
| 459 |  | { | 
| 460 | < | int     i, j, k; | 
| 299 | < | float   val[100]; | 
| 460 | > | float   val; | 
| 461 |  |  | 
| 462 | < | if (rm->ncomp > 100) { | 
| 463 | < | fputs("Unsupported # components in rmx_write_float()\n", stderr); | 
| 464 | < | exit(1); | 
| 304 | < | } | 
| 305 | < | for (i = 0; i < rm->nrows; i++) | 
| 306 | < | for (j = 0; j < rm->ncols; j++) { | 
| 307 | < | for (k = rm->ncomp; k--; ) | 
| 308 | < | val[k] = (float)rmx_lval(rm,i,j,k); | 
| 309 | < | if (fwrite(val, sizeof(val[0]), rm->ncomp, fp) != rm->ncomp) | 
| 462 | > | while (len--) { | 
| 463 | > | val = *dp++; | 
| 464 | > | if (putbinary(&val, sizeof(float), 1, fp) != 1) | 
| 465 |  | return(0); | 
| 466 | < | } | 
| 466 | > | } | 
| 467 |  | return(1); | 
| 468 |  | } | 
| 469 |  |  | 
| 470 |  | static int | 
| 471 | < | rmx_write_double(const RMATRIX *rm, FILE *fp) | 
| 471 | > | rmx_write_rgbe(const double *dp, int nc, int len, FILE *fp) | 
| 472 |  | { | 
| 473 | < | int     i, j, k; | 
| 474 | < | double  val[100]; | 
| 473 | > | COLR    *scan; | 
| 474 | > | int     j; | 
| 475 |  |  | 
| 476 | < | if (rm->ncomp > 100) { | 
| 477 | < | fputs("Unsupported # components in rmx_write_double()\n", stderr); | 
| 478 | < | exit(1); | 
| 476 | > | if ((nc != 1) & (nc != 3)) return(0); | 
| 477 | > | scan = (COLR *)tempbuffer(sizeof(COLR)*len); | 
| 478 | > | if (!scan) return(0); | 
| 479 | > |  | 
| 480 | > | for (j = 0; j < len; j++, dp += nc) | 
| 481 | > | if (nc == 1) | 
| 482 | > | setcolr(scan[j], dp[0], dp[0], dp[0]); | 
| 483 | > | else | 
| 484 | > | setcolr(scan[j], dp[0], dp[1], dp[2]); | 
| 485 | > |  | 
| 486 | > | return(fwritecolrs(scan, len, fp) >= 0); | 
| 487 | > | } | 
| 488 | > |  | 
| 489 | > | static int | 
| 490 | > | rmx_write_spec(const double *dp, int nc, int len, FILE *fp) | 
| 491 | > | { | 
| 492 | > | uby8    *scan; | 
| 493 | > | SCOLOR  scol; | 
| 494 | > | int     j, k; | 
| 495 | > |  | 
| 496 | > | if (nc < 3) return(0); | 
| 497 | > | scan = (uby8 *)tempbuffer((nc+1)*len); | 
| 498 | > | if (!scan) return(0); | 
| 499 | > | for (j = len; j--; dp += nc) { | 
| 500 | > | for (k = nc; k--; ) | 
| 501 | > | scol[k] = dp[k]; | 
| 502 | > | scolor2scolr(scan+j*(nc+1), scol, nc); | 
| 503 |  | } | 
| 504 | < | for (i = 0; i < rm->nrows; i++) | 
| 326 | < | for (j = 0; j < rm->ncols; j++) { | 
| 327 | < | for (k = rm->ncomp; k--; ) | 
| 328 | < | val[k] = rmx_lval(rm,i,j,k); | 
| 329 | < | if (fwrite(val, sizeof(val[0]), rm->ncomp, fp) != rm->ncomp) | 
| 330 | < | return(0); | 
| 331 | < | } | 
| 332 | < | return(1); | 
| 504 | > | return(fwritescolrs(scan, nc, len, fp) >= 0); | 
| 505 |  | } | 
| 506 |  |  | 
| 507 | + | /* Check if CIE XYZ primaries were specified */ | 
| 508 |  | static int | 
| 509 | < | rmx_write_rgbe(const RMATRIX *rm, FILE *fp) | 
| 509 | > | findCIEprims(const char *info) | 
| 510 |  | { | 
| 511 | < | COLOR   *scan = (COLOR *)malloc(sizeof(COLOR)*rm->ncols); | 
| 339 | < | int     i, j; | 
| 511 | > | RGBPRIMS        prims; | 
| 512 |  |  | 
| 513 | < | if (scan == NULL) | 
| 513 | > | if (!info) | 
| 514 |  | return(0); | 
| 515 | < | for (i = 0; i < rm->nrows; i++) { | 
| 516 | < | for (j = rm->ncols; j--; ) | 
| 345 | < | setcolor(scan[j],       rmx_lval(rm,i,j,0), | 
| 346 | < | rmx_lval(rm,i,j,1), | 
| 347 | < | rmx_lval(rm,i,j,2)      ); | 
| 348 | < | if (fwritescan(scan, rm->ncols, fp) < 0) { | 
| 349 | < | free(scan); | 
| 515 | > | info = strstr(info, PRIMARYSTR); | 
| 516 | > | if (!info || !primsval(prims, info)) | 
| 517 |  | return(0); | 
| 518 | < | } | 
| 519 | < | } | 
| 520 | < | free(scan); | 
| 521 | < | return(1); | 
| 518 | > |  | 
| 519 | > | return((prims[RED][CIEX] > .99) & (prims[RED][CIEY] < .01) && | 
| 520 | > | (prims[GRN][CIEX] < .01) & (prims[GRN][CIEY] > .99) && | 
| 521 | > | (prims[BLU][CIEX] < .01) & (prims[BLU][CIEY] < .01)); | 
| 522 |  | } | 
| 523 |  |  | 
| 524 | < | /* Write matrix to file type indicated by dtype */ | 
| 525 | < | long | 
| 526 | < | rmx_write(const RMATRIX *rm, int dtype, FILE *fp) | 
| 524 | > | /* Finish writing header data with resolution and format, returning type used */ | 
| 525 | > | int | 
| 526 | > | rmx_write_header(const RMATRIX *rm, int dtype, FILE *fp) | 
| 527 |  | { | 
| 528 | < | RMATRIX *mydm = NULL; | 
| 362 | < | int     ok = 1; | 
| 363 | < |  | 
| 364 | < | if ((rm == NULL) | (fp == NULL)) | 
| 528 | > | if (!rm | !fp || !rm->mtx | (rm->ncols <= 0)) | 
| 529 |  | return(0); | 
| 366 | – | /* complete header */ | 
| 530 |  | if (rm->info) | 
| 531 |  | fputs(rm->info, fp); | 
| 532 |  | if (dtype == DTfromHeader) | 
| 533 |  | dtype = rm->dtype; | 
| 534 | < | else if ((dtype == DTrgbe) & (rm->dtype == DTxyze)) | 
| 534 | > | else if (dtype == DTrgbe && (rm->dtype == DTxyze || | 
| 535 | > | findCIEprims(rm->info))) | 
| 536 |  | dtype = DTxyze; | 
| 537 | < | else if ((dtype = DTxyze) & (rm->dtype == DTrgbe)) | 
| 537 | > | else if ((dtype == DTxyze) & (rm->dtype == DTrgbe)) | 
| 538 |  | dtype = DTrgbe; | 
| 539 | < | if ((dtype != DTrgbe) & (dtype != DTxyze)) { | 
| 540 | < | fprintf(fp, "NROWS=%d\n", rm->nrows); | 
| 539 | > | if ((dtype == DTspec) & (rm->ncomp < 3)) | 
| 540 | > | return(0); | 
| 541 | > |  | 
| 542 | > | if (dtype == DTascii)                   /* set file type (WINDOWS) */ | 
| 543 | > | SET_FILE_TEXT(fp); | 
| 544 | > | else | 
| 545 | > | SET_FILE_BINARY(fp); | 
| 546 | > | /* write exposure? */ | 
| 547 | > | if (rm->ncomp == 3 && (rm->cexp[RED] != rm->cexp[GRN]) | | 
| 548 | > | (rm->cexp[GRN] != rm->cexp[BLU])) | 
| 549 | > | fputcolcor(rm->cexp, fp); | 
| 550 | > | else if (rm->cexp[GRN] != 1.f) | 
| 551 | > | fputexpos(rm->cexp[GRN], fp); | 
| 552 | > | /* matrix size? */ | 
| 553 | > | if ((dtype > DTspec) | (rm->nrows <= 0)) { | 
| 554 | > | if (rm->nrows > 0) | 
| 555 | > | fprintf(fp, "NROWS=%d\n", rm->nrows); | 
| 556 |  | fprintf(fp, "NCOLS=%d\n", rm->ncols); | 
| 378 | – | fprintf(fp, "NCOMP=%d\n", rm->ncomp); | 
| 379 | – | } else if (rm->ncomp != 3) {            /* wrong # components? */ | 
| 380 | – | double  cmtx[3]; | 
| 381 | – | if (rm->ncomp != 1)             /* only convert grayscale */ | 
| 382 | – | return(0); | 
| 383 | – | cmtx[0] = cmtx[1] = cmtx[2] = 1; | 
| 384 | – | mydm = rmx_transform(rm, 3, cmtx); | 
| 385 | – | if (mydm == NULL) | 
| 386 | – | return(0); | 
| 387 | – | rm = mydm; | 
| 557 |  | } | 
| 558 | < | fputformat((char *)cm_fmt_id[dtype], fp); | 
| 559 | < | fputc('\n', fp); | 
| 560 | < | switch (dtype) {                        /* write data */ | 
| 558 | > | if (dtype >= DTspec) {                  /* # components & split? */ | 
| 559 | > | fputncomp(rm->ncomp, fp); | 
| 560 | > | if (dtype == DTspec || (rm->ncomp > 3 && | 
| 561 | > | memcmp(rm->wlpart, WLPART, sizeof(WLPART)))) | 
| 562 | > | fputwlsplit(rm->wlpart, fp); | 
| 563 | > | } else if ((rm->ncomp != 3) & (rm->ncomp != 1)) | 
| 564 | > | return(0);                      /* wrong # components */ | 
| 565 | > | if ((dtype == DTfloat) | (dtype == DTdouble)) | 
| 566 | > | fputendian(fp);                 /* important to record */ | 
| 567 | > | fputformat(cm_fmt_id[dtype], fp); | 
| 568 | > | fputc('\n', fp);                        /* end of header */ | 
| 569 | > | if ((dtype <= DTspec) & (rm->nrows > 0)) | 
| 570 | > | fprtresolu(rm->ncols, rm->nrows, fp); | 
| 571 | > | return(dtype); | 
| 572 | > | } | 
| 573 | > |  | 
| 574 | > | /* Write out matrix data (usually by row) */ | 
| 575 | > | int | 
| 576 | > | rmx_write_data(const double *dp, int nc, int len, int dtype, FILE *fp) | 
| 577 | > | { | 
| 578 | > | switch (dtype) { | 
| 579 |  | case DTascii: | 
| 580 | < | ok = rmx_write_ascii(rm, fp); | 
| 394 | < | break; | 
| 580 | > | return(rmx_write_ascii(dp, nc, len, fp)); | 
| 581 |  | case DTfloat: | 
| 582 | < | ok = rmx_write_float(rm, fp); | 
| 397 | < | break; | 
| 582 | > | return(rmx_write_float(dp, nc*len, fp)); | 
| 583 |  | case DTdouble: | 
| 584 | < | ok = rmx_write_double(rm, fp); | 
| 400 | < | break; | 
| 584 | > | return(putbinary(dp, sizeof(*dp)*nc, len, fp) == len); | 
| 585 |  | case DTrgbe: | 
| 586 |  | case DTxyze: | 
| 587 | < | fprtresolu(rm->ncols, rm->nrows, fp); | 
| 588 | < | ok = rmx_write_rgbe(rm, fp); | 
| 589 | < | break; | 
| 406 | < | default: | 
| 407 | < | return(0); | 
| 587 | > | return(rmx_write_rgbe(dp, nc, len, fp)); | 
| 588 | > | case DTspec: | 
| 589 | > | return(rmx_write_spec(dp, nc, len, fp)); | 
| 590 |  | } | 
| 591 | < | ok &= (fflush(fp) == 0); | 
| 410 | < | rmx_free(mydm); | 
| 411 | < | return(ftell(fp) * ok);         /* return # bytes written */ | 
| 591 | > | return(0); | 
| 592 |  | } | 
| 593 |  |  | 
| 594 | + | /* Write matrix using file format indicated by dtype */ | 
| 595 | + | int | 
| 596 | + | rmx_write(const RMATRIX *rm, int dtype, FILE *fp) | 
| 597 | + | { | 
| 598 | + | int     ok = 0; | 
| 599 | + | int     i; | 
| 600 | + | /* complete header */ | 
| 601 | + | dtype = rmx_write_header(rm, dtype, fp); | 
| 602 | + | if (dtype <= 0) | 
| 603 | + | return(0); | 
| 604 | + | #ifdef getc_unlocked | 
| 605 | + | flockfile(fp); | 
| 606 | + | #endif | 
| 607 | + | if (dtype == DTdouble)                  /* write all at once? */ | 
| 608 | + | ok = rmx_write_data(rm->mtx, rm->ncomp, | 
| 609 | + | rm->nrows*rm->ncols, dtype, fp); | 
| 610 | + | else                                    /* else row by row */ | 
| 611 | + | for (i = 0; i < rm->nrows; i++) { | 
| 612 | + | ok = rmx_write_data(rmx_val(rm,i,0), rm->ncomp, | 
| 613 | + | rm->ncols, dtype, fp); | 
| 614 | + | if (!ok) break; | 
| 615 | + | } | 
| 616 | + |  | 
| 617 | + | if (ok) ok = (fflush(fp) == 0); | 
| 618 | + | #ifdef getc_unlocked | 
| 619 | + | funlockfile(fp); | 
| 620 | + | #endif | 
| 621 | + | if (!ok) fputs("Error writing matrix\n", stderr); | 
| 622 | + | return(ok); | 
| 623 | + | } | 
| 624 | + |  | 
| 625 |  | /* Allocate and assign square identity matrix with n components */ | 
| 626 |  | RMATRIX * | 
| 627 |  | rmx_identity(const int dim, const int n) | 
| 628 |  | { | 
| 629 |  | RMATRIX *rid = rmx_alloc(dim, dim, n); | 
| 630 | < | int     i; | 
| 630 | > | int     i, k; | 
| 631 |  |  | 
| 632 | < | if (rid == NULL) | 
| 632 | > | if (!rid) | 
| 633 |  | return(NULL); | 
| 634 | < | memset(rid->mtx, 0, sizeof(rid->mtx[0])*dim*dim); | 
| 635 | < | for (i = dim; i--; ) | 
| 636 | < | rmx_lval(rid,i,i,0) = 1; | 
| 637 | < | for (i = n; --i; ) | 
| 638 | < | memcpy(rid->mtx+i*(dim*dim), rid->mtx, | 
| 639 | < | sizeof(rid->mtx[0])*dim*dim); | 
| 634 | > | memset(rid->mtx, 0, array_size(rid)); | 
| 635 | > | for (i = dim; i--; ) { | 
| 636 | > | double      *dp = rmx_lval(rid,i,i); | 
| 637 | > | for (k = n; k--; ) | 
| 638 | > | dp[k] = 1.; | 
| 639 | > | } | 
| 640 |  | return(rid); | 
| 641 |  | } | 
| 642 |  |  | 
| 646 |  | { | 
| 647 |  | RMATRIX *dnew; | 
| 648 |  |  | 
| 649 | < | if (rm == NULL) | 
| 649 | > | if (!rm || !rm->mtx) | 
| 650 |  | return(NULL); | 
| 651 |  | dnew = rmx_alloc(rm->nrows, rm->ncols, rm->ncomp); | 
| 652 | < | if (dnew == NULL) | 
| 652 | > | if (!dnew) | 
| 653 |  | return(NULL); | 
| 654 |  | rmx_addinfo(dnew, rm->info); | 
| 655 |  | dnew->dtype = rm->dtype; | 
| 656 | < | memcpy(dnew->mtx, rm->mtx, | 
| 657 | < | sizeof(rm->mtx[0])*rm->ncomp*rm->nrows*rm->ncols); | 
| 656 | > | copycolor(dnew->cexp, rm->cexp); | 
| 657 | > | memcpy(dnew->wlpart, rm->wlpart, sizeof(dnew->wlpart)); | 
| 658 | > | memcpy(dnew->mtx, rm->mtx, array_size(dnew)); | 
| 659 |  | return(dnew); | 
| 660 |  | } | 
| 661 |  |  | 
| 664 |  | rmx_transpose(const RMATRIX *rm) | 
| 665 |  | { | 
| 666 |  | RMATRIX *dnew; | 
| 667 | < | int     i, j, k; | 
| 667 | > | int     i, j; | 
| 668 |  |  | 
| 669 | < | if (rm == NULL) | 
| 669 | > | if (!rm || !rm->mtx) | 
| 670 |  | return(0); | 
| 671 | + | if ((rm->nrows == 1) | (rm->ncols == 1)) { | 
| 672 | + | dnew = rmx_copy(rm); | 
| 673 | + | if (!dnew) | 
| 674 | + | return(NULL); | 
| 675 | + | dnew->nrows = rm->ncols; | 
| 676 | + | dnew->ncols = rm->nrows; | 
| 677 | + | return(dnew); | 
| 678 | + | } | 
| 679 |  | dnew = rmx_alloc(rm->ncols, rm->nrows, rm->ncomp); | 
| 680 | < | if (dnew == NULL) | 
| 680 | > | if (!dnew) | 
| 681 |  | return(NULL); | 
| 682 |  | if (rm->info) { | 
| 683 |  | rmx_addinfo(dnew, rm->info); | 
| 684 |  | rmx_addinfo(dnew, "Transposed rows and columns\n"); | 
| 685 |  | } | 
| 686 |  | dnew->dtype = rm->dtype; | 
| 687 | < | for (i = dnew->nrows; i--; ) | 
| 688 | < | for (j = dnew->ncols; j--; ) | 
| 689 | < | for (k = dnew->ncomp; k--; ) | 
| 690 | < | rmx_lval(dnew,i,j,k) = rmx_lval(rm,j,i,k); | 
| 687 | > | copycolor(dnew->cexp, rm->cexp); | 
| 688 | > | memcpy(dnew->wlpart, rm->wlpart, sizeof(dnew->wlpart)); | 
| 689 | > | for (j = dnew->ncols; j--; ) | 
| 690 | > | for (i = dnew->nrows; i--; ) | 
| 691 | > | memcpy(rmx_lval(dnew,i,j), rmx_val(rm,j,i), | 
| 692 | > | sizeof(double)*dnew->ncomp); | 
| 693 |  | return(dnew); | 
| 694 |  | } | 
| 695 |  |  | 
| 700 |  | RMATRIX *mres; | 
| 701 |  | int     i, j, k, h; | 
| 702 |  |  | 
| 703 | < | if ((m1 == NULL) | (m2 == NULL) || | 
| 703 | > | if (!m1 | !m2 || !m1->mtx | !m2->mtx | | 
| 704 |  | (m1->ncomp != m2->ncomp) | (m1->ncols != m2->nrows)) | 
| 705 |  | return(NULL); | 
| 706 |  | mres = rmx_alloc(m1->nrows, m2->ncols, m1->ncomp); | 
| 707 | < | if (mres == NULL) | 
| 707 | > | if (!mres) | 
| 708 |  | return(NULL); | 
| 709 |  | i = rmx_newtype(m1->dtype, m2->dtype); | 
| 710 |  | if (i) | 
| 713 |  | rmx_addinfo(mres, rmx_mismatch_warn); | 
| 714 |  | for (i = mres->nrows; i--; ) | 
| 715 |  | for (j = mres->ncols; j--; ) | 
| 716 | < | for (h = m1->ncols; h--; ) { | 
| 717 | < | long double d = 0; | 
| 718 | < | for (k = mres->ncomp; k--; ) | 
| 719 | < | d += (long double)rmx_lval(m1,i,h,k) * | 
| 720 | < | (long double)rmx_lval(m2,h,j,k); | 
| 499 | < | rmx_lval(mres,i,j,k) = (double)d; | 
| 716 | > | for (k = mres->ncomp; k--; ) { | 
| 717 | > | double      d = 0; | 
| 718 | > | for (h = m1->ncols; h--; ) | 
| 719 | > | d += rmx_val(m1,i,h)[k] * rmx_val(m2,h,j)[k]; | 
| 720 | > | rmx_lval(mres,i,j)[k] = d; | 
| 721 |  | } | 
| 722 |  | return(mres); | 
| 723 |  | } | 
| 724 |  |  | 
| 725 | + | /* Element-wise multiplication (or division) of m2 into m1 */ | 
| 726 | + | int | 
| 727 | + | rmx_elemult(RMATRIX *m1, const RMATRIX *m2, int divide) | 
| 728 | + | { | 
| 729 | + | int     zeroDivides = 0; | 
| 730 | + | int     i, j, k; | 
| 731 | + |  | 
| 732 | + | if (!m1 | !m2 || !m1->mtx | !m2->mtx | | 
| 733 | + | (m1->ncols != m2->ncols) | (m1->nrows != m2->nrows)) | 
| 734 | + | return(0); | 
| 735 | + | if ((m2->ncomp > 1) & (m2->ncomp != m1->ncomp)) | 
| 736 | + | return(0); | 
| 737 | + | i = rmx_newtype(m1->dtype, m2->dtype); | 
| 738 | + | if (i) | 
| 739 | + | m1->dtype = i; | 
| 740 | + | else | 
| 741 | + | rmx_addinfo(m1, rmx_mismatch_warn); | 
| 742 | + | for (i = m1->nrows; i--; ) | 
| 743 | + | for (j = m1->ncols; j--; ) | 
| 744 | + | if (divide) { | 
| 745 | + | double      d; | 
| 746 | + | if (m2->ncomp == 1) { | 
| 747 | + | d = rmx_val(m2,i,j)[0]; | 
| 748 | + | if (d == 0) { | 
| 749 | + | ++zeroDivides; | 
| 750 | + | for (k = m1->ncomp; k--; ) | 
| 751 | + | rmx_lval(m1,i,j)[k] = 0; | 
| 752 | + | } else { | 
| 753 | + | d = 1./d; | 
| 754 | + | for (k = m1->ncomp; k--; ) | 
| 755 | + | rmx_lval(m1,i,j)[k] *= d; | 
| 756 | + | } | 
| 757 | + | } else | 
| 758 | + | for (k = m1->ncomp; k--; ) { | 
| 759 | + | d = rmx_val(m2,i,j)[k]; | 
| 760 | + | if (d == 0) { | 
| 761 | + | ++zeroDivides; | 
| 762 | + | rmx_lval(m1,i,j)[k] = 0; | 
| 763 | + | } else | 
| 764 | + | rmx_lval(m1,i,j)[k] /= d; | 
| 765 | + | } | 
| 766 | + | } else { | 
| 767 | + | if (m2->ncomp == 1) { | 
| 768 | + | const double    d = rmx_val(m2,i,j)[0]; | 
| 769 | + | for (k = m1->ncomp; k--; ) | 
| 770 | + | rmx_lval(m1,i,j)[k] *= d; | 
| 771 | + | } else | 
| 772 | + | for (k = m1->ncomp; k--; ) | 
| 773 | + | rmx_lval(m1,i,j)[k] *= rmx_val(m2,i,j)[k]; | 
| 774 | + | } | 
| 775 | + | if (zeroDivides) { | 
| 776 | + | rmx_addinfo(m1, "WARNING: zero divide(s) corrupted results\n"); | 
| 777 | + | errno = ERANGE; | 
| 778 | + | } | 
| 779 | + | return(1); | 
| 780 | + | } | 
| 781 | + |  | 
| 782 |  | /* Sum second matrix into first, applying scale factor beforehand */ | 
| 783 |  | int | 
| 784 |  | rmx_sum(RMATRIX *msum, const RMATRIX *madd, const double sf[]) | 
| 786 |  | double  *mysf = NULL; | 
| 787 |  | int     i, j, k; | 
| 788 |  |  | 
| 789 | < | if ((msum == NULL) | (madd == NULL) || | 
| 789 | > | if (!msum | !madd || !msum->mtx | !madd->mtx | | 
| 790 |  | (msum->nrows != madd->nrows) | | 
| 791 |  | (msum->ncols != madd->ncols) | | 
| 792 |  | (msum->ncomp != madd->ncomp)) | 
| 793 |  | return(0); | 
| 794 | < | if (sf == NULL) { | 
| 794 | > | if (!sf) { | 
| 795 |  | mysf = (double *)malloc(sizeof(double)*msum->ncomp); | 
| 796 | < | if (mysf == NULL) | 
| 796 | > | if (!mysf) | 
| 797 |  | return(0); | 
| 798 |  | for (k = msum->ncomp; k--; ) | 
| 799 |  | mysf[k] = 1; | 
| 805 |  | else | 
| 806 |  | rmx_addinfo(msum, rmx_mismatch_warn); | 
| 807 |  | for (i = msum->nrows; i--; ) | 
| 808 | < | for (j = msum->ncols; j--; ) | 
| 808 | > | for (j = msum->ncols; j--; ) { | 
| 809 | > | const double    *da = rmx_val(madd,i,j); | 
| 810 | > | double          *ds = rmx_lval(msum,i,j); | 
| 811 |  | for (k = msum->ncomp; k--; ) | 
| 812 | < | rmx_lval(msum,i,j,k) += sf[k] * rmx_lval(madd,i,j,k); | 
| 813 | < |  | 
| 814 | < | free(mysf); | 
| 812 | > | ds[k] += sf[k] * da[k]; | 
| 813 | > | } | 
| 814 | > | if (mysf) | 
| 815 | > | free(mysf); | 
| 816 |  | return(1); | 
| 817 |  | } | 
| 818 |  |  | 
| 822 |  | { | 
| 823 |  | int     i, j, k; | 
| 824 |  |  | 
| 825 | < | if ((rm == NULL) | (sf == NULL)) | 
| 825 | > | if (!rm | !sf || !rm->mtx) | 
| 826 |  | return(0); | 
| 827 |  | for (i = rm->nrows; i--; ) | 
| 828 | < | for (j = rm->ncols; j--; ) | 
| 828 | > | for (j = rm->ncols; j--; ) { | 
| 829 | > | double  *dp = rmx_lval(rm,i,j); | 
| 830 |  | for (k = rm->ncomp; k--; ) | 
| 831 | < | rmx_lval(rm,i,j,k) *= sf[k]; | 
| 832 | < |  | 
| 831 | > | dp[k] *= sf[k]; | 
| 832 | > | } | 
| 833 | > | if (rm->info) | 
| 834 | > | rmx_addinfo(rm, "Applied scalar\n"); | 
| 835 | > | /* XXX: should record as exposure for COLR and SCOLR types? */ | 
| 836 |  | return(1); | 
| 837 |  | } | 
| 838 |  |  | 
| 843 |  | int     i, j, ks, kd; | 
| 844 |  | RMATRIX *dnew; | 
| 845 |  |  | 
| 846 | < | if ((msrc == NULL) | (n <= 0) | (cmat == NULL)) | 
| 846 | > | if (!msrc | (n <= 0) | !cmat || !msrc->mtx) | 
| 847 |  | return(NULL); | 
| 848 |  | dnew = rmx_alloc(msrc->nrows, msrc->ncols, n); | 
| 849 | < | if (dnew == NULL) | 
| 849 | > | if (!dnew) | 
| 850 |  | return(NULL); | 
| 851 | + | if (msrc->info) { | 
| 852 | + | char    buf[128]; | 
| 853 | + | sprintf(buf, "Applied %dx%d component transform\n", | 
| 854 | + | dnew->ncomp, msrc->ncomp); | 
| 855 | + | rmx_addinfo(dnew, msrc->info); | 
| 856 | + | rmx_addinfo(dnew, buf); | 
| 857 | + | } | 
| 858 |  | dnew->dtype = msrc->dtype; | 
| 859 |  | for (i = dnew->nrows; i--; ) | 
| 860 | < | for (j = dnew->ncols; j--; ) | 
| 860 | > | for (j = dnew->ncols; j--; ) { | 
| 861 | > | const double    *ds = rmx_val(msrc,i,j); | 
| 862 |  | for (kd = dnew->ncomp; kd--; ) { | 
| 863 |  | double      d = 0; | 
| 864 |  | for (ks = msrc->ncomp; ks--; ) | 
| 865 | < | d += cmat[kd*msrc->ncomp + ks] * rmx_lval(msrc,i,j,ks); | 
| 866 | < | rmx_lval(dnew,i,j,kd) = d; | 
| 865 | > | d += cmat[kd*msrc->ncomp + ks] * ds[ks]; | 
| 866 | > | rmx_lval(dnew,i,j)[kd] = d; | 
| 867 |  | } | 
| 868 | + | } | 
| 869 |  | return(dnew); | 
| 870 |  | } | 
| 871 |  |  | 
| 876 |  | int     i, j; | 
| 877 |  | RMATRIX *dnew; | 
| 878 |  |  | 
| 879 | < | if (cm == NULL) | 
| 879 | > | if (!cm) | 
| 880 |  | return(NULL); | 
| 881 |  | dnew = rmx_alloc(cm->nrows, cm->ncols, 3); | 
| 882 | < | if (dnew == NULL) | 
| 882 | > | if (!dnew) | 
| 883 |  | return(NULL); | 
| 884 |  | dnew->dtype = DTfloat; | 
| 885 |  | for (i = dnew->nrows; i--; ) | 
| 886 |  | for (j = dnew->ncols; j--; ) { | 
| 887 |  | const COLORV    *cv = cm_lval(cm,i,j); | 
| 888 | < | rmx_lval(dnew,i,j,0) = cv[0]; | 
| 889 | < | rmx_lval(dnew,i,j,1) = cv[1]; | 
| 890 | < | rmx_lval(dnew,i,j,2) = cv[2]; | 
| 888 | > | double          *dp = rmx_lval(dnew,i,j); | 
| 889 | > | dp[0] = cv[0]; | 
| 890 | > | dp[1] = cv[1]; | 
| 891 | > | dp[2] = cv[2]; | 
| 892 |  | } | 
| 893 |  | return(dnew); | 
| 894 |  | } | 
| 900 |  | int     i, j; | 
| 901 |  | CMATRIX *cnew; | 
| 902 |  |  | 
| 903 | < | if (rm == NULL || rm->ncomp != 3) | 
| 903 | > | if (!rm || !rm->mtx | (rm->ncomp == 2)) | 
| 904 |  | return(NULL); | 
| 905 |  | cnew = cm_alloc(rm->nrows, rm->ncols); | 
| 906 | < | if (cnew == NULL) | 
| 906 | > | if (!cnew) | 
| 907 |  | return(NULL); | 
| 908 |  | for (i = cnew->nrows; i--; ) | 
| 909 |  | for (j = cnew->ncols; j--; ) { | 
| 910 | < | COLORV  *cv = cm_lval(cnew,i,j); | 
| 911 | < | cv[0] = (COLORV)rmx_lval(rm,i,j,0); | 
| 912 | < | cv[1] = (COLORV)rmx_lval(rm,i,j,1); | 
| 913 | < | cv[2] = (COLORV)rmx_lval(rm,i,j,2); | 
| 910 | > | const double    *dp = rmx_val(rm,i,j); | 
| 911 | > | COLORV          *cv = cm_lval(cnew,i,j); | 
| 912 | > | switch (rm->ncomp) { | 
| 913 | > | case 3: | 
| 914 | > | setcolor(cv, dp[0], dp[1], dp[2]); | 
| 915 | > | break; | 
| 916 | > | case 1: | 
| 917 | > | setcolor(cv, dp[0], dp[0], dp[0]); | 
| 918 | > | break; | 
| 919 | > | default: { | 
| 920 | > | SCOLOR  scol; | 
| 921 | > | int     k; | 
| 922 | > | for (k = rm->ncomp; k--; ) | 
| 923 | > | scol[k] = dp[k]; | 
| 924 | > | scolor2color(cv, scol, rm->ncomp, rm->wlpart); | 
| 925 | > | } break; | 
| 926 | > | } | 
| 927 |  | } | 
| 928 |  | return(cnew); | 
| 929 |  | } |