1 #include <../src/mat/impls/shell/shell.h> /*I "petscmat.h" I*/ 2 3 static PetscErrorCode MatProductSetFromOptions_HT(Mat D) 4 { 5 Mat A, B, C, Ain, Bin, Cin; 6 PetscBool Aistrans, Bistrans, Cistrans; 7 PetscInt Atrans, Btrans, Ctrans; 8 MatProductType ptype; 9 10 PetscFunctionBegin; 11 MatCheckProduct(D, 1); 12 A = D->product->A; 13 B = D->product->B; 14 C = D->product->C; 15 PetscCall(PetscObjectTypeCompare((PetscObject)A, MATHERMITIANTRANSPOSEVIRTUAL, &Aistrans)); 16 PetscCall(PetscObjectTypeCompare((PetscObject)B, MATHERMITIANTRANSPOSEVIRTUAL, &Bistrans)); 17 PetscCall(PetscObjectTypeCompare((PetscObject)C, MATHERMITIANTRANSPOSEVIRTUAL, &Cistrans)); 18 PetscCheck(Aistrans || Bistrans || Cistrans, PetscObjectComm((PetscObject)D), PETSC_ERR_PLIB, "This should not happen"); 19 Atrans = 0; 20 Ain = A; 21 while (Aistrans) { 22 Atrans++; 23 PetscCall(MatHermitianTransposeGetMat(Ain, &Ain)); 24 PetscCall(PetscObjectTypeCompare((PetscObject)Ain, MATHERMITIANTRANSPOSEVIRTUAL, &Aistrans)); 25 } 26 Btrans = 0; 27 Bin = B; 28 while (Bistrans) { 29 Btrans++; 30 PetscCall(MatHermitianTransposeGetMat(Bin, &Bin)); 31 PetscCall(PetscObjectTypeCompare((PetscObject)Bin, MATHERMITIANTRANSPOSEVIRTUAL, &Bistrans)); 32 } 33 Ctrans = 0; 34 Cin = C; 35 while (Cistrans) { 36 Ctrans++; 37 PetscCall(MatHermitianTransposeGetMat(Cin, &Cin)); 38 PetscCall(PetscObjectTypeCompare((PetscObject)Cin, MATHERMITIANTRANSPOSEVIRTUAL, &Cistrans)); 39 } 40 Atrans = Atrans % 2; 41 Btrans = Btrans % 2; 42 Ctrans = Ctrans % 2; 43 ptype = D->product->type; /* same product type by default */ 44 if (Ain->symmetric == PETSC_BOOL3_TRUE) Atrans = 0; 45 if (Bin->symmetric == PETSC_BOOL3_TRUE) Btrans = 0; 46 if (Cin && Cin->symmetric == PETSC_BOOL3_TRUE) Ctrans = 0; 47 48 if (Atrans || Btrans || Ctrans) { 49 PetscCheck(!PetscDefined(USE_COMPLEX), PetscObjectComm((PetscObject)A), PETSC_ERR_SUP, "No support for complex Hermitian transpose matrices"); 50 ptype = MATPRODUCT_UNSPECIFIED; 51 switch (D->product->type) { 52 case MATPRODUCT_AB: 53 if (Atrans && Btrans) { /* At * Bt we do not have support for this */ 54 /* TODO custom implementation ? */ 55 } else if (Atrans) { /* At * B */ 56 ptype = MATPRODUCT_AtB; 57 } else { /* A * Bt */ 58 ptype = MATPRODUCT_ABt; 59 } 60 break; 61 case MATPRODUCT_AtB: 62 if (Atrans && Btrans) { /* A * Bt */ 63 ptype = MATPRODUCT_ABt; 64 } else if (Atrans) { /* A * B */ 65 ptype = MATPRODUCT_AB; 66 } else { /* At * Bt we do not have support for this */ 67 /* TODO custom implementation ? */ 68 } 69 break; 70 case MATPRODUCT_ABt: 71 if (Atrans && Btrans) { /* At * B */ 72 ptype = MATPRODUCT_AtB; 73 } else if (Atrans) { /* At * Bt we do not have support for this */ 74 /* TODO custom implementation ? */ 75 } else { /* A * B */ 76 ptype = MATPRODUCT_AB; 77 } 78 break; 79 case MATPRODUCT_PtAP: 80 if (Atrans) { /* PtAtP */ 81 /* TODO custom implementation ? */ 82 } else { /* RARt */ 83 ptype = MATPRODUCT_RARt; 84 } 85 break; 86 case MATPRODUCT_RARt: 87 if (Atrans) { /* RAtRt */ 88 /* TODO custom implementation ? */ 89 } else { /* PtAP */ 90 ptype = MATPRODUCT_PtAP; 91 } 92 break; 93 case MATPRODUCT_ABC: 94 /* TODO custom implementation ? */ 95 break; 96 default: 97 SETERRQ(PetscObjectComm((PetscObject)D), PETSC_ERR_SUP, "ProductType %s is not supported", MatProductTypes[D->product->type]); 98 } 99 } 100 PetscCall(MatProductReplaceMats(Ain, Bin, Cin, D)); 101 PetscCall(MatProductSetType(D, ptype)); 102 PetscCall(MatProductSetFromOptions(D)); 103 PetscFunctionReturn(PETSC_SUCCESS); 104 } 105 106 static PetscErrorCode MatMult_HT(Mat N, Vec x, Vec y) 107 { 108 Mat A; 109 110 PetscFunctionBegin; 111 PetscCall(MatShellGetContext(N, &A)); 112 PetscCall(MatMultHermitianTranspose(A, x, y)); 113 PetscFunctionReturn(PETSC_SUCCESS); 114 } 115 116 static PetscErrorCode MatMultHermitianTranspose_HT(Mat N, Vec x, Vec y) 117 { 118 Mat A; 119 120 PetscFunctionBegin; 121 PetscCall(MatShellGetContext(N, &A)); 122 PetscCall(MatMult(A, x, y)); 123 PetscFunctionReturn(PETSC_SUCCESS); 124 } 125 126 static PetscErrorCode MatSolve_HT_LU(Mat N, Vec b, Vec x) 127 { 128 Mat A; 129 Vec w; 130 131 PetscFunctionBegin; 132 PetscCall(MatShellGetContext(N, &A)); 133 PetscCall(VecDuplicate(b, &w)); 134 PetscCall(VecCopy(b, w)); 135 PetscCall(VecConjugate(w)); 136 PetscCall(MatSolveTranspose(A, w, x)); 137 PetscCall(VecConjugate(x)); 138 PetscCall(VecDestroy(&w)); 139 PetscFunctionReturn(PETSC_SUCCESS); 140 } 141 142 static PetscErrorCode MatSolveAdd_HT_LU(Mat N, Vec b, Vec y, Vec x) 143 { 144 Mat A; 145 Vec v, w; 146 147 PetscFunctionBegin; 148 PetscCall(MatShellGetContext(N, &A)); 149 PetscCall(VecDuplicate(b, &v)); 150 PetscCall(VecDuplicate(b, &w)); 151 PetscCall(VecCopy(y, v)); 152 PetscCall(VecCopy(b, w)); 153 PetscCall(VecConjugate(v)); 154 PetscCall(VecConjugate(w)); 155 PetscCall(MatSolveTransposeAdd(A, w, v, x)); 156 PetscCall(VecConjugate(x)); 157 PetscCall(VecDestroy(&v)); 158 PetscCall(VecDestroy(&w)); 159 PetscFunctionReturn(PETSC_SUCCESS); 160 } 161 162 static PetscErrorCode MatMatSolve_HT_LU(Mat N, Mat B, Mat X) 163 { 164 Mat A, W; 165 166 PetscFunctionBegin; 167 PetscCall(MatShellGetContext(N, &A)); 168 PetscCall(MatDuplicate(B, MAT_COPY_VALUES, &W)); 169 PetscCall(MatConjugate(W)); 170 PetscCall(MatMatSolveTranspose(A, W, X)); 171 PetscCall(MatConjugate(X)); 172 PetscCall(MatDestroy(&W)); 173 PetscFunctionReturn(PETSC_SUCCESS); 174 } 175 176 static PetscErrorCode MatLUFactor_HT(Mat N, IS row, IS col, const MatFactorInfo *minfo) 177 { 178 Mat A; 179 180 PetscFunctionBegin; 181 PetscCall(MatShellGetContext(N, &A)); 182 PetscCall(MatLUFactor(A, col, row, minfo)); 183 PetscCall(MatShellSetOperation(N, MATOP_SOLVE, (void (*)(void))MatSolve_HT_LU)); 184 PetscCall(MatShellSetOperation(N, MATOP_SOLVE_ADD, (void (*)(void))MatSolveAdd_HT_LU)); 185 PetscCall(MatShellSetOperation(N, MATOP_MAT_SOLVE, (void (*)(void))MatMatSolve_HT_LU)); 186 PetscFunctionReturn(PETSC_SUCCESS); 187 } 188 189 static PetscErrorCode MatSolve_HT_Cholesky(Mat N, Vec b, Vec x) 190 { 191 Mat A; 192 193 PetscFunctionBegin; 194 PetscCall(MatShellGetContext(N, &A)); 195 PetscCall(MatSolve(A, b, x)); 196 PetscFunctionReturn(PETSC_SUCCESS); 197 } 198 199 static PetscErrorCode MatSolveAdd_HT_Cholesky(Mat N, Vec b, Vec y, Vec x) 200 { 201 Mat A; 202 Vec v, w; 203 204 PetscFunctionBegin; 205 PetscCall(MatShellGetContext(N, &A)); 206 PetscCall(VecDuplicate(b, &v)); 207 PetscCall(VecDuplicate(b, &w)); 208 PetscCall(VecCopy(y, v)); 209 PetscCall(VecCopy(b, w)); 210 PetscCall(VecConjugate(v)); 211 PetscCall(VecConjugate(w)); 212 PetscCall(MatSolveTransposeAdd(A, w, v, x)); 213 PetscCall(VecConjugate(x)); 214 PetscCall(VecDestroy(&v)); 215 PetscCall(VecDestroy(&w)); 216 PetscFunctionReturn(PETSC_SUCCESS); 217 } 218 219 static PetscErrorCode MatMatSolve_HT_Cholesky(Mat N, Mat B, Mat X) 220 { 221 Mat A, W; 222 223 PetscFunctionBegin; 224 PetscCall(MatShellGetContext(N, &A)); 225 PetscCall(MatDuplicate(B, MAT_COPY_VALUES, &W)); 226 PetscCall(MatConjugate(W)); 227 PetscCall(MatMatSolveTranspose(A, W, X)); 228 PetscCall(MatConjugate(X)); 229 PetscCall(MatDestroy(&W)); 230 PetscFunctionReturn(PETSC_SUCCESS); 231 } 232 233 static PetscErrorCode MatCholeskyFactor_HT(Mat N, IS perm, const MatFactorInfo *minfo) 234 { 235 Mat A; 236 237 PetscFunctionBegin; 238 PetscCall(MatShellGetContext(N, &A)); 239 PetscCheck(!PetscDefined(USE_COMPLEX) || A->hermitian == PETSC_BOOL3_TRUE, PetscObjectComm((PetscObject)A), PETSC_ERR_SUP, "Cholesky supported only if original matrix is Hermitian"); 240 PetscCall(MatCholeskyFactor(A, perm, minfo)); 241 PetscCall(MatShellSetOperation(N, MATOP_SOLVE, (void (*)(void))MatSolve_HT_Cholesky)); 242 PetscCall(MatShellSetOperation(N, MATOP_SOLVE_ADD, (void (*)(void))MatSolveAdd_HT_Cholesky)); 243 PetscCall(MatShellSetOperation(N, MATOP_MAT_SOLVE, (void (*)(void))MatMatSolve_HT_Cholesky)); 244 PetscFunctionReturn(PETSC_SUCCESS); 245 } 246 247 static PetscErrorCode MatLUFactorNumeric_HT(Mat F, Mat N, const MatFactorInfo *info) 248 { 249 Mat A, FA; 250 251 PetscFunctionBegin; 252 PetscCall(MatShellGetContext(N, &A)); 253 PetscCall(MatShellGetContext(F, &FA)); 254 PetscCall(MatLUFactorNumeric(FA, A, info)); 255 PetscCall(MatShellSetOperation(F, MATOP_SOLVE, (void (*)(void))MatSolve_HT_LU)); 256 PetscCall(MatShellSetOperation(F, MATOP_SOLVE_ADD, (void (*)(void))MatSolveAdd_HT_LU)); 257 PetscCall(MatShellSetOperation(F, MATOP_MAT_SOLVE, (void (*)(void))MatMatSolve_HT_LU)); 258 PetscFunctionReturn(PETSC_SUCCESS); 259 } 260 261 static PetscErrorCode MatLUFactorSymbolic_HT(Mat F, Mat N, IS row, IS col, const MatFactorInfo *info) 262 { 263 Mat A, FA; 264 265 PetscFunctionBegin; 266 PetscCall(MatShellGetContext(N, &A)); 267 PetscCall(MatShellGetContext(F, &FA)); 268 PetscCall(MatLUFactorSymbolic(FA, A, row, col, info)); 269 PetscCall(MatShellSetOperation(F, MATOP_LUFACTOR_NUMERIC, (void (*)(void))MatLUFactorNumeric_HT)); 270 PetscFunctionReturn(PETSC_SUCCESS); 271 } 272 273 static PetscErrorCode MatCholeskyFactorNumeric_HT(Mat F, Mat N, const MatFactorInfo *info) 274 { 275 Mat A, FA; 276 277 PetscFunctionBegin; 278 PetscCall(MatShellGetContext(N, &A)); 279 PetscCall(MatShellGetContext(F, &FA)); 280 PetscCall(MatCholeskyFactorNumeric(FA, A, info)); 281 PetscCall(MatShellSetOperation(F, MATOP_SOLVE, (void (*)(void))MatSolve_HT_Cholesky)); 282 PetscCall(MatShellSetOperation(F, MATOP_SOLVE_ADD, (void (*)(void))MatSolveAdd_HT_Cholesky)); 283 PetscCall(MatShellSetOperation(F, MATOP_MAT_SOLVE, (void (*)(void))MatMatSolve_HT_Cholesky)); 284 PetscFunctionReturn(PETSC_SUCCESS); 285 } 286 287 static PetscErrorCode MatCholeskyFactorSymbolic_HT(Mat F, Mat N, IS perm, const MatFactorInfo *info) 288 { 289 Mat A, FA; 290 291 PetscFunctionBegin; 292 PetscCall(MatShellGetContext(N, &A)); 293 PetscCall(MatShellGetContext(F, &FA)); 294 PetscCall(MatCholeskyFactorSymbolic(FA, A, perm, info)); 295 PetscCall(MatShellSetOperation(F, MATOP_CHOLESKY_FACTOR_NUMERIC, (void (*)(void))MatCholeskyFactorNumeric_HT)); 296 PetscFunctionReturn(PETSC_SUCCESS); 297 } 298 299 static PetscErrorCode MatGetFactor_HT(Mat N, MatSolverType type, MatFactorType ftype, Mat *F) 300 { 301 Mat A, FA; 302 303 PetscFunctionBegin; 304 PetscCall(MatShellGetContext(N, &A)); 305 PetscCall(MatGetFactor(A, type, ftype, &FA)); 306 PetscCall(MatCreateTranspose(FA, F)); 307 if (ftype == MAT_FACTOR_LU) PetscCall(MatShellSetOperation(*F, MATOP_LUFACTOR_SYMBOLIC, (void (*)(void))MatLUFactorSymbolic_HT)); 308 else if (ftype == MAT_FACTOR_CHOLESKY) { 309 PetscCheck(!PetscDefined(USE_COMPLEX) || A->hermitian == PETSC_BOOL3_TRUE, PetscObjectComm((PetscObject)A), PETSC_ERR_SUP, "Cholesky supported only if original matrix is Hermitian"); 310 PetscCall(MatPropagateSymmetryOptions(A, FA)); 311 PetscCall(MatShellSetOperation(*F, MATOP_CHOLESKY_FACTOR_SYMBOLIC, (void (*)(void))MatCholeskyFactorSymbolic_HT)); 312 } else SETERRQ(PetscObjectComm((PetscObject)N), PETSC_ERR_SUP, "Support for factor type %s not implemented in MATTRANSPOSEVIRTUAL", MatFactorTypes[ftype]); 313 (*F)->factortype = ftype; 314 PetscCall(MatDestroy(&FA)); 315 PetscFunctionReturn(PETSC_SUCCESS); 316 } 317 318 static PetscErrorCode MatDestroy_HT(Mat N) 319 { 320 Mat A; 321 322 PetscFunctionBegin; 323 PetscCall(MatShellGetContext(N, &A)); 324 PetscCall(MatDestroy(&A)); 325 PetscCall(PetscObjectComposeFunction((PetscObject)N, "MatHermitianTransposeGetMat_C", NULL)); 326 #if !defined(PETSC_USE_COMPLEX) 327 PetscCall(PetscObjectComposeFunction((PetscObject)N, "MatTransposeGetMat_C", NULL)); 328 #endif 329 PetscCall(PetscObjectComposeFunction((PetscObject)N, "MatProductSetFromOptions_anytype_C", NULL)); 330 PetscCall(PetscObjectComposeFunction((PetscObject)N, "MatShellSetContext_C", NULL)); 331 PetscCall(PetscObjectComposeFunction((PetscObject)N, "MatFactorGetSolverType_C", NULL)); 332 PetscFunctionReturn(PETSC_SUCCESS); 333 } 334 335 static PetscErrorCode MatGetInfo_HT(Mat N, MatInfoType flag, MatInfo *info) 336 { 337 Mat A; 338 339 PetscFunctionBegin; 340 PetscCall(MatShellGetContext(N, &A)); 341 PetscCall(MatGetInfo(A, flag, info)); 342 PetscFunctionReturn(PETSC_SUCCESS); 343 } 344 345 static PetscErrorCode MatFactorGetSolverType_HT(Mat N, MatSolverType *type) 346 { 347 Mat A; 348 349 PetscFunctionBegin; 350 PetscCall(MatShellGetContext(N, &A)); 351 PetscCall(MatFactorGetSolverType(A, type)); 352 PetscFunctionReturn(PETSC_SUCCESS); 353 } 354 355 static PetscErrorCode MatDuplicate_HT(Mat N, MatDuplicateOption op, Mat *m) 356 { 357 Mat A, C; 358 359 PetscFunctionBegin; 360 PetscCall(MatShellGetContext(N, &A)); 361 PetscCall(MatDuplicate(A, op, &C)); 362 PetscCall(MatCreateHermitianTranspose(C, m)); 363 if (op == MAT_COPY_VALUES) PetscCall(MatCopy(N, *m, SAME_NONZERO_PATTERN)); 364 PetscCall(MatDestroy(&C)); 365 PetscFunctionReturn(PETSC_SUCCESS); 366 } 367 368 static PetscErrorCode MatHasOperation_HT(Mat mat, MatOperation op, PetscBool *has) 369 { 370 Mat A; 371 372 PetscFunctionBegin; 373 PetscCall(MatShellGetContext(mat, &A)); 374 *has = PETSC_FALSE; 375 if (op == MATOP_MULT || op == MATOP_MULT_ADD) { 376 PetscCall(MatHasOperation(A, MATOP_MULT_HERMITIAN_TRANSPOSE, has)); 377 if (!*has) PetscCall(MatHasOperation(A, MATOP_MULT_TRANSPOSE, has)); 378 } else if (op == MATOP_MULT_HERMITIAN_TRANSPOSE || op == MATOP_MULT_HERMITIAN_TRANS_ADD || op == MATOP_MULT_TRANSPOSE || op == MATOP_MULT_TRANSPOSE_ADD) { 379 PetscCall(MatHasOperation(A, MATOP_MULT, has)); 380 } else if (((void **)mat->ops)[op]) *has = PETSC_TRUE; 381 PetscFunctionReturn(PETSC_SUCCESS); 382 } 383 384 static PetscErrorCode MatHermitianTransposeGetMat_HT(Mat N, Mat *M) 385 { 386 PetscFunctionBegin; 387 PetscCall(MatShellGetContext(N, M)); 388 PetscFunctionReturn(PETSC_SUCCESS); 389 } 390 391 /*@ 392 MatHermitianTransposeGetMat - Gets the `Mat` object stored inside a `MATHERMITIANTRANSPOSEVIRTUAL` 393 394 Logically Collective 395 396 Input Parameter: 397 . A - the `MATHERMITIANTRANSPOSEVIRTUAL` matrix 398 399 Output Parameter: 400 . M - the matrix object stored inside A 401 402 Level: intermediate 403 404 .seealso: [](ch_matrices), `Mat`, `MATHERMITIANTRANSPOSEVIRTUAL`, `MatCreateHermitianTranspose()` 405 @*/ 406 PetscErrorCode MatHermitianTransposeGetMat(Mat A, Mat *M) 407 { 408 PetscFunctionBegin; 409 PetscValidHeaderSpecific(A, MAT_CLASSID, 1); 410 PetscValidType(A, 1); 411 PetscAssertPointer(M, 2); 412 PetscUseMethod(A, "MatHermitianTransposeGetMat_C", (Mat, Mat *), (A, M)); 413 PetscFunctionReturn(PETSC_SUCCESS); 414 } 415 416 static PetscErrorCode MatGetDiagonal_HT(Mat N, Vec v) 417 { 418 Mat A; 419 420 PetscFunctionBegin; 421 PetscCall(MatShellGetContext(N, &A)); 422 PetscCall(MatGetDiagonal(A, v)); 423 PetscCall(VecConjugate(v)); 424 PetscFunctionReturn(PETSC_SUCCESS); 425 } 426 427 static PetscErrorCode MatCopy_HT(Mat A, Mat B, MatStructure str) 428 { 429 Mat a, b; 430 431 PetscFunctionBegin; 432 PetscCall(MatShellGetContext(A, &a)); 433 PetscCall(MatShellGetContext(B, &b)); 434 PetscCall(MatCopy(a, b, str)); 435 PetscFunctionReturn(PETSC_SUCCESS); 436 } 437 438 static PetscErrorCode MatConvert_HT(Mat N, MatType newtype, MatReuse reuse, Mat *newmat) 439 { 440 Mat A; 441 PetscScalar vscale = 1.0, vshift = 0.0; 442 PetscBool flg; 443 444 PetscFunctionBegin; 445 PetscCall(MatShellGetContext(N, &A)); 446 PetscCall(MatHasOperation(A, MATOP_HERMITIAN_TRANSPOSE, &flg)); 447 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 */ 448 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"); 449 PetscCheck(!((Mat_Shell *)N->data)->axpy, PetscObjectComm((PetscObject)N), PETSC_ERR_SUP, "Cannot call MatConvert() if MatAXPY() has been called on the input Mat"); 450 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"); 451 PetscCheck(!((Mat_Shell *)N->data)->dshift, PetscObjectComm((PetscObject)N), PETSC_ERR_SUP, "Cannot call MatConvert() if MatDiagonalSet() has been called on the input Mat"); 452 vscale = ((Mat_Shell *)N->data)->vscale; 453 vshift = ((Mat_Shell *)N->data)->vshift; 454 } 455 if (flg) { 456 Mat B; 457 458 PetscCall(MatHermitianTranspose(A, MAT_INITIAL_MATRIX, &B)); 459 if (reuse != MAT_INPLACE_MATRIX) { 460 PetscCall(MatConvert(B, newtype, reuse, newmat)); 461 PetscCall(MatDestroy(&B)); 462 } else { 463 PetscCall(MatConvert(B, newtype, MAT_INPLACE_MATRIX, &B)); 464 PetscCall(MatHeaderReplace(N, &B)); 465 } 466 } else { /* use basic converter as fallback */ 467 flg = (PetscBool)(N->ops->getrow != NULL); 468 PetscCall(MatConvert_Basic(N, newtype, reuse, newmat)); 469 } 470 if (flg) { 471 PetscCall(MatScale(*newmat, vscale)); 472 PetscCall(MatShift(*newmat, vshift)); 473 } 474 PetscFunctionReturn(PETSC_SUCCESS); 475 } 476 477 /*MC 478 MATHERMITIANTRANSPOSEVIRTUAL - "hermitiantranspose" - A matrix type that represents a virtual transpose of a matrix 479 480 Level: advanced 481 482 Developer Notes: 483 This is implemented on top of `MATSHELL` to get support for scaling and shifting without requiring duplicate code 484 485 Users can not call `MatShellSetOperation()` operations on this class, there is some error checking for that incorrect usage 486 487 .seealso: [](ch_matrices), `Mat`, `MATTRANSPOSEVIRTUAL`, `Mat`, `MatCreateHermitianTranspose()`, `MatCreateTranspose()` 488 M*/ 489 490 /*@ 491 MatCreateHermitianTranspose - Creates a new matrix object of `MatType` `MATHERMITIANTRANSPOSEVIRTUAL` that behaves like A'* 492 493 Collective 494 495 Input Parameter: 496 . A - the (possibly rectangular) matrix 497 498 Output Parameter: 499 . N - the matrix that represents A'* 500 501 Level: intermediate 502 503 Note: 504 The Hermitian transpose A' is NOT actually formed! Rather the new matrix 505 object performs the matrix-vector product, `MatMult()`, by using the `MatMultHermitianTranspose()` on 506 the original matrix 507 508 .seealso: [](ch_matrices), `Mat`, `MatCreateNormal()`, `MatMult()`, `MatMultHermitianTranspose()`, `MatCreate()`, 509 `MATTRANSPOSEVIRTUAL`, `MatCreateTranspose()`, `MatHermitianTransposeGetMat()`, `MATNORMAL`, `MATNORMALHERMITIAN` 510 @*/ 511 PetscErrorCode MatCreateHermitianTranspose(Mat A, Mat *N) 512 { 513 VecType vtype; 514 515 PetscFunctionBegin; 516 PetscCall(MatCreate(PetscObjectComm((PetscObject)A), N)); 517 PetscCall(PetscLayoutReference(A->rmap, &((*N)->cmap))); 518 PetscCall(PetscLayoutReference(A->cmap, &((*N)->rmap))); 519 PetscCall(MatSetType(*N, MATSHELL)); 520 PetscCall(MatShellSetContext(*N, A)); 521 PetscCall(PetscObjectReference((PetscObject)A)); 522 523 PetscCall(MatSetBlockSizes(*N, PetscAbs(A->cmap->bs), PetscAbs(A->rmap->bs))); 524 PetscCall(MatGetVecType(A, &vtype)); 525 PetscCall(MatSetVecType(*N, vtype)); 526 #if defined(PETSC_HAVE_DEVICE) 527 PetscCall(MatBindToCPU(*N, A->boundtocpu)); 528 #endif 529 PetscCall(MatSetUp(*N)); 530 531 PetscCall(MatShellSetOperation(*N, MATOP_DESTROY, (void (*)(void))MatDestroy_HT)); 532 PetscCall(MatShellSetOperation(*N, MATOP_MULT, (void (*)(void))MatMult_HT)); 533 PetscCall(MatShellSetOperation(*N, MATOP_MULT_HERMITIAN_TRANSPOSE, (void (*)(void))MatMultHermitianTranspose_HT)); 534 #if !defined(PETSC_USE_COMPLEX) 535 PetscCall(MatShellSetOperation(*N, MATOP_MULT_TRANSPOSE, (void (*)(void))MatMultHermitianTranspose_HT)); 536 #endif 537 PetscCall(MatShellSetOperation(*N, MATOP_LUFACTOR, (void (*)(void))MatLUFactor_HT)); 538 PetscCall(MatShellSetOperation(*N, MATOP_CHOLESKYFACTOR, (void (*)(void))MatCholeskyFactor_HT)); 539 PetscCall(MatShellSetOperation(*N, MATOP_GET_FACTOR, (void (*)(void))MatGetFactor_HT)); 540 PetscCall(MatShellSetOperation(*N, MATOP_GETINFO, (void (*)(void))MatGetInfo_HT)); 541 PetscCall(MatShellSetOperation(*N, MATOP_DUPLICATE, (void (*)(void))MatDuplicate_HT)); 542 PetscCall(MatShellSetOperation(*N, MATOP_HAS_OPERATION, (void (*)(void))MatHasOperation_HT)); 543 PetscCall(MatShellSetOperation(*N, MATOP_GET_DIAGONAL, (void (*)(void))MatGetDiagonal_HT)); 544 PetscCall(MatShellSetOperation(*N, MATOP_COPY, (void (*)(void))MatCopy_HT)); 545 PetscCall(MatShellSetOperation(*N, MATOP_CONVERT, (void (*)(void))MatConvert_HT)); 546 547 PetscCall(PetscObjectComposeFunction((PetscObject)*N, "MatHermitianTransposeGetMat_C", MatHermitianTransposeGetMat_HT)); 548 #if !defined(PETSC_USE_COMPLEX) 549 PetscCall(PetscObjectComposeFunction((PetscObject)*N, "MatTransposeGetMat_C", MatHermitianTransposeGetMat_HT)); 550 #endif 551 PetscCall(PetscObjectComposeFunction((PetscObject)*N, "MatProductSetFromOptions_anytype_C", MatProductSetFromOptions_HT)); 552 PetscCall(PetscObjectComposeFunction((PetscObject)*N, "MatFactorGetSolverType_C", MatFactorGetSolverType_HT)); 553 PetscCall(PetscObjectComposeFunction((PetscObject)*N, "MatShellSetContext_C", MatShellSetContext_Immutable)); 554 PetscCall(PetscObjectComposeFunction((PetscObject)*N, "MatShellSetContextDestroy_C", MatShellSetContextDestroy_Immutable)); 555 PetscCall(PetscObjectComposeFunction((PetscObject)*N, "MatShellSetManageScalingShifts_C", MatShellSetManageScalingShifts_Immutable)); 556 PetscCall(PetscObjectChangeTypeName((PetscObject)*N, MATHERMITIANTRANSPOSEVIRTUAL)); 557 PetscFunctionReturn(PETSC_SUCCESS); 558 } 559