1a1f56445SPierre Jolivet #include <../src/mat/impls/shell/shell.h> /*I "petscmat.h" I*/ 285e3dda7SBarry Smith 366976f2fSJacob Faibussowitsch static PetscErrorCode MatMult_Transpose(Mat N, Vec x, Vec y) 4d71ae5a4SJacob Faibussowitsch { 5bf477422SJose E. Roman Mat A; 685e3dda7SBarry Smith 785e3dda7SBarry Smith PetscFunctionBegin; 8bf477422SJose E. Roman PetscCall(MatShellGetContext(N, &A)); 9bf477422SJose E. Roman PetscCall(MatMultTranspose(A, x, y)); 103ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1185e3dda7SBarry Smith } 1285e3dda7SBarry Smith 1366976f2fSJacob Faibussowitsch static PetscErrorCode MatMultTranspose_Transpose(Mat N, Vec x, Vec y) 14d71ae5a4SJacob Faibussowitsch { 15bf477422SJose E. Roman Mat A; 1647a9afc9SBarry Smith 1747a9afc9SBarry Smith PetscFunctionBegin; 18bf477422SJose E. Roman PetscCall(MatShellGetContext(N, &A)); 19bf477422SJose E. Roman PetscCall(MatMult(A, x, y)); 203ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2147a9afc9SBarry Smith } 2247a9afc9SBarry Smith 23f1769f9dSJose E. Roman static PetscErrorCode MatSolve_Transpose_LU(Mat N, Vec b, Vec x) 24f1769f9dSJose E. Roman { 25f1769f9dSJose E. Roman Mat A; 26f1769f9dSJose E. Roman 27f1769f9dSJose E. Roman PetscFunctionBegin; 28f1769f9dSJose E. Roman PetscCall(MatShellGetContext(N, &A)); 29f1769f9dSJose E. Roman PetscCall(MatSolveTranspose(A, b, x)); 30f1769f9dSJose E. Roman PetscFunctionReturn(PETSC_SUCCESS); 31f1769f9dSJose E. Roman } 32f1769f9dSJose E. Roman 33f1769f9dSJose E. Roman static PetscErrorCode MatSolveAdd_Transpose_LU(Mat N, Vec b, Vec y, Vec x) 34f1769f9dSJose E. Roman { 35f1769f9dSJose E. Roman Mat A; 36f1769f9dSJose E. Roman 37f1769f9dSJose E. Roman PetscFunctionBegin; 38f1769f9dSJose E. Roman PetscCall(MatShellGetContext(N, &A)); 39f1769f9dSJose E. Roman PetscCall(MatSolveTransposeAdd(A, b, y, x)); 40f1769f9dSJose E. Roman PetscFunctionReturn(PETSC_SUCCESS); 41f1769f9dSJose E. Roman } 42f1769f9dSJose E. Roman 43f1769f9dSJose E. Roman static PetscErrorCode MatSolveTranspose_Transpose_LU(Mat N, Vec b, Vec x) 44f1769f9dSJose E. Roman { 45f1769f9dSJose E. Roman Mat A; 46f1769f9dSJose E. Roman 47f1769f9dSJose E. Roman PetscFunctionBegin; 48f1769f9dSJose E. Roman PetscCall(MatShellGetContext(N, &A)); 49f1769f9dSJose E. Roman PetscCall(MatSolve(A, b, x)); 50f1769f9dSJose E. Roman PetscFunctionReturn(PETSC_SUCCESS); 51f1769f9dSJose E. Roman } 52f1769f9dSJose E. Roman 53f1769f9dSJose E. Roman static PetscErrorCode MatSolveTransposeAdd_Transpose_LU(Mat N, Vec b, Vec y, Vec x) 54f1769f9dSJose E. Roman { 55f1769f9dSJose E. Roman Mat A; 56f1769f9dSJose E. Roman 57f1769f9dSJose E. Roman PetscFunctionBegin; 58f1769f9dSJose E. Roman PetscCall(MatShellGetContext(N, &A)); 59f1769f9dSJose E. Roman PetscCall(MatSolveAdd(A, b, y, x)); 60f1769f9dSJose E. Roman PetscFunctionReturn(PETSC_SUCCESS); 61f1769f9dSJose E. Roman } 62f1769f9dSJose E. Roman 63f1769f9dSJose E. Roman static PetscErrorCode MatMatSolve_Transpose_LU(Mat N, Mat B, Mat X) 64f1769f9dSJose E. Roman { 65f1769f9dSJose E. Roman Mat A; 66f1769f9dSJose E. Roman 67f1769f9dSJose E. Roman PetscFunctionBegin; 68f1769f9dSJose E. Roman PetscCall(MatShellGetContext(N, &A)); 69f1769f9dSJose E. Roman PetscCall(MatMatSolveTranspose(A, B, X)); 70f1769f9dSJose E. Roman PetscFunctionReturn(PETSC_SUCCESS); 71f1769f9dSJose E. Roman } 72f1769f9dSJose E. Roman 73f1769f9dSJose E. Roman static PetscErrorCode MatMatSolveTranspose_Transpose_LU(Mat N, Mat B, Mat X) 74f1769f9dSJose E. Roman { 75f1769f9dSJose E. Roman Mat A; 76f1769f9dSJose E. Roman 77f1769f9dSJose E. Roman PetscFunctionBegin; 78f1769f9dSJose E. Roman PetscCall(MatShellGetContext(N, &A)); 79f1769f9dSJose E. Roman PetscCall(MatMatSolve(A, B, X)); 80f1769f9dSJose E. Roman PetscFunctionReturn(PETSC_SUCCESS); 81f1769f9dSJose E. Roman } 82f1769f9dSJose E. Roman 83f1769f9dSJose E. Roman static PetscErrorCode MatLUFactor_Transpose(Mat N, IS row, IS col, const MatFactorInfo *minfo) 84f1769f9dSJose E. Roman { 85f1769f9dSJose E. Roman Mat A; 86f1769f9dSJose E. Roman 87f1769f9dSJose E. Roman PetscFunctionBegin; 88f1769f9dSJose E. Roman PetscCall(MatShellGetContext(N, &A)); 89f1769f9dSJose E. Roman PetscCall(MatLUFactor(A, col, row, minfo)); 90f1769f9dSJose E. Roman PetscCall(MatShellSetOperation(N, MATOP_SOLVE, (void (*)(void))MatSolve_Transpose_LU)); 91f1769f9dSJose E. Roman PetscCall(MatShellSetOperation(N, MATOP_SOLVE_ADD, (void (*)(void))MatSolveAdd_Transpose_LU)); 92f1769f9dSJose E. Roman PetscCall(MatShellSetOperation(N, MATOP_SOLVE_TRANSPOSE, (void (*)(void))MatSolveTranspose_Transpose_LU)); 93f1769f9dSJose E. Roman PetscCall(MatShellSetOperation(N, MATOP_SOLVE_TRANSPOSE_ADD, (void (*)(void))MatSolveTransposeAdd_Transpose_LU)); 94f1769f9dSJose E. Roman PetscCall(MatShellSetOperation(N, MATOP_MAT_SOLVE, (void (*)(void))MatMatSolve_Transpose_LU)); 95f1769f9dSJose E. Roman PetscCall(MatShellSetOperation(N, MATOP_MAT_SOLVE_TRANSPOSE, (void (*)(void))MatMatSolveTranspose_Transpose_LU)); 96f1769f9dSJose E. Roman PetscFunctionReturn(PETSC_SUCCESS); 97f1769f9dSJose E. Roman } 98f1769f9dSJose E. Roman 99f1769f9dSJose E. Roman static PetscErrorCode MatSolve_Transpose_Cholesky(Mat N, Vec b, Vec x) 100f1769f9dSJose E. Roman { 101f1769f9dSJose E. Roman Mat A; 102f1769f9dSJose E. Roman 103f1769f9dSJose E. Roman PetscFunctionBegin; 104f1769f9dSJose E. Roman PetscCall(MatShellGetContext(N, &A)); 105f1769f9dSJose E. Roman PetscCall(MatSolveTranspose(A, b, x)); 106f1769f9dSJose E. Roman PetscFunctionReturn(PETSC_SUCCESS); 107f1769f9dSJose E. Roman } 108f1769f9dSJose E. Roman 109f1769f9dSJose E. Roman static PetscErrorCode MatSolveAdd_Transpose_Cholesky(Mat N, Vec b, Vec y, Vec x) 110f1769f9dSJose E. Roman { 111f1769f9dSJose E. Roman Mat A; 112f1769f9dSJose E. Roman 113f1769f9dSJose E. Roman PetscFunctionBegin; 114f1769f9dSJose E. Roman PetscCall(MatShellGetContext(N, &A)); 115f1769f9dSJose E. Roman PetscCall(MatSolveTransposeAdd(A, b, y, x)); 116f1769f9dSJose E. Roman PetscFunctionReturn(PETSC_SUCCESS); 117f1769f9dSJose E. Roman } 118f1769f9dSJose E. Roman 119f1769f9dSJose E. Roman static PetscErrorCode MatSolveTranspose_Transpose_Cholesky(Mat N, Vec b, Vec x) 120f1769f9dSJose E. Roman { 121f1769f9dSJose E. Roman Mat A; 122f1769f9dSJose E. Roman 123f1769f9dSJose E. Roman PetscFunctionBegin; 124f1769f9dSJose E. Roman PetscCall(MatShellGetContext(N, &A)); 125f1769f9dSJose E. Roman PetscCall(MatSolve(A, b, x)); 126f1769f9dSJose E. Roman PetscFunctionReturn(PETSC_SUCCESS); 127f1769f9dSJose E. Roman } 128f1769f9dSJose E. Roman 129f1769f9dSJose E. Roman static PetscErrorCode MatSolveTransposeAdd_Transpose_Cholesky(Mat N, Vec b, Vec y, Vec x) 130f1769f9dSJose E. Roman { 131f1769f9dSJose E. Roman Mat A; 132f1769f9dSJose E. Roman 133f1769f9dSJose E. Roman PetscFunctionBegin; 134f1769f9dSJose E. Roman PetscCall(MatShellGetContext(N, &A)); 135f1769f9dSJose E. Roman PetscCall(MatSolveAdd(A, b, y, x)); 136f1769f9dSJose E. Roman PetscFunctionReturn(PETSC_SUCCESS); 137f1769f9dSJose E. Roman } 138f1769f9dSJose E. Roman 139f1769f9dSJose E. Roman static PetscErrorCode MatMatSolve_Transpose_Cholesky(Mat N, Mat B, Mat X) 140f1769f9dSJose E. Roman { 141f1769f9dSJose E. Roman Mat A; 142f1769f9dSJose E. Roman 143f1769f9dSJose E. Roman PetscFunctionBegin; 144f1769f9dSJose E. Roman PetscCall(MatShellGetContext(N, &A)); 145f1769f9dSJose E. Roman PetscCall(MatMatSolveTranspose(A, B, X)); 146f1769f9dSJose E. Roman PetscFunctionReturn(PETSC_SUCCESS); 147f1769f9dSJose E. Roman } 148f1769f9dSJose E. Roman 149f1769f9dSJose E. Roman static PetscErrorCode MatMatSolveTranspose_Transpose_Cholesky(Mat N, Mat B, Mat X) 150f1769f9dSJose E. Roman { 151f1769f9dSJose E. Roman Mat A; 152f1769f9dSJose E. Roman 153f1769f9dSJose E. Roman PetscFunctionBegin; 154f1769f9dSJose E. Roman PetscCall(MatShellGetContext(N, &A)); 155f1769f9dSJose E. Roman PetscCall(MatMatSolve(A, B, X)); 156f1769f9dSJose E. Roman PetscFunctionReturn(PETSC_SUCCESS); 157f1769f9dSJose E. Roman } 158f1769f9dSJose E. Roman 159f1769f9dSJose E. Roman static PetscErrorCode MatCholeskyFactor_Transpose(Mat N, IS perm, const MatFactorInfo *minfo) 160f1769f9dSJose E. Roman { 161f1769f9dSJose E. Roman Mat A; 162f1769f9dSJose E. Roman 163f1769f9dSJose E. Roman PetscFunctionBegin; 164f1769f9dSJose E. Roman PetscCall(MatShellGetContext(N, &A)); 165f1769f9dSJose E. Roman PetscCall(MatCholeskyFactor(A, perm, minfo)); 166f1769f9dSJose E. Roman PetscCall(MatShellSetOperation(N, MATOP_SOLVE, (void (*)(void))MatSolve_Transpose_Cholesky)); 167f1769f9dSJose E. Roman PetscCall(MatShellSetOperation(N, MATOP_SOLVE_ADD, (void (*)(void))MatSolveAdd_Transpose_Cholesky)); 168f1769f9dSJose E. Roman PetscCall(MatShellSetOperation(N, MATOP_SOLVE_TRANSPOSE, (void (*)(void))MatSolveTranspose_Transpose_Cholesky)); 169f1769f9dSJose E. Roman PetscCall(MatShellSetOperation(N, MATOP_SOLVE_TRANSPOSE_ADD, (void (*)(void))MatSolveTransposeAdd_Transpose_Cholesky)); 170f1769f9dSJose E. Roman PetscCall(MatShellSetOperation(N, MATOP_MAT_SOLVE, (void (*)(void))MatMatSolve_Transpose_Cholesky)); 171f1769f9dSJose E. Roman PetscCall(MatShellSetOperation(N, MATOP_MAT_SOLVE_TRANSPOSE, (void (*)(void))MatMatSolveTranspose_Transpose_Cholesky)); 172f1769f9dSJose E. Roman PetscFunctionReturn(PETSC_SUCCESS); 173f1769f9dSJose E. Roman } 174f1769f9dSJose E. Roman 175*f5bab676SJose E. Roman static PetscErrorCode MatLUFactorNumeric_Transpose(Mat F, Mat N, const MatFactorInfo *info) 176*f5bab676SJose E. Roman { 177*f5bab676SJose E. Roman Mat A, FA; 178*f5bab676SJose E. Roman 179*f5bab676SJose E. Roman PetscFunctionBegin; 180*f5bab676SJose E. Roman PetscCall(MatShellGetContext(N, &A)); 181*f5bab676SJose E. Roman PetscCall(MatShellGetContext(F, &FA)); 182*f5bab676SJose E. Roman PetscCall(MatLUFactorNumeric(FA, A, info)); 183*f5bab676SJose E. Roman PetscCall(MatShellSetOperation(F, MATOP_SOLVE, (void (*)(void))MatSolve_Transpose_LU)); 184*f5bab676SJose E. Roman PetscCall(MatShellSetOperation(F, MATOP_SOLVE_ADD, (void (*)(void))MatSolveAdd_Transpose_LU)); 185*f5bab676SJose E. Roman PetscCall(MatShellSetOperation(F, MATOP_SOLVE_TRANSPOSE, (void (*)(void))MatSolveTranspose_Transpose_LU)); 186*f5bab676SJose E. Roman PetscCall(MatShellSetOperation(F, MATOP_SOLVE_TRANSPOSE_ADD, (void (*)(void))MatSolveTransposeAdd_Transpose_LU)); 187*f5bab676SJose E. Roman PetscCall(MatShellSetOperation(F, MATOP_MAT_SOLVE, (void (*)(void))MatMatSolve_Transpose_LU)); 188*f5bab676SJose E. Roman PetscCall(MatShellSetOperation(F, MATOP_MAT_SOLVE_TRANSPOSE, (void (*)(void))MatMatSolveTranspose_Transpose_LU)); 189*f5bab676SJose E. Roman PetscFunctionReturn(PETSC_SUCCESS); 190*f5bab676SJose E. Roman } 191*f5bab676SJose E. Roman 192*f5bab676SJose E. Roman static PetscErrorCode MatLUFactorSymbolic_Transpose(Mat F, Mat N, IS row, IS col, const MatFactorInfo *info) 193*f5bab676SJose E. Roman { 194*f5bab676SJose E. Roman Mat A, FA; 195*f5bab676SJose E. Roman 196*f5bab676SJose E. Roman PetscFunctionBegin; 197*f5bab676SJose E. Roman PetscCall(MatShellGetContext(N, &A)); 198*f5bab676SJose E. Roman PetscCall(MatShellGetContext(F, &FA)); 199*f5bab676SJose E. Roman PetscCall(MatLUFactorSymbolic(FA, A, row, col, info)); 200*f5bab676SJose E. Roman PetscCall(MatShellSetOperation(F, MATOP_LUFACTOR_NUMERIC, (void (*)(void))MatLUFactorNumeric_Transpose)); 201*f5bab676SJose E. Roman PetscFunctionReturn(PETSC_SUCCESS); 202*f5bab676SJose E. Roman } 203*f5bab676SJose E. Roman 204*f5bab676SJose E. Roman static PetscErrorCode MatCholeskyFactorNumeric_Transpose(Mat F, Mat N, const MatFactorInfo *info) 205*f5bab676SJose E. Roman { 206*f5bab676SJose E. Roman Mat A, FA; 207*f5bab676SJose E. Roman 208*f5bab676SJose E. Roman PetscFunctionBegin; 209*f5bab676SJose E. Roman PetscCall(MatShellGetContext(N, &A)); 210*f5bab676SJose E. Roman PetscCall(MatShellGetContext(F, &FA)); 211*f5bab676SJose E. Roman PetscCall(MatCholeskyFactorNumeric(FA, A, info)); 212*f5bab676SJose E. Roman PetscCall(MatShellSetOperation(F, MATOP_SOLVE, (void (*)(void))MatSolve_Transpose_Cholesky)); 213*f5bab676SJose E. Roman PetscCall(MatShellSetOperation(F, MATOP_SOLVE_ADD, (void (*)(void))MatSolveAdd_Transpose_Cholesky)); 214*f5bab676SJose E. Roman PetscCall(MatShellSetOperation(F, MATOP_SOLVE_TRANSPOSE, (void (*)(void))MatSolveTranspose_Transpose_Cholesky)); 215*f5bab676SJose E. Roman PetscCall(MatShellSetOperation(F, MATOP_SOLVE_TRANSPOSE_ADD, (void (*)(void))MatSolveTransposeAdd_Transpose_Cholesky)); 216*f5bab676SJose E. Roman PetscCall(MatShellSetOperation(F, MATOP_MAT_SOLVE, (void (*)(void))MatMatSolve_Transpose_Cholesky)); 217*f5bab676SJose E. Roman PetscCall(MatShellSetOperation(F, MATOP_MAT_SOLVE_TRANSPOSE, (void (*)(void))MatMatSolveTranspose_Transpose_Cholesky)); 218*f5bab676SJose E. Roman PetscFunctionReturn(PETSC_SUCCESS); 219*f5bab676SJose E. Roman } 220*f5bab676SJose E. Roman 221*f5bab676SJose E. Roman static PetscErrorCode MatCholeskyFactorSymbolic_Transpose(Mat F, Mat N, IS perm, const MatFactorInfo *info) 222*f5bab676SJose E. Roman { 223*f5bab676SJose E. Roman Mat A, FA; 224*f5bab676SJose E. Roman 225*f5bab676SJose E. Roman PetscFunctionBegin; 226*f5bab676SJose E. Roman PetscCall(MatShellGetContext(N, &A)); 227*f5bab676SJose E. Roman PetscCall(MatShellGetContext(F, &FA)); 228*f5bab676SJose E. Roman PetscCall(MatCholeskyFactorSymbolic(FA, A, perm, info)); 229*f5bab676SJose E. Roman PetscCall(MatShellSetOperation(F, MATOP_CHOLESKY_FACTOR_NUMERIC, (void (*)(void))MatCholeskyFactorNumeric_Transpose)); 230*f5bab676SJose E. Roman PetscFunctionReturn(PETSC_SUCCESS); 231*f5bab676SJose E. Roman } 232*f5bab676SJose E. Roman 233*f5bab676SJose E. Roman static PetscErrorCode MatGetFactor_Transpose(Mat N, MatSolverType type, MatFactorType ftype, Mat *F) 234*f5bab676SJose E. Roman { 235*f5bab676SJose E. Roman Mat A, FA; 236*f5bab676SJose E. Roman 237*f5bab676SJose E. Roman PetscFunctionBegin; 238*f5bab676SJose E. Roman PetscCall(MatShellGetContext(N, &A)); 239*f5bab676SJose E. Roman PetscCall(MatGetFactor(A, type, ftype, &FA)); 240*f5bab676SJose E. Roman PetscCall(MatCreateTranspose(FA, F)); 241*f5bab676SJose E. Roman if (ftype == MAT_FACTOR_LU) PetscCall(MatShellSetOperation(*F, MATOP_LUFACTOR_SYMBOLIC, (void (*)(void))MatLUFactorSymbolic_Transpose)); 242*f5bab676SJose E. Roman else if (ftype == MAT_FACTOR_CHOLESKY) { 243*f5bab676SJose E. Roman PetscCall(MatShellSetOperation(*F, MATOP_CHOLESKY_FACTOR_SYMBOLIC, (void (*)(void))MatCholeskyFactorSymbolic_Transpose)); 244*f5bab676SJose E. Roman PetscCall(MatPropagateSymmetryOptions(A, FA)); 245*f5bab676SJose E. Roman } else SETERRQ(PetscObjectComm((PetscObject)N), PETSC_ERR_SUP, "Support for factor type %s not implemented in MATTRANSPOSEVIRTUAL", MatFactorTypes[ftype]); 246*f5bab676SJose E. Roman PetscCall(MatDestroy(&FA)); 247*f5bab676SJose E. Roman PetscFunctionReturn(PETSC_SUCCESS); 248*f5bab676SJose E. Roman } 249*f5bab676SJose E. Roman 25066976f2fSJacob Faibussowitsch static PetscErrorCode MatDestroy_Transpose(Mat N) 251d71ae5a4SJacob Faibussowitsch { 252bf477422SJose E. Roman Mat A; 25385e3dda7SBarry Smith 25485e3dda7SBarry Smith PetscFunctionBegin; 255bf477422SJose E. Roman PetscCall(MatShellGetContext(N, &A)); 256bf477422SJose E. Roman PetscCall(MatDestroy(&A)); 2579566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)N, "MatTransposeGetMat_C", NULL)); 2589566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)N, "MatProductSetFromOptions_anytype_C", NULL)); 259543844c4SJose E. Roman PetscCall(PetscObjectComposeFunction((PetscObject)N, "MatShellSetContext_C", NULL)); 2603ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 26185e3dda7SBarry Smith } 26285e3dda7SBarry Smith 26366976f2fSJacob Faibussowitsch static PetscErrorCode MatDuplicate_Transpose(Mat N, MatDuplicateOption op, Mat *m) 264d71ae5a4SJacob Faibussowitsch { 265a1f56445SPierre Jolivet Mat A, C; 266d0de2241SAndrew Spott 267d0de2241SAndrew Spott PetscFunctionBegin; 268bf477422SJose E. Roman PetscCall(MatShellGetContext(N, &A)); 269a1f56445SPierre Jolivet PetscCall(MatDuplicate(A, op, &C)); 270a1f56445SPierre Jolivet PetscCall(MatCreateTranspose(C, m)); 271f1769f9dSJose E. Roman if (op == MAT_COPY_VALUES) { 272f1769f9dSJose E. Roman PetscCall(MatCopy(N, *m, SAME_NONZERO_PATTERN)); 273f1769f9dSJose E. Roman PetscCall(MatPropagateSymmetryOptions(A, C)); 274f1769f9dSJose E. Roman } 275a1f56445SPierre Jolivet PetscCall(MatDestroy(&C)); 2763ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 277d0de2241SAndrew Spott } 278d0de2241SAndrew Spott 27966976f2fSJacob Faibussowitsch static PetscErrorCode MatHasOperation_Transpose(Mat mat, MatOperation op, PetscBool *has) 280d71ae5a4SJacob Faibussowitsch { 281bf477422SJose E. Roman Mat A; 28252c5f739Sprj- 283bf477422SJose E. Roman PetscFunctionBegin; 284bf477422SJose E. Roman PetscCall(MatShellGetContext(mat, &A)); 28552c5f739Sprj- *has = PETSC_FALSE; 286bf477422SJose E. Roman if (op == MATOP_MULT || op == MATOP_MULT_ADD) { 287bf477422SJose E. Roman PetscCall(MatHasOperation(A, MATOP_MULT_TRANSPOSE, has)); 288bf477422SJose E. Roman } else if (op == MATOP_MULT_TRANSPOSE || op == MATOP_MULT_TRANSPOSE_ADD) { 289bf477422SJose E. Roman PetscCall(MatHasOperation(A, MATOP_MULT, has)); 2903c6db4c4SPierre Jolivet } else if (((void **)mat->ops)[op]) *has = PETSC_TRUE; 2913ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 29252c5f739Sprj- } 29352c5f739Sprj- 294d71ae5a4SJacob Faibussowitsch PETSC_INTERN PetscErrorCode MatProductSetFromOptions_Transpose(Mat D) 295d71ae5a4SJacob Faibussowitsch { 2966718818eSStefano Zampini Mat A, B, C, Ain, Bin, Cin; 2976718818eSStefano Zampini PetscBool Aistrans, Bistrans, Cistrans; 2986718818eSStefano Zampini PetscInt Atrans, Btrans, Ctrans; 2996718818eSStefano Zampini MatProductType ptype; 3006718818eSStefano Zampini 3016718818eSStefano Zampini PetscFunctionBegin; 3026718818eSStefano Zampini MatCheckProduct(D, 1); 3036718818eSStefano Zampini A = D->product->A; 3046718818eSStefano Zampini B = D->product->B; 3056718818eSStefano Zampini C = D->product->C; 306013e2dc7SBarry Smith PetscCall(PetscObjectTypeCompare((PetscObject)A, MATTRANSPOSEVIRTUAL, &Aistrans)); 307013e2dc7SBarry Smith PetscCall(PetscObjectTypeCompare((PetscObject)B, MATTRANSPOSEVIRTUAL, &Bistrans)); 308013e2dc7SBarry Smith PetscCall(PetscObjectTypeCompare((PetscObject)C, MATTRANSPOSEVIRTUAL, &Cistrans)); 309aed4548fSBarry Smith PetscCheck(Aistrans || Bistrans || Cistrans, PetscObjectComm((PetscObject)D), PETSC_ERR_PLIB, "This should not happen"); 3106718818eSStefano Zampini Atrans = 0; 3116718818eSStefano Zampini Ain = A; 3126718818eSStefano Zampini while (Aistrans) { 3136718818eSStefano Zampini Atrans++; 3149566063dSJacob Faibussowitsch PetscCall(MatTransposeGetMat(Ain, &Ain)); 315013e2dc7SBarry Smith PetscCall(PetscObjectTypeCompare((PetscObject)Ain, MATTRANSPOSEVIRTUAL, &Aistrans)); 3166718818eSStefano Zampini } 3176718818eSStefano Zampini Btrans = 0; 3186718818eSStefano Zampini Bin = B; 3196718818eSStefano Zampini while (Bistrans) { 3206718818eSStefano Zampini Btrans++; 3219566063dSJacob Faibussowitsch PetscCall(MatTransposeGetMat(Bin, &Bin)); 322013e2dc7SBarry Smith PetscCall(PetscObjectTypeCompare((PetscObject)Bin, MATTRANSPOSEVIRTUAL, &Bistrans)); 3236718818eSStefano Zampini } 3246718818eSStefano Zampini Ctrans = 0; 3256718818eSStefano Zampini Cin = C; 3266718818eSStefano Zampini while (Cistrans) { 3276718818eSStefano Zampini Ctrans++; 3289566063dSJacob Faibussowitsch PetscCall(MatTransposeGetMat(Cin, &Cin)); 329013e2dc7SBarry Smith PetscCall(PetscObjectTypeCompare((PetscObject)Cin, MATTRANSPOSEVIRTUAL, &Cistrans)); 3306718818eSStefano Zampini } 3316718818eSStefano Zampini Atrans = Atrans % 2; 3326718818eSStefano Zampini Btrans = Btrans % 2; 3336718818eSStefano Zampini Ctrans = Ctrans % 2; 3346718818eSStefano Zampini ptype = D->product->type; /* same product type by default */ 335b94d7dedSBarry Smith if (Ain->symmetric == PETSC_BOOL3_TRUE) Atrans = 0; 336b94d7dedSBarry Smith if (Bin->symmetric == PETSC_BOOL3_TRUE) Btrans = 0; 337b94d7dedSBarry Smith if (Cin && Cin->symmetric == PETSC_BOOL3_TRUE) Ctrans = 0; 3386718818eSStefano Zampini 3396718818eSStefano Zampini if (Atrans || Btrans || Ctrans) { 3406718818eSStefano Zampini ptype = MATPRODUCT_UNSPECIFIED; 3416718818eSStefano Zampini switch (D->product->type) { 3426718818eSStefano Zampini case MATPRODUCT_AB: 3436718818eSStefano Zampini if (Atrans && Btrans) { /* At * Bt we do not have support for this */ 3446718818eSStefano Zampini /* TODO custom implementation ? */ 3456718818eSStefano Zampini } else if (Atrans) { /* At * B */ 3466718818eSStefano Zampini ptype = MATPRODUCT_AtB; 3476718818eSStefano Zampini } else { /* A * Bt */ 3486718818eSStefano Zampini ptype = MATPRODUCT_ABt; 3496718818eSStefano Zampini } 3506718818eSStefano Zampini break; 3516718818eSStefano Zampini case MATPRODUCT_AtB: 3526718818eSStefano Zampini if (Atrans && Btrans) { /* A * Bt */ 3536718818eSStefano Zampini ptype = MATPRODUCT_ABt; 3546718818eSStefano Zampini } else if (Atrans) { /* A * B */ 3556718818eSStefano Zampini ptype = MATPRODUCT_AB; 3566718818eSStefano Zampini } else { /* At * Bt we do not have support for this */ 3576718818eSStefano Zampini /* TODO custom implementation ? */ 3586718818eSStefano Zampini } 3596718818eSStefano Zampini break; 3606718818eSStefano Zampini case MATPRODUCT_ABt: 3616718818eSStefano Zampini if (Atrans && Btrans) { /* At * B */ 3626718818eSStefano Zampini ptype = MATPRODUCT_AtB; 3636718818eSStefano Zampini } else if (Atrans) { /* At * Bt we do not have support for this */ 3646718818eSStefano Zampini /* TODO custom implementation ? */ 3656718818eSStefano Zampini } else { /* A * B */ 3666718818eSStefano Zampini ptype = MATPRODUCT_AB; 3676718818eSStefano Zampini } 3686718818eSStefano Zampini break; 3696718818eSStefano Zampini case MATPRODUCT_PtAP: 3706718818eSStefano Zampini if (Atrans) { /* PtAtP */ 3716718818eSStefano Zampini /* TODO custom implementation ? */ 3726718818eSStefano Zampini } else { /* RARt */ 3736718818eSStefano Zampini ptype = MATPRODUCT_RARt; 3746718818eSStefano Zampini } 3756718818eSStefano Zampini break; 3766718818eSStefano Zampini case MATPRODUCT_RARt: 3776718818eSStefano Zampini if (Atrans) { /* RAtRt */ 3786718818eSStefano Zampini /* TODO custom implementation ? */ 3796718818eSStefano Zampini } else { /* PtAP */ 3806718818eSStefano Zampini ptype = MATPRODUCT_PtAP; 3816718818eSStefano Zampini } 3826718818eSStefano Zampini break; 3836718818eSStefano Zampini case MATPRODUCT_ABC: 3846718818eSStefano Zampini /* TODO custom implementation ? */ 3856718818eSStefano Zampini break; 386d71ae5a4SJacob Faibussowitsch default: 387d71ae5a4SJacob Faibussowitsch SETERRQ(PetscObjectComm((PetscObject)D), PETSC_ERR_SUP, "ProductType %s is not supported", MatProductTypes[D->product->type]); 3886718818eSStefano Zampini } 3896718818eSStefano Zampini } 3909566063dSJacob Faibussowitsch PetscCall(MatProductReplaceMats(Ain, Bin, Cin, D)); 3919566063dSJacob Faibussowitsch PetscCall(MatProductSetType(D, ptype)); 3929566063dSJacob Faibussowitsch PetscCall(MatProductSetFromOptions(D)); 3933ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 3946718818eSStefano Zampini } 3956718818eSStefano Zampini 396bf477422SJose E. Roman static PetscErrorCode MatGetDiagonal_Transpose(Mat N, Vec v) 397d71ae5a4SJacob Faibussowitsch { 398bf477422SJose E. Roman Mat A; 399a0eea678SPierre Jolivet 400a0eea678SPierre Jolivet PetscFunctionBegin; 401bf477422SJose E. Roman PetscCall(MatShellGetContext(N, &A)); 402bf477422SJose E. Roman PetscCall(MatGetDiagonal(A, v)); 4033ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 404a0eea678SPierre Jolivet } 405a0eea678SPierre Jolivet 406a1f56445SPierre Jolivet static PetscErrorCode MatCopy_Transpose(Mat A, Mat B, MatStructure str) 407a1f56445SPierre Jolivet { 408a1f56445SPierre Jolivet Mat a, b; 409a1f56445SPierre Jolivet 410a1f56445SPierre Jolivet PetscFunctionBegin; 411a1f56445SPierre Jolivet PetscCall(MatShellGetContext(A, &a)); 412a1f56445SPierre Jolivet PetscCall(MatShellGetContext(B, &b)); 413a1f56445SPierre Jolivet PetscCall(MatCopy(a, b, str)); 414a1f56445SPierre Jolivet PetscFunctionReturn(PETSC_SUCCESS); 415a1f56445SPierre Jolivet } 416a1f56445SPierre Jolivet 417bf477422SJose E. Roman static PetscErrorCode MatConvert_Transpose(Mat N, MatType newtype, MatReuse reuse, Mat *newmat) 418d71ae5a4SJacob Faibussowitsch { 419bf477422SJose E. Roman Mat A; 420b22c5e46SPierre Jolivet PetscScalar vscale = 1.0, vshift = 0.0; 4216a4403aaSStefano Zampini PetscBool flg; 422a0eea678SPierre Jolivet 423a0eea678SPierre Jolivet PetscFunctionBegin; 424bf477422SJose E. Roman PetscCall(MatShellGetContext(N, &A)); 425bf477422SJose E. Roman PetscCall(MatHasOperation(A, MATOP_TRANSPOSE, &flg)); 426b22c5e46SPierre Jolivet if (flg || N->ops->getrow) { /* if this condition is false, MatConvert_Shell() will be called in MatConvert_Basic(), so the following checks are not needed */ 427b22c5e46SPierre Jolivet PetscCheck(!((Mat_Shell *)N->data)->zrows && !((Mat_Shell *)N->data)->zcols, PetscObjectComm((PetscObject)N), PETSC_ERR_SUP, "Cannot call MatConvert() if MatZeroRows() or MatZeroRowsColumns() has been called on the input Mat"); 428b22c5e46SPierre Jolivet PetscCheck(!((Mat_Shell *)N->data)->axpy, PetscObjectComm((PetscObject)N), PETSC_ERR_SUP, "Cannot call MatConvert() if MatAXPY() has been called on the input Mat"); 429b22c5e46SPierre Jolivet PetscCheck(!((Mat_Shell *)N->data)->left && !((Mat_Shell *)N->data)->right, PetscObjectComm((PetscObject)N), PETSC_ERR_SUP, "Cannot call MatConvert() if MatDiagonalScale() has been called on the input Mat"); 430b22c5e46SPierre Jolivet PetscCheck(!((Mat_Shell *)N->data)->dshift, PetscObjectComm((PetscObject)N), PETSC_ERR_SUP, "Cannot call MatConvert() if MatDiagonalSet() has been called on the input Mat"); 431b22c5e46SPierre Jolivet vscale = ((Mat_Shell *)N->data)->vscale; 432b22c5e46SPierre Jolivet vshift = ((Mat_Shell *)N->data)->vshift; 433b22c5e46SPierre Jolivet } 4346a4403aaSStefano Zampini if (flg) { 4356a4403aaSStefano Zampini Mat B; 4366a4403aaSStefano Zampini 437bf477422SJose E. Roman PetscCall(MatTranspose(A, MAT_INITIAL_MATRIX, &B)); 438ff83db7bSPierre Jolivet if (reuse != MAT_INPLACE_MATRIX) { 4399566063dSJacob Faibussowitsch PetscCall(MatConvert(B, newtype, reuse, newmat)); 4409566063dSJacob Faibussowitsch PetscCall(MatDestroy(&B)); 441ff83db7bSPierre Jolivet } else { 4429566063dSJacob Faibussowitsch PetscCall(MatConvert(B, newtype, MAT_INPLACE_MATRIX, &B)); 443bf477422SJose E. Roman PetscCall(MatHeaderReplace(N, &B)); 444ff83db7bSPierre Jolivet } 4456a4403aaSStefano Zampini } else { /* use basic converter as fallback */ 446b22c5e46SPierre Jolivet flg = (PetscBool)(N->ops->getrow != NULL); 447bf477422SJose E. Roman PetscCall(MatConvert_Basic(N, newtype, reuse, newmat)); 4486a4403aaSStefano Zampini } 449b22c5e46SPierre Jolivet if (flg) { 450b22c5e46SPierre Jolivet PetscCall(MatScale(*newmat, vscale)); 451b22c5e46SPierre Jolivet PetscCall(MatShift(*newmat, vshift)); 452b22c5e46SPierre Jolivet } 4533ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 454a0eea678SPierre Jolivet } 455a0eea678SPierre Jolivet 456bf477422SJose E. Roman static PetscErrorCode MatTransposeGetMat_Transpose(Mat N, Mat *M) 457d71ae5a4SJacob Faibussowitsch { 4588060fb66Sstefano_zampini PetscFunctionBegin; 459bf477422SJose E. Roman PetscCall(MatShellGetContext(N, M)); 4603ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 4618060fb66Sstefano_zampini } 4628060fb66Sstefano_zampini 4638060fb66Sstefano_zampini /*@ 464013e2dc7SBarry Smith MatTransposeGetMat - Gets the `Mat` object stored inside a `MATTRANSPOSEVIRTUAL` 4658060fb66Sstefano_zampini 46620f4b53cSBarry Smith Logically Collective 4678060fb66Sstefano_zampini 4688060fb66Sstefano_zampini Input Parameter: 469013e2dc7SBarry Smith . A - the `MATTRANSPOSEVIRTUAL` matrix 4708060fb66Sstefano_zampini 4718060fb66Sstefano_zampini Output Parameter: 4722ef1f0ffSBarry Smith . M - the matrix object stored inside `A` 4738060fb66Sstefano_zampini 4748060fb66Sstefano_zampini Level: intermediate 4758060fb66Sstefano_zampini 4761cc06b55SBarry Smith .seealso: [](ch_matrices), `Mat`, `MATTRANSPOSEVIRTUAL`, `MatCreateTranspose()` 4778060fb66Sstefano_zampini @*/ 478d71ae5a4SJacob Faibussowitsch PetscErrorCode MatTransposeGetMat(Mat A, Mat *M) 479d71ae5a4SJacob Faibussowitsch { 4808060fb66Sstefano_zampini PetscFunctionBegin; 4818060fb66Sstefano_zampini PetscValidHeaderSpecific(A, MAT_CLASSID, 1); 4828060fb66Sstefano_zampini PetscValidType(A, 1); 4834f572ea9SToby Isaac PetscAssertPointer(M, 2); 484cac4c232SBarry Smith PetscUseMethod(A, "MatTransposeGetMat_C", (Mat, Mat *), (A, M)); 4853ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 4868060fb66Sstefano_zampini } 487d0de2241SAndrew Spott 48811a5261eSBarry Smith /*MC 489013e2dc7SBarry Smith MATTRANSPOSEVIRTUAL - "transpose" - A matrix type that represents a virtual transpose of a matrix 49085e3dda7SBarry Smith 49111a5261eSBarry Smith Level: advanced 49211a5261eSBarry Smith 493a1f56445SPierre Jolivet Developer Notes: 494543844c4SJose E. Roman This is implemented on top of `MATSHELL` to get support for scaling and shifting without requiring duplicate code 495543844c4SJose E. Roman 496543844c4SJose E. Roman Users can not call `MatShellSetOperation()` operations on this class, there is some error checking for that incorrect usage 497543844c4SJose E. Roman 4981cc06b55SBarry Smith .seealso: [](ch_matrices), `Mat`, `MATHERMITIANTRANSPOSEVIRTUAL`, `Mat`, `MatCreateHermitianTranspose()`, `MatCreateTranspose()`, 4992ef1f0ffSBarry Smith `MATNORMALHERMITIAN`, `MATNORMAL` 50011a5261eSBarry Smith M*/ 50111a5261eSBarry Smith 50211a5261eSBarry Smith /*@ 503013e2dc7SBarry Smith MatCreateTranspose - Creates a new matrix `MATTRANSPOSEVIRTUAL` object that behaves like A' 50411a5261eSBarry Smith 505c3339decSBarry Smith Collective 50685e3dda7SBarry Smith 50785e3dda7SBarry Smith Input Parameter: 50885e3dda7SBarry Smith . A - the (possibly rectangular) matrix 50985e3dda7SBarry Smith 51085e3dda7SBarry Smith Output Parameter: 51185e3dda7SBarry Smith . N - the matrix that represents A' 51285e3dda7SBarry Smith 51385e3dda7SBarry Smith Level: intermediate 51485e3dda7SBarry Smith 51511a5261eSBarry Smith Note: 51695452b02SPatrick Sanan The transpose A' is NOT actually formed! Rather the new matrix 51711a5261eSBarry Smith object performs the matrix-vector product by using the `MatMultTranspose()` on 51885e3dda7SBarry Smith the original matrix 51985e3dda7SBarry Smith 5201cc06b55SBarry Smith .seealso: [](ch_matrices), `Mat`, `MATTRANSPOSEVIRTUAL`, `MatCreateNormal()`, `MatMult()`, `MatMultTranspose()`, `MatCreate()`, 5212ef1f0ffSBarry Smith `MATNORMALHERMITIAN` 52285e3dda7SBarry Smith @*/ 523d71ae5a4SJacob Faibussowitsch PetscErrorCode MatCreateTranspose(Mat A, Mat *N) 524d71ae5a4SJacob Faibussowitsch { 525487d878eSStefano Zampini VecType vtype; 52685e3dda7SBarry Smith 52785e3dda7SBarry Smith PetscFunctionBegin; 5289566063dSJacob Faibussowitsch PetscCall(MatCreate(PetscObjectComm((PetscObject)A), N)); 52987971105SStefano Zampini PetscCall(PetscLayoutReference(A->rmap, &((*N)->cmap))); 53087971105SStefano Zampini PetscCall(PetscLayoutReference(A->cmap, &((*N)->rmap))); 531bf477422SJose E. Roman PetscCall(MatSetType(*N, MATSHELL)); 532bf477422SJose E. Roman PetscCall(MatShellSetContext(*N, A)); 5339566063dSJacob Faibussowitsch PetscCall(PetscObjectReference((PetscObject)A)); 53485e3dda7SBarry Smith 5359566063dSJacob Faibussowitsch PetscCall(MatSetBlockSizes(*N, PetscAbs(A->cmap->bs), PetscAbs(A->rmap->bs))); 5369566063dSJacob Faibussowitsch PetscCall(MatGetVecType(A, &vtype)); 5379566063dSJacob Faibussowitsch PetscCall(MatSetVecType(*N, vtype)); 5382487f3f2SStefano Zampini #if defined(PETSC_HAVE_DEVICE) 5399566063dSJacob Faibussowitsch PetscCall(MatBindToCPU(*N, A->boundtocpu)); 5402487f3f2SStefano Zampini #endif 5419566063dSJacob Faibussowitsch PetscCall(MatSetUp(*N)); 542bf477422SJose E. Roman 543bf477422SJose E. Roman PetscCall(MatShellSetOperation(*N, MATOP_DESTROY, (void (*)(void))MatDestroy_Transpose)); 544bf477422SJose E. Roman PetscCall(MatShellSetOperation(*N, MATOP_MULT, (void (*)(void))MatMult_Transpose)); 545bf477422SJose E. Roman PetscCall(MatShellSetOperation(*N, MATOP_MULT_TRANSPOSE, (void (*)(void))MatMultTranspose_Transpose)); 546f1769f9dSJose E. Roman PetscCall(MatShellSetOperation(*N, MATOP_LUFACTOR, (void (*)(void))MatLUFactor_Transpose)); 547f1769f9dSJose E. Roman PetscCall(MatShellSetOperation(*N, MATOP_CHOLESKYFACTOR, (void (*)(void))MatCholeskyFactor_Transpose)); 548*f5bab676SJose E. Roman PetscCall(MatShellSetOperation(*N, MATOP_GET_FACTOR, (void (*)(void))MatGetFactor_Transpose)); 549bf477422SJose E. Roman PetscCall(MatShellSetOperation(*N, MATOP_DUPLICATE, (void (*)(void))MatDuplicate_Transpose)); 550bf477422SJose E. Roman PetscCall(MatShellSetOperation(*N, MATOP_HAS_OPERATION, (void (*)(void))MatHasOperation_Transpose)); 551bf477422SJose E. Roman PetscCall(MatShellSetOperation(*N, MATOP_GET_DIAGONAL, (void (*)(void))MatGetDiagonal_Transpose)); 552a1f56445SPierre Jolivet PetscCall(MatShellSetOperation(*N, MATOP_COPY, (void (*)(void))MatCopy_Transpose)); 553bf477422SJose E. Roman PetscCall(MatShellSetOperation(*N, MATOP_CONVERT, (void (*)(void))MatConvert_Transpose)); 554bf477422SJose E. Roman 555543844c4SJose E. Roman PetscCall(PetscObjectComposeFunction((PetscObject)*N, "MatTransposeGetMat_C", MatTransposeGetMat_Transpose)); 556543844c4SJose E. Roman PetscCall(PetscObjectComposeFunction((PetscObject)*N, "MatProductSetFromOptions_anytype_C", MatProductSetFromOptions_Transpose)); 557a1f56445SPierre Jolivet PetscCall(PetscObjectComposeFunction((PetscObject)*N, "MatShellSetContext_C", MatShellSetContext_Immutable)); 558a1f56445SPierre Jolivet PetscCall(PetscObjectComposeFunction((PetscObject)*N, "MatShellSetContextDestroy_C", MatShellSetContextDestroy_Immutable)); 559a1f56445SPierre Jolivet PetscCall(PetscObjectComposeFunction((PetscObject)*N, "MatShellSetManageScalingShifts_C", MatShellSetManageScalingShifts_Immutable)); 560bf477422SJose E. Roman PetscCall(PetscObjectChangeTypeName((PetscObject)*N, MATTRANSPOSEVIRTUAL)); 5613ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 56285e3dda7SBarry Smith } 563