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.90 by greg, Fri Apr 4 22:47:56 2025 UTC vs.
Revision 2.103 by greg, Thu Oct 30 17:26:45 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 255 | Line 255 | rmx_load_double(rmx_dtype *drp, const RMATRIX *rm, FIL
255		return(1);
256		}
257
258	+	#if DTrmx_native==DTfloat
259	+	#define rmx_load_spec(dp,rm,fp) (freadsscan(dp,(rm)->ncomp,(rm)->ncols,fp) >= 0)
260	+	#else
261		static int
259	–	rmx_load_rgbe(rmx_dtype *drp, const RMATRIX *rm, FILE *fp)
260	–	{
261	–	COLR    *scan;
262	–	COLOR   col;
263	–	int     j;
264	–
265	–	if (rm->ncomp != 3)
266	–	return(0);
267	–	scan = (COLR *)tempbuffer(sizeof(COLR)*rm->ncols);
268	–	if (!scan)
269	–	return(0);
270	–	if (freadcolrs(scan, rm->ncols, fp) < 0)
271	–	return(0);
272	–	for (j = 0; j < rm->ncols; j++) {
273	–	colr_color(col, scan[j]);
274	–	*drp++ = colval(col,RED);
275	–	*drp++ = colval(col,GRN);
276	–	*drp++ = colval(col,BLU);
277	–	}
278	–	return(1);
279	–	}
280	–
281	–	static int
262		rmx_load_spec(rmx_dtype *drp, const RMATRIX *rm, FILE *fp)
263		{
264		COLRV   *scan;
#	Line 299 | Line 279 | rmx_load_spec(rmx_dtype *drp, const RMATRIX *rm, FILE
279		}
280		return(1);
281		}
282	+	#endif
283
284		/* Read matrix header from input stream (cannot be XML) */
285		int
#	Line 345 | Line 326 | rmx_load_row(rmx_dtype *drp, const RMATRIX *rm, FILE *
326		return(rmx_load_double(drp, rm, fp));
327		case DTrgbe:
328		case DTxyze:
348	–	return(rmx_load_rgbe(drp, rm, fp));
329		case DTspec:
330		return(rmx_load_spec(drp, rm, fp));
331		default:
#	Line 389 | Line 369 | rmx_load_data(RMATRIX *rm, FILE *fp)
369
370		/* Load matrix from supported file type */
371		RMATRIX *
372	<	rmx_load(const char *inspec, RMPref rmp)
372	>	rmx_load(const char *inspec)
373		{
374		FILE            *fp;
375		RMATRIX         *dnew;
#	Line 403 | Line 383 | rmx_load(const char *inspec, RMPref rmp)
383		fp = stdin;
384		else if (inspec[0] == '!')
385		fp = popen(inspec+1, "r");
386	<	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 */
386	>	else
387		fp = fopen(inspec, "r");
420	–	}
388		if (!fp) {
389		fprintf(stderr, "Cannot open for reading: %s\n", inspec);
390		return(NULL);
#	Line 437 | Line 404 | rmx_load(const char *inspec, RMPref rmp)
404
405		if (fp != stdin) {                      /* close input stream */
406		if (inspec[0] == '!')
407	<	pclose(fp);
407	>	ok &= pclose(fp)==0;
408		else
409		fclose(fp);
410		}
#	Line 501 | Line 468 | rmx_write_double(const rmx_dtype *dp, int len, FILE *f
468		#endif
469
470		static int
471	<	rmx_write_ascii(const rmx_dtype *dp, int nc, int len, FILE *fp)
471	>	rmx_write_ascii(const rmx_dtype *dp, int ncomp, int len, FILE *fp)
472		{
473		while (len-- > 0) {
474	<	int     k = nc;
474	>	int     k = ncomp;
475		while (k-- > 0)
476		fprintf(fp, " %.7e", *dp++);
477		fputc('\t', fp);
#	Line 513 | Line 480 | rmx_write_ascii(const rmx_dtype *dp, int nc, int len,
480		}
481
482		static int
483	<	rmx_write_rgbe(const rmx_dtype *dp, int nc, int len, FILE *fp)
483	>	rmx_write_rgbe(const rmx_dtype *dp, int ncomp, int len, FILE *fp)
484		{
485		COLR    *scan;
486		int     j;
487
488	<	if ((nc != 1) & (nc != 3)) return(0);
488	>	if ((ncomp != 1) & (ncomp != 3)) return(0);
489		scan = (COLR *)tempbuffer(sizeof(COLR)*len);
490		if (!scan) return(0);
491
492	<	for (j = 0; j < len; j++, dp += nc)
493	<	if (nc == 1)
492	>	for (j = 0; j < len; j++, dp += ncomp)
493	>	if (ncomp == 1)
494		setcolr(scan[j], dp[0], dp[0], dp[0]);
495		else
496		setcolr(scan[j], dp[0], dp[1], dp[2]);
#	Line 531 | Line 498 | rmx_write_rgbe(const rmx_dtype *dp, int nc, int len, F
498		return(fwritecolrs(scan, len, fp) >= 0);
499		}
500
501	+	#if DTrmx_native==DTfloat
502	+	#define rmx_write_spec(dp,nc,ln,fp)     (fwritesscan(dp,nc,ln,fp) >= 0)
503	+	#else
504		static int
505	<	rmx_write_spec(const rmx_dtype *dp, int nc, int len, FILE *fp)
505	>	rmx_write_spec(const rmx_dtype *dp, int ncomp, int len, FILE *fp)
506		{
507		COLRV   *scan;
508		COLORV  scol[MAXCOMP];
509		int     j, k;
510
511	<	if ((nc < 3) | (nc > MAXCOMP)) return(0);
512	<	scan = (COLRV *)tempbuffer((nc+1)*len);
511	>	if ((ncomp < 3) | (ncomp > MAXCOMP)) return(0);
512	>	scan = (COLRV *)tempbuffer((ncomp+1)*len);
513		if (!scan) return(0);
514	<	for (j = 0; j < len; j++, dp += nc) {
515	<	for (k = nc; k--; )
514	>	for (j = 0; j < len; j++, dp += ncomp) {
515	>	for (k = ncomp; k--; )
516		scol[k] = dp[k];
517	<	scolor2scolr(scan+j*(nc+1), scol, nc);
517	>	scolor2scolr(scan+j*(ncomp+1), scol, ncomp);
518		}
519	<	return(fwritescolrs(scan, nc, len, fp) >= 0);
519	>	return(fwritescolrs(scan, ncomp, len, fp) >= 0);
520		}
521	+	#endif
522
523		/* Check if CIE XYZ primaries were specified */
524		static int
#	Line 624 | Line 595 | rmx_write_header(const RMATRIX *rm, int dtype, FILE *f
595
596		/* Write out matrix data (usually by row) */
597		int
598	<	rmx_write_data(const rmx_dtype *dp, int nc, int len, int dtype, FILE *fp)
598	>	rmx_write_data(const rmx_dtype *dp, int ncomp, int len, int dtype, FILE *fp)
599		{
600		switch (dtype) {
601		#if DTrmx_native==DTdouble
602		case DTfloat:
603	<	return(rmx_write_float(dp, nc*len, fp));
603	>	return(rmx_write_float(dp, ncomp*len, fp));
604		#else
605		case DTdouble:
606	<	return(rmx_write_double(dp, nc*len, fp));
606	>	return(rmx_write_double(dp, ncomp*len, fp));
607		#endif
608		case DTrmx_native:
609	<	return(putbinary(dp, sizeof(*dp)*nc, len, fp) == len);
609	>	return(putbinary(dp, sizeof(*dp)*ncomp, len, fp) == len);
610		case DTascii:
611	<	return(rmx_write_ascii(dp, nc, len, fp));
611	>	return(rmx_write_ascii(dp, ncomp, len, fp));
612		case DTrgbe:
613		case DTxyze:
614	<	return(rmx_write_rgbe(dp, nc, len, fp));
614	>	return(rmx_write_rgbe(dp, ncomp, len, fp));
615		case DTspec:
616	<	return(rmx_write_spec(dp, nc, len, fp));
616	>	return(rmx_write_spec(dp, ncomp, len, fp));
617		}
618		return(0);
619		}
#	Line 758 | Line 729 | rmx_transfer_data(RMATRIX *rdst, RMATRIX *rsrc, int do
729		int
730		rmx_transpose(RMATRIX *rm)
731		{
732	<	RMATRIX dnew;
733	<	int     i, j;
732	>	uby8            *bmap;
733	>	rmx_dtype       val[MAXCOMP];
734	>	RMATRIX         dold;
735	>	int             i, j;
736
737		if (!rm || !rm->mtx | (rm->ncomp > MAXCOMP))
738		return(0);
#	Line 772 | Line 745 | rmx_transpose(RMATRIX *rm)
745		return(1);
746		}
747		if (rm->nrows == rm->ncols) {   /* square matrix case */
748	<	rmx_dtype       val[MAXCOMP];
749	<	for (j = rm->ncols; j--; )
750	<	for (i = rm->nrows; i--; ) {
778	<	if (i == j) continue;
779	<	memcpy(val, rmx_val(rm,i,j),
748	>	for (i = rm->nrows; --i > 0; )
749	>	for (j = i; j-- > 0; ) {
750	>	memcpy(val, rmx_val(rm,i,j),
751		sizeof(rmx_dtype)*rm->ncomp);
752	<	memcpy(rmx_lval(rm,i,j), rmx_val(rm,j,i),
752	>	memcpy(rmx_lval(rm,i,j), rmx_val(rm,j,i),
753		sizeof(rmx_dtype)*rm->ncomp);
754	<	memcpy(rmx_val(rm,j,i), val,
754	>	memcpy(rmx_lval(rm,j,i), val,
755		sizeof(rmx_dtype)*rm->ncomp);
756	<	}
757	<	return(1);
756	>	}
757	>	return(1);
758		}
759	<	memset(&dnew, 0, sizeof(dnew));
760	<	dnew.ncols = rm->nrows; dnew.nrows = rm->ncols;
761	<	dnew.ncomp = rm->ncomp;
762	<	if (!rmx_prepare(&dnew))
759	>	#define bmbyte(r,c)     bmap[((r)*rm->ncols+(c))>>3]
760	>	#define bmbit(r,c)      (1 << ((r)*rm->ncols+(c) & 7))
761	>	#define bmop(r,c, op)   (bmbyte(r,c) op bmbit(r,c))
762	>	#define bmtest(r,c)     bmop(r,c,&)
763	>	#define bmset(r,c)      bmop(r,c,|=)
764	>	/* loop completion bitmap */
765	>	bmap = (uby8 *)calloc(((size_t)rm->nrows*rm->ncols+7)>>3, 1);
766	>	if (!bmap)
767		return(0);
768	<	rmx_addinfo(&dnew, rm->info);
769	<	dnew.dtype = rm->dtype;
770	<	copycolor(dnew.cexp, rm->cexp);
771	<	memcpy(dnew.wlpart, rm->wlpart, sizeof(dnew.wlpart));
772	<	for (j = dnew.ncols; j--; )
773	<	for (i = dnew.nrows; i--; )
774	<	memcpy(rmx_lval(&dnew,i,j), rmx_val(rm,j,i),
775	<	sizeof(rmx_dtype)*dnew.ncomp);
776	<	/* and reassign result */
777	<	return(rmx_transfer_data(rm, &dnew, 1));
768	>	dold = *rm;
769	>	rm->ncols = dold.nrows; rm->nrows = dold.ncols;
770	>	for (i = rm->nrows; i--; )      /* try every starting point */
771	>	for (j = rm->ncols; j--; ) {
772	>	int     i0, j0;
773	>	int     i1 = i;
774	>	size_t  j1 = j;
775	>	if (bmtest(i, j))
776	>	continue;       /* traversed loop earlier */
777	>	memcpy(val, rmx_val(rm,i,j),
778	>	sizeof(rmx_dtype)*rm->ncomp);
779	>	for ( ; ; ) {           /* new transpose loop */
780	>	const rmx_dtype     *ds;
781	>	i0 = i1; j0 = j1;
782	>	ds = rmx_val(&dold, j0, i0);
783	>	j1 = (ds - dold.mtx)/dold.ncomp;
784	>	i1 = j1 / rm->ncols;
785	>	j1 -= (size_t)i1*rm->ncols;
786	>	bmset(i1, j1);      /* mark as done */
787	>	if ((i1 == i) & (j1 == j))
788	>	break;          /* back at start */
789	>	memcpy(rmx_lval(rm,i0,j0), ds,
790	>	sizeof(rmx_dtype)*rm->ncomp);
791	>	}                       /* complete the loop */
792	>	memcpy(rmx_lval(rm,i0,j0), val,
793	>	sizeof(rmx_dtype)*rm->ncomp);
794	>	}
795	>	free(bmap);                     /* all done! */
796	>	return(1);
797	>	#undef  bmbyte
798	>	#undef  bmbit
799	>	#undef  bmop
800	>	#undef  bmtest
801	>	#undef  bmset
802		}
803
804		/* Multiply (concatenate) two matrices and allocate the result */
#	Line 979 | Line 978 | rmx_transform(const RMATRIX *msrc, int n, const double
978		return(dnew);
979		}
980
982	–	/* Convert a color matrix to newly allocated RMATRIX buffer */
983	–	RMATRIX *
984	–	rmx_from_cmatrix(const CMATRIX *cm)
985	–	{
986	–	RMATRIX *dnew;
987	–
988	–	if (!cm)
989	–	return(NULL);
990	–	dnew = rmx_alloc(cm->nrows, cm->ncols, 3);
991	–	if (!dnew)
992	–	return(NULL);
993	–
994	–	dnew->dtype = sizeof(COLORV)==sizeof(float) ?
995	–	DTfloat : DTdouble;
996	–
997	–	if (sizeof(COLORV) == sizeof(rmx_dtype)) {
998	–	memcpy(dnew->mtx, cm->cmem, rmx_array_size(dnew));
999	–	} else {
1000	–	int     i, j;
1001	–	for (i = dnew->nrows; i--; )
1002	–	for (j = dnew->ncols; j--; ) {
1003	–	const COLORV    *cv = cm_lval(cm,i,j);
1004	–	rmx_dtype       *dp = rmx_lval(dnew,i,j);
1005	–	dp[0] = cv[0];
1006	–	dp[1] = cv[1];
1007	–	dp[2] = cv[2];
1008	–	}
1009	–	}
1010	–	return(dnew);
1011	–	}
1012	–
1013	–	/* Convert general matrix to newly allocated CMATRIX buffer */
1014	–	CMATRIX *
1015	–	cm_from_rmatrix(const RMATRIX *rm)
1016	–	{
1017	–	CMATRIX *cnew;
1018	–
1019	–	if (!rm || !rm->mtx | (rm->ncomp == 2) | (rm->ncomp > MAXCOMP))
1020	–	return(NULL);
1021	–	cnew = cm_alloc(rm->nrows, rm->ncols);
1022	–	if (!cnew)
1023	–	return(NULL);
1024	–	if ((sizeof(COLORV) == sizeof(rmx_dtype)) & (rm->ncomp == 3)) {
1025	–	memcpy(cnew->cmem, rm->mtx, rmx_array_size(rm));
1026	–	} else {
1027	–	int     i, j;
1028	–	for (i = cnew->nrows; i--; )
1029	–	for (j = cnew->ncols; j--; ) {
1030	–	const rmx_dtype *dp = rmx_val(rm,i,j);
1031	–	COLORV          *cv = cm_lval(cnew,i,j);
1032	–	switch (rm->ncomp) {
1033	–	case 3:
1034	–	setcolor(cv, dp[0], dp[1], dp[2]);
1035	–	break;
1036	–	case 1:
1037	–	setcolor(cv, dp[0], dp[0], dp[0]);
1038	–	break;
1039	–	default: {
1040	–	COLORV  scol[MAXCOMP];
1041	–	int     k;
1042	–	for (k = rm->ncomp; k--; )
1043	–	scol[k] = dp[k];
1044	–	scolor2color(cv, scol, rm->ncomp, rm->wlpart);
1045	–	} break;
1046	–	}
1047	–	}
1048	–	}
1049	–	return(cnew);
1050	–	}

Diff Legend

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