--- ray/src/util/rmatrix.c 2014/09/18 23:20:12 2.10 +++ ray/src/util/rmatrix.c 2016/08/30 14:54:08 2.23 @@ -1,5 +1,5 @@ #ifndef lint -static const char RCSid[] = "$Id: rmatrix.c,v 2.10 2014/09/18 23:20:12 greg Exp $"; +static const char RCSid[] = "$Id: rmatrix.c,v 2.23 2016/08/30 14:54:08 greg Exp $"; #endif /* * General matrix operations. @@ -9,7 +9,10 @@ static const char RCSid[] = "$Id: rmatrix.c,v 2.10 201 #include #include #include +#include "rtio.h" +#include "platform.h" #include "resolu.h" +#include "paths.h" #include "rmatrix.h" static char rmx_mismatch_warn[] = "WARNING: data type mismatch\n"; @@ -46,7 +49,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); @@ -108,9 +113,7 @@ static int rmx_load_ascii(RMATRIX *rm, FILE *fp) { int i, j, k; -#ifdef _WIN32 - _setmode(fileno(fp), _O_TEXT); -#endif + for (i = 0; i < rm->nrows; i++) for (j = 0; j < rm->ncols; j++) for (k = 0; k < rm->ncomp; k++) @@ -131,7 +134,7 @@ rmx_load_float(RMATRIX *rm, FILE *fp) } for (i = 0; i < rm->nrows; i++) for (j = 0; j < rm->ncols; j++) { - if (fread(val, sizeof(val[0]), rm->ncomp, fp) != rm->ncomp) + if (getbinary(val, sizeof(val[0]), rm->ncomp, fp) != rm->ncomp) return(0); for (k = rm->ncomp; k--; ) rmx_lval(rm,i,j,k) = val[k]; @@ -151,7 +154,7 @@ rmx_load_double(RMATRIX *rm, FILE *fp) } for (i = 0; i < rm->nrows; i++) for (j = 0; j < rm->ncols; j++) { - if (fread(val, sizeof(val[0]), rm->ncomp, fp) != rm->ncomp) + if (getbinary(val, sizeof(val[0]), rm->ncomp, fp) != rm->ncomp) return(0); for (k = rm->ncomp; k--; ) rmx_lval(rm,i,j,k) = val[k]; @@ -184,33 +187,36 @@ 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 = ""; -#ifdef _WIN32 - _setmode(fileno(stdin), _O_BINARY); -#endif + if (inspec == NULL) { /* reading from stdin? */ + inspec = ""; + SET_FILE_BINARY(stdin); + } else if (inspec[0] == '!') { + if ((fp = popen(inspec+1, "r")) == NULL) + return(NULL); + SET_FILE_BINARY(stdin); } 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); cm_free(cm); + dnew->dtype = DTascii; return(dnew); } /* else open it ourselves */ - if ((fp = fopen(fname, "rb")) == NULL) + if ((fp = fopen(inspec, "rb")) == NULL) return(NULL); } #ifdef getc_unlocked @@ -244,8 +250,10 @@ rmx_load(const char *fname) dnew->info = dinfo.info; switch (dinfo.dtype) { case DTascii: + SET_FILE_TEXT(stdin); 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 +274,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); } @@ -279,9 +298,7 @@ static int rmx_write_ascii(const RMATRIX *rm, FILE *fp) { int i, j, k; -#ifdef _WIN32 - _setmode(fileno(fp), _O_TEXT); -#endif + for (i = 0; i < rm->nrows; i++) { for (j = 0; j < rm->ncols; j++) { for (k = 0; k < rm->ncomp; k++) @@ -307,7 +324,7 @@ rmx_write_float(const RMATRIX *rm, FILE *fp) for (j = 0; j < rm->ncols; j++) { for (k = rm->ncomp; k--; ) val[k] = (float)rmx_lval(rm,i,j,k); - if (fwrite(val, sizeof(val[0]), rm->ncomp, fp) != rm->ncomp) + if (putbinary(val, sizeof(val[0]), rm->ncomp, fp) != rm->ncomp) return(0); } return(1); @@ -327,7 +344,7 @@ rmx_write_double(const RMATRIX *rm, FILE *fp) for (j = 0; j < rm->ncols; j++) { for (k = rm->ncomp; k--; ) val[k] = rmx_lval(rm,i,j,k); - if (fwrite(val, sizeof(val[0]), rm->ncomp, fp) != rm->ncomp) + if (putbinary(val, sizeof(val[0]), rm->ncomp, fp) != rm->ncomp) return(0); } return(1); @@ -364,6 +381,9 @@ rmx_write(const RMATRIX *rm, int dtype, FILE *fp) if ((rm == NULL) | (fp == NULL)) return(0); +#ifdef getc_unlocked + flockfile(fp); +#endif /* complete header */ if (rm->info) fputs(rm->info, fp); @@ -408,6 +428,9 @@ rmx_write(const RMATRIX *rm, int dtype, FILE *fp) return(0); } ok &= (fflush(fp) == 0); +#ifdef getc_unlocked + funlockfile(fp); +#endif rmx_free(mydm); return(ok); } @@ -417,16 +440,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); }