ViewVC Help

View File | Revision Log | Show Annotations | Download File | Root Listing

root/Development/ray/src/util/rmatrix.c

Comparing ray/src/util/rmatrix.c (file contents):
Revision 2.89 by greg, Fri Apr 4 18:06:48 2025 UTC vs.
Revision 2.102 by greg, Thu Oct 30 16:47:54 2025 UTC

#	Line 20 | Line 20 | static const char      rmx_mismatch_warn[] = "WARNING: data
20
21		/* Initialize a RMATRIX struct but don't allocate array space */
22		RMATRIX *
23	<	rmx_new(int nr, int nc, int n)
23	>	rmx_new(int nr, int nc, int ncomp)
24		{
25		RMATRIX *dnew;
26
27	<	if (n <= 0)
27	>	if (ncomp <= 0)
28		return(NULL);
29
30		dnew = (RMATRIX *)calloc(1, sizeof(RMATRIX));
#	Line 34 | Line 34 | rmx_new(int nr, int nc, int n)
34		dnew->dtype = DTrmx_native;
35		dnew->nrows = nr;
36		dnew->ncols = nc;
37	<	dnew->ncomp = n;
37	>	dnew->ncomp = ncomp;
38		setcolor(dnew->cexp, 1.f, 1.f, 1.f);
39		memcpy(dnew->wlpart, WLPART, sizeof(dnew->wlpart));
40
#	Line 57 | Line 57 | rmx_prepare(RMATRIX *rm)
57
58		/* Call rmx_new() and rmx_prepare() */
59		RMATRIX *
60	<	rmx_alloc(int nr, int nc, int n)
60	>	rmx_alloc(int nr, int nc, int ncomp)
61		{
62	<	RMATRIX *dnew = rmx_new(nr, nc, n);
62	>	RMATRIX *dnew = rmx_new(nr, nc, ncomp);
63
64		if (!rmx_prepare(dnew)) {
65		rmx_free(dnew);
#	Line 278 | Line 278 | rmx_load_rgbe(rmx_dtype *drp, const RMATRIX *rm, FILE
278		return(1);
279		}
280
281	+	#if DTrmx_native==DTfloat
282		static int
283		rmx_load_spec(rmx_dtype *drp, const RMATRIX *rm, FILE *fp)
284		{
285		COLRV   *scan;
286	+	int     j;
287	+
288	+	if ((rm->ncomp < 3) | (rm->ncomp > MAXCOMP))
289	+	return(0);
290	+	scan = (COLRV *)tempbuffer((rm->ncomp+1)*rm->ncols);
291	+	if (!scan)
292	+	return(0);
293	+	if (freadscolrs(scan, rm->ncomp, rm->ncols, fp) < 0)
294	+	return(0);
295	+	for (j = 0; j < rm->ncols; j++, drp += rm->ncomp)
296	+	scolr2scolor(drp, scan+j*(rm->ncomp+1), rm->ncomp);
297	+	return(1);
298	+	}
299	+	#else
300	+	static int
301	+	rmx_load_spec(rmx_dtype *drp, const RMATRIX *rm, FILE *fp)
302	+	{
303	+	COLRV   *scan;
304		COLORV  scol[MAXCOMP];
305		int     j, k;
306
#	Line 299 | Line 318 | rmx_load_spec(rmx_dtype *drp, const RMATRIX *rm, FILE
318		}
319		return(1);
320		}
321	+	#endif
322
323		/* Read matrix header from input stream (cannot be XML) */
324		int
#	Line 389 | Line 409 | rmx_load_data(RMATRIX *rm, FILE *fp)
409
410		/* Load matrix from supported file type */
411		RMATRIX *
412	<	rmx_load(const char *inspec, RMPref rmp)
412	>	rmx_load(const char *inspec)
413		{
414		FILE            *fp;
415		RMATRIX         *dnew;
#	Line 403 | Line 423 | rmx_load(const char *inspec, RMPref rmp)
423		fp = stdin;
424		else if (inspec[0] == '!')
425		fp = popen(inspec+1, "r");
426	<	else {                                  /* check suffix */
407	<	const char      *sp = strrchr(inspec, '.');
408	<	if (sp > inspec && !strcasecmp(sp+1, "XML")) {  /* BSDF? */
409	<	CMATRIX *cm = rmp==RMPnone ? (CMATRIX *)NULL :
410	<	rmp==RMPtrans ? cm_loadBTDF(inspec) :
411	<	cm_loadBRDF(inspec, rmp==RMPreflB) ;
412	<	if (!cm)
413	<	return(NULL);
414	<	dnew = rmx_from_cmatrix(cm);
415	<	cm_free(cm);
416	<	dnew->dtype = DTascii;
417	<	return(dnew);           /* return here */
418	<	}                               /* else open it ourselves */
426	>	else
427		fp = fopen(inspec, "r");
420	–	}
428		if (!fp) {
429		fprintf(stderr, "Cannot open for reading: %s\n", inspec);
430		return(NULL);
#	Line 437 | Line 444 | rmx_load(const char *inspec, RMPref rmp)
444
445		if (fp != stdin) {                      /* close input stream */
446		if (inspec[0] == '!')
447	<	pclose(fp);
447	>	ok &= pclose(fp)==0;
448		else
449		fclose(fp);
450		}
#	Line 501 | Line 508 | rmx_write_double(const rmx_dtype *dp, int len, FILE *f
508		#endif
509
510		static int
511	<	rmx_write_ascii(const rmx_dtype *dp, int nc, int len, FILE *fp)
511	>	rmx_write_ascii(const rmx_dtype *dp, int ncomp, int len, FILE *fp)
512		{
513		while (len-- > 0) {
514	<	int     k = nc;
514	>	int     k = ncomp;
515		while (k-- > 0)
516		fprintf(fp, " %.7e", *dp++);
517		fputc('\t', fp);
#	Line 513 | Line 520 | rmx_write_ascii(const rmx_dtype *dp, int nc, int len,
520		}
521
522		static int
523	<	rmx_write_rgbe(const rmx_dtype *dp, int nc, int len, FILE *fp)
523	>	rmx_write_rgbe(const rmx_dtype *dp, int ncomp, int len, FILE *fp)
524		{
525		COLR    *scan;
526		int     j;
527
528	<	if ((nc != 1) & (nc != 3)) return(0);
528	>	if ((ncomp != 1) & (ncomp != 3)) return(0);
529		scan = (COLR *)tempbuffer(sizeof(COLR)*len);
530		if (!scan) return(0);
531
532	<	for (j = 0; j < len; j++, dp += nc)
533	<	if (nc == 1)
532	>	for (j = 0; j < len; j++, dp += ncomp)
533	>	if (ncomp == 1)
534		setcolr(scan[j], dp[0], dp[0], dp[0]);
535		else
536		setcolr(scan[j], dp[0], dp[1], dp[2]);
#	Line 531 | Line 538 | rmx_write_rgbe(const rmx_dtype *dp, int nc, int len, F
538		return(fwritecolrs(scan, len, fp) >= 0);
539		}
540
541	+	#if DTrmx_native==DTfloat
542		static int
543	<	rmx_write_spec(const rmx_dtype *dp, int nc, int len, FILE *fp)
543	>	rmx_write_spec(const rmx_dtype *dp, int ncomp, int len, FILE *fp)
544		{
545		COLRV   *scan;
546	+	int     j;
547	+
548	+	if ((ncomp < 3) | (ncomp > MAXCOMP)) return(0);
549	+	scan = (COLRV *)tempbuffer((ncomp+1)*len);
550	+	if (!scan) return(0);
551	+	for (j = 0; j < len; j++, dp += ncomp)
552	+	scolor2scolr(scan+j*(ncomp+1), dp, ncomp);
553	+
554	+	return(fwritescolrs(scan, ncomp, len, fp) >= 0);
555	+	}
556	+	#else
557	+	static int
558	+	rmx_write_spec(const rmx_dtype *dp, int ncomp, int len, FILE *fp)
559	+	{
560	+	COLRV   *scan;
561		COLORV  scol[MAXCOMP];
562		int     j, k;
563
564	<	if ((nc < 3) | (nc > MAXCOMP)) return(0);
565	<	scan = (COLRV *)tempbuffer((nc+1)*len);
564	>	if ((ncomp < 3) | (ncomp > MAXCOMP)) return(0);
565	>	scan = (COLRV *)tempbuffer((ncomp+1)*len);
566		if (!scan) return(0);
567	<	for (j = 0; j < len; j++, dp += nc) {
568	<	for (k = nc; k--; )
567	>	for (j = 0; j < len; j++, dp += ncomp) {
568	>	for (k = ncomp; k--; )
569		scol[k] = dp[k];
570	<	scolor2scolr(scan+j*(nc+1), scol, nc);
570	>	scolor2scolr(scan+j*(ncomp+1), scol, ncomp);
571		}
572	<	return(fwritescolrs(scan, nc, len, fp) >= 0);
572	>	return(fwritescolrs(scan, ncomp, len, fp) >= 0);
573		}
574	+	#endif
575
576		/* Check if CIE XYZ primaries were specified */
577		static int
#	Line 574 | Line 598 | rmx_write_header(const RMATRIX *rm, int dtype, FILE *f
598		return(0);
599		if (rm->info)
600		fputs(rm->info, fp);
601	<	if (dtype == DTfromHeader)
601	>	if (dtype == DTfromHeader) {
602		dtype = rm->dtype;
603	<	else if (dtype == DTrgbe && (rm->dtype == DTxyze ||
603	>	#if DTrmx_native==DTfloat
604	>	if (dtype == DTdouble)          /* but stored as float? */
605	>	dtype = DTfloat;
606	>	#endif
607	>	} else if (dtype == DTrgbe && (rm->dtype == DTxyze ||
608		findCIEprims(rm->info)))
609		dtype = DTxyze;
610		else if ((dtype == DTxyze) & (rm->dtype == DTrgbe))
#	Line 620 | Line 648 | rmx_write_header(const RMATRIX *rm, int dtype, FILE *f
648
649		/* Write out matrix data (usually by row) */
650		int
651	<	rmx_write_data(const rmx_dtype *dp, int nc, int len, int dtype, FILE *fp)
651	>	rmx_write_data(const rmx_dtype *dp, int ncomp, int len, int dtype, FILE *fp)
652		{
653		switch (dtype) {
654		#if DTrmx_native==DTdouble
655		case DTfloat:
656	<	return(rmx_write_float(dp, nc*len, fp));
656	>	return(rmx_write_float(dp, ncomp*len, fp));
657		#else
658		case DTdouble:
659	<	return(rmx_write_double(dp, nc*len, fp));
659	>	return(rmx_write_double(dp, ncomp*len, fp));
660		#endif
661		case DTrmx_native:
662	<	return(putbinary(dp, sizeof(*dp)*nc, len, fp) == len);
662	>	return(putbinary(dp, sizeof(*dp)*ncomp, len, fp) == len);
663		case DTascii:
664	<	return(rmx_write_ascii(dp, nc, len, fp));
664	>	return(rmx_write_ascii(dp, ncomp, len, fp));
665		case DTrgbe:
666		case DTxyze:
667	<	return(rmx_write_rgbe(dp, nc, len, fp));
667	>	return(rmx_write_rgbe(dp, ncomp, len, fp));
668		case DTspec:
669	<	return(rmx_write_spec(dp, nc, len, fp));
669	>	return(rmx_write_spec(dp, ncomp, len, fp));
670		}
671		return(0);
672		}
#	Line 754 | Line 782 | rmx_transfer_data(RMATRIX *rdst, RMATRIX *rsrc, int do
782		int
783		rmx_transpose(RMATRIX *rm)
784		{
785	<	RMATRIX dnew;
786	<	int     i, j;
785	>	uby8            *bmap;
786	>	rmx_dtype       val[MAXCOMP];
787	>	RMATRIX         dold;
788	>	int             i, j;
789
790		if (!rm || !rm->mtx | (rm->ncomp > MAXCOMP))
791		return(0);
#	Line 768 | Line 798 | rmx_transpose(RMATRIX *rm)
798		return(1);
799		}
800		if (rm->nrows == rm->ncols) {   /* square matrix case */
801	<	rmx_dtype       val[MAXCOMP];
802	<	for (j = rm->ncols; j--; )
803	<	for (i = rm->nrows; i--; ) {
774	<	if (i == j) continue;
775	<	memcpy(val, rmx_val(rm,i,j),
801	>	for (i = rm->nrows; --i > 0; )
802	>	for (j = i; j-- > 0; ) {
803	>	memcpy(val, rmx_val(rm,i,j),
804		sizeof(rmx_dtype)*rm->ncomp);
805	<	memcpy(rmx_lval(rm,i,j), rmx_val(rm,j,i),
805	>	memcpy(rmx_lval(rm,i,j), rmx_val(rm,j,i),
806		sizeof(rmx_dtype)*rm->ncomp);
807	<	memcpy(rmx_val(rm,j,i), val,
807	>	memcpy(rmx_lval(rm,j,i), val,
808		sizeof(rmx_dtype)*rm->ncomp);
809	<	}
810	<	return(1);
809	>	}
810	>	return(1);
811		}
812	<	memset(&dnew, 0, sizeof(dnew));
813	<	dnew.ncols = rm->nrows; dnew.nrows = rm->ncols;
814	<	dnew.ncomp = rm->ncomp;
815	<	if (!rmx_prepare(&dnew))
812	>	#define bmbyte(r,c)     bmap[((r)*rm->ncols+(c))>>3]
813	>	#define bmbit(r,c)      (1 << ((r)*rm->ncols+(c) & 7))
814	>	#define bmop(r,c, op)   (bmbyte(r,c) op bmbit(r,c))
815	>	#define bmtest(r,c)     bmop(r,c,&)
816	>	#define bmset(r,c)      bmop(r,c,|=)
817	>	/* loop completion bitmap */
818	>	bmap = (uby8 *)calloc(((size_t)rm->nrows*rm->ncols+7)>>3, 1);
819	>	if (!bmap)
820		return(0);
821	<	rmx_addinfo(&dnew, rm->info);
822	<	dnew.dtype = rm->dtype;
823	<	copycolor(dnew.cexp, rm->cexp);
824	<	memcpy(dnew.wlpart, rm->wlpart, sizeof(dnew.wlpart));
825	<	for (j = dnew.ncols; j--; )
826	<	for (i = dnew.nrows; i--; )
827	<	memcpy(rmx_lval(&dnew,i,j), rmx_val(rm,j,i),
828	<	sizeof(rmx_dtype)*dnew.ncomp);
829	<	/* and reassign result */
830	<	return(rmx_transfer_data(rm, &dnew, 1));
821	>	dold = *rm;
822	>	rm->ncols = dold.nrows; rm->nrows = dold.ncols;
823	>	for (i = rm->nrows; i--; )      /* try every starting point */
824	>	for (j = rm->ncols; j--; ) {
825	>	int     i0, j0;
826	>	int     i1 = i;
827	>	size_t  j1 = j;
828	>	if (bmtest(i, j))
829	>	continue;       /* traversed loop earlier */
830	>	memcpy(val, rmx_val(rm,i,j),
831	>	sizeof(rmx_dtype)*rm->ncomp);
832	>	for ( ; ; ) {           /* new transpose loop */
833	>	const rmx_dtype     *ds;
834	>	i0 = i1; j0 = j1;
835	>	ds = rmx_val(&dold, j0, i0);
836	>	j1 = (ds - dold.mtx)/dold.ncomp;
837	>	i1 = j1 / rm->ncols;
838	>	j1 -= (size_t)i1*rm->ncols;
839	>	bmset(i1, j1);      /* mark as done */
840	>	if ((i1 == i) & (j1 == j))
841	>	break;          /* back at start */
842	>	memcpy(rmx_lval(rm,i0,j0), ds,
843	>	sizeof(rmx_dtype)*rm->ncomp);
844	>	}                       /* complete the loop */
845	>	memcpy(rmx_lval(rm,i0,j0), val,
846	>	sizeof(rmx_dtype)*rm->ncomp);
847	>	}
848	>	free(bmap);                     /* all done! */
849	>	return(1);
850	>	#undef  bmbyte
851	>	#undef  bmbit
852	>	#undef  bmop
853	>	#undef  bmtest
854	>	#undef  bmset
855		}
856
857		/* Multiply (concatenate) two matrices and allocate the result */
#	Line 975 | Line 1031 | rmx_transform(const RMATRIX *msrc, int n, const double
1031		return(dnew);
1032		}
1033
978	–	/* Convert a color matrix to newly allocated RMATRIX buffer */
979	–	RMATRIX *
980	–	rmx_from_cmatrix(const CMATRIX *cm)
981	–	{
982	–	RMATRIX *dnew;
983	–
984	–	if (!cm)
985	–	return(NULL);
986	–	dnew = rmx_alloc(cm->nrows, cm->ncols, 3);
987	–	if (!dnew)
988	–	return(NULL);
989	–
990	–	dnew->dtype = sizeof(COLORV)==sizeof(float) ?
991	–	DTfloat : DTdouble;
992	–
993	–	if (sizeof(COLORV) == sizeof(rmx_dtype)) {
994	–	memcpy(dnew->mtx, cm->cmem, rmx_array_size(dnew));
995	–	} else {
996	–	int     i, j;
997	–	for (i = dnew->nrows; i--; )
998	–	for (j = dnew->ncols; j--; ) {
999	–	const COLORV    *cv = cm_lval(cm,i,j);
1000	–	rmx_dtype       *dp = rmx_lval(dnew,i,j);
1001	–	dp[0] = cv[0];
1002	–	dp[1] = cv[1];
1003	–	dp[2] = cv[2];
1004	–	}
1005	–	}
1006	–	return(dnew);
1007	–	}
1008	–
1009	–	/* Convert general matrix to newly allocated CMATRIX buffer */
1010	–	CMATRIX *
1011	–	cm_from_rmatrix(const RMATRIX *rm)
1012	–	{
1013	–	CMATRIX *cnew;
1014	–
1015	–	if (!rm || !rm->mtx | (rm->ncomp == 2) | (rm->ncomp > MAXCOMP))
1016	–	return(NULL);
1017	–	cnew = cm_alloc(rm->nrows, rm->ncols);
1018	–	if (!cnew)
1019	–	return(NULL);
1020	–	if ((sizeof(COLORV) == sizeof(rmx_dtype)) & (rm->ncomp == 3)) {
1021	–	memcpy(cnew->cmem, rm->mtx, rmx_array_size(rm));
1022	–	} else {
1023	–	int     i, j;
1024	–	for (i = cnew->nrows; i--; )
1025	–	for (j = cnew->ncols; j--; ) {
1026	–	const rmx_dtype *dp = rmx_val(rm,i,j);
1027	–	COLORV          *cv = cm_lval(cnew,i,j);
1028	–	switch (rm->ncomp) {
1029	–	case 3:
1030	–	setcolor(cv, dp[0], dp[1], dp[2]);
1031	–	break;
1032	–	case 1:
1033	–	setcolor(cv, dp[0], dp[0], dp[0]);
1034	–	break;
1035	–	default: {
1036	–	COLORV  scol[MAXCOMP];
1037	–	int     k;
1038	–	for (k = rm->ncomp; k--; )
1039	–	scol[k] = dp[k];
1040	–	scolor2color(cv, scol, rm->ncomp, rm->wlpart);
1041	–	} break;
1042	–	}
1043	–	}
1044	–	}
1045	–	return(cnew);
1046	–	}

Diff Legend

–	Removed lines
+	Added lines
<	Changed lines (old)
>	Changed lines (new)