--- ray/src/util/rmatrix.c 2014/08/02 17:10:43 2.6 +++ ray/src/util/rmatrix.c 2015/05/04 20:53:21 2.13 @@ -1,5 +1,5 @@ #ifndef lint -static const char RCSid[] = "$Id: rmatrix.c,v 2.6 2014/08/02 17:10:43 greg Exp $"; +static const char RCSid[] = "$Id: rmatrix.c,v 2.13 2015/05/04 20:53:21 greg Exp $"; #endif /* * General matrix operations. @@ -59,9 +59,10 @@ rmx_addinfo(RMATRIX *rm, const char *info) { if (!info || !*info) return(0); - if (!rm->info) + if (!rm->info) { rm->info = (char *)malloc(strlen(info)+1); - else + if (rm->info) rm->info[0] = '\0'; + } else rm->info = (char *)realloc(rm->info, strlen(rm->info)+strlen(info)+1); if (!rm->info) @@ -183,22 +184,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); @@ -206,7 +216,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 @@ -242,6 +252,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)) @@ -262,11 +273,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); } @@ -352,7 +374,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; @@ -367,7 +389,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); @@ -405,7 +427,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 */ @@ -413,16 +435,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); } @@ -488,9 +508,9 @@ rmx_multiply(const RMATRIX *m1, const RMATRIX *m2) rmx_addinfo(mres, rmx_mismatch_warn); for (i = mres->nrows; i--; ) for (j = mres->ncols; j--; ) - for (h = m1->ncols; h--; ) { + for (k = mres->ncomp; k--; ) { long double d = 0; - for (k = mres->ncomp; k--; ) + for (h = m1->ncols; h--; ) d += (long double)rmx_lval(m1,i,h,k) * (long double)rmx_lval(m2,h,j,k); rmx_lval(mres,i,j,k) = (double)d;