xref: /petsc/src/mat/impls/aij/seq/matptap.c (revision 7e44a18ef85fdf2ba969602d2c492a61eef88079)
1eb9c0419SKris Buschelman /*
2eb9c0419SKris Buschelman   Defines projective product routines where A is a SeqAIJ matrix
3eb9c0419SKris Buschelman           C = P^T * A * P
4eb9c0419SKris Buschelman */
5eb9c0419SKris Buschelman 
6eb9c0419SKris Buschelman #include "src/mat/impls/aij/seq/aij.h"
7eb9c0419SKris Buschelman #include "src/mat/utils/freespace.h"
8eb9c0419SKris Buschelman 
9*7e44a18eSKris Buschelman EXTERN int MatSeqAIJPtAP(Mat,Mat,Mat*);
10*7e44a18eSKris Buschelman EXTERN int MatSeqAIJPtAPSymbolic(Mat,Mat,Mat*);
11*7e44a18eSKris Buschelman EXTERN int MatSeqAIJPtAPNumeric(Mat,Mat,Mat);
12*7e44a18eSKris Buschelman EXTERN int RegisterMatMatMultRoutines_Private(Mat);
13eb9c0419SKris Buschelman 
14eb9c0419SKris Buschelman static int MATSeqAIJ_PtAP         = 0;
15eb9c0419SKris Buschelman static int MATSeqAIJ_PtAPSymbolic = 0;
16eb9c0419SKris Buschelman static int MATSeqAIJ_PtAPNumeric  = 0;
17eb9c0419SKris Buschelman 
18eb9c0419SKris Buschelman /*
19eb9c0419SKris Buschelman      MatSeqAIJPtAP - Creates the SeqAIJ matrix product, C,
20eb9c0419SKris Buschelman            of SeqAIJ matrix A and matrix P:
21eb9c0419SKris Buschelman                  C = P^T * A * P;
22eb9c0419SKris Buschelman 
23eb9c0419SKris Buschelman      Note: C is assumed to be uncreated.
24eb9c0419SKris Buschelman            If this is not the case, Destroy C before calling this routine.
25eb9c0419SKris Buschelman */
26eb9c0419SKris Buschelman #undef __FUNCT__
27eb9c0419SKris Buschelman #define __FUNCT__ "MatSeqAIJPtAP"
28eb9c0419SKris Buschelman int MatSeqAIJPtAP(Mat A,Mat P,Mat *C) {
29eb9c0419SKris Buschelman   int ierr;
30eb9c0419SKris Buschelman   char funct[80];
31eb9c0419SKris Buschelman 
32eb9c0419SKris Buschelman   PetscFunctionBegin;
33eb9c0419SKris Buschelman 
34eb9c0419SKris Buschelman   ierr = PetscLogEventBegin(MATSeqAIJ_PtAP,A,P,0,0);CHKERRQ(ierr);
35eb9c0419SKris Buschelman 
36eb9c0419SKris Buschelman   ierr = MatSeqAIJPtAPSymbolic(A,P,C);CHKERRQ(ierr);
37eb9c0419SKris Buschelman 
38eb9c0419SKris Buschelman   /* Avoid additional error checking included in */
39eb9c0419SKris Buschelman /*   ierr = MatSeqAIJApplyPtAPNumeric(A,P,*C);CHKERRQ(ierr); */
40eb9c0419SKris Buschelman 
41eb9c0419SKris Buschelman   /* Query A for ApplyPtAPNumeric implementation based on types of P */
42eb9c0419SKris Buschelman   ierr = PetscStrcpy(funct,"MatApplyPtAPNumeric_seqaij_");CHKERRQ(ierr);
43eb9c0419SKris Buschelman   ierr = PetscStrcat(funct,P->type_name);CHKERRQ(ierr);
44eb9c0419SKris Buschelman   ierr = PetscTryMethod(A,funct,(Mat,Mat,Mat),(A,P,*C));CHKERRQ(ierr);
45eb9c0419SKris Buschelman 
46eb9c0419SKris Buschelman   ierr = PetscLogEventEnd(MATSeqAIJ_PtAP,A,P,0,0);CHKERRQ(ierr);
47eb9c0419SKris Buschelman 
48eb9c0419SKris Buschelman   PetscFunctionReturn(0);
49eb9c0419SKris Buschelman }
50eb9c0419SKris Buschelman 
51eb9c0419SKris Buschelman /*
52eb9c0419SKris Buschelman      MatSeqAIJPtAPSymbolic - Creates the (i,j) structure of the SeqAIJ matrix product, C,
53eb9c0419SKris Buschelman            of SeqAIJ matrix A and matrix P, according to:
54eb9c0419SKris Buschelman                  C = P^T * A * P;
55eb9c0419SKris Buschelman 
56eb9c0419SKris Buschelman      Note: C is assumed to be uncreated.
57eb9c0419SKris Buschelman            If this is not the case, Destroy C before calling this routine.
58eb9c0419SKris Buschelman */
59eb9c0419SKris Buschelman #undef __FUNCT__
60eb9c0419SKris Buschelman #define __FUNCT__ "MatSeqAIJPtAPSymbolic"
61eb9c0419SKris Buschelman int MatSeqAIJPtAPSymbolic(Mat A,Mat P,Mat *C) {
62eb9c0419SKris Buschelman   int ierr;
63eb9c0419SKris Buschelman   char funct[80];
64eb9c0419SKris Buschelman 
65eb9c0419SKris Buschelman   PetscFunctionBegin;
66eb9c0419SKris Buschelman 
67eb9c0419SKris Buschelman   PetscValidPointer(C);
68eb9c0419SKris Buschelman 
69eb9c0419SKris Buschelman   PetscValidHeaderSpecific(A,MAT_COOKIE);
70eb9c0419SKris Buschelman   PetscValidType(A);
71eb9c0419SKris Buschelman   MatPreallocated(A);
72eb9c0419SKris Buschelman   if (!A->assembled) SETERRQ(PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
73eb9c0419SKris Buschelman   if (A->factor) SETERRQ(PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
74eb9c0419SKris Buschelman 
75eb9c0419SKris Buschelman   PetscValidHeaderSpecific(P,MAT_COOKIE);
76eb9c0419SKris Buschelman   PetscValidType(P);
77eb9c0419SKris Buschelman   MatPreallocated(P);
78eb9c0419SKris Buschelman   if (!P->assembled) SETERRQ(PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
79eb9c0419SKris Buschelman   if (P->factor) SETERRQ(PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
80eb9c0419SKris Buschelman 
81eb9c0419SKris Buschelman   if (P->M!=A->N) SETERRQ2(PETSC_ERR_ARG_SIZ,"Matrix dimensions are incompatible, %d != %d",P->M,A->N);
82eb9c0419SKris Buschelman   if (A->M!=A->N) SETERRQ2(PETSC_ERR_ARG_SIZ,"Matrix 'A' must be square, %d != %d",A->M,A->N);
83eb9c0419SKris Buschelman 
84eb9c0419SKris Buschelman   /* Query A for ApplyPtAP implementation based on types of P */
85eb9c0419SKris Buschelman   ierr = PetscStrcpy(funct,"MatApplyPtAPSymbolic_seqaij_");CHKERRQ(ierr);
86eb9c0419SKris Buschelman   ierr = PetscStrcat(funct,P->type_name);CHKERRQ(ierr);
87eb9c0419SKris Buschelman   ierr = PetscTryMethod(A,funct,(Mat,Mat,Mat*),(A,P,C));CHKERRQ(ierr);
88eb9c0419SKris Buschelman 
89eb9c0419SKris Buschelman   PetscFunctionReturn(0);
90eb9c0419SKris Buschelman }
91eb9c0419SKris Buschelman 
92eb9c0419SKris Buschelman EXTERN_C_BEGIN
93eb9c0419SKris Buschelman #undef __FUNCT__
94eb9c0419SKris Buschelman #define __FUNCT__ "MatApplyPtAPSymbolic_SeqAIJ_SeqAIJ"
95eb9c0419SKris Buschelman int MatApplyPtAPSymbolic_SeqAIJ_SeqAIJ(Mat A,Mat P,Mat *C) {
96eb9c0419SKris Buschelman   int            ierr;
97eb9c0419SKris Buschelman   FreeSpaceList  free_space=PETSC_NULL,current_space=PETSC_NULL;
98eb9c0419SKris Buschelman   Mat_SeqAIJ     *a=(Mat_SeqAIJ*)A->data,*p=(Mat_SeqAIJ*)P->data,*c;
99eb9c0419SKris Buschelman   int            *pti,*ptj,*ptJ,*ai=a->i,*aj=a->j,*ajj,*pi=p->i,*pj=p->j,*pjj;
100eb9c0419SKris Buschelman   int            *ci,*cj,*denserow,*sparserow,*ptadenserow,*ptasparserow,*ptaj;
101eb9c0419SKris Buschelman   int            an=A->N,am=A->M,pn=P->N,pm=P->M;
102eb9c0419SKris Buschelman   int            i,j,k,ptnzi,arow,anzj,ptanzi,prow,pnzj,cnzi;
103eb9c0419SKris Buschelman   MatScalar      *ca;
104eb9c0419SKris Buschelman 
105eb9c0419SKris Buschelman   PetscFunctionBegin;
106eb9c0419SKris Buschelman 
107eb9c0419SKris Buschelman   /* Start timer */
108eb9c0419SKris Buschelman   ierr = PetscLogEventBegin(MATSeqAIJ_PtAPSymbolic,A,P,0,0);CHKERRQ(ierr);
109eb9c0419SKris Buschelman 
110eb9c0419SKris Buschelman   /* Get ij structure of P^T */
111eb9c0419SKris Buschelman   ierr = MatGetSymbolicTranspose_SeqAIJ(P,&pti,&ptj);CHKERRQ(ierr);
112eb9c0419SKris Buschelman   ptJ=ptj;
113eb9c0419SKris Buschelman 
114eb9c0419SKris Buschelman   /* Allocate ci array, arrays for fill computation and */
115eb9c0419SKris Buschelman   /* free space for accumulating nonzero column info */
1163985e5eaSKris Buschelman   ierr = PetscMalloc((pn+1)*sizeof(int),&ci);CHKERRQ(ierr);
117eb9c0419SKris Buschelman   ci[0] = 0;
118eb9c0419SKris Buschelman 
119eb9c0419SKris Buschelman   ierr = PetscMalloc((2*pn+2*an+1)*sizeof(int),&ptadenserow);CHKERRQ(ierr);
120eb9c0419SKris Buschelman   ierr = PetscMemzero(ptadenserow,(2*pn+2*an+1)*sizeof(int));CHKERRQ(ierr);
121eb9c0419SKris Buschelman   ptasparserow = ptadenserow  + an;
122eb9c0419SKris Buschelman   denserow     = ptasparserow + an;
123eb9c0419SKris Buschelman   sparserow    = denserow     + pn;
124eb9c0419SKris Buschelman 
125eb9c0419SKris Buschelman   /* Set initial free space to be nnz(A) scaled by aspect ratio of P. */
126eb9c0419SKris Buschelman   /* This should be reasonable if sparsity of PtAP is similar to that of A. */
127eb9c0419SKris Buschelman   ierr          = GetMoreSpace((ai[am]/pm)*pn,&free_space);
128eb9c0419SKris Buschelman   current_space = free_space;
129eb9c0419SKris Buschelman 
130eb9c0419SKris Buschelman   /* Determine symbolic info for each row of C: */
131eb9c0419SKris Buschelman   for (i=0;i<pn;i++) {
132eb9c0419SKris Buschelman     ptnzi  = pti[i+1] - pti[i];
133eb9c0419SKris Buschelman     ptanzi = 0;
134eb9c0419SKris Buschelman     /* Determine symbolic row of PtA: */
135eb9c0419SKris Buschelman     for (j=0;j<ptnzi;j++) {
136eb9c0419SKris Buschelman       arow = *ptJ++;
137eb9c0419SKris Buschelman       anzj = ai[arow+1] - ai[arow];
138eb9c0419SKris Buschelman       ajj  = aj + ai[arow];
139eb9c0419SKris Buschelman       for (k=0;k<anzj;k++) {
140eb9c0419SKris Buschelman         if (!ptadenserow[ajj[k]]) {
141eb9c0419SKris Buschelman           ptadenserow[ajj[k]]    = -1;
142eb9c0419SKris Buschelman           ptasparserow[ptanzi++] = ajj[k];
143eb9c0419SKris Buschelman         }
144eb9c0419SKris Buschelman       }
145eb9c0419SKris Buschelman     }
146eb9c0419SKris Buschelman       /* Using symbolic info for row of PtA, determine symbolic info for row of C: */
147eb9c0419SKris Buschelman     ptaj = ptasparserow;
148eb9c0419SKris Buschelman     cnzi   = 0;
149eb9c0419SKris Buschelman     for (j=0;j<ptanzi;j++) {
150eb9c0419SKris Buschelman       prow = *ptaj++;
151eb9c0419SKris Buschelman       pnzj = pi[prow+1] - pi[prow];
152eb9c0419SKris Buschelman       pjj  = pj + pi[prow];
153eb9c0419SKris Buschelman       for (k=0;k<pnzj;k++) {
154eb9c0419SKris Buschelman         if (!denserow[pjj[k]]) {
155eb9c0419SKris Buschelman             denserow[pjj[k]]  = -1;
156eb9c0419SKris Buschelman             sparserow[cnzi++] = pjj[k];
157eb9c0419SKris Buschelman         }
158eb9c0419SKris Buschelman       }
159eb9c0419SKris Buschelman     }
160eb9c0419SKris Buschelman 
161eb9c0419SKris Buschelman     /* sort sparserow */
162eb9c0419SKris Buschelman     ierr = PetscSortInt(cnzi,sparserow);CHKERRQ(ierr);
163eb9c0419SKris Buschelman 
164eb9c0419SKris Buschelman     /* If free space is not available, make more free space */
165eb9c0419SKris Buschelman     /* Double the amount of total space in the list */
166eb9c0419SKris Buschelman     if (current_space->local_remaining<cnzi) {
167eb9c0419SKris Buschelman       ierr = GetMoreSpace(current_space->total_array_size,&current_space);CHKERRQ(ierr);
168eb9c0419SKris Buschelman     }
169eb9c0419SKris Buschelman 
170eb9c0419SKris Buschelman     /* Copy data into free space, and zero out denserows */
171eb9c0419SKris Buschelman     ierr = PetscMemcpy(current_space->array,sparserow,cnzi*sizeof(int));CHKERRQ(ierr);
172eb9c0419SKris Buschelman     current_space->array           += cnzi;
173eb9c0419SKris Buschelman     current_space->local_used      += cnzi;
174eb9c0419SKris Buschelman     current_space->local_remaining -= cnzi;
175eb9c0419SKris Buschelman 
176eb9c0419SKris Buschelman     for (j=0;j<ptanzi;j++) {
177eb9c0419SKris Buschelman       ptadenserow[ptasparserow[j]] = 0;
178eb9c0419SKris Buschelman     }
179eb9c0419SKris Buschelman     for (j=0;j<cnzi;j++) {
180eb9c0419SKris Buschelman       denserow[sparserow[j]] = 0;
181eb9c0419SKris Buschelman     }
182eb9c0419SKris Buschelman       /* Aside: Perhaps we should save the pta info for the numerical factorization. */
183eb9c0419SKris Buschelman       /*        For now, we will recompute what is needed. */
184eb9c0419SKris Buschelman     ci[i+1] = ci[i] + cnzi;
185eb9c0419SKris Buschelman   }
186eb9c0419SKris Buschelman   /* nnz is now stored in ci[ptm], column indices are in the list of free space */
187eb9c0419SKris Buschelman   /* Allocate space for cj, initialize cj, and */
188eb9c0419SKris Buschelman   /* destroy list of free space and other temporary array(s) */
189eb9c0419SKris Buschelman   ierr = PetscMalloc((ci[pn]+1)*sizeof(int),&cj);CHKERRQ(ierr);
190eb9c0419SKris Buschelman   ierr = MakeSpaceContiguous(&free_space,cj);CHKERRQ(ierr);
191eb9c0419SKris Buschelman   ierr = PetscFree(ptadenserow);CHKERRQ(ierr);
192eb9c0419SKris Buschelman 
193eb9c0419SKris Buschelman   /* Allocate space for ca */
194eb9c0419SKris Buschelman   ierr = PetscMalloc((ci[pn]+1)*sizeof(MatScalar),&ca);CHKERRQ(ierr);
195eb9c0419SKris Buschelman   ierr = PetscMemzero(ca,(ci[pn]+1)*sizeof(MatScalar));CHKERRQ(ierr);
196eb9c0419SKris Buschelman 
197eb9c0419SKris Buschelman   /* put together the new matrix */
198eb9c0419SKris Buschelman   ierr = MatCreateSeqAIJWithArrays(A->comm,pn,pn,ci,cj,ca,C);CHKERRQ(ierr);
199eb9c0419SKris Buschelman 
200eb9c0419SKris Buschelman   /* MatCreateSeqAIJWithArrays flags matrix so PETSc doesn't free the user's arrays. */
201eb9c0419SKris Buschelman   /* Since these are PETSc arrays, change flags to free them as necessary. */
202eb9c0419SKris Buschelman   c = (Mat_SeqAIJ *)((*C)->data);
203eb9c0419SKris Buschelman   c->freedata = PETSC_TRUE;
204eb9c0419SKris Buschelman   c->nonew    = 0;
205eb9c0419SKris Buschelman 
206eb9c0419SKris Buschelman   /* Clean up. */
207eb9c0419SKris Buschelman   ierr = MatRestoreSymbolicTranspose_SeqAIJ(P,&pti,&ptj);CHKERRQ(ierr);
208eb9c0419SKris Buschelman 
209eb9c0419SKris Buschelman   ierr = PetscLogEventEnd(MATSeqAIJ_PtAPSymbolic,A,P,0,0);CHKERRQ(ierr);
210eb9c0419SKris Buschelman   PetscFunctionReturn(0);
211eb9c0419SKris Buschelman }
212eb9c0419SKris Buschelman EXTERN_C_END
213eb9c0419SKris Buschelman 
2143985e5eaSKris Buschelman #include "src/mat/impls/maij/maij.h"
2153985e5eaSKris Buschelman EXTERN_C_BEGIN
2163985e5eaSKris Buschelman #undef __FUNCT__
2173985e5eaSKris Buschelman #define __FUNCT__ "MatApplyPtAPSymbolic_SeqAIJ_SeqMAIJ"
2183985e5eaSKris Buschelman int MatApplyPtAPSymbolic_SeqAIJ_SeqMAIJ(Mat A,Mat PP,Mat *C) {
2195c66b693SKris Buschelman   /* This routine requires testing -- I don't think it works. */
2203985e5eaSKris Buschelman   int            ierr;
2213985e5eaSKris Buschelman   FreeSpaceList  free_space=PETSC_NULL,current_space=PETSC_NULL;
2223985e5eaSKris Buschelman   Mat_SeqMAIJ    *pp=(Mat_SeqMAIJ*)PP->data;
2233985e5eaSKris Buschelman   Mat            P=pp->AIJ;
2243985e5eaSKris Buschelman   Mat_SeqAIJ     *a=(Mat_SeqAIJ*)A->data,*p=(Mat_SeqAIJ*)P->data,*c;
2253985e5eaSKris Buschelman   int            *pti,*ptj,*ptJ,*ai=a->i,*aj=a->j,*ajj,*pi=p->i,*pj=p->j,*pjj;
2263985e5eaSKris Buschelman   int            *ci,*cj,*denserow,*sparserow,*ptadenserow,*ptasparserow,*ptaj;
2273985e5eaSKris Buschelman   int            an=A->N,am=A->M,pn=P->N,pm=P->M,ppdof=pp->dof;
228fe05a634SKris Buschelman   int            i,j,k,dof,pdof,ptnzi,arow,anzj,ptanzi,prow,pnzj,cnzi;
2293985e5eaSKris Buschelman   MatScalar      *ca;
2303985e5eaSKris Buschelman 
2313985e5eaSKris Buschelman   PetscFunctionBegin;
2323985e5eaSKris Buschelman   /* Start timer */
2333985e5eaSKris Buschelman   ierr = PetscLogEventBegin(MATSeqAIJ_PtAPSymbolic,A,PP,0,0);CHKERRQ(ierr);
2343985e5eaSKris Buschelman 
2353985e5eaSKris Buschelman   /* Get ij structure of P^T */
2363985e5eaSKris Buschelman   ierr = MatGetSymbolicTranspose_SeqAIJ(P,&pti,&ptj);CHKERRQ(ierr);
2373985e5eaSKris Buschelman 
2383985e5eaSKris Buschelman   /* Allocate ci array, arrays for fill computation and */
2393985e5eaSKris Buschelman   /* free space for accumulating nonzero column info */
2403985e5eaSKris Buschelman   ierr = PetscMalloc((pn+1)*sizeof(int),&ci);CHKERRQ(ierr);
2413985e5eaSKris Buschelman   ci[0] = 0;
2423985e5eaSKris Buschelman 
2433985e5eaSKris Buschelman   ierr = PetscMalloc((2*pn+2*an+1)*sizeof(int),&ptadenserow);CHKERRQ(ierr);
2443985e5eaSKris Buschelman   ierr = PetscMemzero(ptadenserow,(2*pn+2*an+1)*sizeof(int));CHKERRQ(ierr);
2453985e5eaSKris Buschelman   ptasparserow = ptadenserow  + an;
2463985e5eaSKris Buschelman   denserow     = ptasparserow + an;
2473985e5eaSKris Buschelman   sparserow    = denserow     + pn;
2483985e5eaSKris Buschelman 
2493985e5eaSKris Buschelman   /* Set initial free space to be nnz(A) scaled by aspect ratio of P. */
2503985e5eaSKris Buschelman   /* This should be reasonable if sparsity of PtAP is similar to that of A. */
2513985e5eaSKris Buschelman   ierr          = GetMoreSpace((ai[am]/pm)*pn,&free_space);
2523985e5eaSKris Buschelman   current_space = free_space;
2533985e5eaSKris Buschelman 
2543985e5eaSKris Buschelman   /* Determine symbolic info for each row of C: */
2553985e5eaSKris Buschelman   for (i=0;i<pn/ppdof;i++) {
2563985e5eaSKris Buschelman     ptnzi  = pti[i+1] - pti[i];
2573985e5eaSKris Buschelman     ptanzi = 0;
2583985e5eaSKris Buschelman     ptJ    = ptj + pti[i];
2593985e5eaSKris Buschelman     for (dof=0;dof<ppdof;dof++) {
2603985e5eaSKris Buschelman     /* Determine symbolic row of PtA: */
2613985e5eaSKris Buschelman       for (j=0;j<ptnzi;j++) {
2623985e5eaSKris Buschelman         arow = ptJ[j] + dof;
2633985e5eaSKris Buschelman         anzj = ai[arow+1] - ai[arow];
2643985e5eaSKris Buschelman         ajj  = aj + ai[arow];
2653985e5eaSKris Buschelman         for (k=0;k<anzj;k++) {
2663985e5eaSKris Buschelman           if (!ptadenserow[ajj[k]]) {
2673985e5eaSKris Buschelman             ptadenserow[ajj[k]]    = -1;
2683985e5eaSKris Buschelman             ptasparserow[ptanzi++] = ajj[k];
2693985e5eaSKris Buschelman           }
2703985e5eaSKris Buschelman         }
2713985e5eaSKris Buschelman       }
2723985e5eaSKris Buschelman       /* Using symbolic info for row of PtA, determine symbolic info for row of C: */
2733985e5eaSKris Buschelman       ptaj = ptasparserow;
2743985e5eaSKris Buschelman       cnzi   = 0;
2753985e5eaSKris Buschelman       for (j=0;j<ptanzi;j++) {
276fe05a634SKris Buschelman         pdof = *ptaj%dof;
2773985e5eaSKris Buschelman         prow = (*ptaj++)/dof;
2783985e5eaSKris Buschelman         pnzj = pi[prow+1] - pi[prow];
2793985e5eaSKris Buschelman         pjj  = pj + pi[prow];
2803985e5eaSKris Buschelman         for (k=0;k<pnzj;k++) {
281fe05a634SKris Buschelman           if (!denserow[pjj[k]+pdof]) {
282fe05a634SKris Buschelman             denserow[pjj[k]+pdof] = -1;
283fe05a634SKris Buschelman             sparserow[cnzi++]     = pjj[k]+pdof;
2843985e5eaSKris Buschelman           }
2853985e5eaSKris Buschelman         }
2863985e5eaSKris Buschelman       }
2873985e5eaSKris Buschelman 
2883985e5eaSKris Buschelman       /* sort sparserow */
2893985e5eaSKris Buschelman       ierr = PetscSortInt(cnzi,sparserow);CHKERRQ(ierr);
2903985e5eaSKris Buschelman 
2913985e5eaSKris Buschelman       /* If free space is not available, make more free space */
2923985e5eaSKris Buschelman       /* Double the amount of total space in the list */
2933985e5eaSKris Buschelman       if (current_space->local_remaining<cnzi) {
2943985e5eaSKris Buschelman         ierr = GetMoreSpace(current_space->total_array_size,&current_space);CHKERRQ(ierr);
2953985e5eaSKris Buschelman       }
2963985e5eaSKris Buschelman 
2973985e5eaSKris Buschelman       /* Copy data into free space, and zero out denserows */
2983985e5eaSKris Buschelman       ierr = PetscMemcpy(current_space->array,sparserow,cnzi*sizeof(int));CHKERRQ(ierr);
2993985e5eaSKris Buschelman       current_space->array           += cnzi;
3003985e5eaSKris Buschelman       current_space->local_used      += cnzi;
3013985e5eaSKris Buschelman       current_space->local_remaining -= cnzi;
3023985e5eaSKris Buschelman 
3033985e5eaSKris Buschelman       for (j=0;j<ptanzi;j++) {
3043985e5eaSKris Buschelman         ptadenserow[ptasparserow[j]] = 0;
3053985e5eaSKris Buschelman       }
3063985e5eaSKris Buschelman       for (j=0;j<cnzi;j++) {
3073985e5eaSKris Buschelman         denserow[sparserow[j]] = 0;
3083985e5eaSKris Buschelman       }
3093985e5eaSKris Buschelman       /* Aside: Perhaps we should save the pta info for the numerical factorization. */
3103985e5eaSKris Buschelman       /*        For now, we will recompute what is needed. */
3113985e5eaSKris Buschelman       ci[i+1+dof] = ci[i+dof] + cnzi;
3123985e5eaSKris Buschelman     }
3133985e5eaSKris Buschelman   }
3143985e5eaSKris Buschelman   /* nnz is now stored in ci[ptm], column indices are in the list of free space */
3153985e5eaSKris Buschelman   /* Allocate space for cj, initialize cj, and */
3163985e5eaSKris Buschelman   /* destroy list of free space and other temporary array(s) */
3173985e5eaSKris Buschelman   ierr = PetscMalloc((ci[pn]+1)*sizeof(int),&cj);CHKERRQ(ierr);
3183985e5eaSKris Buschelman   ierr = MakeSpaceContiguous(&free_space,cj);CHKERRQ(ierr);
3193985e5eaSKris Buschelman   ierr = PetscFree(ptadenserow);CHKERRQ(ierr);
3203985e5eaSKris Buschelman 
3213985e5eaSKris Buschelman   /* Allocate space for ca */
3223985e5eaSKris Buschelman   ierr = PetscMalloc((ci[pn]+1)*sizeof(MatScalar),&ca);CHKERRQ(ierr);
3233985e5eaSKris Buschelman   ierr = PetscMemzero(ca,(ci[pn]+1)*sizeof(MatScalar));CHKERRQ(ierr);
3243985e5eaSKris Buschelman 
3253985e5eaSKris Buschelman   /* put together the new matrix */
3263985e5eaSKris Buschelman   ierr = MatCreateSeqAIJWithArrays(A->comm,pn,pn,ci,cj,ca,C);CHKERRQ(ierr);
3273985e5eaSKris Buschelman 
3283985e5eaSKris Buschelman   /* MatCreateSeqAIJWithArrays flags matrix so PETSc doesn't free the user's arrays. */
3293985e5eaSKris Buschelman   /* Since these are PETSc arrays, change flags to free them as necessary. */
3303985e5eaSKris Buschelman   c = (Mat_SeqAIJ *)((*C)->data);
3313985e5eaSKris Buschelman   c->freedata = PETSC_TRUE;
3323985e5eaSKris Buschelman   c->nonew    = 0;
3333985e5eaSKris Buschelman 
3343985e5eaSKris Buschelman   /* Clean up. */
3353985e5eaSKris Buschelman   ierr = MatRestoreSymbolicTranspose_SeqAIJ(P,&pti,&ptj);CHKERRQ(ierr);
3363985e5eaSKris Buschelman 
3373985e5eaSKris Buschelman   ierr = PetscLogEventEnd(MATSeqAIJ_PtAPSymbolic,A,PP,0,0);CHKERRQ(ierr);
3383985e5eaSKris Buschelman   PetscFunctionReturn(0);
3393985e5eaSKris Buschelman }
3403985e5eaSKris Buschelman EXTERN_C_END
3413985e5eaSKris Buschelman 
342eb9c0419SKris Buschelman /*
343eb9c0419SKris Buschelman      MatSeqAIJPtAPNumeric - Computes the SeqAIJ matrix product, C,
344eb9c0419SKris Buschelman            of SeqAIJ matrix A and matrix P, according to:
345eb9c0419SKris Buschelman                  C = P^T * A * P
346eb9c0419SKris Buschelman      Note: C must have been created by calling MatSeqAIJApplyPtAPSymbolic.
347eb9c0419SKris Buschelman */
348eb9c0419SKris Buschelman #undef __FUNCT__
349eb9c0419SKris Buschelman #define __FUNCT__ "MatSeqAIJPtAPNumeric"
350eb9c0419SKris Buschelman int MatSeqAIJPtAPNumeric(Mat A,Mat P,Mat C) {
351eb9c0419SKris Buschelman   int ierr;
352eb9c0419SKris Buschelman   char funct[80];
353eb9c0419SKris Buschelman 
354eb9c0419SKris Buschelman   PetscFunctionBegin;
355eb9c0419SKris Buschelman 
356eb9c0419SKris Buschelman   PetscValidHeaderSpecific(A,MAT_COOKIE);
357eb9c0419SKris Buschelman   PetscValidType(A);
358eb9c0419SKris Buschelman   MatPreallocated(A);
359eb9c0419SKris Buschelman   if (!A->assembled) SETERRQ(PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
360eb9c0419SKris Buschelman   if (A->factor) SETERRQ(PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
361eb9c0419SKris Buschelman 
362eb9c0419SKris Buschelman   PetscValidHeaderSpecific(P,MAT_COOKIE);
363eb9c0419SKris Buschelman   PetscValidType(P);
364eb9c0419SKris Buschelman   MatPreallocated(P);
365eb9c0419SKris Buschelman   if (!P->assembled) SETERRQ(PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
366eb9c0419SKris Buschelman   if (P->factor) SETERRQ(PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
367eb9c0419SKris Buschelman 
368eb9c0419SKris Buschelman   PetscValidHeaderSpecific(C,MAT_COOKIE);
369eb9c0419SKris Buschelman   PetscValidType(C);
370eb9c0419SKris Buschelman   MatPreallocated(C);
371eb9c0419SKris Buschelman   if (!C->assembled) SETERRQ(PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
372eb9c0419SKris Buschelman   if (C->factor) SETERRQ(PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
373eb9c0419SKris Buschelman 
374eb9c0419SKris Buschelman   if (P->N!=C->M) SETERRQ2(PETSC_ERR_ARG_SIZ,"Matrix dimensions are incompatible, %d != %d",P->N,C->M);
375eb9c0419SKris Buschelman   if (P->M!=A->N) SETERRQ2(PETSC_ERR_ARG_SIZ,"Matrix dimensions are incompatible, %d != %d",P->M,A->N);
376eb9c0419SKris Buschelman   if (A->M!=A->N) SETERRQ2(PETSC_ERR_ARG_SIZ,"Matrix 'A' must be square, %d != %d",A->M,A->N);
377eb9c0419SKris Buschelman   if (P->N!=C->N) SETERRQ2(PETSC_ERR_ARG_SIZ,"Matrix dimensions are incompatible, %d != %d",P->N,C->N);
378eb9c0419SKris Buschelman 
379eb9c0419SKris Buschelman   /* Query A for ApplyPtAP implementation based on types of P */
380eb9c0419SKris Buschelman   ierr = PetscStrcpy(funct,"MatApplyPtAPNumeric_seqaij_");CHKERRQ(ierr);
381eb9c0419SKris Buschelman   ierr = PetscStrcat(funct,P->type_name);CHKERRQ(ierr);
382eb9c0419SKris Buschelman   ierr = PetscTryMethod(A,funct,(Mat,Mat,Mat),(A,P,C));CHKERRQ(ierr);
383eb9c0419SKris Buschelman 
384eb9c0419SKris Buschelman   PetscFunctionReturn(0);
385eb9c0419SKris Buschelman }
386eb9c0419SKris Buschelman 
387eb9c0419SKris Buschelman EXTERN_C_BEGIN
388eb9c0419SKris Buschelman #undef __FUNCT__
389eb9c0419SKris Buschelman #define __FUNCT__ "MatApplyPtAPNumeric_SeqAIJ_SeqAIJ"
390eb9c0419SKris Buschelman int MatApplyPtAPNumeric_SeqAIJ_SeqAIJ(Mat A,Mat P,Mat C) {
391eb9c0419SKris Buschelman   int        ierr,flops=0;
392eb9c0419SKris Buschelman   Mat_SeqAIJ *a  = (Mat_SeqAIJ *) A->data;
393eb9c0419SKris Buschelman   Mat_SeqAIJ *p  = (Mat_SeqAIJ *) P->data;
394eb9c0419SKris Buschelman   Mat_SeqAIJ *c  = (Mat_SeqAIJ *) C->data;
395eb9c0419SKris Buschelman   int        *ai=a->i,*aj=a->j,*apj,*apjdense,*pi=p->i,*pj=p->j,*pJ=p->j,*pjj;
396eb9c0419SKris Buschelman   int        *ci=c->i,*cj=c->j,*cjj;
397eb9c0419SKris Buschelman   int        am=A->M,cn=C->N,cm=C->M;
398eb9c0419SKris Buschelman   int        i,j,k,anzi,pnzi,apnzj,nextap,pnzj,prow,crow;
399eb9c0419SKris Buschelman   MatScalar  *aa=a->a,*apa,*pa=p->a,*pA=p->a,*paj,*ca=c->a,*caj;
400eb9c0419SKris Buschelman 
401eb9c0419SKris Buschelman   PetscFunctionBegin;
402eb9c0419SKris Buschelman   ierr = PetscLogEventBegin(MATSeqAIJ_PtAPNumeric,A,P,C,0);CHKERRQ(ierr);
403eb9c0419SKris Buschelman 
404eb9c0419SKris Buschelman   /* Allocate temporary array for storage of one row of A*P */
405eb9c0419SKris Buschelman   ierr = PetscMalloc(cn*(sizeof(MatScalar)+2*sizeof(int)),&apa);CHKERRQ(ierr);
406eb9c0419SKris Buschelman   ierr = PetscMemzero(apa,cn*(sizeof(MatScalar)+2*sizeof(int)));CHKERRQ(ierr);
407eb9c0419SKris Buschelman 
408eb9c0419SKris Buschelman   apj      = (int *)(apa + cn);
409eb9c0419SKris Buschelman   apjdense = apj + cn;
410eb9c0419SKris Buschelman 
411eb9c0419SKris Buschelman   /* Clear old values in C */
412eb9c0419SKris Buschelman   ierr = PetscMemzero(ca,ci[cm]*sizeof(MatScalar));CHKERRQ(ierr);
413eb9c0419SKris Buschelman 
414eb9c0419SKris Buschelman   for (i=0;i<am;i++) {
415eb9c0419SKris Buschelman     /* Form sparse row of A*P */
416eb9c0419SKris Buschelman     anzi  = ai[i+1] - ai[i];
417eb9c0419SKris Buschelman     apnzj = 0;
418eb9c0419SKris Buschelman     for (j=0;j<anzi;j++) {
419eb9c0419SKris Buschelman       prow = *aj++;
420eb9c0419SKris Buschelman       pnzj = pi[prow+1] - pi[prow];
421eb9c0419SKris Buschelman       pjj  = pj + pi[prow];
422eb9c0419SKris Buschelman       paj  = pa + pi[prow];
423eb9c0419SKris Buschelman       for (k=0;k<pnzj;k++) {
424eb9c0419SKris Buschelman         if (!apjdense[pjj[k]]) {
425eb9c0419SKris Buschelman           apjdense[pjj[k]] = -1;
426eb9c0419SKris Buschelman           apj[apnzj++]     = pjj[k];
427eb9c0419SKris Buschelman         }
428eb9c0419SKris Buschelman         apa[pjj[k]] += (*aa)*paj[k];
429eb9c0419SKris Buschelman       }
430eb9c0419SKris Buschelman       flops += 2*pnzj;
431eb9c0419SKris Buschelman       aa++;
432eb9c0419SKris Buschelman     }
433eb9c0419SKris Buschelman 
434eb9c0419SKris Buschelman     /* Sort the j index array for quick sparse axpy. */
435eb9c0419SKris Buschelman     ierr = PetscSortInt(apnzj,apj);CHKERRQ(ierr);
436eb9c0419SKris Buschelman 
437eb9c0419SKris Buschelman     /* Compute P^T*A*P using outer product (P^T)[:,j]*(A*P)[j,:]. */
438eb9c0419SKris Buschelman     pnzi = pi[i+1] - pi[i];
439eb9c0419SKris Buschelman     for (j=0;j<pnzi;j++) {
440eb9c0419SKris Buschelman       nextap = 0;
441eb9c0419SKris Buschelman       crow   = *pJ++;
442eb9c0419SKris Buschelman       cjj    = cj + ci[crow];
443eb9c0419SKris Buschelman       caj    = ca + ci[crow];
444eb9c0419SKris Buschelman       /* Perform sparse axpy operation.  Note cjj includes apj. */
445eb9c0419SKris Buschelman       for (k=0;nextap<apnzj;k++) {
446eb9c0419SKris Buschelman         if (cjj[k]==apj[nextap]) {
447eb9c0419SKris Buschelman           caj[k] += (*pA)*apa[apj[nextap++]];
448eb9c0419SKris Buschelman         }
449eb9c0419SKris Buschelman       }
450eb9c0419SKris Buschelman       flops += 2*apnzj;
451eb9c0419SKris Buschelman       pA++;
452eb9c0419SKris Buschelman     }
453eb9c0419SKris Buschelman 
454eb9c0419SKris Buschelman     /* Zero the current row info for A*P */
455eb9c0419SKris Buschelman     for (j=0;j<apnzj;j++) {
456eb9c0419SKris Buschelman       apa[apj[j]]      = 0.;
457eb9c0419SKris Buschelman       apjdense[apj[j]] = 0;
458eb9c0419SKris Buschelman     }
459eb9c0419SKris Buschelman   }
460eb9c0419SKris Buschelman 
461eb9c0419SKris Buschelman   /* Assemble the final matrix and clean up */
462eb9c0419SKris Buschelman   ierr = MatAssemblyBegin(C,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
463eb9c0419SKris Buschelman   ierr = MatAssemblyEnd(C,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
464eb9c0419SKris Buschelman   ierr = PetscFree(apa);CHKERRQ(ierr);
465eb9c0419SKris Buschelman   ierr = PetscLogFlops(flops);CHKERRQ(ierr);
466eb9c0419SKris Buschelman   ierr = PetscLogEventEnd(MATSeqAIJ_PtAPNumeric,A,P,C,0);CHKERRQ(ierr);
467eb9c0419SKris Buschelman 
468eb9c0419SKris Buschelman   PetscFunctionReturn(0);
469eb9c0419SKris Buschelman }
470eb9c0419SKris Buschelman EXTERN_C_END
471eb9c0419SKris Buschelman 
472eb9c0419SKris Buschelman #undef __FUNCT__
473eb9c0419SKris Buschelman #define __FUNCT__ "RegisterApplyPtAPRoutines_Private"
474eb9c0419SKris Buschelman int RegisterApplyPtAPRoutines_Private(Mat A) {
475eb9c0419SKris Buschelman   int ierr;
476eb9c0419SKris Buschelman 
477eb9c0419SKris Buschelman   PetscFunctionBegin;
478eb9c0419SKris Buschelman 
479eb9c0419SKris Buschelman   if (!MATSeqAIJ_PtAP) {
480eb9c0419SKris Buschelman     ierr = PetscLogEventRegister(&MATSeqAIJ_PtAP,"MatSeqAIJApplyPtAP",MAT_COOKIE);CHKERRQ(ierr);
481eb9c0419SKris Buschelman   }
482eb9c0419SKris Buschelman 
483eb9c0419SKris Buschelman   if (!MATSeqAIJ_PtAPSymbolic) {
484eb9c0419SKris Buschelman     ierr = PetscLogEventRegister(&MATSeqAIJ_PtAPSymbolic,"MatSeqAIJApplyPtAPSymbolic",MAT_COOKIE);CHKERRQ(ierr);
485eb9c0419SKris Buschelman   }
486eb9c0419SKris Buschelman   ierr = PetscObjectComposeFunctionDynamic((PetscObject)A,"MatApplyPtAPSymbolic_seqaij_seqaij",
487eb9c0419SKris Buschelman                                            "MatApplyPtAPSymbolic_SeqAIJ_SeqAIJ",
488eb9c0419SKris Buschelman                                            MatApplyPtAPSymbolic_SeqAIJ_SeqAIJ);CHKERRQ(ierr);
489eb9c0419SKris Buschelman 
490eb9c0419SKris Buschelman   if (!MATSeqAIJ_PtAPNumeric) {
491eb9c0419SKris Buschelman     ierr = PetscLogEventRegister(&MATSeqAIJ_PtAPNumeric,"MatSeqAIJApplyPtAPNumeric",MAT_COOKIE);CHKERRQ(ierr);
492eb9c0419SKris Buschelman   }
493eb9c0419SKris Buschelman   ierr = PetscObjectComposeFunctionDynamic((PetscObject)A,"MatApplyPtAPNumeric_seqaij_seqaij",
494eb9c0419SKris Buschelman                                            "MatApplyPtAPNumeric_SeqAIJ_SeqAIJ",
495eb9c0419SKris Buschelman                                            MatApplyPtAPNumeric_SeqAIJ_SeqAIJ);CHKERRQ(ierr);
4965c66b693SKris Buschelman   ierr = RegisterMatMatMultRoutines_Private(A);CHKERRQ(ierr);
497eb9c0419SKris Buschelman   PetscFunctionReturn(0);
498eb9c0419SKris Buschelman }
499