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

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

Parent Directory Parent Directory | Revision Log Revision Log


Revision 3642 - (show annotations)
Thu Oct 27 03:41:51 2011 UTC (7 years, 11 months ago) by caltinay
Original Path: trunk/paso/src/SparseMatrix_getSubmatrix.c
File MIME type: text/plain
File size: 6101 byte(s)
Assorted spelling/comment fixes in paso.

1
2 /*******************************************************
3 *
4 * Copyright (c) 2003-2010 by University of Queensland
5 * 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
14
15 /**************************************************************/
16
17 /* Paso: SparseMatrix */
18
19 /**************************************************************/
20
21 /* Copyrights by ACcESS Australia 2003, 2004,2005 */
22 /* Author: Lutz Gross, l.gross@uq.edu.au */
23
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 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){
40 Paso_Pattern* sub_pattern=NULL;
41 Paso_SparseMatrix* out=NULL;
42 index_t index_offset=(A->type & MATRIX_FORMAT_OFFSET1 ? 1:0);
43 int i,k,tmp,m,subpattern_row;
44 int type=A->type;
45 Esys_resetError();
46 if (A->type & MATRIX_FORMAT_CSC) {
47 Esys_setError(TYPE_ERROR, "Paso_SparseMatrix_getSubmatrix: gathering submatrices supports CSR matrix format only.");
48 } else {
49 sub_pattern=Paso_Pattern_getSubpattern(A->pattern,n_row_sub,n_col_sub,row_list,new_col_index);
50 if (Esys_noError()) {
51 /* create the return object */
52 out=Paso_SparseMatrix_alloc(type,sub_pattern,A->row_block_size,A->col_block_size,TRUE);
53 if (Esys_noError()) {
54 #pragma omp parallel for private(i,k,m,subpattern_row,tmp) schedule(static)
55 for (i=0;i<n_row_sub;++i) {
56 subpattern_row=row_list[i];
57 for (k=A->pattern->ptr[subpattern_row]-index_offset;k<A->pattern->ptr[subpattern_row+1]-index_offset;++k) {
58 tmp=new_col_index[A->pattern->index[k]-index_offset];
59 if (tmp>-1) {
60 #pragma ivdep
61 for (m=out->pattern->ptr[i]-index_offset;m<out->pattern->ptr[i+1]-index_offset;++m) {
62 if (out->pattern->index[m]==tmp+index_offset) {
63 Paso_copyShortDouble(A->block_size,&(A->val[k*A->block_size]),&(out->val[m*A->block_size]));
64 break;
65 }
66 }
67 }
68 }
69 }
70 }
71 }
72 Paso_Pattern_free(sub_pattern);
73 }
74 return out;
75 }
76
77 Paso_SparseMatrix* Paso_SparseMatrix_getBlock(Paso_SparseMatrix* A, int blockid){
78
79 dim_t blocksize=A->row_block_size,i;
80 dim_t n=A->numRows;
81 index_t iptr;
82
83 Paso_SparseMatrix* out=NULL;
84
85 out=Paso_SparseMatrix_alloc(A->type,A->pattern, 1, 1, 0);
86
87 if (blocksize==1) {
88 if (blockid==1) {
89 #pragma omp parallel for private(i,iptr) schedule(static)
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 Esys_setError(VALUE_ERROR, "Paso_SparseMatrix_getBlock: Requested and actual block sizes do not match.");
98 }
99 }
100 else if (blocksize==2) {
101 if (blockid==1) {
102 #pragma omp parallel for private(i,iptr) schedule(static)
103 for(i=0;i<n;i++) {
104 for (iptr=A->pattern->ptr[i];iptr<A->pattern->ptr[i+1]; ++iptr) {
105 out->val[iptr]=A->val[4*iptr];
106 }
107 }
108 }
109 else if (blockid==2) {
110 #pragma omp parallel for private(i,iptr) schedule(static)
111 for(i=0;i<n;i++) {
112 for (iptr=A->pattern->ptr[i];iptr<A->pattern->ptr[i+1]; ++iptr) {
113 out->val[iptr]=A->val[4*iptr+3];
114 }
115 }
116 }
117 else {
118 Esys_setError(VALUE_ERROR,"Paso_SparseMatrix_getBlock: Requested and actual block sizes do not match.");
119 }
120 }
121 else if (blocksize==3) {
122 if (blockid==1) {
123 #pragma omp parallel for private(i,iptr) schedule(static)
124 for(i=0;i<n;i++) {
125 for (iptr=A->pattern->ptr[i];iptr<A->pattern->ptr[i+1]; ++iptr) {
126 out->val[iptr]=A->val[9*iptr];
127 }
128 }
129 }
130 else if (blockid==2) {
131 #pragma omp parallel for private(i,iptr) schedule(static)
132 for(i=0;i<n;i++) {
133 for (iptr=A->pattern->ptr[i];iptr<A->pattern->ptr[i+1]; ++iptr) {
134 out->val[iptr]=A->val[9*iptr+4];
135 }
136 }
137 }
138 else if (blockid==3) {
139 #pragma omp parallel for private(i,iptr) schedule(static)
140 for(i=0;i<n;i++) {
141 for (iptr=A->pattern->ptr[i];iptr<A->pattern->ptr[i+1]; ++iptr) {
142 out->val[iptr]=A->val[9*iptr+8];
143 }
144 }
145 }
146 else {
147 Esys_setError(VALUE_ERROR,"Paso_SparseMatrix_getBlock: Requested and actual block sizes do not match.");
148 }
149 }
150
151 return out;
152 }

  ViewVC Help
Powered by ViewVC 1.1.26