| 1 |
/* RCSid $Id: rmatrix.h,v 2.16 2023/11/21 01:30:20 greg Exp $ */ |
| 2 |
/* |
| 3 |
* Header file for general matrix routines. |
| 4 |
*/ |
| 5 |
|
| 6 |
#ifndef _RAD_RMATRIX_H_ |
| 7 |
#define _RAD_RMATRIX_H_ |
| 8 |
|
| 9 |
#include "cmatrix.h" |
| 10 |
|
| 11 |
#ifdef __cplusplus |
| 12 |
extern "C" { |
| 13 |
#endif |
| 14 |
|
| 15 |
/* Preferred BSDF component: |
| 16 |
transmission, reflection front (normal side), reflection back */ |
| 17 |
typedef enum {RMPtrans=0, RMPreflF, RMPreflB} RMPref; |
| 18 |
|
| 19 |
/* General [row][col][cmp] component matrix */ |
| 20 |
typedef struct { |
| 21 |
char *info; |
| 22 |
void *mapped; |
| 23 |
double *mtx; |
| 24 |
COLOR cexp; |
| 25 |
float wlpart[4]; |
| 26 |
int nrows, ncols; |
| 27 |
short ncomp; |
| 28 |
uby8 dtype; |
| 29 |
uby8 swapin; |
| 30 |
} RMATRIX; |
| 31 |
|
| 32 |
#define rmx_lval(rm,r,c) ((rm)->mtx + (rm)->ncomp*((c)+(size_t)(rm)->ncols*(r))) |
| 33 |
#define rmx_val rmx_lval |
| 34 |
|
| 35 |
/* Initialize a RMATRIX struct but don't allocate array space */ |
| 36 |
extern RMATRIX *rmx_new(int nr, int nc, int n); |
| 37 |
|
| 38 |
/* Prepare a RMATRIX for writing (allocate array if needed) */ |
| 39 |
extern int rmx_prepare(RMATRIX *rm); |
| 40 |
|
| 41 |
/* Call rmx_new() and rmx_prepare() */ |
| 42 |
extern RMATRIX *rmx_alloc(int nr, int nc, int n); |
| 43 |
|
| 44 |
/* Free a RMATRIX array */ |
| 45 |
extern void rmx_free(RMATRIX *rm); |
| 46 |
|
| 47 |
/* Resolve data type based on two input types (returns 0 for mismatch) */ |
| 48 |
extern int rmx_newtype(int dtyp1, int dtyp2); |
| 49 |
|
| 50 |
/* Read matrix header from input stream (cannot be XML) */ |
| 51 |
extern int rmx_load_header(RMATRIX *rm, FILE *fp); |
| 52 |
|
| 53 |
/* Allocate & load post-header data from stream given type set in rm->dtype */ |
| 54 |
extern int rmx_load_data(RMATRIX *rm, FILE *fp); |
| 55 |
|
| 56 |
/* Load matrix from supported file type (NULL for stdin, '!' with command) */ |
| 57 |
extern RMATRIX *rmx_load(const char *inspec, RMPref rmp); |
| 58 |
|
| 59 |
/* Append header information associated with matrix data */ |
| 60 |
extern int rmx_addinfo(RMATRIX *rm, const char *info); |
| 61 |
|
| 62 |
/* Write matrix to file type indicated by dtype */ |
| 63 |
extern int rmx_write(const RMATRIX *rm, int dtype, FILE *fp); |
| 64 |
|
| 65 |
/* Allocate and assign square identity matrix with n components */ |
| 66 |
extern RMATRIX *rmx_identity(int dim, int n); |
| 67 |
|
| 68 |
/* Duplicate the given matrix */ |
| 69 |
extern RMATRIX *rmx_copy(const RMATRIX *rm); |
| 70 |
|
| 71 |
/* Allocate and assign transposed matrix */ |
| 72 |
extern RMATRIX *rmx_transpose(const RMATRIX *rm); |
| 73 |
|
| 74 |
/* Multiply (concatenate) two matrices and allocate the result */ |
| 75 |
extern RMATRIX *rmx_multiply(const RMATRIX *m1, const RMATRIX *m2); |
| 76 |
|
| 77 |
/* Element-wise multiplication (or division) of m2 into m1 */ |
| 78 |
extern int rmx_elemult(RMATRIX *m1, const RMATRIX *m2, int divide); |
| 79 |
|
| 80 |
/* Sum second matrix into first, applying scale factor beforehand */ |
| 81 |
extern int rmx_sum(RMATRIX *msum, const RMATRIX *madd, const double sf[]); |
| 82 |
|
| 83 |
/* Scale the given matrix by the indicated scalar component vector */ |
| 84 |
extern int rmx_scale(RMATRIX *rm, const double sf[]); |
| 85 |
|
| 86 |
/* Allocate new matrix and apply component transformation */ |
| 87 |
extern RMATRIX *rmx_transform(const RMATRIX *msrc, int n, const double cmat[]); |
| 88 |
|
| 89 |
/* Convert a color matrix to newly allocated RMATRIX buffer */ |
| 90 |
extern RMATRIX *rmx_from_cmatrix(const CMATRIX *cm); |
| 91 |
|
| 92 |
/* Convert general matrix to newly allocated CMATRIX buffer */ |
| 93 |
extern CMATRIX *cm_from_rmatrix(const RMATRIX *rm); |
| 94 |
|
| 95 |
#ifdef __cplusplus |
| 96 |
} |
| 97 |
#endif |
| 98 |
#endif /* _RAD_RMATRIX_H_ */ |
