xref: /petsc/src/tao/matrix/adamat.c (revision 0e6606419405417ae11ca0744afbced7b5e3372a)
1a7e14dcfSSatish Balay #include "adamat.h"                /*I  "mat.h"  I*/
2a7e14dcfSSatish Balay 
3a7e14dcfSSatish Balay #undef __FUNCT__
4a7e14dcfSSatish Balay #define __FUNCT__ "MatCreateADA"
5a7e14dcfSSatish Balay /*@C
6a7e14dcfSSatish Balay    MatCreateADA - Creates a matrix M=A^T D1 A + D2 where D1, D2 are diagonal
7a7e14dcfSSatish Balay 
8a7e14dcfSSatish Balay    Collective on matrix
9a7e14dcfSSatish Balay 
10a7e14dcfSSatish Balay    Input Parameters:
11a7e14dcfSSatish Balay +  mat - matrix of arbitrary type
12a7e14dcfSSatish Balay .  d1 - A vector with diagonal elements of D1
13a7e14dcfSSatish Balay -  d2 - A vector
14a7e14dcfSSatish Balay 
15a7e14dcfSSatish Balay    Output Parameters:
16a7e14dcfSSatish Balay .  J - New matrix whose operations are defined in terms of mat, D1, and D2.
17a7e14dcfSSatish Balay 
18a7e14dcfSSatish Balay    Notes:
19a7e14dcfSSatish Balay    The user provides the input data and is responsible for destroying
20a7e14dcfSSatish Balay    this data after matrix J has been destroyed.
21a7e14dcfSSatish Balay    The operation MatMult(A,D2,D1) must be well defined.
22a7e14dcfSSatish Balay    Before calling the operation MatGetDiagonal(), the function
23a7e14dcfSSatish Balay    MatADAComputeDiagonal() must be called.  The matrices A and D1 must
24a7e14dcfSSatish Balay    be the same during calls to MatADAComputeDiagonal() and
25a7e14dcfSSatish Balay    MatGetDiagonal().
26a7e14dcfSSatish Balay 
27a7e14dcfSSatish Balay    Level: developer
28a7e14dcfSSatish Balay 
29a7e14dcfSSatish Balay .seealso: MatCreate()
30a7e14dcfSSatish Balay @*/
31a7e14dcfSSatish Balay PetscErrorCode MatCreateADA(Mat mat,Vec d1, Vec d2, Mat *J)
32a7e14dcfSSatish Balay {
33a7e14dcfSSatish Balay   MPI_Comm       comm=((PetscObject)mat)->comm;
34a7e14dcfSSatish Balay   TaoMatADACtx   ctx;
35a7e14dcfSSatish Balay   PetscErrorCode ierr;
36a7e14dcfSSatish Balay   PetscInt       nloc,n;
37a7e14dcfSSatish Balay 
38a7e14dcfSSatish Balay   PetscFunctionBegin;
393c9e27cfSGeoffrey Irving   ierr = PetscNew(&ctx);CHKERRQ(ierr);
40a7e14dcfSSatish Balay   ctx->A=mat;
41a7e14dcfSSatish Balay   ctx->D1=d1;
42a7e14dcfSSatish Balay   ctx->D2=d2;
43a7e14dcfSSatish Balay   if (d1){
44a7e14dcfSSatish Balay     ierr = VecDuplicate(d1,&ctx->W);CHKERRQ(ierr);
45a7e14dcfSSatish Balay     ierr = PetscObjectReference((PetscObject)d1);CHKERRQ(ierr);
46a7e14dcfSSatish Balay   } else {
47*0e660641SBarry Smith     ctx->W = NULL;
48a7e14dcfSSatish Balay   }
49a7e14dcfSSatish Balay   if (d2){
50a7e14dcfSSatish Balay     ierr = VecDuplicate(d2,&ctx->W2);CHKERRQ(ierr);
51a7e14dcfSSatish Balay     ierr = VecDuplicate(d2,&ctx->ADADiag);CHKERRQ(ierr);
52a7e14dcfSSatish Balay     ierr =  PetscObjectReference((PetscObject)d2);CHKERRQ(ierr);
53a7e14dcfSSatish Balay   } else {
54*0e660641SBarry Smith     ctx->W2      = NULL;
55*0e660641SBarry Smith     ctx->ADADiag = NULL;
56a7e14dcfSSatish Balay   }
57a7e14dcfSSatish Balay 
58a7e14dcfSSatish Balay   ctx->GotDiag=0;
59a7e14dcfSSatish Balay   ierr =  PetscObjectReference((PetscObject)mat);CHKERRQ(ierr);
60a7e14dcfSSatish Balay 
61a7e14dcfSSatish Balay   ierr=VecGetLocalSize(d2,&nloc);CHKERRQ(ierr);
62a7e14dcfSSatish Balay   ierr=VecGetSize(d2,&n);CHKERRQ(ierr);
63a7e14dcfSSatish Balay 
64a7e14dcfSSatish Balay   ierr = MatCreateShell(comm,nloc,nloc,n,n,ctx,J);CHKERRQ(ierr);
65a7e14dcfSSatish Balay 
66a7e14dcfSSatish Balay   ierr = MatShellSetOperation(*J,MATOP_MULT,(void(*)(void))MatMult_ADA);CHKERRQ(ierr);
67a7e14dcfSSatish Balay   ierr = MatShellSetOperation(*J,MATOP_DESTROY,(void(*)(void))MatDestroy_ADA);CHKERRQ(ierr);
68a7e14dcfSSatish Balay   ierr = MatShellSetOperation(*J,MATOP_VIEW,(void(*)(void))MatView_ADA);CHKERRQ(ierr);
69a7e14dcfSSatish Balay   ierr = MatShellSetOperation(*J,MATOP_MULT_TRANSPOSE,(void(*)(void))MatMultTranspose_ADA);CHKERRQ(ierr);
70a7e14dcfSSatish Balay   ierr = MatShellSetOperation(*J,MATOP_DIAGONAL_SET,(void(*)(void))MatDiagonalSet_ADA);CHKERRQ(ierr);
71a7e14dcfSSatish Balay   ierr = MatShellSetOperation(*J,MATOP_SHIFT,(void(*)(void))MatShift_ADA);CHKERRQ(ierr);
72a7e14dcfSSatish Balay   ierr = MatShellSetOperation(*J,MATOP_EQUAL,(void(*)(void))MatEqual_ADA);CHKERRQ(ierr);
73a7e14dcfSSatish Balay   ierr = MatShellSetOperation(*J,MATOP_SCALE,(void(*)(void))MatScale_ADA);CHKERRQ(ierr);
74a7e14dcfSSatish Balay   ierr = MatShellSetOperation(*J,MATOP_TRANSPOSE,(void(*)(void))MatTranspose_ADA);CHKERRQ(ierr);
75a7e14dcfSSatish Balay   ierr = MatShellSetOperation(*J,MATOP_GET_DIAGONAL,(void(*)(void))MatGetDiagonal_ADA);CHKERRQ(ierr);
76a7e14dcfSSatish Balay   ierr = MatShellSetOperation(*J,MATOP_GET_SUBMATRICES,(void(*)(void))MatGetSubMatrices_ADA);CHKERRQ(ierr);
77a7e14dcfSSatish Balay   ierr = MatShellSetOperation(*J,MATOP_NORM,(void(*)(void))MatNorm_ADA);CHKERRQ(ierr);
78a7e14dcfSSatish Balay   ierr = MatShellSetOperation(*J,MATOP_DUPLICATE,(void(*)(void))MatDuplicate_ADA);CHKERRQ(ierr);
79a7e14dcfSSatish Balay   ierr = MatShellSetOperation(*J,MATOP_GET_SUBMATRIX,(void(*)(void))MatGetSubMatrix_ADA);CHKERRQ(ierr);
80a7e14dcfSSatish Balay 
81a7e14dcfSSatish Balay   ierr = PetscLogObjectParent((PetscObject)(*J),(PetscObject)ctx->W);CHKERRQ(ierr);
82a7e14dcfSSatish Balay   ierr = PetscLogObjectParent((PetscObject)mat,(PetscObject)(*J));CHKERRQ(ierr);
83a7e14dcfSSatish Balay 
84a7e14dcfSSatish Balay   ierr = MatSetOption(*J,MAT_SYMMETRIC,PETSC_TRUE);CHKERRQ(ierr);
85a7e14dcfSSatish Balay   PetscFunctionReturn(0);
86a7e14dcfSSatish Balay }
87a7e14dcfSSatish Balay 
88a7e14dcfSSatish Balay #undef __FUNCT__
89a7e14dcfSSatish Balay #define __FUNCT__ "MatMult_ADA"
90a7e14dcfSSatish Balay PetscErrorCode MatMult_ADA(Mat mat,Vec a,Vec y)
91a7e14dcfSSatish Balay {
92a7e14dcfSSatish Balay   TaoMatADACtx   ctx;
93a7e14dcfSSatish Balay   PetscReal      one = 1.0;
94a7e14dcfSSatish Balay   PetscErrorCode ierr;
95a7e14dcfSSatish Balay 
96a7e14dcfSSatish Balay   PetscFunctionBegin;
97a7e14dcfSSatish Balay   ierr = MatShellGetContext(mat,(void **)&ctx);CHKERRQ(ierr);
98a7e14dcfSSatish Balay   ierr = MatMult(ctx->A,a,ctx->W);CHKERRQ(ierr);
99a7e14dcfSSatish Balay   if (ctx->D1){
100a7e14dcfSSatish Balay     ierr = VecPointwiseMult(ctx->W,ctx->D1,ctx->W);CHKERRQ(ierr);
101a7e14dcfSSatish Balay   }
102a7e14dcfSSatish Balay   ierr = MatMultTranspose(ctx->A,ctx->W,y);CHKERRQ(ierr);
103a7e14dcfSSatish Balay   if (ctx->D2){
104a7e14dcfSSatish Balay     ierr = VecPointwiseMult(ctx->W2, ctx->D2, a);CHKERRQ(ierr);
105a7e14dcfSSatish Balay     ierr = VecAXPY(y, one, ctx->W2);CHKERRQ(ierr);
106a7e14dcfSSatish Balay   }
107a7e14dcfSSatish Balay   PetscFunctionReturn(0);
108a7e14dcfSSatish Balay }
109a7e14dcfSSatish Balay 
110a7e14dcfSSatish Balay #undef __FUNCT__
111a7e14dcfSSatish Balay #define __FUNCT__ "MatMultTranspose_ADA"
112a7e14dcfSSatish Balay PetscErrorCode MatMultTranspose_ADA(Mat mat,Vec a,Vec y)
113a7e14dcfSSatish Balay {
114a7e14dcfSSatish Balay   PetscErrorCode ierr;
115a7e14dcfSSatish Balay 
116a7e14dcfSSatish Balay   PetscFunctionBegin;
117a7e14dcfSSatish Balay   ierr = MatMult_ADA(mat,a,y);CHKERRQ(ierr);
118a7e14dcfSSatish Balay   PetscFunctionReturn(0);
119a7e14dcfSSatish Balay }
120a7e14dcfSSatish Balay 
121a7e14dcfSSatish Balay #undef __FUNCT__
122a7e14dcfSSatish Balay #define __FUNCT__ "MatDiagonalSet_ADA"
123a7e14dcfSSatish Balay PetscErrorCode MatDiagonalSet_ADA(Vec D, Mat M)
124a7e14dcfSSatish Balay {
125a7e14dcfSSatish Balay   TaoMatADACtx   ctx;
126a7e14dcfSSatish Balay   PetscReal      zero=0.0,one = 1.0;
127a7e14dcfSSatish Balay   PetscErrorCode ierr;
128a7e14dcfSSatish Balay 
129a7e14dcfSSatish Balay   PetscFunctionBegin;
130a7e14dcfSSatish Balay   ierr = MatShellGetContext(M,(void **)&ctx);CHKERRQ(ierr);
131*0e660641SBarry Smith   if (!ctx->D2){
132a7e14dcfSSatish Balay     ierr = VecDuplicate(D,&ctx->D2);CHKERRQ(ierr);
133a7e14dcfSSatish Balay     ierr = VecSet(ctx->D2, zero);CHKERRQ(ierr);
134a7e14dcfSSatish Balay   }
135a7e14dcfSSatish Balay   ierr = VecAXPY(ctx->D2, one, D);CHKERRQ(ierr);
136a7e14dcfSSatish Balay   PetscFunctionReturn(0);
137a7e14dcfSSatish Balay }
138a7e14dcfSSatish Balay 
139a7e14dcfSSatish Balay #undef __FUNCT__
140a7e14dcfSSatish Balay #define __FUNCT__ "MatDestroy_ADA"
141a7e14dcfSSatish Balay PetscErrorCode MatDestroy_ADA(Mat mat)
142a7e14dcfSSatish Balay {
143a7e14dcfSSatish Balay   PetscErrorCode ierr;
144a7e14dcfSSatish Balay   TaoMatADACtx   ctx;
145a7e14dcfSSatish Balay 
146a7e14dcfSSatish Balay   PetscFunctionBegin;
147a7e14dcfSSatish Balay   ierr=MatShellGetContext(mat,(void **)&ctx);CHKERRQ(ierr);
148a7e14dcfSSatish Balay   ierr=VecDestroy(&ctx->W);CHKERRQ(ierr);
149a7e14dcfSSatish Balay   ierr=VecDestroy(&ctx->W2);CHKERRQ(ierr);
150a7e14dcfSSatish Balay   ierr=VecDestroy(&ctx->ADADiag);CHKERRQ(ierr);
151a7e14dcfSSatish Balay   ierr=MatDestroy(&ctx->A);CHKERRQ(ierr);
152a7e14dcfSSatish Balay   ierr=VecDestroy(&ctx->D1);CHKERRQ(ierr);
153a7e14dcfSSatish Balay   ierr=VecDestroy(&ctx->D2);CHKERRQ(ierr);
154a7e14dcfSSatish Balay   ierr = PetscFree(ctx);CHKERRQ(ierr);
155a7e14dcfSSatish Balay   PetscFunctionReturn(0);
156a7e14dcfSSatish Balay }
157a7e14dcfSSatish Balay 
158a7e14dcfSSatish Balay #undef __FUNCT__
159a7e14dcfSSatish Balay #define __FUNCT__ "MatView_ADA"
160a7e14dcfSSatish Balay PetscErrorCode MatView_ADA(Mat mat,PetscViewer viewer)
161a7e14dcfSSatish Balay {
162a7e14dcfSSatish Balay   PetscFunctionBegin;
163a7e14dcfSSatish Balay   PetscFunctionReturn(0);
164a7e14dcfSSatish Balay }
165a7e14dcfSSatish Balay 
166a7e14dcfSSatish Balay #undef __FUNCT__
167a7e14dcfSSatish Balay #define __FUNCT__ "MatShift_ADA"
168a7e14dcfSSatish Balay PetscErrorCode MatShift_ADA(Mat Y, PetscReal a)
169a7e14dcfSSatish Balay {
170a7e14dcfSSatish Balay   PetscErrorCode ierr;
171a7e14dcfSSatish Balay   TaoMatADACtx   ctx;
172a7e14dcfSSatish Balay 
173a7e14dcfSSatish Balay   PetscFunctionBegin;
174a7e14dcfSSatish Balay   ierr = MatShellGetContext(Y,(void **)&ctx);CHKERRQ(ierr);
175a7e14dcfSSatish Balay   ierr = VecShift(ctx->D2,a);CHKERRQ(ierr);
176a7e14dcfSSatish Balay   PetscFunctionReturn(0);
177a7e14dcfSSatish Balay }
178a7e14dcfSSatish Balay 
179a7e14dcfSSatish Balay #undef __FUNCT__
180a7e14dcfSSatish Balay #define __FUNCT__ "MatDuplicate_ADA"
181a7e14dcfSSatish Balay PetscErrorCode MatDuplicate_ADA(Mat mat,MatDuplicateOption op,Mat *M)
182a7e14dcfSSatish Balay {
183a7e14dcfSSatish Balay   PetscErrorCode    ierr;
184a7e14dcfSSatish Balay   TaoMatADACtx      ctx;
185a7e14dcfSSatish Balay   Mat               A2;
186a7e14dcfSSatish Balay   Vec               D1b=NULL,D2b;
187a7e14dcfSSatish Balay 
188a7e14dcfSSatish Balay   PetscFunctionBegin;
189a7e14dcfSSatish Balay   ierr = MatShellGetContext(mat,(void **)&ctx);CHKERRQ(ierr);
190a7e14dcfSSatish Balay   ierr = MatDuplicate(ctx->A,op,&A2);CHKERRQ(ierr);
191a7e14dcfSSatish Balay   if (ctx->D1){
192a7e14dcfSSatish Balay     ierr = VecDuplicate(ctx->D1,&D1b);CHKERRQ(ierr);
193a7e14dcfSSatish Balay     ierr = VecCopy(ctx->D1,D1b);CHKERRQ(ierr);
194a7e14dcfSSatish Balay   }
195a7e14dcfSSatish Balay   ierr = VecDuplicate(ctx->D2,&D2b);CHKERRQ(ierr);
196a7e14dcfSSatish Balay   ierr = VecCopy(ctx->D2,D2b);CHKERRQ(ierr);
197a7e14dcfSSatish Balay   ierr = MatCreateADA(A2,D1b,D2b,M);CHKERRQ(ierr);
198a7e14dcfSSatish Balay   if (ctx->D1){
199a7e14dcfSSatish Balay     ierr = PetscObjectDereference((PetscObject)D1b);CHKERRQ(ierr);
200a7e14dcfSSatish Balay   }
201a7e14dcfSSatish Balay   ierr = PetscObjectDereference((PetscObject)D2b);CHKERRQ(ierr);
202a7e14dcfSSatish Balay   ierr = PetscObjectDereference((PetscObject)A2);CHKERRQ(ierr);
203a7e14dcfSSatish Balay   PetscFunctionReturn(0);
204a7e14dcfSSatish Balay }
205a7e14dcfSSatish Balay 
206a7e14dcfSSatish Balay #undef __FUNCT__
207a7e14dcfSSatish Balay #define __FUNCT__ "MatEqual_ADA"
208a7e14dcfSSatish Balay PetscErrorCode MatEqual_ADA(Mat A,Mat B,PetscBool *flg)
209a7e14dcfSSatish Balay {
210a7e14dcfSSatish Balay   PetscErrorCode ierr;
211a7e14dcfSSatish Balay   TaoMatADACtx   ctx1,ctx2;
212a7e14dcfSSatish Balay 
213a7e14dcfSSatish Balay   PetscFunctionBegin;
214a7e14dcfSSatish Balay   ierr = MatShellGetContext(A,(void **)&ctx1);CHKERRQ(ierr);
215a7e14dcfSSatish Balay   ierr = MatShellGetContext(B,(void **)&ctx2);CHKERRQ(ierr);
216a7e14dcfSSatish Balay   ierr = VecEqual(ctx1->D2,ctx2->D2,flg);CHKERRQ(ierr);
217a7e14dcfSSatish Balay   if (*flg==PETSC_TRUE){
218a7e14dcfSSatish Balay     ierr = VecEqual(ctx1->D1,ctx2->D1,flg);CHKERRQ(ierr);
219a7e14dcfSSatish Balay   }
220a7e14dcfSSatish Balay   if (*flg==PETSC_TRUE){
221a7e14dcfSSatish Balay     ierr = MatEqual(ctx1->A,ctx2->A,flg);CHKERRQ(ierr);
222a7e14dcfSSatish Balay   }
223a7e14dcfSSatish Balay   PetscFunctionReturn(0);
224a7e14dcfSSatish Balay }
225a7e14dcfSSatish Balay 
226a7e14dcfSSatish Balay #undef __FUNCT__
227a7e14dcfSSatish Balay #define __FUNCT__ "MatScale_ADA"
228a7e14dcfSSatish Balay PetscErrorCode MatScale_ADA(Mat mat, PetscReal a)
229a7e14dcfSSatish Balay {
230a7e14dcfSSatish Balay   PetscErrorCode ierr;
231a7e14dcfSSatish Balay   TaoMatADACtx   ctx;
232a7e14dcfSSatish Balay 
233a7e14dcfSSatish Balay   PetscFunctionBegin;
234a7e14dcfSSatish Balay   ierr = MatShellGetContext(mat,(void **)&ctx);CHKERRQ(ierr);
235a7e14dcfSSatish Balay   ierr = VecScale(ctx->D1,a);CHKERRQ(ierr);
236a7e14dcfSSatish Balay   if (ctx->D2){
237a7e14dcfSSatish Balay     ierr = VecScale(ctx->D2,a);CHKERRQ(ierr);
238a7e14dcfSSatish Balay   }
239a7e14dcfSSatish Balay   PetscFunctionReturn(0);
240a7e14dcfSSatish Balay }
241a7e14dcfSSatish Balay 
242a7e14dcfSSatish Balay #undef __FUNCT__
243a7e14dcfSSatish Balay #define __FUNCT__ "MatTranspose_ADA"
244a7e14dcfSSatish Balay PetscErrorCode MatTranspose_ADA(Mat mat,Mat *B)
245a7e14dcfSSatish Balay {
246a7e14dcfSSatish Balay   PetscErrorCode ierr;
247a7e14dcfSSatish Balay   TaoMatADACtx   ctx;
248a7e14dcfSSatish Balay 
249a7e14dcfSSatish Balay   PetscFunctionBegin;
250a7e14dcfSSatish Balay   if (*B){
251a7e14dcfSSatish Balay     ierr = MatShellGetContext(mat,(void **)&ctx);CHKERRQ(ierr);
252a7e14dcfSSatish Balay     ierr = MatDuplicate(mat,MAT_COPY_VALUES,B);CHKERRQ(ierr);
253a7e14dcfSSatish Balay   }
254a7e14dcfSSatish Balay   PetscFunctionReturn(0);
255a7e14dcfSSatish Balay }
256a7e14dcfSSatish Balay 
257a7e14dcfSSatish Balay #undef __FUNCT__
258a7e14dcfSSatish Balay #define __FUNCT__ "MatADAComputeDiagonal"
259a7e14dcfSSatish Balay PetscErrorCode MatADAComputeDiagonal(Mat mat)
260a7e14dcfSSatish Balay {
261a7e14dcfSSatish Balay   PetscErrorCode ierr;
262a7e14dcfSSatish Balay   PetscInt        i,m,n,low,high;
263a7e14dcfSSatish Balay   PetscReal       *dtemp,*dptr;
264a7e14dcfSSatish Balay   TaoMatADACtx    ctx;
265a7e14dcfSSatish Balay 
266a7e14dcfSSatish Balay   PetscFunctionBegin;
267a7e14dcfSSatish Balay   ierr = MatShellGetContext(mat,(void **)&ctx);CHKERRQ(ierr);
268a7e14dcfSSatish Balay   ierr = MatGetOwnershipRange(mat, &low, &high);CHKERRQ(ierr);
269a7e14dcfSSatish Balay   ierr = MatGetSize(mat,&m,&n);CHKERRQ(ierr);
270a7e14dcfSSatish Balay 
271*0e660641SBarry Smith   ierr = PetscMalloc1(n,&dtemp );CHKERRQ(ierr);
272a7e14dcfSSatish Balay 
273a7e14dcfSSatish Balay   for (i=0; i<n; i++){
274a7e14dcfSSatish Balay     ierr = MatGetColumnVector(ctx->A, ctx->W, i);CHKERRQ(ierr);
275a7e14dcfSSatish Balay     ierr = VecPointwiseMult(ctx->W,ctx->W,ctx->W);CHKERRQ(ierr);
276a7e14dcfSSatish Balay     ierr = VecDotBegin(ctx->D1, ctx->W,dtemp+i);CHKERRQ(ierr);
277a7e14dcfSSatish Balay   }
278a7e14dcfSSatish Balay   for (i=0; i<n; i++){
279a7e14dcfSSatish Balay     ierr = VecDotEnd(ctx->D1, ctx->W,dtemp+i);CHKERRQ(ierr);
280a7e14dcfSSatish Balay   }
281a7e14dcfSSatish Balay 
282a7e14dcfSSatish Balay   ierr = VecGetArray(ctx->ADADiag,&dptr);CHKERRQ(ierr);
283a7e14dcfSSatish Balay   for (i=low; i<high; i++){
284a7e14dcfSSatish Balay     dptr[i-low]= dtemp[i];
285a7e14dcfSSatish Balay   }
286a7e14dcfSSatish Balay   ierr = VecRestoreArray(ctx->ADADiag,&dptr);CHKERRQ(ierr);
287a7e14dcfSSatish Balay   ierr = PetscFree(dtemp);CHKERRQ(ierr);
288a7e14dcfSSatish Balay   PetscFunctionReturn(0);
289a7e14dcfSSatish Balay }
290a7e14dcfSSatish Balay 
291a7e14dcfSSatish Balay #undef __FUNCT__
292a7e14dcfSSatish Balay #define __FUNCT__ "MatGetDiagonal_ADA"
293a7e14dcfSSatish Balay PetscErrorCode MatGetDiagonal_ADA(Mat mat,Vec v)
294a7e14dcfSSatish Balay {
295a7e14dcfSSatish Balay   PetscErrorCode  ierr;
296a7e14dcfSSatish Balay   PetscReal       one=1.0;
297a7e14dcfSSatish Balay   TaoMatADACtx    ctx;
298a7e14dcfSSatish Balay 
299a7e14dcfSSatish Balay   PetscFunctionBegin;
300a7e14dcfSSatish Balay   ierr = MatShellGetContext(mat,(void **)&ctx);CHKERRQ(ierr);
301a7e14dcfSSatish Balay   ierr = MatADAComputeDiagonal(mat);CHKERRQ(ierr);
302a7e14dcfSSatish Balay   ierr=VecCopy(ctx->ADADiag,v);CHKERRQ(ierr);
303a7e14dcfSSatish Balay   if (ctx->D2){
304a7e14dcfSSatish Balay     ierr=VecAXPY(v, one, ctx->D2);CHKERRQ(ierr);
305a7e14dcfSSatish Balay   }
306a7e14dcfSSatish Balay   PetscFunctionReturn(0);
307a7e14dcfSSatish Balay }
308a7e14dcfSSatish Balay 
309a7e14dcfSSatish Balay #undef __FUNCT__
310a7e14dcfSSatish Balay #define __FUNCT__ "MatGetSubMatrices_ADA"
311a7e14dcfSSatish Balay PetscErrorCode MatGetSubMatrices_ADA(Mat A,PetscInt n, IS *irow,IS *icol,MatReuse scall,Mat **B)
312a7e14dcfSSatish Balay {
313a7e14dcfSSatish Balay   PetscErrorCode ierr;
314a7e14dcfSSatish Balay   PetscInt       i;
315a7e14dcfSSatish Balay 
316a7e14dcfSSatish Balay   PetscFunctionBegin;
317a7e14dcfSSatish Balay   if (scall == MAT_INITIAL_MATRIX) {
318*0e660641SBarry Smith     ierr = PetscMalloc1(n+1,B );CHKERRQ(ierr);
319a7e14dcfSSatish Balay   }
320a7e14dcfSSatish Balay   for ( i=0; i<n; i++ ) {
321a7e14dcfSSatish Balay     ierr = MatGetSubMatrix_ADA(A,irow[i],icol[i],scall,&(*B)[i]);CHKERRQ(ierr);
322a7e14dcfSSatish Balay   }
323a7e14dcfSSatish Balay   PetscFunctionReturn(0);
324a7e14dcfSSatish Balay }
325a7e14dcfSSatish Balay 
326a7e14dcfSSatish Balay #undef __FUNCT__
327a7e14dcfSSatish Balay #define __FUNCT__ "MatGetSubMatrix_ADA"
328*0e660641SBarry Smith PetscErrorCode MatGetSubMatrix_ADA(Mat mat,IS isrow,IS iscol,MatReuse cll, Mat *newmat)
329a7e14dcfSSatish Balay {
330a7e14dcfSSatish Balay   PetscErrorCode ierr;
331a7e14dcfSSatish Balay   PetscInt       low,high;
332a7e14dcfSSatish Balay   PetscInt       n,nlocal,i;
333a7e14dcfSSatish Balay   const PetscInt *iptr;
334a7e14dcfSSatish Balay   PetscReal       *dptr,*ddptr,zero=0.0;
335a7e14dcfSSatish Balay   VecType         type_name;
336a7e14dcfSSatish Balay   IS              ISrow;
337a7e14dcfSSatish Balay   Vec             D1,D2;
338a7e14dcfSSatish Balay   Mat             Atemp;
339a7e14dcfSSatish Balay   TaoMatADACtx    ctx;
340a7e14dcfSSatish Balay 
341a7e14dcfSSatish Balay   PetscFunctionBegin;
342a7e14dcfSSatish Balay   ierr = MatShellGetContext(mat,(void **)&ctx);CHKERRQ(ierr);
343a7e14dcfSSatish Balay 
344a7e14dcfSSatish Balay   ierr = MatGetOwnershipRange(ctx->A,&low,&high);CHKERRQ(ierr);
345a7e14dcfSSatish Balay   ierr = ISCreateStride(((PetscObject)mat)->comm,high-low,low,1,&ISrow);CHKERRQ(ierr);
346a7e14dcfSSatish Balay   ierr = MatGetSubMatrix(ctx->A,ISrow,iscol,cll,&Atemp);CHKERRQ(ierr);
347a7e14dcfSSatish Balay   ierr = ISDestroy(&ISrow);CHKERRQ(ierr);
348a7e14dcfSSatish Balay 
349a7e14dcfSSatish Balay   if (ctx->D1){
350a7e14dcfSSatish Balay     ierr=VecDuplicate(ctx->D1,&D1);CHKERRQ(ierr);
351a7e14dcfSSatish Balay     ierr=VecCopy(ctx->D1,D1);CHKERRQ(ierr);
352a7e14dcfSSatish Balay   } else {
3536c23d075SBarry Smith     D1 = NULL;
354a7e14dcfSSatish Balay   }
355a7e14dcfSSatish Balay 
356a7e14dcfSSatish Balay   if (ctx->D2){
357a7e14dcfSSatish Balay     ierr=ISGetSize(isrow,&n);CHKERRQ(ierr);
358a7e14dcfSSatish Balay     ierr=ISGetLocalSize(isrow,&nlocal);CHKERRQ(ierr);
359a7e14dcfSSatish Balay     ierr=VecCreate(((PetscObject)(ctx->D2))->comm,&D2);CHKERRQ(ierr);
360a7e14dcfSSatish Balay     ierr=VecGetType(ctx->D2,&type_name);CHKERRQ(ierr);
361a7e14dcfSSatish Balay     ierr=VecSetSizes(D2,nlocal,n);CHKERRQ(ierr);
362a7e14dcfSSatish Balay     ierr=VecSetType(D2,type_name);CHKERRQ(ierr);
363a7e14dcfSSatish Balay     ierr=VecSet(D2, zero);CHKERRQ(ierr);
364a7e14dcfSSatish Balay     ierr=VecGetArray(ctx->D2, &dptr);CHKERRQ(ierr);
365a7e14dcfSSatish Balay     ierr=VecGetArray(D2, &ddptr);CHKERRQ(ierr);
366a7e14dcfSSatish Balay     ierr=ISGetIndices(isrow,&iptr);CHKERRQ(ierr);
367a7e14dcfSSatish Balay     for (i=0;i<nlocal;i++){
368a7e14dcfSSatish Balay       ddptr[i] = dptr[iptr[i]-low];
369a7e14dcfSSatish Balay     }
370a7e14dcfSSatish Balay     ierr=ISRestoreIndices(isrow,&iptr);CHKERRQ(ierr);
371a7e14dcfSSatish Balay     ierr=VecRestoreArray(D2, &ddptr);CHKERRQ(ierr);
372a7e14dcfSSatish Balay     ierr=VecRestoreArray(ctx->D2, &dptr);CHKERRQ(ierr);
373a7e14dcfSSatish Balay   } else {
3746c23d075SBarry Smith     D2 = NULL;
375a7e14dcfSSatish Balay   }
376a7e14dcfSSatish Balay 
377a7e14dcfSSatish Balay   ierr = MatCreateADA(Atemp,D1,D2,newmat);CHKERRQ(ierr);
378a7e14dcfSSatish Balay   ierr = MatShellGetContext(*newmat,(void **)&ctx);CHKERRQ(ierr);
379a7e14dcfSSatish Balay   ierr = PetscObjectDereference((PetscObject)Atemp);CHKERRQ(ierr);
380a7e14dcfSSatish Balay   if (ctx->D1){
381a7e14dcfSSatish Balay     ierr = PetscObjectDereference((PetscObject)D1);CHKERRQ(ierr);
382a7e14dcfSSatish Balay   }
383a7e14dcfSSatish Balay   if (ctx->D2){
384a7e14dcfSSatish Balay     ierr = PetscObjectDereference((PetscObject)D2);CHKERRQ(ierr);
385a7e14dcfSSatish Balay   }
386a7e14dcfSSatish Balay   PetscFunctionReturn(0);
387a7e14dcfSSatish Balay }
388a7e14dcfSSatish Balay 
389a7e14dcfSSatish Balay #undef __FUNCT__
390a7e14dcfSSatish Balay #define __FUNCT__ "MatGetRowADA"
391a7e14dcfSSatish Balay PetscErrorCode MatGetRowADA(Mat mat,PetscInt row,PetscInt *ncols,PetscInt **cols,PetscReal **vals)
392a7e14dcfSSatish Balay {
393a7e14dcfSSatish Balay   PetscErrorCode ierr;
394a7e14dcfSSatish Balay   PetscInt       m,n;
395a7e14dcfSSatish Balay 
396a7e14dcfSSatish Balay   PetscFunctionBegin;
397a7e14dcfSSatish Balay   ierr = MatGetSize(mat,&m,&n);CHKERRQ(ierr);
398a7e14dcfSSatish Balay 
399a7e14dcfSSatish Balay   if (*ncols>0){
400*0e660641SBarry Smith     ierr = PetscMalloc1(*ncols,cols );CHKERRQ(ierr);
401*0e660641SBarry Smith     ierr = PetscMalloc1(*ncols,vals );CHKERRQ(ierr);
402a7e14dcfSSatish Balay   } else {
4036c23d075SBarry Smith     *cols=NULL;
4046c23d075SBarry Smith     *vals=NULL;
405a7e14dcfSSatish Balay   }
406a7e14dcfSSatish Balay   PetscFunctionReturn(0);
407a7e14dcfSSatish Balay }
408a7e14dcfSSatish Balay 
409a7e14dcfSSatish Balay #undef __FUNCT__
410a7e14dcfSSatish Balay #define __FUNCT__ "MatRestoreRowADA"
411a7e14dcfSSatish Balay PetscErrorCode MatRestoreRowADA(Mat mat,PetscInt row,PetscInt *ncols,PetscInt **cols,PetscReal **vals)
412a7e14dcfSSatish Balay {
413a7e14dcfSSatish Balay   PetscErrorCode ierr;
414*0e660641SBarry Smith 
415a7e14dcfSSatish Balay   PetscFunctionBegin;
416a7e14dcfSSatish Balay   if (*ncols>0){
417a7e14dcfSSatish Balay     ierr = PetscFree(*cols); CHKERRQ(ierr);
418a7e14dcfSSatish Balay     ierr = PetscFree(*vals); CHKERRQ(ierr);
419a7e14dcfSSatish Balay   }
4206c23d075SBarry Smith   *cols=NULL;
4216c23d075SBarry Smith   *vals=NULL;
422a7e14dcfSSatish Balay   PetscFunctionReturn(0);
423a7e14dcfSSatish Balay }
424a7e14dcfSSatish Balay 
425a7e14dcfSSatish Balay #undef __FUNCT__
426a7e14dcfSSatish Balay #define __FUNCT__ "MatGetColumnVectorADA"
427a7e14dcfSSatish Balay PetscErrorCode MatGetColumnVector_ADA(Mat mat,Vec Y, PetscInt col)
428a7e14dcfSSatish Balay {
429a7e14dcfSSatish Balay   PetscErrorCode ierr;
430a7e14dcfSSatish Balay   PetscInt       low,high;
431a7e14dcfSSatish Balay   PetscReal      zero=0.0,one=1.0;
432a7e14dcfSSatish Balay 
433a7e14dcfSSatish Balay   PetscFunctionBegin;
434a7e14dcfSSatish Balay   ierr=VecSet(Y, zero);CHKERRQ(ierr);
435a7e14dcfSSatish Balay   ierr=VecGetOwnershipRange(Y,&low,&high);CHKERRQ(ierr);
436a7e14dcfSSatish Balay   if (col>=low && col<high){
437a7e14dcfSSatish Balay     ierr=VecSetValue(Y,col,one,INSERT_VALUES);CHKERRQ(ierr);
438a7e14dcfSSatish Balay   }
439a7e14dcfSSatish Balay   ierr=VecAssemblyBegin(Y);CHKERRQ(ierr);
440a7e14dcfSSatish Balay   ierr=VecAssemblyEnd(Y);CHKERRQ(ierr);
441a7e14dcfSSatish Balay   ierr=MatMult_ADA(mat,Y,Y);CHKERRQ(ierr);
442a7e14dcfSSatish Balay   PetscFunctionReturn(0);
443a7e14dcfSSatish Balay }
444a7e14dcfSSatish Balay 
445a7e14dcfSSatish Balay PetscErrorCode MatConvert_ADA(Mat mat,MatType newtype,Mat *NewMat)
446a7e14dcfSSatish Balay {
447a7e14dcfSSatish Balay   PetscErrorCode ierr;
448a7e14dcfSSatish Balay   PetscMPIInt    size;
449a7e14dcfSSatish Balay   PetscBool      sametype, issame, isdense, isseqdense;
450a7e14dcfSSatish Balay   TaoMatADACtx   ctx;
451a7e14dcfSSatish Balay 
452a7e14dcfSSatish Balay   PetscFunctionBegin;
453a7e14dcfSSatish Balay   ierr = MatShellGetContext(mat,(void **)&ctx);CHKERRQ(ierr);
45487f595a5SBarry Smith   ierr = MPI_Comm_size(((PetscObject)mat)->comm,&size);CHKERRQ(ierr);
455a7e14dcfSSatish Balay 
456a7e14dcfSSatish Balay   ierr = PetscObjectTypeCompare((PetscObject)mat,newtype,&sametype);CHKERRQ(ierr);
457a7e14dcfSSatish Balay   ierr = PetscObjectTypeCompare((PetscObject)mat,MATSAME,&issame);CHKERRQ(ierr);
458a7e14dcfSSatish Balay   ierr = PetscObjectTypeCompare((PetscObject)mat,MATMPIDENSE,&isdense);CHKERRQ(ierr);
459a7e14dcfSSatish Balay   ierr = PetscObjectTypeCompare((PetscObject)mat,MATSEQDENSE,&isseqdense);CHKERRQ(ierr);
460a7e14dcfSSatish Balay 
461a7e14dcfSSatish Balay   if (sametype || issame) {
462a7e14dcfSSatish Balay     ierr=MatDuplicate(mat,MAT_COPY_VALUES,NewMat);CHKERRQ(ierr);
463a7e14dcfSSatish Balay   } else if (isdense) {
464a7e14dcfSSatish Balay     PetscInt  i,j,low,high,m,n,M,N;
465a7e14dcfSSatish Balay     PetscReal *dptr;
466a7e14dcfSSatish Balay     Vec       X;
467a7e14dcfSSatish Balay 
468a7e14dcfSSatish Balay     ierr = VecDuplicate(ctx->D2,&X);CHKERRQ(ierr);
469a7e14dcfSSatish Balay     ierr=MatGetSize(mat,&M,&N);CHKERRQ(ierr);
470a7e14dcfSSatish Balay     ierr=MatGetLocalSize(mat,&m,&n);CHKERRQ(ierr);
47187f595a5SBarry Smith     ierr = MatCreateDense(((PetscObject)mat)->comm,m,m,N,N,NULL,NewMat);CHKERRQ(ierr);
472a7e14dcfSSatish Balay     ierr = MatGetOwnershipRange(*NewMat,&low,&high);CHKERRQ(ierr);
473a7e14dcfSSatish Balay     for (i=0;i<M;i++){
474a7e14dcfSSatish Balay       ierr = MatGetColumnVector_ADA(mat,X,i);CHKERRQ(ierr);
475a7e14dcfSSatish Balay       ierr = VecGetArray(X,&dptr);CHKERRQ(ierr);
476a7e14dcfSSatish Balay       for (j=0; j<high-low; j++){
477a7e14dcfSSatish Balay         ierr = MatSetValue(*NewMat,low+j,i,dptr[j],INSERT_VALUES);CHKERRQ(ierr);
478a7e14dcfSSatish Balay       }
479a7e14dcfSSatish Balay       ierr=VecRestoreArray(X,&dptr);CHKERRQ(ierr);
480a7e14dcfSSatish Balay     }
481a7e14dcfSSatish Balay     ierr=MatAssemblyBegin(*NewMat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
482a7e14dcfSSatish Balay     ierr=MatAssemblyEnd(*NewMat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
483a7e14dcfSSatish Balay     ierr = VecDestroy(&X);CHKERRQ(ierr);
484a7e14dcfSSatish Balay   } else if (isseqdense && size==1){
485a7e14dcfSSatish Balay     PetscInt   i,j,low,high,m,n,M,N;
486a7e14dcfSSatish Balay     PetscReal *dptr;
487a7e14dcfSSatish Balay     Vec       X;
488a7e14dcfSSatish Balay 
489a7e14dcfSSatish Balay     ierr = VecDuplicate(ctx->D2,&X);CHKERRQ(ierr);
490a7e14dcfSSatish Balay     ierr = MatGetSize(mat,&M,&N);CHKERRQ(ierr);
491a7e14dcfSSatish Balay     ierr = MatGetLocalSize(mat,&m,&n);CHKERRQ(ierr);
49287f595a5SBarry Smith     ierr = MatCreateSeqDense(((PetscObject)mat)->comm,N,N,NULL,NewMat);CHKERRQ(ierr);
493a7e14dcfSSatish Balay     ierr = MatGetOwnershipRange(*NewMat,&low,&high);CHKERRQ(ierr);
494a7e14dcfSSatish Balay     for (i=0;i<M;i++){
495a7e14dcfSSatish Balay       ierr = MatGetColumnVector_ADA(mat,X,i);CHKERRQ(ierr);
496a7e14dcfSSatish Balay       ierr = VecGetArray(X,&dptr);CHKERRQ(ierr);
497a7e14dcfSSatish Balay       for (j=0; j<high-low; j++){
498a7e14dcfSSatish Balay         ierr = MatSetValue(*NewMat,low+j,i,dptr[j],INSERT_VALUES);CHKERRQ(ierr);
499a7e14dcfSSatish Balay       }
500a7e14dcfSSatish Balay       ierr=VecRestoreArray(X,&dptr);CHKERRQ(ierr);
501a7e14dcfSSatish Balay     }
502a7e14dcfSSatish Balay     ierr=MatAssemblyBegin(*NewMat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
503a7e14dcfSSatish Balay     ierr=MatAssemblyEnd(*NewMat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
504a7e14dcfSSatish Balay     ierr=VecDestroy(&X);CHKERRQ(ierr);
50587f595a5SBarry Smith   } else SETERRQ(PETSC_COMM_SELF,1,"No support to convert objects to that type");
506a7e14dcfSSatish Balay   PetscFunctionReturn(0);
507a7e14dcfSSatish Balay }
508a7e14dcfSSatish Balay 
509a7e14dcfSSatish Balay #undef __FUNCT__
510a7e14dcfSSatish Balay #define __FUNCT__ "MatNorm_ADA"
511a7e14dcfSSatish Balay PetscErrorCode MatNorm_ADA(Mat mat,NormType type,PetscReal *norm)
512a7e14dcfSSatish Balay {
513a7e14dcfSSatish Balay   PetscErrorCode ierr;
514a7e14dcfSSatish Balay   TaoMatADACtx   ctx;
515a7e14dcfSSatish Balay 
516a7e14dcfSSatish Balay   PetscFunctionBegin;
517a7e14dcfSSatish Balay   ierr = MatShellGetContext(mat,(void **)&ctx);CHKERRQ(ierr);
518a7e14dcfSSatish Balay   if (type == NORM_FROBENIUS) {
519a7e14dcfSSatish Balay     *norm = 1.0;
520a7e14dcfSSatish Balay   } else if (type == NORM_1 || type == NORM_INFINITY) {
521a7e14dcfSSatish Balay     *norm = 1.0;
52287f595a5SBarry Smith   } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"No two norm");
523a7e14dcfSSatish Balay   PetscFunctionReturn(0);
524a7e14dcfSSatish Balay }
525