src/util/rmtxop.c

#ifndef lint
static const char RCSid[] = "$Id: rmtxop.c,v 2.25 2023/11/29 18:57:10 greg Exp $";
#endif
/*
 * General component matrix operations.
 */

#include <errno.h>
#include <ctype.h>
#include "rtio.h"
#include "resolu.h"
#include "rmatrix.h"
#include "platform.h"

#define MAXCOMP         MAXCSAMP        /* #components we support */

/* Unary matrix operation(s) */
typedef struct {
        double          sca[MAXCOMP];           /* scalar coefficients */
        double          cmat[MAXCOMP*MAXCOMP];  /* component transformation */
        short           nsf;                    /* number of scalars */
        short           clen;                   /* number of coefficients */
        char            csym[11];               /* symbolic coefficients */
        char            transpose;              /* do transpose? */
} RUNARYOP;

/* Matrix input source and requested operation(s) */
typedef struct {
        const char      *inspec;                /* input specification */
        RMPref          rmp;                    /* matrix preference */
        RUNARYOP        preop;                  /* unary operation(s) */
        RMATRIX         *mtx;                   /* original matrix if loaded */
        int             binop;                  /* binary op with next (or 0) */
} ROPMAT;

int     verbose = 0;                    /* verbose reporting? */

/* Load matrix */
static int
loadmatrix(ROPMAT *rop)
{
        if (rop->mtx != NULL)           /* already loaded? */
                return(0);

        rop->mtx = rmx_load(rop->inspec, rop->rmp);

        return(!rop->mtx ? -1 : 1);
}

/* Compute conversion row from spectrum to one channel of RGB */
static void
rgbrow(ROPMAT *rop, int r, int p)
{
        const int       nc = rop->mtx->ncomp;
        const float *   wlp = rop->mtx->wlpart;
        int             i;

        for (i = nc; i--; ) {
                int     nmEnd = wlp[0] + (wlp[3] - wlp[0])*i/nc;
                int     nmStart = wlp[0] + (wlp[3] - wlp[0])*(i+1)/nc;
                COLOR   crgb;
                spec_rgb(crgb, nmStart, nmEnd);
                rop->preop.cmat[r*nc+i] = crgb[p];
        }
}

/* Compute conversion row from spectrum to one channel of XYZ */
static void
xyzrow(ROPMAT *rop, int r, int p)
{
        const int       nc = rop->mtx->ncomp;
        const float *   wlp = rop->mtx->wlpart;
        int             i;

        for (i = nc; i--; ) {
                int     nmEnd = wlp[0] + (wlp[3] - wlp[0])*i/nc;
                int     nmStart = wlp[0] + (wlp[3] - wlp[0])*(i+1)/nc;
                COLOR   cxyz;
                spec_cie(cxyz, nmStart, nmEnd);
                rop->preop.cmat[r*nc+i] = cxyz[p];
        }
}

/* Use the spectral sensitivity function to compute matrix coefficients */
static void
sensrow(ROPMAT *rop, int r, double (*sf)(SCOLOR sc, int ncs, const float wlpt[4]))
{
        const int       nc = rop->mtx->ncomp;
        int             i;

        for (i = nc; i--; ) {
                SCOLOR  sclr;
                scolorblack(sclr);
                sclr[i] = 1.f;
                rop->preop.cmat[r*nc+i] = (*sf)(sclr, nc, rop->mtx->wlpart);
        }
}

/* Check/set symbolic transform */
static int
checksymbolic(ROPMAT *rop)
{
        const int       nc = rop->mtx->ncomp;
        const int       dt = rop->mtx->dtype;
        int             i, j;

        if (nc < 3) {
                fprintf(stderr, "%s: -c '%s' requires at least 3 components\n",
                                rop->inspec, rop->preop.csym);
                return(-1);
        }
        rop->preop.clen = strlen(rop->preop.csym) * nc;
        if (rop->preop.clen > MAXCOMP*MAXCOMP) {
                fprintf(stderr, "%s: -c '%s' results in too many components\n",
                                rop->inspec, rop->preop.csym);
                return(-1);
        }
        for (j = 0; rop->preop.csym[j]; j++) {
                int     comp = 0;
                switch (rop->preop.csym[j]) {
                case 'B':
                        ++comp;
                        /* fall through */
                case 'G':
                        ++comp;
                        /* fall through */
                case 'R':
                        if (dt == DTxyze) {
                                for (i = 3; i--; )
                                        rop->preop.cmat[j*nc+i] = 1./WHTEFFICACY *
                                                        xyz2rgbmat[comp][i];
                        } else if (nc == 3)
                                rop->preop.cmat[j*nc+comp] = 1.;
                        else
                                rgbrow(rop, j, comp);
                        break;
                case 'Z':
                        ++comp;
                        /* fall through */
                case 'Y':
                        ++comp;
                        /* fall through */
                case 'X':
                        if (dt == DTxyze) {
                                rop->preop.cmat[j*nc+comp] = 1.;
                        } else if (nc == 3) {
                                for (i = 3; i--; )
                                        rop->preop.cmat[j*nc+i] =
                                                        rgb2xyzmat[comp][i];
                        } else if (comp == CIEY)
                                sensrow(rop, j, scolor2photopic);
                        else
                                xyzrow(rop, j, comp);

                        for (i = nc*(dt != DTxyze); i--; )
                                rop->preop.cmat[j*nc+i] *= WHTEFFICACY;
                        break;
                case 'S':               /* scotopic (il)luminance */
                        sensrow(rop, j, scolor2scotopic);
                        for (i = nc; i--; )
                                rop->preop.cmat[j*nc+i] *= WHTSCOTOPIC;
                        break;
                case 'M':               /* melanopic (il)luminance */
                        sensrow(rop, j, scolor2melanopic);
                        for (i = nc; i--; )
                                rop->preop.cmat[j*nc+i] *= WHTMELANOPIC;
                        break;
                case 'A':               /* average component */
                        for (i = nc; i--; )
                                rop->preop.cmat[j*nc+i] = 1./(double)nc;
                        break;
                default:
                        fprintf(stderr, "%s: -c '%c' unsupported\n",
                                rop->inspec, rop->preop.csym[j]);
                        return(-1);
                }
        }
                                        /* return recommended output type */
        if (!strcmp(rop->preop.csym, "XYZ")) {
                if (dt <= DTspec)
                        return(DTxyze);
        } else if (!strcmp(rop->preop.csym, "RGB")) {
                if (dt <= DTspec)
                        return(DTrgbe);
        }
        if ((nc > 3) & (dt <= DTspec))
                return(DTfloat);        /* probably not actual spectrum */
        return(0);
}

