--- ray/src/util/rmatrix.c 2014/08/27 13:33:47 2.8 +++ ray/src/util/rmatrix.c 2015/07/22 04:29:56 2.16 @@ -1,5 +1,5 @@ #ifndef lint -static const char RCSid[] = "$Id: rmatrix.c,v 2.8 2014/08/27 13:33:47 greg Exp $"; +static const char RCSid[] = "$Id: rmatrix.c,v 2.16 2015/07/22 04:29:56 greg Exp $"; #endif /* * General matrix operations. @@ -10,6 +10,7 @@ static const char RCSid[] = "$Id: rmatrix.c,v 2.8 2014 #include #include #include "resolu.h" +#include "rtprocess.h" #include "rmatrix.h" static char rmx_mismatch_warn[] = "WARNING: data type mismatch\n"; @@ -46,7 +47,9 @@ rmx_free(RMATRIX *rm) int rmx_newtype(int dtyp1, int dtyp2) { - if ((dtyp1==DTxyze) | (dtyp1==DTrgbe) && dtyp1 != dtyp2) + if ((dtyp1==DTxyze) | (dtyp1==DTrgbe) | + (dtyp2==DTxyze) | (dtyp2==DTrgbe) + && dtyp1 != dtyp2) return(0); if (dtyp1 < dtyp2) return(dtyp1); @@ -184,25 +187,31 @@ rmx_load_rgbe(RMATRIX *rm, FILE *fp) /* Load matrix from supported file type */ RMATRIX * -rmx_load(const char *fname) +rmx_load(const char *inspec) { FILE *fp = stdin; RMATRIX dinfo; RMATRIX *dnew; - if (fname == NULL) { /* reading from stdin? */ - fname = ""; + if (inspec == NULL) { /* reading from stdin? */ + inspec = ""; #ifdef _WIN32 _setmode(fileno(stdin), _O_BINARY); #endif + } else if (inspec[0] == '!') { + if ((fp = popen(inspec+1, "r")) == NULL) + return(NULL); +#ifdef _WIN32 + _setmode(fileno(fp), _O_BINARY); +#endif } else { - const char *sp = fname; /* check suffix */ + const char *sp = inspec; /* check suffix */ while (*sp) ++sp; - while (sp > fname && sp[-1] != '.') + while (sp > inspec && sp[-1] != '.') --sp; if (!strcasecmp(sp, "XML")) { /* assume it's a BSDF */ - CMATRIX *cm = cm_loadBTDF((char *)fname); + CMATRIX *cm = cm_loadBTDF((char *)inspec); if (cm == NULL) return(NULL); dnew = rmx_from_cmatrix(cm); @@ -210,7 +219,7 @@ rmx_load(const char *fname) return(dnew); } /* else open it ourselves */ - if ((fp = fopen(fname, "rb")) == NULL) + if ((fp = fopen(inspec, "rb")) == NULL) return(NULL); } #ifdef getc_unlocked @@ -246,6 +255,7 @@ rmx_load(const char *fname) case DTascii: if (!rmx_load_ascii(dnew, fp)) goto loaderr; + dnew->dtype = DTascii; /* should leave double? */ break; case DTfloat: if (!rmx_load_float(dnew, fp)) @@ -266,11 +276,22 @@ rmx_load(const char *fname) default: goto loaderr; } - if (fp != stdin) - fclose(fp); + if (fp != stdin) { + if (inspec[0] == '!') + pclose(fp); + else + fclose(fp); + } +#ifdef getc_unlocked + else + funlockfile(fp); +#endif return(dnew); loaderr: /* should report error? */ - fclose(fp); + if (inspec[0] == '!') + pclose(fp); + else + fclose(fp); rmx_free(dnew); return(NULL); } @@ -356,7 +377,7 @@ rmx_write_rgbe(const RMATRIX *rm, FILE *fp) } /* Write matrix to file type indicated by dtype */ -long +int rmx_write(const RMATRIX *rm, int dtype, FILE *fp) { RMATRIX *mydm = NULL; @@ -371,7 +392,7 @@ rmx_write(const RMATRIX *rm, int dtype, FILE *fp) dtype = rm->dtype; else if ((dtype == DTrgbe) & (rm->dtype == DTxyze)) dtype = DTxyze; - else if ((dtype = DTxyze) & (rm->dtype == DTrgbe)) + else if ((dtype == DTxyze) & (rm->dtype == DTrgbe)) dtype = DTrgbe; if ((dtype != DTrgbe) & (dtype != DTxyze)) { fprintf(fp, "NROWS=%d\n", rm->nrows); @@ -409,7 +430,7 @@ rmx_write(const RMATRIX *rm, int dtype, FILE *fp) } ok &= (fflush(fp) == 0); rmx_free(mydm); - return(ftell(fp) * ok); /* return # bytes written */ + return(ok); } /* Allocate and assign square identity matrix with n components */ @@ -417,16 +438,14 @@ RMATRIX * rmx_identity(const int dim, const int n) { RMATRIX *rid = rmx_alloc(dim, dim, n); - int i; + int i, k; if (rid == NULL) return(NULL); - memset(rid->mtx, 0, sizeof(rid->mtx[0])*dim*dim); + memset(rid->mtx, 0, sizeof(rid->mtx[0])*n*dim*dim); for (i = dim; i--; ) - rmx_lval(rid,i,i,0) = 1; - for (i = n; --i; ) - memcpy(rid->mtx+i*(dim*dim), rid->mtx, - sizeof(rid->mtx[0])*dim*dim); + for (k = n; k--; ) + rmx_lval(rid,i,i,k) = 1; return(rid); } @@ -495,8 +514,7 @@ rmx_multiply(const RMATRIX *m1, const RMATRIX *m2) for (k = mres->ncomp; k--; ) { long double d = 0; for (h = m1->ncols; h--; ) - d += (long double)rmx_lval(m1,i,h,k) * - (long double)rmx_lval(m2,h,j,k); + d += rmx_lval(m1,i,h,k) * rmx_lval(m2,h,j,k); rmx_lval(mres,i,j,k) = (double)d; } return(mres);