diff options
author | Alberto Ruiz <aruiz@um.es> | 2010-02-20 17:08:59 +0000 |
---|---|---|
committer | Alberto Ruiz <aruiz@um.es> | 2010-02-20 17:08:59 +0000 |
commit | dedd74ee85ee1bf468552107760541b31e6f1878 (patch) | |
tree | 2726f637f48498eef1af22ce03741a3353d8ec08 /examples/devel/ej2/functions.c | |
parent | dc9cdee87db0181774cb230610e4b24f9ef3f89a (diff) |
added simpler devel example
Diffstat (limited to 'examples/devel/ej2/functions.c')
-rw-r--r-- | examples/devel/ej2/functions.c | 24 |
1 files changed, 24 insertions, 0 deletions
diff --git a/examples/devel/ej2/functions.c b/examples/devel/ej2/functions.c new file mode 100644 index 0000000..4dcd377 --- /dev/null +++ b/examples/devel/ej2/functions.c | |||
@@ -0,0 +1,24 @@ | |||
1 | /* general element order */ | ||
2 | |||
3 | typedef struct { double r, i; } doublecomplex; | ||
4 | |||
5 | #define DVEC(A) int A##n, double*A##p | ||
6 | #define CVEC(A) int A##n, doublecomplex*A##p | ||
7 | #define DMAT(A) int A##r, int A##c, double*A##p | ||
8 | #define CMAT(A) int A##r, int A##c, doublecomplex*A##p | ||
9 | |||
10 | #define AT(M,r,c) (M##p[(r)*sr+(c)*sc]) | ||
11 | |||
12 | int c_diag(int ro, DMAT(m),DVEC(y),DMAT(z)) { | ||
13 | int i,j,sr,sc; | ||
14 | if (ro==1) { sr = mc; sc = 1;} else { sr = 1; sc = mr;} | ||
15 | for (j=0; j<yn; j++) { | ||
16 | yp[j] = AT(m,j,j); | ||
17 | } | ||
18 | for (i=0; i<mr; i++) { | ||
19 | for (j=0; j<mc; j++) { | ||
20 | AT(z,i,j) = i==j?yp[i]:0; | ||
21 | } | ||
22 | } | ||
23 | return 0; | ||
24 | } | ||