xref: /petsc/src/ksp/pc/impls/bddc/bddcgraph.c (revision e49050b469fe929c9147b2e09cde6f59023ec235)
1674ae819SStefano Zampini #include <petsc-private/petscimpl.h>
2674ae819SStefano Zampini #include "bddcprivate.h"
3674ae819SStefano Zampini #include "bddcstructs.h"
4674ae819SStefano Zampini 
5674ae819SStefano Zampini #define NEUMANN_MARK -1
6674ae819SStefano Zampini #define DIRICHLET_MARK -2
7674ae819SStefano Zampini #define SPECIAL_MARK -3
8674ae819SStefano Zampini 
9674ae819SStefano Zampini #undef __FUNCT__
10674ae819SStefano Zampini #define __FUNCT__ "PCBDDCGraphASCIIView"
11*e49050b4SStefano Zampini PetscErrorCode PCBDDCGraphASCIIView(PCBDDCGraph graph, PetscInt verbosity_level, PetscViewer viewer)
12674ae819SStefano Zampini {
13674ae819SStefano Zampini   PetscInt       i,j;
14674ae819SStefano Zampini   PetscErrorCode ierr;
15674ae819SStefano Zampini 
16674ae819SStefano Zampini   PetscFunctionBegin;
17674ae819SStefano Zampini   ierr = PetscViewerASCIISynchronizedAllow(viewer,PETSC_TRUE);CHKERRQ(ierr);
18674ae819SStefano Zampini   ierr = PetscViewerASCIIPrintf(viewer,"--------------------------------------------------\n");CHKERRQ(ierr);
19674ae819SStefano Zampini   ierr = PetscViewerFlush(viewer);CHKERRQ(ierr);
20674ae819SStefano Zampini   ierr = PetscViewerASCIISynchronizedPrintf(viewer,"Local BDDC graph for subdomain %04d\n",PetscGlobalRank);CHKERRQ(ierr);
21674ae819SStefano Zampini   ierr = PetscViewerASCIISynchronizedPrintf(viewer,"Number of vertices %d\n",graph->nvtxs);CHKERRQ(ierr);
22*e49050b4SStefano Zampini   if (verbosity_level > 1) {
23674ae819SStefano Zampini     for (i=0;i<graph->nvtxs;i++) {
24674ae819SStefano Zampini       ierr = PetscViewerASCIISynchronizedPrintf(viewer,"%d:\n",i);CHKERRQ(ierr);
25674ae819SStefano Zampini       ierr = PetscViewerASCIISynchronizedPrintf(viewer,"   touched: %d\n",graph->touched[i]);CHKERRQ(ierr);
26674ae819SStefano Zampini       ierr = PetscViewerASCIISynchronizedPrintf(viewer,"   which_dof: %d\n",graph->which_dof[i]);CHKERRQ(ierr);
27674ae819SStefano Zampini       ierr = PetscViewerASCIISynchronizedPrintf(viewer,"   neighbours: %d\n",graph->count[i]);CHKERRQ(ierr);
28674ae819SStefano Zampini       if (graph->count[i]) {
29674ae819SStefano Zampini         ierr = PetscViewerASCIISynchronizedPrintf(viewer,"     set of neighbours:");CHKERRQ(ierr);
30674ae819SStefano Zampini         for (j=0;j<graph->count[i];j++) {
31674ae819SStefano Zampini           ierr = PetscViewerASCIISynchronizedPrintf(viewer," %d",graph->neighbours_set[i][j]);CHKERRQ(ierr);
32674ae819SStefano Zampini         }
33674ae819SStefano Zampini         ierr = PetscViewerASCIISynchronizedPrintf(viewer,"\n");CHKERRQ(ierr);
34674ae819SStefano Zampini       }
35*e49050b4SStefano Zampini       if (verbosity_level > 2) {
36674ae819SStefano Zampini         if (graph->xadj && graph->adjncy) {
37674ae819SStefano Zampini           ierr = PetscViewerASCIISynchronizedPrintf(viewer,"   local adj list:");CHKERRQ(ierr);
38674ae819SStefano Zampini           for (j=graph->xadj[i];j<graph->xadj[i+1];j++) {
39674ae819SStefano Zampini             ierr = PetscViewerASCIISynchronizedPrintf(viewer," %d",graph->adjncy[j]);CHKERRQ(ierr);
40674ae819SStefano Zampini           }
41674ae819SStefano Zampini           ierr = PetscViewerASCIISynchronizedPrintf(viewer,"\n");CHKERRQ(ierr);
42674ae819SStefano Zampini         }
43*e49050b4SStefano Zampini       }
44674ae819SStefano Zampini       ierr = PetscViewerASCIISynchronizedPrintf(viewer,"   interface subset id: %d\n",graph->subset[i]);CHKERRQ(ierr);
45674ae819SStefano Zampini       if (graph->subset[i] && graph->subset_ncc) {
46674ae819SStefano Zampini         ierr = PetscViewerASCIISynchronizedPrintf(viewer,"   ncc for subset: %d\n",graph->subset_ncc[graph->subset[i]-1]);CHKERRQ(ierr);
47674ae819SStefano Zampini       }
48674ae819SStefano Zampini     }
49*e49050b4SStefano Zampini   }
50674ae819SStefano Zampini   ierr = PetscViewerASCIISynchronizedPrintf(viewer,"Total number of connected components %d\n",graph->ncc);CHKERRQ(ierr);
51674ae819SStefano Zampini   for (i=0;i<graph->ncc;i++) {
52674ae819SStefano Zampini     ierr = PetscViewerASCIISynchronizedPrintf(viewer,"  %d (neighs:",i);CHKERRQ(ierr);
53674ae819SStefano Zampini     PetscInt node_num=graph->queue[graph->cptr[i]];
54674ae819SStefano Zampini     for (j=0;j<graph->count[node_num];j++) {
55674ae819SStefano Zampini       ierr = PetscViewerASCIISynchronizedPrintf(viewer," %d",graph->neighbours_set[node_num][j]);CHKERRQ(ierr);
56674ae819SStefano Zampini     }
57674ae819SStefano Zampini     ierr = PetscViewerASCIISynchronizedPrintf(viewer,"):");CHKERRQ(ierr);
58674ae819SStefano Zampini     for (j=graph->cptr[i];j<graph->cptr[i+1];j++) {
59674ae819SStefano Zampini       ierr = PetscViewerASCIISynchronizedPrintf(viewer," %d",graph->queue[j]);CHKERRQ(ierr);
60674ae819SStefano Zampini     }
61674ae819SStefano Zampini     ierr = PetscViewerASCIISynchronizedPrintf(viewer,"\n");CHKERRQ(ierr);
62674ae819SStefano Zampini   }
63*e49050b4SStefano Zampini   if (graph->custom_minimal_size > 1 && verbosity_level > 1) {
64674ae819SStefano Zampini     ierr = PetscViewerASCIISynchronizedPrintf(viewer,"Custom minimal size %d\n",graph->custom_minimal_size);CHKERRQ(ierr);
65674ae819SStefano Zampini   }
66674ae819SStefano Zampini   ierr = PetscViewerFlush(viewer);CHKERRQ(ierr);
67674ae819SStefano Zampini   PetscFunctionReturn(0);
68674ae819SStefano Zampini }
69674ae819SStefano Zampini 
70674ae819SStefano Zampini #undef __FUNCT__
71674ae819SStefano Zampini #define __FUNCT__ "PCBDDCGraphGetCandidatesIS"
72674ae819SStefano 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)
73674ae819SStefano Zampini {
74674ae819SStefano Zampini   IS             *ISForFaces,*ISForEdges,ISForVertices;
75674ae819SStefano Zampini   PetscInt       i,j,nfc,nec,nvc,*idx;
76674ae819SStefano Zampini   PetscBool      twodim_flag;
77674ae819SStefano Zampini   PetscErrorCode ierr;
78674ae819SStefano Zampini 
79674ae819SStefano Zampini   PetscFunctionBegin;
80674ae819SStefano Zampini   /* loop on ccs to evalute number of faces, edges and vertices */
81674ae819SStefano Zampini   nfc = 0;
82674ae819SStefano Zampini   nec = 0;
83674ae819SStefano Zampini   nvc = 0;
84674ae819SStefano Zampini   twodim_flag = PETSC_FALSE;
85674ae819SStefano Zampini   for (i=0;i<graph->ncc;i++) {
86674ae819SStefano Zampini     if (graph->cptr[i+1]-graph->cptr[i] > graph->custom_minimal_size) {
87674ae819SStefano Zampini       if (graph->count[graph->queue[graph->cptr[i]]] == 1) {
88674ae819SStefano Zampini         nfc++;
89674ae819SStefano Zampini       } else { /* note that nec will be zero in 2d */
90674ae819SStefano Zampini         nec++;
91674ae819SStefano Zampini       }
92674ae819SStefano Zampini     } else {
93674ae819SStefano Zampini       nvc += graph->cptr[i+1]-graph->cptr[i];
94674ae819SStefano Zampini     }
95674ae819SStefano Zampini   }
96674ae819SStefano Zampini   if (!nec) { /* we are in a 2d case -> no faces, only edges */
97674ae819SStefano Zampini     nec = nfc;
98674ae819SStefano Zampini     nfc = 0;
99674ae819SStefano Zampini     twodim_flag = PETSC_TRUE;
100674ae819SStefano Zampini   }
101674ae819SStefano Zampini   /* allocate IS arrays for faces, edges. Vertices need a single index set. */
102674ae819SStefano Zampini   ISForFaces = 0;
103674ae819SStefano Zampini   ISForEdges = 0;
104674ae819SStefano Zampini   ISForVertices = 0;
105674ae819SStefano Zampini   if (use_faces) {
106674ae819SStefano Zampini     ierr = PetscMalloc(nfc*sizeof(IS),&ISForFaces);CHKERRQ(ierr);
107674ae819SStefano Zampini   }
108674ae819SStefano Zampini   if (use_edges) {
109674ae819SStefano Zampini     ierr = PetscMalloc(nec*sizeof(IS),&ISForEdges);CHKERRQ(ierr);
110674ae819SStefano Zampini   }
111674ae819SStefano Zampini   if (use_vertices) {
112674ae819SStefano Zampini     ierr = PetscMalloc(nvc*sizeof(PetscInt),&idx);CHKERRQ(ierr);
113674ae819SStefano Zampini   }
114674ae819SStefano Zampini   /* loop on ccs to compute index sets for faces and edges */
115674ae819SStefano Zampini   nfc = 0;
116674ae819SStefano Zampini   nec = 0;
117674ae819SStefano Zampini   nvc = 0;
118674ae819SStefano Zampini   for (i=0;i<graph->ncc;i++) {
119674ae819SStefano Zampini     if (graph->cptr[i+1]-graph->cptr[i] > graph->custom_minimal_size) {
120674ae819SStefano Zampini       if (graph->count[graph->queue[graph->cptr[i]]] == 1) {
121674ae819SStefano Zampini         if (twodim_flag) {
122674ae819SStefano Zampini           if (use_edges) {
123674ae819SStefano 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);
124674ae819SStefano Zampini             nec++;
125674ae819SStefano Zampini           }
126674ae819SStefano Zampini         } else {
127674ae819SStefano Zampini           if (use_faces) {
128674ae819SStefano 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);
129674ae819SStefano Zampini             nfc++;
130674ae819SStefano Zampini           }
131674ae819SStefano Zampini         }
132674ae819SStefano Zampini       } else {
133674ae819SStefano Zampini         if (use_edges) {
134674ae819SStefano 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);
135674ae819SStefano Zampini           nec++;
136674ae819SStefano Zampini         }
137674ae819SStefano Zampini       }
138674ae819SStefano Zampini     }
139674ae819SStefano Zampini   }
140674ae819SStefano Zampini   /* index set for vertices */
141674ae819SStefano Zampini   if (use_vertices) {
142674ae819SStefano Zampini     for (i=0;i<graph->ncc;i++) {
143674ae819SStefano Zampini       if (graph->cptr[i+1]-graph->cptr[i] <= graph->custom_minimal_size) {
144674ae819SStefano Zampini         for (j=graph->cptr[i];j<graph->cptr[i+1];j++) {
145674ae819SStefano Zampini           idx[nvc]=graph->queue[j];
146674ae819SStefano Zampini           nvc++;
147674ae819SStefano Zampini         }
148674ae819SStefano Zampini       }
149674ae819SStefano Zampini     }
150674ae819SStefano Zampini     /* sort vertex set (by local ordering) */
151674ae819SStefano Zampini     ierr = PetscSortInt(nvc,idx);CHKERRQ(ierr);
152674ae819SStefano Zampini     ierr = ISCreateGeneral(PETSC_COMM_SELF,nvc,idx,PETSC_OWN_POINTER,&ISForVertices);CHKERRQ(ierr);
153674ae819SStefano Zampini   }
154674ae819SStefano Zampini   /* get back info */
155674ae819SStefano Zampini   *n_faces = nfc;
156674ae819SStefano Zampini   *FacesIS = ISForFaces;
157674ae819SStefano Zampini   *n_edges = nec;
158674ae819SStefano Zampini   *EdgesIS = ISForEdges;
159674ae819SStefano Zampini   *VerticesIS = ISForVertices;
160674ae819SStefano Zampini   PetscFunctionReturn(0);
161674ae819SStefano Zampini }
162674ae819SStefano Zampini 
163674ae819SStefano Zampini #undef __FUNCT__
164674ae819SStefano Zampini #define __FUNCT__ "PCBDDCGraphComputeConnectedComponents"
165674ae819SStefano Zampini PetscErrorCode PCBDDCGraphComputeConnectedComponents(PCBDDCGraph graph)
166674ae819SStefano Zampini {
167674ae819SStefano Zampini   PetscInt    adapt_interface,adapt_interface_reduced;
168674ae819SStefano Zampini   MPI_Comm    interface_comm;
169674ae819SStefano Zampini   MPI_Request *send_requests;
170674ae819SStefano Zampini   MPI_Request *recv_requests;
171674ae819SStefano Zampini   PetscInt    *aux_new_xadj,*new_xadj,*new_adjncy,**temp_buffer;
172674ae819SStefano Zampini   PetscInt    i,j,k,s,sum_requests,buffer_size,size_of_recv,temp_buffer_size;
173674ae819SStefano Zampini   PetscMPIInt rank,neigh,tag,mpi_buffer_size;
174674ae819SStefano Zampini   PetscInt    *cum_recv_counts,*subset_to_nodes_indices,*recv_buffer_subset,*nodes_to_temp_buffer_indices;
175674ae819SStefano Zampini   PetscInt    *send_buffer,*recv_buffer,*queue_in_global_numbering,*sizes_of_sends,*add_to_subset;
176674ae819SStefano Zampini   PetscInt    start_of_recv,start_of_send,size_of_send,global_subset_counter,ins_val;
177674ae819SStefano Zampini   PetscBool   *subset_cc_adapt,same_set;
178674ae819SStefano Zampini   PetscErrorCode ierr;
179674ae819SStefano Zampini 
180674ae819SStefano Zampini   PetscFunctionBegin;
181674ae819SStefano Zampini   /* compute connected components locally */
182674ae819SStefano Zampini   ierr = PetscObjectGetComm((PetscObject)(graph->l2gmap),&interface_comm);CHKERRQ(ierr);
183674ae819SStefano Zampini   ierr = PCBDDCGraphComputeConnectedComponentsLocal(graph);CHKERRQ(ierr);
184674ae819SStefano Zampini   /* check consistency of connected components among neighbouring subdomains -> it adapt them in case it is needed */
185674ae819SStefano Zampini   adapt_interface = 0;
186674ae819SStefano Zampini   adapt_interface_reduced = 0;
187674ae819SStefano Zampini   for (i=0;i<graph->n_subsets;i++) {
188674ae819SStefano Zampini     /* We are not sure that on a given subset of the local interface,
189674ae819SStefano Zampini        with two connected components, the latters be the same among sharing subdomains */
190674ae819SStefano Zampini     if (graph->subset_ncc[i] > 1) {
191674ae819SStefano Zampini       adapt_interface=1;
192674ae819SStefano Zampini       break;
193674ae819SStefano Zampini     }
194674ae819SStefano Zampini   }
195674ae819SStefano Zampini   ierr = MPI_Allreduce(&adapt_interface,&adapt_interface_reduced,1,MPIU_INT,MPI_LOR,interface_comm);CHKERRQ(ierr);
196674ae819SStefano Zampini 
197674ae819SStefano Zampini   if (graph->n_subsets && adapt_interface_reduced) {
198674ae819SStefano Zampini     /* Retrict adjacency graph using information from previously computed connected components */
199674ae819SStefano Zampini     ierr = PetscMalloc(graph->nvtxs*sizeof(PetscInt),&aux_new_xadj);CHKERRQ(ierr);
200674ae819SStefano Zampini     for (i=0;i<graph->nvtxs;i++) {
201674ae819SStefano Zampini       aux_new_xadj[i]=1;
202674ae819SStefano Zampini     }
203674ae819SStefano Zampini     for (i=0;i<graph->ncc;i++) {
204674ae819SStefano Zampini       k = graph->cptr[i+1]-graph->cptr[i];
205674ae819SStefano Zampini       for (j=0;j<k;j++) {
206674ae819SStefano Zampini         aux_new_xadj[graph->queue[graph->cptr[i]+j]]=k;
207674ae819SStefano Zampini       }
208674ae819SStefano Zampini     }
209674ae819SStefano Zampini     j = 0;
210674ae819SStefano Zampini     for (i=0;i<graph->nvtxs;i++) {
211674ae819SStefano Zampini       j += aux_new_xadj[i];
212674ae819SStefano Zampini     }
213674ae819SStefano Zampini     ierr = PetscMalloc((graph->nvtxs+1)*sizeof(PetscInt),&new_xadj);CHKERRQ(ierr);
214674ae819SStefano Zampini     ierr = PetscMalloc(j*sizeof(PetscInt),&new_adjncy);CHKERRQ(ierr);
215674ae819SStefano Zampini     new_xadj[0]=0;
216674ae819SStefano Zampini     for (i=0;i<graph->nvtxs;i++) {
217674ae819SStefano Zampini       new_xadj[i+1]=new_xadj[i]+aux_new_xadj[i];
218674ae819SStefano Zampini       if (aux_new_xadj[i]==1) {
219674ae819SStefano Zampini         new_adjncy[new_xadj[i]]=i;
220674ae819SStefano Zampini       }
221674ae819SStefano Zampini     }
222674ae819SStefano Zampini     ierr = PetscFree(aux_new_xadj);CHKERRQ(ierr);
223674ae819SStefano Zampini     for (i=0;i<graph->ncc;i++) {
224674ae819SStefano Zampini       k = graph->cptr[i+1]-graph->cptr[i];
225674ae819SStefano Zampini       for (j=0;j<k;j++) {
226674ae819SStefano Zampini         ierr = PetscMemcpy(&new_adjncy[new_xadj[graph->queue[graph->cptr[i]+j]]],&graph->queue[graph->cptr[i]],k*sizeof(PetscInt));CHKERRQ(ierr);
227674ae819SStefano Zampini       }
228674ae819SStefano Zampini     }
229674ae819SStefano Zampini     /* set new CSR into graph */
230674ae819SStefano Zampini     ierr = PetscFree(graph->xadj);CHKERRQ(ierr);
231674ae819SStefano Zampini     ierr = PetscFree(graph->adjncy);CHKERRQ(ierr);
232674ae819SStefano Zampini     graph->xadj = new_xadj;
233674ae819SStefano Zampini     graph->adjncy = new_adjncy;
234674ae819SStefano Zampini     /* allocate some space */
235674ae819SStefano Zampini     ierr = MPI_Comm_rank(interface_comm,&rank);CHKERRQ(ierr);
236674ae819SStefano Zampini     ierr = PetscMalloc((graph->n_subsets+1)*sizeof(*cum_recv_counts),&cum_recv_counts);CHKERRQ(ierr);
237674ae819SStefano Zampini     ierr = PetscMemzero(cum_recv_counts,(graph->n_subsets+1)*sizeof(*cum_recv_counts));CHKERRQ(ierr);
238674ae819SStefano Zampini     ierr = PetscMalloc(graph->n_subsets*sizeof(*subset_to_nodes_indices),&subset_to_nodes_indices);CHKERRQ(ierr);
239674ae819SStefano Zampini     /* first count how many neighbours per connected component I will receive from */
240674ae819SStefano Zampini     cum_recv_counts[0]=0;
241674ae819SStefano Zampini     for (i=1;i<graph->n_subsets+1;i++) {
242674ae819SStefano Zampini       j = 0;
243674ae819SStefano Zampini       while (graph->subset[j] != i) {
244674ae819SStefano Zampini         j++;
245674ae819SStefano Zampini       }
246674ae819SStefano Zampini       subset_to_nodes_indices[i-1]=j;
247674ae819SStefano Zampini       /* We don't want sends/recvs_to/from_self -> here I don't count myself  */
248674ae819SStefano Zampini       cum_recv_counts[i]=cum_recv_counts[i-1]+graph->count[j];
249674ae819SStefano Zampini     }
250674ae819SStefano Zampini     ierr = PetscMalloc(2*cum_recv_counts[graph->n_subsets]*sizeof(*recv_buffer_subset),&recv_buffer_subset);CHKERRQ(ierr);
251674ae819SStefano Zampini     ierr = PetscMalloc(cum_recv_counts[graph->n_subsets]*sizeof(MPI_Request),&send_requests);CHKERRQ(ierr);
252674ae819SStefano Zampini     ierr = PetscMalloc(cum_recv_counts[graph->n_subsets]*sizeof(MPI_Request),&recv_requests);CHKERRQ(ierr);
253674ae819SStefano Zampini     for (i=0;i<cum_recv_counts[graph->n_subsets];i++) {
254674ae819SStefano Zampini       send_requests[i]=MPI_REQUEST_NULL;
255674ae819SStefano Zampini       recv_requests[i]=MPI_REQUEST_NULL;
256674ae819SStefano Zampini     }
257674ae819SStefano Zampini     /* exchange with my neighbours the number of my connected components on the shared interface */
258674ae819SStefano Zampini     sum_requests = 0;
259674ae819SStefano Zampini     for (i=0;i<graph->n_subsets;i++) {
260674ae819SStefano Zampini       j = subset_to_nodes_indices[i];
261674ae819SStefano Zampini       for (k=0;k<graph->count[j];k++) {
262674ae819SStefano Zampini         ierr = PetscMPIIntCast(graph->neighbours_set[j][k],&neigh);CHKERRQ(ierr);
263674ae819SStefano Zampini         ierr = PetscMPIIntCast((PetscInt)(rank+1)*graph->count[j],&tag);CHKERRQ(ierr);
264674ae819SStefano Zampini         ierr = MPI_Isend(&graph->subset_ncc[i],1,MPIU_INT,neigh,tag,interface_comm,&send_requests[sum_requests]);CHKERRQ(ierr);
265674ae819SStefano Zampini         ierr = PetscMPIIntCast((PetscInt)(neigh+1)*graph->count[j],&tag);CHKERRQ(ierr);
266674ae819SStefano Zampini         ierr = MPI_Irecv(&recv_buffer_subset[sum_requests],1,MPIU_INT,neigh,tag,interface_comm,&recv_requests[sum_requests]);CHKERRQ(ierr);
267674ae819SStefano Zampini         sum_requests++;
268674ae819SStefano Zampini       }
269674ae819SStefano Zampini     }
270674ae819SStefano Zampini     ierr = MPI_Waitall(sum_requests,recv_requests,MPI_STATUSES_IGNORE);CHKERRQ(ierr);
271674ae819SStefano Zampini     ierr = MPI_Waitall(sum_requests,send_requests,MPI_STATUSES_IGNORE);CHKERRQ(ierr);
272674ae819SStefano Zampini     /* determine the connected component I need to adapt */
273674ae819SStefano Zampini     ierr = PetscMalloc(graph->n_subsets*sizeof(*subset_cc_adapt),&subset_cc_adapt);CHKERRQ(ierr);
274674ae819SStefano Zampini     ierr = PetscMemzero(subset_cc_adapt,graph->n_subsets*sizeof(*subset_cc_adapt));CHKERRQ(ierr);
275674ae819SStefano Zampini     for (i=0;i<graph->n_subsets;i++) {
276674ae819SStefano Zampini       for (j=cum_recv_counts[i];j<cum_recv_counts[i+1];j++){
277674ae819SStefano Zampini         /* The first condition is natural (someone has a different number of ccs than me), the second one is just to be safe */
278674ae819SStefano Zampini         if ( graph->subset_ncc[i] != recv_buffer_subset[j] || graph->subset_ncc[i] > 1 ) {
279674ae819SStefano Zampini           subset_cc_adapt[i] = PETSC_TRUE;
280674ae819SStefano Zampini           break;
281674ae819SStefano Zampini         }
282674ae819SStefano Zampini       }
283674ae819SStefano Zampini     }
284674ae819SStefano Zampini     buffer_size = 0;
285674ae819SStefano Zampini     for (i=0;i<graph->n_subsets;i++) {
286674ae819SStefano Zampini       if (subset_cc_adapt[i]) {
287674ae819SStefano Zampini         for (j=i;j<graph->ncc;j++) {
288674ae819SStefano Zampini           if (graph->subset[graph->queue[graph->cptr[j]]] == i+1) { /* WARNING -> subset values goes from 1 to graph->n_subsets included */
289674ae819SStefano Zampini             buffer_size += 1 + graph->cptr[j+1]-graph->cptr[j];
290674ae819SStefano Zampini           }
291674ae819SStefano Zampini         }
292674ae819SStefano Zampini       }
293674ae819SStefano Zampini     }
294674ae819SStefano Zampini     ierr = PetscMalloc(buffer_size*sizeof(*send_buffer),&send_buffer);CHKERRQ(ierr);
295674ae819SStefano Zampini     /* now get from neighbours their ccs (in global numbering) and adapt them (in case it is needed) */
296674ae819SStefano Zampini     ierr = PetscMalloc(graph->nvtxs*sizeof(*queue_in_global_numbering),&queue_in_global_numbering);CHKERRQ(ierr);
297674ae819SStefano Zampini     ierr = ISLocalToGlobalMappingApply(graph->l2gmap,graph->cptr[graph->ncc],graph->queue,queue_in_global_numbering);CHKERRQ(ierr);
298674ae819SStefano Zampini     /* determine how much data to send (size of each queue plus the global indices) and communicate it to neighbours */
299674ae819SStefano Zampini     ierr = PetscMalloc(graph->n_subsets*sizeof(*sizes_of_sends),&sizes_of_sends);CHKERRQ(ierr);
300674ae819SStefano Zampini     ierr = PetscMemzero(sizes_of_sends,graph->n_subsets*sizeof(*sizes_of_sends));CHKERRQ(ierr);
301674ae819SStefano Zampini     sum_requests = 0;
302674ae819SStefano Zampini     start_of_send = 0;
303674ae819SStefano Zampini     start_of_recv = cum_recv_counts[graph->n_subsets];
304674ae819SStefano Zampini     for (i=0;i<graph->n_subsets;i++) {
305674ae819SStefano Zampini       if (subset_cc_adapt[i]) {
306674ae819SStefano Zampini         size_of_send = 0;
307674ae819SStefano Zampini         for (j=i;j<graph->ncc;j++) {
308674ae819SStefano Zampini           if (graph->subset[graph->queue[graph->cptr[j]]] == i+1) { /* WARNING -> subset values goes from 1 to graph->n_subsets included */
309674ae819SStefano Zampini             send_buffer[start_of_send+size_of_send]=graph->cptr[j+1]-graph->cptr[j];
310674ae819SStefano Zampini             size_of_send += 1;
311674ae819SStefano Zampini             ierr = PetscMemcpy(&send_buffer[start_of_send+size_of_send],
312674ae819SStefano Zampini                                &queue_in_global_numbering[graph->cptr[j]],
313674ae819SStefano Zampini                                (graph->cptr[j+1]-graph->cptr[j])*sizeof(*send_buffer));CHKERRQ(ierr);
314674ae819SStefano Zampini             size_of_send = size_of_send+graph->cptr[j+1]-graph->cptr[j];
315674ae819SStefano Zampini           }
316674ae819SStefano Zampini         }
317674ae819SStefano Zampini         j = subset_to_nodes_indices[i];
318674ae819SStefano Zampini         sizes_of_sends[i] = size_of_send;
319674ae819SStefano Zampini         for (k=0;k<graph->count[j];k++) {
320674ae819SStefano Zampini           ierr = PetscMPIIntCast(graph->neighbours_set[j][k],&neigh);CHKERRQ(ierr);
321674ae819SStefano Zampini           ierr = PetscMPIIntCast((PetscInt)(rank+1)*graph->count[j],&tag);CHKERRQ(ierr);
322674ae819SStefano Zampini           ierr = MPI_Isend(&sizes_of_sends[i],1,MPIU_INT,neigh,tag,interface_comm,&send_requests[sum_requests]);CHKERRQ(ierr);
323674ae819SStefano Zampini           ierr = PetscMPIIntCast((PetscInt)(neigh+1)*graph->count[j],&tag);CHKERRQ(ierr);
324674ae819SStefano 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);
325674ae819SStefano Zampini           sum_requests++;
326674ae819SStefano Zampini         }
327674ae819SStefano Zampini         start_of_send += size_of_send;
328674ae819SStefano Zampini       }
329674ae819SStefano Zampini     }
330674ae819SStefano Zampini     ierr = MPI_Waitall(sum_requests,send_requests,MPI_STATUSES_IGNORE);CHKERRQ(ierr);
331674ae819SStefano Zampini     ierr = MPI_Waitall(sum_requests,recv_requests,MPI_STATUSES_IGNORE);CHKERRQ(ierr);
332674ae819SStefano Zampini     buffer_size = 0;
333674ae819SStefano Zampini     for (k=0;k<sum_requests;k++) {
334674ae819SStefano Zampini       buffer_size += recv_buffer_subset[start_of_recv+k];
335674ae819SStefano Zampini     }
336674ae819SStefano Zampini     ierr = PetscMalloc(buffer_size*sizeof(*recv_buffer),&recv_buffer);CHKERRQ(ierr);
337674ae819SStefano Zampini     /* now exchange the data */
338674ae819SStefano Zampini     start_of_recv = 0;
339674ae819SStefano Zampini     start_of_send = 0;
340674ae819SStefano Zampini     sum_requests = 0;
341674ae819SStefano Zampini     for (i=0;i<graph->n_subsets;i++) {
342674ae819SStefano Zampini       if (subset_cc_adapt[i]) {
343674ae819SStefano Zampini         size_of_send = sizes_of_sends[i];
344674ae819SStefano Zampini         j = subset_to_nodes_indices[i];
345674ae819SStefano Zampini         for (k=0;k<graph->count[j];k++) {
346674ae819SStefano Zampini           ierr = PetscMPIIntCast(graph->neighbours_set[j][k],&neigh);CHKERRQ(ierr);
347674ae819SStefano Zampini           ierr = PetscMPIIntCast((PetscInt)(rank+1)*graph->count[j],&tag);CHKERRQ(ierr);
348674ae819SStefano Zampini           ierr = PetscMPIIntCast(size_of_send,&mpi_buffer_size);CHKERRQ(ierr);
349674ae819SStefano Zampini           ierr = MPI_Isend(&send_buffer[start_of_send],mpi_buffer_size,MPIU_INT,neigh,tag,interface_comm,&send_requests[sum_requests]);CHKERRQ(ierr);
350674ae819SStefano Zampini           size_of_recv = recv_buffer_subset[cum_recv_counts[graph->n_subsets]+sum_requests];
351674ae819SStefano Zampini           ierr = PetscMPIIntCast((PetscInt)(neigh+1)*graph->count[j],&tag);CHKERRQ(ierr);
352674ae819SStefano Zampini           ierr = PetscMPIIntCast(size_of_recv,&mpi_buffer_size);CHKERRQ(ierr);
353674ae819SStefano Zampini           ierr = MPI_Irecv(&recv_buffer[start_of_recv],mpi_buffer_size,MPIU_INT,neigh,tag,interface_comm,&recv_requests[sum_requests]);CHKERRQ(ierr);
354674ae819SStefano Zampini           start_of_recv += size_of_recv;
355674ae819SStefano Zampini           sum_requests++;
356674ae819SStefano Zampini         }
357674ae819SStefano Zampini         start_of_send += size_of_send;
358674ae819SStefano Zampini       }
359674ae819SStefano Zampini     }
360674ae819SStefano Zampini     ierr = MPI_Waitall(sum_requests,recv_requests,MPI_STATUSES_IGNORE);CHKERRQ(ierr);
361674ae819SStefano Zampini     ierr = MPI_Waitall(sum_requests,send_requests,MPI_STATUSES_IGNORE);CHKERRQ(ierr);
362674ae819SStefano Zampini     for (j=0;j<buffer_size;) {
363674ae819SStefano Zampini        ierr = ISGlobalToLocalMappingApply(graph->l2gmap,IS_GTOLM_MASK,recv_buffer[j],&recv_buffer[j+1],&recv_buffer[j],&recv_buffer[j+1]);CHKERRQ(ierr);
364674ae819SStefano Zampini        k = recv_buffer[j]+1;
365674ae819SStefano Zampini        j += k;
366674ae819SStefano Zampini     }
367674ae819SStefano Zampini     sum_requests = cum_recv_counts[graph->n_subsets];
368674ae819SStefano Zampini     start_of_recv = 0;
369674ae819SStefano Zampini     ierr = PetscMalloc(graph->nvtxs*sizeof(*nodes_to_temp_buffer_indices),&nodes_to_temp_buffer_indices);CHKERRQ(ierr);
370674ae819SStefano Zampini     global_subset_counter = 0;
371674ae819SStefano Zampini     for (i=0;i<graph->n_subsets;i++) {
372674ae819SStefano Zampini       if (subset_cc_adapt[i]) {
373674ae819SStefano Zampini         temp_buffer_size = 0;
374674ae819SStefano Zampini         /* find nodes on the shared interface we need to adapt */
375674ae819SStefano Zampini         for (j=0;j<graph->nvtxs;j++) {
376674ae819SStefano Zampini           if (graph->subset[j]==i+1) {
377674ae819SStefano Zampini             nodes_to_temp_buffer_indices[j] = temp_buffer_size;
378674ae819SStefano Zampini             temp_buffer_size++;
379674ae819SStefano Zampini           } else {
380674ae819SStefano Zampini             nodes_to_temp_buffer_indices[j] = -1;
381674ae819SStefano Zampini           }
382674ae819SStefano Zampini         }
383674ae819SStefano Zampini         /* allocate some temporary space */
384674ae819SStefano Zampini         ierr = PetscMalloc(temp_buffer_size*sizeof(PetscInt*),&temp_buffer);CHKERRQ(ierr);
385674ae819SStefano Zampini         ierr = PetscMalloc(temp_buffer_size*(cum_recv_counts[i+1]-cum_recv_counts[i])*sizeof(PetscInt),&temp_buffer[0]);CHKERRQ(ierr);
386674ae819SStefano Zampini         ierr = PetscMemzero(temp_buffer[0],temp_buffer_size*(cum_recv_counts[i+1]-cum_recv_counts[i])*sizeof(PetscInt));CHKERRQ(ierr);
387674ae819SStefano Zampini         for (j=1;j<temp_buffer_size;j++) {
388674ae819SStefano Zampini           temp_buffer[j] = temp_buffer[j-1]+cum_recv_counts[i+1]-cum_recv_counts[i];
389674ae819SStefano Zampini         }
390674ae819SStefano Zampini         /* analyze contributions from neighbouring subdomains for i-th conn comp
391674ae819SStefano Zampini            temp buffer structure:
392674ae819SStefano Zampini            supposing part of the interface has dimension 5 (for example with global dofs 0,1,2,3,4)
393674ae819SStefano Zampini            3 neighs procs with structured connected components:
394674ae819SStefano Zampini              neigh 0: [0 1 4], [2 3];  (2 connected components)
395674ae819SStefano Zampini              neigh 1: [0 1], [2 3 4];  (2 connected components)
396674ae819SStefano Zampini              neigh 2: [0 4], [1], [2 3]; (3 connected components)
397674ae819SStefano Zampini            tempbuffer (row-oriented) will be filled as:
398674ae819SStefano Zampini              [ 0, 0, 0;
399674ae819SStefano Zampini                0, 0, 1;
400674ae819SStefano Zampini                1, 1, 2;
401674ae819SStefano Zampini                1, 1, 2;
402674ae819SStefano Zampini                0, 1, 0; ];
403674ae819SStefano Zampini            This way we can simply find intersections of ccs among neighs.
404674ae819SStefano Zampini            For the example above, the graph->subset array will be modified to reproduce the following 4 connected components [0], [1], [2 3], [4];
405674ae819SStefano Zampini                                                                                                                                    */
406674ae819SStefano Zampini         for (j=0;j<cum_recv_counts[i+1]-cum_recv_counts[i];j++) {
407674ae819SStefano Zampini           ins_val = 0;
408674ae819SStefano Zampini           size_of_recv = recv_buffer_subset[sum_requests];  /* total size of recv from neighs */
409674ae819SStefano Zampini           for (buffer_size=0;buffer_size<size_of_recv;) {  /* loop until all data from neighs has been taken into account */
410674ae819SStefano Zampini             for (k=1;k<recv_buffer[buffer_size+start_of_recv]+1;k++) { /* filling properly temp_buffer using data from a single recv */
411674ae819SStefano Zampini               temp_buffer[nodes_to_temp_buffer_indices[recv_buffer[start_of_recv+buffer_size+k]]][j] = ins_val;
412674ae819SStefano Zampini             }
413674ae819SStefano Zampini             buffer_size += k;
414674ae819SStefano Zampini             ins_val++;
415674ae819SStefano Zampini           }
416674ae819SStefano Zampini           start_of_recv += size_of_recv;
417674ae819SStefano Zampini           sum_requests++;
418674ae819SStefano Zampini         }
419674ae819SStefano Zampini         ierr = PetscMalloc(temp_buffer_size*sizeof(*add_to_subset),&add_to_subset);CHKERRQ(ierr);
420674ae819SStefano Zampini         ierr = PetscMemzero(add_to_subset,temp_buffer_size*sizeof(*add_to_subset));CHKERRQ(ierr);
421674ae819SStefano Zampini         for (j=0;j<temp_buffer_size;j++) {
422674ae819SStefano Zampini           if (!add_to_subset[j]) { /* found a new cc  */
423674ae819SStefano Zampini             global_subset_counter++;
424674ae819SStefano Zampini             add_to_subset[j] = global_subset_counter;
425674ae819SStefano Zampini             for (k=j+1;k<temp_buffer_size;k++) { /* check for other nodes in new cc */
426674ae819SStefano Zampini               same_set = PETSC_TRUE;
427674ae819SStefano Zampini               for (s=0;s<cum_recv_counts[i+1]-cum_recv_counts[i];s++) {
428674ae819SStefano Zampini                 if (temp_buffer[j][s]!=temp_buffer[k][s]) {
429674ae819SStefano Zampini                   same_set = PETSC_FALSE;
430674ae819SStefano Zampini                   break;
431674ae819SStefano Zampini                 }
432674ae819SStefano Zampini               }
433674ae819SStefano Zampini               if (same_set) {
434674ae819SStefano Zampini                 add_to_subset[k] = global_subset_counter;
435674ae819SStefano Zampini               }
436674ae819SStefano Zampini             }
437674ae819SStefano Zampini           }
438674ae819SStefano Zampini         }
439674ae819SStefano Zampini         /* insert new data in subset array */
440674ae819SStefano Zampini         temp_buffer_size = 0;
441674ae819SStefano Zampini         for (j=0;j<graph->nvtxs;j++) {
442674ae819SStefano Zampini           if (graph->subset[j]==i+1) {
443674ae819SStefano Zampini             graph->subset[j] = graph->n_subsets+add_to_subset[temp_buffer_size];
444674ae819SStefano Zampini             temp_buffer_size++;
445674ae819SStefano Zampini           }
446674ae819SStefano Zampini         }
447674ae819SStefano Zampini         ierr = PetscFree(temp_buffer[0]);CHKERRQ(ierr);
448674ae819SStefano Zampini         ierr = PetscFree(temp_buffer);CHKERRQ(ierr);
449674ae819SStefano Zampini         ierr = PetscFree(add_to_subset);CHKERRQ(ierr);
450674ae819SStefano Zampini       }
451674ae819SStefano Zampini     }
452674ae819SStefano Zampini     ierr = PetscFree(nodes_to_temp_buffer_indices);CHKERRQ(ierr);
453674ae819SStefano Zampini     ierr = PetscFree(sizes_of_sends);CHKERRQ(ierr);
454674ae819SStefano Zampini     ierr = PetscFree(send_requests);CHKERRQ(ierr);
455674ae819SStefano Zampini     ierr = PetscFree(recv_requests);CHKERRQ(ierr);
456674ae819SStefano Zampini     ierr = PetscFree(recv_buffer);CHKERRQ(ierr);
457674ae819SStefano Zampini     ierr = PetscFree(recv_buffer_subset);CHKERRQ(ierr);
458674ae819SStefano Zampini     ierr = PetscFree(send_buffer);CHKERRQ(ierr);
459674ae819SStefano Zampini     ierr = PetscFree(cum_recv_counts);CHKERRQ(ierr);
460674ae819SStefano Zampini     ierr = PetscFree(subset_to_nodes_indices);CHKERRQ(ierr);
461674ae819SStefano Zampini     ierr = PetscFree(subset_cc_adapt);CHKERRQ(ierr);
462674ae819SStefano Zampini     /* We are ready to find for connected components consistent among neighbouring subdomains */
463674ae819SStefano Zampini     if (global_subset_counter) {
464674ae819SStefano Zampini       ierr = PetscMemzero(graph->touched,graph->nvtxs*sizeof(*graph->touched));CHKERRQ(ierr);
465674ae819SStefano Zampini       global_subset_counter = 0;
466674ae819SStefano Zampini       for (i=0;i<graph->nvtxs;i++) {
467674ae819SStefano Zampini         if (graph->subset[i] && !graph->touched[i]) {
468674ae819SStefano Zampini           global_subset_counter++;
469674ae819SStefano Zampini           for (j=i+1;j<graph->nvtxs;j++) {
470674ae819SStefano Zampini             if (!graph->touched[j] && graph->subset[j]==graph->subset[i]) {
471674ae819SStefano Zampini               graph->subset[j] = global_subset_counter;
472674ae819SStefano Zampini               graph->touched[j] = PETSC_TRUE;
473674ae819SStefano Zampini             }
474674ae819SStefano Zampini           }
475674ae819SStefano Zampini           graph->subset[i] = global_subset_counter;
476674ae819SStefano Zampini           graph->touched[i] = PETSC_TRUE;
477674ae819SStefano Zampini         }
478674ae819SStefano Zampini       }
479674ae819SStefano Zampini       /* refine connected components locally */
480674ae819SStefano Zampini       ierr = PCBDDCGraphComputeConnectedComponentsLocal(graph);CHKERRQ(ierr);
481674ae819SStefano Zampini     }
482674ae819SStefano Zampini     ierr = PetscFree(queue_in_global_numbering);CHKERRQ(ierr);
483674ae819SStefano Zampini   }
484674ae819SStefano Zampini   PetscFunctionReturn(0);
485674ae819SStefano Zampini }
486674ae819SStefano Zampini 
487674ae819SStefano Zampini /* The following code has been adapted from function IsConnectedSubdomain contained
488674ae819SStefano Zampini    in source file contig.c of METIS library (version 5.0.1)
489674ae819SStefano Zampini    It finds connected components for each subset  */
490674ae819SStefano Zampini #undef __FUNCT__
491674ae819SStefano Zampini #define __FUNCT__ "PCBDDCGraphComputeConnectedComponentsLocal"
492674ae819SStefano Zampini PetscErrorCode PCBDDCGraphComputeConnectedComponentsLocal(PCBDDCGraph graph)
493674ae819SStefano Zampini {
494674ae819SStefano Zampini   PetscInt       i,j,k,first,last,nleft,ncc,pid,cum_queue,n,ncc_pid;
495674ae819SStefano Zampini   PetscInt       *queue_global;
496674ae819SStefano Zampini   PetscErrorCode ierr;
497674ae819SStefano Zampini 
498674ae819SStefano Zampini   PetscFunctionBegin;
499674ae819SStefano Zampini   /* quiet return if no csr info is available */
500674ae819SStefano Zampini   if (!graph->xadj || !graph->adjncy) {
501674ae819SStefano Zampini     PetscFunctionReturn(0);
502674ae819SStefano Zampini   }
503674ae819SStefano Zampini 
504674ae819SStefano Zampini   /* reset any previous search of connected components */
505674ae819SStefano Zampini   ierr = PetscMemzero(graph->touched,graph->nvtxs*sizeof(*graph->touched));CHKERRQ(ierr);
506674ae819SStefano Zampini   graph->n_subsets = 0;
507674ae819SStefano Zampini   for (i=0;i<graph->nvtxs;i++) {
508674ae819SStefano Zampini     if (graph->which_dof[i] == DIRICHLET_MARK || !graph->count[i]) {
509674ae819SStefano Zampini       graph->touched[i] = PETSC_TRUE;
510674ae819SStefano Zampini       graph->subset[i] = 0;
511674ae819SStefano Zampini     }
512674ae819SStefano Zampini     graph->n_subsets = PetscMax(graph->n_subsets,graph->subset[i]);
513674ae819SStefano Zampini   }
514674ae819SStefano Zampini   ierr = PetscFree(graph->subset_ncc);CHKERRQ(ierr);
515674ae819SStefano Zampini   ierr = PetscMalloc(graph->n_subsets*sizeof(*graph->subset_ncc),&graph->subset_ncc);CHKERRQ(ierr);
516674ae819SStefano Zampini   ierr = PetscMemzero(graph->subset_ncc,graph->n_subsets*sizeof(*graph->subset_ncc));CHKERRQ(ierr);
517674ae819SStefano Zampini   ierr = PetscMemzero(graph->cptr,(graph->nvtxs+1)*sizeof(*graph->cptr));CHKERRQ(ierr);
518674ae819SStefano Zampini   ierr = PetscMemzero(graph->queue,graph->nvtxs*sizeof(*graph->queue));CHKERRQ(ierr);
519674ae819SStefano Zampini 
520674ae819SStefano Zampini   /* begin search for connected components */
521674ae819SStefano Zampini   cum_queue = 0;
522674ae819SStefano Zampini   ncc = 0;
523674ae819SStefano Zampini   for (n=0;n<graph->n_subsets;n++) {
524674ae819SStefano Zampini     pid = n+1;  /* partition labeled by 0 is discarded */
525674ae819SStefano Zampini     nleft = 0;
526674ae819SStefano Zampini     for (i=0;i<graph->nvtxs;i++) {
527674ae819SStefano Zampini       if (graph->subset[i] == pid) {
528674ae819SStefano Zampini         nleft++;
529674ae819SStefano Zampini       }
530674ae819SStefano Zampini     }
531674ae819SStefano Zampini     for (i=0; i<graph->nvtxs; i++) {
532674ae819SStefano Zampini       if (graph->subset[i] == pid) {
533674ae819SStefano Zampini         break;
534674ae819SStefano Zampini       }
535674ae819SStefano Zampini     }
536674ae819SStefano Zampini     graph->touched[i] = PETSC_TRUE;
537674ae819SStefano Zampini     graph->queue[cum_queue] = i;
538674ae819SStefano Zampini     first = 0;
539674ae819SStefano Zampini     last = 1;
540674ae819SStefano Zampini     graph->cptr[ncc] = cum_queue;
541674ae819SStefano Zampini     ncc_pid = 0;
542674ae819SStefano Zampini     while (first != nleft) {
543674ae819SStefano Zampini       if (first == last) {
544674ae819SStefano Zampini         graph->cptr[++ncc] = first+cum_queue;
545674ae819SStefano Zampini         ncc_pid++;
546674ae819SStefano Zampini         for (i=0; i<graph->nvtxs; i++) {
547674ae819SStefano Zampini           if (graph->subset[i] == pid && !graph->touched[i]) {
548674ae819SStefano Zampini             break;
549674ae819SStefano Zampini           }
550674ae819SStefano Zampini         }
551674ae819SStefano Zampini         graph->queue[cum_queue+last] = i;
552674ae819SStefano Zampini         last++;
553674ae819SStefano Zampini         graph->touched[i] = PETSC_TRUE;
554674ae819SStefano Zampini       }
555674ae819SStefano Zampini       i = graph->queue[cum_queue+first];
556674ae819SStefano Zampini       first++;
557674ae819SStefano Zampini       for (j=graph->xadj[i];j<graph->xadj[i+1];j++) {
558674ae819SStefano Zampini         k = graph->adjncy[j];
559674ae819SStefano Zampini         if (graph->subset[k] == pid && !graph->touched[k]) {
560674ae819SStefano Zampini           graph->queue[cum_queue+last] = k;
561674ae819SStefano Zampini           last++;
562674ae819SStefano Zampini           graph->touched[k] = PETSC_TRUE;
563674ae819SStefano Zampini         }
564674ae819SStefano Zampini       }
565674ae819SStefano Zampini     }
566674ae819SStefano Zampini     graph->cptr[++ncc] = first+cum_queue;
567674ae819SStefano Zampini     ncc_pid++;
568674ae819SStefano Zampini     cum_queue = graph->cptr[ncc];
569674ae819SStefano Zampini     graph->subset_ncc[n] = ncc_pid;
570674ae819SStefano Zampini   }
571674ae819SStefano Zampini   graph->ncc = ncc;
572674ae819SStefano Zampini   /* For consistency among neighbours, I need to sort (by global ordering) each connected component */
573674ae819SStefano Zampini   ierr = PetscMalloc(graph->cptr[graph->ncc]*sizeof(*queue_global),&queue_global);CHKERRQ(ierr);
574674ae819SStefano Zampini   ierr = ISLocalToGlobalMappingApply(graph->l2gmap,graph->cptr[graph->ncc],graph->queue,queue_global);CHKERRQ(ierr);
575674ae819SStefano Zampini   for (i=0;i<graph->ncc;i++) {
576674ae819SStefano Zampini     ierr = PetscSortIntWithArray(graph->cptr[i+1]-graph->cptr[i],&queue_global[graph->cptr[i]],&graph->queue[graph->cptr[i]]);CHKERRQ(ierr);
577674ae819SStefano Zampini   }
578674ae819SStefano Zampini   ierr = PetscFree(queue_global);CHKERRQ(ierr);
579674ae819SStefano Zampini   PetscFunctionReturn(0);
580674ae819SStefano Zampini }
581674ae819SStefano Zampini 
582674ae819SStefano Zampini #undef __FUNCT__
583674ae819SStefano Zampini #define __FUNCT__ "PCBDDCGraphSetUp"
584674ae819SStefano Zampini PetscErrorCode PCBDDCGraphSetUp(PCBDDCGraph graph, PetscInt custom_minimal_size, IS neumann_is, IS dirichlet_is, PetscInt n_ISForDofs, IS ISForDofs[], IS custom_primal_vertices)
585674ae819SStefano Zampini {
586674ae819SStefano Zampini   VecScatter     scatter_ctx;
587674ae819SStefano Zampini   Vec            local_vec,local_vec2,global_vec;
588674ae819SStefano Zampini   IS             to,from;
589674ae819SStefano Zampini   MPI_Comm       comm;
590674ae819SStefano Zampini   PetscScalar    *array,*array2;
591674ae819SStefano Zampini   const PetscInt *is_indices;
592674ae819SStefano Zampini   PetscInt       n_neigh,*neigh,*n_shared,**shared;
593674ae819SStefano Zampini   PetscInt       i,j,k,s,total_counts,nodes_touched,is_size;
594674ae819SStefano Zampini   PetscErrorCode ierr;
595674ae819SStefano Zampini   PetscBool      same_set;
596674ae819SStefano Zampini 
597674ae819SStefano Zampini   PetscFunctionBegin;
598674ae819SStefano Zampini   ierr = PetscObjectGetComm((PetscObject)(graph->l2gmap),&comm);CHKERRQ(ierr);
599674ae819SStefano Zampini   /* custom_minimal_size */
600674ae819SStefano Zampini   graph->custom_minimal_size = PetscMax(graph->custom_minimal_size,custom_minimal_size);
601674ae819SStefano Zampini   /* get info l2gmap and allocate work vectors  */
602674ae819SStefano Zampini   ierr = ISLocalToGlobalMappingGetInfo(graph->l2gmap,&n_neigh,&neigh,&n_shared,&shared);CHKERRQ(ierr);
603674ae819SStefano Zampini   ierr = ISLocalToGlobalMappingGetIndices(graph->l2gmap,&is_indices);CHKERRQ(ierr);
604674ae819SStefano Zampini   j = 0;
605674ae819SStefano Zampini   for (i=0;i<graph->nvtxs;i++) {
606674ae819SStefano Zampini     j = PetscMax(j,is_indices[i]);
607674ae819SStefano Zampini   }
608674ae819SStefano Zampini   ierr = MPI_Allreduce(&j,&i,1,MPIU_INT,MPI_MAX,comm);CHKERRQ(ierr);
609674ae819SStefano Zampini   i++;
610674ae819SStefano Zampini   ierr = VecCreate(PETSC_COMM_SELF,&local_vec);CHKERRQ(ierr);
611674ae819SStefano Zampini   ierr = VecSetSizes(local_vec,PETSC_DECIDE,graph->nvtxs);CHKERRQ(ierr);
612674ae819SStefano Zampini   ierr = VecSetType(local_vec,VECSTANDARD);CHKERRQ(ierr);
613674ae819SStefano Zampini   ierr = VecDuplicate(local_vec,&local_vec2);CHKERRQ(ierr);
614674ae819SStefano Zampini   ierr = VecCreate(comm,&global_vec);CHKERRQ(ierr);
615674ae819SStefano Zampini   ierr = VecSetSizes(global_vec,PETSC_DECIDE,i);CHKERRQ(ierr);
616674ae819SStefano Zampini   ierr = VecSetType(global_vec,VECSTANDARD);CHKERRQ(ierr);
617674ae819SStefano Zampini   ierr = ISCreateStride(PETSC_COMM_SELF,graph->nvtxs,0,1,&to);CHKERRQ(ierr);
618674ae819SStefano Zampini   ierr = ISLocalToGlobalMappingApplyIS(graph->l2gmap,to,&from);CHKERRQ(ierr);
619674ae819SStefano Zampini   ierr = VecScatterCreate(global_vec,from,local_vec,to,&scatter_ctx);CHKERRQ(ierr);
620674ae819SStefano Zampini   ierr = ISDestroy(&to);CHKERRQ(ierr);
621674ae819SStefano Zampini   ierr = ISDestroy(&from);CHKERRQ(ierr);
622674ae819SStefano Zampini   /* Count total number of neigh per node */
623674ae819SStefano Zampini   k=0;
624674ae819SStefano Zampini   for (i=1;i<n_neigh;i++){
625674ae819SStefano Zampini     k += n_shared[i];
626674ae819SStefano Zampini     for (j=0;j<n_shared[i];j++){
627674ae819SStefano Zampini       graph->count[shared[i][j]] += 1;
628674ae819SStefano Zampini     }
629674ae819SStefano Zampini   }
630674ae819SStefano Zampini   /* Allocate space for storing the set of neighbours for each node */
631674ae819SStefano Zampini   if (graph->nvtxs) {
632674ae819SStefano Zampini     ierr = PetscMalloc(k*sizeof(PetscInt),&graph->neighbours_set[0]);CHKERRQ(ierr);
633674ae819SStefano Zampini   }
634674ae819SStefano Zampini   for (i=1;i<graph->nvtxs;i++) { /* dont count myself */
635674ae819SStefano Zampini     graph->neighbours_set[i]=graph->neighbours_set[i-1]+graph->count[i-1];
636674ae819SStefano Zampini   }
637674ae819SStefano Zampini   /* Get information for sharing subdomains */
638674ae819SStefano Zampini   ierr = PetscMemzero(graph->count,graph->nvtxs*sizeof(*graph->count));CHKERRQ(ierr);
639674ae819SStefano Zampini   for (i=1;i<n_neigh;i++) { /* dont count myself */
640674ae819SStefano Zampini     s = n_shared[i];
641674ae819SStefano Zampini     for (j=0;j<s;j++) {
642674ae819SStefano Zampini       k = shared[i][j];
643674ae819SStefano Zampini       graph->neighbours_set[k][graph->count[k]] = neigh[i];
644674ae819SStefano Zampini       graph->count[k] += 1;
645674ae819SStefano Zampini     }
646674ae819SStefano Zampini   }
647674ae819SStefano Zampini   /* sort set of sharing subdomains */
648674ae819SStefano Zampini   for (i=0;i<graph->nvtxs;i++) {
649674ae819SStefano Zampini     ierr = PetscSortInt(graph->count[i],graph->neighbours_set[i]);CHKERRQ(ierr);
650674ae819SStefano Zampini   }
651674ae819SStefano Zampini   /* Get info for dofs splitting */
652674ae819SStefano Zampini   for (i=0;i<n_ISForDofs;i++) {
653674ae819SStefano Zampini     ierr = ISGetSize(ISForDofs[i],&is_size);CHKERRQ(ierr);
654674ae819SStefano Zampini     ierr = ISGetIndices(ISForDofs[i],(const PetscInt**)&is_indices);CHKERRQ(ierr);
655674ae819SStefano Zampini     for (j=0;j<is_size;j++) {
656674ae819SStefano Zampini       graph->which_dof[is_indices[j]]=i;
657674ae819SStefano Zampini     }
658674ae819SStefano Zampini     ierr = ISRestoreIndices(ISForDofs[i],(const PetscInt**)&is_indices);CHKERRQ(ierr);
659674ae819SStefano Zampini   }
660674ae819SStefano Zampini   /* Take into account Neumann nodes */
661674ae819SStefano Zampini   ierr = VecSet(local_vec,0.0);CHKERRQ(ierr);
662674ae819SStefano Zampini   ierr = VecSet(local_vec2,0.0);CHKERRQ(ierr);
663674ae819SStefano Zampini   if (neumann_is) {
664674ae819SStefano Zampini     ierr = VecGetArray(local_vec,&array);CHKERRQ(ierr);
665674ae819SStefano Zampini     ierr = ISGetSize(neumann_is,&is_size);CHKERRQ(ierr);
666674ae819SStefano Zampini     ierr = ISGetIndices(neumann_is,(const PetscInt**)&is_indices);CHKERRQ(ierr);
667674ae819SStefano Zampini     for (i=0;i<is_size;i++) {
668674ae819SStefano Zampini       array[is_indices[i]] = 1.0;
669674ae819SStefano Zampini     }
670674ae819SStefano Zampini     ierr = ISRestoreIndices(neumann_is,(const PetscInt**)&is_indices);CHKERRQ(ierr);
671674ae819SStefano Zampini     ierr = VecRestoreArray(local_vec,&array);CHKERRQ(ierr);
672674ae819SStefano Zampini   }
673674ae819SStefano Zampini   /* Neumann nodes: impose consistency among neighbours */
674674ae819SStefano Zampini   ierr = VecSet(global_vec,0.0);CHKERRQ(ierr);
675674ae819SStefano Zampini   ierr = VecScatterBegin(scatter_ctx,local_vec,global_vec,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
676674ae819SStefano Zampini   ierr = VecScatterEnd(scatter_ctx,local_vec,global_vec,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
677674ae819SStefano Zampini   ierr = VecScatterBegin(scatter_ctx,global_vec,local_vec,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
678674ae819SStefano Zampini   ierr = VecScatterEnd(scatter_ctx,global_vec,local_vec,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
679674ae819SStefano Zampini   ierr = VecGetArray(local_vec,&array);CHKERRQ(ierr);
680674ae819SStefano Zampini   for (i=0;i<graph->nvtxs;i++) {
681674ae819SStefano Zampini     if (array[i] > 0.0) {
682674ae819SStefano Zampini       graph->which_dof[i] = NEUMANN_MARK;
683674ae819SStefano Zampini     }
684674ae819SStefano Zampini   }
685674ae819SStefano Zampini   ierr = VecRestoreArray(local_vec,&array);CHKERRQ(ierr);
686674ae819SStefano Zampini   /* Take into account Dirichlet nodes */
687674ae819SStefano Zampini   ierr = VecSet(local_vec2,0.0);CHKERRQ(ierr);
688674ae819SStefano Zampini   if (dirichlet_is) {
689674ae819SStefano Zampini     ierr = VecGetArray(local_vec,&array);CHKERRQ(ierr);
690674ae819SStefano Zampini     ierr = VecGetArray(local_vec2,&array2);CHKERRQ(ierr);
691674ae819SStefano Zampini     ierr = ISGetSize(dirichlet_is,&is_size);CHKERRQ(ierr);
692674ae819SStefano Zampini     ierr = ISGetIndices(dirichlet_is,(const PetscInt**)&is_indices);CHKERRQ(ierr);
693674ae819SStefano Zampini     for (i=0;i<is_size;i++){
694674ae819SStefano Zampini       k = is_indices[i];
695674ae819SStefano Zampini       if (graph->count[k] && !graph->touched[k]) {
696674ae819SStefano Zampini         if (array[k] > 0.0) {
697674ae819SStefano 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);
698674ae819SStefano Zampini         }
699674ae819SStefano Zampini         array2[k] = 1.0;
700674ae819SStefano Zampini       }
701674ae819SStefano Zampini     }
702674ae819SStefano Zampini     ierr = ISRestoreIndices(dirichlet_is,(const PetscInt**)&is_indices);CHKERRQ(ierr);
703674ae819SStefano Zampini     ierr = VecRestoreArray(local_vec,&array);CHKERRQ(ierr);
704674ae819SStefano Zampini     ierr = VecRestoreArray(local_vec2,&array2);CHKERRQ(ierr);
705674ae819SStefano Zampini   }
706674ae819SStefano Zampini   /* Dirichlet nodes: impose consistency among neighbours */
707674ae819SStefano Zampini   ierr = VecSet(global_vec,0.0);CHKERRQ(ierr);
708674ae819SStefano Zampini   ierr = VecScatterBegin(scatter_ctx,local_vec2,global_vec,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
709674ae819SStefano Zampini   ierr = VecScatterEnd(scatter_ctx,local_vec2,global_vec,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
710674ae819SStefano Zampini   ierr = VecScatterBegin(scatter_ctx,global_vec,local_vec,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
711674ae819SStefano Zampini   ierr = VecScatterEnd(scatter_ctx,global_vec,local_vec,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
712674ae819SStefano Zampini   ierr = VecGetArray(local_vec,&array);CHKERRQ(ierr);
713674ae819SStefano Zampini   for (i=0;i<graph->nvtxs;i++) {
714674ae819SStefano Zampini     if (array[i] > 0.0) {
715674ae819SStefano Zampini       graph->touched[i] = PETSC_TRUE;
716674ae819SStefano Zampini       graph->subset[i] = 0; /* dirichlet nodes treated as internal -> is it ok? */
717674ae819SStefano Zampini       graph->which_dof[i] = DIRICHLET_MARK;
718674ae819SStefano Zampini     }
719674ae819SStefano Zampini   }
720674ae819SStefano Zampini   ierr = VecRestoreArray(local_vec,&array);CHKERRQ(ierr);
721674ae819SStefano Zampini   /* mark special nodes -> each will become a single node equivalence class */
722674ae819SStefano Zampini   if (custom_primal_vertices) {
723674ae819SStefano Zampini     ierr = ISGetSize(custom_primal_vertices,&is_size);CHKERRQ(ierr);
724674ae819SStefano Zampini     ierr = ISGetIndices(custom_primal_vertices,(const PetscInt**)&is_indices);CHKERRQ(ierr);
725674ae819SStefano Zampini     for (i=0;i<is_size;i++) {
726674ae819SStefano Zampini       graph->which_dof[is_indices[i]] = SPECIAL_MARK-i;
727674ae819SStefano Zampini     }
728674ae819SStefano Zampini     ierr = ISRestoreIndices(custom_primal_vertices,(const PetscInt**)&is_indices);CHKERRQ(ierr);
729674ae819SStefano Zampini   }
730674ae819SStefano Zampini   /* mark interior nodes as touched and belonging to partition number 0 */
731674ae819SStefano Zampini   for (i=0;i<graph->nvtxs;i++) {
732674ae819SStefano Zampini     if (!graph->count[i]) {
733674ae819SStefano Zampini       graph->touched[i] = PETSC_TRUE;
734674ae819SStefano Zampini       graph->subset[i] = 0;
735674ae819SStefano Zampini     }
736674ae819SStefano Zampini   }
737674ae819SStefano Zampini   /* init graph structure and compute default subsets */
738674ae819SStefano Zampini   nodes_touched=0;
739674ae819SStefano Zampini   for (i=0;i<graph->nvtxs;i++) {
740674ae819SStefano Zampini     if (graph->touched[i]) {
741674ae819SStefano Zampini       nodes_touched++;
742674ae819SStefano Zampini     }
743674ae819SStefano Zampini   }
744674ae819SStefano Zampini   i = 0;
745674ae819SStefano Zampini   graph->ncc = 0;
746674ae819SStefano Zampini   total_counts = 0;
747674ae819SStefano Zampini   while (nodes_touched<graph->nvtxs) {
748674ae819SStefano Zampini     /*  find first untouched node in local ordering */
749674ae819SStefano Zampini     while (graph->touched[i]) {
750674ae819SStefano Zampini       i++;
751674ae819SStefano Zampini     }
752674ae819SStefano Zampini     graph->touched[i] = PETSC_TRUE;
753674ae819SStefano Zampini     graph->subset[i] = graph->ncc+1;
754674ae819SStefano Zampini     graph->cptr[graph->ncc] = total_counts;
755674ae819SStefano Zampini     graph->queue[total_counts] = i;
756674ae819SStefano Zampini     total_counts++;
757674ae819SStefano Zampini     nodes_touched++;
758674ae819SStefano Zampini     /* now find all other nodes having the same set of sharing subdomains */
759674ae819SStefano Zampini     for (j=i+1;j<graph->nvtxs;j++){
760674ae819SStefano Zampini       /* check for same number of sharing subdomains and dof number */
761674ae819SStefano Zampini       if (!graph->touched[j] && graph->count[i]==graph->count[j] && graph->which_dof[i] == graph->which_dof[j] ){
762674ae819SStefano Zampini         /* check for same set of sharing subdomains */
763674ae819SStefano Zampini         same_set=PETSC_TRUE;
764674ae819SStefano Zampini         for (k=0;k<graph->count[j];k++){
765674ae819SStefano Zampini           if (graph->neighbours_set[i][k]!=graph->neighbours_set[j][k]) {
766674ae819SStefano Zampini             same_set=PETSC_FALSE;
767674ae819SStefano Zampini           }
768674ae819SStefano Zampini         }
769674ae819SStefano Zampini         /* I found a friend of mine */
770674ae819SStefano Zampini         if (same_set) {
771674ae819SStefano Zampini           graph->subset[j]=graph->ncc+1;
772674ae819SStefano Zampini           graph->touched[j]=PETSC_TRUE;
773674ae819SStefano Zampini           nodes_touched++;
774674ae819SStefano Zampini           graph->queue[total_counts] = j;
775674ae819SStefano Zampini           total_counts++;
776674ae819SStefano Zampini         }
777674ae819SStefano Zampini       }
778674ae819SStefano Zampini     }
779674ae819SStefano Zampini     graph->ncc++;
780674ae819SStefano Zampini   }
781674ae819SStefano Zampini   /* set default number of subsets (at this point no info on csr graph has been taken into account, so n_subsets = ncc */
782674ae819SStefano Zampini   graph->n_subsets = graph->ncc;
783674ae819SStefano Zampini   ierr = PetscMalloc(graph->n_subsets*sizeof(*graph->subset_ncc),&graph->subset_ncc);CHKERRQ(ierr);
784674ae819SStefano Zampini   for (i=0;i<graph->n_subsets;i++) {
785674ae819SStefano Zampini     graph->subset_ncc[i] = 1;
786674ae819SStefano Zampini   }
787674ae819SStefano Zampini   /* final pointer */
788674ae819SStefano Zampini   graph->cptr[graph->ncc] = total_counts;
789674ae819SStefano Zampini   /* free memory allocated by ISLocalToGlobalMappingGetInfo */
790674ae819SStefano Zampini   ierr = ISLocalToGlobalMappingRestoreInfo(graph->l2gmap,&n_neigh,&neigh,&n_shared,&shared);CHKERRQ(ierr);
791674ae819SStefano Zampini   /* free objects */
792674ae819SStefano Zampini   ierr = VecDestroy(&local_vec);CHKERRQ(ierr);
793674ae819SStefano Zampini   ierr = VecDestroy(&local_vec2);CHKERRQ(ierr);
794674ae819SStefano Zampini   ierr = VecDestroy(&global_vec);CHKERRQ(ierr);
795674ae819SStefano Zampini   ierr = VecScatterDestroy(&scatter_ctx);CHKERRQ(ierr);
796674ae819SStefano Zampini   PetscFunctionReturn(0);
797674ae819SStefano Zampini }
798674ae819SStefano Zampini 
799674ae819SStefano Zampini #undef __FUNCT__
800674ae819SStefano Zampini #define __FUNCT__ "PCBDDCGraphResetCSR"
801674ae819SStefano Zampini PetscErrorCode PCBDDCGraphResetCSR(PCBDDCGraph graph)
802674ae819SStefano Zampini {
803674ae819SStefano Zampini   PetscErrorCode ierr;
804674ae819SStefano Zampini 
805674ae819SStefano Zampini   PetscFunctionBegin;
806674ae819SStefano Zampini   ierr = PetscFree(graph->xadj);CHKERRQ(ierr);
807674ae819SStefano Zampini   ierr = PetscFree(graph->adjncy);CHKERRQ(ierr);
808674ae819SStefano Zampini   PetscFunctionReturn(0);
809674ae819SStefano Zampini }
810674ae819SStefano Zampini 
811674ae819SStefano Zampini #undef __FUNCT__
812674ae819SStefano Zampini #define __FUNCT__ "PCBDDCGraphReset"
813674ae819SStefano Zampini PetscErrorCode PCBDDCGraphReset(PCBDDCGraph graph)
814674ae819SStefano Zampini {
815674ae819SStefano Zampini   PetscErrorCode ierr;
816674ae819SStefano Zampini 
817674ae819SStefano Zampini   PetscFunctionBegin;
818674ae819SStefano Zampini   ierr = ISLocalToGlobalMappingDestroy(&graph->l2gmap);CHKERRQ(ierr);
819674ae819SStefano Zampini   ierr = PetscFree(graph->subset_ncc);CHKERRQ(ierr);
820674ae819SStefano Zampini   if (graph->nvtxs) {
821674ae819SStefano Zampini     ierr = PetscFree(graph->neighbours_set[0]);CHKERRQ(ierr);
822674ae819SStefano Zampini   }
823674ae819SStefano Zampini   ierr = PetscFree7(graph->touched,
824674ae819SStefano Zampini                     graph->count,
825674ae819SStefano Zampini                     graph->neighbours_set,
826674ae819SStefano Zampini                     graph->subset,
827674ae819SStefano Zampini                     graph->which_dof,
828674ae819SStefano Zampini                     graph->cptr,
829674ae819SStefano Zampini                     graph->queue);CHKERRQ(ierr);
830674ae819SStefano Zampini   graph->nvtxs = 0;
831674ae819SStefano Zampini   graph->n_subsets = 0;
832674ae819SStefano Zampini   graph->custom_minimal_size = 1;
833674ae819SStefano Zampini   PetscFunctionReturn(0);
834674ae819SStefano Zampini }
835674ae819SStefano Zampini 
836674ae819SStefano Zampini #undef __FUNCT__
837674ae819SStefano Zampini #define __FUNCT__ "PCBDDCGraphInit"
838674ae819SStefano Zampini PetscErrorCode PCBDDCGraphInit(PCBDDCGraph graph, ISLocalToGlobalMapping l2gmap)
839674ae819SStefano Zampini {
840674ae819SStefano Zampini   PetscInt       n;
841674ae819SStefano Zampini   PetscErrorCode ierr;
842674ae819SStefano Zampini 
843674ae819SStefano Zampini   PetscFunctionBegin;
844674ae819SStefano Zampini   PetscValidPointer(graph,1);
845674ae819SStefano Zampini   PetscValidHeaderSpecific(l2gmap,IS_LTOGM_CLASSID,2);
846674ae819SStefano Zampini   /* reset graph if already allocated */
847674ae819SStefano Zampini   if (graph->nvtxs) {
848674ae819SStefano Zampini     ierr = PCBDDCGraphReset(graph);CHKERRQ(ierr);
849674ae819SStefano Zampini   }
850674ae819SStefano Zampini   /* set number of vertices */
851674ae819SStefano Zampini   ierr = PetscObjectReference((PetscObject)l2gmap);CHKERRQ(ierr);
852674ae819SStefano Zampini   graph->l2gmap = l2gmap;
853674ae819SStefano Zampini   ierr = ISLocalToGlobalMappingGetSize(l2gmap,&n);CHKERRQ(ierr);
854674ae819SStefano Zampini   graph->nvtxs = n;
855674ae819SStefano Zampini   /* allocate used space */
856674ae819SStefano Zampini   ierr = PetscMalloc7(graph->nvtxs,PetscBool,&graph->touched,
857674ae819SStefano Zampini                       graph->nvtxs,PetscInt,&graph->count,
858674ae819SStefano Zampini                       graph->nvtxs,PetscInt*,&graph->neighbours_set,
859674ae819SStefano Zampini                       graph->nvtxs,PetscInt,&graph->subset,
860674ae819SStefano Zampini                       graph->nvtxs,PetscInt,&graph->which_dof,
861674ae819SStefano Zampini                       graph->nvtxs+1,PetscInt,&graph->cptr,
862674ae819SStefano Zampini                       graph->nvtxs,PetscInt,&graph->queue);CHKERRQ(ierr);
863674ae819SStefano Zampini   /* zeroes memory */
864674ae819SStefano Zampini   ierr = PetscMemzero(graph->touched,graph->nvtxs*sizeof(PetscBool));CHKERRQ(ierr);
865674ae819SStefano Zampini   ierr = PetscMemzero(graph->count,graph->nvtxs*sizeof(PetscInt));CHKERRQ(ierr);
866674ae819SStefano Zampini   ierr = PetscMemzero(graph->subset,graph->nvtxs*sizeof(PetscInt));CHKERRQ(ierr);
867674ae819SStefano Zampini   ierr = PetscMemzero(graph->which_dof,graph->nvtxs*sizeof(PetscInt));CHKERRQ(ierr);
868674ae819SStefano Zampini   ierr = PetscMemzero(graph->cptr,(graph->nvtxs+1)*sizeof(PetscInt));CHKERRQ(ierr);
869674ae819SStefano Zampini   ierr = PetscMemzero(graph->queue,graph->nvtxs*sizeof(PetscInt));CHKERRQ(ierr);
870674ae819SStefano Zampini   /* zeroes first pointer to neighbour set */
871674ae819SStefano Zampini   if (graph->nvtxs) {
872674ae819SStefano Zampini     graph->neighbours_set[0] = 0;
873674ae819SStefano Zampini   }
874674ae819SStefano Zampini   /* zeroes workspace for values of ncc */
875674ae819SStefano Zampini   graph->subset_ncc = 0;
876674ae819SStefano Zampini   PetscFunctionReturn(0);
877674ae819SStefano Zampini }
878674ae819SStefano Zampini 
879674ae819SStefano Zampini #undef __FUNCT__
880674ae819SStefano Zampini #define __FUNCT__ "PCBDDCGraphDestroy"
881674ae819SStefano Zampini PetscErrorCode PCBDDCGraphDestroy(PCBDDCGraph* graph)
882674ae819SStefano Zampini {
883674ae819SStefano Zampini   PetscErrorCode ierr;
884674ae819SStefano Zampini 
885674ae819SStefano Zampini   PetscFunctionBegin;
886674ae819SStefano Zampini   ierr = PCBDDCGraphReset(*graph);CHKERRQ(ierr);
887674ae819SStefano Zampini   ierr = PetscFree(*graph);CHKERRQ(ierr);
888674ae819SStefano Zampini   PetscFunctionReturn(0);
889674ae819SStefano Zampini }
890674ae819SStefano Zampini 
891674ae819SStefano Zampini #undef __FUNCT__
892674ae819SStefano Zampini #define __FUNCT__ "PCBDDCGraphCreate"
893674ae819SStefano Zampini PetscErrorCode PCBDDCGraphCreate(PCBDDCGraph *graph)
894674ae819SStefano Zampini {
895674ae819SStefano Zampini   PCBDDCGraph    new_graph;
896674ae819SStefano Zampini   PetscErrorCode ierr;
897674ae819SStefano Zampini 
898674ae819SStefano Zampini   PetscFunctionBegin;
899674ae819SStefano Zampini   ierr = PetscMalloc(sizeof(*new_graph),&new_graph);CHKERRQ(ierr);
900674ae819SStefano Zampini   /* local to global mapping of dofs */
901674ae819SStefano Zampini   new_graph->l2gmap = 0;
902674ae819SStefano Zampini   /* vertex size */
903674ae819SStefano Zampini   new_graph->nvtxs = 0;
904674ae819SStefano Zampini   new_graph->n_subsets = 0;
905674ae819SStefano Zampini   new_graph->custom_minimal_size = 1;
906674ae819SStefano Zampini   /* zeroes ponters */
907674ae819SStefano Zampini   new_graph->neighbours_set = 0;
908674ae819SStefano Zampini   new_graph->subset = 0;
909674ae819SStefano Zampini   new_graph->which_dof = 0;
910674ae819SStefano Zampini   new_graph->cptr = 0;
911674ae819SStefano Zampini   new_graph->queue = 0;
912674ae819SStefano Zampini   new_graph->count = 0;
913674ae819SStefano Zampini   new_graph->subset_ncc = 0;
914674ae819SStefano Zampini   new_graph->touched = 0;
915674ae819SStefano Zampini   /* zeroes pointers to csr graph of local nodes connectivity (optional data) */
916674ae819SStefano Zampini   new_graph->xadj = 0;
917674ae819SStefano Zampini   new_graph->adjncy = 0;
918674ae819SStefano Zampini   *graph = new_graph;
919674ae819SStefano Zampini   PetscFunctionReturn(0);
920674ae819SStefano Zampini }
921