#include <../src/tao/matrix/adamat.h> /*I "mat.h" I*/ #undef __FUNCT__ #define __FUNCT__ "MatCreateADA" /*@C MatCreateADA - Creates a matrix M=A^T D1 A + D2 where D1, D2 are diagonal Collective on matrix Input Parameters: + mat - matrix of arbitrary type . d1 - A vector with diagonal elements of D1 - d2 - A vector Output Parameters: . J - New matrix whose operations are defined in terms of mat, D1, and D2. Notes: The user provides the input data and is responsible for destroying this data after matrix J has been destroyed. The operation MatMult(A,D2,D1) must be well defined. Before calling the operation MatGetDiagonal(), the function MatADAComputeDiagonal() must be called. The matrices A and D1 must be the same during calls to MatADAComputeDiagonal() and MatGetDiagonal(). Level: developer .seealso: MatCreate() @*/ PetscErrorCode MatCreateADA(Mat mat,Vec d1, Vec d2, Mat *J) { MPI_Comm comm=((PetscObject)mat)->comm; TaoMatADACtx ctx; PetscErrorCode ierr; PetscInt nloc,n; PetscFunctionBegin; ierr = PetscNew(&ctx);CHKERRQ(ierr); ctx->A=mat; ctx->D1=d1; ctx->D2=d2; if (d1){ ierr = VecDuplicate(d1,&ctx->W);CHKERRQ(ierr); ierr = PetscObjectReference((PetscObject)d1);CHKERRQ(ierr); } else { ctx->W = NULL; } if (d2){ ierr = VecDuplicate(d2,&ctx->W2);CHKERRQ(ierr); ierr = VecDuplicate(d2,&ctx->ADADiag);CHKERRQ(ierr); ierr = PetscObjectReference((PetscObject)d2);CHKERRQ(ierr); } else { ctx->W2 = NULL; ctx->ADADiag = NULL; } ctx->GotDiag=0; ierr = PetscObjectReference((PetscObject)mat);CHKERRQ(ierr); ierr=VecGetLocalSize(d2,&nloc);CHKERRQ(ierr); ierr=VecGetSize(d2,&n);CHKERRQ(ierr); ierr = MatCreateShell(comm,nloc,nloc,n,n,ctx,J);CHKERRQ(ierr); ierr = MatShellSetOperation(*J,MATOP_MULT,(void(*)(void))MatMult_ADA);CHKERRQ(ierr); ierr = MatShellSetOperation(*J,MATOP_DESTROY,(void(*)(void))MatDestroy_ADA);CHKERRQ(ierr); ierr = MatShellSetOperation(*J,MATOP_VIEW,(void(*)(void))MatView_ADA);CHKERRQ(ierr); ierr = MatShellSetOperation(*J,MATOP_MULT_TRANSPOSE,(void(*)(void))MatMultTranspose_ADA);CHKERRQ(ierr); ierr = MatShellSetOperation(*J,MATOP_DIAGONAL_SET,(void(*)(void))MatDiagonalSet_ADA);CHKERRQ(ierr); ierr = MatShellSetOperation(*J,MATOP_SHIFT,(void(*)(void))MatShift_ADA);CHKERRQ(ierr); ierr = MatShellSetOperation(*J,MATOP_EQUAL,(void(*)(void))MatEqual_ADA);CHKERRQ(ierr); ierr = MatShellSetOperation(*J,MATOP_SCALE,(void(*)(void))MatScale_ADA);CHKERRQ(ierr); ierr = MatShellSetOperation(*J,MATOP_TRANSPOSE,(void(*)(void))MatTranspose_ADA);CHKERRQ(ierr); ierr = MatShellSetOperation(*J,MATOP_GET_DIAGONAL,(void(*)(void))MatGetDiagonal_ADA);CHKERRQ(ierr); ierr = MatShellSetOperation(*J,MATOP_GET_SUBMATRICES,(void(*)(void))MatGetSubMatrices_ADA);CHKERRQ(ierr); ierr = MatShellSetOperation(*J,MATOP_NORM,(void(*)(void))MatNorm_ADA);CHKERRQ(ierr); ierr = MatShellSetOperation(*J,MATOP_DUPLICATE,(void(*)(void))MatDuplicate_ADA);CHKERRQ(ierr); ierr = MatShellSetOperation(*J,MATOP_GET_SUBMATRIX,(void(*)(void))MatGetSubMatrix_ADA);CHKERRQ(ierr); ierr = PetscLogObjectParent((PetscObject)(*J),(PetscObject)ctx->W);CHKERRQ(ierr); ierr = PetscLogObjectParent((PetscObject)mat,(PetscObject)(*J));CHKERRQ(ierr); ierr = MatSetOption(*J,MAT_SYMMETRIC,PETSC_TRUE);CHKERRQ(ierr); PetscFunctionReturn(0); } #undef __FUNCT__ #define __FUNCT__ "MatMult_ADA" PetscErrorCode MatMult_ADA(Mat mat,Vec a,Vec y) { TaoMatADACtx ctx; PetscReal one = 1.0; PetscErrorCode ierr; PetscFunctionBegin; ierr = MatShellGetContext(mat,(void **)&ctx);CHKERRQ(ierr); ierr = MatMult(ctx->A,a,ctx->W);CHKERRQ(ierr); if (ctx->D1){ ierr = VecPointwiseMult(ctx->W,ctx->D1,ctx->W);CHKERRQ(ierr); } ierr = MatMultTranspose(ctx->A,ctx->W,y);CHKERRQ(ierr); if (ctx->D2){ ierr = VecPointwiseMult(ctx->W2, ctx->D2, a);CHKERRQ(ierr); ierr = VecAXPY(y, one, ctx->W2);CHKERRQ(ierr); } PetscFunctionReturn(0); } #undef __FUNCT__ #define __FUNCT__ "MatMultTranspose_ADA" PetscErrorCode MatMultTranspose_ADA(Mat mat,Vec a,Vec y) { PetscErrorCode ierr; PetscFunctionBegin; ierr = MatMult_ADA(mat,a,y);CHKERRQ(ierr); PetscFunctionReturn(0); } #undef __FUNCT__ #define __FUNCT__ "MatDiagonalSet_ADA" PetscErrorCode MatDiagonalSet_ADA(Vec D, Mat M) { TaoMatADACtx ctx; PetscReal zero=0.0,one = 1.0; PetscErrorCode ierr; PetscFunctionBegin; ierr = MatShellGetContext(M,(void **)&ctx);CHKERRQ(ierr); if (!ctx->D2){ ierr = VecDuplicate(D,&ctx->D2);CHKERRQ(ierr); ierr = VecSet(ctx->D2, zero);CHKERRQ(ierr); } ierr = VecAXPY(ctx->D2, one, D);CHKERRQ(ierr); PetscFunctionReturn(0); } #undef __FUNCT__ #define __FUNCT__ "MatDestroy_ADA" PetscErrorCode MatDestroy_ADA(Mat mat) { PetscErrorCode ierr; TaoMatADACtx ctx; PetscFunctionBegin; ierr=MatShellGetContext(mat,(void **)&ctx);CHKERRQ(ierr); ierr=VecDestroy(&ctx->W);CHKERRQ(ierr); ierr=VecDestroy(&ctx->W2);CHKERRQ(ierr); ierr=VecDestroy(&ctx->ADADiag);CHKERRQ(ierr); ierr=MatDestroy(&ctx->A);CHKERRQ(ierr); ierr=VecDestroy(&ctx->D1);CHKERRQ(ierr); ierr=VecDestroy(&ctx->D2);CHKERRQ(ierr); ierr = PetscFree(ctx);CHKERRQ(ierr); PetscFunctionReturn(0); } #undef __FUNCT__ #define __FUNCT__ "MatView_ADA" PetscErrorCode MatView_ADA(Mat mat,PetscViewer viewer) { PetscFunctionBegin; PetscFunctionReturn(0); } #undef __FUNCT__ #define __FUNCT__ "MatShift_ADA" PetscErrorCode MatShift_ADA(Mat Y, PetscReal a) { PetscErrorCode ierr; TaoMatADACtx ctx; PetscFunctionBegin; ierr = MatShellGetContext(Y,(void **)&ctx);CHKERRQ(ierr); ierr = VecShift(ctx->D2,a);CHKERRQ(ierr); PetscFunctionReturn(0); } #undef __FUNCT__ #define __FUNCT__ "MatDuplicate_ADA" PetscErrorCode MatDuplicate_ADA(Mat mat,MatDuplicateOption op,Mat *M) { PetscErrorCode ierr; TaoMatADACtx ctx; Mat A2; Vec D1b=NULL,D2b; PetscFunctionBegin; ierr = MatShellGetContext(mat,(void **)&ctx);CHKERRQ(ierr); ierr = MatDuplicate(ctx->A,op,&A2);CHKERRQ(ierr); if (ctx->D1){ ierr = VecDuplicate(ctx->D1,&D1b);CHKERRQ(ierr); ierr = VecCopy(ctx->D1,D1b);CHKERRQ(ierr); } ierr = VecDuplicate(ctx->D2,&D2b);CHKERRQ(ierr); ierr = VecCopy(ctx->D2,D2b);CHKERRQ(ierr); ierr = MatCreateADA(A2,D1b,D2b,M);CHKERRQ(ierr); if (ctx->D1){ ierr = PetscObjectDereference((PetscObject)D1b);CHKERRQ(ierr); } ierr = PetscObjectDereference((PetscObject)D2b);CHKERRQ(ierr); ierr = PetscObjectDereference((PetscObject)A2);CHKERRQ(ierr); PetscFunctionReturn(0); } #undef __FUNCT__ #define __FUNCT__ "MatEqual_ADA" PetscErrorCode MatEqual_ADA(Mat A,Mat B,PetscBool *flg) { PetscErrorCode ierr; TaoMatADACtx ctx1,ctx2; PetscFunctionBegin; ierr = MatShellGetContext(A,(void **)&ctx1);CHKERRQ(ierr); ierr = MatShellGetContext(B,(void **)&ctx2);CHKERRQ(ierr); ierr = VecEqual(ctx1->D2,ctx2->D2,flg);CHKERRQ(ierr); if (*flg==PETSC_TRUE){ ierr = VecEqual(ctx1->D1,ctx2->D1,flg);CHKERRQ(ierr); } if (*flg==PETSC_TRUE){ ierr = MatEqual(ctx1->A,ctx2->A,flg);CHKERRQ(ierr); } PetscFunctionReturn(0); } #undef __FUNCT__ #define __FUNCT__ "MatScale_ADA" PetscErrorCode MatScale_ADA(Mat mat, PetscReal a) { PetscErrorCode ierr; TaoMatADACtx ctx; PetscFunctionBegin; ierr = MatShellGetContext(mat,(void **)&ctx);CHKERRQ(ierr); ierr = VecScale(ctx->D1,a);CHKERRQ(ierr); if (ctx->D2){ ierr = VecScale(ctx->D2,a);CHKERRQ(ierr); } PetscFunctionReturn(0); } #undef __FUNCT__ #define __FUNCT__ "MatTranspose_ADA" PetscErrorCode MatTranspose_ADA(Mat mat,Mat *B) { PetscErrorCode ierr; TaoMatADACtx ctx; PetscFunctionBegin; if (*B){ ierr = MatShellGetContext(mat,(void **)&ctx);CHKERRQ(ierr); ierr = MatDuplicate(mat,MAT_COPY_VALUES,B);CHKERRQ(ierr); } PetscFunctionReturn(0); } #undef __FUNCT__ #define __FUNCT__ "MatADAComputeDiagonal" PetscErrorCode MatADAComputeDiagonal(Mat mat) { PetscErrorCode ierr; PetscInt i,m,n,low,high; PetscScalar *dtemp,*dptr; TaoMatADACtx ctx; PetscFunctionBegin; ierr = MatShellGetContext(mat,(void **)&ctx);CHKERRQ(ierr); ierr = MatGetOwnershipRange(mat, &low, &high);CHKERRQ(ierr); ierr = MatGetSize(mat,&m,&n);CHKERRQ(ierr); ierr = PetscMalloc1(n,&dtemp );CHKERRQ(ierr); for (i=0; iA, ctx->W, i);CHKERRQ(ierr); ierr = VecPointwiseMult(ctx->W,ctx->W,ctx->W);CHKERRQ(ierr); ierr = VecDotBegin(ctx->D1, ctx->W,dtemp+i);CHKERRQ(ierr); } for (i=0; iD1, ctx->W,dtemp+i);CHKERRQ(ierr); } ierr = VecGetArray(ctx->ADADiag,&dptr);CHKERRQ(ierr); for (i=low; iADADiag,&dptr);CHKERRQ(ierr); ierr = PetscFree(dtemp);CHKERRQ(ierr); PetscFunctionReturn(0); } #undef __FUNCT__ #define __FUNCT__ "MatGetDiagonal_ADA" PetscErrorCode MatGetDiagonal_ADA(Mat mat,Vec v) { PetscErrorCode ierr; PetscReal one=1.0; TaoMatADACtx ctx; PetscFunctionBegin; ierr = MatShellGetContext(mat,(void **)&ctx);CHKERRQ(ierr); ierr = MatADAComputeDiagonal(mat);CHKERRQ(ierr); ierr=VecCopy(ctx->ADADiag,v);CHKERRQ(ierr); if (ctx->D2){ ierr=VecAXPY(v, one, ctx->D2);CHKERRQ(ierr); } PetscFunctionReturn(0); } #undef __FUNCT__ #define __FUNCT__ "MatGetSubMatrices_ADA" PetscErrorCode MatGetSubMatrices_ADA(Mat A,PetscInt n, IS *irow,IS *icol,MatReuse scall,Mat **B) { PetscErrorCode ierr; PetscInt i; PetscFunctionBegin; if (scall == MAT_INITIAL_MATRIX) { ierr = PetscMalloc1(n+1,B );CHKERRQ(ierr); } for ( i=0; iA,&low,&high);CHKERRQ(ierr); ierr = ISCreateStride(((PetscObject)mat)->comm,high-low,low,1,&ISrow);CHKERRQ(ierr); ierr = MatGetSubMatrix(ctx->A,ISrow,iscol,cll,&Atemp);CHKERRQ(ierr); ierr = ISDestroy(&ISrow);CHKERRQ(ierr); if (ctx->D1){ ierr=VecDuplicate(ctx->D1,&D1);CHKERRQ(ierr); ierr=VecCopy(ctx->D1,D1);CHKERRQ(ierr); } else { D1 = NULL; } if (ctx->D2){ Vec D2sub; ierr=VecGetSubVector(ctx->D2,isrow,&D2sub);CHKERRQ(ierr); ierr=VecDuplicate(D2sub,&D2);CHKERRQ(ierr); ierr=VecCopy(D2sub,D2);CHKERRQ(ierr); ierr=VecRestoreSubVector(ctx->D2,isrow,&D2sub);CHKERRQ(ierr); } else { D2 = NULL; } ierr = MatCreateADA(Atemp,D1,D2,newmat);CHKERRQ(ierr); ierr = MatShellGetContext(*newmat,(void **)&ctx);CHKERRQ(ierr); ierr = PetscObjectDereference((PetscObject)Atemp);CHKERRQ(ierr); if (ctx->D1){ ierr = PetscObjectDereference((PetscObject)D1);CHKERRQ(ierr); } if (ctx->D2){ ierr = PetscObjectDereference((PetscObject)D2);CHKERRQ(ierr); } PetscFunctionReturn(0); } #undef __FUNCT__ #define __FUNCT__ "MatGetRowADA" PetscErrorCode MatGetRowADA(Mat mat,PetscInt row,PetscInt *ncols,PetscInt **cols,PetscReal **vals) { PetscErrorCode ierr; PetscInt m,n; PetscFunctionBegin; ierr = MatGetSize(mat,&m,&n);CHKERRQ(ierr); if (*ncols>0){ ierr = PetscMalloc1(*ncols,cols );CHKERRQ(ierr); ierr = PetscMalloc1(*ncols,vals );CHKERRQ(ierr); } else { *cols=NULL; *vals=NULL; } PetscFunctionReturn(0); } #undef __FUNCT__ #define __FUNCT__ "MatRestoreRowADA" PetscErrorCode MatRestoreRowADA(Mat mat,PetscInt row,PetscInt *ncols,PetscInt **cols,PetscReal **vals) { PetscErrorCode ierr; PetscFunctionBegin; if (*ncols>0){ ierr = PetscFree(*cols); CHKERRQ(ierr); ierr = PetscFree(*vals); CHKERRQ(ierr); } *cols=NULL; *vals=NULL; PetscFunctionReturn(0); } #undef __FUNCT__ #define __FUNCT__ "MatGetColumnVectorADA" PetscErrorCode MatGetColumnVector_ADA(Mat mat,Vec Y, PetscInt col) { PetscErrorCode ierr; PetscInt low,high; PetscScalar zero=0.0,one=1.0; PetscFunctionBegin; ierr=VecSet(Y, zero);CHKERRQ(ierr); ierr=VecGetOwnershipRange(Y,&low,&high);CHKERRQ(ierr); if (col>=low && colcomm,&size);CHKERRQ(ierr); ierr = PetscObjectTypeCompare((PetscObject)mat,newtype,&sametype);CHKERRQ(ierr); ierr = PetscObjectTypeCompare((PetscObject)mat,MATSAME,&issame);CHKERRQ(ierr); ierr = PetscObjectTypeCompare((PetscObject)mat,MATMPIDENSE,&isdense);CHKERRQ(ierr); ierr = PetscObjectTypeCompare((PetscObject)mat,MATSEQDENSE,&isseqdense);CHKERRQ(ierr); if (sametype || issame) { ierr=MatDuplicate(mat,MAT_COPY_VALUES,NewMat);CHKERRQ(ierr); } else if (isdense) { PetscInt i,j,low,high,m,n,M,N; const PetscScalar *dptr; Vec X; ierr = VecDuplicate(ctx->D2,&X);CHKERRQ(ierr); ierr=MatGetSize(mat,&M,&N);CHKERRQ(ierr); ierr=MatGetLocalSize(mat,&m,&n);CHKERRQ(ierr); ierr = MatCreateDense(((PetscObject)mat)->comm,m,m,N,N,NULL,NewMat);CHKERRQ(ierr); ierr = MatGetOwnershipRange(*NewMat,&low,&high);CHKERRQ(ierr); for (i=0;iD2,&X);CHKERRQ(ierr); ierr = MatGetSize(mat,&M,&N);CHKERRQ(ierr); ierr = MatGetLocalSize(mat,&m,&n);CHKERRQ(ierr); ierr = MatCreateSeqDense(((PetscObject)mat)->comm,N,N,NULL,NewMat);CHKERRQ(ierr); ierr = MatGetOwnershipRange(*NewMat,&low,&high);CHKERRQ(ierr); for (i=0;i