src/util/rmtxop.c

#ifndef lint
static const char RCSid[] = "$Id: rmtxop.c,v 2.36 2025/03/28 00:06:36 greg Exp $";
#endif
/*
 * General component matrix operations.
 */

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

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

/* Unary matrix operation(s) */
typedef struct {
        double          cmat[MAXCOMP*MAXCOMP];  /* component transformation */
        double          sca[MAXCOMP];           /* scalar coefficients */
        const char      *csym;                  /* symbolic coefs or file */
        short           clen;                   /* number of coefficients */
        short           nsf;                    /* number of scalars */
        short           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 */
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);
}

extern int      checksymbolic(ROPMAT *rop);

/* Check/set transform based on a reference input file */
int
checkreffile(ROPMAT *rop)
{
        static const char       *curRF = NULL;
        static RMATRIX          refm;
        const int               nc = rop->mtx->ncomp;
        int                     i;

        if (!curRF || strcmp(rop->preop.csym, curRF)) {
                FILE    *fp = fopen(rop->preop.csym, "rb");
                if (!rmx_load_header(&refm, fp)) {
                        fprintf(stderr, "%s: cannot read info header\n",
                                        rop->preop.csym);
                        curRF = NULL;
                        if (fp) fclose(fp);
                        return(-1);
                }
                fclose(fp);
                curRF = rop->preop.csym;
        }
        if (refm.ncomp == 3) {
                rop->preop.csym = (refm.dtype == DTxyze) ? "XYZ" : "RGB";
                return(checksymbolic(rop));
        }
        if (refm.ncomp == 2) {
                fprintf(stderr, "%s: cannot convert to 2 components\n",
                                curRF);
                return(-1);
        }
        if (refm.ncomp == 1) {
                rop->preop.csym = "Y";          /* XXX big assumption */
                return(checksymbolic(rop));
        }
        if (refm.ncomp == nc &&
                        !memcmp(refm.wlpart, rop->mtx->wlpart, sizeof(refm.wlpart)))
                return(0);                      /* nothing to do */

        if ((nc <= 3) | (nc > MAXCSAMP) | (refm.ncomp > MAXCSAMP)) {
                fprintf(stderr, "%s: cannot resample from %d to %d components\n",
                                curRF, nc, refm.ncomp);
                return(-1);
        }
        rop->preop.clen = refm.ncomp * nc;      /* compute spec to ref */

        for (i = 0; i < nc; i++) {
                SCOLOR  scstim, scresp;
                int     j;
                memset(scstim, 0, sizeof(COLORV)*nc);
                scstim[i] = 1.f;
                convertscolor(scresp, refm.ncomp, refm.wlpart[0], refm.wlpart[3],
                                scstim, nc, rop->mtx->wlpart[0], rop->mtx->wlpart[3]);
                for (j = refm.ncomp; j-- > 0; )
                        rop->preop.cmat[j*nc + i] = scresp[j];
        }
        memcpy(rop->mtx->wlpart, refm.wlpart, sizeof(rop->mtx->wlpart));
        return(0);
}

/* Compute conversion row from spectrum to one channel of RGB */
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 */
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 */
void
sensrow(ROPMAT *rop, int r, double (*sf)(const SCOLOR sc, int ncs, const float wlpt[4]))
{
        const int       nc = rop->mtx->ncomp;
        int             i;

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

/* Check/set symbolic transform */
int
checksymbolic(ROPMAT *rop)
{
        const int       nc = rop->mtx->ncomp;
        const int       dt = rop->mtx->dtype;
        double          cf = 1;
        int             i, j;
                                        /* check suffix => reference file */
        if (strchr(rop->preop.csym, '.') > rop->preop.csym)
                return(checkreffile(rop));

        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':
                case 'b':
                        ++comp;
                        /* fall through */
                case 'G':
                case 'g':
                        ++comp;
                        /* fall through */
                case 'R':
                case 'r':
                        if (rop->preop.csym[j] <= 'Z')
                                cf = 1./WHTEFFICACY;
                        if (dt == DTxyze) {
                                for (i = 3; i--; )
                                        rop->preop.cmat[j*nc+i] = cf*xyz2rgbmat[comp][i];
                        } else if (nc == 3)
                                rop->preop.cmat[j*nc+comp] = 1.;
                        else
                                rgbrow(rop, j, comp);
                        break;
                case 'Z':
                case 'z':
                        ++comp;
                        /* fall through */
                case 'Y':
                case 'y':
                        ++comp;
                        /* fall through */
                case 'X':
                case 'x':
                        if ((rop->preop.csym[j] <= 'Z') & (dt != DTxyze))
                                cf = WHTEFFICACY;
                        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*(cf != 1); i--; )
                                rop->preop.cmat[j*nc+i] *= cf;
                        break;
                case 'S':               /* scotopic (il)luminance */
                        cf = WHTSCOTOPIC;
                        /* fall through */
                case 's':
                        sensrow(rop, j, scolor2scotopic);
                        for (i = nc*(cf != 1); i--; )
                                rop->preop.cmat[j*nc+i] *= cf;
                        break;
                case 'M':               /* melanopic (il)luminance */
                        cf = WHTMELANOPIC;
                        /* fall through */
                case 'm':
                        sensrow(rop, j, scolor2melanopic);
                        for (i = nc*(cf != 1); i--; )
                                rop->preop.cmat[j*nc+i] *= cf;
                        break;
                case 'A':               /* average component */
                case 'a':
                        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 (!strcasecmp(rop->preop.csym, "XYZ")) {
                if (dt <= DTspec)
                        return(DTxyze);
        } else if (!strcasecmp(rop->preop.csym, "RGB")) {
                if (dt <= DTspec)
                        return(DTrgbe);
        } else if (dt == DTspec)
                return(DTfloat);        /* probably not actual spectrum */
        return(0);
}

/* Get matrix and perform unary operations */
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 &&                  /* 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 != rop->preop.clen) {
                                fprintf(stderr, "%s: -s must have one or %d factors\n",
                                                rop->inspec,
                                                rop->preop.clen/rop->mtx->ncomp);
                                goto failure;
                        } else {
                                for (i = rop->preop.nsf; i--; )
                                        for (j = rop->mtx->ncomp; j--; )
                                                rop->preop.cmat[i*rop->mtx->ncomp+j]
                                                                *= 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? */
                if (!rmx_transpose(rop->mtx)) {
                        fputs(rop->inspec, stderr);
                        fputs(": transpose failed\n", stderr);
                        goto failure;
                }
                if (verbose) {
                        fputs(rop->inspec, stderr);
                        fputs(": transposed rows and columns\n", stderr);
                }
        }
        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 */
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 */
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 */
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? */
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));
}

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);
}

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