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