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