Lib/matrix.h

Go to the documentation of this file.

#ifndef  __JBXL_MATRIX_LIBRARY_H_
#define  __JBXL_MATRIX_LIBRARY_H_


#include "common.h"
#include "mt.h"

#include <math.h>
#include <float.h>



#ifndef va_start
    #include <stdarg.h>
#endif


typedef struct _ivector {
    int x;                  
    int y;                  
    int z;                  
    double n;               
} ivector;


typedef struct _vector {
    double x;               
    double y;               
    double z;               
    double n;               
} vector;


typedef struct _quaternion {
    double s;               
    double x;               
    double y;               
    double z;               
    double n;               
} quaternion;



typedef struct _imatrix {
    int n;                  
    int r;                  
    int *sz;                
    int *mx;                
} imatrix;


typedef struct _matrix {
    int n;                  
    int r;                  
    int *sz;                
    double *mx;             
} matrix;


#define  in_vector(a, b)   ((a).x*(b).x+(a).y*(b).y+(a).z*(b).z)                              
#define  add_vector(a, b)  set_vector((a).x+(b).x,(a).y+(b).y,(a).z+(b).z)                    
#define  sub_vector(a, b)  set_vector((a).x-(b).x,(a).y-(b).y,(a).z-(b).z)                    
#define  add_ivector(a, b) set_ivector((a).x+(b).x,(a).y+(b).y,(a).z+(b).z)                   
#define  sub_ivector(a, b) set_ivector((a).x-(b).x,(a).y-(b).y,(a).z-(b).z)                   
#define  normal_vector(a)  (a).r = sqrt((double)((a).x*(a).x)+(a).y*(a).y+(a).z*(a).z);       


#define Vt(m, i)      ((m).mx[(i)-1])                                                         
#define Mx(m, i, j)    ((m).mx[(j)-1+(m).sz[1]*((i)-1)])                                      

#define Mx1(m, i)                ((m).mx[(i)-1])                                              
#define Mx2(m, i, j)       ((m).mx[(j)-1+(m).sz[1]*((i)-1)])                                  
#define Mx3(m, i, j, k)    ((m).mx[(k)-1+(m).sz[2]*((j)-1)+(m).sz[1]*(m).sz[2]*((i)-1)])      
#define Mx4(m, i, j, k, l) ((m).mx[(l)-1+(m).sz[3]*((k)-1)+(m).sz[2]*(m).sz[3]*((j)-1) +\
                            (m).sz[1]*(m).sz[2]*(m).sz[3]*((i)-1)])                           


vector   unit_vector(vector a);
vector   unit_ivector(ivector a);
vector   set_vector(double x, double y, double z);
ivector  set_ivector(int x, int y, int z);
vector   ex_vector(vector a, vector b);

ivector  f2ivector(vector a);
vector   i2vector(ivector a);

matrix   make_matrix1(int n);
matrix   make_matrix2(int n, int m);
matrix   make_matrix (int n, int* sz);

imatrix  make_imatrix1(int n);
imatrix  make_imatrix2(int n, int m);
imatrix  make_imatrix (int n, int* sz);


double*  get_matrix(matrix a, ...);
matrix   add_matrix(matrix a, matrix b);
matrix   sub_matrix(matrix a, matrix b);
matrix   mlt_matrix(matrix a, matrix b);
void     copy_matrix(matrix a, matrix b);
void     free_matrix(matrix* a);
void     clear_matrix(matrix* a);
void     print_matrix(FILE* fp, matrix a);

int*     get_imatrix(imatrix a, ...);
imatrix  add_imatrix(imatrix a, imatrix b);
imatrix  sub_imatrix(imatrix a, imatrix b);
imatrix  mlt_imatrix(imatrix a, imatrix b);
void     copy_imatrix(imatrix a, imatrix b);
void     free_imatrix(imatrix* a);
void     clear_imatrix(imatrix* a);
void     print_imatrix(FILE* fp, imatrix a);

//
matrix   decompQR(matrix xx, imatrix col);
matrix   trans_matrix(matrix a);
matrix   minimum2(matrix a, matrix x);
matrix   invrU_matrix(matrix x);


#endif

---