12eac72dbSBarry Smith /* 2f8256253SLois Curfman McInnes An index set is a generalization of a subset of integers. Index sets 3f8256253SLois Curfman McInnes are used for defining scatters and gathers. 42eac72dbSBarry Smith */ 56524c165SJacob Faibussowitsch #ifndef PETSCIS_H 626bd1501SBarry Smith #define PETSCIS_H 7ac09b921SBarry Smith 82c8e378dSBarry Smith #include <petscsys.h> 90c312b8eSJed Brown #include <petscsftypes.h> 10ea844a1aSMatthew Knepley #include <petscsectiontypes.h> 114914ba2bSBarry Smith #include <petscistypes.h> /*I "petscis.h" I*/ 122eac72dbSBarry Smith 13ac09b921SBarry Smith /* SUBMANSEC = IS */ 14ac09b921SBarry Smith 1597b48c8fSBarry Smith #define IS_FILE_CLASSID 1211218 16014dd563SJed Brown PETSC_EXTERN PetscClassId IS_CLASSID; 17f0479e8cSBarry Smith 18607a6623SBarry Smith PETSC_EXTERN PetscErrorCode ISInitializePackage(void); 192b6de112SBarry Smith 2076bdecfbSBarry Smith /*J 218f6c3df8SBarry Smith ISType - String with the name of a PETSc index set type 2227bdab1eSBarry Smith 2316a05f60SBarry Smith Values: 249c89aa79SPierre Jolivet + `ISGENERAL` - the values are stored with an array of indices and generally have no structure 2516a05f60SBarry Smith . `ISSTRIDE` - the values have a simple structure of an initial offset and then a step size between values 2616a05f60SBarry Smith - `ISBLOCK` - values are an array of indices, each representing a block of values 2716a05f60SBarry Smith 2827bdab1eSBarry Smith Level: beginner 2927bdab1eSBarry Smith 3016a05f60SBarry Smith .seealso: `ISSetType()`, `IS`, `ISCreateGeneral()`, `ISCreateStride()`, `ISCreateBlock()`, `ISCreate()`, `ISRegister()`, 3116a05f60SBarry Smith `VecScatterCreate()`, `MatGetSubMatrices()` 3276bdecfbSBarry Smith J*/ 3319fd82e9SBarry Smith typedef const char *ISType; 3427bdab1eSBarry Smith #define ISGENERAL "general" 3527bdab1eSBarry Smith #define ISSTRIDE "stride" 3627bdab1eSBarry Smith #define ISBLOCK "block" 3727bdab1eSBarry Smith 3827bdab1eSBarry Smith /* Dynamic creation and loading functions */ 39140e18c1SBarry Smith PETSC_EXTERN PetscFunctionList ISList; 4019fd82e9SBarry Smith PETSC_EXTERN PetscErrorCode ISSetType(IS, ISType); 4119fd82e9SBarry Smith PETSC_EXTERN PetscErrorCode ISGetType(IS, ISType *); 42bdf89e91SBarry Smith PETSC_EXTERN PetscErrorCode ISRegister(const char[], PetscErrorCode (*)(IS)); 43*da8c939bSJacob Faibussowitsch PETSC_EXTERN PetscErrorCode ISRegisterAll(void); 44014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISCreate(MPI_Comm, IS *); 4527bdab1eSBarry Smith 46014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISDestroy(IS *); 47014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISSetPermutation(IS); 48014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISPermutation(IS, PetscBool *); 49014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISSetIdentity(IS); 50014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISIdentity(IS, PetscBool *); 51014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISContiguousLocal(IS, PetscInt, PetscInt, PetscInt *, PetscBool *); 5208480c60SBarry Smith 532a1da528SToby Isaac /*E 542a1da528SToby Isaac ISInfo - Info that may either be computed or set as known for an index set 552a1da528SToby Isaac 5616a05f60SBarry Smith Level: intermediate 572a1da528SToby Isaac 582a1da528SToby Isaac Developer Notes: 592a1da528SToby Isaac Entries that are negative need not be called collectively by all processes. 602a1da528SToby Isaac 6187497f52SBarry Smith Any additions/changes here MUST also be made in include/petsc/finclude/petscis.h 6287497f52SBarry Smith 6387497f52SBarry Smith Any additions/changes here must also be made in src/vec/vec/interface/dlregisvec.c in ISInfos[] 6487497f52SBarry Smith 6516a05f60SBarry Smith .seealso: `IS`, `ISType`, `ISSetInfo()` 662a1da528SToby Isaac E*/ 679371c9d4SSatish Balay typedef enum { 689371c9d4SSatish Balay IS_INFO_MIN = -1, 692a1da528SToby Isaac IS_SORTED = 0, 702a1da528SToby Isaac IS_UNIQUE = 1, 712a1da528SToby Isaac IS_PERMUTATION = 2, 722a1da528SToby Isaac IS_INTERVAL = 3, 732a1da528SToby Isaac IS_IDENTITY = 4, 749371c9d4SSatish Balay IS_INFO_MAX = 5 759371c9d4SSatish Balay } ISInfo; 762a1da528SToby Isaac 779371c9d4SSatish Balay typedef enum { 789371c9d4SSatish Balay IS_LOCAL, 799371c9d4SSatish Balay IS_GLOBAL 809371c9d4SSatish Balay } ISInfoType; 812a1da528SToby Isaac 822a1da528SToby Isaac PETSC_EXTERN PetscErrorCode ISSetInfo(IS, ISInfo, ISInfoType, PetscBool, PetscBool); 83657dc977SToby Isaac PETSC_EXTERN PetscErrorCode ISGetInfo(IS, ISInfo, ISInfoType, PetscBool, PetscBool *); 842a1da528SToby Isaac PETSC_EXTERN PetscErrorCode ISClearInfoCache(IS, PetscBool); 85014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISGetIndices(IS, const PetscInt *[]); 86014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISRestoreIndices(IS, const PetscInt *[]); 87014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISGetTotalIndices(IS, const PetscInt *[]); 88014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISRestoreTotalIndices(IS, const PetscInt *[]); 89014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISGetNonlocalIndices(IS, const PetscInt *[]); 90014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISRestoreNonlocalIndices(IS, const PetscInt *[]); 91d60670a5SStefano Zampini PETSC_EXTERN PetscErrorCode ISGetNonlocalIS(IS, IS *); 92d60670a5SStefano Zampini PETSC_EXTERN PetscErrorCode ISRestoreNonlocalIS(IS, IS *); 93014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISGetSize(IS, PetscInt *); 94014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISGetLocalSize(IS, PetscInt *); 95014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISInvertPermutation(IS, PetscInt, IS *); 96014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISView(IS, PetscViewer); 97fe2efc57SMark PETSC_EXTERN PetscErrorCode ISViewFromOptions(IS, PetscObject, const char[]); 98235f7792SMatthew G. Knepley PETSC_EXTERN PetscErrorCode ISLoad(IS, PetscViewer); 99014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISEqual(IS, IS, PetscBool *); 100e8386968SVaclav Hapla PETSC_EXTERN PetscErrorCode ISEqualUnsorted(IS, IS, PetscBool *); 101014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISSort(IS); 102b080c0f9SMatthew G. Knepley PETSC_EXTERN PetscErrorCode ISSortRemoveDups(IS); 103014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISSorted(IS, PetscBool *); 104014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISDifference(IS, IS, IS *); 105014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISSum(IS, IS, IS *); 106014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISExpand(IS, IS, IS *); 1073daafeecSToby Isaac PETSC_EXTERN PetscErrorCode ISIntersect(IS, IS, IS *); 108132da990SBarry Smith PETSC_EXTERN PetscErrorCode ISGetMinMax(IS, PetscInt *, PetscInt *); 109612dd529SBarry Smith 110c3c3c9f4SToby Isaac PETSC_EXTERN PetscErrorCode ISLocate(IS, PetscInt, PetscInt *); 1119305a4c7SMatthew G. Knepley PETSC_EXTERN PetscErrorCode ISGetPointRange(IS, PetscInt *, PetscInt *, const PetscInt **); 1129305a4c7SMatthew G. Knepley PETSC_EXTERN PetscErrorCode ISRestorePointRange(IS, PetscInt *, PetscInt *, const PetscInt **); 1139305a4c7SMatthew G. Knepley PETSC_EXTERN PetscErrorCode ISGetPointSubrange(IS, PetscInt, PetscInt, const PetscInt *); 114c3c3c9f4SToby Isaac 115014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISGetBlockSize(IS, PetscInt *); 116014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISSetBlockSize(IS, PetscInt); 117c16cb8f2SBarry Smith 118014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISToGeneral(IS); 11938f40f24SLois Curfman McInnes 120014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISDuplicate(IS, IS *); 121014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISCopy(IS, IS); 1229fdaf958SVaclav Hapla PETSC_EXTERN PetscErrorCode ISShift(IS, PetscInt, IS); 123014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISAllGather(IS, IS *); 124014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISComplement(IS, PetscInt, PetscInt, IS *); 125014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISConcatenate(MPI_Comm, PetscInt, const IS[], IS *); 126bb74729cSDmitry Karpeev PETSC_EXTERN PetscErrorCode ISListToPair(MPI_Comm, PetscInt, IS[], IS *, IS *); 127bb74729cSDmitry Karpeev PETSC_EXTERN PetscErrorCode ISPairToList(IS, IS, PetscInt *, IS *[]); 128bb74729cSDmitry Karpeev PETSC_EXTERN PetscErrorCode ISEmbed(IS, IS, PetscBool, IS *); 1295ea6c424SDmitry Karpeev PETSC_EXTERN PetscErrorCode ISSortPermutation(IS, PetscBool, IS *); 130014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISOnComm(IS, MPI_Comm, PetscCopyMode, IS *); 1316583bcc1SStefano Zampini PETSC_EXTERN PetscErrorCode ISRenumber(IS, IS, PetscInt *, IS *); 1326c04a3c5SFande Kong PETSC_EXTERN PetscErrorCode ISCreateSubIS(IS, IS, IS *); 133d64ed03dSBarry Smith 13484933e40SVaclav Hapla /* ISGENERAL specific */ 13584933e40SVaclav Hapla PETSC_EXTERN PetscErrorCode ISCreateGeneral(MPI_Comm, PetscInt, const PetscInt[], PetscCopyMode, IS *); 13684933e40SVaclav Hapla PETSC_EXTERN PetscErrorCode ISGeneralSetIndices(IS, PetscInt, const PetscInt[], PetscCopyMode); 1379bc6f2f1SVaclav Hapla PETSC_EXTERN PetscErrorCode ISGeneralSetIndicesFromMask(IS, PetscInt, PetscInt, const PetscBool[]); 13853bb9c4cSVaclav Hapla PETSC_EXTERN PetscErrorCode ISGeneralFilter(IS, PetscInt, PetscInt); 13953bb9c4cSVaclav Hapla 14084933e40SVaclav Hapla /* ISBLOCK specific */ 14184933e40SVaclav Hapla PETSC_EXTERN PetscErrorCode ISCreateBlock(MPI_Comm, PetscInt, PetscInt, const PetscInt[], PetscCopyMode, IS *); 14284933e40SVaclav Hapla PETSC_EXTERN PetscErrorCode ISBlockSetIndices(IS, PetscInt, PetscInt, const PetscInt[], PetscCopyMode); 14384933e40SVaclav Hapla PETSC_EXTERN PetscErrorCode ISBlockGetIndices(IS, const PetscInt *[]); 14484933e40SVaclav Hapla PETSC_EXTERN PetscErrorCode ISBlockRestoreIndices(IS, const PetscInt *[]); 14584933e40SVaclav Hapla PETSC_EXTERN PetscErrorCode ISBlockGetLocalSize(IS, PetscInt *); 14684933e40SVaclav Hapla PETSC_EXTERN PetscErrorCode ISBlockGetSize(IS, PetscInt *); 14784933e40SVaclav Hapla 14884933e40SVaclav Hapla /* ISSTRIDE specific */ 14984933e40SVaclav Hapla PETSC_EXTERN PetscErrorCode ISCreateStride(MPI_Comm, PetscInt, PetscInt, PetscInt, IS *); 15084933e40SVaclav Hapla PETSC_EXTERN PetscErrorCode ISStrideSetStride(IS, PetscInt, PetscInt, PetscInt); 15184933e40SVaclav Hapla PETSC_EXTERN PetscErrorCode ISStrideGetInfo(IS, PetscInt *, PetscInt *); 15284933e40SVaclav Hapla 153014dd563SJed Brown PETSC_EXTERN PetscClassId IS_LTOGM_CLASSID; 15456cd22aeSBarry Smith 1555c20da3cSBarry Smith /*E 1560040bde1SJunchao Zhang ISGlobalToLocalMappingMode - Indicates mapping behavior if global indices are missing 1575c20da3cSBarry Smith 15816a05f60SBarry Smith Values: 15916a05f60SBarry Smith + `IS_GTOLM_MASK` - missing global indices are masked by mapping them to a local index of -1 16016a05f60SBarry Smith - `IS_GTOLM_DROP` - missing global indices are dropped 1615c20da3cSBarry Smith 1625c20da3cSBarry Smith Level: beginner 1635c20da3cSBarry Smith 164db781477SPatrick Sanan .seealso: `ISGlobalToLocalMappingApplyBlock()`, `ISGlobalToLocalMappingApply()` 1655c20da3cSBarry Smith E*/ 1669371c9d4SSatish Balay typedef enum { 1679371c9d4SSatish Balay IS_GTOLM_MASK, 1689371c9d4SSatish Balay IS_GTOLM_DROP 1699371c9d4SSatish Balay } ISGlobalToLocalMappingMode; 17090f02eecSBarry Smith 171413f72f0SBarry Smith /*J 172413f72f0SBarry Smith ISLocalToGlobalMappingType - String with the name of a mapping method 173413f72f0SBarry Smith 17416a05f60SBarry Smith Values: 17516a05f60SBarry Smith + `ISLOCALTOGLOBALMAPPINGBASIC` - a non-memory scalable way of storing `ISLocalToGlobalMapping` that allows applying `ISGlobalToLocalMappingApply()` efficiently 17616a05f60SBarry Smith - `ISLOCALTOGLOBALMAPPINGHASH` - a memory scalable way of storing `ISLocalToGlobalMapping` that allows applying `ISGlobalToLocalMappingApply()` reasonably efficiently 17716a05f60SBarry Smith 178413f72f0SBarry Smith Level: beginner 179413f72f0SBarry Smith 18016a05f60SBarry Smith .seealso: `ISLocalToGlobalMapping`, `ISLocalToGlobalMappingSetType()`, `ISLocalToGlobalSetFromOptions()`, `ISGlobalToLocalMappingMode` 181413f72f0SBarry Smith J*/ 182413f72f0SBarry Smith typedef const char *ISLocalToGlobalMappingType; 183413f72f0SBarry Smith #define ISLOCALTOGLOBALMAPPINGBASIC "basic" 184413f72f0SBarry Smith #define ISLOCALTOGLOBALMAPPINGHASH "hash" 185413f72f0SBarry Smith 186413f72f0SBarry Smith PETSC_EXTERN PetscErrorCode ISLocalToGlobalMappingSetType(ISLocalToGlobalMapping, ISLocalToGlobalMappingType); 187a0d79125SStefano Zampini PETSC_EXTERN PetscErrorCode ISLocalToGlobalMappingGetType(ISLocalToGlobalMapping, ISLocalToGlobalMappingType *); 1881d36bdfdSBarry Smith PETSC_EXTERN PetscErrorCode ISLocalToGlobalMappingRegister(const char[], PetscErrorCode (*)(ISLocalToGlobalMapping)); 189413f72f0SBarry Smith PETSC_EXTERN PetscErrorCode ISLocalToGlobalMappingRegisterAll(void); 190f0413b6fSBarry Smith PETSC_EXTERN PetscErrorCode ISLocalToGlobalMappingCreate(MPI_Comm, PetscInt, PetscInt, const PetscInt[], PetscCopyMode, ISLocalToGlobalMapping *); 191014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISLocalToGlobalMappingCreateIS(IS, ISLocalToGlobalMapping *); 192014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISLocalToGlobalMappingCreateSF(PetscSF, PetscInt, ISLocalToGlobalMapping *); 1937e99dc12SLawrence Mitchell PETSC_EXTERN PetscErrorCode ISLocalToGlobalMappingSetFromOptions(ISLocalToGlobalMapping); 194413f72f0SBarry Smith PETSC_EXTERN PetscErrorCode ISLocalToGlobalMappingSetUp(ISLocalToGlobalMapping); 195014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISLocalToGlobalMappingView(ISLocalToGlobalMapping, PetscViewer); 196fe2efc57SMark PETSC_EXTERN PetscErrorCode ISLocalToGlobalMappingViewFromOptions(ISLocalToGlobalMapping, PetscObject, const char[]); 197662df134SStefano Zampini 198014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISLocalToGlobalMappingDestroy(ISLocalToGlobalMapping *); 19904a59952SBarry Smith PETSC_EXTERN PetscErrorCode ISLocalToGlobalMappingApply(ISLocalToGlobalMapping, PetscInt, const PetscInt[], PetscInt[]); 20045b6f7e9SBarry Smith PETSC_EXTERN PetscErrorCode ISLocalToGlobalMappingApplyBlock(ISLocalToGlobalMapping, PetscInt, const PetscInt[], PetscInt[]); 201014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISLocalToGlobalMappingApplyIS(ISLocalToGlobalMapping, IS, IS *); 202413f72f0SBarry Smith PETSC_EXTERN PetscErrorCode ISGlobalToLocalMappingApply(ISLocalToGlobalMapping, ISGlobalToLocalMappingMode, PetscInt, const PetscInt[], PetscInt *, PetscInt[]); 203413f72f0SBarry Smith PETSC_EXTERN PetscErrorCode ISGlobalToLocalMappingApplyBlock(ISLocalToGlobalMapping, ISGlobalToLocalMappingMode, PetscInt, const PetscInt[], PetscInt *, PetscInt[]); 204413f72f0SBarry Smith PETSC_EXTERN PetscErrorCode ISGlobalToLocalMappingApplyIS(ISLocalToGlobalMapping, ISGlobalToLocalMappingMode, IS, IS *); 205014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISLocalToGlobalMappingGetSize(ISLocalToGlobalMapping, PetscInt *); 2061bd0b88eSStefano Zampini PETSC_EXTERN PetscErrorCode ISLocalToGlobalMappingGetNodeInfo(ISLocalToGlobalMapping, PetscInt *, PetscInt *[], PetscInt **[]); 2071bd0b88eSStefano Zampini PETSC_EXTERN PetscErrorCode ISLocalToGlobalMappingRestoreNodeInfo(ISLocalToGlobalMapping, PetscInt *, PetscInt *[], PetscInt **[]); 208014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISLocalToGlobalMappingGetInfo(ISLocalToGlobalMapping, PetscInt *, PetscInt *[], PetscInt *[], PetscInt **[]); 209014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISLocalToGlobalMappingRestoreInfo(ISLocalToGlobalMapping, PetscInt *, PetscInt *[], PetscInt *[], PetscInt **[]); 2106a818285SBarry Smith PETSC_EXTERN PetscErrorCode ISLocalToGlobalMappingGetBlockInfo(ISLocalToGlobalMapping, PetscInt *, PetscInt *[], PetscInt *[], PetscInt **[]); 2116a818285SBarry Smith PETSC_EXTERN PetscErrorCode ISLocalToGlobalMappingRestoreBlockInfo(ISLocalToGlobalMapping, PetscInt *, PetscInt *[], PetscInt *[], PetscInt **[]); 212014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISLocalToGlobalMappingGetIndices(ISLocalToGlobalMapping, const PetscInt **); 213014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISLocalToGlobalMappingRestoreIndices(ISLocalToGlobalMapping, const PetscInt **); 21445b6f7e9SBarry Smith PETSC_EXTERN PetscErrorCode ISLocalToGlobalMappingGetBlockIndices(ISLocalToGlobalMapping, const PetscInt **); 21545b6f7e9SBarry Smith PETSC_EXTERN PetscErrorCode ISLocalToGlobalMappingRestoreBlockIndices(ISLocalToGlobalMapping, const PetscInt **); 216014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISLocalToGlobalMappingConcatenate(MPI_Comm, PetscInt, const ISLocalToGlobalMapping[], ISLocalToGlobalMapping *); 21745b6f7e9SBarry Smith PETSC_EXTERN PetscErrorCode ISLocalToGlobalMappingGetBlockSize(ISLocalToGlobalMapping, PetscInt *); 21863fa5c83Sstefano_zampini PETSC_EXTERN PetscErrorCode ISLocalToGlobalMappingSetBlockSize(ISLocalToGlobalMapping, PetscInt); 2196658fb44Sstefano_zampini PETSC_EXTERN PetscErrorCode ISLocalToGlobalMappingDuplicate(ISLocalToGlobalMapping, ISLocalToGlobalMapping *); 22045b6f7e9SBarry Smith 221b9617806SBarry Smith /*E 222b9617806SBarry Smith ISColoringType - determines if the coloring is for the entire parallel grid/graph/matrix 223b9617806SBarry Smith or for just the local ghosted portion 224b9617806SBarry Smith 22516a05f60SBarry Smith Values: 22616a05f60SBarry Smith + `IS_COLORING_GLOBAL` - does not include the colors for ghost points, this is used when the function 22716a05f60SBarry Smith is called synchronously in parallel. This requires generating a "parallel coloring". 22816a05f60SBarry Smith - `IS_COLORING_LOCAL` - includes colors for ghost points, this is used when the function can be called 22916a05f60SBarry Smith separately on individual processes with the ghost points already filled in. Does not 23016a05f60SBarry Smith require a "parallel coloring", rather each process colors its local + ghost part. 23116a05f60SBarry Smith Using this can result in much less parallel communication. Currently only works 23216a05f60SBarry Smith with `DMDA` and if you call `MatFDColoringSetFunction()` with the local function. 23316a05f60SBarry Smith 234b9617806SBarry Smith Level: beginner 235b9617806SBarry Smith 23616a05f60SBarry Smith .seealso: `ISColoring`, `ISColoringSetType()`, `ISColoringGetType()`, `DMCreateColoring()` 237b9617806SBarry Smith E*/ 2389371c9d4SSatish Balay typedef enum { 2399371c9d4SSatish Balay IS_COLORING_GLOBAL, 2409371c9d4SSatish Balay IS_COLORING_LOCAL 2419371c9d4SSatish Balay } ISColoringType; 2426a6fc655SJed Brown PETSC_EXTERN const char *const ISColoringTypes[]; 243569ea7c4SPierre Jolivet typedef unsigned PETSC_IS_COLORING_VALUE_TYPE ISColoringValue; 244569ea7c4SPierre Jolivet #define IS_COLORING_MAX PETSC_IS_COLORING_MAX 245569ea7c4SPierre Jolivet #define MPIU_COLORING_VALUE PETSC_MPIU_IS_COLORING_VALUE_TYPE 246014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISAllGatherColors(MPI_Comm, PetscInt, ISColoringValue *, PetscInt *, ISColoringValue *[]); 247dde82324SBarry Smith 248aaf3ff59SMatthew G. Knepley PETSC_EXTERN PetscErrorCode ISColoringCreate(MPI_Comm, PetscInt, PetscInt, const ISColoringValue[], PetscCopyMode, ISColoring *); 249014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISColoringDestroy(ISColoring *); 250014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISColoringView(ISColoring, PetscViewer); 2518aec7d55SBarry Smith PETSC_EXTERN PetscErrorCode ISColoringViewFromOptions(ISColoring, PetscObject, const char[]); 252071fcb05SBarry Smith PETSC_EXTERN PetscErrorCode ISColoringGetIS(ISColoring, PetscCopyMode, PetscInt *, IS *[]); 253071fcb05SBarry Smith PETSC_EXTERN PetscErrorCode ISColoringRestoreIS(ISColoring, PetscCopyMode, IS *[]); 25449b734d2SBarry Smith PETSC_EXTERN PetscErrorCode ISColoringReference(ISColoring); 25549b734d2SBarry Smith PETSC_EXTERN PetscErrorCode ISColoringSetType(ISColoring, ISColoringType); 256bdaf1daeSBarry Smith PETSC_EXTERN PetscErrorCode ISColoringGetType(ISColoring, ISColoringType *); 257bdaf1daeSBarry Smith PETSC_EXTERN PetscErrorCode ISColoringGetColors(ISColoring, PetscInt *, PetscInt *, const ISColoringValue **); 2583a7fca6bSBarry Smith 25994f0491fSFande Kong PETSC_EXTERN PetscErrorCode ISBuildTwoSided(IS, IS, IS *); 260014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISPartitioningToNumbering(IS, IS *); 261014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISPartitioningCount(IS, PetscInt, PetscInt[]); 262dbef8a1cSBarry Smith 263014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISCompressIndicesGeneral(PetscInt, PetscInt, PetscInt, PetscInt, const IS[], IS[]); 264edd03b47SJacob Faibussowitsch PETSC_DEPRECATED_FUNCTION(3, 19, 0, "ISCompressIndicesGeneral()", ) static inline PetscErrorCode ISCompressIndicesSorted(PetscInt n, PetscInt bs, PetscInt imax, const IS is_in[], IS is_out[]) 265e37d522bSPierre Jolivet { 266e37d522bSPierre Jolivet return ISCompressIndicesGeneral(n, bs, n, imax, is_in, is_out); 267e37d522bSPierre Jolivet } 268014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISExpandIndicesGeneral(PetscInt, PetscInt, PetscInt, PetscInt, const IS[], IS[]); 269d9489beaSHong Zhang 27069ce434fSBarry Smith struct _n_PetscLayout { 27169ce434fSBarry Smith MPI_Comm comm; 27238a25198SStefano Zampini PetscMPIInt size; 27369ce434fSBarry Smith PetscInt n, N; /* local, global vector size */ 27469ce434fSBarry Smith PetscInt rstart, rend; /* local start, local end + 1 */ 27569ce434fSBarry Smith PetscInt *range; /* the offset of each processor */ 2769621ec18SVaclav Hapla PetscBool range_alloc; /* should range be freed in Destroy? */ 27733d57670SJed Brown PetscInt bs; /* number of elements in each block (generally for multi-component 27833d57670SJed Brown * problems). Defaults to -1 and can be arbitrarily lazy so always use 27933d57670SJed Brown * PetscAbs(map->bs) when accessing directly and expecting result to be 28033d57670SJed Brown * positive. Do NOT multiply above numbers by bs */ 2812a7a6963SBarry Smith PetscInt refcnt; /* MPI Vecs obtained with VecDuplicate() and from MatCreateVecs() reuse map of input object */ 28269ce434fSBarry Smith ISLocalToGlobalMapping mapping; /* mapping used in Vec/MatSetValuesLocal() */ 283ca5434daSLawrence Mitchell PetscBool setupcalled; /* Forbid setup more than once */ 284ca5434daSLawrence Mitchell PetscInt oldn, oldN; /* Checking if setup is allowed */ 285ca5434daSLawrence Mitchell PetscInt oldbs; /* And again */ 28669ce434fSBarry Smith }; 28769ce434fSBarry Smith 2887e3c27c9SBarry Smith /*@C 28969ce434fSBarry Smith PetscLayoutFindOwner - Find the owning rank for a global index 29069ce434fSBarry Smith 29120f4b53cSBarry Smith Not Collective; No Fortran Support 29269ce434fSBarry Smith 29369ce434fSBarry Smith Input Parameters: 29469ce434fSBarry Smith + map - the layout 29569ce434fSBarry Smith - idx - global index to find the owner of 29669ce434fSBarry Smith 29769ce434fSBarry Smith Output Parameter: 29869ce434fSBarry Smith . owner - the owning rank 29969ce434fSBarry Smith 30069ce434fSBarry Smith Level: developer 30169ce434fSBarry Smith 30216a05f60SBarry Smith .seealso: `PetscLayout`, `PetscLayoutFindOwnerIndex()` 30369ce434fSBarry Smith @*/ 304d71ae5a4SJacob Faibussowitsch static inline PetscErrorCode PetscLayoutFindOwner(PetscLayout map, PetscInt idx, PetscMPIInt *owner) 305d71ae5a4SJacob Faibussowitsch { 30669ce434fSBarry Smith PetscMPIInt lo = 0, hi, t; 30769ce434fSBarry Smith 30869ce434fSBarry Smith PetscFunctionBegin; 30969ce434fSBarry Smith *owner = -1; /* GCC erroneously issues warning about possibly uninitialized use when error condition */ 31038a25198SStefano Zampini #if defined(PETSC_USE_DEBUG) 3112c71b3e2SJacob Faibussowitsch PetscCheck((map->n >= 0) && (map->N >= 0) && (map->range), PETSC_COMM_SELF, PETSC_ERR_ARG_WRONGSTATE, "PetscLayoutSetUp() must be called first"); 3122c71b3e2SJacob Faibussowitsch PetscCheck(idx >= 0 && idx <= map->N, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Index %" PetscInt_FMT " is out of range", idx); 31338a25198SStefano Zampini #endif 31438a25198SStefano Zampini hi = map->size; 31569ce434fSBarry Smith while (hi - lo > 1) { 31669ce434fSBarry Smith t = lo + (hi - lo) / 2; 31769ce434fSBarry Smith if (idx < map->range[t]) hi = t; 31869ce434fSBarry Smith else lo = t; 31969ce434fSBarry Smith } 32069ce434fSBarry Smith *owner = lo; 3213ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 32269ce434fSBarry Smith } 32369ce434fSBarry Smith 32469ce434fSBarry Smith /*@C 32516a05f60SBarry Smith PetscLayoutFindOwnerIndex - Find the owning MPI rank and the local index on that rank for a global index 32669ce434fSBarry Smith 32720f4b53cSBarry Smith Not Collective; No Fortran Support 32869ce434fSBarry Smith 32969ce434fSBarry Smith Input Parameters: 33069ce434fSBarry Smith + map - the layout 33169ce434fSBarry Smith - idx - global index to find the owner of 33269ce434fSBarry Smith 333d8d19677SJose E. Roman Output Parameters: 33469ce434fSBarry Smith + owner - the owning rank 33516a05f60SBarry Smith - lidx - local index used by the owner for `idx` 33669ce434fSBarry Smith 33769ce434fSBarry Smith Level: developer 33869ce434fSBarry Smith 33916a05f60SBarry Smith .seealso: `PetscLayout`, `PetscLayoutFindOwner()` 34069ce434fSBarry Smith @*/ 341d71ae5a4SJacob Faibussowitsch static inline PetscErrorCode PetscLayoutFindOwnerIndex(PetscLayout map, PetscInt idx, PetscMPIInt *owner, PetscInt *lidx) 342d71ae5a4SJacob Faibussowitsch { 34369ce434fSBarry Smith PetscMPIInt lo = 0, hi, t; 34469ce434fSBarry Smith 34569ce434fSBarry Smith PetscFunctionBegin; 34638a25198SStefano Zampini #if defined(PETSC_USE_DEBUG) 3472c71b3e2SJacob Faibussowitsch PetscCheck((map->n >= 0) && (map->N >= 0) && (map->range), PETSC_COMM_SELF, PETSC_ERR_ARG_WRONGSTATE, "PetscLayoutSetUp() must be called first"); 3482c71b3e2SJacob Faibussowitsch PetscCheck(idx >= 0 && idx <= map->N, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Index %" PetscInt_FMT " is out of range", idx); 34938a25198SStefano Zampini #endif 35038a25198SStefano Zampini hi = map->size; 35169ce434fSBarry Smith while (hi - lo > 1) { 35269ce434fSBarry Smith t = lo + (hi - lo) / 2; 35369ce434fSBarry Smith if (idx < map->range[t]) hi = t; 35469ce434fSBarry Smith else lo = t; 35569ce434fSBarry Smith } 35604799504SMark Adams if (owner) *owner = lo; 35704799504SMark Adams if (lidx) *lidx = idx - map->range[lo]; 3583ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 35969ce434fSBarry Smith } 3609e03d832SJed Brown 36101e13f73SMatthew G. Knepley PETSC_EXTERN PetscErrorCode PetscLayoutCreate(MPI_Comm, PetscLayout *); 3629621ec18SVaclav Hapla PETSC_EXTERN PetscErrorCode PetscLayoutCreateFromSizes(MPI_Comm, PetscInt, PetscInt, PetscInt, PetscLayout *); 3639621ec18SVaclav Hapla PETSC_EXTERN PetscErrorCode PetscLayoutCreateFromRanges(MPI_Comm, const PetscInt[], PetscCopyMode, PetscInt, PetscLayout *); 36401e13f73SMatthew G. Knepley PETSC_EXTERN PetscErrorCode PetscLayoutSetUp(PetscLayout); 36501e13f73SMatthew G. Knepley PETSC_EXTERN PetscErrorCode PetscLayoutDestroy(PetscLayout *); 36601e13f73SMatthew G. Knepley PETSC_EXTERN PetscErrorCode PetscLayoutDuplicate(PetscLayout, PetscLayout *); 36701e13f73SMatthew G. Knepley PETSC_EXTERN PetscErrorCode PetscLayoutReference(PetscLayout, PetscLayout *); 36801e13f73SMatthew G. Knepley PETSC_EXTERN PetscErrorCode PetscLayoutSetLocalSize(PetscLayout, PetscInt); 36901e13f73SMatthew G. Knepley PETSC_EXTERN PetscErrorCode PetscLayoutGetLocalSize(PetscLayout, PetscInt *); 37001e13f73SMatthew G. Knepley PETSC_EXTERN PetscErrorCode PetscLayoutSetSize(PetscLayout, PetscInt); 37101e13f73SMatthew G. Knepley PETSC_EXTERN PetscErrorCode PetscLayoutGetSize(PetscLayout, PetscInt *); 37201e13f73SMatthew G. Knepley PETSC_EXTERN PetscErrorCode PetscLayoutSetBlockSize(PetscLayout, PetscInt); 37301e13f73SMatthew G. Knepley PETSC_EXTERN PetscErrorCode PetscLayoutGetBlockSize(PetscLayout, PetscInt *); 37401e13f73SMatthew G. Knepley PETSC_EXTERN PetscErrorCode PetscLayoutGetRange(PetscLayout, PetscInt *, PetscInt *); 37501e13f73SMatthew G. Knepley PETSC_EXTERN PetscErrorCode PetscLayoutGetRanges(PetscLayout, const PetscInt *[]); 376f92d6284SStefano Zampini PETSC_EXTERN PetscErrorCode PetscLayoutCompare(PetscLayout, PetscLayout, PetscBool *); 37701e13f73SMatthew G. Knepley PETSC_EXTERN PetscErrorCode PetscLayoutSetISLocalToGlobalMapping(PetscLayout, ISLocalToGlobalMapping); 37807acc2aeSBarry Smith PETSC_EXTERN PetscErrorCode PetscLayoutMapLocal(PetscLayout, PetscInt, const PetscInt[], PetscInt *, PetscInt **, PetscInt **); 37901e13f73SMatthew G. Knepley 380ce605777SToby Isaac PETSC_EXTERN PetscErrorCode PetscParallelSortInt(PetscLayout, PetscLayout, PetscInt *, PetscInt *); 381ce605777SToby Isaac 382ce605777SToby Isaac PETSC_EXTERN PetscErrorCode ISGetLayout(IS, PetscLayout *); 3834e1d6e84SVaclav Hapla PETSC_EXTERN PetscErrorCode ISSetLayout(IS, PetscLayout); 384ce605777SToby Isaac 385a2ce50c7SBarry Smith #endif 386