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

Comparing ray/src/util/rcomb.c (file contents):
Revision 2.22 by greg, Tue Jun 4 22:05:23 2024 UTC vs.
Revision 2.29 by greg, Fri Apr 4 18:06:48 2025 UTC

# Line 21 | Line 21 | static const char RCSid[] = "$Id$";
21   #define M_PI    3.14159265358979323846
22   #endif
23  
24 #define MAXCOMP         MAXCSAMP        /* #components we support */
25
24   /* Unary matrix operation(s) */
25   typedef struct {
26          double          cmat[MAXCOMP*MAXCOMP];  /* component transformation */
# Line 174 | Line 172 | xyzrow(ROPMAT *rop, int r, int p)
172  
173   /* Use the spectral sensitivity function to compute matrix coefficients */
174   void
175 < sensrow(ROPMAT *rop, int r, double (*sf)(SCOLOR sc, int ncs, const float wlpt[4]))
175 > sensrow(ROPMAT *rop, int r, double (*sf)(const SCOLOR sc, int ncs, const float wlpt[4]))
176   {
177          const int       nc = rop->imx.ncomp;
178          int             i;
# Line 488 | Line 486 | initialize(RMATRIX *imp)
486                  restype = mop[i].rmp->dtype;
487                  if (!imp->dtype || (restype = rmx_newtype(restype, imp->dtype)) > 0)
488                          imp->dtype = restype;
489 <                else
489 >                else if (!nowarn)
490                          fprintf(stderr, "%s: warning - data type mismatch\n",
491                                          mop[i].inspec);
492                  if (!i) {
# Line 552 | Line 550 | spawned_children(int np)
550  
551   #if defined(_WIN32) || defined(_WIN64)
552          if (np > 1) {
553 <                fputs("Warning: only one process under Windows\n", stderr);
553 >                if (!nowarn)
554 >                        fputs("Warning: only one process under Windows\n", stderr);
555                  np = 1;
556          } else
557   #endif
# Line 570 | Line 569 | spawned_children(int np)
569                  }
570          }
571                                  /* prep output row buffer(s) */
572 <        if (mcat || mop[nmats].preop.clen > 0) {
572 >        if (mop[nmats].preop.clen > 0) {
573                  if (!split_input(&mop[nmats]))  /* need separate buffer */
574 <                        return(0);
575 <                if (mop[nmats].preop.clen > 0)
576 <                        mop[nmats].rmp->ncomp = mop[nmats].preop.clen /
578 <                                                mop[nmats].imx.ncomp;
574 >                        goto memerror;
575 >                mop[nmats].rmp->ncomp = mop[nmats].preop.clen /
576 >                                        mop[nmats].imx.ncomp;
577          }
578          mop[nmats].imx.nrows = 1;
579          if (!rmx_prepare(&mop[nmats].imx))
# Line 600 | Line 598 | spawned_children(int np)
598                  goto memerror;
599          for (i = nchildren; i--; ) cproc[i] = sp_inactive;
600          cproc[nchildren-1].flags |= PF_FILT_OUT;
601 <                                /* start each child */
602 <        for (i = 0; i < nchildren; i++) {
603 <                rv = open_process(&cproc[i], NULL);
604 <                if (rv <= 0) break;
607 <        }
601 >                                /* start each child from parent */
602 >        for (i = 0; i < nchildren; i++)
603 >                if ((rv = open_process(&cproc[i], NULL)) <= 0)
604 >                        break;  /* child breaks here */
605          if (rv < 0) {
606 <                perror("fork");
606 >                perror("fork"); /* WTH? */
607                  close_processes(cproc, i);
608                  exit(1);
609          }
610 <        if (rv) {               /* are we the parent? */
611 <                i = nchildren-1;        /* last child is sole reader */
612 <                while (i-- > 0) {
613 <                        close(cproc[i].r);
614 <                        cproc[i].r = -1;
610 >        if (i != nchildren-1) { /* last child is sole reader */
611 >                int     j = i;
612 >                while (j-- > 0) {
613 >                        close(cproc[j].r);
614 >                        cproc[j].r = -1;
615                  }
619                return(1);      /* parent return value */
616          }
617 <        inchild = i;            /* our child index */
617 >        if (rv > 0)
618 >                return(1);      /* parent return value */
619 >
620 >        inchild = i;            /* else set our child index */
621          while (i-- > 0)         /* only parent writes siblings */
622                  close(cproc[i].w);
623  
624 +        i = nmats;              /* close matrix streams (carefully) */
625 +        while (i-- > 0) {
626 +                if (mop[i].infp != stdin) {
627 +                        close(fileno(mop[i].infp));     /* avoid lseek() */
628 +                        fclose(mop[i].infp);            /* ! pclose() */
629 +                }
630 +                mop[i].infp = NULL;
631 +        }
632 +        fpurge(stdin);          /* discard previously buffered input */
633 +
634          if (inchild == nchildren-1)
635                  return(-1);     /* output process return value */
636  
637 <        i = inchild;            /* won't read from siblings */
629 <        while (i-- > 0)
630 <                close(cproc[i].r);
631 <        i = nmats;              /* redirect input matrix streams */
637 >        i = nmats;              /* get matrix rows from parent */
638          while (i-- > 0) {
633                if (mop[i].infp != stdin)
634                        fclose(mop[i].infp);    /* ! pclose() */
639                  mop[i].infp = stdin;
640                  mop[i].imx.dtype = DTrmx_native;
641                  mop[i].imx.pflags &= ~RMF_SWAPIN;
642          }
639        fpurge(stdin);          /* discard any previous matrix input */
643   #ifdef getc_unlocked
644          flockfile(stdin);
645   #endif
# Line 668 | Line 671 | parent_loop(void)
671                  if (!rmx_load_row(mop[i].imx.mtx, &mop[i].imx, mop[i].infp)) {
672                          if (cur_row > in_nrows) /* unknown #input rows? */
673                                  break;
674 <                        fprintf(stderr, "%s: parent read error at row %d\n",
674 >                        fprintf(stderr, "%s: load error at row %d\n",
675                                          mop[i].inspec, cur_row);
676                          return(0);
677                  }
# Line 676 | Line 679 | parent_loop(void)
679                  break;
680              for (i = 0; i < nmats; i++)
681                  if (writebuf(wfd, mop[i].imx.mtx, rmx_array_size(&mop[i].imx))
682 <                                        != rmx_array_size(&mop[i].imx))
682 >                                        != rmx_array_size(&mop[i].imx)) {
683 >                        fprintf(stderr, "%s: write error at row %d\n",
684 >                                        mop[i].inspec, cur_row);
685                          return(0);
686 +                }
687          }
688 <        i = close_processes(cproc, nchildren);
688 >        i = close_processes(cproc, nchildren);  /* collect family */
689          free(cproc); cproc = NULL; nchildren = 0;
690          if (i < 0) {
691 <                fputs("Warning: lost child in parent_loop()\n", stderr);
691 >                if (!nowarn)
692 >                        fputs("Warning: lost child process\n", stderr);
693                  return(1);
694          }
695          if (i > 0) {
# Line 690 | Line 697 | parent_loop(void)
697                  return(0);
698          }
699          return(1);                              /* return success! */
693 memerror:
694        fputs("Out of memory in parent_loop()\n", stderr);
695        exit(1);
700   }
701  
702   int
# Line 706 | Line 710 | combine_input(void)
710          int             co_set;
711          int             i;
712  
713 <        if (mcat && mcat_last &&
714 <                        !(tmp = rmx_alloc(1, res->imx.ncols, res->rmp->ncomp)))
715 <                goto memerror;
713 >        if (mcat_last && !(tmp = rmx_alloc(1, res->imx.ncols, res->rmp->ncomp))) {
714 >                fputs("Out of buffer space in combine_input()\n", stderr);
715 >                return(0);
716 >        }
717                                          /* figure out what the user set */
718          co_set = fundefined("co");
719          if (!co_set)
# Line 730 | Line 735 | combine_input(void)
735                  if (!rmx_load_row(mop[i].imx.mtx, &mop[i].imx, mop[i].infp)) {
736                          if (cur_row > in_nrows) /* unknown #input rows? */
737                                  break;
738 <                        fprintf(stderr, "%s: read error at row %d\n",
738 >                        fprintf(stderr, "%s: load error at row %d\n",
739                                          mop[i].inspec, cur_row);
740                          return(0);
741                  }
# Line 776 | Line 781 | combine_input(void)
781              }
782              rmx_free(mres); mres = NULL;
783              if (!rmx_write_data(res->rmp->mtx, res->rmp->ncomp,
784 <                                res->rmp->ncols, res->rmp->dtype, stdout))
784 >                                res->rmp->ncols, res->rmp->dtype, stdout) ||
785 >                                 (inchild >= 0 && fflush(stdout) == EOF)) {
786 >                fprintf(stderr, "Conversion/write error at row %d\n",
787 >                                cur_row);
788                  return(0);
789 <            if (inchild >= 0 && fflush(stdout) == EOF)
782 <                return(0);
789 >            }
790          }
791          return(inchild >= 0 || fflush(stdout) != EOF);
785 memerror:
786        fputs("Out of buffer space in combine_input()\n", stderr);
787        return(0);
792   multerror:
793 <        fputs("Unexpected matrix multiply error in combine_input()\n", stderr);
793 >        fputs("Unexpected matrix multiply error\n", stderr);
794          return(0);
795   }
796  
# Line 794 | Line 798 | int
798   output_loop(void)
799   {
800          const size_t    row_size = rmx_array_size(mop[nmats].rmp);
797        int             i = nmats;
801          int             cur_child = 0;
802 +        int             i = nmats;
803  
804 <        if (mop[nmats].rmp != &mop[nmats].imx)          /* output is split? */
801 <                rmx_reset(&mop[nmats].imx);
802 <        while (i-- > 0) {                               /* close input matrices */
803 <                fclose(mop[i].infp);            /* ! pclose() */
804 <                mop[i].infp = NULL;
804 >        while (i-- > 0) {                               /* free input buffers */
805                  rmx_reset(&mop[i].imx);
806                  if (mop[i].rmp != &mop[i].imx) {
807                          rmx_free(mop[i].rmp);
808                          mop[i].rmp = &mop[i].imx;
809                  }
810          }
811 +        if (mop[nmats].rmp != &mop[nmats].imx)          /* output is split? */
812 +                rmx_reset(&mop[nmats].imx);
813   #ifdef getc_unlocked
814          flockfile(stdout);                              /* we own this, now */
815   #endif
# Line 817 | Line 819 | output_loop(void)
819                  if (!rv)                                /* out of rows? */
820                          break;
821                  if (rv != row_size) {
822 <                        fputs("Read error in output loop\n", stderr);
822 >                        fputs("Read error\n", stderr);
823                          return(0);
824                  }                                       /* do final conversion */
825                  if (!rmx_write_data(mop[nmats].rmp->mtx, mop[nmats].rmp->ncomp,
826                                  mop[nmats].rmp->ncols, mop[nmats].rmp->dtype, stdout)) {
827 <                        fputs("Conversion/write error in output loop\n", stderr);
827 >                        fputs("Conversion/write error\n", stderr);
828                          return(0);
829                  }
830                  cur_child++;
831 <                cur_child *= (cur_child < inchild);
831 >                cur_child *= (cur_child < inchild);     /* loop over workers */
832          }
833          return(fflush(stdout) != EOF);
834   }
# Line 848 | Line 850 | resize_inparr(int n2alloc)
850  
851          if (n2alloc == nall)
852                  return;
853 <        for (i = nall; i > n2alloc; i--) {
853 >        for (i = nall; i-- > n2alloc; ) {
854                  rmx_reset(&mop[i].imx);
855                  if (mop[i].rmp != &mop[i].imx)
856                          rmx_free(mop[i].rmp);
# Line 943 | Line 945 | main(int argc, char *argv[])
945                                  }
946                                  break;
947                          case 'C':
948 +                                mcat_last = 0;
949                                  if (!n || isflt(argv[i+1]))
950                                          goto userr;
951                                  defCsym = mop[nmats].preop.csym = argv[++i];
952                                  mop[nmats].preop.clen = 0;
950                                mcat_last = 0;
953                                  break;
954                          case 'c':
955 +                                mcat_last = 0;
956                                  if (n && !isflt(argv[i+1])) {
957                                          mop[nmats].preop.csym = argv[++i];
958                                          mop[nmats].preop.clen = 0;
956                                        mcat_last = 0;
959                                          break;
960                                  }
961                                  if (n > MAXCOMP*MAXCOMP) n = MAXCOMP*MAXCOMP;
# Line 966 | Line 968 | main(int argc, char *argv[])
968                                          goto userr;
969                                  }
970                                  mop[nmats].preop.csym = NULL;
969                                mcat_last = 0;
971                                  break;
972                          case 'm':
973 +                                mcat_last = 1;
974                                  if (!n) goto userr;
975                                  if (argv[++i][0] == '-' && !argv[i][1]) {
976                                          if (stdin_used++) goto stdin_error;
977                                          mcat_spec = stdin_name;
978                                  } else
979                                          mcat_spec = argv[i];
978                                mcat_last = 1;
980                                  break;
981                          default:
982                                  fprintf(stderr, "%s: unknown option '%s'\n",

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines