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