xref: /petsc/src/ksp/pc/impls/bddc/bddcgraph.c (revision 674ae81933ad1188dbe7c87a2be01dd8bb4076e0)
1*674ae819SStefano Zampini #include <petsc-private/petscimpl.h>
2*674ae819SStefano Zampini #include "bddcprivate.h"
3*674ae819SStefano Zampini #include "bddcstructs.h"
4*674ae819SStefano Zampini 
5*674ae819SStefano Zampini #define NEUMANN_MARK -1
6*674ae819SStefano Zampini #define DIRICHLET_MARK -2
7*674ae819SStefano Zampini #define SPECIAL_MARK -3
8*674ae819SStefano Zampini 
9*674ae819SStefano Zampini #undef __FUNCT__
10*674ae819SStefano Zampini #define __FUNCT__ "PCBDDCGraphASCIIView"
11*674ae819SStefano Zampini PetscErrorCode PCBDDCGraphASCIIView(PCBDDCGraph graph,PetscViewer viewer)
12*674ae819SStefano Zampini {
13*674ae819SStefano Zampini   PetscInt       i,j;
14*674ae819SStefano Zampini   PetscErrorCode ierr;
15*674ae819SStefano Zampini 
16*674ae819SStefano Zampini   PetscFunctionBegin;
17*674ae819SStefano Zampini   ierr = PetscViewerASCIISynchronizedAllow(viewer,PETSC_TRUE);CHKERRQ(ierr);
18*674ae819SStefano Zampini   ierr = PetscViewerASCIIPrintf(viewer,"--------------------------------------------------\n");CHKERRQ(ierr);
19*674ae819SStefano Zampini   ierr = PetscViewerFlush(viewer);CHKERRQ(ierr);
20*674ae819SStefano Zampini   ierr = PetscViewerASCIISynchronizedPrintf(viewer,"Local BDDC graph for subdomain %04d\n",PetscGlobalRank);CHKERRQ(ierr);
21*674ae819SStefano Zampini   ierr = PetscViewerASCIISynchronizedPrintf(viewer,"Number of vertices %d\n",graph->nvtxs);CHKERRQ(ierr);
22*674ae819SStefano Zampini   for (i=0;i<graph->nvtxs;i++) {
23*674ae819SStefano Zampini     ierr = PetscViewerASCIISynchronizedPrintf(viewer,"%d:\n",i);CHKERRQ(ierr);
24*674ae819SStefano Zampini     ierr = PetscViewerASCIISynchronizedPrintf(viewer,"   touched: %d\n",graph->touched[i]);CHKERRQ(ierr);
25*674ae819SStefano Zampini     ierr = PetscViewerASCIISynchronizedPrintf(viewer,"   which_dof: %d\n",graph->which_dof[i]);CHKERRQ(ierr);
26*674ae819SStefano Zampini     ierr = PetscViewerASCIISynchronizedPrintf(viewer,"   neighbours: %d\n",graph->count[i]);CHKERRQ(ierr);
27*674ae819SStefano Zampini     if (graph->count[i]) {
28*674ae819SStefano Zampini       ierr = PetscViewerASCIISynchronizedPrintf(viewer,"     set of neighbours:");CHKERRQ(ierr);
29*674ae819SStefano Zampini       for (j=0;j<graph->count[i];j++) {
30*674ae819SStefano Zampini         ierr = PetscViewerASCIISynchronizedPrintf(viewer," %d",graph->neighbours_set[i][j]);CHKERRQ(ierr);
31*674ae819SStefano Zampini       }
32*674ae819SStefano Zampini       ierr = PetscViewerASCIISynchronizedPrintf(viewer,"\n");CHKERRQ(ierr);
33*674ae819SStefano Zampini     }
34*674ae819SStefano Zampini /*
35*674ae819SStefano Zampini     if (graph->xadj && graph->adjncy) {
36*674ae819SStefano Zampini       ierr = PetscViewerASCIISynchronizedPrintf(viewer,"   local adj list:");CHKERRQ(ierr);
37*674ae819SStefano Zampini       for (j=graph->xadj[i];j<graph->xadj[i+1];j++) {
38*674ae819SStefano Zampini         ierr = PetscViewerASCIISynchronizedPrintf(viewer," %d",graph->adjncy[j]);CHKERRQ(ierr);
39*674ae819SStefano Zampini       }
40*674ae819SStefano Zampini       ierr = PetscViewerASCIISynchronizedPrintf(viewer,"\n");CHKERRQ(ierr);
41*674ae819SStefano Zampini     }
42*674ae819SStefano Zampini */
43*674ae819SStefano Zampini     ierr = PetscViewerASCIISynchronizedPrintf(viewer,"   interface subset id: %d\n",graph->subset[i]);CHKERRQ(ierr);
44*674ae819SStefano Zampini     if (graph->subset[i] && graph->subset_ncc) {
45*674ae819SStefano Zampini       ierr = PetscViewerASCIISynchronizedPrintf(viewer,"   ncc for subset: %d\n",graph->subset_ncc[graph->subset[i]-1]);CHKERRQ(ierr);
46*674ae819SStefano Zampini     }
47*674ae819SStefano Zampini   }
48*674ae819SStefano Zampini   ierr = PetscViewerASCIISynchronizedPrintf(viewer,"Total number of connected components %d\n",graph->ncc);CHKERRQ(ierr);
49*674ae819SStefano Zampini   for (i=0;i<graph->ncc;i++) {
50*674ae819SStefano Zampini     ierr = PetscViewerASCIISynchronizedPrintf(viewer,"  %d (neighs:",i);CHKERRQ(ierr);
51*674ae819SStefano Zampini     PetscInt node_num=graph->queue[graph->cptr[i]];
52*674ae819SStefano Zampini     for (j=0;j<graph->count[node_num];j++) {
53*674ae819SStefano Zampini       ierr = PetscViewerASCIISynchronizedPrintf(viewer," %d",graph->neighbours_set[node_num][j]);CHKERRQ(ierr);
54*674ae819SStefano Zampini     }
55*674ae819SStefano Zampini     ierr = PetscViewerASCIISynchronizedPrintf(viewer,"):");CHKERRQ(ierr);
56*674ae819SStefano Zampini     for (j=graph->cptr[i];j<graph->cptr[i+1];j++) {
57*674ae819SStefano Zampini       ierr = PetscViewerASCIISynchronizedPrintf(viewer," %d",graph->queue[j]);CHKERRQ(ierr);
58*674ae819SStefano Zampini     }
59*674ae819SStefano Zampini     ierr = PetscViewerASCIISynchronizedPrintf(viewer,"\n");CHKERRQ(ierr);
60*674ae819SStefano Zampini   }
61*674ae819SStefano Zampini   if (graph->custom_minimal_size > 1) {
62*674ae819SStefano Zampini     ierr = PetscViewerASCIISynchronizedPrintf(viewer,"Custom minimal size %d\n",graph->custom_minimal_size);CHKERRQ(ierr);
63*674ae819SStefano Zampini   }
64*674ae819SStefano Zampini   ierr = PetscViewerFlush(viewer);CHKERRQ(ierr);
65*674ae819SStefano Zampini   PetscFunctionReturn(0);
66*674ae819SStefano Zampini }
67*674ae819SStefano Zampini 
68*674ae819SStefano Zampini #undef __FUNCT__
69*674ae819SStefano Zampini #define __FUNCT__ "PCBDDCGraphGetCandidatesIS"
70*674ae819SStefano Zampini PetscErrorCode PCBDDCGraphGetCandidatesIS(PCBDDCGraph graph, PetscBool use_faces, PetscBool use_edges, PetscBool use_vertices, PetscInt *n_faces, IS *FacesIS[], PetscInt *n_edges, IS *EdgesIS[], IS *VerticesIS)
71*674ae819SStefano Zampini {
72*674ae819SStefano Zampini   IS             *ISForFaces,*ISForEdges,ISForVertices;
73*674ae819SStefano Zampini   PetscInt       i,j,nfc,nec,nvc,*idx;
74*674ae819SStefano Zampini   PetscBool      twodim_flag;
75*674ae819SStefano Zampini   PetscErrorCode ierr;
76*674ae819SStefano Zampini 
77*674ae819SStefano Zampini   PetscFunctionBegin;
78*674ae819SStefano Zampini   /* loop on ccs to evalute number of faces, edges and vertices */
79*674ae819SStefano Zampini   nfc = 0;
80*674ae819SStefano Zampini   nec = 0;
81*674ae819SStefano Zampini   nvc = 0;
82*674ae819SStefano Zampini   twodim_flag = PETSC_FALSE;
83*674ae819SStefano Zampini   for (i=0;i<graph->ncc;i++) {
84*674ae819SStefano Zampini     if (graph->cptr[i+1]-graph->cptr[i] > graph->custom_minimal_size) {
85*674ae819SStefano Zampini       if (graph->count[graph->queue[graph->cptr[i]]] == 1) {
86*674ae819SStefano Zampini         nfc++;
87*674ae819SStefano Zampini       } else { /* note that nec will be zero in 2d */
88*674ae819SStefano Zampini         nec++;
89*674ae819SStefano Zampini       }
90*674ae819SStefano Zampini     } else {
91*674ae819SStefano Zampini       nvc += graph->cptr[i+1]-graph->cptr[i];
92*674ae819SStefano Zampini     }
93*674ae819SStefano Zampini   }
94*674ae819SStefano Zampini   if (!nec) { /* we are in a 2d case -> no faces, only edges */
95*674ae819SStefano Zampini     nec = nfc;
96*674ae819SStefano Zampini     nfc = 0;
97*674ae819SStefano Zampini     twodim_flag = PETSC_TRUE;
98*674ae819SStefano Zampini   }
99*674ae819SStefano Zampini   /* allocate IS arrays for faces, edges. Vertices need a single index set. */
100*674ae819SStefano Zampini   ISForFaces = 0;
101*674ae819SStefano Zampini   ISForEdges = 0;
102*674ae819SStefano Zampini   ISForVertices = 0;
103*674ae819SStefano Zampini   if (use_faces) {
104*674ae819SStefano Zampini     ierr = PetscMalloc(nfc*sizeof(IS),&ISForFaces);CHKERRQ(ierr);
105*674ae819SStefano Zampini   }
106*674ae819SStefano Zampini   if (use_edges) {
107*674ae819SStefano Zampini     ierr = PetscMalloc(nec*sizeof(IS),&ISForEdges);CHKERRQ(ierr);
108*674ae819SStefano Zampini   }
109*674ae819SStefano Zampini   if (use_vertices) {
110*674ae819SStefano Zampini     ierr = PetscMalloc(nvc*sizeof(PetscInt),&idx);CHKERRQ(ierr);
111*674ae819SStefano Zampini   }
112*674ae819SStefano Zampini   /* loop on ccs to compute index sets for faces and edges */
113*674ae819SStefano Zampini   nfc = 0;
114*674ae819SStefano Zampini   nec = 0;
115*674ae819SStefano Zampini   nvc = 0;
116*674ae819SStefano Zampini   for (i=0;i<graph->ncc;i++) {
117*674ae819SStefano Zampini     if (graph->cptr[i+1]-graph->cptr[i] > graph->custom_minimal_size) {
118*674ae819SStefano Zampini       if (graph->count[graph->queue[graph->cptr[i]]] == 1) {
119*674ae819SStefano Zampini         if (twodim_flag) {
120*674ae819SStefano Zampini           if (use_edges) {
121*674ae819SStefano Zampini             ierr = ISCreateGeneral(PETSC_COMM_SELF,graph->cptr[i+1]-graph->cptr[i],&graph->queue[graph->cptr[i]],PETSC_COPY_VALUES,&ISForEdges[nec]);CHKERRQ(ierr);
122*674ae819SStefano Zampini             nec++;
123*674ae819SStefano Zampini           }
124*674ae819SStefano Zampini         } else {
125*674ae819SStefano Zampini           if (use_faces) {
126*674ae819SStefano Zampini             ierr = ISCreateGeneral(PETSC_COMM_SELF,graph->cptr[i+1]-graph->cptr[i],&graph->queue[graph->cptr[i]],PETSC_COPY_VALUES,&ISForFaces[nfc]);CHKERRQ(ierr);
127*674ae819SStefano Zampini             nfc++;
128*674ae819SStefano Zampini           }
129*674ae819SStefano Zampini         }
130*674ae819SStefano Zampini       } else {
131*674ae819SStefano Zampini         if (use_edges) {
132*674ae819SStefano Zampini           ierr = ISCreateGeneral(PETSC_COMM_SELF,graph->cptr[i+1]-graph->cptr[i],&graph->queue[graph->cptr[i]],PETSC_COPY_VALUES,&ISForEdges[nec]);CHKERRQ(ierr);
133*674ae819SStefano Zampini           nec++;
134*674ae819SStefano Zampini         }
135*674ae819SStefano Zampini       }
136*674ae819SStefano Zampini     }
137*674ae819SStefano Zampini   }
138*674ae819SStefano Zampini   /* index set for vertices */
139*674ae819SStefano Zampini   if (use_vertices) {
140*674ae819SStefano Zampini     for (i=0;i<graph->ncc;i++) {
141*674ae819SStefano Zampini       if (graph->cptr[i+1]-graph->cptr[i] <= graph->custom_minimal_size) {
142*674ae819SStefano Zampini         for (j=graph->cptr[i];j<graph->cptr[i+1];j++) {
143*674ae819SStefano Zampini           idx[nvc]=graph->queue[j];
144*674ae819SStefano Zampini           nvc++;
145*674ae819SStefano Zampini         }
146*674ae819SStefano Zampini       }
147*674ae819SStefano Zampini     }
148*674ae819SStefano Zampini     /* sort vertex set (by local ordering) */
149*674ae819SStefano Zampini     ierr = PetscSortInt(nvc,idx);CHKERRQ(ierr);
150*674ae819SStefano Zampini     ierr = ISCreateGeneral(PETSC_COMM_SELF,nvc,idx,PETSC_OWN_POINTER,&ISForVertices);CHKERRQ(ierr);
151*674ae819SStefano Zampini   }
152*674ae819SStefano Zampini   /* get back info */
153*674ae819SStefano Zampini   *n_faces = nfc;
154*674ae819SStefano Zampini   *FacesIS = ISForFaces;
155*674ae819SStefano Zampini   *n_edges = nec;
156*674ae819SStefano Zampini   *EdgesIS = ISForEdges;
157*674ae819SStefano Zampini   *VerticesIS = ISForVertices;
158*674ae819SStefano Zampini   PetscFunctionReturn(0);
159*674ae819SStefano Zampini }
160*674ae819SStefano Zampini 
161*674ae819SStefano Zampini #undef __FUNCT__
162*674ae819SStefano Zampini #define __FUNCT__ "PCBDDCGraphComputeConnectedComponents"
163*674ae819SStefano Zampini PetscErrorCode PCBDDCGraphComputeConnectedComponents(PCBDDCGraph graph)
164*674ae819SStefano Zampini {
165*674ae819SStefano Zampini   PetscInt    adapt_interface,adapt_interface_reduced;
166*674ae819SStefano Zampini   MPI_Comm    interface_comm;
167*674ae819SStefano Zampini   MPI_Request *send_requests;
168*674ae819SStefano Zampini   MPI_Request *recv_requests;
169*674ae819SStefano Zampini   PetscInt    *aux_new_xadj,*new_xadj,*new_adjncy,**temp_buffer;
170*674ae819SStefano Zampini   PetscInt    i,j,k,s,sum_requests,buffer_size,size_of_recv,temp_buffer_size;
171*674ae819SStefano Zampini   PetscMPIInt rank,neigh,tag,mpi_buffer_size;
172*674ae819SStefano Zampini   PetscInt    *cum_recv_counts,*subset_to_nodes_indices,*recv_buffer_subset,*nodes_to_temp_buffer_indices;
173*674ae819SStefano Zampini   PetscInt    *send_buffer,*recv_buffer,*queue_in_global_numbering,*sizes_of_sends,*add_to_subset;
174*674ae819SStefano Zampini   PetscInt    start_of_recv,start_of_send,size_of_send,global_subset_counter,ins_val;
175*674ae819SStefano Zampini   PetscBool   *subset_cc_adapt,same_set;
176*674ae819SStefano Zampini   PetscErrorCode ierr;
177*674ae819SStefano Zampini 
178*674ae819SStefano Zampini   PetscFunctionBegin;
179*674ae819SStefano Zampini   /* compute connected components locally */
180*674ae819SStefano Zampini   ierr = PetscObjectGetComm((PetscObject)(graph->l2gmap),&interface_comm);CHKERRQ(ierr);
181*674ae819SStefano Zampini   ierr = PCBDDCGraphComputeConnectedComponentsLocal(graph);CHKERRQ(ierr);
182*674ae819SStefano Zampini   /* check consistency of connected components among neighbouring subdomains -> it adapt them in case it is needed */
183*674ae819SStefano Zampini   adapt_interface = 0;
184*674ae819SStefano Zampini   adapt_interface_reduced = 0;
185*674ae819SStefano Zampini   for (i=0;i<graph->n_subsets;i++) {
186*674ae819SStefano Zampini     /* We are not sure that on a given subset of the local interface,
187*674ae819SStefano Zampini        with two connected components, the latters be the same among sharing subdomains */
188*674ae819SStefano Zampini     if (graph->subset_ncc[i] > 1) {
189*674ae819SStefano Zampini       adapt_interface=1;
190*674ae819SStefano Zampini       break;
191*674ae819SStefano Zampini     }
192*674ae819SStefano Zampini   }
193*674ae819SStefano Zampini   ierr = MPI_Allreduce(&adapt_interface,&adapt_interface_reduced,1,MPIU_INT,MPI_LOR,interface_comm);CHKERRQ(ierr);
194*674ae819SStefano Zampini 
195*674ae819SStefano Zampini   if (graph->n_subsets && adapt_interface_reduced) {
196*674ae819SStefano Zampini     /* Retrict adjacency graph using information from previously computed connected components */
197*674ae819SStefano Zampini     ierr = PetscMalloc(graph->nvtxs*sizeof(PetscInt),&aux_new_xadj);CHKERRQ(ierr);
198*674ae819SStefano Zampini     for (i=0;i<graph->nvtxs;i++) {
199*674ae819SStefano Zampini       aux_new_xadj[i]=1;
200*674ae819SStefano Zampini     }
201*674ae819SStefano Zampini     for (i=0;i<graph->ncc;i++) {
202*674ae819SStefano Zampini       k = graph->cptr[i+1]-graph->cptr[i];
203*674ae819SStefano Zampini       for (j=0;j<k;j++) {
204*674ae819SStefano Zampini         aux_new_xadj[graph->queue[graph->cptr[i]+j]]=k;
205*674ae819SStefano Zampini       }
206*674ae819SStefano Zampini     }
207*674ae819SStefano Zampini     j = 0;
208*674ae819SStefano Zampini     for (i=0;i<graph->nvtxs;i++) {
209*674ae819SStefano Zampini       j += aux_new_xadj[i];
210*674ae819SStefano Zampini     }
211*674ae819SStefano Zampini     ierr = PetscMalloc((graph->nvtxs+1)*sizeof(PetscInt),&new_xadj);CHKERRQ(ierr);
212*674ae819SStefano Zampini     ierr = PetscMalloc(j*sizeof(PetscInt),&new_adjncy);CHKERRQ(ierr);
213*674ae819SStefano Zampini     new_xadj[0]=0;
214*674ae819SStefano Zampini     for (i=0;i<graph->nvtxs;i++) {
215*674ae819SStefano Zampini       new_xadj[i+1]=new_xadj[i]+aux_new_xadj[i];
216*674ae819SStefano Zampini       if (aux_new_xadj[i]==1) {
217*674ae819SStefano Zampini         new_adjncy[new_xadj[i]]=i;
218*674ae819SStefano Zampini       }
219*674ae819SStefano Zampini     }
220*674ae819SStefano Zampini     ierr = PetscFree(aux_new_xadj);CHKERRQ(ierr);
221*674ae819SStefano Zampini     for (i=0;i<graph->ncc;i++) {
222*674ae819SStefano Zampini       k = graph->cptr[i+1]-graph->cptr[i];
223*674ae819SStefano Zampini       for (j=0;j<k;j++) {
224*674ae819SStefano Zampini         ierr = PetscMemcpy(&new_adjncy[new_xadj[graph->queue[graph->cptr[i]+j]]],&graph->queue[graph->cptr[i]],k*sizeof(PetscInt));CHKERRQ(ierr);
225*674ae819SStefano Zampini       }
226*674ae819SStefano Zampini     }
227*674ae819SStefano Zampini     /* set new CSR into graph */
228*674ae819SStefano Zampini     ierr = PetscFree(graph->xadj);CHKERRQ(ierr);
229*674ae819SStefano Zampini     ierr = PetscFree(graph->adjncy);CHKERRQ(ierr);
230*674ae819SStefano Zampini     graph->xadj = new_xadj;
231*674ae819SStefano Zampini     graph->adjncy = new_adjncy;
232*674ae819SStefano Zampini     /* allocate some space */
233*674ae819SStefano Zampini     ierr = MPI_Comm_rank(interface_comm,&rank);CHKERRQ(ierr);
234*674ae819SStefano Zampini     ierr = PetscMalloc((graph->n_subsets+1)*sizeof(*cum_recv_counts),&cum_recv_counts);CHKERRQ(ierr);
235*674ae819SStefano Zampini     ierr = PetscMemzero(cum_recv_counts,(graph->n_subsets+1)*sizeof(*cum_recv_counts));CHKERRQ(ierr);
236*674ae819SStefano Zampini     ierr = PetscMalloc(graph->n_subsets*sizeof(*subset_to_nodes_indices),&subset_to_nodes_indices);CHKERRQ(ierr);
237*674ae819SStefano Zampini     /* first count how many neighbours per connected component I will receive from */
238*674ae819SStefano Zampini     cum_recv_counts[0]=0;
239*674ae819SStefano Zampini     for (i=1;i<graph->n_subsets+1;i++) {
240*674ae819SStefano Zampini       j = 0;
241*674ae819SStefano Zampini       while (graph->subset[j] != i) {
242*674ae819SStefano Zampini         j++;
243*674ae819SStefano Zampini       }
244*674ae819SStefano Zampini       subset_to_nodes_indices[i-1]=j;
245*674ae819SStefano Zampini       /* We don't want sends/recvs_to/from_self -> here I don't count myself  */
246*674ae819SStefano Zampini       cum_recv_counts[i]=cum_recv_counts[i-1]+graph->count[j];
247*674ae819SStefano Zampini     }
248*674ae819SStefano Zampini     ierr = PetscMalloc(2*cum_recv_counts[graph->n_subsets]*sizeof(*recv_buffer_subset),&recv_buffer_subset);CHKERRQ(ierr);
249*674ae819SStefano Zampini     ierr = PetscMalloc(cum_recv_counts[graph->n_subsets]*sizeof(MPI_Request),&send_requests);CHKERRQ(ierr);
250*674ae819SStefano Zampini     ierr = PetscMalloc(cum_recv_counts[graph->n_subsets]*sizeof(MPI_Request),&recv_requests);CHKERRQ(ierr);
251*674ae819SStefano Zampini     for (i=0;i<cum_recv_counts[graph->n_subsets];i++) {
252*674ae819SStefano Zampini       send_requests[i]=MPI_REQUEST_NULL;
253*674ae819SStefano Zampini       recv_requests[i]=MPI_REQUEST_NULL;
254*674ae819SStefano Zampini     }
255*674ae819SStefano Zampini     /* exchange with my neighbours the number of my connected components on the shared interface */
256*674ae819SStefano Zampini     sum_requests = 0;
257*674ae819SStefano Zampini     for (i=0;i<graph->n_subsets;i++) {
258*674ae819SStefano Zampini       j = subset_to_nodes_indices[i];
259*674ae819SStefano Zampini       for (k=0;k<graph->count[j];k++) {
260*674ae819SStefano Zampini         ierr = PetscMPIIntCast(graph->neighbours_set[j][k],&neigh);CHKERRQ(ierr);
261*674ae819SStefano Zampini         ierr = PetscMPIIntCast((PetscInt)(rank+1)*graph->count[j],&tag);CHKERRQ(ierr);
262*674ae819SStefano Zampini         ierr = MPI_Isend(&graph->subset_ncc[i],1,MPIU_INT,neigh,tag,interface_comm,&send_requests[sum_requests]);CHKERRQ(ierr);
263*674ae819SStefano Zampini         ierr = PetscMPIIntCast((PetscInt)(neigh+1)*graph->count[j],&tag);CHKERRQ(ierr);
264*674ae819SStefano Zampini         ierr = MPI_Irecv(&recv_buffer_subset[sum_requests],1,MPIU_INT,neigh,tag,interface_comm,&recv_requests[sum_requests]);CHKERRQ(ierr);
265*674ae819SStefano Zampini         sum_requests++;
266*674ae819SStefano Zampini       }
267*674ae819SStefano Zampini     }
268*674ae819SStefano Zampini     ierr = MPI_Waitall(sum_requests,recv_requests,MPI_STATUSES_IGNORE);CHKERRQ(ierr);
269*674ae819SStefano Zampini     ierr = MPI_Waitall(sum_requests,send_requests,MPI_STATUSES_IGNORE);CHKERRQ(ierr);
270*674ae819SStefano Zampini     /* determine the connected component I need to adapt */
271*674ae819SStefano Zampini     ierr = PetscMalloc(graph->n_subsets*sizeof(*subset_cc_adapt),&subset_cc_adapt);CHKERRQ(ierr);
272*674ae819SStefano Zampini     ierr = PetscMemzero(subset_cc_adapt,graph->n_subsets*sizeof(*subset_cc_adapt));CHKERRQ(ierr);
273*674ae819SStefano Zampini     for (i=0;i<graph->n_subsets;i++) {
274*674ae819SStefano Zampini       for (j=cum_recv_counts[i];j<cum_recv_counts[i+1];j++){
275*674ae819SStefano Zampini         /* The first condition is natural (someone has a different number of ccs than me), the second one is just to be safe */
276*674ae819SStefano Zampini         if ( graph->subset_ncc[i] != recv_buffer_subset[j] || graph->subset_ncc[i] > 1 ) {
277*674ae819SStefano Zampini           subset_cc_adapt[i] = PETSC_TRUE;
278*674ae819SStefano Zampini           break;
279*674ae819SStefano Zampini         }
280*674ae819SStefano Zampini       }
281*674ae819SStefano Zampini     }
282*674ae819SStefano Zampini     buffer_size = 0;
283*674ae819SStefano Zampini     for (i=0;i<graph->n_subsets;i++) {
284*674ae819SStefano Zampini       if (subset_cc_adapt[i]) {
285*674ae819SStefano Zampini         for (j=i;j<graph->ncc;j++) {
286*674ae819SStefano Zampini           if (graph->subset[graph->queue[graph->cptr[j]]] == i+1) { /* WARNING -> subset values goes from 1 to graph->n_subsets included */
287*674ae819SStefano Zampini             buffer_size += 1 + graph->cptr[j+1]-graph->cptr[j];
288*674ae819SStefano Zampini           }
289*674ae819SStefano Zampini         }
290*674ae819SStefano Zampini       }
291*674ae819SStefano Zampini     }
292*674ae819SStefano Zampini     ierr = PetscMalloc(buffer_size*sizeof(*send_buffer),&send_buffer);CHKERRQ(ierr);
293*674ae819SStefano Zampini     /* now get from neighbours their ccs (in global numbering) and adapt them (in case it is needed) */
294*674ae819SStefano Zampini     ierr = PetscMalloc(graph->nvtxs*sizeof(*queue_in_global_numbering),&queue_in_global_numbering);CHKERRQ(ierr);
295*674ae819SStefano Zampini     ierr = ISLocalToGlobalMappingApply(graph->l2gmap,graph->cptr[graph->ncc],graph->queue,queue_in_global_numbering);CHKERRQ(ierr);
296*674ae819SStefano Zampini     /* determine how much data to send (size of each queue plus the global indices) and communicate it to neighbours */
297*674ae819SStefano Zampini     ierr = PetscMalloc(graph->n_subsets*sizeof(*sizes_of_sends),&sizes_of_sends);CHKERRQ(ierr);
298*674ae819SStefano Zampini     ierr = PetscMemzero(sizes_of_sends,graph->n_subsets*sizeof(*sizes_of_sends));CHKERRQ(ierr);
299*674ae819SStefano Zampini     sum_requests = 0;
300*674ae819SStefano Zampini     start_of_send = 0;
301*674ae819SStefano Zampini     start_of_recv = cum_recv_counts[graph->n_subsets];
302*674ae819SStefano Zampini     for (i=0;i<graph->n_subsets;i++) {
303*674ae819SStefano Zampini       if (subset_cc_adapt[i]) {
304*674ae819SStefano Zampini         size_of_send = 0;
305*674ae819SStefano Zampini         for (j=i;j<graph->ncc;j++) {
306*674ae819SStefano Zampini           if (graph->subset[graph->queue[graph->cptr[j]]] == i+1) { /* WARNING -> subset values goes from 1 to graph->n_subsets included */
307*674ae819SStefano Zampini             send_buffer[start_of_send+size_of_send]=graph->cptr[j+1]-graph->cptr[j];
308*674ae819SStefano Zampini             size_of_send += 1;
309*674ae819SStefano Zampini             ierr = PetscMemcpy(&send_buffer[start_of_send+size_of_send],
310*674ae819SStefano Zampini                                &queue_in_global_numbering[graph->cptr[j]],
311*674ae819SStefano Zampini                                (graph->cptr[j+1]-graph->cptr[j])*sizeof(*send_buffer));CHKERRQ(ierr);
312*674ae819SStefano Zampini             size_of_send = size_of_send+graph->cptr[j+1]-graph->cptr[j];
313*674ae819SStefano Zampini           }
314*674ae819SStefano Zampini         }
315*674ae819SStefano Zampini         j = subset_to_nodes_indices[i];
316*674ae819SStefano Zampini         sizes_of_sends[i] = size_of_send;
317*674ae819SStefano Zampini         for (k=0;k<graph->count[j];k++) {
318*674ae819SStefano Zampini           ierr = PetscMPIIntCast(graph->neighbours_set[j][k],&neigh);CHKERRQ(ierr);
319*674ae819SStefano Zampini           ierr = PetscMPIIntCast((PetscInt)(rank+1)*graph->count[j],&tag);CHKERRQ(ierr);
320*674ae819SStefano Zampini           ierr = MPI_Isend(&sizes_of_sends[i],1,MPIU_INT,neigh,tag,interface_comm,&send_requests[sum_requests]);CHKERRQ(ierr);
321*674ae819SStefano Zampini           ierr = PetscMPIIntCast((PetscInt)(neigh+1)*graph->count[j],&tag);CHKERRQ(ierr);
322*674ae819SStefano Zampini           ierr = MPI_Irecv(&recv_buffer_subset[sum_requests+start_of_recv],1,MPIU_INT,neigh,tag,interface_comm,&recv_requests[sum_requests]);CHKERRQ(ierr);
323*674ae819SStefano Zampini           sum_requests++;
324*674ae819SStefano Zampini         }
325*674ae819SStefano Zampini         start_of_send += size_of_send;
326*674ae819SStefano Zampini       }
327*674ae819SStefano Zampini     }
328*674ae819SStefano Zampini     ierr = MPI_Waitall(sum_requests,send_requests,MPI_STATUSES_IGNORE);CHKERRQ(ierr);
329*674ae819SStefano Zampini     ierr = MPI_Waitall(sum_requests,recv_requests,MPI_STATUSES_IGNORE);CHKERRQ(ierr);
330*674ae819SStefano Zampini     buffer_size = 0;
331*674ae819SStefano Zampini     for (k=0;k<sum_requests;k++) {
332*674ae819SStefano Zampini       buffer_size += recv_buffer_subset[start_of_recv+k];
333*674ae819SStefano Zampini     }
334*674ae819SStefano Zampini     ierr = PetscMalloc(buffer_size*sizeof(*recv_buffer),&recv_buffer);CHKERRQ(ierr);
335*674ae819SStefano Zampini     /* now exchange the data */
336*674ae819SStefano Zampini     start_of_recv = 0;
337*674ae819SStefano Zampini     start_of_send = 0;
338*674ae819SStefano Zampini     sum_requests = 0;
339*674ae819SStefano Zampini     for (i=0;i<graph->n_subsets;i++) {
340*674ae819SStefano Zampini       if (subset_cc_adapt[i]) {
341*674ae819SStefano Zampini         size_of_send = sizes_of_sends[i];
342*674ae819SStefano Zampini         j = subset_to_nodes_indices[i];
343*674ae819SStefano Zampini         for (k=0;k<graph->count[j];k++) {
344*674ae819SStefano Zampini           ierr = PetscMPIIntCast(graph->neighbours_set[j][k],&neigh);CHKERRQ(ierr);
345*674ae819SStefano Zampini           ierr = PetscMPIIntCast((PetscInt)(rank+1)*graph->count[j],&tag);CHKERRQ(ierr);
346*674ae819SStefano Zampini           ierr = PetscMPIIntCast(size_of_send,&mpi_buffer_size);CHKERRQ(ierr);
347*674ae819SStefano Zampini           ierr = MPI_Isend(&send_buffer[start_of_send],mpi_buffer_size,MPIU_INT,neigh,tag,interface_comm,&send_requests[sum_requests]);CHKERRQ(ierr);
348*674ae819SStefano Zampini           size_of_recv = recv_buffer_subset[cum_recv_counts[graph->n_subsets]+sum_requests];
349*674ae819SStefano Zampini           ierr = PetscMPIIntCast((PetscInt)(neigh+1)*graph->count[j],&tag);CHKERRQ(ierr);
350*674ae819SStefano Zampini           ierr = PetscMPIIntCast(size_of_recv,&mpi_buffer_size);CHKERRQ(ierr);
351*674ae819SStefano Zampini           ierr = MPI_Irecv(&recv_buffer[start_of_recv],mpi_buffer_size,MPIU_INT,neigh,tag,interface_comm,&recv_requests[sum_requests]);CHKERRQ(ierr);
352*674ae819SStefano Zampini           start_of_recv += size_of_recv;
353*674ae819SStefano Zampini           sum_requests++;
354*674ae819SStefano Zampini         }
355*674ae819SStefano Zampini         start_of_send += size_of_send;
356*674ae819SStefano Zampini       }
357*674ae819SStefano Zampini     }
358*674ae819SStefano Zampini     ierr = MPI_Waitall(sum_requests,recv_requests,MPI_STATUSES_IGNORE);CHKERRQ(ierr);
359*674ae819SStefano Zampini     ierr = MPI_Waitall(sum_requests,send_requests,MPI_STATUSES_IGNORE);CHKERRQ(ierr);
360*674ae819SStefano Zampini     for (j=0;j<buffer_size;) {
361*674ae819SStefano Zampini        ierr = ISGlobalToLocalMappingApply(graph->l2gmap,IS_GTOLM_MASK,recv_buffer[j],&recv_buffer[j+1],&recv_buffer[j],&recv_buffer[j+1]);CHKERRQ(ierr);
362*674ae819SStefano Zampini        k = recv_buffer[j]+1;
363*674ae819SStefano Zampini        j += k;
364*674ae819SStefano Zampini     }
365*674ae819SStefano Zampini     sum_requests = cum_recv_counts[graph->n_subsets];
366*674ae819SStefano Zampini     start_of_recv = 0;
367*674ae819SStefano Zampini     ierr = PetscMalloc(graph->nvtxs*sizeof(*nodes_to_temp_buffer_indices),&nodes_to_temp_buffer_indices);CHKERRQ(ierr);
368*674ae819SStefano Zampini     global_subset_counter = 0;
369*674ae819SStefano Zampini     for (i=0;i<graph->n_subsets;i++) {
370*674ae819SStefano Zampini       if (subset_cc_adapt[i]) {
371*674ae819SStefano Zampini         temp_buffer_size = 0;
372*674ae819SStefano Zampini         /* find nodes on the shared interface we need to adapt */
373*674ae819SStefano Zampini         for (j=0;j<graph->nvtxs;j++) {
374*674ae819SStefano Zampini           if (graph->subset[j]==i+1) {
375*674ae819SStefano Zampini             nodes_to_temp_buffer_indices[j] = temp_buffer_size;
376*674ae819SStefano Zampini             temp_buffer_size++;
377*674ae819SStefano Zampini           } else {
378*674ae819SStefano Zampini             nodes_to_temp_buffer_indices[j] = -1;
379*674ae819SStefano Zampini           }
380*674ae819SStefano Zampini         }
381*674ae819SStefano Zampini         /* allocate some temporary space */
382*674ae819SStefano Zampini         ierr = PetscMalloc(temp_buffer_size*sizeof(PetscInt*),&temp_buffer);CHKERRQ(ierr);
383*674ae819SStefano Zampini         ierr = PetscMalloc(temp_buffer_size*(cum_recv_counts[i+1]-cum_recv_counts[i])*sizeof(PetscInt),&temp_buffer[0]);CHKERRQ(ierr);
384*674ae819SStefano Zampini         ierr = PetscMemzero(temp_buffer[0],temp_buffer_size*(cum_recv_counts[i+1]-cum_recv_counts[i])*sizeof(PetscInt));CHKERRQ(ierr);
385*674ae819SStefano Zampini         for (j=1;j<temp_buffer_size;j++) {
386*674ae819SStefano Zampini           temp_buffer[j] = temp_buffer[j-1]+cum_recv_counts[i+1]-cum_recv_counts[i];
387*674ae819SStefano Zampini         }
388*674ae819SStefano Zampini         /* analyze contributions from neighbouring subdomains for i-th conn comp
389*674ae819SStefano Zampini            temp buffer structure:
390*674ae819SStefano Zampini            supposing part of the interface has dimension 5 (for example with global dofs 0,1,2,3,4)
391*674ae819SStefano Zampini            3 neighs procs with structured connected components:
392*674ae819SStefano Zampini              neigh 0: [0 1 4], [2 3];  (2 connected components)
393*674ae819SStefano Zampini              neigh 1: [0 1], [2 3 4];  (2 connected components)
394*674ae819SStefano Zampini              neigh 2: [0 4], [1], [2 3]; (3 connected components)
395*674ae819SStefano Zampini            tempbuffer (row-oriented) will be filled as:
396*674ae819SStefano Zampini              [ 0, 0, 0;
397*674ae819SStefano Zampini                0, 0, 1;
398*674ae819SStefano Zampini                1, 1, 2;
399*674ae819SStefano Zampini                1, 1, 2;
400*674ae819SStefano Zampini                0, 1, 0; ];
401*674ae819SStefano Zampini            This way we can simply find intersections of ccs among neighs.
402*674ae819SStefano Zampini            For the example above, the graph->subset array will be modified to reproduce the following 4 connected components [0], [1], [2 3], [4];
403*674ae819SStefano Zampini                                                                                                                                    */
404*674ae819SStefano Zampini         for (j=0;j<cum_recv_counts[i+1]-cum_recv_counts[i];j++) {
405*674ae819SStefano Zampini           ins_val = 0;
406*674ae819SStefano Zampini           size_of_recv = recv_buffer_subset[sum_requests];  /* total size of recv from neighs */
407*674ae819SStefano Zampini           for (buffer_size=0;buffer_size<size_of_recv;) {  /* loop until all data from neighs has been taken into account */
408*674ae819SStefano Zampini             for (k=1;k<recv_buffer[buffer_size+start_of_recv]+1;k++) { /* filling properly temp_buffer using data from a single recv */
409*674ae819SStefano Zampini               temp_buffer[nodes_to_temp_buffer_indices[recv_buffer[start_of_recv+buffer_size+k]]][j] = ins_val;
410*674ae819SStefano Zampini             }
411*674ae819SStefano Zampini             buffer_size += k;
412*674ae819SStefano Zampini             ins_val++;
413*674ae819SStefano Zampini           }
414*674ae819SStefano Zampini           start_of_recv += size_of_recv;
415*674ae819SStefano Zampini           sum_requests++;
416*674ae819SStefano Zampini         }
417*674ae819SStefano Zampini         ierr = PetscMalloc(temp_buffer_size*sizeof(*add_to_subset),&add_to_subset);CHKERRQ(ierr);
418*674ae819SStefano Zampini         ierr = PetscMemzero(add_to_subset,temp_buffer_size*sizeof(*add_to_subset));CHKERRQ(ierr);
419*674ae819SStefano Zampini         for (j=0;j<temp_buffer_size;j++) {
420*674ae819SStefano Zampini           if (!add_to_subset[j]) { /* found a new cc  */
421*674ae819SStefano Zampini             global_subset_counter++;
422*674ae819SStefano Zampini             add_to_subset[j] = global_subset_counter;
423*674ae819SStefano Zampini             for (k=j+1;k<temp_buffer_size;k++) { /* check for other nodes in new cc */
424*674ae819SStefano Zampini               same_set = PETSC_TRUE;
425*674ae819SStefano Zampini               for (s=0;s<cum_recv_counts[i+1]-cum_recv_counts[i];s++) {
426*674ae819SStefano Zampini                 if (temp_buffer[j][s]!=temp_buffer[k][s]) {
427*674ae819SStefano Zampini                   same_set = PETSC_FALSE;
428*674ae819SStefano Zampini                   break;
429*674ae819SStefano Zampini                 }
430*674ae819SStefano Zampini               }
431*674ae819SStefano Zampini               if (same_set) {
432*674ae819SStefano Zampini                 add_to_subset[k] = global_subset_counter;
433*674ae819SStefano Zampini               }
434*674ae819SStefano Zampini             }
435*674ae819SStefano Zampini           }
436*674ae819SStefano Zampini         }
437*674ae819SStefano Zampini         /* insert new data in subset array */
438*674ae819SStefano Zampini         temp_buffer_size = 0;
439*674ae819SStefano Zampini         for (j=0;j<graph->nvtxs;j++) {
440*674ae819SStefano Zampini           if (graph->subset[j]==i+1) {
441*674ae819SStefano Zampini             graph->subset[j] = graph->n_subsets+add_to_subset[temp_buffer_size];
442*674ae819SStefano Zampini             temp_buffer_size++;
443*674ae819SStefano Zampini           }
444*674ae819SStefano Zampini         }
445*674ae819SStefano Zampini         ierr = PetscFree(temp_buffer[0]);CHKERRQ(ierr);
446*674ae819SStefano Zampini         ierr = PetscFree(temp_buffer);CHKERRQ(ierr);
447*674ae819SStefano Zampini         ierr = PetscFree(add_to_subset);CHKERRQ(ierr);
448*674ae819SStefano Zampini       }
449*674ae819SStefano Zampini     }
450*674ae819SStefano Zampini     ierr = PetscFree(nodes_to_temp_buffer_indices);CHKERRQ(ierr);
451*674ae819SStefano Zampini     ierr = PetscFree(sizes_of_sends);CHKERRQ(ierr);
452*674ae819SStefano Zampini     ierr = PetscFree(send_requests);CHKERRQ(ierr);
453*674ae819SStefano Zampini     ierr = PetscFree(recv_requests);CHKERRQ(ierr);
454*674ae819SStefano Zampini     ierr = PetscFree(recv_buffer);CHKERRQ(ierr);
455*674ae819SStefano Zampini     ierr = PetscFree(recv_buffer_subset);CHKERRQ(ierr);
456*674ae819SStefano Zampini     ierr = PetscFree(send_buffer);CHKERRQ(ierr);
457*674ae819SStefano Zampini     ierr = PetscFree(cum_recv_counts);CHKERRQ(ierr);
458*674ae819SStefano Zampini     ierr = PetscFree(subset_to_nodes_indices);CHKERRQ(ierr);
459*674ae819SStefano Zampini     ierr = PetscFree(subset_cc_adapt);CHKERRQ(ierr);
460*674ae819SStefano Zampini     /* We are ready to find for connected components consistent among neighbouring subdomains */
461*674ae819SStefano Zampini     if (global_subset_counter) {
462*674ae819SStefano Zampini       ierr = PetscMemzero(graph->touched,graph->nvtxs*sizeof(*graph->touched));CHKERRQ(ierr);
463*674ae819SStefano Zampini       global_subset_counter = 0;
464*674ae819SStefano Zampini       for (i=0;i<graph->nvtxs;i++) {
465*674ae819SStefano Zampini         if (graph->subset[i] && !graph->touched[i]) {
466*674ae819SStefano Zampini           global_subset_counter++;
467*674ae819SStefano Zampini           for (j=i+1;j<graph->nvtxs;j++) {
468*674ae819SStefano Zampini             if (!graph->touched[j] && graph->subset[j]==graph->subset[i]) {
469*674ae819SStefano Zampini               graph->subset[j] = global_subset_counter;
470*674ae819SStefano Zampini               graph->touched[j] = PETSC_TRUE;
471*674ae819SStefano Zampini             }
472*674ae819SStefano Zampini           }
473*674ae819SStefano Zampini           graph->subset[i] = global_subset_counter;
474*674ae819SStefano Zampini           graph->touched[i] = PETSC_TRUE;
475*674ae819SStefano Zampini         }
476*674ae819SStefano Zampini       }
477*674ae819SStefano Zampini       /* refine connected components locally */
478*674ae819SStefano Zampini       ierr = PCBDDCGraphComputeConnectedComponentsLocal(graph);CHKERRQ(ierr);
479*674ae819SStefano Zampini     }
480*674ae819SStefano Zampini     ierr = PetscFree(queue_in_global_numbering);CHKERRQ(ierr);
481*674ae819SStefano Zampini   }
482*674ae819SStefano Zampini   PetscFunctionReturn(0);
483*674ae819SStefano Zampini }
484*674ae819SStefano Zampini 
485*674ae819SStefano Zampini /* The following code has been adapted from function IsConnectedSubdomain contained
486*674ae819SStefano Zampini    in source file contig.c of METIS library (version 5.0.1)
487*674ae819SStefano Zampini    It finds connected components for each subset  */
488*674ae819SStefano Zampini #undef __FUNCT__
489*674ae819SStefano Zampini #define __FUNCT__ "PCBDDCGraphComputeConnectedComponentsLocal"
490*674ae819SStefano Zampini PetscErrorCode PCBDDCGraphComputeConnectedComponentsLocal(PCBDDCGraph graph)
491*674ae819SStefano Zampini {
492*674ae819SStefano Zampini   PetscInt       i,j,k,first,last,nleft,ncc,pid,cum_queue,n,ncc_pid;
493*674ae819SStefano Zampini   PetscInt       *queue_global;
494*674ae819SStefano Zampini   PetscErrorCode ierr;
495*674ae819SStefano Zampini 
496*674ae819SStefano Zampini   PetscFunctionBegin;
497*674ae819SStefano Zampini   /* quiet return if no csr info is available */
498*674ae819SStefano Zampini   if (!graph->xadj || !graph->adjncy) {
499*674ae819SStefano Zampini     PetscFunctionReturn(0);
500*674ae819SStefano Zampini   }
501*674ae819SStefano Zampini 
502*674ae819SStefano Zampini   /* reset any previous search of connected components */
503*674ae819SStefano Zampini   ierr = PetscMemzero(graph->touched,graph->nvtxs*sizeof(*graph->touched));CHKERRQ(ierr);
504*674ae819SStefano Zampini   graph->n_subsets = 0;
505*674ae819SStefano Zampini   for (i=0;i<graph->nvtxs;i++) {
506*674ae819SStefano Zampini     if (graph->which_dof[i] == DIRICHLET_MARK || !graph->count[i]) {
507*674ae819SStefano Zampini       graph->touched[i] = PETSC_TRUE;
508*674ae819SStefano Zampini       graph->subset[i] = 0;
509*674ae819SStefano Zampini     }
510*674ae819SStefano Zampini     graph->n_subsets = PetscMax(graph->n_subsets,graph->subset[i]);
511*674ae819SStefano Zampini   }
512*674ae819SStefano Zampini   ierr = PetscFree(graph->subset_ncc);CHKERRQ(ierr);
513*674ae819SStefano Zampini   ierr = PetscMalloc(graph->n_subsets*sizeof(*graph->subset_ncc),&graph->subset_ncc);CHKERRQ(ierr);
514*674ae819SStefano Zampini   ierr = PetscMemzero(graph->subset_ncc,graph->n_subsets*sizeof(*graph->subset_ncc));CHKERRQ(ierr);
515*674ae819SStefano Zampini   ierr = PetscMemzero(graph->cptr,(graph->nvtxs+1)*sizeof(*graph->cptr));CHKERRQ(ierr);
516*674ae819SStefano Zampini   ierr = PetscMemzero(graph->queue,graph->nvtxs*sizeof(*graph->queue));CHKERRQ(ierr);
517*674ae819SStefano Zampini 
518*674ae819SStefano Zampini   /* begin search for connected components */
519*674ae819SStefano Zampini   cum_queue = 0;
520*674ae819SStefano Zampini   ncc = 0;
521*674ae819SStefano Zampini   for (n=0;n<graph->n_subsets;n++) {
522*674ae819SStefano Zampini     pid = n+1;  /* partition labeled by 0 is discarded */
523*674ae819SStefano Zampini     nleft = 0;
524*674ae819SStefano Zampini     for (i=0;i<graph->nvtxs;i++) {
525*674ae819SStefano Zampini       if (graph->subset[i] == pid) {
526*674ae819SStefano Zampini         nleft++;
527*674ae819SStefano Zampini       }
528*674ae819SStefano Zampini     }
529*674ae819SStefano Zampini     for (i=0; i<graph->nvtxs; i++) {
530*674ae819SStefano Zampini       if (graph->subset[i] == pid) {
531*674ae819SStefano Zampini         break;
532*674ae819SStefano Zampini       }
533*674ae819SStefano Zampini     }
534*674ae819SStefano Zampini     graph->touched[i] = PETSC_TRUE;
535*674ae819SStefano Zampini     graph->queue[cum_queue] = i;
536*674ae819SStefano Zampini     first = 0;
537*674ae819SStefano Zampini     last = 1;
538*674ae819SStefano Zampini     graph->cptr[ncc] = cum_queue;
539*674ae819SStefano Zampini     ncc_pid = 0;
540*674ae819SStefano Zampini     while (first != nleft) {
541*674ae819SStefano Zampini       if (first == last) {
542*674ae819SStefano Zampini         graph->cptr[++ncc] = first+cum_queue;
543*674ae819SStefano Zampini         ncc_pid++;
544*674ae819SStefano Zampini         for (i=0; i<graph->nvtxs; i++) {
545*674ae819SStefano Zampini           if (graph->subset[i] == pid && !graph->touched[i]) {
546*674ae819SStefano Zampini             break;
547*674ae819SStefano Zampini           }
548*674ae819SStefano Zampini         }
549*674ae819SStefano Zampini         graph->queue[cum_queue+last] = i;
550*674ae819SStefano Zampini         last++;
551*674ae819SStefano Zampini         graph->touched[i] = PETSC_TRUE;
552*674ae819SStefano Zampini       }
553*674ae819SStefano Zampini       i = graph->queue[cum_queue+first];
554*674ae819SStefano Zampini       first++;
555*674ae819SStefano Zampini       for (j=graph->xadj[i];j<graph->xadj[i+1];j++) {
556*674ae819SStefano Zampini         k = graph->adjncy[j];
557*674ae819SStefano Zampini         if (graph->subset[k] == pid && !graph->touched[k]) {
558*674ae819SStefano Zampini           graph->queue[cum_queue+last] = k;
559*674ae819SStefano Zampini           last++;
560*674ae819SStefano Zampini           graph->touched[k] = PETSC_TRUE;
561*674ae819SStefano Zampini         }
562*674ae819SStefano Zampini       }
563*674ae819SStefano Zampini     }
564*674ae819SStefano Zampini     graph->cptr[++ncc] = first+cum_queue;
565*674ae819SStefano Zampini     ncc_pid++;
566*674ae819SStefano Zampini     cum_queue = graph->cptr[ncc];
567*674ae819SStefano Zampini     graph->subset_ncc[n] = ncc_pid;
568*674ae819SStefano Zampini   }
569*674ae819SStefano Zampini   graph->ncc = ncc;
570*674ae819SStefano Zampini   /* For consistency among neighbours, I need to sort (by global ordering) each connected component */
571*674ae819SStefano Zampini   ierr = PetscMalloc(graph->cptr[graph->ncc]*sizeof(*queue_global),&queue_global);CHKERRQ(ierr);
572*674ae819SStefano Zampini   ierr = ISLocalToGlobalMappingApply(graph->l2gmap,graph->cptr[graph->ncc],graph->queue,queue_global);CHKERRQ(ierr);
573*674ae819SStefano Zampini   for (i=0;i<graph->ncc;i++) {
574*674ae819SStefano Zampini     ierr = PetscSortIntWithArray(graph->cptr[i+1]-graph->cptr[i],&queue_global[graph->cptr[i]],&graph->queue[graph->cptr[i]]);CHKERRQ(ierr);
575*674ae819SStefano Zampini   }
576*674ae819SStefano Zampini   ierr = PetscFree(queue_global);CHKERRQ(ierr);
577*674ae819SStefano Zampini   PetscFunctionReturn(0);
578*674ae819SStefano Zampini }
579*674ae819SStefano Zampini 
580*674ae819SStefano Zampini #undef __FUNCT__
581*674ae819SStefano Zampini #define __FUNCT__ "PCBDDCGraphSetUp"
582*674ae819SStefano Zampini PetscErrorCode PCBDDCGraphSetUp(PCBDDCGraph graph, PetscInt custom_minimal_size, IS neumann_is, IS dirichlet_is, PetscInt n_ISForDofs, IS ISForDofs[], IS custom_primal_vertices)
583*674ae819SStefano Zampini {
584*674ae819SStefano Zampini   VecScatter     scatter_ctx;
585*674ae819SStefano Zampini   Vec            local_vec,local_vec2,global_vec;
586*674ae819SStefano Zampini   IS             to,from;
587*674ae819SStefano Zampini   MPI_Comm       comm;
588*674ae819SStefano Zampini   PetscScalar    *array,*array2;
589*674ae819SStefano Zampini   const PetscInt *is_indices;
590*674ae819SStefano Zampini   PetscInt       n_neigh,*neigh,*n_shared,**shared;
591*674ae819SStefano Zampini   PetscInt       i,j,k,s,total_counts,nodes_touched,is_size;
592*674ae819SStefano Zampini   PetscErrorCode ierr;
593*674ae819SStefano Zampini   PetscBool      same_set;
594*674ae819SStefano Zampini 
595*674ae819SStefano Zampini   PetscFunctionBegin;
596*674ae819SStefano Zampini   ierr = PetscObjectGetComm((PetscObject)(graph->l2gmap),&comm);CHKERRQ(ierr);
597*674ae819SStefano Zampini   /* custom_minimal_size */
598*674ae819SStefano Zampini   graph->custom_minimal_size = PetscMax(graph->custom_minimal_size,custom_minimal_size);
599*674ae819SStefano Zampini   /* get info l2gmap and allocate work vectors  */
600*674ae819SStefano Zampini   ierr = ISLocalToGlobalMappingGetInfo(graph->l2gmap,&n_neigh,&neigh,&n_shared,&shared);CHKERRQ(ierr);
601*674ae819SStefano Zampini   ierr = ISLocalToGlobalMappingGetIndices(graph->l2gmap,&is_indices);CHKERRQ(ierr);
602*674ae819SStefano Zampini   j = 0;
603*674ae819SStefano Zampini   for (i=0;i<graph->nvtxs;i++) {
604*674ae819SStefano Zampini     j = PetscMax(j,is_indices[i]);
605*674ae819SStefano Zampini   }
606*674ae819SStefano Zampini   ierr = MPI_Allreduce(&j,&i,1,MPIU_INT,MPI_MAX,comm);CHKERRQ(ierr);
607*674ae819SStefano Zampini   i++;
608*674ae819SStefano Zampini   ierr = VecCreate(PETSC_COMM_SELF,&local_vec);CHKERRQ(ierr);
609*674ae819SStefano Zampini   ierr = VecSetSizes(local_vec,PETSC_DECIDE,graph->nvtxs);CHKERRQ(ierr);
610*674ae819SStefano Zampini   ierr = VecSetType(local_vec,VECSTANDARD);CHKERRQ(ierr);
611*674ae819SStefano Zampini   ierr = VecDuplicate(local_vec,&local_vec2);CHKERRQ(ierr);
612*674ae819SStefano Zampini   ierr = VecCreate(comm,&global_vec);CHKERRQ(ierr);
613*674ae819SStefano Zampini   ierr = VecSetSizes(global_vec,PETSC_DECIDE,i);CHKERRQ(ierr);
614*674ae819SStefano Zampini   ierr = VecSetType(global_vec,VECSTANDARD);CHKERRQ(ierr);
615*674ae819SStefano Zampini   ierr = ISCreateStride(PETSC_COMM_SELF,graph->nvtxs,0,1,&to);CHKERRQ(ierr);
616*674ae819SStefano Zampini   ierr = ISLocalToGlobalMappingApplyIS(graph->l2gmap,to,&from);CHKERRQ(ierr);
617*674ae819SStefano Zampini   ierr = VecScatterCreate(global_vec,from,local_vec,to,&scatter_ctx);CHKERRQ(ierr);
618*674ae819SStefano Zampini   ierr = ISDestroy(&to);CHKERRQ(ierr);
619*674ae819SStefano Zampini   ierr = ISDestroy(&from);CHKERRQ(ierr);
620*674ae819SStefano Zampini   /* Count total number of neigh per node */
621*674ae819SStefano Zampini   k=0;
622*674ae819SStefano Zampini   for (i=1;i<n_neigh;i++){
623*674ae819SStefano Zampini     k += n_shared[i];
624*674ae819SStefano Zampini     for (j=0;j<n_shared[i];j++){
625*674ae819SStefano Zampini       graph->count[shared[i][j]] += 1;
626*674ae819SStefano Zampini     }
627*674ae819SStefano Zampini   }
628*674ae819SStefano Zampini   /* Allocate space for storing the set of neighbours for each node */
629*674ae819SStefano Zampini   if (graph->nvtxs) {
630*674ae819SStefano Zampini     ierr = PetscMalloc(k*sizeof(PetscInt),&graph->neighbours_set[0]);CHKERRQ(ierr);
631*674ae819SStefano Zampini   }
632*674ae819SStefano Zampini   for (i=1;i<graph->nvtxs;i++) { /* dont count myself */
633*674ae819SStefano Zampini     graph->neighbours_set[i]=graph->neighbours_set[i-1]+graph->count[i-1];
634*674ae819SStefano Zampini   }
635*674ae819SStefano Zampini   /* Get information for sharing subdomains */
636*674ae819SStefano Zampini   ierr = PetscMemzero(graph->count,graph->nvtxs*sizeof(*graph->count));CHKERRQ(ierr);
637*674ae819SStefano Zampini   for (i=1;i<n_neigh;i++) { /* dont count myself */
638*674ae819SStefano Zampini     s = n_shared[i];
639*674ae819SStefano Zampini     for (j=0;j<s;j++) {
640*674ae819SStefano Zampini       k = shared[i][j];
641*674ae819SStefano Zampini       graph->neighbours_set[k][graph->count[k]] = neigh[i];
642*674ae819SStefano Zampini       graph->count[k] += 1;
643*674ae819SStefano Zampini     }
644*674ae819SStefano Zampini   }
645*674ae819SStefano Zampini   /* sort set of sharing subdomains */
646*674ae819SStefano Zampini   for (i=0;i<graph->nvtxs;i++) {
647*674ae819SStefano Zampini     ierr = PetscSortInt(graph->count[i],graph->neighbours_set[i]);CHKERRQ(ierr);
648*674ae819SStefano Zampini   }
649*674ae819SStefano Zampini   /* Get info for dofs splitting */
650*674ae819SStefano Zampini   for (i=0;i<n_ISForDofs;i++) {
651*674ae819SStefano Zampini     ierr = ISGetSize(ISForDofs[i],&is_size);CHKERRQ(ierr);
652*674ae819SStefano Zampini     ierr = ISGetIndices(ISForDofs[i],(const PetscInt**)&is_indices);CHKERRQ(ierr);
653*674ae819SStefano Zampini     for (j=0;j<is_size;j++) {
654*674ae819SStefano Zampini       graph->which_dof[is_indices[j]]=i;
655*674ae819SStefano Zampini     }
656*674ae819SStefano Zampini     ierr = ISRestoreIndices(ISForDofs[i],(const PetscInt**)&is_indices);CHKERRQ(ierr);
657*674ae819SStefano Zampini   }
658*674ae819SStefano Zampini   /* Take into account Neumann nodes */
659*674ae819SStefano Zampini   ierr = VecSet(local_vec,0.0);CHKERRQ(ierr);
660*674ae819SStefano Zampini   ierr = VecSet(local_vec2,0.0);CHKERRQ(ierr);
661*674ae819SStefano Zampini   if (neumann_is) {
662*674ae819SStefano Zampini     ierr = VecGetArray(local_vec,&array);CHKERRQ(ierr);
663*674ae819SStefano Zampini     ierr = ISGetSize(neumann_is,&is_size);CHKERRQ(ierr);
664*674ae819SStefano Zampini     ierr = ISGetIndices(neumann_is,(const PetscInt**)&is_indices);CHKERRQ(ierr);
665*674ae819SStefano Zampini     for (i=0;i<is_size;i++) {
666*674ae819SStefano Zampini       array[is_indices[i]] = 1.0;
667*674ae819SStefano Zampini     }
668*674ae819SStefano Zampini     ierr = ISRestoreIndices(neumann_is,(const PetscInt**)&is_indices);CHKERRQ(ierr);
669*674ae819SStefano Zampini     ierr = VecRestoreArray(local_vec,&array);CHKERRQ(ierr);
670*674ae819SStefano Zampini   }
671*674ae819SStefano Zampini   /* Neumann nodes: impose consistency among neighbours */
672*674ae819SStefano Zampini   ierr = VecSet(global_vec,0.0);CHKERRQ(ierr);
673*674ae819SStefano Zampini   ierr = VecScatterBegin(scatter_ctx,local_vec,global_vec,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
674*674ae819SStefano Zampini   ierr = VecScatterEnd(scatter_ctx,local_vec,global_vec,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
675*674ae819SStefano Zampini   ierr = VecScatterBegin(scatter_ctx,global_vec,local_vec,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
676*674ae819SStefano Zampini   ierr = VecScatterEnd(scatter_ctx,global_vec,local_vec,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
677*674ae819SStefano Zampini   ierr = VecGetArray(local_vec,&array);CHKERRQ(ierr);
678*674ae819SStefano Zampini   for (i=0;i<graph->nvtxs;i++) {
679*674ae819SStefano Zampini     if (array[i] > 0.0) {
680*674ae819SStefano Zampini       graph->which_dof[i] = NEUMANN_MARK;
681*674ae819SStefano Zampini     }
682*674ae819SStefano Zampini   }
683*674ae819SStefano Zampini   ierr = VecRestoreArray(local_vec,&array);CHKERRQ(ierr);
684*674ae819SStefano Zampini   /* Take into account Dirichlet nodes */
685*674ae819SStefano Zampini   ierr = VecSet(local_vec2,0.0);CHKERRQ(ierr);
686*674ae819SStefano Zampini   if (dirichlet_is) {
687*674ae819SStefano Zampini     ierr = VecGetArray(local_vec,&array);CHKERRQ(ierr);
688*674ae819SStefano Zampini     ierr = VecGetArray(local_vec2,&array2);CHKERRQ(ierr);
689*674ae819SStefano Zampini     ierr = ISGetSize(dirichlet_is,&is_size);CHKERRQ(ierr);
690*674ae819SStefano Zampini     ierr = ISGetIndices(dirichlet_is,(const PetscInt**)&is_indices);CHKERRQ(ierr);
691*674ae819SStefano Zampini     for (i=0;i<is_size;i++){
692*674ae819SStefano Zampini       k = is_indices[i];
693*674ae819SStefano Zampini       if (graph->count[k] && !graph->touched[k]) {
694*674ae819SStefano Zampini         if (array[k] > 0.0) {
695*674ae819SStefano Zampini           SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_USER,"BDDC cannot have boundary nodes which are marked Neumann and Dirichlet at the same time! Local node %d is wrong!\n",k);
696*674ae819SStefano Zampini         }
697*674ae819SStefano Zampini         array2[k] = 1.0;
698*674ae819SStefano Zampini       }
699*674ae819SStefano Zampini     }
700*674ae819SStefano Zampini     ierr = ISRestoreIndices(dirichlet_is,(const PetscInt**)&is_indices);CHKERRQ(ierr);
701*674ae819SStefano Zampini     ierr = VecRestoreArray(local_vec,&array);CHKERRQ(ierr);
702*674ae819SStefano Zampini     ierr = VecRestoreArray(local_vec2,&array2);CHKERRQ(ierr);
703*674ae819SStefano Zampini   }
704*674ae819SStefano Zampini   /* Dirichlet nodes: impose consistency among neighbours */
705*674ae819SStefano Zampini   ierr = VecSet(global_vec,0.0);CHKERRQ(ierr);
706*674ae819SStefano Zampini   ierr = VecScatterBegin(scatter_ctx,local_vec2,global_vec,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
707*674ae819SStefano Zampini   ierr = VecScatterEnd(scatter_ctx,local_vec2,global_vec,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
708*674ae819SStefano Zampini   ierr = VecScatterBegin(scatter_ctx,global_vec,local_vec,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
709*674ae819SStefano Zampini   ierr = VecScatterEnd(scatter_ctx,global_vec,local_vec,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
710*674ae819SStefano Zampini   ierr = VecGetArray(local_vec,&array);CHKERRQ(ierr);
711*674ae819SStefano Zampini   for (i=0;i<graph->nvtxs;i++) {
712*674ae819SStefano Zampini     if (array[i] > 0.0) {
713*674ae819SStefano Zampini       graph->touched[i] = PETSC_TRUE;
714*674ae819SStefano Zampini       graph->subset[i] = 0; /* dirichlet nodes treated as internal -> is it ok? */
715*674ae819SStefano Zampini       graph->which_dof[i] = DIRICHLET_MARK;
716*674ae819SStefano Zampini     }
717*674ae819SStefano Zampini   }
718*674ae819SStefano Zampini   ierr = VecRestoreArray(local_vec,&array);CHKERRQ(ierr);
719*674ae819SStefano Zampini   /* mark special nodes -> each will become a single node equivalence class */
720*674ae819SStefano Zampini   if (custom_primal_vertices) {
721*674ae819SStefano Zampini     ierr = ISGetSize(custom_primal_vertices,&is_size);CHKERRQ(ierr);
722*674ae819SStefano Zampini     ierr = ISGetIndices(custom_primal_vertices,(const PetscInt**)&is_indices);CHKERRQ(ierr);
723*674ae819SStefano Zampini     for (i=0;i<is_size;i++) {
724*674ae819SStefano Zampini       graph->which_dof[is_indices[i]] = SPECIAL_MARK-i;
725*674ae819SStefano Zampini     }
726*674ae819SStefano Zampini     ierr = ISRestoreIndices(custom_primal_vertices,(const PetscInt**)&is_indices);CHKERRQ(ierr);
727*674ae819SStefano Zampini   }
728*674ae819SStefano Zampini   /* mark interior nodes as touched and belonging to partition number 0 */
729*674ae819SStefano Zampini   for (i=0;i<graph->nvtxs;i++) {
730*674ae819SStefano Zampini     if (!graph->count[i]) {
731*674ae819SStefano Zampini       graph->touched[i] = PETSC_TRUE;
732*674ae819SStefano Zampini       graph->subset[i] = 0;
733*674ae819SStefano Zampini     }
734*674ae819SStefano Zampini   }
735*674ae819SStefano Zampini   /* init graph structure and compute default subsets */
736*674ae819SStefano Zampini   nodes_touched=0;
737*674ae819SStefano Zampini   for (i=0;i<graph->nvtxs;i++) {
738*674ae819SStefano Zampini     if (graph->touched[i]) {
739*674ae819SStefano Zampini       nodes_touched++;
740*674ae819SStefano Zampini     }
741*674ae819SStefano Zampini   }
742*674ae819SStefano Zampini   i = 0;
743*674ae819SStefano Zampini   graph->ncc = 0;
744*674ae819SStefano Zampini   total_counts = 0;
745*674ae819SStefano Zampini   while (nodes_touched<graph->nvtxs) {
746*674ae819SStefano Zampini     /*  find first untouched node in local ordering */
747*674ae819SStefano Zampini     while (graph->touched[i]) {
748*674ae819SStefano Zampini       i++;
749*674ae819SStefano Zampini     }
750*674ae819SStefano Zampini     graph->touched[i] = PETSC_TRUE;
751*674ae819SStefano Zampini     graph->subset[i] = graph->ncc+1;
752*674ae819SStefano Zampini     graph->cptr[graph->ncc] = total_counts;
753*674ae819SStefano Zampini     graph->queue[total_counts] = i;
754*674ae819SStefano Zampini     total_counts++;
755*674ae819SStefano Zampini     nodes_touched++;
756*674ae819SStefano Zampini     /* now find all other nodes having the same set of sharing subdomains */
757*674ae819SStefano Zampini     for (j=i+1;j<graph->nvtxs;j++){
758*674ae819SStefano Zampini       /* check for same number of sharing subdomains and dof number */
759*674ae819SStefano Zampini       if (!graph->touched[j] && graph->count[i]==graph->count[j] && graph->which_dof[i] == graph->which_dof[j] ){
760*674ae819SStefano Zampini         /* check for same set of sharing subdomains */
761*674ae819SStefano Zampini         same_set=PETSC_TRUE;
762*674ae819SStefano Zampini         for (k=0;k<graph->count[j];k++){
763*674ae819SStefano Zampini           if (graph->neighbours_set[i][k]!=graph->neighbours_set[j][k]) {
764*674ae819SStefano Zampini             same_set=PETSC_FALSE;
765*674ae819SStefano Zampini           }
766*674ae819SStefano Zampini         }
767*674ae819SStefano Zampini         /* I found a friend of mine */
768*674ae819SStefano Zampini         if (same_set) {
769*674ae819SStefano Zampini           graph->subset[j]=graph->ncc+1;
770*674ae819SStefano Zampini           graph->touched[j]=PETSC_TRUE;
771*674ae819SStefano Zampini           nodes_touched++;
772*674ae819SStefano Zampini           graph->queue[total_counts] = j;
773*674ae819SStefano Zampini           total_counts++;
774*674ae819SStefano Zampini         }
775*674ae819SStefano Zampini       }
776*674ae819SStefano Zampini     }
777*674ae819SStefano Zampini     graph->ncc++;
778*674ae819SStefano Zampini   }
779*674ae819SStefano Zampini   /* set default number of subsets (at this point no info on csr graph has been taken into account, so n_subsets = ncc */
780*674ae819SStefano Zampini   graph->n_subsets = graph->ncc;
781*674ae819SStefano Zampini   ierr = PetscMalloc(graph->n_subsets*sizeof(*graph->subset_ncc),&graph->subset_ncc);CHKERRQ(ierr);
782*674ae819SStefano Zampini   for (i=0;i<graph->n_subsets;i++) {
783*674ae819SStefano Zampini     graph->subset_ncc[i] = 1;
784*674ae819SStefano Zampini   }
785*674ae819SStefano Zampini   /* final pointer */
786*674ae819SStefano Zampini   graph->cptr[graph->ncc] = total_counts;
787*674ae819SStefano Zampini   /* free memory allocated by ISLocalToGlobalMappingGetInfo */
788*674ae819SStefano Zampini   ierr = ISLocalToGlobalMappingRestoreInfo(graph->l2gmap,&n_neigh,&neigh,&n_shared,&shared);CHKERRQ(ierr);
789*674ae819SStefano Zampini   /* free objects */
790*674ae819SStefano Zampini   ierr = VecDestroy(&local_vec);CHKERRQ(ierr);
791*674ae819SStefano Zampini   ierr = VecDestroy(&local_vec2);CHKERRQ(ierr);
792*674ae819SStefano Zampini   ierr = VecDestroy(&global_vec);CHKERRQ(ierr);
793*674ae819SStefano Zampini   ierr = VecScatterDestroy(&scatter_ctx);CHKERRQ(ierr);
794*674ae819SStefano Zampini   PetscFunctionReturn(0);
795*674ae819SStefano Zampini }
796*674ae819SStefano Zampini 
797*674ae819SStefano Zampini #undef __FUNCT__
798*674ae819SStefano Zampini #define __FUNCT__ "PCBDDCGraphResetCSR"
799*674ae819SStefano Zampini PetscErrorCode PCBDDCGraphResetCSR(PCBDDCGraph graph)
800*674ae819SStefano Zampini {
801*674ae819SStefano Zampini   PetscErrorCode ierr;
802*674ae819SStefano Zampini 
803*674ae819SStefano Zampini   PetscFunctionBegin;
804*674ae819SStefano Zampini   ierr = PetscFree(graph->xadj);CHKERRQ(ierr);
805*674ae819SStefano Zampini   ierr = PetscFree(graph->adjncy);CHKERRQ(ierr);
806*674ae819SStefano Zampini   PetscFunctionReturn(0);
807*674ae819SStefano Zampini }
808*674ae819SStefano Zampini 
809*674ae819SStefano Zampini #undef __FUNCT__
810*674ae819SStefano Zampini #define __FUNCT__ "PCBDDCGraphReset"
811*674ae819SStefano Zampini PetscErrorCode PCBDDCGraphReset(PCBDDCGraph graph)
812*674ae819SStefano Zampini {
813*674ae819SStefano Zampini   PetscErrorCode ierr;
814*674ae819SStefano Zampini 
815*674ae819SStefano Zampini   PetscFunctionBegin;
816*674ae819SStefano Zampini   ierr = ISLocalToGlobalMappingDestroy(&graph->l2gmap);CHKERRQ(ierr);
817*674ae819SStefano Zampini   ierr = PetscFree(graph->subset_ncc);CHKERRQ(ierr);
818*674ae819SStefano Zampini   if (graph->nvtxs) {
819*674ae819SStefano Zampini     ierr = PetscFree(graph->neighbours_set[0]);CHKERRQ(ierr);
820*674ae819SStefano Zampini   }
821*674ae819SStefano Zampini   ierr = PetscFree7(graph->touched,
822*674ae819SStefano Zampini                     graph->count,
823*674ae819SStefano Zampini                     graph->neighbours_set,
824*674ae819SStefano Zampini                     graph->subset,
825*674ae819SStefano Zampini                     graph->which_dof,
826*674ae819SStefano Zampini                     graph->cptr,
827*674ae819SStefano Zampini                     graph->queue);CHKERRQ(ierr);
828*674ae819SStefano Zampini   graph->nvtxs = 0;
829*674ae819SStefano Zampini   graph->n_subsets = 0;
830*674ae819SStefano Zampini   graph->custom_minimal_size = 1;
831*674ae819SStefano Zampini   PetscFunctionReturn(0);
832*674ae819SStefano Zampini }
833*674ae819SStefano Zampini 
834*674ae819SStefano Zampini #undef __FUNCT__
835*674ae819SStefano Zampini #define __FUNCT__ "PCBDDCGraphInit"
836*674ae819SStefano Zampini PetscErrorCode PCBDDCGraphInit(PCBDDCGraph graph, ISLocalToGlobalMapping l2gmap)
837*674ae819SStefano Zampini {
838*674ae819SStefano Zampini   PetscInt       n;
839*674ae819SStefano Zampini   PetscErrorCode ierr;
840*674ae819SStefano Zampini 
841*674ae819SStefano Zampini   PetscFunctionBegin;
842*674ae819SStefano Zampini   PetscValidPointer(graph,1);
843*674ae819SStefano Zampini   PetscValidHeaderSpecific(l2gmap,IS_LTOGM_CLASSID,2);
844*674ae819SStefano Zampini   /* reset graph if already allocated */
845*674ae819SStefano Zampini   if (graph->nvtxs) {
846*674ae819SStefano Zampini     ierr = PCBDDCGraphReset(graph);CHKERRQ(ierr);
847*674ae819SStefano Zampini   }
848*674ae819SStefano Zampini   /* set number of vertices */
849*674ae819SStefano Zampini   ierr = PetscObjectReference((PetscObject)l2gmap);CHKERRQ(ierr);
850*674ae819SStefano Zampini   graph->l2gmap = l2gmap;
851*674ae819SStefano Zampini   ierr = ISLocalToGlobalMappingGetSize(l2gmap,&n);CHKERRQ(ierr);
852*674ae819SStefano Zampini   graph->nvtxs = n;
853*674ae819SStefano Zampini   /* allocate used space */
854*674ae819SStefano Zampini   ierr = PetscMalloc7(graph->nvtxs,PetscBool,&graph->touched,
855*674ae819SStefano Zampini                       graph->nvtxs,PetscInt,&graph->count,
856*674ae819SStefano Zampini                       graph->nvtxs,PetscInt*,&graph->neighbours_set,
857*674ae819SStefano Zampini                       graph->nvtxs,PetscInt,&graph->subset,
858*674ae819SStefano Zampini                       graph->nvtxs,PetscInt,&graph->which_dof,
859*674ae819SStefano Zampini                       graph->nvtxs+1,PetscInt,&graph->cptr,
860*674ae819SStefano Zampini                       graph->nvtxs,PetscInt,&graph->queue);CHKERRQ(ierr);
861*674ae819SStefano Zampini   /* zeroes memory */
862*674ae819SStefano Zampini   ierr = PetscMemzero(graph->touched,graph->nvtxs*sizeof(PetscBool));CHKERRQ(ierr);
863*674ae819SStefano Zampini   ierr = PetscMemzero(graph->count,graph->nvtxs*sizeof(PetscInt));CHKERRQ(ierr);
864*674ae819SStefano Zampini   ierr = PetscMemzero(graph->subset,graph->nvtxs*sizeof(PetscInt));CHKERRQ(ierr);
865*674ae819SStefano Zampini   ierr = PetscMemzero(graph->which_dof,graph->nvtxs*sizeof(PetscInt));CHKERRQ(ierr);
866*674ae819SStefano Zampini   ierr = PetscMemzero(graph->cptr,(graph->nvtxs+1)*sizeof(PetscInt));CHKERRQ(ierr);
867*674ae819SStefano Zampini   ierr = PetscMemzero(graph->queue,graph->nvtxs*sizeof(PetscInt));CHKERRQ(ierr);
868*674ae819SStefano Zampini   /* zeroes first pointer to neighbour set */
869*674ae819SStefano Zampini   if (graph->nvtxs) {
870*674ae819SStefano Zampini     graph->neighbours_set[0] = 0;
871*674ae819SStefano Zampini   }
872*674ae819SStefano Zampini   /* zeroes workspace for values of ncc */
873*674ae819SStefano Zampini   graph->subset_ncc = 0;
874*674ae819SStefano Zampini   PetscFunctionReturn(0);
875*674ae819SStefano Zampini }
876*674ae819SStefano Zampini 
877*674ae819SStefano Zampini #undef __FUNCT__
878*674ae819SStefano Zampini #define __FUNCT__ "PCBDDCGraphDestroy"
879*674ae819SStefano Zampini PetscErrorCode PCBDDCGraphDestroy(PCBDDCGraph* graph)
880*674ae819SStefano Zampini {
881*674ae819SStefano Zampini   PetscErrorCode ierr;
882*674ae819SStefano Zampini 
883*674ae819SStefano Zampini   PetscFunctionBegin;
884*674ae819SStefano Zampini   ierr = PCBDDCGraphReset(*graph);CHKERRQ(ierr);
885*674ae819SStefano Zampini   ierr = PetscFree(*graph);CHKERRQ(ierr);
886*674ae819SStefano Zampini   PetscFunctionReturn(0);
887*674ae819SStefano Zampini }
888*674ae819SStefano Zampini 
889*674ae819SStefano Zampini #undef __FUNCT__
890*674ae819SStefano Zampini #define __FUNCT__ "PCBDDCGraphCreate"
891*674ae819SStefano Zampini PetscErrorCode PCBDDCGraphCreate(PCBDDCGraph *graph)
892*674ae819SStefano Zampini {
893*674ae819SStefano Zampini   PCBDDCGraph    new_graph;
894*674ae819SStefano Zampini   PetscErrorCode ierr;
895*674ae819SStefano Zampini 
896*674ae819SStefano Zampini   PetscFunctionBegin;
897*674ae819SStefano Zampini   ierr = PetscMalloc(sizeof(*new_graph),&new_graph);CHKERRQ(ierr);
898*674ae819SStefano Zampini   /* local to global mapping of dofs */
899*674ae819SStefano Zampini   new_graph->l2gmap = 0;
900*674ae819SStefano Zampini   /* vertex size */
901*674ae819SStefano Zampini   new_graph->nvtxs = 0;
902*674ae819SStefano Zampini   new_graph->n_subsets = 0;
903*674ae819SStefano Zampini   new_graph->custom_minimal_size = 1;
904*674ae819SStefano Zampini   /* zeroes ponters */
905*674ae819SStefano Zampini   new_graph->neighbours_set = 0;
906*674ae819SStefano Zampini   new_graph->subset = 0;
907*674ae819SStefano Zampini   new_graph->which_dof = 0;
908*674ae819SStefano Zampini   new_graph->cptr = 0;
909*674ae819SStefano Zampini   new_graph->queue = 0;
910*674ae819SStefano Zampini   new_graph->count = 0;
911*674ae819SStefano Zampini   new_graph->subset_ncc = 0;
912*674ae819SStefano Zampini   new_graph->touched = 0;
913*674ae819SStefano Zampini   /* zeroes pointers to csr graph of local nodes connectivity (optional data) */
914*674ae819SStefano Zampini   new_graph->xadj = 0;
915*674ae819SStefano Zampini   new_graph->adjncy = 0;
916*674ae819SStefano Zampini   *graph = new_graph;
917*674ae819SStefano Zampini   PetscFunctionReturn(0);
918*674ae819SStefano Zampini }
919