/* Get matrix and perform unary operations */
static RMATRIX *
loadop(ROPMAT *rop)
{
        int     outtype = 0;
        RMATRIX *mres;
        int     i, j;

        if (loadmatrix(rop) < 0)                /* make sure we're loaded */
                return(NULL);

        if (rop->preop.csym[0] &&               /* symbolic transform? */
                        (outtype = checksymbolic(rop)) < 0)
                goto failure;
        if (rop->preop.clen > 0) {              /* apply component transform? */
                if (rop->preop.clen % rop->mtx->ncomp) {
                        fprintf(stderr, "%s: -c must have N x %d coefficients\n",
                                        rop->inspec, rop->mtx->ncomp);
                        goto failure;
                }
                if (rop->preop.nsf > 0) {       /* scale transform, first */
                        if (rop->preop.nsf == 1) {
                                for (i = rop->preop.clen; i--; )
                                        rop->preop.cmat[i] *= rop->preop.sca[0];
                        } else if (rop->preop.nsf != rop->mtx->ncomp) {
                                fprintf(stderr, "%s: -s must have one or %d factors\n",
                                                rop->inspec, rop->mtx->ncomp);
                                goto failure;
                        } else {
                                for (j = rop->preop.clen/rop->preop.nsf; j--; )
                                        for (i = rop->preop.nsf; i--; )
                                                rop->preop.cmat[j*rop->preop.nsf+i] *=
                                                                rop->preop.sca[i];
                        }
                }
                mres = rmx_transform(rop->mtx, rop->preop.clen/rop->mtx->ncomp,
                                        rop->preop.cmat);
                if (mres == NULL) {
                        fprintf(stderr, "%s: matrix transform failed\n",
                                                rop->inspec);
                        goto failure;
                }
                if (verbose)
                        fprintf(stderr, "%s: applied %d x %d transform%s\n",
                                        rop->inspec, mres->ncomp,
                                        rop->mtx->ncomp,
                                        rop->preop.nsf ? " (* scalar)" : "");
                rop->preop.nsf = 0;             /* now folded in */
                if ((mres->ncomp > 3) & (mres->dtype <= DTspec))
                        outtype = DTfloat;      /* probably not actual spectrum */
                rmx_free(rop->mtx);
                rop->mtx = mres;
        }
        if (rop->preop.nsf > 0) {               /* apply scalar(s)? */
                if (rop->preop.nsf == 1) {
                        for (i = rop->mtx->ncomp; --i; )
                                rop->preop.sca[i] = rop->preop.sca[0];
                } else if (rop->preop.nsf != rop->mtx->ncomp) {
                        fprintf(stderr, "%s: -s must have one or %d factors\n",
                                        rop->inspec, rop->mtx->ncomp);
                        goto failure;
                }
                if (!rmx_scale(rop->mtx, rop->preop.sca)) {
                        fputs(rop->inspec, stderr);
                        fputs(": scalar operation failed\n", stderr);
                        goto failure;
                }
                if (verbose) {
                        fputs(rop->inspec, stderr);
                        fputs(": applied scalar (", stderr);
                        for (i = 0; i < rop->preop.nsf; i++)
                                fprintf(stderr, " %f", rop->preop.sca[i]);
                        fputs(" )\n", stderr);
                }
        }
        if (rop->preop.transpose) {             /* transpose matrix? */
                mres = rmx_transpose(rop->mtx);
                if (mres == NULL) {
                        fputs(rop->inspec, stderr);
                        fputs(": transpose failed\n", stderr);
                        goto failure;
                }
                if (verbose) {
                        fputs(rop->inspec, stderr);
                        fputs(": transposed rows and columns\n", stderr);
                }
                rmx_free(rop->mtx);
                rop->mtx = mres;
        }
        mres = rop->mtx;
        rop->mtx = NULL;
        if (outtype)
                mres->dtype = outtype;
        return(mres);
failure:
        rmx_free(rop->mtx);
        return(rop->mtx = NULL);
}

