12362add9SBarry Smith 2b45d2f2cSJed Brown #include <petsc-private/isimpl.h> /*I "petscis.h" I*/ 30c312b8eSJed Brown #include <petscsf.h> 4665c2dedSJed Brown #include <petscviewer.h> 52362add9SBarry Smith 67087cfbeSBarry Smith PetscClassId IS_LTOGM_CLASSID; 78e58c17dSMatthew Knepley 804a59952SBarry Smith 94a2ae208SSatish Balay #undef __FUNCT__ 10743c9b42SStefano Zampini #define __FUNCT__ "ISG2LMapApply" 11186d4ecdSBarry Smith PetscErrorCode ISG2LMapApply(ISLocalToGlobalMapping mapping,PetscInt n,const PetscInt in[],PetscInt out[]) 12186d4ecdSBarry Smith { 13186d4ecdSBarry Smith PetscErrorCode ierr; 14743c9b42SStefano Zampini PetscInt i,start,end; 15186d4ecdSBarry Smith 16186d4ecdSBarry Smith PetscFunctionBegin; 17186d4ecdSBarry Smith if (!mapping->globals) { 18186d4ecdSBarry Smith ierr = ISGlobalToLocalMappingApply(mapping,IS_GTOLM_MASK,0,0,0,0);CHKERRQ(ierr); 19186d4ecdSBarry Smith } 20743c9b42SStefano Zampini start = mapping->globalstart; 21743c9b42SStefano Zampini end = mapping->globalend; 22186d4ecdSBarry Smith for (i=0; i<n; i++) { 23186d4ecdSBarry Smith if (in[i] < 0) out[i] = in[i]; 24186d4ecdSBarry Smith else if (in[i] < start) out[i] = -1; 25186d4ecdSBarry Smith else if (in[i] > end) out[i] = -1; 26743c9b42SStefano Zampini else out[i] = mapping->globals[in[i] - start]; 27186d4ecdSBarry Smith } 28186d4ecdSBarry Smith PetscFunctionReturn(0); 29186d4ecdSBarry Smith } 30186d4ecdSBarry Smith 31186d4ecdSBarry Smith 32186d4ecdSBarry Smith #undef __FUNCT__ 334a2ae208SSatish Balay #define __FUNCT__ "ISLocalToGlobalMappingGetSize" 34565245c5SBarry Smith /*@ 35107e9a97SBarry Smith ISLocalToGlobalMappingGetSize - Gets the local size of a local to global mapping 363b9aefa3SBarry Smith 373b9aefa3SBarry Smith Not Collective 383b9aefa3SBarry Smith 393b9aefa3SBarry Smith Input Parameter: 403b9aefa3SBarry Smith . ltog - local to global mapping 413b9aefa3SBarry Smith 423b9aefa3SBarry Smith Output Parameter: 43107e9a97SBarry Smith . n - the number of entries in the local mapping, ISLocalToGlobalMappingGetIndices() returns an array of this length 443b9aefa3SBarry Smith 453b9aefa3SBarry Smith Level: advanced 463b9aefa3SBarry Smith 47273d9f13SBarry Smith Concepts: mapping^local to global 483b9aefa3SBarry Smith 493b9aefa3SBarry Smith .seealso: ISLocalToGlobalMappingDestroy(), ISLocalToGlobalMappingCreate() 503b9aefa3SBarry Smith @*/ 517087cfbeSBarry Smith PetscErrorCode ISLocalToGlobalMappingGetSize(ISLocalToGlobalMapping mapping,PetscInt *n) 523b9aefa3SBarry Smith { 533b9aefa3SBarry Smith PetscFunctionBegin; 540700a824SBarry Smith PetscValidHeaderSpecific(mapping,IS_LTOGM_CLASSID,1); 554482741eSBarry Smith PetscValidIntPointer(n,2); 56107e9a97SBarry Smith *n = mapping->bs*mapping->n; 573b9aefa3SBarry Smith PetscFunctionReturn(0); 583b9aefa3SBarry Smith } 593b9aefa3SBarry Smith 604a2ae208SSatish Balay #undef __FUNCT__ 614a2ae208SSatish Balay #define __FUNCT__ "ISLocalToGlobalMappingView" 625a5d4f66SBarry Smith /*@C 635a5d4f66SBarry Smith ISLocalToGlobalMappingView - View a local to global mapping 645a5d4f66SBarry Smith 65b9cd556bSLois Curfman McInnes Not Collective 66b9cd556bSLois Curfman McInnes 675a5d4f66SBarry Smith Input Parameters: 683b9aefa3SBarry Smith + ltog - local to global mapping 693b9aefa3SBarry Smith - viewer - viewer 705a5d4f66SBarry Smith 71a997ad1aSLois Curfman McInnes Level: advanced 72a997ad1aSLois Curfman McInnes 73273d9f13SBarry Smith Concepts: mapping^local to global 745a5d4f66SBarry Smith 755a5d4f66SBarry Smith .seealso: ISLocalToGlobalMappingDestroy(), ISLocalToGlobalMappingCreate() 765a5d4f66SBarry Smith @*/ 777087cfbeSBarry Smith PetscErrorCode ISLocalToGlobalMappingView(ISLocalToGlobalMapping mapping,PetscViewer viewer) 785a5d4f66SBarry Smith { 7932dcc486SBarry Smith PetscInt i; 8032dcc486SBarry Smith PetscMPIInt rank; 81ace3abfcSBarry Smith PetscBool iascii; 826849ba73SBarry Smith PetscErrorCode ierr; 835a5d4f66SBarry Smith 845a5d4f66SBarry Smith PetscFunctionBegin; 850700a824SBarry Smith PetscValidHeaderSpecific(mapping,IS_LTOGM_CLASSID,1); 863050cee2SBarry Smith if (!viewer) { 87ce94432eSBarry Smith ierr = PetscViewerASCIIGetStdout(PetscObjectComm((PetscObject)mapping),&viewer);CHKERRQ(ierr); 883050cee2SBarry Smith } 890700a824SBarry Smith PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2); 905a5d4f66SBarry Smith 91ce94432eSBarry Smith ierr = MPI_Comm_rank(PetscObjectComm((PetscObject)mapping),&rank);CHKERRQ(ierr); 92251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr); 9332077d6dSBarry Smith if (iascii) { 9498c3331eSBarry Smith ierr = PetscObjectPrintClassNamePrefixType((PetscObject)mapping,viewer);CHKERRQ(ierr); 957b23a99aSBarry Smith ierr = PetscViewerASCIISynchronizedAllow(viewer,PETSC_TRUE);CHKERRQ(ierr); 965a5d4f66SBarry Smith for (i=0; i<mapping->n; i++) { 977904a332SBarry Smith ierr = PetscViewerASCIISynchronizedPrintf(viewer,"[%d] %D %D\n",rank,i,mapping->indices[i]);CHKERRQ(ierr); 986831982aSBarry Smith } 99b0a32e0cSBarry Smith ierr = PetscViewerFlush(viewer);CHKERRQ(ierr); 1007b23a99aSBarry Smith ierr = PetscViewerASCIISynchronizedAllow(viewer,PETSC_FALSE);CHKERRQ(ierr); 1017b23a99aSBarry Smith } else SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Viewer type %s not supported for ISLocalToGlobalMapping",((PetscObject)viewer)->type_name); 1025a5d4f66SBarry Smith PetscFunctionReturn(0); 1035a5d4f66SBarry Smith } 1045a5d4f66SBarry Smith 1054a2ae208SSatish Balay #undef __FUNCT__ 1064a2ae208SSatish Balay #define __FUNCT__ "ISLocalToGlobalMappingCreateIS" 1071f428162SBarry Smith /*@ 1082bdab257SBarry Smith ISLocalToGlobalMappingCreateIS - Creates a mapping between a local (0 to n) 1092bdab257SBarry Smith ordering and a global parallel ordering. 1102bdab257SBarry Smith 1110f5bd95cSBarry Smith Not collective 112b9cd556bSLois Curfman McInnes 113a997ad1aSLois Curfman McInnes Input Parameter: 1148c03b21aSDmitry Karpeev . is - index set containing the global numbers for each local number 1152bdab257SBarry Smith 116a997ad1aSLois Curfman McInnes Output Parameter: 1172bdab257SBarry Smith . mapping - new mapping data structure 1182bdab257SBarry Smith 119f0413b6fSBarry Smith Notes: the block size of the IS determines the block size of the mapping 120a997ad1aSLois Curfman McInnes Level: advanced 121a997ad1aSLois Curfman McInnes 122273d9f13SBarry Smith Concepts: mapping^local to global 1232bdab257SBarry Smith 1242bdab257SBarry Smith .seealso: ISLocalToGlobalMappingDestroy(), ISLocalToGlobalMappingCreate() 1252bdab257SBarry Smith @*/ 1267087cfbeSBarry Smith PetscErrorCode ISLocalToGlobalMappingCreateIS(IS is,ISLocalToGlobalMapping *mapping) 1272bdab257SBarry Smith { 1286849ba73SBarry Smith PetscErrorCode ierr; 1293bbf0e92SBarry Smith PetscInt n,bs; 1305d0c19d7SBarry Smith const PetscInt *indices; 1312bdab257SBarry Smith MPI_Comm comm; 1323bbf0e92SBarry Smith PetscBool isblock; 1333a40ed3dSBarry Smith 1343a40ed3dSBarry Smith PetscFunctionBegin; 1350700a824SBarry Smith PetscValidHeaderSpecific(is,IS_CLASSID,1); 1364482741eSBarry Smith PetscValidPointer(mapping,2); 1372bdab257SBarry Smith 1382bdab257SBarry Smith ierr = PetscObjectGetComm((PetscObject)is,&comm);CHKERRQ(ierr); 1393b9aefa3SBarry Smith ierr = ISGetLocalSize(is,&n);CHKERRQ(ierr); 1403bbf0e92SBarry Smith ierr = PetscObjectTypeCompare((PetscObject)is,ISBLOCK,&isblock);CHKERRQ(ierr); 1413bbf0e92SBarry Smith ierr = ISGetBlockSize(is,&bs);CHKERRQ(ierr); 142f0413b6fSBarry Smith /* if (!isblock) { */ 143f0413b6fSBarry Smith ierr = ISGetIndices(is,&indices);CHKERRQ(ierr); 144f0413b6fSBarry Smith ierr = ISLocalToGlobalMappingCreate(comm,1,n,indices,PETSC_COPY_VALUES,mapping);CHKERRQ(ierr); 1452bdab257SBarry Smith ierr = ISRestoreIndices(is,&indices);CHKERRQ(ierr); 146f0413b6fSBarry Smith /* } else { 147f0413b6fSBarry Smith ierr = ISBlockGetIndices(is,&indices);CHKERRQ(ierr); 148f0413b6fSBarry Smith ierr = ISLocalToGlobalMappingCreate(comm,bs,n,indices,PETSC_COPY_VALUES,mapping);CHKERRQ(ierr); 149f0413b6fSBarry Smith ierr = ISBlockRestoreIndices(is,&indices);CHKERRQ(ierr); 150f0413b6fSBarry Smith }*/ 1513a40ed3dSBarry Smith PetscFunctionReturn(0); 1522bdab257SBarry Smith } 1535a5d4f66SBarry Smith 154a4d96a55SJed Brown #undef __FUNCT__ 155a4d96a55SJed Brown #define __FUNCT__ "ISLocalToGlobalMappingCreateSF" 156a4d96a55SJed Brown /*@C 157a4d96a55SJed Brown ISLocalToGlobalMappingCreateSF - Creates a mapping between a local (0 to n) 158a4d96a55SJed Brown ordering and a global parallel ordering. 159a4d96a55SJed Brown 160a4d96a55SJed Brown Collective 161a4d96a55SJed Brown 162a4d96a55SJed Brown Input Parameter: 163a4d96a55SJed Brown + sf - star forest mapping contiguous local indices to (rank, offset) 164a4d96a55SJed Brown - start - first global index on this process 165a4d96a55SJed Brown 166a4d96a55SJed Brown Output Parameter: 167a4d96a55SJed Brown . mapping - new mapping data structure 168a4d96a55SJed Brown 169a4d96a55SJed Brown Level: advanced 170a4d96a55SJed Brown 171a4d96a55SJed Brown Concepts: mapping^local to global 172a4d96a55SJed Brown 173a4d96a55SJed Brown .seealso: ISLocalToGlobalMappingDestroy(), ISLocalToGlobalMappingCreate(), ISLocalToGlobalMappingCreateIS() 174a4d96a55SJed Brown @*/ 175a4d96a55SJed Brown PetscErrorCode ISLocalToGlobalMappingCreateSF(PetscSF sf,PetscInt start,ISLocalToGlobalMapping *mapping) 176a4d96a55SJed Brown { 177a4d96a55SJed Brown PetscErrorCode ierr; 178a4d96a55SJed Brown PetscInt i,maxlocal,nroots,nleaves,*globals,*ltog; 179a4d96a55SJed Brown const PetscInt *ilocal; 180a4d96a55SJed Brown MPI_Comm comm; 181a4d96a55SJed Brown 182a4d96a55SJed Brown PetscFunctionBegin; 183a4d96a55SJed Brown PetscValidHeaderSpecific(sf,PETSCSF_CLASSID,1); 184a4d96a55SJed Brown PetscValidPointer(mapping,3); 185a4d96a55SJed Brown 186a4d96a55SJed Brown ierr = PetscObjectGetComm((PetscObject)sf,&comm);CHKERRQ(ierr); 1870298fd71SBarry Smith ierr = PetscSFGetGraph(sf,&nroots,&nleaves,&ilocal,NULL);CHKERRQ(ierr); 188f6e5521dSKarl Rupp if (ilocal) { 189f6e5521dSKarl Rupp for (i=0,maxlocal=0; i<nleaves; i++) maxlocal = PetscMax(maxlocal,ilocal[i]+1); 190f6e5521dSKarl Rupp } 191a4d96a55SJed Brown else maxlocal = nleaves; 192785e854fSJed Brown ierr = PetscMalloc1(nroots,&globals);CHKERRQ(ierr); 193785e854fSJed Brown ierr = PetscMalloc1(maxlocal,<og);CHKERRQ(ierr); 194a4d96a55SJed Brown for (i=0; i<nroots; i++) globals[i] = start + i; 195a4d96a55SJed Brown for (i=0; i<maxlocal; i++) ltog[i] = -1; 196a4d96a55SJed Brown ierr = PetscSFBcastBegin(sf,MPIU_INT,globals,ltog);CHKERRQ(ierr); 197a4d96a55SJed Brown ierr = PetscSFBcastEnd(sf,MPIU_INT,globals,ltog);CHKERRQ(ierr); 198f0413b6fSBarry Smith ierr = ISLocalToGlobalMappingCreate(comm,1,maxlocal,ltog,PETSC_OWN_POINTER,mapping);CHKERRQ(ierr); 199a4d96a55SJed Brown ierr = PetscFree(globals);CHKERRQ(ierr); 200a4d96a55SJed Brown PetscFunctionReturn(0); 201a4d96a55SJed Brown } 202b46b645bSBarry Smith 2034a2ae208SSatish Balay #undef __FUNCT__ 20445b6f7e9SBarry Smith #define __FUNCT__ "ISLocalToGlobalMappingGetBlockSize" 20545b6f7e9SBarry Smith /*@ 20645b6f7e9SBarry Smith ISLocalToGlobalMappingGetBlockSize - Gets the blocksize of the mapping 20745b6f7e9SBarry Smith ordering and a global parallel ordering. 20845b6f7e9SBarry Smith 20945b6f7e9SBarry Smith Not Collective 21045b6f7e9SBarry Smith 21145b6f7e9SBarry Smith Input Parameters: 21245b6f7e9SBarry Smith . mapping - mapping data structure 21345b6f7e9SBarry Smith 21445b6f7e9SBarry Smith Output Parameter: 21545b6f7e9SBarry Smith . bs - the blocksize 21645b6f7e9SBarry Smith 21745b6f7e9SBarry Smith Level: advanced 21845b6f7e9SBarry Smith 21945b6f7e9SBarry Smith Concepts: mapping^local to global 22045b6f7e9SBarry Smith 22145b6f7e9SBarry Smith .seealso: ISLocalToGlobalMappingDestroy(), ISLocalToGlobalMappingCreateIS() 22245b6f7e9SBarry Smith @*/ 22345b6f7e9SBarry Smith PetscErrorCode ISLocalToGlobalMappingGetBlockSize(ISLocalToGlobalMapping mapping,PetscInt *bs) 22445b6f7e9SBarry Smith { 22545b6f7e9SBarry Smith PetscFunctionBegin; 22645b6f7e9SBarry Smith *bs = mapping->bs; 22745b6f7e9SBarry Smith PetscFunctionReturn(0); 22845b6f7e9SBarry Smith } 22945b6f7e9SBarry Smith 23045b6f7e9SBarry Smith #undef __FUNCT__ 2314a2ae208SSatish Balay #define __FUNCT__ "ISLocalToGlobalMappingCreate" 232ba5bb76aSSatish Balay /*@ 23390f02eecSBarry Smith ISLocalToGlobalMappingCreate - Creates a mapping between a local (0 to n) 23490f02eecSBarry Smith ordering and a global parallel ordering. 2352362add9SBarry Smith 23689d82c54SBarry Smith Not Collective, but communicator may have more than one process 237b9cd556bSLois Curfman McInnes 2382362add9SBarry Smith Input Parameters: 23989d82c54SBarry Smith + comm - MPI communicator 240f0413b6fSBarry Smith . bs - the block size 24190f02eecSBarry Smith . n - the number of local elements 242f0413b6fSBarry Smith . indices - the global index for each local element, these do not need to be in increasing order (sorted), these values should not be scaled by the blocksize bs 243d5ad8652SBarry Smith - mode - see PetscCopyMode 2442362add9SBarry Smith 245a997ad1aSLois Curfman McInnes Output Parameter: 24690f02eecSBarry Smith . mapping - new mapping data structure 2472362add9SBarry Smith 248f0413b6fSBarry Smith Notes: There is one integer value in indices per block and it represents the actual indices bs*idx + j, where j=0,..,bs-1 249a997ad1aSLois Curfman McInnes Level: advanced 250a997ad1aSLois Curfman McInnes 251273d9f13SBarry Smith Concepts: mapping^local to global 2522362add9SBarry Smith 253d5ad8652SBarry Smith .seealso: ISLocalToGlobalMappingDestroy(), ISLocalToGlobalMappingCreateIS() 2542362add9SBarry Smith @*/ 255f0413b6fSBarry Smith PetscErrorCode ISLocalToGlobalMappingCreate(MPI_Comm cm,PetscInt bs,PetscInt n,const PetscInt indices[],PetscCopyMode mode,ISLocalToGlobalMapping *mapping) 2562362add9SBarry Smith { 2576849ba73SBarry Smith PetscErrorCode ierr; 25832dcc486SBarry Smith PetscInt *in; 259b46b645bSBarry Smith 260b46b645bSBarry Smith PetscFunctionBegin; 26173911063SBarry Smith if (n) PetscValidIntPointer(indices,3); 2624482741eSBarry Smith PetscValidPointer(mapping,4); 263b46b645bSBarry Smith 2640298fd71SBarry Smith *mapping = NULL; 265607a6623SBarry Smith ierr = ISInitializePackage();CHKERRQ(ierr); 2662362add9SBarry Smith 26767c2884eSBarry Smith ierr = PetscHeaderCreate(*mapping,_p_ISLocalToGlobalMapping,int,IS_LTOGM_CLASSID,"ISLocalToGlobalMapping","Local to global mapping","IS", 26852e6d16bSBarry Smith cm,ISLocalToGlobalMappingDestroy,ISLocalToGlobalMappingView);CHKERRQ(ierr); 269d4bb536fSBarry Smith (*mapping)->n = n; 270f0413b6fSBarry Smith (*mapping)->bs = bs; 271d4bb536fSBarry Smith /* 272d4bb536fSBarry Smith Do not create the global to local mapping. This is only created if 273d4bb536fSBarry Smith ISGlobalToLocalMapping() is called 274d4bb536fSBarry Smith */ 275d4bb536fSBarry Smith (*mapping)->globals = 0; 276d5ad8652SBarry Smith if (mode == PETSC_COPY_VALUES) { 277785e854fSJed Brown ierr = PetscMalloc1(n,&in);CHKERRQ(ierr); 278d5ad8652SBarry Smith ierr = PetscMemcpy(in,indices,n*sizeof(PetscInt));CHKERRQ(ierr); 279d5ad8652SBarry Smith (*mapping)->indices = in; 2806389a1a1SBarry Smith ierr = PetscLogObjectMemory((PetscObject)*mapping,n*sizeof(PetscInt));CHKERRQ(ierr); 2816389a1a1SBarry Smith } else if (mode == PETSC_OWN_POINTER) { 2826389a1a1SBarry Smith (*mapping)->indices = (PetscInt*)indices; 2836389a1a1SBarry Smith ierr = PetscLogObjectMemory((PetscObject)*mapping,n*sizeof(PetscInt));CHKERRQ(ierr); 2846389a1a1SBarry Smith } 285f6e5521dSKarl Rupp else SETERRQ(cm,PETSC_ERR_SUP,"Cannot currently use PETSC_USE_POINTER"); 2863a40ed3dSBarry Smith PetscFunctionReturn(0); 2872362add9SBarry Smith } 2882362add9SBarry Smith 2894a2ae208SSatish Balay #undef __FUNCT__ 2904a2ae208SSatish Balay #define __FUNCT__ "ISLocalToGlobalMappingDestroy" 29190f02eecSBarry Smith /*@ 29290f02eecSBarry Smith ISLocalToGlobalMappingDestroy - Destroys a mapping between a local (0 to n) 29390f02eecSBarry Smith ordering and a global parallel ordering. 29490f02eecSBarry Smith 2950f5bd95cSBarry Smith Note Collective 296b9cd556bSLois Curfman McInnes 29790f02eecSBarry Smith Input Parameters: 29890f02eecSBarry Smith . mapping - mapping data structure 29990f02eecSBarry Smith 300a997ad1aSLois Curfman McInnes Level: advanced 301a997ad1aSLois Curfman McInnes 3023acfe500SLois Curfman McInnes .seealso: ISLocalToGlobalMappingCreate() 30390f02eecSBarry Smith @*/ 3046bf464f9SBarry Smith PetscErrorCode ISLocalToGlobalMappingDestroy(ISLocalToGlobalMapping *mapping) 30590f02eecSBarry Smith { 306dfbe8321SBarry Smith PetscErrorCode ierr; 3075fd66863SKarl Rupp 3083a40ed3dSBarry Smith PetscFunctionBegin; 3096bf464f9SBarry Smith if (!*mapping) PetscFunctionReturn(0); 3106bf464f9SBarry Smith PetscValidHeaderSpecific((*mapping),IS_LTOGM_CLASSID,1); 311997056adSBarry Smith if (--((PetscObject)(*mapping))->refct > 0) {*mapping = 0;PetscFunctionReturn(0);} 3126bf464f9SBarry Smith ierr = PetscFree((*mapping)->indices);CHKERRQ(ierr); 3136bf464f9SBarry Smith ierr = PetscFree((*mapping)->globals);CHKERRQ(ierr); 314d38fa0fbSBarry Smith ierr = PetscHeaderDestroy(mapping);CHKERRQ(ierr); 315992144d0SBarry Smith *mapping = 0; 3163a40ed3dSBarry Smith PetscFunctionReturn(0); 31790f02eecSBarry Smith } 31890f02eecSBarry Smith 3194a2ae208SSatish Balay #undef __FUNCT__ 3204a2ae208SSatish Balay #define __FUNCT__ "ISLocalToGlobalMappingApplyIS" 32190f02eecSBarry Smith /*@ 3223acfe500SLois Curfman McInnes ISLocalToGlobalMappingApplyIS - Creates from an IS in the local numbering 3233acfe500SLois Curfman McInnes a new index set using the global numbering defined in an ISLocalToGlobalMapping 3243acfe500SLois Curfman McInnes context. 32590f02eecSBarry Smith 326b9cd556bSLois Curfman McInnes Not collective 327b9cd556bSLois Curfman McInnes 32890f02eecSBarry Smith Input Parameters: 329b9cd556bSLois Curfman McInnes + mapping - mapping between local and global numbering 330b9cd556bSLois Curfman McInnes - is - index set in local numbering 33190f02eecSBarry Smith 33290f02eecSBarry Smith Output Parameters: 33390f02eecSBarry Smith . newis - index set in global numbering 33490f02eecSBarry Smith 335a997ad1aSLois Curfman McInnes Level: advanced 336a997ad1aSLois Curfman McInnes 337273d9f13SBarry Smith Concepts: mapping^local to global 3383acfe500SLois Curfman McInnes 33990f02eecSBarry Smith .seealso: ISLocalToGlobalMappingApply(), ISLocalToGlobalMappingCreate(), 340d4bb536fSBarry Smith ISLocalToGlobalMappingDestroy(), ISGlobalToLocalMappingApply() 34190f02eecSBarry Smith @*/ 3427087cfbeSBarry Smith PetscErrorCode ISLocalToGlobalMappingApplyIS(ISLocalToGlobalMapping mapping,IS is,IS *newis) 34390f02eecSBarry Smith { 3446849ba73SBarry Smith PetscErrorCode ierr; 345*e24637baSBarry Smith PetscInt n,*idxout; 3465d0c19d7SBarry Smith const PetscInt *idxin; 3473a40ed3dSBarry Smith 3483a40ed3dSBarry Smith PetscFunctionBegin; 3490700a824SBarry Smith PetscValidHeaderSpecific(mapping,IS_LTOGM_CLASSID,1); 3500700a824SBarry Smith PetscValidHeaderSpecific(is,IS_CLASSID,2); 3514482741eSBarry Smith PetscValidPointer(newis,3); 35290f02eecSBarry Smith 3533b9aefa3SBarry Smith ierr = ISGetLocalSize(is,&n);CHKERRQ(ierr); 35490f02eecSBarry Smith ierr = ISGetIndices(is,&idxin);CHKERRQ(ierr); 355785e854fSJed Brown ierr = PetscMalloc1(n,&idxout);CHKERRQ(ierr); 356*e24637baSBarry Smith ierr = ISLocalToGlobalMappingApply(mapping,n,idxin,idxout);CHKERRQ(ierr); 3573b9aefa3SBarry Smith ierr = ISRestoreIndices(is,&idxin);CHKERRQ(ierr); 358543f3098SMatthew G. Knepley ierr = ISCreateGeneral(PetscObjectComm((PetscObject)is),n,idxout,PETSC_OWN_POINTER,newis);CHKERRQ(ierr); 3593a40ed3dSBarry Smith PetscFunctionReturn(0); 36090f02eecSBarry Smith } 36190f02eecSBarry Smith 362afcb2eb5SJed Brown #undef __FUNCT__ 363afcb2eb5SJed Brown #define __FUNCT__ "ISLocalToGlobalMappingApply" 364b89cb25eSSatish Balay /*@ 3653acfe500SLois Curfman McInnes ISLocalToGlobalMappingApply - Takes a list of integers in a local numbering 3663acfe500SLois Curfman McInnes and converts them to the global numbering. 36790f02eecSBarry Smith 368b9cd556bSLois Curfman McInnes Not collective 369b9cd556bSLois Curfman McInnes 370bb25748dSBarry Smith Input Parameters: 371b9cd556bSLois Curfman McInnes + mapping - the local to global mapping context 372bb25748dSBarry Smith . N - number of integers 373b9cd556bSLois Curfman McInnes - in - input indices in local numbering 374bb25748dSBarry Smith 375bb25748dSBarry Smith Output Parameter: 376bb25748dSBarry Smith . out - indices in global numbering 377bb25748dSBarry Smith 378b9cd556bSLois Curfman McInnes Notes: 379b9cd556bSLois Curfman McInnes The in and out array parameters may be identical. 380d4bb536fSBarry Smith 381a997ad1aSLois Curfman McInnes Level: advanced 382a997ad1aSLois Curfman McInnes 38345b6f7e9SBarry Smith .seealso: ISLocalToGlobalMappingApplyBlock(), ISLocalToGlobalMappingCreate(),ISLocalToGlobalMappingDestroy(), 3840752156aSBarry Smith ISLocalToGlobalMappingApplyIS(),AOCreateBasic(),AOApplicationToPetsc(), 385d4bb536fSBarry Smith AOPetscToApplication(), ISGlobalToLocalMappingApply() 386bb25748dSBarry Smith 387273d9f13SBarry Smith Concepts: mapping^local to global 388afcb2eb5SJed Brown @*/ 389afcb2eb5SJed Brown PetscErrorCode ISLocalToGlobalMappingApply(ISLocalToGlobalMapping mapping,PetscInt N,const PetscInt in[],PetscInt out[]) 390afcb2eb5SJed Brown { 391*e24637baSBarry Smith PetscInt i,bs = mapping->bs,Nmax = bs*mapping->n; 39245b6f7e9SBarry Smith const PetscInt *idx = mapping->indices; 39345b6f7e9SBarry Smith 39445b6f7e9SBarry Smith PetscFunctionBegin; 39545b6f7e9SBarry Smith if (bs == 1) { 39645b6f7e9SBarry Smith for (i=0; i<N; i++) { 39745b6f7e9SBarry Smith if (in[i] < 0) { 39845b6f7e9SBarry Smith out[i] = in[i]; 39945b6f7e9SBarry Smith continue; 40045b6f7e9SBarry Smith } 401*e24637baSBarry Smith if (in[i] >= Nmax) SETERRQ3(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Local index %D too large %D (max) at %D",in[i],Nmax-1,i); 40245b6f7e9SBarry Smith out[i] = idx[in[i]]; 40345b6f7e9SBarry Smith } 40445b6f7e9SBarry Smith } else { 40545b6f7e9SBarry Smith for (i=0; i<N; i++) { 40645b6f7e9SBarry Smith if (in[i] < 0) { 40745b6f7e9SBarry Smith out[i] = in[i]; 40845b6f7e9SBarry Smith continue; 40945b6f7e9SBarry Smith } 410*e24637baSBarry Smith if (in[i] >= Nmax) SETERRQ3(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Local index %D too large %D (max) at %D",in[i],Nmax-1,i); 41145b6f7e9SBarry Smith out[i] = idx[in[i]/bs]*bs + (in[i] % bs); 41245b6f7e9SBarry Smith } 41345b6f7e9SBarry Smith } 41445b6f7e9SBarry Smith PetscFunctionReturn(0); 41545b6f7e9SBarry Smith } 41645b6f7e9SBarry Smith 41745b6f7e9SBarry Smith #undef __FUNCT__ 41845b6f7e9SBarry Smith #define __FUNCT__ "ISLocalToGlobalMappingApplyBlock" 41945b6f7e9SBarry Smith /*@ 42045b6f7e9SBarry Smith ISLocalToGlobalMappingApplyBlock - Takes a list of integers in a local numbering and converts them to the global numbering. 42145b6f7e9SBarry Smith 42245b6f7e9SBarry Smith Not collective 42345b6f7e9SBarry Smith 42445b6f7e9SBarry Smith Input Parameters: 42545b6f7e9SBarry Smith + mapping - the local to global mapping context 42645b6f7e9SBarry Smith . N - number of integers 42745b6f7e9SBarry Smith - in - input indices in local numbering 42845b6f7e9SBarry Smith 42945b6f7e9SBarry Smith Output Parameter: 43045b6f7e9SBarry Smith . out - indices in global numbering 43145b6f7e9SBarry Smith 43245b6f7e9SBarry Smith Notes: 43345b6f7e9SBarry Smith The in and out array parameters may be identical. 43445b6f7e9SBarry Smith 43545b6f7e9SBarry Smith Level: advanced 43645b6f7e9SBarry Smith 43745b6f7e9SBarry Smith .seealso: ISLocalToGlobalMappingApply(), ISLocalToGlobalMappingCreate(),ISLocalToGlobalMappingDestroy(), 43845b6f7e9SBarry Smith ISLocalToGlobalMappingApplyIS(),AOCreateBasic(),AOApplicationToPetsc(), 43945b6f7e9SBarry Smith AOPetscToApplication(), ISGlobalToLocalMappingApply() 44045b6f7e9SBarry Smith 44145b6f7e9SBarry Smith Concepts: mapping^local to global 44245b6f7e9SBarry Smith @*/ 44345b6f7e9SBarry Smith PetscErrorCode ISLocalToGlobalMappingApplyBlock(ISLocalToGlobalMapping mapping,PetscInt N,const PetscInt in[],PetscInt out[]) 44445b6f7e9SBarry Smith { 445afcb2eb5SJed Brown PetscInt i,Nmax = mapping->n; 446afcb2eb5SJed Brown const PetscInt *idx = mapping->indices; 447d4bb536fSBarry Smith 448afcb2eb5SJed Brown PetscFunctionBegin; 449afcb2eb5SJed Brown for (i=0; i<N; i++) { 450afcb2eb5SJed Brown if (in[i] < 0) { 451afcb2eb5SJed Brown out[i] = in[i]; 452afcb2eb5SJed Brown continue; 453afcb2eb5SJed Brown } 454*e24637baSBarry Smith if (in[i] >= Nmax) SETERRQ3(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Local block index %D too large %D (max) at %D",in[i],Nmax-1,i); 455afcb2eb5SJed Brown out[i] = idx[in[i]]; 456afcb2eb5SJed Brown } 457afcb2eb5SJed Brown PetscFunctionReturn(0); 458afcb2eb5SJed Brown } 459d4bb536fSBarry Smith 460d4bb536fSBarry Smith /* -----------------------------------------------------------------------------------------*/ 461d4bb536fSBarry Smith 4624a2ae208SSatish Balay #undef __FUNCT__ 4634a2ae208SSatish Balay #define __FUNCT__ "ISGlobalToLocalMappingSetUp_Private" 464d4bb536fSBarry Smith /* 465d4bb536fSBarry Smith Creates the global fields in the ISLocalToGlobalMapping structure 466d4bb536fSBarry Smith */ 4676849ba73SBarry Smith static PetscErrorCode ISGlobalToLocalMappingSetUp_Private(ISLocalToGlobalMapping mapping) 468d4bb536fSBarry Smith { 4696849ba73SBarry Smith PetscErrorCode ierr; 47032dcc486SBarry Smith PetscInt i,*idx = mapping->indices,n = mapping->n,end,start,*globals; 471d4bb536fSBarry Smith 4723a40ed3dSBarry Smith PetscFunctionBegin; 473d4bb536fSBarry Smith end = 0; 474ec268f7cSJed Brown start = PETSC_MAX_INT; 475d4bb536fSBarry Smith 476d4bb536fSBarry Smith for (i=0; i<n; i++) { 477d4bb536fSBarry Smith if (idx[i] < 0) continue; 478d4bb536fSBarry Smith if (idx[i] < start) start = idx[i]; 479d4bb536fSBarry Smith if (idx[i] > end) end = idx[i]; 480d4bb536fSBarry Smith } 481d4bb536fSBarry Smith if (start > end) {start = 0; end = -1;} 482d4bb536fSBarry Smith mapping->globalstart = start; 483d4bb536fSBarry Smith mapping->globalend = end; 484d4bb536fSBarry Smith 485785e854fSJed Brown ierr = PetscMalloc1((end-start+2),&globals);CHKERRQ(ierr); 486b0a32e0cSBarry Smith mapping->globals = globals; 487f6e5521dSKarl Rupp for (i=0; i<end-start+1; i++) globals[i] = -1; 488d4bb536fSBarry Smith for (i=0; i<n; i++) { 489d4bb536fSBarry Smith if (idx[i] < 0) continue; 490d4bb536fSBarry Smith globals[idx[i] - start] = i; 491d4bb536fSBarry Smith } 492d4bb536fSBarry Smith 4933bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)mapping,(end-start+1)*sizeof(PetscInt));CHKERRQ(ierr); 4943a40ed3dSBarry Smith PetscFunctionReturn(0); 495d4bb536fSBarry Smith } 496d4bb536fSBarry Smith 4974a2ae208SSatish Balay #undef __FUNCT__ 4984a2ae208SSatish Balay #define __FUNCT__ "ISGlobalToLocalMappingApply" 499d4bb536fSBarry Smith /*@ 500a997ad1aSLois Curfman McInnes ISGlobalToLocalMappingApply - Provides the local numbering for a list of integers 501a997ad1aSLois Curfman McInnes specified with a global numbering. 502d4bb536fSBarry Smith 503b9cd556bSLois Curfman McInnes Not collective 504b9cd556bSLois Curfman McInnes 505d4bb536fSBarry Smith Input Parameters: 506b9cd556bSLois Curfman McInnes + mapping - mapping between local and global numbering 507d4bb536fSBarry Smith . type - IS_GTOLM_MASK - replaces global indices with no local value with -1 508d4bb536fSBarry Smith IS_GTOLM_DROP - drops the indices with no local value from the output list 509d4bb536fSBarry Smith . n - number of global indices to map 510b9cd556bSLois Curfman McInnes - idx - global indices to map 511d4bb536fSBarry Smith 512d4bb536fSBarry Smith Output Parameters: 513b9cd556bSLois Curfman McInnes + nout - number of indices in output array (if type == IS_GTOLM_MASK then nout = n) 514b9cd556bSLois Curfman McInnes - idxout - local index of each global index, one must pass in an array long enough 515e182c471SBarry Smith to hold all the indices. You can call ISGlobalToLocalMappingApply() with 5160298fd71SBarry Smith idxout == NULL to determine the required length (returned in nout) 517e182c471SBarry Smith and then allocate the required space and call ISGlobalToLocalMappingApply() 518e182c471SBarry Smith a second time to set the values. 519d4bb536fSBarry Smith 520b9cd556bSLois Curfman McInnes Notes: 5210298fd71SBarry Smith Either nout or idxout may be NULL. idx and idxout may be identical. 522d4bb536fSBarry Smith 5230f5bd95cSBarry Smith This is not scalable in memory usage. Each processor requires O(Nglobal) size 5240f5bd95cSBarry Smith array to compute these. 5250f5bd95cSBarry Smith 526a997ad1aSLois Curfman McInnes Level: advanced 527a997ad1aSLois Curfman McInnes 52832fd6b96SBarry Smith Developer Note: The manual page states that idx and idxout may be identical but the calling 52932fd6b96SBarry Smith sequence declares idx as const so it cannot be the same as idxout. 53032fd6b96SBarry Smith 531273d9f13SBarry Smith Concepts: mapping^global to local 532d4bb536fSBarry Smith 533d4bb536fSBarry Smith .seealso: ISLocalToGlobalMappingApply(), ISLocalToGlobalMappingCreate(), 534d4bb536fSBarry Smith ISLocalToGlobalMappingDestroy() 535d4bb536fSBarry Smith @*/ 5367087cfbeSBarry Smith PetscErrorCode ISGlobalToLocalMappingApply(ISLocalToGlobalMapping mapping,ISGlobalToLocalMappingType type, 53732dcc486SBarry Smith PetscInt n,const PetscInt idx[],PetscInt *nout,PetscInt idxout[]) 538d4bb536fSBarry Smith { 53932dcc486SBarry Smith PetscInt i,*globals,nf = 0,tmp,start,end; 5406849ba73SBarry Smith PetscErrorCode ierr; 541d4bb536fSBarry Smith 5423a40ed3dSBarry Smith PetscFunctionBegin; 5430700a824SBarry Smith PetscValidHeaderSpecific(mapping,IS_LTOGM_CLASSID,1); 544d4bb536fSBarry Smith if (!mapping->globals) { 545d4bb536fSBarry Smith ierr = ISGlobalToLocalMappingSetUp_Private(mapping);CHKERRQ(ierr); 546d4bb536fSBarry Smith } 547d4bb536fSBarry Smith globals = mapping->globals; 548d4bb536fSBarry Smith start = mapping->globalstart; 549d4bb536fSBarry Smith end = mapping->globalend; 550d4bb536fSBarry Smith 551d4bb536fSBarry Smith if (type == IS_GTOLM_MASK) { 552d4bb536fSBarry Smith if (idxout) { 553d4bb536fSBarry Smith for (i=0; i<n; i++) { 554d4bb536fSBarry Smith if (idx[i] < 0) idxout[i] = idx[i]; 555d4bb536fSBarry Smith else if (idx[i] < start) idxout[i] = -1; 556d4bb536fSBarry Smith else if (idx[i] > end) idxout[i] = -1; 557d4bb536fSBarry Smith else idxout[i] = globals[idx[i] - start]; 558d4bb536fSBarry Smith } 559d4bb536fSBarry Smith } 560d4bb536fSBarry Smith if (nout) *nout = n; 561d4bb536fSBarry Smith } else { 562d4bb536fSBarry Smith if (idxout) { 563d4bb536fSBarry Smith for (i=0; i<n; i++) { 564d4bb536fSBarry Smith if (idx[i] < 0) continue; 565d4bb536fSBarry Smith if (idx[i] < start) continue; 566d4bb536fSBarry Smith if (idx[i] > end) continue; 567d4bb536fSBarry Smith tmp = globals[idx[i] - start]; 568d4bb536fSBarry Smith if (tmp < 0) continue; 569d4bb536fSBarry Smith idxout[nf++] = tmp; 570d4bb536fSBarry Smith } 571d4bb536fSBarry Smith } else { 572d4bb536fSBarry Smith for (i=0; i<n; i++) { 573d4bb536fSBarry Smith if (idx[i] < 0) continue; 574d4bb536fSBarry Smith if (idx[i] < start) continue; 575d4bb536fSBarry Smith if (idx[i] > end) continue; 576d4bb536fSBarry Smith tmp = globals[idx[i] - start]; 577d4bb536fSBarry Smith if (tmp < 0) continue; 578d4bb536fSBarry Smith nf++; 579d4bb536fSBarry Smith } 580d4bb536fSBarry Smith } 581d4bb536fSBarry Smith if (nout) *nout = nf; 582d4bb536fSBarry Smith } 5833a40ed3dSBarry Smith PetscFunctionReturn(0); 584d4bb536fSBarry Smith } 58590f02eecSBarry Smith 5864a2ae208SSatish Balay #undef __FUNCT__ 5874a2ae208SSatish Balay #define __FUNCT__ "ISLocalToGlobalMappingGetInfo" 58889d82c54SBarry Smith /*@C 58989d82c54SBarry Smith ISLocalToGlobalMappingGetInfo - Gets the neighbor information for each processor and 59089d82c54SBarry Smith each index shared by more than one processor 59189d82c54SBarry Smith 59289d82c54SBarry Smith Collective on ISLocalToGlobalMapping 59389d82c54SBarry Smith 59489d82c54SBarry Smith Input Parameters: 59589d82c54SBarry Smith . mapping - the mapping from local to global indexing 59689d82c54SBarry Smith 59789d82c54SBarry Smith Output Parameter: 59889d82c54SBarry Smith + nproc - number of processors that are connected to this one 59989d82c54SBarry Smith . proc - neighboring processors 60007b52d57SBarry Smith . numproc - number of indices for each subdomain (processor) 6013463a7baSJed Brown - indices - indices of nodes (in local numbering) shared with neighbors (sorted by global numbering) 60289d82c54SBarry Smith 60389d82c54SBarry Smith Level: advanced 60489d82c54SBarry Smith 605273d9f13SBarry Smith Concepts: mapping^local to global 60689d82c54SBarry Smith 6072cfcea29SBarry Smith Fortran Usage: 6082cfcea29SBarry Smith $ ISLocalToGlobalMpngGetInfoSize(ISLocalToGlobalMapping,PetscInt nproc,PetscInt numprocmax,ierr) followed by 6092cfcea29SBarry Smith $ ISLocalToGlobalMappingGetInfo(ISLocalToGlobalMapping,PetscInt nproc, PetscInt procs[nproc],PetscInt numprocs[nproc], 6102cfcea29SBarry Smith PetscInt indices[nproc][numprocmax],ierr) 6112cfcea29SBarry Smith There is no ISLocalToGlobalMappingRestoreInfo() in Fortran. You must make sure that procs[], numprocs[] and 6122cfcea29SBarry Smith indices[][] are large enough arrays, either by allocating them dynamically or defining static ones large enough. 6132cfcea29SBarry Smith 6142cfcea29SBarry Smith 61507b52d57SBarry Smith .seealso: ISLocalToGlobalMappingDestroy(), ISLocalToGlobalMappingCreateIS(), ISLocalToGlobalMappingCreate(), 61607b52d57SBarry Smith ISLocalToGlobalMappingRestoreInfo() 61789d82c54SBarry Smith @*/ 6187087cfbeSBarry Smith PetscErrorCode ISLocalToGlobalMappingGetInfo(ISLocalToGlobalMapping mapping,PetscInt *nproc,PetscInt *procs[],PetscInt *numprocs[],PetscInt **indices[]) 61989d82c54SBarry Smith { 6206849ba73SBarry Smith PetscErrorCode ierr; 62197f1f81fSBarry Smith PetscMPIInt size,rank,tag1,tag2,tag3,*len,*source,imdex; 62232dcc486SBarry Smith PetscInt i,n = mapping->n,Ng,ng,max = 0,*lindices = mapping->indices; 62332dcc486SBarry Smith PetscInt *nprocs,*owner,nsends,*sends,j,*starts,nmax,nrecvs,*recvs,proc; 62497f1f81fSBarry Smith PetscInt cnt,scale,*ownedsenders,*nownedsenders,rstart,nowned; 62532dcc486SBarry Smith PetscInt node,nownedm,nt,*sends2,nsends2,*starts2,*lens2,*dest,nrecvs2,*starts3,*recvs2,k,*bprocs,*tmp; 62632dcc486SBarry Smith PetscInt first_procs,first_numprocs,*first_indices; 62789d82c54SBarry Smith MPI_Request *recv_waits,*send_waits; 62830dcb7c9SBarry Smith MPI_Status recv_status,*send_status,*recv_statuses; 629ce94432eSBarry Smith MPI_Comm comm; 630ace3abfcSBarry Smith PetscBool debug = PETSC_FALSE; 63189d82c54SBarry Smith 63289d82c54SBarry Smith PetscFunctionBegin; 6330700a824SBarry Smith PetscValidHeaderSpecific(mapping,IS_LTOGM_CLASSID,1); 634ce94432eSBarry Smith ierr = PetscObjectGetComm((PetscObject)mapping,&comm);CHKERRQ(ierr); 63524cf384cSBarry Smith ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr); 63624cf384cSBarry Smith ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr); 63724cf384cSBarry Smith if (size == 1) { 63824cf384cSBarry Smith *nproc = 0; 6390298fd71SBarry Smith *procs = NULL; 64032dcc486SBarry Smith ierr = PetscMalloc(sizeof(PetscInt),numprocs);CHKERRQ(ierr); 6411e2105dcSBarry Smith (*numprocs)[0] = 0; 64232dcc486SBarry Smith ierr = PetscMalloc(sizeof(PetscInt*),indices);CHKERRQ(ierr); 6430298fd71SBarry Smith (*indices)[0] = NULL; 64424cf384cSBarry Smith PetscFunctionReturn(0); 64524cf384cSBarry Smith } 64624cf384cSBarry Smith 6470298fd71SBarry Smith ierr = PetscOptionsGetBool(NULL,"-islocaltoglobalmappinggetinfo_debug",&debug,NULL);CHKERRQ(ierr); 64807b52d57SBarry Smith 6493677ff5aSBarry Smith /* 6503677ff5aSBarry Smith Notes on ISLocalToGlobalMappingGetInfo 6513677ff5aSBarry Smith 6523677ff5aSBarry Smith globally owned node - the nodes that have been assigned to this processor in global 6533677ff5aSBarry Smith numbering, just for this routine. 6543677ff5aSBarry Smith 6553677ff5aSBarry Smith nontrivial globally owned node - node assigned to this processor that is on a subdomain 6563677ff5aSBarry Smith boundary (i.e. is has more than one local owner) 6573677ff5aSBarry Smith 6583677ff5aSBarry Smith locally owned node - node that exists on this processors subdomain 6593677ff5aSBarry Smith 6603677ff5aSBarry Smith nontrivial locally owned node - node that is not in the interior (i.e. has more than one 6613677ff5aSBarry Smith local subdomain 6623677ff5aSBarry Smith */ 66324cf384cSBarry Smith ierr = PetscObjectGetNewTag((PetscObject)mapping,&tag1);CHKERRQ(ierr); 66424cf384cSBarry Smith ierr = PetscObjectGetNewTag((PetscObject)mapping,&tag2);CHKERRQ(ierr); 66524cf384cSBarry Smith ierr = PetscObjectGetNewTag((PetscObject)mapping,&tag3);CHKERRQ(ierr); 66689d82c54SBarry Smith 66789d82c54SBarry Smith for (i=0; i<n; i++) { 66889d82c54SBarry Smith if (lindices[i] > max) max = lindices[i]; 66989d82c54SBarry Smith } 67032dcc486SBarry Smith ierr = MPI_Allreduce(&max,&Ng,1,MPIU_INT,MPI_MAX,comm);CHKERRQ(ierr); 67178058e43SBarry Smith Ng++; 67289d82c54SBarry Smith ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr); 67389d82c54SBarry Smith ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr); 674bc8ff85bSBarry Smith scale = Ng/size + 1; 675a2e34c3dSBarry Smith ng = scale; if (rank == size-1) ng = Ng - scale*(size-1); ng = PetscMax(1,ng); 676caba0dd0SBarry Smith rstart = scale*rank; 67789d82c54SBarry Smith 67889d82c54SBarry Smith /* determine ownership ranges of global indices */ 679785e854fSJed Brown ierr = PetscMalloc1(2*size,&nprocs);CHKERRQ(ierr); 68032dcc486SBarry Smith ierr = PetscMemzero(nprocs,2*size*sizeof(PetscInt));CHKERRQ(ierr); 68189d82c54SBarry Smith 68289d82c54SBarry Smith /* determine owners of each local node */ 683785e854fSJed Brown ierr = PetscMalloc1(n,&owner);CHKERRQ(ierr); 68489d82c54SBarry Smith for (i=0; i<n; i++) { 6853677ff5aSBarry Smith proc = lindices[i]/scale; /* processor that globally owns this index */ 68627c402fcSBarry Smith nprocs[2*proc+1] = 1; /* processor globally owns at least one of ours */ 6873677ff5aSBarry Smith owner[i] = proc; 68827c402fcSBarry Smith nprocs[2*proc]++; /* count of how many that processor globally owns of ours */ 68989d82c54SBarry Smith } 69027c402fcSBarry Smith nsends = 0; for (i=0; i<size; i++) nsends += nprocs[2*i+1]; 6917904a332SBarry Smith ierr = PetscInfo1(mapping,"Number of global owners for my local data %D\n",nsends);CHKERRQ(ierr); 69289d82c54SBarry Smith 69389d82c54SBarry Smith /* inform other processors of number of messages and max length*/ 69427c402fcSBarry Smith ierr = PetscMaxSum(comm,nprocs,&nmax,&nrecvs);CHKERRQ(ierr); 6957904a332SBarry Smith ierr = PetscInfo1(mapping,"Number of local owners for my global data %D\n",nrecvs);CHKERRQ(ierr); 69689d82c54SBarry Smith 69789d82c54SBarry Smith /* post receives for owned rows */ 698785e854fSJed Brown ierr = PetscMalloc1((2*nrecvs+1)*(nmax+1),&recvs);CHKERRQ(ierr); 699785e854fSJed Brown ierr = PetscMalloc1((nrecvs+1),&recv_waits);CHKERRQ(ierr); 70089d82c54SBarry Smith for (i=0; i<nrecvs; i++) { 70132dcc486SBarry Smith ierr = MPI_Irecv(recvs+2*nmax*i,2*nmax,MPIU_INT,MPI_ANY_SOURCE,tag1,comm,recv_waits+i);CHKERRQ(ierr); 70289d82c54SBarry Smith } 70389d82c54SBarry Smith 70489d82c54SBarry Smith /* pack messages containing lists of local nodes to owners */ 705785e854fSJed Brown ierr = PetscMalloc1((2*n+1),&sends);CHKERRQ(ierr); 706785e854fSJed Brown ierr = PetscMalloc1((size+1),&starts);CHKERRQ(ierr); 70789d82c54SBarry Smith starts[0] = 0; 708f6e5521dSKarl Rupp for (i=1; i<size; i++) starts[i] = starts[i-1] + 2*nprocs[2*i-2]; 70989d82c54SBarry Smith for (i=0; i<n; i++) { 71089d82c54SBarry Smith sends[starts[owner[i]]++] = lindices[i]; 71130dcb7c9SBarry Smith sends[starts[owner[i]]++] = i; 71289d82c54SBarry Smith } 71389d82c54SBarry Smith ierr = PetscFree(owner);CHKERRQ(ierr); 71489d82c54SBarry Smith starts[0] = 0; 715f6e5521dSKarl Rupp for (i=1; i<size; i++) starts[i] = starts[i-1] + 2*nprocs[2*i-2]; 71689d82c54SBarry Smith 71789d82c54SBarry Smith /* send the messages */ 718785e854fSJed Brown ierr = PetscMalloc1((nsends+1),&send_waits);CHKERRQ(ierr); 719785e854fSJed Brown ierr = PetscMalloc1((nsends+1),&dest);CHKERRQ(ierr); 72089d82c54SBarry Smith cnt = 0; 72189d82c54SBarry Smith for (i=0; i<size; i++) { 72227c402fcSBarry Smith if (nprocs[2*i]) { 72332dcc486SBarry Smith ierr = MPI_Isend(sends+starts[i],2*nprocs[2*i],MPIU_INT,i,tag1,comm,send_waits+cnt);CHKERRQ(ierr); 72430dcb7c9SBarry Smith dest[cnt] = i; 72589d82c54SBarry Smith cnt++; 72689d82c54SBarry Smith } 72789d82c54SBarry Smith } 72889d82c54SBarry Smith ierr = PetscFree(starts);CHKERRQ(ierr); 72989d82c54SBarry Smith 73089d82c54SBarry Smith /* wait on receives */ 731785e854fSJed Brown ierr = PetscMalloc1((nrecvs+1),&source);CHKERRQ(ierr); 732785e854fSJed Brown ierr = PetscMalloc1((nrecvs+1),&len);CHKERRQ(ierr); 73389d82c54SBarry Smith cnt = nrecvs; 734785e854fSJed Brown ierr = PetscMalloc1((ng+1),&nownedsenders);CHKERRQ(ierr); 73532dcc486SBarry Smith ierr = PetscMemzero(nownedsenders,ng*sizeof(PetscInt));CHKERRQ(ierr); 73689d82c54SBarry Smith while (cnt) { 73789d82c54SBarry Smith ierr = MPI_Waitany(nrecvs,recv_waits,&imdex,&recv_status);CHKERRQ(ierr); 73889d82c54SBarry Smith /* unpack receives into our local space */ 73932dcc486SBarry Smith ierr = MPI_Get_count(&recv_status,MPIU_INT,&len[imdex]);CHKERRQ(ierr); 74089d82c54SBarry Smith source[imdex] = recv_status.MPI_SOURCE; 74130dcb7c9SBarry Smith len[imdex] = len[imdex]/2; 742caba0dd0SBarry Smith /* count how many local owners for each of my global owned indices */ 74330dcb7c9SBarry Smith for (i=0; i<len[imdex]; i++) nownedsenders[recvs[2*imdex*nmax+2*i]-rstart]++; 74489d82c54SBarry Smith cnt--; 74589d82c54SBarry Smith } 74689d82c54SBarry Smith ierr = PetscFree(recv_waits);CHKERRQ(ierr); 74789d82c54SBarry Smith 74830dcb7c9SBarry Smith /* count how many globally owned indices are on an edge multiplied by how many processors own them. */ 749bc8ff85bSBarry Smith nowned = 0; 750bc8ff85bSBarry Smith nownedm = 0; 751bc8ff85bSBarry Smith for (i=0; i<ng; i++) { 752bc8ff85bSBarry Smith if (nownedsenders[i] > 1) {nownedm += nownedsenders[i]; nowned++;} 753bc8ff85bSBarry Smith } 754bc8ff85bSBarry Smith 755bc8ff85bSBarry Smith /* create single array to contain rank of all local owners of each globally owned index */ 756785e854fSJed Brown ierr = PetscMalloc1((nownedm+1),&ownedsenders);CHKERRQ(ierr); 757785e854fSJed Brown ierr = PetscMalloc1((ng+1),&starts);CHKERRQ(ierr); 758bc8ff85bSBarry Smith starts[0] = 0; 759bc8ff85bSBarry Smith for (i=1; i<ng; i++) { 760bc8ff85bSBarry Smith if (nownedsenders[i-1] > 1) starts[i] = starts[i-1] + nownedsenders[i-1]; 761bc8ff85bSBarry Smith else starts[i] = starts[i-1]; 762bc8ff85bSBarry Smith } 763bc8ff85bSBarry Smith 76430dcb7c9SBarry Smith /* for each nontrival globally owned node list all arriving processors */ 765bc8ff85bSBarry Smith for (i=0; i<nrecvs; i++) { 766bc8ff85bSBarry Smith for (j=0; j<len[i]; j++) { 76730dcb7c9SBarry Smith node = recvs[2*i*nmax+2*j]-rstart; 768f6e5521dSKarl Rupp if (nownedsenders[node] > 1) ownedsenders[starts[node]++] = source[i]; 769bc8ff85bSBarry Smith } 770bc8ff85bSBarry Smith } 771bc8ff85bSBarry Smith 77207b52d57SBarry Smith if (debug) { /* ----------------------------------- */ 77330dcb7c9SBarry Smith starts[0] = 0; 77430dcb7c9SBarry Smith for (i=1; i<ng; i++) { 77530dcb7c9SBarry Smith if (nownedsenders[i-1] > 1) starts[i] = starts[i-1] + nownedsenders[i-1]; 77630dcb7c9SBarry Smith else starts[i] = starts[i-1]; 77730dcb7c9SBarry Smith } 77830dcb7c9SBarry Smith for (i=0; i<ng; i++) { 77930dcb7c9SBarry Smith if (nownedsenders[i] > 1) { 7807904a332SBarry Smith ierr = PetscSynchronizedPrintf(comm,"[%d] global node %D local owner processors: ",rank,i+rstart);CHKERRQ(ierr); 78130dcb7c9SBarry Smith for (j=0; j<nownedsenders[i]; j++) { 7827904a332SBarry Smith ierr = PetscSynchronizedPrintf(comm,"%D ",ownedsenders[starts[i]+j]);CHKERRQ(ierr); 78330dcb7c9SBarry Smith } 78430dcb7c9SBarry Smith ierr = PetscSynchronizedPrintf(comm,"\n");CHKERRQ(ierr); 78530dcb7c9SBarry Smith } 78630dcb7c9SBarry Smith } 7870ec8b6e3SBarry Smith ierr = PetscSynchronizedFlush(comm,PETSC_STDOUT);CHKERRQ(ierr); 78807b52d57SBarry Smith } /* ----------------------------------- */ 78930dcb7c9SBarry Smith 7903677ff5aSBarry Smith /* wait on original sends */ 7913a96401aSBarry Smith if (nsends) { 792785e854fSJed Brown ierr = PetscMalloc1(nsends,&send_status);CHKERRQ(ierr); 7933a96401aSBarry Smith ierr = MPI_Waitall(nsends,send_waits,send_status);CHKERRQ(ierr); 7943a96401aSBarry Smith ierr = PetscFree(send_status);CHKERRQ(ierr); 7953a96401aSBarry Smith } 79689d82c54SBarry Smith ierr = PetscFree(send_waits);CHKERRQ(ierr); 7973a96401aSBarry Smith ierr = PetscFree(sends);CHKERRQ(ierr); 7983677ff5aSBarry Smith ierr = PetscFree(nprocs);CHKERRQ(ierr); 7993677ff5aSBarry Smith 8003677ff5aSBarry Smith /* pack messages to send back to local owners */ 80130dcb7c9SBarry Smith starts[0] = 0; 80230dcb7c9SBarry Smith for (i=1; i<ng; i++) { 80330dcb7c9SBarry Smith if (nownedsenders[i-1] > 1) starts[i] = starts[i-1] + nownedsenders[i-1]; 80430dcb7c9SBarry Smith else starts[i] = starts[i-1]; 80530dcb7c9SBarry Smith } 80630dcb7c9SBarry Smith nsends2 = nrecvs; 807785e854fSJed Brown ierr = PetscMalloc1((nsends2+1),&nprocs);CHKERRQ(ierr); /* length of each message */ 80830dcb7c9SBarry Smith for (i=0; i<nrecvs; i++) { 80930dcb7c9SBarry Smith nprocs[i] = 1; 81030dcb7c9SBarry Smith for (j=0; j<len[i]; j++) { 81130dcb7c9SBarry Smith node = recvs[2*i*nmax+2*j]-rstart; 812f6e5521dSKarl Rupp if (nownedsenders[node] > 1) nprocs[i] += 2 + nownedsenders[node]; 81330dcb7c9SBarry Smith } 81430dcb7c9SBarry Smith } 815f6e5521dSKarl Rupp nt = 0; 816f6e5521dSKarl Rupp for (i=0; i<nsends2; i++) nt += nprocs[i]; 817f6e5521dSKarl Rupp 818785e854fSJed Brown ierr = PetscMalloc1((nt+1),&sends2);CHKERRQ(ierr); 819785e854fSJed Brown ierr = PetscMalloc1((nsends2+1),&starts2);CHKERRQ(ierr); 820f6e5521dSKarl Rupp 821f6e5521dSKarl Rupp starts2[0] = 0; 822f6e5521dSKarl Rupp for (i=1; i<nsends2; i++) starts2[i] = starts2[i-1] + nprocs[i-1]; 82330dcb7c9SBarry Smith /* 82430dcb7c9SBarry Smith Each message is 1 + nprocs[i] long, and consists of 82530dcb7c9SBarry Smith (0) the number of nodes being sent back 82630dcb7c9SBarry Smith (1) the local node number, 82730dcb7c9SBarry Smith (2) the number of processors sharing it, 82830dcb7c9SBarry Smith (3) the processors sharing it 82930dcb7c9SBarry Smith */ 83030dcb7c9SBarry Smith for (i=0; i<nsends2; i++) { 83130dcb7c9SBarry Smith cnt = 1; 83230dcb7c9SBarry Smith sends2[starts2[i]] = 0; 83330dcb7c9SBarry Smith for (j=0; j<len[i]; j++) { 83430dcb7c9SBarry Smith node = recvs[2*i*nmax+2*j]-rstart; 83530dcb7c9SBarry Smith if (nownedsenders[node] > 1) { 83630dcb7c9SBarry Smith sends2[starts2[i]]++; 83730dcb7c9SBarry Smith sends2[starts2[i]+cnt++] = recvs[2*i*nmax+2*j+1]; 83830dcb7c9SBarry Smith sends2[starts2[i]+cnt++] = nownedsenders[node]; 83932dcc486SBarry Smith ierr = PetscMemcpy(&sends2[starts2[i]+cnt],&ownedsenders[starts[node]],nownedsenders[node]*sizeof(PetscInt));CHKERRQ(ierr); 84030dcb7c9SBarry Smith cnt += nownedsenders[node]; 84130dcb7c9SBarry Smith } 84230dcb7c9SBarry Smith } 84330dcb7c9SBarry Smith } 84430dcb7c9SBarry Smith 84530dcb7c9SBarry Smith /* receive the message lengths */ 84630dcb7c9SBarry Smith nrecvs2 = nsends; 847785e854fSJed Brown ierr = PetscMalloc1((nrecvs2+1),&lens2);CHKERRQ(ierr); 848785e854fSJed Brown ierr = PetscMalloc1((nrecvs2+1),&starts3);CHKERRQ(ierr); 849785e854fSJed Brown ierr = PetscMalloc1((nrecvs2+1),&recv_waits);CHKERRQ(ierr); 85030dcb7c9SBarry Smith for (i=0; i<nrecvs2; i++) { 851d44834fbSBarry Smith ierr = MPI_Irecv(&lens2[i],1,MPIU_INT,dest[i],tag2,comm,recv_waits+i);CHKERRQ(ierr); 85230dcb7c9SBarry Smith } 853d44834fbSBarry Smith 8548a8e0b3aSBarry Smith /* send the message lengths */ 8558a8e0b3aSBarry Smith for (i=0; i<nsends2; i++) { 8568a8e0b3aSBarry Smith ierr = MPI_Send(&nprocs[i],1,MPIU_INT,source[i],tag2,comm);CHKERRQ(ierr); 8578a8e0b3aSBarry Smith } 8588a8e0b3aSBarry Smith 859d44834fbSBarry Smith /* wait on receives of lens */ 8600c468ba9SBarry Smith if (nrecvs2) { 861785e854fSJed Brown ierr = PetscMalloc1(nrecvs2,&recv_statuses);CHKERRQ(ierr); 862d44834fbSBarry Smith ierr = MPI_Waitall(nrecvs2,recv_waits,recv_statuses);CHKERRQ(ierr); 863d44834fbSBarry Smith ierr = PetscFree(recv_statuses);CHKERRQ(ierr); 8640c468ba9SBarry Smith } 865a2ea699eSBarry Smith ierr = PetscFree(recv_waits);CHKERRQ(ierr); 866d44834fbSBarry Smith 86730dcb7c9SBarry Smith starts3[0] = 0; 868d44834fbSBarry Smith nt = 0; 86930dcb7c9SBarry Smith for (i=0; i<nrecvs2-1; i++) { 87030dcb7c9SBarry Smith starts3[i+1] = starts3[i] + lens2[i]; 871d44834fbSBarry Smith nt += lens2[i]; 87230dcb7c9SBarry Smith } 87376466f69SStefano Zampini if (nrecvs2) nt += lens2[nrecvs2-1]; 874d44834fbSBarry Smith 875785e854fSJed Brown ierr = PetscMalloc1((nt+1),&recvs2);CHKERRQ(ierr); 876785e854fSJed Brown ierr = PetscMalloc1((nrecvs2+1),&recv_waits);CHKERRQ(ierr); 87752b72c4aSBarry Smith for (i=0; i<nrecvs2; i++) { 87832dcc486SBarry Smith ierr = MPI_Irecv(recvs2+starts3[i],lens2[i],MPIU_INT,dest[i],tag3,comm,recv_waits+i);CHKERRQ(ierr); 87930dcb7c9SBarry Smith } 88030dcb7c9SBarry Smith 88130dcb7c9SBarry Smith /* send the messages */ 882785e854fSJed Brown ierr = PetscMalloc1((nsends2+1),&send_waits);CHKERRQ(ierr); 88330dcb7c9SBarry Smith for (i=0; i<nsends2; i++) { 88432dcc486SBarry Smith ierr = MPI_Isend(sends2+starts2[i],nprocs[i],MPIU_INT,source[i],tag3,comm,send_waits+i);CHKERRQ(ierr); 88530dcb7c9SBarry Smith } 88630dcb7c9SBarry Smith 88730dcb7c9SBarry Smith /* wait on receives */ 8880c468ba9SBarry Smith if (nrecvs2) { 889785e854fSJed Brown ierr = PetscMalloc1(nrecvs2,&recv_statuses);CHKERRQ(ierr); 89030dcb7c9SBarry Smith ierr = MPI_Waitall(nrecvs2,recv_waits,recv_statuses);CHKERRQ(ierr); 89130dcb7c9SBarry Smith ierr = PetscFree(recv_statuses);CHKERRQ(ierr); 8920c468ba9SBarry Smith } 89330dcb7c9SBarry Smith ierr = PetscFree(recv_waits);CHKERRQ(ierr); 89430dcb7c9SBarry Smith ierr = PetscFree(nprocs);CHKERRQ(ierr); 89530dcb7c9SBarry Smith 89607b52d57SBarry Smith if (debug) { /* ----------------------------------- */ 89730dcb7c9SBarry Smith cnt = 0; 89830dcb7c9SBarry Smith for (i=0; i<nrecvs2; i++) { 89930dcb7c9SBarry Smith nt = recvs2[cnt++]; 90030dcb7c9SBarry Smith for (j=0; j<nt; j++) { 9017904a332SBarry Smith ierr = PetscSynchronizedPrintf(comm,"[%d] local node %D number of subdomains %D: ",rank,recvs2[cnt],recvs2[cnt+1]);CHKERRQ(ierr); 90230dcb7c9SBarry Smith for (k=0; k<recvs2[cnt+1]; k++) { 9037904a332SBarry Smith ierr = PetscSynchronizedPrintf(comm,"%D ",recvs2[cnt+2+k]);CHKERRQ(ierr); 90430dcb7c9SBarry Smith } 90530dcb7c9SBarry Smith cnt += 2 + recvs2[cnt+1]; 90630dcb7c9SBarry Smith ierr = PetscSynchronizedPrintf(comm,"\n");CHKERRQ(ierr); 90730dcb7c9SBarry Smith } 90830dcb7c9SBarry Smith } 9090ec8b6e3SBarry Smith ierr = PetscSynchronizedFlush(comm,PETSC_STDOUT);CHKERRQ(ierr); 91007b52d57SBarry Smith } /* ----------------------------------- */ 91130dcb7c9SBarry Smith 91230dcb7c9SBarry Smith /* count number subdomains for each local node */ 913785e854fSJed Brown ierr = PetscMalloc1(size,&nprocs);CHKERRQ(ierr); 91432dcc486SBarry Smith ierr = PetscMemzero(nprocs,size*sizeof(PetscInt));CHKERRQ(ierr); 91530dcb7c9SBarry Smith cnt = 0; 91630dcb7c9SBarry Smith for (i=0; i<nrecvs2; i++) { 91730dcb7c9SBarry Smith nt = recvs2[cnt++]; 91830dcb7c9SBarry Smith for (j=0; j<nt; j++) { 919f6e5521dSKarl Rupp for (k=0; k<recvs2[cnt+1]; k++) nprocs[recvs2[cnt+2+k]]++; 92030dcb7c9SBarry Smith cnt += 2 + recvs2[cnt+1]; 92130dcb7c9SBarry Smith } 92230dcb7c9SBarry Smith } 92330dcb7c9SBarry Smith nt = 0; for (i=0; i<size; i++) nt += (nprocs[i] > 0); 92430dcb7c9SBarry Smith *nproc = nt; 925785e854fSJed Brown ierr = PetscMalloc1((nt+1),procs);CHKERRQ(ierr); 926785e854fSJed Brown ierr = PetscMalloc1((nt+1),numprocs);CHKERRQ(ierr); 927785e854fSJed Brown ierr = PetscMalloc1((nt+1),indices);CHKERRQ(ierr); 9280298fd71SBarry Smith for (i=0;i<nt+1;i++) (*indices)[i]=NULL; 929785e854fSJed Brown ierr = PetscMalloc1(size,&bprocs);CHKERRQ(ierr); 93030dcb7c9SBarry Smith cnt = 0; 93130dcb7c9SBarry Smith for (i=0; i<size; i++) { 93230dcb7c9SBarry Smith if (nprocs[i] > 0) { 93330dcb7c9SBarry Smith bprocs[i] = cnt; 93430dcb7c9SBarry Smith (*procs)[cnt] = i; 93530dcb7c9SBarry Smith (*numprocs)[cnt] = nprocs[i]; 936785e854fSJed Brown ierr = PetscMalloc1(nprocs[i],&(*indices)[cnt]);CHKERRQ(ierr); 93730dcb7c9SBarry Smith cnt++; 93830dcb7c9SBarry Smith } 93930dcb7c9SBarry Smith } 94030dcb7c9SBarry Smith 94130dcb7c9SBarry Smith /* make the list of subdomains for each nontrivial local node */ 94232dcc486SBarry Smith ierr = PetscMemzero(*numprocs,nt*sizeof(PetscInt));CHKERRQ(ierr); 94330dcb7c9SBarry Smith cnt = 0; 94430dcb7c9SBarry Smith for (i=0; i<nrecvs2; i++) { 94530dcb7c9SBarry Smith nt = recvs2[cnt++]; 94630dcb7c9SBarry Smith for (j=0; j<nt; j++) { 947f6e5521dSKarl Rupp for (k=0; k<recvs2[cnt+1]; k++) (*indices)[bprocs[recvs2[cnt+2+k]]][(*numprocs)[bprocs[recvs2[cnt+2+k]]]++] = recvs2[cnt]; 94830dcb7c9SBarry Smith cnt += 2 + recvs2[cnt+1]; 94930dcb7c9SBarry Smith } 95030dcb7c9SBarry Smith } 95130dcb7c9SBarry Smith ierr = PetscFree(bprocs);CHKERRQ(ierr); 95207b52d57SBarry Smith ierr = PetscFree(recvs2);CHKERRQ(ierr); 95330dcb7c9SBarry Smith 95407b52d57SBarry Smith /* sort the node indexing by their global numbers */ 95507b52d57SBarry Smith nt = *nproc; 95607b52d57SBarry Smith for (i=0; i<nt; i++) { 957785e854fSJed Brown ierr = PetscMalloc1(((*numprocs)[i]),&tmp);CHKERRQ(ierr); 958f6e5521dSKarl Rupp for (j=0; j<(*numprocs)[i]; j++) tmp[j] = lindices[(*indices)[i][j]]; 95907b52d57SBarry Smith ierr = PetscSortIntWithArray((*numprocs)[i],tmp,(*indices)[i]);CHKERRQ(ierr); 96007b52d57SBarry Smith ierr = PetscFree(tmp);CHKERRQ(ierr); 96107b52d57SBarry Smith } 96207b52d57SBarry Smith 96307b52d57SBarry Smith if (debug) { /* ----------------------------------- */ 96430dcb7c9SBarry Smith nt = *nproc; 96530dcb7c9SBarry Smith for (i=0; i<nt; i++) { 9667904a332SBarry Smith ierr = PetscSynchronizedPrintf(comm,"[%d] subdomain %D number of indices %D: ",rank,(*procs)[i],(*numprocs)[i]);CHKERRQ(ierr); 96730dcb7c9SBarry Smith for (j=0; j<(*numprocs)[i]; j++) { 9687904a332SBarry Smith ierr = PetscSynchronizedPrintf(comm,"%D ",(*indices)[i][j]);CHKERRQ(ierr); 96930dcb7c9SBarry Smith } 97030dcb7c9SBarry Smith ierr = PetscSynchronizedPrintf(comm,"\n");CHKERRQ(ierr); 97130dcb7c9SBarry Smith } 9720ec8b6e3SBarry Smith ierr = PetscSynchronizedFlush(comm,PETSC_STDOUT);CHKERRQ(ierr); 97307b52d57SBarry Smith } /* ----------------------------------- */ 97430dcb7c9SBarry Smith 97530dcb7c9SBarry Smith /* wait on sends */ 97630dcb7c9SBarry Smith if (nsends2) { 977785e854fSJed Brown ierr = PetscMalloc1(nsends2,&send_status);CHKERRQ(ierr); 97830dcb7c9SBarry Smith ierr = MPI_Waitall(nsends2,send_waits,send_status);CHKERRQ(ierr); 97930dcb7c9SBarry Smith ierr = PetscFree(send_status);CHKERRQ(ierr); 98030dcb7c9SBarry Smith } 98130dcb7c9SBarry Smith 98230dcb7c9SBarry Smith ierr = PetscFree(starts3);CHKERRQ(ierr); 98330dcb7c9SBarry Smith ierr = PetscFree(dest);CHKERRQ(ierr); 98430dcb7c9SBarry Smith ierr = PetscFree(send_waits);CHKERRQ(ierr); 9853677ff5aSBarry Smith 986bc8ff85bSBarry Smith ierr = PetscFree(nownedsenders);CHKERRQ(ierr); 987bc8ff85bSBarry Smith ierr = PetscFree(ownedsenders);CHKERRQ(ierr); 988bc8ff85bSBarry Smith ierr = PetscFree(starts);CHKERRQ(ierr); 98930dcb7c9SBarry Smith ierr = PetscFree(starts2);CHKERRQ(ierr); 99030dcb7c9SBarry Smith ierr = PetscFree(lens2);CHKERRQ(ierr); 99189d82c54SBarry Smith 99289d82c54SBarry Smith ierr = PetscFree(source);CHKERRQ(ierr); 99397f1f81fSBarry Smith ierr = PetscFree(len);CHKERRQ(ierr); 99489d82c54SBarry Smith ierr = PetscFree(recvs);CHKERRQ(ierr); 9953a96401aSBarry Smith ierr = PetscFree(nprocs);CHKERRQ(ierr); 99630dcb7c9SBarry Smith ierr = PetscFree(sends2);CHKERRQ(ierr); 99724cf384cSBarry Smith 99824cf384cSBarry Smith /* put the information about myself as the first entry in the list */ 99924cf384cSBarry Smith first_procs = (*procs)[0]; 100024cf384cSBarry Smith first_numprocs = (*numprocs)[0]; 100124cf384cSBarry Smith first_indices = (*indices)[0]; 100224cf384cSBarry Smith for (i=0; i<*nproc; i++) { 100324cf384cSBarry Smith if ((*procs)[i] == rank) { 100424cf384cSBarry Smith (*procs)[0] = (*procs)[i]; 100524cf384cSBarry Smith (*numprocs)[0] = (*numprocs)[i]; 100624cf384cSBarry Smith (*indices)[0] = (*indices)[i]; 100724cf384cSBarry Smith (*procs)[i] = first_procs; 100824cf384cSBarry Smith (*numprocs)[i] = first_numprocs; 100924cf384cSBarry Smith (*indices)[i] = first_indices; 101024cf384cSBarry Smith break; 101124cf384cSBarry Smith } 101224cf384cSBarry Smith } 101389d82c54SBarry Smith PetscFunctionReturn(0); 101489d82c54SBarry Smith } 101589d82c54SBarry Smith 10164a2ae208SSatish Balay #undef __FUNCT__ 10174a2ae208SSatish Balay #define __FUNCT__ "ISLocalToGlobalMappingRestoreInfo" 101807b52d57SBarry Smith /*@C 101907b52d57SBarry Smith ISLocalToGlobalMappingRestoreInfo - Frees the memory allocated by ISLocalToGlobalMappingGetInfo() 102089d82c54SBarry Smith 102107b52d57SBarry Smith Collective on ISLocalToGlobalMapping 102207b52d57SBarry Smith 102307b52d57SBarry Smith Input Parameters: 102407b52d57SBarry Smith . mapping - the mapping from local to global indexing 102507b52d57SBarry Smith 102607b52d57SBarry Smith Output Parameter: 102707b52d57SBarry Smith + nproc - number of processors that are connected to this one 102807b52d57SBarry Smith . proc - neighboring processors 102907b52d57SBarry Smith . numproc - number of indices for each processor 103007b52d57SBarry Smith - indices - indices of local nodes shared with neighbor (sorted by global numbering) 103107b52d57SBarry Smith 103207b52d57SBarry Smith Level: advanced 103307b52d57SBarry Smith 103407b52d57SBarry Smith .seealso: ISLocalToGlobalMappingDestroy(), ISLocalToGlobalMappingCreateIS(), ISLocalToGlobalMappingCreate(), 103507b52d57SBarry Smith ISLocalToGlobalMappingGetInfo() 103607b52d57SBarry Smith @*/ 10377087cfbeSBarry Smith PetscErrorCode ISLocalToGlobalMappingRestoreInfo(ISLocalToGlobalMapping mapping,PetscInt *nproc,PetscInt *procs[],PetscInt *numprocs[],PetscInt **indices[]) 103807b52d57SBarry Smith { 10396849ba73SBarry Smith PetscErrorCode ierr; 104032dcc486SBarry Smith PetscInt i; 104107b52d57SBarry Smith 104207b52d57SBarry Smith PetscFunctionBegin; 104305b42c5fSBarry Smith ierr = PetscFree(*procs);CHKERRQ(ierr); 104405b42c5fSBarry Smith ierr = PetscFree(*numprocs);CHKERRQ(ierr); 104500ff320aSBarry Smith if (*indices) { 104605b42c5fSBarry Smith ierr = PetscFree((*indices)[0]);CHKERRQ(ierr); 104700ff320aSBarry Smith for (i=1; i<*nproc; i++) { 104805b42c5fSBarry Smith ierr = PetscFree((*indices)[i]);CHKERRQ(ierr); 104907b52d57SBarry Smith } 105007b52d57SBarry Smith ierr = PetscFree(*indices);CHKERRQ(ierr); 105124cf384cSBarry Smith } 105207b52d57SBarry Smith PetscFunctionReturn(0); 105307b52d57SBarry Smith } 105486994e45SJed Brown 105586994e45SJed Brown #undef __FUNCT__ 105686994e45SJed Brown #define __FUNCT__ "ISLocalToGlobalMappingGetIndices" 105786994e45SJed Brown /*@C 1058107e9a97SBarry Smith ISLocalToGlobalMappingGetIndices - Get global indices for every local point that is mapped 105986994e45SJed Brown 106086994e45SJed Brown Not Collective 106186994e45SJed Brown 106286994e45SJed Brown Input Arguments: 106386994e45SJed Brown . ltog - local to global mapping 106486994e45SJed Brown 106586994e45SJed Brown Output Arguments: 1066565245c5SBarry Smith . array - array of indices, the length of this array may be obtained with ISLocalToGlobalMappingGetSize() 106786994e45SJed Brown 106886994e45SJed Brown Level: advanced 106986994e45SJed Brown 1070107e9a97SBarry Smith Notes: ISLocalToGlobalMappingGetSize() returns the length the this array 1071107e9a97SBarry Smith 1072107e9a97SBarry Smith .seealso: ISLocalToGlobalMappingCreate(), ISLocalToGlobalMappingApply(), ISLocalToGlobalMappingRestoreIndices(), ISLocalToGlobalMappingGetBlockIndices(), ISLocalToGlobalMappingRestoreBlockIndices() 107386994e45SJed Brown @*/ 10747087cfbeSBarry Smith PetscErrorCode ISLocalToGlobalMappingGetIndices(ISLocalToGlobalMapping ltog,const PetscInt **array) 107586994e45SJed Brown { 107686994e45SJed Brown PetscFunctionBegin; 107786994e45SJed Brown PetscValidHeaderSpecific(ltog,IS_LTOGM_CLASSID,1); 107886994e45SJed Brown PetscValidPointer(array,2); 107945b6f7e9SBarry Smith if (ltog->bs == 1) { 108086994e45SJed Brown *array = ltog->indices; 108145b6f7e9SBarry Smith } else { 108245b6f7e9SBarry Smith PetscInt *jj,k,i,j,n = ltog->n, bs = ltog->bs; 108345b6f7e9SBarry Smith const PetscInt *ii; 108445b6f7e9SBarry Smith PetscErrorCode ierr; 108545b6f7e9SBarry Smith 108645b6f7e9SBarry Smith ierr = PetscMalloc1(bs*n,&jj);CHKERRQ(ierr); 108745b6f7e9SBarry Smith *array = jj; 108845b6f7e9SBarry Smith k = 0; 108945b6f7e9SBarry Smith ii = ltog->indices; 109045b6f7e9SBarry Smith for (i=0; i<n; i++) 109145b6f7e9SBarry Smith for (j=0; j<bs; j++) 109245b6f7e9SBarry Smith jj[k++] = bs*ii[i] + j; 109345b6f7e9SBarry Smith } 109486994e45SJed Brown PetscFunctionReturn(0); 109586994e45SJed Brown } 109686994e45SJed Brown 109786994e45SJed Brown #undef __FUNCT__ 109886994e45SJed Brown #define __FUNCT__ "ISLocalToGlobalMappingRestoreIndices" 109986994e45SJed Brown /*@C 110086994e45SJed Brown ISLocalToGlobalMappingRestoreIndices - Restore indices obtained with ISLocalToGlobalMappingRestoreIndices() 110186994e45SJed Brown 110286994e45SJed Brown Not Collective 110386994e45SJed Brown 110486994e45SJed Brown Input Arguments: 110586994e45SJed Brown + ltog - local to global mapping 110686994e45SJed Brown - array - array of indices 110786994e45SJed Brown 110886994e45SJed Brown Level: advanced 110986994e45SJed Brown 111086994e45SJed Brown .seealso: ISLocalToGlobalMappingCreate(), ISLocalToGlobalMappingApply(), ISLocalToGlobalMappingGetIndices() 111186994e45SJed Brown @*/ 11127087cfbeSBarry Smith PetscErrorCode ISLocalToGlobalMappingRestoreIndices(ISLocalToGlobalMapping ltog,const PetscInt **array) 111386994e45SJed Brown { 111486994e45SJed Brown PetscFunctionBegin; 111586994e45SJed Brown PetscValidHeaderSpecific(ltog,IS_LTOGM_CLASSID,1); 111686994e45SJed Brown PetscValidPointer(array,2); 111745b6f7e9SBarry Smith if (ltog->bs == 1 && *array != ltog->indices) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_BADPTR,"Trying to return mismatched pointer"); 111845b6f7e9SBarry Smith 111945b6f7e9SBarry Smith if (ltog->bs > 1) { 112045b6f7e9SBarry Smith PetscErrorCode ierr; 112145b6f7e9SBarry Smith ierr = PetscFree(*(void**)array);CHKERRQ(ierr); 112245b6f7e9SBarry Smith } 112345b6f7e9SBarry Smith PetscFunctionReturn(0); 112445b6f7e9SBarry Smith } 112545b6f7e9SBarry Smith 112645b6f7e9SBarry Smith #undef __FUNCT__ 112745b6f7e9SBarry Smith #define __FUNCT__ "ISLocalToGlobalMappingGetBlockIndices" 112845b6f7e9SBarry Smith /*@C 112945b6f7e9SBarry Smith ISLocalToGlobalMappingGetBlockIndices - Get global indices for every local block 113045b6f7e9SBarry Smith 113145b6f7e9SBarry Smith Not Collective 113245b6f7e9SBarry Smith 113345b6f7e9SBarry Smith Input Arguments: 113445b6f7e9SBarry Smith . ltog - local to global mapping 113545b6f7e9SBarry Smith 113645b6f7e9SBarry Smith Output Arguments: 113745b6f7e9SBarry Smith . array - array of indices 113845b6f7e9SBarry Smith 113945b6f7e9SBarry Smith Level: advanced 114045b6f7e9SBarry Smith 114145b6f7e9SBarry Smith .seealso: ISLocalToGlobalMappingCreate(), ISLocalToGlobalMappingApply(), ISLocalToGlobalMappingRestoreBlockIndices() 114245b6f7e9SBarry Smith @*/ 114345b6f7e9SBarry Smith PetscErrorCode ISLocalToGlobalMappingGetBlockIndices(ISLocalToGlobalMapping ltog,const PetscInt **array) 114445b6f7e9SBarry Smith { 114545b6f7e9SBarry Smith PetscFunctionBegin; 114645b6f7e9SBarry Smith PetscValidHeaderSpecific(ltog,IS_LTOGM_CLASSID,1); 114745b6f7e9SBarry Smith PetscValidPointer(array,2); 114845b6f7e9SBarry Smith *array = ltog->indices; 114945b6f7e9SBarry Smith PetscFunctionReturn(0); 115045b6f7e9SBarry Smith } 115145b6f7e9SBarry Smith 115245b6f7e9SBarry Smith #undef __FUNCT__ 115345b6f7e9SBarry Smith #define __FUNCT__ "ISLocalToGlobalMappingRestoreBlockIndices" 115445b6f7e9SBarry Smith /*@C 115545b6f7e9SBarry Smith ISLocalToGlobalMappingRestoreBlockIndices - Restore indices obtained with ISLocalToGlobalMappingGetBlockIndices() 115645b6f7e9SBarry Smith 115745b6f7e9SBarry Smith Not Collective 115845b6f7e9SBarry Smith 115945b6f7e9SBarry Smith Input Arguments: 116045b6f7e9SBarry Smith + ltog - local to global mapping 116145b6f7e9SBarry Smith - array - array of indices 116245b6f7e9SBarry Smith 116345b6f7e9SBarry Smith Level: advanced 116445b6f7e9SBarry Smith 116545b6f7e9SBarry Smith .seealso: ISLocalToGlobalMappingCreate(), ISLocalToGlobalMappingApply(), ISLocalToGlobalMappingGetIndices() 116645b6f7e9SBarry Smith @*/ 116745b6f7e9SBarry Smith PetscErrorCode ISLocalToGlobalMappingRestoreBlockIndices(ISLocalToGlobalMapping ltog,const PetscInt **array) 116845b6f7e9SBarry Smith { 116945b6f7e9SBarry Smith PetscFunctionBegin; 117045b6f7e9SBarry Smith PetscValidHeaderSpecific(ltog,IS_LTOGM_CLASSID,1); 117145b6f7e9SBarry Smith PetscValidPointer(array,2); 117286994e45SJed Brown if (*array != ltog->indices) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_BADPTR,"Trying to return mismatched pointer"); 11730298fd71SBarry Smith *array = NULL; 117486994e45SJed Brown PetscFunctionReturn(0); 117586994e45SJed Brown } 1176f7efa3c7SJed Brown 1177f7efa3c7SJed Brown #undef __FUNCT__ 1178f7efa3c7SJed Brown #define __FUNCT__ "ISLocalToGlobalMappingConcatenate" 1179f7efa3c7SJed Brown /*@C 1180f7efa3c7SJed Brown ISLocalToGlobalMappingConcatenate - Create a new mapping that concatenates a list of mappings 1181f7efa3c7SJed Brown 1182f7efa3c7SJed Brown Not Collective 1183f7efa3c7SJed Brown 1184f7efa3c7SJed Brown Input Arguments: 1185f7efa3c7SJed Brown + comm - communicator for the new mapping, must contain the communicator of every mapping to concatenate 1186f7efa3c7SJed Brown . n - number of mappings to concatenate 1187f7efa3c7SJed Brown - ltogs - local to global mappings 1188f7efa3c7SJed Brown 1189f7efa3c7SJed Brown Output Arguments: 1190f7efa3c7SJed Brown . ltogcat - new mapping 1191f7efa3c7SJed Brown 1192f7efa3c7SJed Brown Level: advanced 1193f7efa3c7SJed Brown 1194f7efa3c7SJed Brown .seealso: ISLocalToGlobalMappingCreate() 1195f7efa3c7SJed Brown @*/ 1196f7efa3c7SJed Brown PetscErrorCode ISLocalToGlobalMappingConcatenate(MPI_Comm comm,PetscInt n,const ISLocalToGlobalMapping ltogs[],ISLocalToGlobalMapping *ltogcat) 1197f7efa3c7SJed Brown { 1198f7efa3c7SJed Brown PetscInt i,cnt,m,*idx; 1199f7efa3c7SJed Brown PetscErrorCode ierr; 1200f7efa3c7SJed Brown 1201f7efa3c7SJed Brown PetscFunctionBegin; 1202f7efa3c7SJed Brown if (n < 0) SETERRQ1(comm,PETSC_ERR_ARG_OUTOFRANGE,"Must have a non-negative number of mappings, given %D",n); 1203f7efa3c7SJed Brown if (n > 0) PetscValidPointer(ltogs,3); 1204f7efa3c7SJed Brown for (i=0; i<n; i++) PetscValidHeaderSpecific(ltogs[i],IS_LTOGM_CLASSID,3); 1205f7efa3c7SJed Brown PetscValidPointer(ltogcat,4); 1206f7efa3c7SJed Brown for (cnt=0,i=0; i<n; i++) { 1207f7efa3c7SJed Brown ierr = ISLocalToGlobalMappingGetSize(ltogs[i],&m);CHKERRQ(ierr); 1208f7efa3c7SJed Brown cnt += m; 1209f7efa3c7SJed Brown } 1210785e854fSJed Brown ierr = PetscMalloc1(cnt,&idx);CHKERRQ(ierr); 1211f7efa3c7SJed Brown for (cnt=0,i=0; i<n; i++) { 1212f7efa3c7SJed Brown const PetscInt *subidx; 1213f7efa3c7SJed Brown ierr = ISLocalToGlobalMappingGetSize(ltogs[i],&m);CHKERRQ(ierr); 1214f7efa3c7SJed Brown ierr = ISLocalToGlobalMappingGetIndices(ltogs[i],&subidx);CHKERRQ(ierr); 1215f7efa3c7SJed Brown ierr = PetscMemcpy(&idx[cnt],subidx,m*sizeof(PetscInt));CHKERRQ(ierr); 1216f7efa3c7SJed Brown ierr = ISLocalToGlobalMappingRestoreIndices(ltogs[i],&subidx);CHKERRQ(ierr); 1217f7efa3c7SJed Brown cnt += m; 1218f7efa3c7SJed Brown } 1219f0413b6fSBarry Smith ierr = ISLocalToGlobalMappingCreate(comm,1,cnt,idx,PETSC_OWN_POINTER,ltogcat);CHKERRQ(ierr); 1220f7efa3c7SJed Brown PetscFunctionReturn(0); 1221f7efa3c7SJed Brown } 122204a59952SBarry Smith 122304a59952SBarry Smith 1224