aboutsummaryrefslogtreecommitdiffstats
path: root/matrix.h
blob: 254a2e6842531e92f647dd5ec7f636a81047300c (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
/* 
 * matrix.h, matrix.c: Liner equation solver using LU decomposition.
 * $Id: matrix.h,v 1.1 2001/11/08 05:15:21 a-ito Exp $
 * 
 * by K.Okabe  Aug. 1999
 * 
 * You can use,copy,modify and distribute this program without any permission.
 */

#ifndef _MATRIX_H
#include <math.h>
#include <string.h>

/* 
 * Types.
 */

struct matrix {
    double *me;
    int dim;
};

struct vector {
    double *ve;
    int dim;
};

typedef struct matrix *Matrix;
typedef struct vector *Vector;

/* 
 * Macros.
 */

#define M_VAL(m,i,j) ((m)->me[(i)*(m)->dim +(j)])
#define V_VAL(v,i) ((v)->ve[i])

/* 
 * Compatible macros with those in Meschach Library.
 */

#define m_entry(m,i,j) (M_VAL(m,i,j))
#define v_entry(v,i) (V_VAL(v,i))
#ifdef __CYGWIN__
#define m_copy(m1,m2) (bcopy((const char *)(m1)->me,(char *)(m2)->me,(m1)->dim*(m1)->dim*sizeof(double)))
#else				/* not __CYGWIN__ */
#define m_copy(m1,m2) (bcopy((m1)->me,(m2)->me,(m1)->dim*(m1)->dim*sizeof(double)))
#endif				/* not __CYGWIN__ */
#define v_free(v) ((v)=NULL)
#define m_free(m) ((m)=NULL)
#define px_free(px) ((px)=NULL)
#define m_set_val(m,i,j,x) (M_VAL(m,i,j) = (x))
#define m_add_val(m,i,j,x) (M_VAL(m,i,j) += (x))
#define v_set_val(v,i,x) (V_VAL(v,i) = (x))
#define v_add_val(v,i,x) (V_VAL(v,i) += (x))
#define m_get(r,c) (new_matrix(r))
#define v_get(n) (new_vector(n))
#define px_get(n) (New_N(int,n))
typedef struct matrix MAT;
typedef struct vector VEC;
typedef int PERM;

extern int LUfactor(Matrix, int *);
extern Matrix m_inverse(Matrix, Matrix);
extern Matrix LUinverse(Matrix, int *, Matrix);
extern int LUsolve(Matrix, int *, Vector, Vector);
extern int Lsolve(Matrix, Vector, Vector, double);
extern int Usolve(Matrix, Vector, Vector, double);
extern Matrix new_matrix(int);
extern Vector new_vector(int);

#define _MATRIX_H
#endif				/* _MATRIX_H */