/* Execute binary operation, free matrix arguments and return new result */
static RMATRIX *
binaryop(const char *inspec, RMATRIX *mleft, int op, RMATRIX *mright)
{
        RMATRIX *mres = NULL;
        int     i;

        if ((mleft == NULL) | (mright == NULL))
                return(NULL);
        switch (op) {
        case '.':                       /* concatenate */
                if (mleft->ncomp != mright->ncomp) {
                        fputs(inspec, stderr);
                        fputs(": # components do not match\n", stderr);
                } else if (mleft->ncols != mright->nrows) {
                        fputs(inspec, stderr);
                        fputs(": mismatched dimensions\n",
                                        stderr);
                } else
                        mres = rmx_multiply(mleft, mright);
                rmx_free(mleft);
                rmx_free(mright);
                if (mres == NULL) {
                        fputs(inspec, stderr);
                        fputs(": concatenation failed\n", stderr);
                        return(NULL);
                }
                if (verbose) {
                        fputs(inspec, stderr);
                        fputs(": concatenated matrix\n", stderr);
                }
                break;
        case '+':
                if (!rmx_sum(mleft, mright, NULL)) {
                        fputs(inspec, stderr);
                        fputs(": matrix sum failed\n", stderr);
                        rmx_free(mleft);
                        rmx_free(mright);
                        return(NULL);
                }
                if (verbose) {
                        fputs(inspec, stderr);
                        fputs(": added in matrix\n", stderr);
                }
                rmx_free(mright);
                mres = mleft;
                break;
        case '*':
        case '/': {
                const char *    tnam = (op == '/') ?
                                        "division" : "multiplication";
                errno = 0;
                if (!rmx_elemult(mleft, mright, (op == '/'))) {
                        fprintf(stderr, "%s: element-wise %s failed\n",
                                        inspec, tnam);
                        rmx_free(mleft);
                        rmx_free(mright);
                        return(NULL);
                }
                if (errno)
                        fprintf(stderr,
                                "%s: warning - error during element-wise %s\n",
                                        inspec, tnam);
                else if (verbose)
                        fprintf(stderr, "%s: element-wise %s\n", inspec, tnam);
                rmx_free(mright);
                mres = mleft;
                } break;
        default:
                fprintf(stderr, "%s: unknown operation '%c'\n", inspec, op);
                rmx_free(mleft);
                rmx_free(mright);
                return(NULL);
        }
        return(mres);
}

/* Perform matrix operations from left to right */
static RMATRIX *
op_left2right(ROPMAT *mop)
{
        RMATRIX *mleft = loadop(mop);

        while (mop->binop) {
                if (mleft == NULL)
                        break;
                mleft = binaryop(mop[1].inspec,
                                mleft, mop->binop, loadop(mop+1));
                mop++;
        }
        return(mleft);
}

/* Perform matrix operations from right to left */
static RMATRIX *
op_right2left(ROPMAT *mop)
{
        RMATRIX *mright;
        int     rpos = 0;
                                        /* find end of list */
        while (mop[rpos].binop)
                if (mop[rpos++].binop != '.') {
                        fputs(
                "Right-to-left evaluation only for matrix multiplication!\n",
                                        stderr);
                        return(NULL);
                }
        mright = loadop(mop+rpos);
        while (rpos-- > 0) {
                if (mright == NULL)
                        break;
                mright = binaryop(mop[rpos+1].inspec,
                                loadop(mop+rpos), mop[rpos].binop, mright);
        }
        return(mright);
}

#define t_nrows(mop)    ((mop)->preop.transpose ? (mop)->mtx->ncols \
                                                : (mop)->mtx->nrows)
#define t_ncols(mop)    ((mop)->preop.transpose ? (mop)->mtx->nrows \
                                                : (mop)->mtx->ncols)

/* Should we prefer concatenating from rightmost matrix towards left? */
static int
prefer_right2left(ROPMAT *mop)
{
        int     mri = 0;

        while (mop[mri].binop)          /* find rightmost matrix */
                if (mop[mri++].binop != '.')
                        return(0);      /* pre-empt reversal for other ops */

        if (mri <= 1)
                return(0);              /* won't matter */

        if (loadmatrix(mop+mri) < 0)    /* load rightmost cat */
                return(1);              /* fail will bail in a moment */

        if (t_ncols(mop+mri) == 1)
                return(1);              /* definitely better R->L */

        if (t_ncols(mop+mri) >= t_nrows(mop+mri))
                return(0);              /* ...probably worse */

        if (loadmatrix(mop) < 0)        /* load leftmost */
                return(0);              /* fail will bail in a moment */

        return(t_ncols(mop+mri) < t_nrows(mop));
}

static int
get_factors(double da[], int n, char *av[])
{
        int     ac;

        for (ac = 0; ac < n && isflt(av[ac]); ac++)
                da[ac] = atof(av[ac]);
        return(ac);
}

static ROPMAT *
resize_moparr(ROPMAT *mop, int n2alloc)
{
        int     nmats = 0;
        int     i;

        while (mop[nmats++].binop)
                ;
        for (i = nmats; i > n2alloc; i--)
                rmx_free(mop[i].mtx);
        mop = (ROPMAT *)realloc(mop, n2alloc*sizeof(ROPMAT));
        if (mop == NULL) {
                fputs("Out of memory in resize_moparr()\n", stderr);
                exit(1);
        }
        if (n2alloc > nmats)
                memset(mop+nmats, 0, (n2alloc-nmats)*sizeof(ROPMAT));
        return(mop);
}

