xref: /petsc/src/mat/impls/transpose/transm.c (revision f5bab676488eab7ccb90dc895dfba268b72c04e0)
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