/[escript]/trunk/paso/src/SparseMatrix_getSubmatrix.cpp
ViewVC logotype

Annotation of /trunk/paso/src/SparseMatrix_getSubmatrix.cpp

Parent Directory Parent Directory | Revision Log Revision Log


Revision 2659 - (hide annotations)
Thu Sep 10 03:54:50 2009 UTC (10 years, 2 months ago) by artak
Original Path: trunk/paso/src/SparseMatrix_getSubmatrix.c
File MIME type: text/plain
File size: 5529 byte(s)
first attempt of AMG on blocks
1 ksteube 1315
2     /*******************************************************
3 ksteube 1811 *
4 jfenwick 2548 * Copyright (c) 2003-2009 by University of Queensland
5 ksteube 1811 * Earth Systems Science Computational Center (ESSCC)
6     * http://www.uq.edu.au/esscc
7     *
8     * Primary Business: Queensland, Australia
9     * Licensed under the Open Software License version 3.0
10     * http://www.opensource.org/licenses/osl-3.0.php
11     *
12     *******************************************************/
13 ksteube 1315
14 ksteube 1811
15 ksteube 1315 /**************************************************************/
16    
17     /* Paso: SparseMatrix */
18    
19     /**************************************************************/
20    
21     /* Copyrights by ACcESS Australia 2003, 2004,2005 */
22 jfenwick 2608 /* Author: Lutz Gross, l.gross@uq.edu.au */
23 ksteube 1315
24     /**************************************************************/
25    
26     #include "Paso.h"
27     #include "SparseMatrix.h"
28     #include "PasoUtil.h"
29    
30     /**************************************************************
31    
32     returns the submatrix of A where rows are gathered by index row_list
33     and columns are selected by non-negative values of new_col_index.
34     if new_col_index[i]>-1 new_col_index[i] gives the column of i in
35     the returned submatrix
36    
37     */
38    
39    
40     Paso_SparseMatrix* Paso_SparseMatrix_getSubmatrix(Paso_SparseMatrix* A,int n_row_sub,int n_col_sub, index_t* row_list,index_t* new_col_index){
41     Paso_Pattern* sub_pattern=NULL;
42     Paso_SparseMatrix* out=NULL;
43     index_t index_offset=(A->type & MATRIX_FORMAT_OFFSET1 ? 1:0);
44     int i,k,tmp,m,subpattern_row;
45     int type=A->type;
46     Paso_resetError();
47     if (A->type & MATRIX_FORMAT_CSC) {
48     Paso_setError(TYPE_ERROR,"gathering submatrices supports CSR matrix format only.");
49     } else {
50     sub_pattern=Paso_Pattern_getSubpattern(A->pattern,n_row_sub,n_col_sub,row_list,new_col_index);
51     if (Paso_noError()) {
52     /* create the return object */
53 gross 2551 out=Paso_SparseMatrix_alloc(type,sub_pattern,A->row_block_size,A->col_block_size,TRUE);
54 ksteube 1315 if (Paso_noError()) {
55     #pragma omp parallel for private(i,k,m,subpattern_row,tmp) schedule(static)
56     for (i=0;i<n_row_sub;++i) {
57     subpattern_row=row_list[i];
58     for (k=A->pattern->ptr[subpattern_row]-index_offset;k<A->pattern->ptr[subpattern_row+1]-index_offset;++k) {
59     tmp=new_col_index[A->pattern->index[k]-index_offset];
60     if (tmp>-1) {
61     #pragma ivdep
62     for (m=out->pattern->ptr[i]-index_offset;m<out->pattern->ptr[i+1]-index_offset;++m) {
63     if (out->pattern->index[m]==tmp+index_offset) {
64 gross 1639 Paso_copyShortDouble(A->block_size,&(A->val[k*A->block_size]),&(out->val[m*A->block_size]));
65 ksteube 1315 break;
66     }
67     }
68     }
69     }
70     }
71     }
72     }
73     Paso_Pattern_free(sub_pattern);
74     }
75     return out;
76     }
77 artak 2659
78     Paso_SparseMatrix* Paso_SparseMatrix_getBlock(Paso_SparseMatrix* A, int blockid){
79    
80     dim_t blocksize=A->row_block_size,i;
81     dim_t n=A->numRows;
82     index_t iptr;
83    
84     Paso_SparseMatrix* out=NULL;
85    
86     out=Paso_SparseMatrix_alloc(A->type,A->pattern, 1, 1, 0);
87    
88     if (blocksize==1) {
89     if (blockid==1) {
90     for(i=0;i<n;++i) {
91     for (iptr=A->pattern->ptr[i];iptr<A->pattern->ptr[i+1]; ++iptr) {
92     out->val[iptr]=A->val[iptr];
93     }
94     }
95     }
96     else {
97     Paso_setError(VALUE_ERROR,"Requested and actual block sizes do not match.");
98     }
99     }
100     else if (blocksize==2) {
101     if (blockid==1) {
102     for(i=0;i<n;i++) {
103     for (iptr=A->pattern->ptr[i];iptr<A->pattern->ptr[i+1]; ++iptr) {
104     out->val[iptr]=A->val[4*iptr];
105     }
106     }
107     }
108     else if (blockid==2) {
109     for(i=0;i<n;i++) {
110     for (iptr=A->pattern->ptr[i];iptr<A->pattern->ptr[i+1]; ++iptr) {
111     out->val[iptr]=A->val[4*iptr+3];
112     }
113     }
114     }
115     else {
116     Paso_setError(VALUE_ERROR,"Requested and actual block sizes do not match.");
117     }
118     }
119     else if (blocksize==3) {
120     if (blockid==1) {
121     for(i=0;i<n;i++) {
122     for (iptr=A->pattern->ptr[i];iptr<A->pattern->ptr[i+1]; ++iptr) {
123     out->val[iptr]=A->val[9*iptr];
124     }
125     }
126     }
127     else if (blockid==2) {
128     for(i=0;i<n;i++) {
129     for (iptr=A->pattern->ptr[i];iptr<A->pattern->ptr[i+1]; ++iptr) {
130     out->val[iptr]=A->val[9*iptr+4];
131     }
132     }
133     }
134     else if (blockid==3) {
135     for(i=0;i<n;i++) {
136     for (iptr=A->pattern->ptr[i];iptr<A->pattern->ptr[i+1]; ++iptr) {
137     out->val[iptr]=A->val[9*iptr+8];
138     }
139     }
140     }
141     else {
142     Paso_setError(VALUE_ERROR,"Requested and actual block sizes do not match.");
143     }
144     }
145    
146     return out;
147     }

  ViewVC Help
Powered by ViewVC 1.1.26