/* Load one or more matrices and operate on them, sending results to stdout */
int
main(int argc, char *argv[])
{
        int     outfmt = DTfromHeader;
        int     nall = 2;
        ROPMAT  *mop = (ROPMAT *)calloc(nall, sizeof(ROPMAT));
        int     nmats = 0;
        RMATRIX *mres = NULL;
        int     stdin_used = 0;
        int     i;
                                        /* get options and arguments */
        for (i = 1; i < argc; i++) {
                if (argv[i][0] && !argv[i][1] &&
                                strchr(".+*/", argv[i][0]) != NULL) {
                        if (!nmats || mop[nmats-1].binop) {
                                fprintf(stderr,
                        "%s: missing matrix argument before '%c' operation\n",
                                                argv[0], argv[i][0]);
                                return(1);
                        }
                        mop[nmats-1].binop = argv[i][0];
                } else if (argv[i][0] != '-' || !argv[i][1]) {
                        if (argv[i][0] == '-') {
                                if (stdin_used++) {
                                        fprintf(stderr,
                        "%s: standard input used for more than one matrix\n",
                                                argv[0]);
                                        return(1);
                                }
                                mop[nmats].inspec = stdin_name;
                        } else
                                mop[nmats].inspec = argv[i];
                        if (nmats > 0 && !mop[nmats-1].binop)
                                mop[nmats-1].binop = '.';
                        nmats++;
                } else {
                        int     n = argc-1 - i;
                        switch (argv[i][1]) {   /* get option */
                        case 'v':
                                verbose++;
                                break;
                        case 'f':
                                switch (argv[i][2]) {
                                case 'd':
                                        outfmt = DTdouble;
                                        break;
                                case 'f':
                                        outfmt = DTfloat;
                                        break;
                                case 'a':
                                        outfmt = DTascii;
                                        break;
                                case 'c':
                                        outfmt = DTrgbe;
                                        break;
                                default:
                                        goto userr;
                                }
                                break;
                        case 't':
                                mop[nmats].preop.transpose = 1;
                                break;
                        case 's':
                                if (n > MAXCOMP) n = MAXCOMP;
                                i += mop[nmats].preop.nsf =
                                        get_factors(mop[nmats].preop.sca,
                                                        n, argv+i+1);
                                if (mop[nmats].preop.nsf <= 0) {
                                        fprintf(stderr, "%s: -s missing arguments\n",
                                                        argv[0]);
                                        goto userr;
                                }
                                break;
                        case 'c':
                                if (n && isupper(argv[i+1][0])) {
                                        strlcpy(mop[nmats].preop.csym,
                                                argv[++i],
                                                sizeof(mop[0].preop.csym));
                                        mop[nmats].preop.clen = 0;
                                        break;
                                }
                                if (n > MAXCOMP*MAXCOMP) n = MAXCOMP*MAXCOMP;
                                i += mop[nmats].preop.clen =
                                        get_factors(mop[nmats].preop.cmat,
                                                        n, argv+i+1);
                                if (mop[nmats].preop.clen <= 0) {
                                        fprintf(stderr, "%s: -c missing arguments\n",
                                                        argv[0]);
                                        goto userr;
                                }
                                mop[nmats].preop.csym[0] = '\0';
                                break;
                        case 'r':
                                if (argv[i][2] == 'f')
                                        mop[nmats].rmp = RMPreflF;
                                else if (argv[i][2] == 'b')
                                        mop[nmats].rmp = RMPreflB;
                                else
                                        goto userr;
                                break;
                        default:
                                fprintf(stderr, "%s: unknown operation '%s'\n",
                                                argv[0], argv[i]);
                                goto userr;
                        }
                }
                if (nmats >= nall)
                        mop = resize_moparr(mop, nall += 2);
        }
        if (mop[0].inspec == NULL)      /* nothing to do? */
                goto userr;
        if (mop[nmats-1].binop) {
                fprintf(stderr,
                        "%s: missing matrix argument after '%c' operation\n",
                                argv[0], mop[nmats-1].binop);
                return(1);
        }
                                        /* favor quicker concatenation */
        mop[nmats].mtx = prefer_right2left(mop) ? op_right2left(mop)
                                                : op_left2right(mop);
        if (mop[nmats].mtx == NULL)
                return(1);
                                        /* apply trailing unary operations */
        mop[nmats].inspec = "trailing_ops";
        mres = loadop(mop+nmats);
        if (mres == NULL)
                return(1);
        if (outfmt == DTfromHeader)     /* check data type */
                outfmt = mres->dtype;
        if (outfmt == DTrgbe) {
                if (mres->ncomp > 3)
                        outfmt = DTspec;
                else if (mres->dtype == DTxyze)
                        outfmt = DTxyze;
        }
        newheader("RADIANCE", stdout);  /* write result to stdout */
        printargs(argc, argv, stdout);
        return(rmx_write(mres, outfmt, stdout) ? 0 : 1);
userr:
        fprintf(stderr,
        "Usage: %s [-v][-f{adfc}][-t][-s sf .. | -c ce ..][-rf|-rb] m1 [.+*/] .. > mres\n",
                        argv[0]);
        return(1);
}
Revision:	2.26
Committed:	Fri Dec 1 02:05:00 2023 UTC (4 months, 3 weeks ago) by greg
Content type:	text/plain
Branch:	MAIN
Changes since 2.25:	+10 -11 lines
Log Message:	refactor(rmtxop): Changed rmatrix library interface for by row matrix output
#	User	Rev	Content
1	greg	2.1	#ifndef lint
2	greg	2.26	static const char RCSid[] = "$Id: rmtxop.c,v 2.25 2023/11/29 18:57:10 greg Exp $";
3	greg	2.1	#endif
4			/*
5			* General component matrix operations.
6			*/
7
8	greg	2.11	#include <errno.h>
9	greg	2.22	#include <ctype.h>
10	greg	2.1	#include "rtio.h"
11			#include "resolu.h"
12			#include "rmatrix.h"
13	greg	2.10	#include "platform.h"
14	greg	2.1
15	greg	2.22	#define MAXCOMP MAXCSAMP /* #components we support */
16	greg	2.1
17	greg	2.22	/* Unary matrix operation(s) */
18	greg	2.1	typedef struct {
19			double sca[MAXCOMP]; /* scalar coefficients */
20			double cmat[MAXCOMPMAXCOMP]; / component transformation */
21	greg	2.13	short nsf; /* number of scalars */
22			short clen; /* number of coefficients */
23	greg	2.22	char csym[11]; /* symbolic coefficients */
24			char transpose; /* do transpose? */
25	greg	2.13	} RUNARYOP;
26
27	greg	2.22	/* Matrix input source and requested operation(s) */
28	greg	2.13	typedef struct {
29			const char inspec; / input specification */
30	greg	2.18	RMPref rmp; /* matrix preference */
31	greg	2.13	RUNARYOP preop; /* unary operation(s) */
32			RMATRIX mtx; / original matrix if loaded */
33			int binop; /* binary op with next (or 0) */
34			} ROPMAT;
35	greg	2.1
36			int verbose = 0; /* verbose reporting? */
37
38	greg	2.13	/* Load matrix */
39			static int
40			loadmatrix(ROPMAT *rop)
41	greg	2.1	{
42	greg	2.20	if (rop->mtx != NULL) /* already loaded? */
43	greg	2.13	return(0);
44
45	greg	2.18	rop->mtx = rmx_load(rop->inspec, rop->rmp);
46	greg	2.23
47			return(!rop->mtx ? -1 : 1);
48	greg	2.1	}
49
50	greg	2.22	/* Compute conversion row from spectrum to one channel of RGB */
51			static void
52			rgbrow(ROPMAT *rop, int r, int p)
53			{
54			const int nc = rop->mtx->ncomp;
55			const float * wlp = rop->mtx->wlpart;
56			int i;
57
58			for (i = nc; i--; ) {
59			int nmEnd = wlp[0] + (wlp[3] - wlp[0])*i/nc;
60			int nmStart = wlp[0] + (wlp[3] - wlp[0])*(i+1)/nc;
61			COLOR crgb;
62			spec_rgb(crgb, nmStart, nmEnd);
63			rop->preop.cmat[r*nc+i] = crgb[p];
64			}
65			}
66
67			/* Compute conversion row from spectrum to one channel of XYZ */
68			static void
69			xyzrow(ROPMAT *rop, int r, int p)
70			{
71			const int nc = rop->mtx->ncomp;
72			const float * wlp = rop->mtx->wlpart;
73			int i;
74
75			for (i = nc; i--; ) {
76			int nmEnd = wlp[0] + (wlp[3] - wlp[0])*i/nc;
77			int nmStart = wlp[0] + (wlp[3] - wlp[0])*(i+1)/nc;
78			COLOR cxyz;
79			spec_cie(cxyz, nmStart, nmEnd);
80			rop->preop.cmat[r*nc+i] = cxyz[p];
81			}
82			}
83
84			/* Use the spectral sensitivity function to compute matrix coefficients */
85			static void
86			sensrow(ROPMAT rop, int r, double (sf)(SCOLOR sc, int ncs, const float wlpt[4]))
87			{
88			const int nc = rop->mtx->ncomp;
89			int i;
90
91			for (i = nc; i--; ) {
92			SCOLOR sclr;
93			scolorblack(sclr);
94	greg	2.24	sclr[i] = 1.f;
95	greg	2.22	rop->preop.cmat[rnc+i] = (sf)(sclr, nc, rop->mtx->wlpart);
96			}
97			}
98
99			/* Check/set symbolic transform */
100			static int
101			checksymbolic(ROPMAT *rop)
102			{
103			const int nc = rop->mtx->ncomp;
104			const int dt = rop->mtx->dtype;
105			int i, j;
106
107			if (nc < 3) {
108			fprintf(stderr, "%s: -c '%s' requires at least 3 components\n",
109			rop->inspec, rop->preop.csym);
110			return(-1);
111			}
112			rop->preop.clen = strlen(rop->preop.csym) * nc;
113			if (rop->preop.clen > MAXCOMP*MAXCOMP) {
114			fprintf(stderr, "%s: -c '%s' results in too many components\n",
115			rop->inspec, rop->preop.csym);
116			return(-1);
117			}
118			for (j = 0; rop->preop.csym[j]; j++) {
119			int comp = 0;
120			switch (rop->preop.csym[j]) {
121			case 'B':
122			++comp;
123			/* fall through */
124			case 'G':
125			++comp;
126			/* fall through */
127			case 'R':
128			if (dt == DTxyze) {
129			for (i = 3; i--; )
130			rop->preop.cmat[jnc+i] = 1./WHTEFFICACY
131			xyz2rgbmat[comp][i];
132			} else if (nc == 3)
133			rop->preop.cmat[j*nc+comp] = 1.;
134			else
135			rgbrow(rop, j, comp);
136			break;
137			case 'Z':
138			++comp;
139			/* fall through */
140			case 'Y':
141			++comp;
142			/* fall through */
143			case 'X':
144			if (dt == DTxyze) {
145			rop->preop.cmat[j*nc+comp] = 1.;
146			} else if (nc == 3) {
147			for (i = 3; i--; )
148			rop->preop.cmat[j*nc+i] =
149			rgb2xyzmat[comp][i];
150			} else if (comp == CIEY)
151			sensrow(rop, j, scolor2photopic);
152			else
153			xyzrow(rop, j, comp);
154
155			for (i = nc*(dt != DTxyze); i--; )
156			rop->preop.cmat[jnc+i] = WHTEFFICACY;
157			break;
158	greg	2.24	case 'S': /* scotopic (il)luminance */
159	greg	2.22	sensrow(rop, j, scolor2scotopic);
160			for (i = nc; i--; )
161			rop->preop.cmat[jnc+i] = WHTSCOTOPIC;
162			break;
163	greg	2.24	case 'M': /* melanopic (il)luminance */
164	greg	2.22	sensrow(rop, j, scolor2melanopic);
165			for (i = nc; i--; )
166			rop->preop.cmat[jnc+i] = WHTMELANOPIC;
167			break;
168	greg	2.24	case 'A': /* average component */
169			for (i = nc; i--; )
170	greg	2.25	rop->preop.cmat[j*nc+i] = 1./(double)nc;
171	greg	2.24	break;
172	greg	2.22	default:
173			fprintf(stderr, "%s: -c '%c' unsupported\n",
174			rop->inspec, rop->preop.csym[j]);
175			return(-1);
176			}
177			}
178			/* return recommended output type */
179			if (!strcmp(rop->preop.csym, "XYZ")) {
180			if (dt <= DTspec)
181			return(DTxyze);
182			} else if (!strcmp(rop->preop.csym, "RGB")) {
183			if (dt <= DTspec)
184			return(DTrgbe);
185			}
186			if ((nc > 3) & (dt <= DTspec))
187			return(DTfloat); /* probably not actual spectrum */
188			return(0);
189			}
190
191	greg	2.13	/* Get matrix and perform unary operations */
192	greg	2.1	static RMATRIX *
193	greg	2.13	loadop(ROPMAT *rop)
194	greg	2.1	{
195	greg	2.22	int outtype = 0;
196	greg	2.13	RMATRIX *mres;
197	greg	2.22	int i, j;
198	greg	2.1
199	greg	2.13	if (loadmatrix(rop) < 0) /* make sure we're loaded */
200	greg	2.1	return(NULL);
201	greg	2.13
202	greg	2.22	if (rop->preop.csym[0] && /* symbolic transform? */
203			(outtype = checksymbolic(rop)) < 0)
204			goto failure;
205			if (rop->preop.clen > 0) { /* apply component transform? */
206			if (rop->preop.clen % rop->mtx->ncomp) {
207			fprintf(stderr, "%s: -c must have N x %d coefficients\n",
208			rop->inspec, rop->mtx->ncomp);
209			goto failure;
210			}
211			if (rop->preop.nsf > 0) { /* scale transform, first */
212			if (rop->preop.nsf == 1) {
213			for (i = rop->preop.clen; i--; )
214			rop->preop.cmat[i] *= rop->preop.sca[0];
215			} else if (rop->preop.nsf != rop->mtx->ncomp) {
216			fprintf(stderr, "%s: -s must have one or %d factors\n",
217			rop->inspec, rop->mtx->ncomp);
218			goto failure;
219			} else {
220			for (j = rop->preop.clen/rop->preop.nsf; j--; )
221			for (i = rop->preop.nsf; i--; )
222			rop->preop.cmat[jrop->preop.nsf+i] =
223			rop->preop.sca[i];
224			}
225			}
226			mres = rmx_transform(rop->mtx, rop->preop.clen/rop->mtx->ncomp,
227			rop->preop.cmat);
228			if (mres == NULL) {
229			fprintf(stderr, "%s: matrix transform failed\n",
230			rop->inspec);
231	greg	2.13	goto failure;
232	greg	2.1	}
233	greg	2.22	if (verbose)
234			fprintf(stderr, "%s: applied %d x %d transform%s\n",
235			rop->inspec, mres->ncomp,
236			rop->mtx->ncomp,
237			rop->preop.nsf ? " (* scalar)" : "");
238	greg	2.26	rop->preop.nsf = 0; /* now folded in */
239	greg	2.22	if ((mres->ncomp > 3) & (mres->dtype <= DTspec))
240			outtype = DTfloat; /* probably not actual spectrum */
241			rmx_free(rop->mtx);
242			rop->mtx = mres;
243			}
244			if (rop->preop.nsf > 0) { /* apply scalar(s)? */
245	greg	2.13	if (rop->preop.nsf == 1) {
246			for (i = rop->mtx->ncomp; --i; )
247			rop->preop.sca[i] = rop->preop.sca[0];
248			} else if (rop->preop.nsf != rop->mtx->ncomp) {
249	greg	2.1	fprintf(stderr, "%s: -s must have one or %d factors\n",
250	greg	2.13	rop->inspec, rop->mtx->ncomp);
251			goto failure;
252	greg	2.1	}
253	greg	2.13	if (!rmx_scale(rop->mtx, rop->preop.sca)) {
254			fputs(rop->inspec, stderr);
255	greg	2.1	fputs(": scalar operation failed\n", stderr);
256	greg	2.13	goto failure;
257	greg	2.1	}
258			if (verbose) {
259	greg	2.13	fputs(rop->inspec, stderr);
260	greg	2.1	fputs(": applied scalar (", stderr);
261	greg	2.13	for (i = 0; i < rop->preop.nsf; i++)
262			fprintf(stderr, " %f", rop->preop.sca[i]);
263	greg	2.1	fputs(" )\n", stderr);
264			}
265			}
266	greg	2.22	if (rop->preop.transpose) { /* transpose matrix? */
267	greg	2.13	mres = rmx_transpose(rop->mtx);
268			if (mres == NULL) {
269			fputs(rop->inspec, stderr);
270	greg	2.12	fputs(": transpose failed\n", stderr);
271	greg	2.13	goto failure;
272	greg	2.12	}
273			if (verbose) {
274	greg	2.13	fputs(rop->inspec, stderr);
275	greg	2.12	fputs(": transposed rows and columns\n", stderr);
276			}
277	greg	2.13	rmx_free(rop->mtx);
278			rop->mtx = mres;
279			}
280			mres = rop->mtx;
281			rop->mtx = NULL;
282	greg	2.22	if (outtype)
283			mres->dtype = outtype;
284	greg	2.13	return(mres);
285			failure:
286			rmx_free(rop->mtx);
287			return(rop->mtx = NULL);
288			}
289
290			/* Execute binary operation, free matrix arguments and return new result */
291			static RMATRIX *
292			binaryop(const char inspec, RMATRIX mleft, int op, RMATRIX *mright)
293			{
294			RMATRIX *mres = NULL;
295			int i;
296
297			if ((mleft == NULL) \| (mright == NULL))
298			return(NULL);
299			switch (op) {
300			case '.': /* concatenate */
301	greg	2.16	if (mleft->ncomp != mright->ncomp) {
302			fputs(inspec, stderr);
303			fputs(": # components do not match\n", stderr);
304			} else if (mleft->ncols != mright->nrows) {
305			fputs(inspec, stderr);
306			fputs(": mismatched dimensions\n",
307			stderr);
308			} else
309			mres = rmx_multiply(mleft, mright);
310	greg	2.13	rmx_free(mleft);
311	greg	2.12	rmx_free(mright);
312	greg	2.1	if (mres == NULL) {
313	greg	2.13	fputs(inspec, stderr);
314	greg	2.16	fputs(": concatenation failed\n", stderr);
315	greg	2.1	return(NULL);
316			}
317			if (verbose) {
318	greg	2.13	fputs(inspec, stderr);
319	greg	2.1	fputs(": concatenated matrix\n", stderr);
320			}
321	greg	2.13	break;
322			case '+':
323			if (!rmx_sum(mleft, mright, NULL)) {
324			fputs(inspec, stderr);
325	greg	2.1	fputs(": matrix sum failed\n", stderr);
326	greg	2.13	rmx_free(mleft);
327	greg	2.1	rmx_free(mright);
328			return(NULL);
329			}
330			if (verbose) {
331	greg	2.13	fputs(inspec, stderr);
332	greg	2.1	fputs(": added in matrix\n", stderr);
333			}
334			rmx_free(mright);
335	greg	2.13	mres = mleft;
336			break;
337			case '*':
338			case '/': {
339			const char * tnam = (op == '/') ?
340	greg	2.11	"division" : "multiplication";
341			errno = 0;
342	greg	2.13	if (!rmx_elemult(mleft, mright, (op == '/'))) {
343	greg	2.11	fprintf(stderr, "%s: element-wise %s failed\n",
344	greg	2.13	inspec, tnam);
345			rmx_free(mleft);
346	greg	2.11	rmx_free(mright);
347			return(NULL);
348			}
349			if (errno)
350			fprintf(stderr,
351			"%s: warning - error during element-wise %s\n",
352	greg	2.13	inspec, tnam);
353	greg	2.11	else if (verbose)
354	greg	2.13	fprintf(stderr, "%s: element-wise %s\n", inspec, tnam);
355	greg	2.11	rmx_free(mright);
356	greg	2.13	mres = mleft;
357			} break;
358			default:
359			fprintf(stderr, "%s: unknown operation '%c'\n", inspec, op);
360			rmx_free(mleft);
361	greg	2.1	rmx_free(mright);
362			return(NULL);
363			}
364	greg	2.13	return(mres);
365			}
366
367			/* Perform matrix operations from left to right */
368			static RMATRIX *
369			op_left2right(ROPMAT *mop)
370			{
371			RMATRIX *mleft = loadop(mop);
372
373			while (mop->binop) {
374			if (mleft == NULL)
375			break;
376			mleft = binaryop(mop[1].inspec,
377			mleft, mop->binop, loadop(mop+1));
378			mop++;
379			}
380	greg	2.1	return(mleft);
381			}
382
383	greg	2.13	/* Perform matrix operations from right to left */
384			static RMATRIX *
385			op_right2left(ROPMAT *mop)
386			{
387			RMATRIX *mright;
388			int rpos = 0;
389			/* find end of list */
390			while (mop[rpos].binop)
391	greg	2.14	if (mop[rpos++].binop != '.') {
392			fputs(
393			"Right-to-left evaluation only for matrix multiplication!\n",
394			stderr);
395			return(NULL);
396			}
397	greg	2.13	mright = loadop(mop+rpos);
398			while (rpos-- > 0) {
399			if (mright == NULL)
400			break;
401	greg	2.20	mright = binaryop(mop[rpos+1].inspec,
402	greg	2.13	loadop(mop+rpos), mop[rpos].binop, mright);
403			}
404			return(mright);
405			}
406
407			#define t_nrows(mop) ((mop)->preop.transpose ? (mop)->mtx->ncols \
408			: (mop)->mtx->nrows)
409			#define t_ncols(mop) ((mop)->preop.transpose ? (mop)->mtx->nrows \
410			: (mop)->mtx->ncols)
411
412			/* Should we prefer concatenating from rightmost matrix towards left? */
413			static int
414			prefer_right2left(ROPMAT *mop)
415			{
416			int mri = 0;
417
418			while (mop[mri].binop) /* find rightmost matrix */
419			if (mop[mri++].binop != '.')
420			return(0); /* pre-empt reversal for other ops */
421
422			if (mri <= 1)
423			return(0); /* won't matter */
424
425			if (loadmatrix(mop+mri) < 0) /* load rightmost cat */
426			return(1); /* fail will bail in a moment */
427
428			if (t_ncols(mop+mri) == 1)
429			return(1); /* definitely better R->L */
430
431			if (t_ncols(mop+mri) >= t_nrows(mop+mri))
432			return(0); /* ...probably worse */
433
434			if (loadmatrix(mop) < 0) /* load leftmost */
435			return(0); /* fail will bail in a moment */
436
437			return(t_ncols(mop+mri) < t_nrows(mop));
438			}
439
440	greg	2.1	static int
441			get_factors(double da[], int n, char *av[])
442			{
443			int ac;
444
445			for (ac = 0; ac < n && isflt(av[ac]); ac++)
446			da[ac] = atof(av[ac]);
447			return(ac);
448			}
449
450	greg	2.13	static ROPMAT *
451	greg	2.26	resize_moparr(ROPMAT *mop, int n2alloc)
452	greg	2.13	{
453			int nmats = 0;
454	greg	2.26	int i;
455	greg	2.13
456			while (mop[nmats++].binop)
457			;
458	greg	2.26	for (i = nmats; i > n2alloc; i--)
459			rmx_free(mop[i].mtx);
460	greg	2.13	mop = (ROPMAT )realloc(mop, n2allocsizeof(ROPMAT));
461			if (mop == NULL) {
462	greg	2.26	fputs("Out of memory in resize_moparr()\n", stderr);
463	greg	2.13	exit(1);
464			}
465			if (n2alloc > nmats)
466			memset(mop+nmats, 0, (n2alloc-nmats)*sizeof(ROPMAT));
467			return(mop);
468			}
469
470	greg	2.1	/* Load one or more matrices and operate on them, sending results to stdout */
471			int
472			main(int argc, char *argv[])
473			{
474	greg	2.7	int outfmt = DTfromHeader;
475	greg	2.13	int nall = 2;
476			ROPMAT mop = (ROPMAT )calloc(nall, sizeof(ROPMAT));
477			int nmats = 0;
478	greg	2.1	RMATRIX *mres = NULL;
479	greg	2.13	int stdin_used = 0;
480	greg	2.1	int i;
481			/* get options and arguments */
482	greg	2.13	for (i = 1; i < argc; i++) {
483	greg	2.11	if (argv[i][0] && !argv[i][1] &&
484	greg	2.13	strchr(".+*/", argv[i][0]) != NULL) {
485	greg	2.15	if (!nmats \|\| mop[nmats-1].binop) {
486	greg	2.14	fprintf(stderr,
487	greg	2.15	"%s: missing matrix argument before '%c' operation\n",
488	greg	2.14	argv[0], argv[i][0]);
489	greg	2.13	return(1);
490			}
491	greg	2.15	mop[nmats-1].binop = argv[i][0];
492	greg	2.1	} else if (argv[i][0] != '-' \|\| !argv[i][1]) {
493	greg	2.13	if (argv[i][0] == '-') {
494			if (stdin_used++) {
495			fprintf(stderr,
496			"%s: standard input used for more than one matrix\n",
497			argv[0]);
498			return(1);
499			}
500			mop[nmats].inspec = stdin_name;
501			} else
502			mop[nmats].inspec = argv[i];
503			if (nmats > 0 && !mop[nmats-1].binop)
504			mop[nmats-1].binop = '.';
505			nmats++;
506	greg	2.1	} else {
507			int n = argc-1 - i;
508			switch (argv[i][1]) { /* get option */
509			case 'v':
510	greg	2.14	verbose++;
511	greg	2.1	break;
512			case 'f':
513			switch (argv[i][2]) {
514			case 'd':
515			outfmt = DTdouble;
516			break;
517			case 'f':
518			outfmt = DTfloat;
519			break;
520			case 'a':
521			outfmt = DTascii;
522			break;
523			case 'c':
524			outfmt = DTrgbe;
525			break;
526			default:
527			goto userr;
528			}
529			break;
530			case 't':
531	greg	2.13	mop[nmats].preop.transpose = 1;
532	greg	2.1	break;
533			case 's':
534			if (n > MAXCOMP) n = MAXCOMP;
535	greg	2.13	i += mop[nmats].preop.nsf =
536			get_factors(mop[nmats].preop.sca,
537			n, argv+i+1);
538	greg	2.22	if (mop[nmats].preop.nsf <= 0) {
539			fprintf(stderr, "%s: -s missing arguments\n",
540			argv[0]);
541			goto userr;
542			}
543	greg	2.1	break;
544			case 'c':
545	greg	2.22	if (n && isupper(argv[i+1][0])) {
546			strlcpy(mop[nmats].preop.csym,
547			argv[++i],
548			sizeof(mop[0].preop.csym));
549			mop[nmats].preop.clen = 0;
550			break;
551			}
552	greg	2.1	if (n > MAXCOMPMAXCOMP) n = MAXCOMPMAXCOMP;
553	greg	2.13	i += mop[nmats].preop.clen =
554			get_factors(mop[nmats].preop.cmat,
555			n, argv+i+1);
556	greg	2.22	if (mop[nmats].preop.clen <= 0) {
557			fprintf(stderr, "%s: -c missing arguments\n",
558			argv[0]);
559			goto userr;
560			}
561			mop[nmats].preop.csym[0] = '\0';
562	greg	2.1	break;
563	greg	2.18	case 'r':
564			if (argv[i][2] == 'f')
565			mop[nmats].rmp = RMPreflF;
566			else if (argv[i][2] == 'b')
567			mop[nmats].rmp = RMPreflB;
568			else
569			goto userr;
570			break;
571	greg	2.1	default:
572			fprintf(stderr, "%s: unknown operation '%s'\n",
573			argv[0], argv[i]);
574			goto userr;
575			}
576			}
577	greg	2.14	if (nmats >= nall)
578	greg	2.26	mop = resize_moparr(mop, nall += 2);
579	greg	2.13	}
580			if (mop[0].inspec == NULL) /* nothing to do? */
581	greg	2.1	goto userr;
582	greg	2.15	if (mop[nmats-1].binop) {
583			fprintf(stderr,
584			"%s: missing matrix argument after '%c' operation\n",
585			argv[0], mop[nmats-1].binop);
586			return(1);
587			}
588	greg	2.13	/* favor quicker concatenation */
589	greg	2.14	mop[nmats].mtx = prefer_right2left(mop) ? op_right2left(mop)
590			: op_left2right(mop);
591			if (mop[nmats].mtx == NULL)
592			return(1);
593			/* apply trailing unary operations */
594			mop[nmats].inspec = "trailing_ops";
595			mres = loadop(mop+nmats);
596			if (mres == NULL)
597	greg	2.13	return(1);
598	greg	2.22	if (outfmt == DTfromHeader) /* check data type */
599	greg	2.6	outfmt = mres->dtype;
600	greg	2.22	if (outfmt == DTrgbe) {
601			if (mres->ncomp > 3)
602			outfmt = DTspec;
603			else if (mres->dtype == DTxyze)
604			outfmt = DTxyze;
605			}
606	greg	2.26	newheader("RADIANCE", stdout); /* write result to stdout */
607	greg	2.1	printargs(argc, argv, stdout);
608	greg	2.23	return(rmx_write(mres, outfmt, stdout) ? 0 : 1);
609	greg	2.1	userr:
610			fprintf(stderr,
611	greg	2.26	"Usage: %s [-v][-f{adfc}][-t][-s sf .. \| -c ce ..][-rf\|-rb] m1 [.+*/] .. > mres\n",
612	greg	2.1	argv[0]);
613			return(1);
614			}