ViewVC Help
View File | Revision Log | Show Annotations | Download File | Root Listing
root/radiance/ray/src/util/rmatrix.c
(Generate patch)

Comparing ray/src/util/rmatrix.c (file contents):
Revision 2.61 by greg, Mon Nov 27 22:04:45 2023 UTC vs.
Revision 2.62 by greg, Tue Nov 28 00:39:56 2023 UTC

# Line 302 | Line 302 | rmx_load_spec(RMATRIX *rm, FILE *fp)
302          return(1);
303   }
304  
305 + /* Read matrix header from input stream (cannot be XML) */
306 + int
307 + rmx_load_header(RMATRIX *rm, FILE *fp)
308 + {
309 +        if (!rm | !fp)
310 +                return(-1);
311 +        if (rm->info) {                         /* clear state */
312 +                free(rm->info);
313 +                rm->info = NULL;
314 +        }
315 +        if (rm->mtx) {                          /* ...and data */
316 + #ifdef MAP_FILE
317 +                if (rm->mapped) {
318 +                        munmap(rm->mapped, mapped_size(rm));
319 +                        rm->mapped = NULL;
320 +                } else
321 + #endif
322 +                        free(rm->mtx);
323 +                rm->mtx = NULL;
324 +        }
325 +        if (rm->nrows | rm->ncols | !rm->dtype) {
326 +                rm->nrows = rm->ncols = 0;
327 +                rm->ncomp = 3;
328 +                setcolor(rm->cexp, 1.f, 1.f, 1.f);
329 +                memcpy(rm->wlpart, WLPART, sizeof(rm->wlpart));
330 +        }
331 +        SET_FILE_BINARY(fp);
332 +        rm->dtype = DTascii;                    /* assumed w/o FORMAT */
333 +        if (getheader(fp, get_dminfo, rm) < 0) {
334 +                fputs("Unrecognized matrix format\n", stderr);
335 +                return(-1);
336 +        }
337 +                                                /* resolution string? */
338 +        if ((rm->nrows <= 0) | (rm->ncols <= 0)) {
339 +                if (!fscnresolu(&rm->ncols, &rm->nrows, fp))
340 +                        return(-1);
341 +                if ((rm->dtype == DTrgbe) | (rm->dtype == DTxyze) &&
342 +                                rm->ncomp != 3)
343 +                        return(-1);
344 +        }
345 +        return(rm->dtype);
346 + }
347 +
348 + /* Allocate & load post-header data from stream given type set in rm->dtype */
349 + int
350 + rmx_load_data(RMATRIX *rm, FILE *fp)
351 + {
352 +        switch (rm->dtype) {
353 +        case DTascii:
354 +                SET_FILE_TEXT(fp);
355 +                return(rmx_load_ascii(rm, fp));
356 +        case DTfloat:
357 +                return(rmx_load_float(rm, fp));
358 +        case DTdouble:
359 +                return(rmx_load_double(rm, fp));
360 +        case DTrgbe:
361 +        case DTxyze:
362 +                return(rmx_load_rgbe(rm, fp));
363 +        case DTspec:
364 +                return(rmx_load_spec(rm, fp));
365 +        default:
366 +                fputs("Unsupported data type in rmx_loaddata()\n", stderr);
367 +        }
368 +        return(0);
369 + }
370 +
371   /* Load matrix from supported file type */
372   RMATRIX *
373   rmx_load(const char *inspec, RMPref rmp)
374   {
375          FILE            *fp;
376          RMATRIX         *dnew;
377 +        int             ok;
378  
379          if (!inspec)
380                  inspec = stdin_name;
381          else if (!*inspec)
382                  return(NULL);
383 <        if (inspec == stdin_name) {             /* reading from stdin? */
383 >        if (inspec == stdin_name)               /* reading from stdin? */
384                  fp = stdin;
385 <        } else if (inspec[0] == '!') {
386 <                if (!(fp = popen(inspec+1, "r")))
387 <                        return(NULL);
321 <        } else {
385 >        else if (inspec[0] == '!')
386 >                fp = popen(inspec+1, "r");
387 >        else {
388                  const char      *sp = inspec;   /* check suffix */
389                  while (*sp)
390                          ++sp;
# Line 332 | Line 398 | rmx_load(const char *inspec, RMPref rmp)
398                          dnew = rmx_from_cmatrix(cm);
399                          cm_free(cm);
400                          dnew->dtype = DTascii;
401 <                        return(dnew);
402 <                }
403 <                                                /* else open it ourselves */
338 <                if (!(fp = fopen(inspec, "r")))
339 <                        return(NULL);
401 >                        return(dnew);           /* return here */
402 >                }                               /* else open it ourselves */
403 >                fp = fopen(inspec, "r");
404          }
405 <        SET_FILE_BINARY(fp);
405 >        if (!fp)
406 >                return(NULL);
407   #ifdef getc_unlocked
408          flockfile(fp);
409   #endif
410 <        if (!(dnew = rmx_new(0,0,3))) {
411 <                fclose(fp);
410 >                                                /* load header info */
411 >        if (rmx_load_header(dnew = rmx_new(0,0,3), fp) < 0) {
412 >                fprintf(stderr, "Bad header in: %s\n", inspec);
413 >                if (inspec[0] == '!') pclose(fp);
414 >                else fclose(fp);
415 >                rmx_free(dnew);
416                  return(NULL);
417          }
418 <        dnew->dtype = DTascii;                  /* assumed w/o FORMAT */
419 <        if (getheader(fp, get_dminfo, dnew) < 0) {
420 <                fclose(fp);
352 <                return(NULL);
353 <        }
354 <        if ((dnew->nrows <= 0) | (dnew->ncols <= 0)) {
355 <                if (!fscnresolu(&dnew->ncols, &dnew->nrows, fp)) {
356 <                        fclose(fp);
357 <                        return(NULL);
358 <                }
359 <                if ((dnew->dtype == DTrgbe) | (dnew->dtype == DTxyze) &&
360 <                                dnew->ncomp != 3) {
361 <                        fclose(fp);
362 <                        return(NULL);
363 <                }
364 <        }
365 <        switch (dnew->dtype) {
366 <        case DTascii:
367 <                SET_FILE_TEXT(fp);
368 <                if (!rmx_load_ascii(dnew, fp))
369 <                        goto loaderr;
370 <                dnew->dtype = DTascii;          /* should leave double? */
371 <                break;
372 <        case DTfloat:
373 <                if (!rmx_load_float(dnew, fp))
374 <                        goto loaderr;
375 <                dnew->dtype = DTfloat;
376 <                break;
377 <        case DTdouble:
378 <                if (!rmx_load_double(dnew, fp))
379 <                        goto loaderr;
380 <                dnew->dtype = DTdouble;
381 <                break;
382 <        case DTrgbe:
383 <        case DTxyze:
384 <                if (!rmx_load_rgbe(dnew, fp))
385 <                        goto loaderr;
386 <                break;
387 <        case DTspec:
388 <                if (!rmx_load_spec(dnew, fp))
389 <                        goto loaderr;
390 <                break;
391 <        default:
392 <                goto loaderr;
393 <        }
394 <        if (fp != stdin) {
418 >        ok = rmx_load_data(dnew, fp);           /* allocate & load data */
419 >
420 >        if (fp != stdin) {                      /* close input stream */
421                  if (inspec[0] == '!')
422                          pclose(fp);
423                  else
# Line 401 | Line 427 | rmx_load(const char *inspec, RMPref rmp)
427          else
428                  funlockfile(fp);
429   #endif
430 +        if (!ok) {                              /* load failure? */
431 +                fprintf(stderr, "Error loading data from: %s\n", inspec);
432 +                rmx_free(dnew);
433 +                return(NULL);
434 +        }
435                                                  /* undo exposure? */
436          if ((dnew->cexp[0] != 1.f) |
437                          (dnew->cexp[1] != 1.f) | (dnew->cexp[2] != 1.f)) {
# Line 409 | Line 440 | rmx_load(const char *inspec, RMPref rmp)
440                  cmlt[0] = 1./dnew->cexp[0];
441                  cmlt[1] = 1./dnew->cexp[1];
442                  cmlt[2] = 1./dnew->cexp[2];
443 <                if (dnew->ncomp > MAXCSAMP)
444 <                        goto loaderr;
443 >                if (dnew->ncomp > MAXCSAMP) {
444 >                        fprintf(stderr, "Excess spectral components in: %s\n",
445 >                                        inspec);
446 >                        rmx_free(dnew);
447 >                        return(NULL);
448 >                }
449                  for (i = dnew->ncomp; i-- > 3; )
450                          cmlt[i] = cmlt[1];
451                  rmx_scale(dnew, cmlt);
452                  setcolor(dnew->cexp, 1.f, 1.f, 1.f);
453          }
454          return(dnew);
420 – loaderr:                                        /* should report error? */
421 –        if (inspec[0] == '!')
422 –                pclose(fp);
423 –        else
424 –                fclose(fp);
425 –        rmx_free(dnew);
426 –        return(NULL);
455   }
456  
457   static int
# Line 436 | Line 464 | rmx_write_ascii(const RMATRIX *rm, FILE *fp)
464  
465          for (i = 0; i < rm->nrows; i++) {
466              for (j = 0; j < rm->ncols; j++) {
467 <                const double    *dp = rmx_lval(rm,i,j);
467 >                const double    *dp = rmx_val(rm,i,j);
468                  for (k = 0; k < rm->ncomp; k++)
469                      fprintf(fp, fmt, dp[k]);
470                  fputc('\t', fp);
# Line 458 | Line 486 | rmx_write_float(const RMATRIX *rm, FILE *fp)
486          }
487          for (i = 0; i < rm->nrows; i++)
488              for (j = 0; j < rm->ncols; j++) {
489 <                const double    *dp = rmx_lval(rm,i,j);
489 >                const double    *dp = rmx_val(rm,i,j);
490                  for (k = rm->ncomp; k--; )
491                      val[k] = (float)dp[k];
492                  if (putbinary(val, sizeof(float), rm->ncomp, fp) != rm->ncomp)
# Line 473 | Line 501 | rmx_write_double(const RMATRIX *rm, FILE *fp)
501          int     i;
502  
503          for (i = 0; i < rm->nrows; i++)
504 <                if (putbinary(rmx_lval(rm,i,0), sizeof(double)*rm->ncomp,
504 >                if (putbinary(rmx_val(rm,i,0), sizeof(double)*rm->ncomp,
505                                          rm->ncols, fp) != rm->ncols)
506                          return(0);
507          return(1);
# Line 489 | Line 517 | rmx_write_rgbe(const RMATRIX *rm, FILE *fp)
517                  return(0);
518          for (i = 0; i < rm->nrows; i++) {
519              for (j = rm->ncols; j--; ) {
520 <                const double    *dp = rmx_lval(rm,i,j);
520 >                const double    *dp = rmx_val(rm,i,j);
521                  if (rm->ncomp == 1)
522                          setcolr(scan[j], dp[0], dp[0], dp[0]);
523                  else
# Line 516 | Line 544 | rmx_write_spec(const RMATRIX *rm, FILE *fp)
544                  return(0);
545          for (i = 0; i < rm->nrows; i++) {
546              for (j = rm->ncols; j--; ) {
547 <                const double    *dp = rmx_lval(rm,i,j);
547 >                const double    *dp = rmx_val(rm,i,j);
548                  for (k = rm->ncomp; k--; )
549                          scol[k] = dp[k];
550                  scolor2scolr(scan+j*(rm->ncomp+1), scol, rm->ncomp);
# Line 616 | Line 644 | rmx_write(const RMATRIX *rm, int dtype, FILE *fp)
644   #ifdef getc_unlocked
645          funlockfile(fp);
646   #endif
647 +        if (!ok) fputs("Error writing matrix\n", stderr);
648          return(ok);
649   }
650  
# Line 685 | Line 714 | rmx_transpose(const RMATRIX *rm)
714          memcpy(dnew->wlpart, rm->wlpart, sizeof(dnew->wlpart));
715          for (j = dnew->ncols; j--; )
716              for (i = dnew->nrows; i--; )
717 <                memcpy(rmx_lval(dnew,i,j), rmx_lval(rm,j,i),
717 >                memcpy(rmx_lval(dnew,i,j), rmx_val(rm,j,i),
718                                  sizeof(double)*dnew->ncomp);
719          return(dnew);
720   }
# Line 712 | Line 741 | rmx_multiply(const RMATRIX *m1, const RMATRIX *m2)
741                  for (k = mres->ncomp; k--; ) {
742                      double      d = 0;
743                      for (h = m1->ncols; h--; )
744 <                        d += rmx_lval(m1,i,h)[k] * rmx_lval(m2,h,j)[k];
744 >                        d += rmx_val(m1,i,h)[k] * rmx_val(m2,h,j)[k];
745                      rmx_lval(mres,i,j)[k] = d;
746                  }
747          return(mres);
# Line 739 | Line 768 | rmx_elemult(RMATRIX *m1, const RMATRIX *m2, int divide
768                  if (divide) {
769                      double      d;
770                      if (m2->ncomp == 1) {
771 <                        d = rmx_lval(m2,i,j)[0];
771 >                        d = rmx_val(m2,i,j)[0];
772                          if (d == 0) {
773                              ++zeroDivides;
774                              for (k = m1->ncomp; k--; )
# Line 751 | Line 780 | rmx_elemult(RMATRIX *m1, const RMATRIX *m2, int divide
780                          }
781                      } else
782                          for (k = m1->ncomp; k--; ) {
783 <                            d = rmx_lval(m2,i,j)[k];
783 >                            d = rmx_val(m2,i,j)[k];
784                              if (d == 0) {
785                                  ++zeroDivides;
786                                  rmx_lval(m1,i,j)[k] = 0;
# Line 760 | Line 789 | rmx_elemult(RMATRIX *m1, const RMATRIX *m2, int divide
789                          }
790                  } else {
791                      if (m2->ncomp == 1) {
792 <                        const double    d = rmx_lval(m2,i,j)[0];
792 >                        const double    d = rmx_val(m2,i,j)[0];
793                          for (k = m1->ncomp; k--; )
794                              rmx_lval(m1,i,j)[k] *= d;
795                      } else
796                          for (k = m1->ncomp; k--; )
797 <                            rmx_lval(m1,i,j)[k] *= rmx_lval(m2,i,j)[k];
797 >                            rmx_lval(m1,i,j)[k] *= rmx_val(m2,i,j)[k];
798                  }
799          if (zeroDivides) {
800                  rmx_addinfo(m1, "WARNING: zero divide(s) corrupted results\n");
# Line 801 | Line 830 | rmx_sum(RMATRIX *msum, const RMATRIX *madd, const doub
830                  rmx_addinfo(msum, rmx_mismatch_warn);
831          for (i = msum->nrows; i--; )
832              for (j = msum->ncols; j--; ) {
833 <                const double    *da = rmx_lval(madd,i,j);
833 >                const double    *da = rmx_val(madd,i,j);
834                  double          *ds = rmx_lval(msum,i,j);
835                  for (k = msum->ncomp; k--; )
836                       ds[k] += sf[k] * da[k];
# Line 853 | Line 882 | rmx_transform(const RMATRIX *msrc, int n, const double
882          dnew->dtype = msrc->dtype;
883          for (i = dnew->nrows; i--; )
884              for (j = dnew->ncols; j--; ) {
885 <                const double    *ds = rmx_lval(msrc,i,j);
885 >                const double    *ds = rmx_val(msrc,i,j);
886                  for (kd = dnew->ncomp; kd--; ) {
887                      double      d = 0;
888                      for (ks = msrc->ncomp; ks--; )
# Line 902 | Line 931 | cm_from_rmatrix(const RMATRIX *rm)
931                  return(NULL);
932          for (i = cnew->nrows; i--; )
933              for (j = cnew->ncols; j--; ) {
934 <                const double    *dp = rmx_lval(rm,i,j);
934 >                const double    *dp = rmx_val(rm,i,j);
935                  COLORV          *cv = cm_lval(cnew,i,j);
936                  switch (rm->ncomp) {
937                  case 3:

Diff Legend

– Removed lines
+ Added lines
< Changed lines
> Changed lines