--- ray/src/util/rmatrix.h 2023/12/06 17:57:34 2.20 +++ ray/src/util/rmatrix.h 2025/04/04 18:06:48 2.26 @@ -1,4 +1,4 @@ -/* RCSid $Id: rmatrix.h,v 2.20 2023/12/06 17:57:34 greg Exp $ */ +/* RCSid $Id: rmatrix.h,v 2.26 2025/04/04 18:06:48 greg Exp $ */ /* * Header file for general matrix routines. */ @@ -16,22 +16,43 @@ extern "C" { none, transmission, reflection front (normal side), reflection back */ typedef enum {RMPnone=-1, RMPtrans=0, RMPreflF, RMPreflB} RMPref; +/* RMATRIX flags (usually private): + need to swap input, we should free memory */ +#define RMF_SWAPIN 1 +#define RMF_FREEMEM 2 + +#ifndef MAXCOMP +#define MAXCOMP MAXCSAMP /* #components we support */ +#endif + /* Set in-core data type */ +#if !defined(DTrmx_native) || DTrmx_native==DTfloat +#define DTrmx_native DTfloat +#define rmx_dtype float +#define rmx_scanfmt "%f" +#elif DTrmx_native==DTdouble +#define rmx_dtype double +#define rmx_scanfmt "%lf" +#endif + /* General [row][col][cmp] component matrix */ typedef struct { - char *info; - void *mapped; - double *mtx; - COLOR cexp; - float wlpart[4]; - int nrows, ncols; - short ncomp; - uby8 dtype; - uby8 swapin; + char *info; + void *mapped; + rmx_dtype *mtx; + COLOR cexp; + float wlpart[4]; + int nrows, ncols; + short ncomp; + uby8 dtype; + uby8 pflags; } RMATRIX; #define rmx_lval(rm,r,c) ((rm)->mtx + (rm)->ncomp*((c)+(size_t)(rm)->ncols*(r))) #define rmx_val rmx_lval +#define rmx_array_size(rm) (sizeof(rmx_dtype)*(rm)->nrows*(rm)->ncols*(rm)->ncomp) +#define rmx_mapped_size(rm) ((char *)(rm)->mtx + rmx_array_size(rm) - (char *)(rm)->mapped) + /* Initialize a RMATRIX struct but don't allocate array space */ extern RMATRIX *rmx_new(int nr, int nc, int n); @@ -53,8 +74,8 @@ extern int rmx_newtype(int dtyp1, int dtyp2); /* Read matrix header from input stream (cannot be XML) */ extern int rmx_load_header(RMATRIX *rm, FILE *fp); -/* Load next row as double (cannot be XML) */ -extern int rmx_load_row(double *drp, const RMATRIX *rm, FILE *fp); +/* Load next row as rmx_dtype (cannot be XML) */ +extern int rmx_load_row(rmx_dtype *drp, const RMATRIX *rm, FILE *fp); /* Allocate & load post-header data from stream given type set in rm->dtype */ extern int rmx_load_data(RMATRIX *rm, FILE *fp); @@ -69,7 +90,7 @@ extern int rmx_addinfo(RMATRIX *rm, const char *info); extern int rmx_write_header(const RMATRIX *rm, int dtype, FILE *fp); /* Write out matrix data (usually by row) */ -extern int rmx_write_data(const double *dp, int nc, int len, +extern int rmx_write_data(const rmx_dtype *dp, int nc, int len, int dtype, FILE *fp); /* Write matrix using file format indicated by dtype */ @@ -84,8 +105,8 @@ extern RMATRIX *rmx_copy(const RMATRIX *rm); /* Replace data in first matrix with data from second */ extern int rmx_transfer_data(RMATRIX *rdst, RMATRIX *rsrc, int dometa); -/* Allocate and assign transposed matrix */ -extern RMATRIX *rmx_transpose(const RMATRIX *rm); +/* Transpose the given matrix */ +extern int rmx_transpose(RMATRIX *rm); /* Multiply (concatenate) two matrices and allocate the result */ extern RMATRIX *rmx_multiply(const RMATRIX *m1, const RMATRIX *m2);