xref: /petsc/include/petscis.h (revision 0c312b8e6c5493a795e997bb0b5f8ef7b1287755)
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 */
50a835dfdSSatish Balay #if !defined(__PETSCIS_H)
60a835dfdSSatish Balay #define __PETSCIS_H
72c8e378dSBarry Smith #include <petscsys.h>
8665c2dedSJed Brown #include <petscviewertypes.h>
9*0c312b8eSJed Brown #include <petscsftypes.h>
102eac72dbSBarry Smith 
1197b48c8fSBarry Smith #define IS_FILE_CLASSID 1211218
12014dd563SJed Brown PETSC_EXTERN PetscClassId IS_CLASSID;
13f0479e8cSBarry Smith 
14607a6623SBarry Smith PETSC_EXTERN PetscErrorCode ISInitializePackage(void);
152b6de112SBarry Smith 
165c20da3cSBarry Smith /*S
17f22f69f0SBarry Smith      IS - Abstract PETSc object that allows indexing.
185c20da3cSBarry Smith 
195c20da3cSBarry Smith    Level: beginner
205c20da3cSBarry Smith 
215c20da3cSBarry Smith   Concepts: indexing, stride
225c20da3cSBarry Smith 
235c20da3cSBarry Smith .seealso:  ISCreateGeneral(), ISCreateBlock(), ISCreateStride(), ISGetIndices(), ISDestroy()
245c20da3cSBarry Smith S*/
25f09e8eb9SSatish Balay typedef struct _p_IS* IS;
262eac72dbSBarry Smith 
2776bdecfbSBarry Smith /*J
2827bdab1eSBarry Smith     ISType - String with the name of a PETSc vector or the creation function
2927bdab1eSBarry Smith        with an optional dynamic library name, for example
3027bdab1eSBarry Smith        http://www.mcs.anl.gov/petsc/lib.a:myveccreate()
3127bdab1eSBarry Smith 
3227bdab1eSBarry Smith    Level: beginner
3327bdab1eSBarry Smith 
3427bdab1eSBarry Smith .seealso: ISSetType(), IS
3576bdecfbSBarry Smith J*/
3619fd82e9SBarry Smith typedef const char* ISType;
3727bdab1eSBarry Smith #define ISGENERAL      "general"
3827bdab1eSBarry Smith #define ISSTRIDE       "stride"
3927bdab1eSBarry Smith #define ISBLOCK        "block"
4027bdab1eSBarry Smith 
4127bdab1eSBarry Smith /* Dynamic creation and loading functions */
42140e18c1SBarry Smith PETSC_EXTERN PetscFunctionList ISList;
43014dd563SJed Brown PETSC_EXTERN PetscBool         ISRegisterAllCalled;
4419fd82e9SBarry Smith PETSC_EXTERN PetscErrorCode ISSetType(IS, ISType);
4519fd82e9SBarry Smith PETSC_EXTERN PetscErrorCode ISGetType(IS, ISType *);
46bdf89e91SBarry Smith PETSC_EXTERN PetscErrorCode ISRegister(const char[],PetscErrorCode (*)(IS));
47607a6623SBarry Smith PETSC_EXTERN PetscErrorCode ISRegisterAll(void);
48014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISRegisterDestroy(void);
49014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISCreate(MPI_Comm,IS*);
5027bdab1eSBarry Smith 
51639f9d9dSBarry Smith /*
52639f9d9dSBarry Smith     Default index set data structures that PETSc provides.
53639f9d9dSBarry Smith */
54014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISCreateGeneral(MPI_Comm,PetscInt,const PetscInt[],PetscCopyMode,IS *);
55014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISGeneralSetIndices(IS,PetscInt,const PetscInt[],PetscCopyMode);
56014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISCreateBlock(MPI_Comm,PetscInt,PetscInt,const PetscInt[],PetscCopyMode,IS *);
57014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISBlockSetIndices(IS,PetscInt,PetscInt,const PetscInt[],PetscCopyMode);
58014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISCreateStride(MPI_Comm,PetscInt,PetscInt,PetscInt,IS *);
59014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISStrideSetStride(IS,PetscInt,PetscInt,PetscInt);
604b0e389bSBarry Smith 
61014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISDestroy(IS*);
62014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISSetPermutation(IS);
63014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISPermutation(IS,PetscBool *);
64014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISSetIdentity(IS);
65014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISIdentity(IS,PetscBool *);
66014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISContiguousLocal(IS,PetscInt,PetscInt,PetscInt*,PetscBool*);
6708480c60SBarry Smith 
68014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISGetIndices(IS,const PetscInt *[]);
69014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISRestoreIndices(IS,const PetscInt *[]);
70014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISGetTotalIndices(IS,const PetscInt *[]);
71014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISRestoreTotalIndices(IS,const PetscInt *[]);
72014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISGetNonlocalIndices(IS,const PetscInt *[]);
73014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISRestoreNonlocalIndices(IS,const PetscInt *[]);
74014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISGetNonlocalIS(IS, IS *is);
75014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISRestoreNonlocalIS(IS, IS *is);
76014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISGetSize(IS,PetscInt *);
77014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISGetLocalSize(IS,PetscInt *);
78014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISInvertPermutation(IS,PetscInt,IS*);
79014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISView(IS,PetscViewer);
80014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISEqual(IS,IS,PetscBool  *);
81014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISSort(IS);
82014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISSorted(IS,PetscBool  *);
83014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISDifference(IS,IS,IS*);
84014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISSum(IS,IS,IS*);
85014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISExpand(IS,IS,IS*);
86132da990SBarry Smith PETSC_EXTERN PetscErrorCode ISGetMinMax(IS,PetscInt*,PetscInt*);
87612dd529SBarry Smith 
88014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISBlockGetIndices(IS,const PetscInt *[]);
89014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISBlockRestoreIndices(IS,const PetscInt *[]);
90014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISBlockGetLocalSize(IS,PetscInt *);
91014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISBlockGetSize(IS,PetscInt *);
92014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISGetBlockSize(IS,PetscInt*);
93014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISSetBlockSize(IS,PetscInt);
94c16cb8f2SBarry Smith 
95014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISStrideGetInfo(IS,PetscInt *,PetscInt*);
96c16cb8f2SBarry Smith 
97014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISToGeneral(IS);
9838f40f24SLois Curfman McInnes 
99014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISDuplicate(IS,IS*);
100014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISCopy(IS,IS);
101014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISAllGather(IS,IS*);
102014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISComplement(IS,PetscInt,PetscInt,IS*);
103014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISConcatenate(MPI_Comm,PetscInt,const IS[],IS*);
104bb74729cSDmitry Karpeev PETSC_EXTERN PetscErrorCode ISListToPair(MPI_Comm,PetscInt, IS[],IS*,IS*);
105bb74729cSDmitry Karpeev PETSC_EXTERN PetscErrorCode ISPairToList(IS,IS,PetscInt*, IS *[]);
106bb74729cSDmitry Karpeev PETSC_EXTERN PetscErrorCode ISEmbed(IS,IS,PetscBool,IS*);
107014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISOnComm(IS,MPI_Comm,PetscCopyMode,IS*);
108d64ed03dSBarry Smith 
10956cd22aeSBarry Smith /* --------------------------------------------------------------------------*/
110014dd563SJed Brown PETSC_EXTERN PetscClassId IS_LTOGM_CLASSID;
11156cd22aeSBarry Smith 
1125c20da3cSBarry Smith /*S
113d9ffb7b8SBarry Smith    ISLocalToGlobalMapping - mappings from an arbitrary
11490f02eecSBarry Smith       local ordering from 0 to n-1 to a global PETSc ordering
115d4bb536fSBarry Smith       used by a vector or matrix.
116d4bb536fSBarry Smith 
1175c20da3cSBarry Smith    Level: intermediate
1185c20da3cSBarry Smith 
119d4bb536fSBarry Smith    Note: mapping from Local to Global is scalable; but Global
120eec0b4cfSBarry Smith   to Local may not be if the range of global values represented locally
121d4bb536fSBarry Smith   is very large.
12274637425SBarry Smith 
12374637425SBarry Smith    Note: the ISLocalToGlobalMapping is actually a private object; it is included
124992144d0SBarry Smith   here for the inline function ISLocalToGlobalMappingApply() to allow it to be inlined since
12574637425SBarry Smith   it is used so often.
12674637425SBarry Smith 
1275c20da3cSBarry Smith .seealso:  ISLocalToGlobalMappingCreate()
1285c20da3cSBarry Smith S*/
129f09e8eb9SSatish Balay typedef struct _p_ISLocalToGlobalMapping* ISLocalToGlobalMapping;
1305c20da3cSBarry Smith 
1315c20da3cSBarry Smith /*E
1325c20da3cSBarry Smith     ISGlobalToLocalMappingType - Indicates if missing global indices are
1335c20da3cSBarry Smith 
1345c20da3cSBarry Smith    IS_GTOLM_MASK - missing global indices are replaced with -1
1355c20da3cSBarry Smith    IS_GTOLM_DROP - missing global indices are dropped
1365c20da3cSBarry Smith 
1375c20da3cSBarry Smith    Level: beginner
1385c20da3cSBarry Smith 
1395c20da3cSBarry Smith .seealso: ISGlobalToLocalMappingApply()
1405c20da3cSBarry Smith 
1415c20da3cSBarry Smith E*/
142987e4450SSatish Balay typedef enum {IS_GTOLM_MASK,IS_GTOLM_DROP} ISGlobalToLocalMappingType;
14390f02eecSBarry Smith 
144014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISLocalToGlobalMappingCreate(MPI_Comm,PetscInt,const PetscInt[],PetscCopyMode,ISLocalToGlobalMapping*);
145014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISLocalToGlobalMappingCreateIS(IS,ISLocalToGlobalMapping *);
146014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISLocalToGlobalMappingCreateSF(PetscSF,PetscInt,ISLocalToGlobalMapping*);
147014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISLocalToGlobalMappingView(ISLocalToGlobalMapping,PetscViewer);
148014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISLocalToGlobalMappingDestroy(ISLocalToGlobalMapping*);
149014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISLocalToGlobalMappingApplyIS(ISLocalToGlobalMapping,IS,IS*);
150afcb2eb5SJed Brown PETSC_EXTERN PetscErrorCode ISLocalToGlobalMappingApply(ISLocalToGlobalMapping,PetscInt,const PetscInt[],PetscInt[]);
151014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISGlobalToLocalMappingApply(ISLocalToGlobalMapping,ISGlobalToLocalMappingType,PetscInt,const PetscInt[],PetscInt*,PetscInt[]);
152014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISLocalToGlobalMappingGetSize(ISLocalToGlobalMapping,PetscInt*);
153014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISLocalToGlobalMappingGetInfo(ISLocalToGlobalMapping,PetscInt*,PetscInt*[],PetscInt*[],PetscInt**[]);
154014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISLocalToGlobalMappingRestoreInfo(ISLocalToGlobalMapping,PetscInt*,PetscInt*[],PetscInt*[],PetscInt**[]);
155014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISLocalToGlobalMappingGetIndices(ISLocalToGlobalMapping,const PetscInt**);
156014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISLocalToGlobalMappingRestoreIndices(ISLocalToGlobalMapping,const PetscInt**);
157014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISLocalToGlobalMappingBlock(ISLocalToGlobalMapping,PetscInt,ISLocalToGlobalMapping*);
158014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISLocalToGlobalMappingUnBlock(ISLocalToGlobalMapping,PetscInt,ISLocalToGlobalMapping*);
159014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISLocalToGlobalMappingConcatenate(MPI_Comm,PetscInt,const ISLocalToGlobalMapping[],ISLocalToGlobalMapping*);
160186d4ecdSBarry Smith PETSC_EXTERN PetscErrorCode ISG2LMapApply(ISLocalToGlobalMapping,PetscInt,const PetscInt[],PetscInt[]);
16174637425SBarry Smith 
16256cd22aeSBarry Smith /* --------------------------------------------------------------------------*/
163b9617806SBarry Smith /*E
164b9617806SBarry Smith     ISColoringType - determines if the coloring is for the entire parallel grid/graph/matrix
165b9617806SBarry Smith                      or for just the local ghosted portion
166b9617806SBarry Smith 
167b9617806SBarry Smith     Level: beginner
168b9617806SBarry Smith 
1698ee2e534SBarry Smith $   IS_COLORING_GLOBAL - does not include the colors for ghost points, this is used when the function
1708ee2e534SBarry Smith $                        is called synchronously in parallel. This requires generating a "parallel coloring".
1718ee2e534SBarry Smith $   IS_COLORING_GHOSTED - includes colors for ghost points, this is used when the function can be called
1728ee2e534SBarry Smith $                         seperately on individual processes with the ghost points already filled in. Does not
1738ee2e534SBarry Smith $                         require a "parallel coloring", rather each process colors its local + ghost part.
1748ee2e534SBarry Smith $                         Using this can result in much less parallel communication. In the paradigm of
1759a42bb27SBarry Smith $                         DMGetLocalVector() and DMGetGlobalVector() this could be called IS_COLORING_LOCAL
17673d7d85fSBarry Smith 
177e727c939SJed Brown .seealso: DMCreateColoring()
178b9617806SBarry Smith E*/
1798ee2e534SBarry Smith typedef enum {IS_COLORING_GLOBAL,IS_COLORING_GHOSTED} ISColoringType;
1806a6fc655SJed Brown PETSC_EXTERN const char *const ISColoringTypes[];
1816c09f170SSatish Balay typedef unsigned PETSC_IS_COLOR_VALUE_TYPE ISColoringValue;
182014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISAllGatherColors(MPI_Comm,PetscInt,ISColoringValue*,PetscInt*,ISColoringValue*[]);
183dde82324SBarry Smith 
1845c20da3cSBarry Smith /*S
185dde82324SBarry Smith      ISColoring - sets of IS's that define a coloring
186639f9d9dSBarry Smith               of the underlying indices
1875c20da3cSBarry Smith 
1885c20da3cSBarry Smith    Level: intermediate
1895c20da3cSBarry Smith 
1905c20da3cSBarry Smith     Notes:
191b9617806SBarry Smith         One should not access the *is records below directly because they may not yet
1925c20da3cSBarry Smith     have been created. One should use ISColoringGetIS() to make sure they are
1935c20da3cSBarry Smith     created when needed.
1945c20da3cSBarry Smith 
195a5057860SBarry Smith     Developer Note: this is not a PetscObject
196a5057860SBarry Smith 
1975c20da3cSBarry Smith .seealso:  ISColoringCreate(), ISColoringGetIS(), ISColoringView(), ISColoringGetIS()
1985c20da3cSBarry Smith S*/
19995fbd943SSatish Balay struct _n_ISColoring {
200a7cc72afSBarry Smith   PetscInt        refct;
201a7cc72afSBarry Smith   PetscInt        n;                /* number of colors */
2025c20da3cSBarry Smith   IS              *is;              /* for each color indicates columns */
203639f9d9dSBarry Smith   MPI_Comm        comm;
20408b6dcc0SBarry Smith   ISColoringValue *colors;          /* for each column indicates color */
20532dcc486SBarry Smith   PetscInt        N;                /* number of columns */
206b9617806SBarry Smith   ISColoringType  ctype;
207639f9d9dSBarry Smith };
20895fbd943SSatish Balay typedef struct _n_ISColoring* ISColoring;
209639f9d9dSBarry Smith 
210014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISColoringCreate(MPI_Comm,PetscInt,PetscInt,const ISColoringValue[],ISColoring*);
211014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISColoringDestroy(ISColoring*);
212014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISColoringView(ISColoring,PetscViewer);
213014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISColoringGetIS(ISColoring,PetscInt*,IS*[]);
214014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISColoringRestoreIS(ISColoring,IS*[]);
21549b734d2SBarry Smith PETSC_EXTERN PetscErrorCode ISColoringReference(ISColoring);
21649b734d2SBarry Smith PETSC_EXTERN PetscErrorCode ISColoringSetType(ISColoring,ISColoringType);
21749b734d2SBarry Smith 
2183a7fca6bSBarry Smith 
219dbef8a1cSBarry Smith /* --------------------------------------------------------------------------*/
220dbef8a1cSBarry Smith 
221014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISPartitioningToNumbering(IS,IS*);
222014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISPartitioningCount(IS,PetscInt,PetscInt[]);
223dbef8a1cSBarry Smith 
224014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISCompressIndicesGeneral(PetscInt,PetscInt,PetscInt,PetscInt,const IS[],IS[]);
225014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISCompressIndicesSorted(PetscInt,PetscInt,PetscInt,const IS[],IS[]);
226014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISExpandIndicesGeneral(PetscInt,PetscInt,PetscInt,PetscInt,const IS[],IS[]);
227d9489beaSHong Zhang 
22869ce434fSBarry Smith /*S
22969ce434fSBarry Smith      PetscLayout - defines layout of vectors and matrices across processes (which rows are owned by which processes)
23069ce434fSBarry Smith 
23169ce434fSBarry Smith    Level: developer
23269ce434fSBarry Smith 
23369ce434fSBarry Smith 
23469ce434fSBarry Smith .seealso:  PetscLayoutCreate(), PetscLayoutDestroy()
23569ce434fSBarry Smith S*/
23669ce434fSBarry Smith typedef struct _n_PetscLayout* PetscLayout;
23769ce434fSBarry Smith struct _n_PetscLayout{
23869ce434fSBarry Smith   MPI_Comm               comm;
23969ce434fSBarry Smith   PetscInt               n,N;         /* local, global vector size */
24069ce434fSBarry Smith   PetscInt               rstart,rend; /* local start, local end + 1 */
24169ce434fSBarry Smith   PetscInt               *range;      /* the offset of each processor */
24269ce434fSBarry Smith   PetscInt               bs;          /* number of elements in each block (generally for multi-component problems) Do NOT multiply above numbers by bs */
24369ce434fSBarry Smith   PetscInt               refcnt;      /* MPI Vecs obtained with VecDuplicate() and from MatGetVecs() reuse map of input object */
24469ce434fSBarry Smith   ISLocalToGlobalMapping mapping;     /* mapping used in Vec/MatSetValuesLocal() */
24569ce434fSBarry Smith   ISLocalToGlobalMapping bmapping;    /* mapping used in Vec/MatSetValuesBlockedLocal() */
24669ce434fSBarry Smith   PetscInt               *trstarts;   /* local start for each thread */
24769ce434fSBarry Smith };
24869ce434fSBarry Smith 
24969ce434fSBarry Smith PETSC_EXTERN PetscErrorCode PetscLayoutCreate(MPI_Comm,PetscLayout*);
25069ce434fSBarry Smith PETSC_EXTERN PetscErrorCode PetscLayoutSetUp(PetscLayout);
25169ce434fSBarry Smith PETSC_EXTERN PetscErrorCode PetscLayoutDestroy(PetscLayout*);
25269ce434fSBarry Smith PETSC_EXTERN PetscErrorCode PetscLayoutDuplicate(PetscLayout,PetscLayout*);
25369ce434fSBarry Smith PETSC_EXTERN PetscErrorCode PetscLayoutReference(PetscLayout,PetscLayout*);
25469ce434fSBarry Smith PETSC_EXTERN PetscErrorCode PetscLayoutSetLocalSize(PetscLayout,PetscInt);
25569ce434fSBarry Smith PETSC_EXTERN PetscErrorCode PetscLayoutGetLocalSize(PetscLayout,PetscInt *);
25669ce434fSBarry Smith PETSC_EXTERN PetscErrorCode PetscLayoutSetSize(PetscLayout,PetscInt);
25769ce434fSBarry Smith PETSC_EXTERN PetscErrorCode PetscLayoutGetSize(PetscLayout,PetscInt *);
25869ce434fSBarry Smith PETSC_EXTERN PetscErrorCode PetscLayoutSetBlockSize(PetscLayout,PetscInt);
25969ce434fSBarry Smith PETSC_EXTERN PetscErrorCode PetscLayoutGetBlockSize(PetscLayout,PetscInt*);
26069ce434fSBarry Smith PETSC_EXTERN PetscErrorCode PetscLayoutGetRange(PetscLayout,PetscInt *,PetscInt *);
26169ce434fSBarry Smith PETSC_EXTERN PetscErrorCode PetscLayoutGetRanges(PetscLayout,const PetscInt *[]);
26269ce434fSBarry Smith PETSC_EXTERN PetscErrorCode PetscLayoutSetISLocalToGlobalMapping(PetscLayout,ISLocalToGlobalMapping);
26369ce434fSBarry Smith PETSC_EXTERN PetscErrorCode PetscLayoutSetISLocalToGlobalMappingBlock(PetscLayout,ISLocalToGlobalMapping);
26469ce434fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSFSetGraphLayout(PetscSF,PetscLayout,PetscInt,const PetscInt*,PetscCopyMode,const PetscInt*);
26569ce434fSBarry Smith 
26669ce434fSBarry Smith #undef __FUNCT__
26769ce434fSBarry Smith #define __FUNCT__ "PetscLayoutFindOwner"
26869ce434fSBarry Smith /*@C
26969ce434fSBarry Smith      PetscLayoutFindOwner - Find the owning rank for a global index
27069ce434fSBarry Smith 
27169ce434fSBarry Smith     Not Collective
27269ce434fSBarry Smith 
27369ce434fSBarry Smith    Input Parameters:
27469ce434fSBarry Smith +    map - the layout
27569ce434fSBarry Smith -    idx - global index to find the owner of
27669ce434fSBarry Smith 
27769ce434fSBarry Smith    Output Parameter:
27869ce434fSBarry Smith .    owner - the owning rank
27969ce434fSBarry Smith 
28069ce434fSBarry Smith    Level: developer
28169ce434fSBarry Smith 
28269ce434fSBarry Smith     Fortran Notes:
28369ce434fSBarry Smith       Not available from Fortran
28469ce434fSBarry Smith 
28569ce434fSBarry Smith @*/
28669ce434fSBarry Smith PETSC_STATIC_INLINE PetscErrorCode PetscLayoutFindOwner(PetscLayout map,PetscInt idx,PetscInt *owner)
28769ce434fSBarry Smith {
28869ce434fSBarry Smith   PetscErrorCode ierr;
28969ce434fSBarry Smith   PetscMPIInt    lo = 0,hi,t;
29069ce434fSBarry Smith 
29169ce434fSBarry Smith   PetscFunctionBegin;
29269ce434fSBarry Smith   *owner = -1;                  /* GCC erroneously issues warning about possibly uninitialized use when error condition */
29369ce434fSBarry Smith   if (!((map->n >= 0) && (map->N >= 0) && (map->range))) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"PetscLayoutSetUp() must be called first");
29469ce434fSBarry Smith   if (idx < 0 || idx > map->N) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Index %D is out of range",idx);
29569ce434fSBarry Smith   ierr = MPI_Comm_size(map->comm,&hi);CHKERRQ(ierr);
29669ce434fSBarry Smith   while (hi - lo > 1) {
29769ce434fSBarry Smith     t = lo + (hi - lo) / 2;
29869ce434fSBarry Smith     if (idx < map->range[t]) hi = t;
29969ce434fSBarry Smith     else                     lo = t;
30069ce434fSBarry Smith   }
30169ce434fSBarry Smith   *owner = lo;
30269ce434fSBarry Smith   PetscFunctionReturn(0);
30369ce434fSBarry Smith }
30469ce434fSBarry Smith 
30569ce434fSBarry Smith #undef __FUNCT__
30669ce434fSBarry Smith #define __FUNCT__ "PetscLayoutFindOwnerIndex"
30769ce434fSBarry Smith /*@C
30869ce434fSBarry Smith      PetscLayoutFindOwnerIndex - Find the owning rank and the local index for a global index
30969ce434fSBarry Smith 
31069ce434fSBarry Smith     Not Collective
31169ce434fSBarry Smith 
31269ce434fSBarry Smith    Input Parameters:
31369ce434fSBarry Smith +    map   - the layout
31469ce434fSBarry Smith -    idx   - global index to find the owner of
31569ce434fSBarry Smith 
31669ce434fSBarry Smith    Output Parameter:
31769ce434fSBarry Smith +    owner - the owning rank
31869ce434fSBarry Smith -    lidx  - local index used by the owner for idx
31969ce434fSBarry Smith 
32069ce434fSBarry Smith    Level: developer
32169ce434fSBarry Smith 
32269ce434fSBarry Smith     Fortran Notes:
32369ce434fSBarry Smith       Not available from Fortran
32469ce434fSBarry Smith 
32569ce434fSBarry Smith @*/
32669ce434fSBarry Smith PETSC_STATIC_INLINE PetscErrorCode PetscLayoutFindOwnerIndex(PetscLayout map,PetscInt idx,PetscInt *owner, PetscInt *lidx)
32769ce434fSBarry Smith {
32869ce434fSBarry Smith   PetscErrorCode ierr;
32969ce434fSBarry Smith   PetscMPIInt    lo = 0,hi,t;
33069ce434fSBarry Smith 
33169ce434fSBarry Smith   PetscFunctionBegin;
33269ce434fSBarry Smith   if (!((map->n >= 0) && (map->N >= 0) && (map->range))) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"PetscLayoutSetUp() must be called first");
33369ce434fSBarry Smith   if (idx < 0 || idx > map->N) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Index %D is out of range",idx);
33469ce434fSBarry Smith   ierr = MPI_Comm_size(map->comm,&hi);CHKERRQ(ierr);
33569ce434fSBarry Smith   while (hi - lo > 1) {
33669ce434fSBarry Smith     t = lo + (hi - lo) / 2;
33769ce434fSBarry Smith     if (idx < map->range[t]) hi = t;
33869ce434fSBarry Smith     else                     lo = t;
33969ce434fSBarry Smith   }
34069ce434fSBarry Smith   *owner = lo;
34169ce434fSBarry Smith   *lidx  = idx-map->range[lo];
34269ce434fSBarry Smith   PetscFunctionReturn(0);
34369ce434fSBarry Smith }
3449e03d832SJed Brown 
3451d799100SJed Brown PETSC_EXTERN PetscClassId PETSC_SECTION_CLASSID;
3461d799100SJed Brown 
347d53a3d6fSBarry Smith /*S
34886ae641aSJed Brown   PetscSection - Mapping from integers in a designated range to contiguous sets of integers.
34986ae641aSJed Brown 
35086ae641aSJed Brown   In contrast to IS, which maps from integers to single integers, the range of a PetscSection is in the space of
35186ae641aSJed Brown   contiguous sets of integers. These ranges are frequently interpreted as domains of other array-like objects,
35286ae641aSJed Brown   especially other PetscSections, Vecs, and ISs. The domain is set with PetscSectionSetChart() and does not need to
35386ae641aSJed Brown   start at 0. For each point in the domain of a PetscSection, the output set is represented through an offset and a
35486ae641aSJed Brown   count, which are set using PetscSectionSetOffset() and PetscSectionSetDof() respectively. Lookup is typically using
35586ae641aSJed Brown   accessors or routines like VecGetValuesSection().
356d53a3d6fSBarry Smith 
357d53a3d6fSBarry Smith   Level: developer
358d53a3d6fSBarry Smith 
359d53a3d6fSBarry Smith .seealso:  PetscSectionCreate(), PetscSectionDestroy()
360d53a3d6fSBarry Smith S*/
3611d799100SJed Brown typedef struct _p_PetscSection *PetscSection;
362d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionCreate(MPI_Comm,PetscSection*);
363d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionClone(PetscSection, PetscSection*);
364d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionGetNumFields(PetscSection, PetscInt *);
365d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionSetNumFields(PetscSection, PetscInt);
366d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionGetFieldName(PetscSection, PetscInt, const char *[]);
367d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionSetFieldName(PetscSection, PetscInt, const char []);
368d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionGetFieldComponents(PetscSection, PetscInt, PetscInt *);
369d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionSetFieldComponents(PetscSection, PetscInt, PetscInt);
370d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionGetChart(PetscSection, PetscInt *, PetscInt *);
371d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionSetChart(PetscSection, PetscInt, PetscInt);
372d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionGetDof(PetscSection, PetscInt, PetscInt*);
373d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionSetDof(PetscSection, PetscInt, PetscInt);
374d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionAddDof(PetscSection, PetscInt, PetscInt);
375d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionGetFieldDof(PetscSection, PetscInt, PetscInt, PetscInt*);
376d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionSetFieldDof(PetscSection, PetscInt, PetscInt, PetscInt);
377d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionAddFieldDof(PetscSection, PetscInt, PetscInt, PetscInt);
378d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionGetConstraintDof(PetscSection, PetscInt, PetscInt*);
379d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionSetConstraintDof(PetscSection, PetscInt, PetscInt);
380d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionAddConstraintDof(PetscSection, PetscInt, PetscInt);
381d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionGetFieldConstraintDof(PetscSection, PetscInt, PetscInt, PetscInt*);
382d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionSetFieldConstraintDof(PetscSection, PetscInt, PetscInt, PetscInt);
383d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionAddFieldConstraintDof(PetscSection, PetscInt, PetscInt, PetscInt);
384d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionGetConstraintIndices(PetscSection, PetscInt, const PetscInt**);
385d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionSetConstraintIndices(PetscSection, PetscInt, const PetscInt*);
386d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionGetFieldConstraintIndices(PetscSection, PetscInt, PetscInt, const PetscInt**);
387d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionSetFieldConstraintIndices(PetscSection, PetscInt, PetscInt, const PetscInt*);
388d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionSetUpBC(PetscSection);
389d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionSetUp(PetscSection);
390d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionGetMaxDof(PetscSection, PetscInt*);
391d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionGetStorageSize(PetscSection, PetscInt*);
392d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionGetConstrainedStorageSize(PetscSection, PetscInt*);
393d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionGetOffset(PetscSection, PetscInt, PetscInt*);
394d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionSetOffset(PetscSection, PetscInt, PetscInt);
395d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionGetFieldOffset(PetscSection, PetscInt, PetscInt, PetscInt*);
396d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionSetFieldOffset(PetscSection, PetscInt, PetscInt, PetscInt);
397d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionGetOffsetRange(PetscSection, PetscInt *, PetscInt *);
398d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionView(PetscSection, PetscViewer);
399d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionReset(PetscSection);
400d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionDestroy(PetscSection*);
401d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionCreateGlobalSection(PetscSection, PetscSF, PetscBool, PetscSection *);
402d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionCreateGlobalSectionCensored(PetscSection, PetscSF, PetscBool, PetscInt, const PetscInt [], PetscSection *);
403d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionCreateSubsection(PetscSection, PetscInt, PetscInt [], PetscSection *);
404d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionCreateSubmeshSection(PetscSection, IS, PetscSection *);
405d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionGetPointLayout(MPI_Comm, PetscSection, PetscLayout *);
406d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionGetValueLayout(MPI_Comm, PetscSection, PetscLayout *);
407d53a3d6fSBarry Smith 
408d53a3d6fSBarry Smith /* PetscSF support */
409d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSFConvertPartition(PetscSF, PetscSection, IS, ISLocalToGlobalMapping *, PetscSF *);
410d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSFCreateRemoteOffsets(PetscSF, PetscSection, PetscSection, PetscInt **);
411d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSFDistributeSection(PetscSF, PetscSection, PetscInt **, PetscSection);
412d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSFCreateSectionSF(PetscSF, PetscSection, PetscInt [], PetscSection, PetscSF *);
413d53a3d6fSBarry Smith 
4149e03d832SJed Brown /* Reset __FUNCT__ in case the user does not define it themselves */
4159e03d832SJed Brown #undef __FUNCT__
4169e03d832SJed Brown #define __FUNCT__ "User provided function"
4179e03d832SJed Brown 
418a2ce50c7SBarry Smith #endif
419