xref: /petsc/include/petscis.h (revision d53a3d6fea7331c598f751e9a4893fffdfd7b7a2)
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>
82c8e378dSBarry Smith #include <petscsf.h>
92eac72dbSBarry Smith 
1097b48c8fSBarry Smith #define IS_FILE_CLASSID 1211218
11014dd563SJed Brown PETSC_EXTERN PetscClassId IS_CLASSID;
12f0479e8cSBarry Smith 
13014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISInitializePackage(const char[]);
142b6de112SBarry Smith 
155c20da3cSBarry Smith /*S
16f22f69f0SBarry Smith      IS - Abstract PETSc object that allows indexing.
175c20da3cSBarry Smith 
185c20da3cSBarry Smith    Level: beginner
195c20da3cSBarry Smith 
205c20da3cSBarry Smith   Concepts: indexing, stride
215c20da3cSBarry Smith 
225c20da3cSBarry Smith .seealso:  ISCreateGeneral(), ISCreateBlock(), ISCreateStride(), ISGetIndices(), ISDestroy()
235c20da3cSBarry Smith S*/
24f09e8eb9SSatish Balay typedef struct _p_IS* IS;
252eac72dbSBarry Smith 
2676bdecfbSBarry Smith /*J
2727bdab1eSBarry Smith     ISType - String with the name of a PETSc vector or the creation function
2827bdab1eSBarry Smith        with an optional dynamic library name, for example
2927bdab1eSBarry Smith        http://www.mcs.anl.gov/petsc/lib.a:myveccreate()
3027bdab1eSBarry Smith 
3127bdab1eSBarry Smith    Level: beginner
3227bdab1eSBarry Smith 
3327bdab1eSBarry Smith .seealso: ISSetType(), IS
3476bdecfbSBarry Smith J*/
3519fd82e9SBarry Smith typedef const char* ISType;
3627bdab1eSBarry Smith #define ISGENERAL      "general"
3727bdab1eSBarry Smith #define ISSTRIDE       "stride"
3827bdab1eSBarry Smith #define ISBLOCK        "block"
3927bdab1eSBarry Smith 
4027bdab1eSBarry Smith /* Dynamic creation and loading functions */
41140e18c1SBarry Smith PETSC_EXTERN PetscFunctionList ISList;
42014dd563SJed Brown PETSC_EXTERN PetscBool         ISRegisterAllCalled;
4319fd82e9SBarry Smith PETSC_EXTERN PetscErrorCode ISSetType(IS, ISType);
4419fd82e9SBarry Smith PETSC_EXTERN PetscErrorCode ISGetType(IS, ISType *);
45014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISRegister(const char[],const char[],const char[],PetscErrorCode (*)(IS));
46014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISRegisterAll(const char []);
47014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISRegisterDestroy(void);
48014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISCreate(MPI_Comm,IS*);
4927bdab1eSBarry Smith 
5027bdab1eSBarry Smith /*MC
51f2ba6396SBarry Smith   ISRegisterDynamic - Adds a new index set implementation
5227bdab1eSBarry Smith 
5327bdab1eSBarry Smith   Synopsis:
54f2ba6396SBarry Smith   #include "petscis.h"
5527bdab1eSBarry Smith   PetscErrorCode ISRegisterDynamic(const char *name, const char *path, const char *func_name, PetscErrorCode (*create_func)(IS))
5627bdab1eSBarry Smith 
5727bdab1eSBarry Smith   Not Collective
5827bdab1eSBarry Smith 
5927bdab1eSBarry Smith   Input Parameters:
6027bdab1eSBarry Smith + name        - The name of a new user-defined creation routine
6127bdab1eSBarry Smith . path        - The path (either absolute or relative) of the library containing this routine
6227bdab1eSBarry Smith . func_name   - The name of routine to create method context
6327bdab1eSBarry Smith - create_func - The creation routine itself
6427bdab1eSBarry Smith 
6527bdab1eSBarry Smith   Notes:
6627bdab1eSBarry Smith   ISRegisterDynamic() may be called multiple times to add several user-defined vectors
6727bdab1eSBarry Smith 
6827bdab1eSBarry Smith   If dynamic libraries are used, then the fourth input argument (routine_create) is ignored.
6927bdab1eSBarry Smith 
7027bdab1eSBarry Smith   Sample usage:
7127bdab1eSBarry Smith .vb
72bb7a0047SJed Brown     ISRegisterDynamic("my_is_name","/home/username/my_lib/lib/libO/solaris/libmy.a", "MyISCreate", MyISCreate);
7327bdab1eSBarry Smith .ve
7427bdab1eSBarry Smith 
7527bdab1eSBarry Smith   Then, your vector type can be chosen with the procedural interface via
7627bdab1eSBarry Smith .vb
7727bdab1eSBarry Smith     ISCreate(MPI_Comm, IS *);
78bb7a0047SJed Brown     ISSetType(IS,"my_is_name");
7927bdab1eSBarry Smith .ve
8027bdab1eSBarry Smith    or at runtime via the option
8127bdab1eSBarry Smith .vb
82bb7a0047SJed Brown     -is_type my_is_name
8327bdab1eSBarry Smith .ve
8427bdab1eSBarry Smith 
8527bdab1eSBarry Smith   Notes: $PETSC_ARCH occuring in pathname will be replaced with appropriate values.
8627bdab1eSBarry Smith          If your function is not being put into a shared library then use ISRegister() instead
8727bdab1eSBarry Smith 
88bb7a0047SJed Brown   This is no ISSetFromOptions() and the current implementations do not have a way to dynamically determine type, so
89bb7a0047SJed Brown   dynamic registration of custom IS types will be of limited use to users.
90bb7a0047SJed Brown 
91bb7a0047SJed Brown   Level: developer
9227bdab1eSBarry Smith 
9327bdab1eSBarry Smith .keywords: IS, register
9427bdab1eSBarry Smith .seealso: ISRegisterAll(), ISRegisterDestroy(), ISRegister()
9527bdab1eSBarry Smith M*/
9627bdab1eSBarry Smith #if defined(PETSC_USE_DYNAMIC_LIBRARIES)
9727bdab1eSBarry Smith #define ISRegisterDynamic(a,b,c,d) ISRegister(a,b,c,0)
9827bdab1eSBarry Smith #else
9927bdab1eSBarry Smith #define ISRegisterDynamic(a,b,c,d) ISRegister(a,b,c,d)
10027bdab1eSBarry Smith #endif
10127bdab1eSBarry Smith 
102639f9d9dSBarry Smith /*
103639f9d9dSBarry Smith     Default index set data structures that PETSc provides.
104639f9d9dSBarry Smith */
105014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISCreateGeneral(MPI_Comm,PetscInt,const PetscInt[],PetscCopyMode,IS *);
106014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISGeneralSetIndices(IS,PetscInt,const PetscInt[],PetscCopyMode);
107014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISCreateBlock(MPI_Comm,PetscInt,PetscInt,const PetscInt[],PetscCopyMode,IS *);
108014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISBlockSetIndices(IS,PetscInt,PetscInt,const PetscInt[],PetscCopyMode);
109014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISCreateStride(MPI_Comm,PetscInt,PetscInt,PetscInt,IS *);
110014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISStrideSetStride(IS,PetscInt,PetscInt,PetscInt);
1114b0e389bSBarry Smith 
112014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISDestroy(IS*);
113014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISSetPermutation(IS);
114014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISPermutation(IS,PetscBool *);
115014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISSetIdentity(IS);
116014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISIdentity(IS,PetscBool *);
117014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISContiguousLocal(IS,PetscInt,PetscInt,PetscInt*,PetscBool*);
11808480c60SBarry Smith 
119014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISGetIndices(IS,const PetscInt *[]);
120014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISRestoreIndices(IS,const PetscInt *[]);
121014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISGetTotalIndices(IS,const PetscInt *[]);
122014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISRestoreTotalIndices(IS,const PetscInt *[]);
123014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISGetNonlocalIndices(IS,const PetscInt *[]);
124014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISRestoreNonlocalIndices(IS,const PetscInt *[]);
125014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISGetNonlocalIS(IS, IS *is);
126014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISRestoreNonlocalIS(IS, IS *is);
127014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISGetSize(IS,PetscInt *);
128014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISGetLocalSize(IS,PetscInt *);
129014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISInvertPermutation(IS,PetscInt,IS*);
130014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISView(IS,PetscViewer);
131014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISEqual(IS,IS,PetscBool  *);
132014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISSort(IS);
133014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISSorted(IS,PetscBool  *);
134014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISDifference(IS,IS,IS*);
135014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISSum(IS,IS,IS*);
136014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISExpand(IS,IS,IS*);
137132da990SBarry Smith PETSC_EXTERN PetscErrorCode ISGetMinMax(IS,PetscInt*,PetscInt*);
138612dd529SBarry Smith 
139014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISBlockGetIndices(IS,const PetscInt *[]);
140014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISBlockRestoreIndices(IS,const PetscInt *[]);
141014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISBlockGetLocalSize(IS,PetscInt *);
142014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISBlockGetSize(IS,PetscInt *);
143014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISGetBlockSize(IS,PetscInt*);
144014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISSetBlockSize(IS,PetscInt);
145c16cb8f2SBarry Smith 
146014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISStrideGetInfo(IS,PetscInt *,PetscInt*);
147c16cb8f2SBarry Smith 
148014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISToGeneral(IS);
14938f40f24SLois Curfman McInnes 
150014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISDuplicate(IS,IS*);
151014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISCopy(IS,IS);
152014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISAllGather(IS,IS*);
153014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISComplement(IS,PetscInt,PetscInt,IS*);
154014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISConcatenate(MPI_Comm,PetscInt,const IS[],IS*);
155bb74729cSDmitry Karpeev PETSC_EXTERN PetscErrorCode ISListToPair(MPI_Comm,PetscInt, IS[],IS*,IS*);
156bb74729cSDmitry Karpeev PETSC_EXTERN PetscErrorCode ISPairToList(IS,IS,PetscInt*, IS *[]);
157bb74729cSDmitry Karpeev PETSC_EXTERN PetscErrorCode ISEmbed(IS,IS,PetscBool,IS*);
158014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISOnComm(IS,MPI_Comm,PetscCopyMode,IS*);
159d64ed03dSBarry Smith 
16056cd22aeSBarry Smith /* --------------------------------------------------------------------------*/
161014dd563SJed Brown PETSC_EXTERN PetscClassId IS_LTOGM_CLASSID;
16256cd22aeSBarry Smith 
1635c20da3cSBarry Smith /*S
164d9ffb7b8SBarry Smith    ISLocalToGlobalMapping - mappings from an arbitrary
16590f02eecSBarry Smith       local ordering from 0 to n-1 to a global PETSc ordering
166d4bb536fSBarry Smith       used by a vector or matrix.
167d4bb536fSBarry Smith 
1685c20da3cSBarry Smith    Level: intermediate
1695c20da3cSBarry Smith 
170d4bb536fSBarry Smith    Note: mapping from Local to Global is scalable; but Global
171eec0b4cfSBarry Smith   to Local may not be if the range of global values represented locally
172d4bb536fSBarry Smith   is very large.
17374637425SBarry Smith 
17474637425SBarry Smith    Note: the ISLocalToGlobalMapping is actually a private object; it is included
175992144d0SBarry Smith   here for the inline function ISLocalToGlobalMappingApply() to allow it to be inlined since
17674637425SBarry Smith   it is used so often.
17774637425SBarry Smith 
1785c20da3cSBarry Smith .seealso:  ISLocalToGlobalMappingCreate()
1795c20da3cSBarry Smith S*/
180f09e8eb9SSatish Balay typedef struct _p_ISLocalToGlobalMapping* ISLocalToGlobalMapping;
1815c20da3cSBarry Smith 
1825c20da3cSBarry Smith /*E
1835c20da3cSBarry Smith     ISGlobalToLocalMappingType - Indicates if missing global indices are
1845c20da3cSBarry Smith 
1855c20da3cSBarry Smith    IS_GTOLM_MASK - missing global indices are replaced with -1
1865c20da3cSBarry Smith    IS_GTOLM_DROP - missing global indices are dropped
1875c20da3cSBarry Smith 
1885c20da3cSBarry Smith    Level: beginner
1895c20da3cSBarry Smith 
1905c20da3cSBarry Smith .seealso: ISGlobalToLocalMappingApply()
1915c20da3cSBarry Smith 
1925c20da3cSBarry Smith E*/
193987e4450SSatish Balay typedef enum {IS_GTOLM_MASK,IS_GTOLM_DROP} ISGlobalToLocalMappingType;
19490f02eecSBarry Smith 
195014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISLocalToGlobalMappingCreate(MPI_Comm,PetscInt,const PetscInt[],PetscCopyMode,ISLocalToGlobalMapping*);
196014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISLocalToGlobalMappingCreateIS(IS,ISLocalToGlobalMapping *);
197014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISLocalToGlobalMappingCreateSF(PetscSF,PetscInt,ISLocalToGlobalMapping*);
198014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISLocalToGlobalMappingView(ISLocalToGlobalMapping,PetscViewer);
199014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISLocalToGlobalMappingDestroy(ISLocalToGlobalMapping*);
200014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISLocalToGlobalMappingApplyIS(ISLocalToGlobalMapping,IS,IS*);
201afcb2eb5SJed Brown PETSC_EXTERN PetscErrorCode ISLocalToGlobalMappingApply(ISLocalToGlobalMapping,PetscInt,const PetscInt[],PetscInt[]);
202014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISGlobalToLocalMappingApply(ISLocalToGlobalMapping,ISGlobalToLocalMappingType,PetscInt,const PetscInt[],PetscInt*,PetscInt[]);
203014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISLocalToGlobalMappingGetSize(ISLocalToGlobalMapping,PetscInt*);
204014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISLocalToGlobalMappingGetInfo(ISLocalToGlobalMapping,PetscInt*,PetscInt*[],PetscInt*[],PetscInt**[]);
205014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISLocalToGlobalMappingRestoreInfo(ISLocalToGlobalMapping,PetscInt*,PetscInt*[],PetscInt*[],PetscInt**[]);
206014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISLocalToGlobalMappingGetIndices(ISLocalToGlobalMapping,const PetscInt**);
207014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISLocalToGlobalMappingRestoreIndices(ISLocalToGlobalMapping,const PetscInt**);
208014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISLocalToGlobalMappingBlock(ISLocalToGlobalMapping,PetscInt,ISLocalToGlobalMapping*);
209014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISLocalToGlobalMappingUnBlock(ISLocalToGlobalMapping,PetscInt,ISLocalToGlobalMapping*);
210014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISLocalToGlobalMappingConcatenate(MPI_Comm,PetscInt,const ISLocalToGlobalMapping[],ISLocalToGlobalMapping*);
211186d4ecdSBarry Smith PETSC_EXTERN PetscErrorCode ISG2LMapApply(ISLocalToGlobalMapping,PetscInt,const PetscInt[],PetscInt[]);
21274637425SBarry Smith 
21356cd22aeSBarry Smith /* --------------------------------------------------------------------------*/
214b9617806SBarry Smith /*E
215b9617806SBarry Smith     ISColoringType - determines if the coloring is for the entire parallel grid/graph/matrix
216b9617806SBarry Smith                      or for just the local ghosted portion
217b9617806SBarry Smith 
218b9617806SBarry Smith     Level: beginner
219b9617806SBarry Smith 
2208ee2e534SBarry Smith $   IS_COLORING_GLOBAL - does not include the colors for ghost points, this is used when the function
2218ee2e534SBarry Smith $                        is called synchronously in parallel. This requires generating a "parallel coloring".
2228ee2e534SBarry Smith $   IS_COLORING_GHOSTED - includes colors for ghost points, this is used when the function can be called
2238ee2e534SBarry Smith $                         seperately on individual processes with the ghost points already filled in. Does not
2248ee2e534SBarry Smith $                         require a "parallel coloring", rather each process colors its local + ghost part.
2258ee2e534SBarry Smith $                         Using this can result in much less parallel communication. In the paradigm of
2269a42bb27SBarry Smith $                         DMGetLocalVector() and DMGetGlobalVector() this could be called IS_COLORING_LOCAL
22773d7d85fSBarry Smith 
228e727c939SJed Brown .seealso: DMCreateColoring()
229b9617806SBarry Smith E*/
2308ee2e534SBarry Smith typedef enum {IS_COLORING_GLOBAL,IS_COLORING_GHOSTED} ISColoringType;
2316a6fc655SJed Brown PETSC_EXTERN const char *const ISColoringTypes[];
2326c09f170SSatish Balay typedef unsigned PETSC_IS_COLOR_VALUE_TYPE ISColoringValue;
233014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISAllGatherColors(MPI_Comm,PetscInt,ISColoringValue*,PetscInt*,ISColoringValue*[]);
234dde82324SBarry Smith 
2355c20da3cSBarry Smith /*S
236dde82324SBarry Smith      ISColoring - sets of IS's that define a coloring
237639f9d9dSBarry Smith               of the underlying indices
2385c20da3cSBarry Smith 
2395c20da3cSBarry Smith    Level: intermediate
2405c20da3cSBarry Smith 
2415c20da3cSBarry Smith     Notes:
242b9617806SBarry Smith         One should not access the *is records below directly because they may not yet
2435c20da3cSBarry Smith     have been created. One should use ISColoringGetIS() to make sure they are
2445c20da3cSBarry Smith     created when needed.
2455c20da3cSBarry Smith 
246a5057860SBarry Smith     Developer Note: this is not a PetscObject
247a5057860SBarry Smith 
2485c20da3cSBarry Smith .seealso:  ISColoringCreate(), ISColoringGetIS(), ISColoringView(), ISColoringGetIS()
2495c20da3cSBarry Smith S*/
25095fbd943SSatish Balay struct _n_ISColoring {
251a7cc72afSBarry Smith   PetscInt        refct;
252a7cc72afSBarry Smith   PetscInt        n;                /* number of colors */
2535c20da3cSBarry Smith   IS              *is;              /* for each color indicates columns */
254639f9d9dSBarry Smith   MPI_Comm        comm;
25508b6dcc0SBarry Smith   ISColoringValue *colors;          /* for each column indicates color */
25632dcc486SBarry Smith   PetscInt        N;                /* number of columns */
257b9617806SBarry Smith   ISColoringType  ctype;
258639f9d9dSBarry Smith };
25995fbd943SSatish Balay typedef struct _n_ISColoring* ISColoring;
260639f9d9dSBarry Smith 
261014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISColoringCreate(MPI_Comm,PetscInt,PetscInt,const ISColoringValue[],ISColoring*);
262014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISColoringDestroy(ISColoring*);
263014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISColoringView(ISColoring,PetscViewer);
264014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISColoringGetIS(ISColoring,PetscInt*,IS*[]);
265014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISColoringRestoreIS(ISColoring,IS*[]);
26649b734d2SBarry Smith PETSC_EXTERN PetscErrorCode ISColoringReference(ISColoring);
26749b734d2SBarry Smith PETSC_EXTERN PetscErrorCode ISColoringSetType(ISColoring,ISColoringType);
26849b734d2SBarry Smith 
2693a7fca6bSBarry Smith 
270dbef8a1cSBarry Smith /* --------------------------------------------------------------------------*/
271dbef8a1cSBarry Smith 
272014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISPartitioningToNumbering(IS,IS*);
273014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISPartitioningCount(IS,PetscInt,PetscInt[]);
274dbef8a1cSBarry Smith 
275014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISCompressIndicesGeneral(PetscInt,PetscInt,PetscInt,PetscInt,const IS[],IS[]);
276014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISCompressIndicesSorted(PetscInt,PetscInt,PetscInt,const IS[],IS[]);
277014dd563SJed Brown PETSC_EXTERN PetscErrorCode ISExpandIndicesGeneral(PetscInt,PetscInt,PetscInt,PetscInt,const IS[],IS[]);
278d9489beaSHong Zhang 
27969ce434fSBarry Smith /*S
28069ce434fSBarry Smith      PetscLayout - defines layout of vectors and matrices across processes (which rows are owned by which processes)
28169ce434fSBarry Smith 
28269ce434fSBarry Smith    Level: developer
28369ce434fSBarry Smith 
28469ce434fSBarry Smith 
28569ce434fSBarry Smith .seealso:  PetscLayoutCreate(), PetscLayoutDestroy()
28669ce434fSBarry Smith S*/
28769ce434fSBarry Smith typedef struct _n_PetscLayout* PetscLayout;
28869ce434fSBarry Smith struct _n_PetscLayout{
28969ce434fSBarry Smith   MPI_Comm               comm;
29069ce434fSBarry Smith   PetscInt               n,N;         /* local, global vector size */
29169ce434fSBarry Smith   PetscInt               rstart,rend; /* local start, local end + 1 */
29269ce434fSBarry Smith   PetscInt               *range;      /* the offset of each processor */
29369ce434fSBarry Smith   PetscInt               bs;          /* number of elements in each block (generally for multi-component problems) Do NOT multiply above numbers by bs */
29469ce434fSBarry Smith   PetscInt               refcnt;      /* MPI Vecs obtained with VecDuplicate() and from MatGetVecs() reuse map of input object */
29569ce434fSBarry Smith   ISLocalToGlobalMapping mapping;     /* mapping used in Vec/MatSetValuesLocal() */
29669ce434fSBarry Smith   ISLocalToGlobalMapping bmapping;    /* mapping used in Vec/MatSetValuesBlockedLocal() */
29769ce434fSBarry Smith   PetscInt               *trstarts;   /* local start for each thread */
29869ce434fSBarry Smith };
29969ce434fSBarry Smith 
30069ce434fSBarry Smith PETSC_EXTERN PetscErrorCode PetscLayoutCreate(MPI_Comm,PetscLayout*);
30169ce434fSBarry Smith PETSC_EXTERN PetscErrorCode PetscLayoutSetUp(PetscLayout);
30269ce434fSBarry Smith PETSC_EXTERN PetscErrorCode PetscLayoutDestroy(PetscLayout*);
30369ce434fSBarry Smith PETSC_EXTERN PetscErrorCode PetscLayoutDuplicate(PetscLayout,PetscLayout*);
30469ce434fSBarry Smith PETSC_EXTERN PetscErrorCode PetscLayoutReference(PetscLayout,PetscLayout*);
30569ce434fSBarry Smith PETSC_EXTERN PetscErrorCode PetscLayoutSetLocalSize(PetscLayout,PetscInt);
30669ce434fSBarry Smith PETSC_EXTERN PetscErrorCode PetscLayoutGetLocalSize(PetscLayout,PetscInt *);
30769ce434fSBarry Smith PETSC_EXTERN PetscErrorCode PetscLayoutSetSize(PetscLayout,PetscInt);
30869ce434fSBarry Smith PETSC_EXTERN PetscErrorCode PetscLayoutGetSize(PetscLayout,PetscInt *);
30969ce434fSBarry Smith PETSC_EXTERN PetscErrorCode PetscLayoutSetBlockSize(PetscLayout,PetscInt);
31069ce434fSBarry Smith PETSC_EXTERN PetscErrorCode PetscLayoutGetBlockSize(PetscLayout,PetscInt*);
31169ce434fSBarry Smith PETSC_EXTERN PetscErrorCode PetscLayoutGetRange(PetscLayout,PetscInt *,PetscInt *);
31269ce434fSBarry Smith PETSC_EXTERN PetscErrorCode PetscLayoutGetRanges(PetscLayout,const PetscInt *[]);
31369ce434fSBarry Smith PETSC_EXTERN PetscErrorCode PetscLayoutSetISLocalToGlobalMapping(PetscLayout,ISLocalToGlobalMapping);
31469ce434fSBarry Smith PETSC_EXTERN PetscErrorCode PetscLayoutSetISLocalToGlobalMappingBlock(PetscLayout,ISLocalToGlobalMapping);
31569ce434fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSFSetGraphLayout(PetscSF,PetscLayout,PetscInt,const PetscInt*,PetscCopyMode,const PetscInt*);
31669ce434fSBarry Smith 
31769ce434fSBarry Smith #undef __FUNCT__
31869ce434fSBarry Smith #define __FUNCT__ "PetscLayoutFindOwner"
31969ce434fSBarry Smith /*@C
32069ce434fSBarry Smith      PetscLayoutFindOwner - Find the owning rank for a global index
32169ce434fSBarry Smith 
32269ce434fSBarry Smith     Not Collective
32369ce434fSBarry Smith 
32469ce434fSBarry Smith    Input Parameters:
32569ce434fSBarry Smith +    map - the layout
32669ce434fSBarry Smith -    idx - global index to find the owner of
32769ce434fSBarry Smith 
32869ce434fSBarry Smith    Output Parameter:
32969ce434fSBarry Smith .    owner - the owning rank
33069ce434fSBarry Smith 
33169ce434fSBarry Smith    Level: developer
33269ce434fSBarry Smith 
33369ce434fSBarry Smith     Fortran Notes:
33469ce434fSBarry Smith       Not available from Fortran
33569ce434fSBarry Smith 
33669ce434fSBarry Smith @*/
33769ce434fSBarry Smith PETSC_STATIC_INLINE PetscErrorCode PetscLayoutFindOwner(PetscLayout map,PetscInt idx,PetscInt *owner)
33869ce434fSBarry Smith {
33969ce434fSBarry Smith   PetscErrorCode ierr;
34069ce434fSBarry Smith   PetscMPIInt    lo = 0,hi,t;
34169ce434fSBarry Smith 
34269ce434fSBarry Smith   PetscFunctionBegin;
34369ce434fSBarry Smith   *owner = -1;                  /* GCC erroneously issues warning about possibly uninitialized use when error condition */
34469ce434fSBarry Smith   if (!((map->n >= 0) && (map->N >= 0) && (map->range))) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"PetscLayoutSetUp() must be called first");
34569ce434fSBarry Smith   if (idx < 0 || idx > map->N) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Index %D is out of range",idx);
34669ce434fSBarry Smith   ierr = MPI_Comm_size(map->comm,&hi);CHKERRQ(ierr);
34769ce434fSBarry Smith   while (hi - lo > 1) {
34869ce434fSBarry Smith     t = lo + (hi - lo) / 2;
34969ce434fSBarry Smith     if (idx < map->range[t]) hi = t;
35069ce434fSBarry Smith     else                     lo = t;
35169ce434fSBarry Smith   }
35269ce434fSBarry Smith   *owner = lo;
35369ce434fSBarry Smith   PetscFunctionReturn(0);
35469ce434fSBarry Smith }
35569ce434fSBarry Smith 
35669ce434fSBarry Smith #undef __FUNCT__
35769ce434fSBarry Smith #define __FUNCT__ "PetscLayoutFindOwnerIndex"
35869ce434fSBarry Smith /*@C
35969ce434fSBarry Smith      PetscLayoutFindOwnerIndex - Find the owning rank and the local index for a global index
36069ce434fSBarry Smith 
36169ce434fSBarry Smith     Not Collective
36269ce434fSBarry Smith 
36369ce434fSBarry Smith    Input Parameters:
36469ce434fSBarry Smith +    map   - the layout
36569ce434fSBarry Smith -    idx   - global index to find the owner of
36669ce434fSBarry Smith 
36769ce434fSBarry Smith    Output Parameter:
36869ce434fSBarry Smith +    owner - the owning rank
36969ce434fSBarry Smith -    lidx  - local index used by the owner for idx
37069ce434fSBarry Smith 
37169ce434fSBarry Smith    Level: developer
37269ce434fSBarry Smith 
37369ce434fSBarry Smith     Fortran Notes:
37469ce434fSBarry Smith       Not available from Fortran
37569ce434fSBarry Smith 
37669ce434fSBarry Smith @*/
37769ce434fSBarry Smith PETSC_STATIC_INLINE PetscErrorCode PetscLayoutFindOwnerIndex(PetscLayout map,PetscInt idx,PetscInt *owner, PetscInt *lidx)
37869ce434fSBarry Smith {
37969ce434fSBarry Smith   PetscErrorCode ierr;
38069ce434fSBarry Smith   PetscMPIInt    lo = 0,hi,t;
38169ce434fSBarry Smith 
38269ce434fSBarry Smith   PetscFunctionBegin;
38369ce434fSBarry Smith   if (!((map->n >= 0) && (map->N >= 0) && (map->range))) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"PetscLayoutSetUp() must be called first");
38469ce434fSBarry Smith   if (idx < 0 || idx > map->N) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Index %D is out of range",idx);
38569ce434fSBarry Smith   ierr = MPI_Comm_size(map->comm,&hi);CHKERRQ(ierr);
38669ce434fSBarry Smith   while (hi - lo > 1) {
38769ce434fSBarry Smith     t = lo + (hi - lo) / 2;
38869ce434fSBarry Smith     if (idx < map->range[t]) hi = t;
38969ce434fSBarry Smith     else                     lo = t;
39069ce434fSBarry Smith   }
39169ce434fSBarry Smith   *owner = lo;
39269ce434fSBarry Smith   *lidx  = idx-map->range[lo];
39369ce434fSBarry Smith   PetscFunctionReturn(0);
39469ce434fSBarry Smith }
3959e03d832SJed Brown 
396*d53a3d6fSBarry Smith /*S
397*d53a3d6fSBarry Smith   PetscSection - This is a mapping from DMMESH points to sets of values, which is
398*d53a3d6fSBarry Smith   our presentation of a fibre bundle.
399*d53a3d6fSBarry Smith 
400*d53a3d6fSBarry Smith   Level: developer
401*d53a3d6fSBarry Smith 
402*d53a3d6fSBarry Smith .seealso:  PetscSectionCreate(), PetscSectionDestroy()
403*d53a3d6fSBarry Smith S*/
404*d53a3d6fSBarry Smith typedef struct _n_PetscSection *PetscSection;
405*d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionCreate(MPI_Comm,PetscSection*);
406*d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionClone(PetscSection, PetscSection*);
407*d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionGetNumFields(PetscSection, PetscInt *);
408*d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionSetNumFields(PetscSection, PetscInt);
409*d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionGetFieldName(PetscSection, PetscInt, const char *[]);
410*d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionSetFieldName(PetscSection, PetscInt, const char []);
411*d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionGetFieldComponents(PetscSection, PetscInt, PetscInt *);
412*d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionSetFieldComponents(PetscSection, PetscInt, PetscInt);
413*d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionGetChart(PetscSection, PetscInt *, PetscInt *);
414*d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionSetChart(PetscSection, PetscInt, PetscInt);
415*d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionGetDof(PetscSection, PetscInt, PetscInt*);
416*d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionSetDof(PetscSection, PetscInt, PetscInt);
417*d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionAddDof(PetscSection, PetscInt, PetscInt);
418*d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionGetFieldDof(PetscSection, PetscInt, PetscInt, PetscInt*);
419*d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionSetFieldDof(PetscSection, PetscInt, PetscInt, PetscInt);
420*d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionAddFieldDof(PetscSection, PetscInt, PetscInt, PetscInt);
421*d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionGetConstraintDof(PetscSection, PetscInt, PetscInt*);
422*d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionSetConstraintDof(PetscSection, PetscInt, PetscInt);
423*d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionAddConstraintDof(PetscSection, PetscInt, PetscInt);
424*d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionGetFieldConstraintDof(PetscSection, PetscInt, PetscInt, PetscInt*);
425*d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionSetFieldConstraintDof(PetscSection, PetscInt, PetscInt, PetscInt);
426*d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionAddFieldConstraintDof(PetscSection, PetscInt, PetscInt, PetscInt);
427*d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionGetConstraintIndices(PetscSection, PetscInt, const PetscInt**);
428*d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionSetConstraintIndices(PetscSection, PetscInt, const PetscInt*);
429*d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionGetFieldConstraintIndices(PetscSection, PetscInt, PetscInt, const PetscInt**);
430*d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionSetFieldConstraintIndices(PetscSection, PetscInt, PetscInt, const PetscInt*);
431*d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionSetUpBC(PetscSection);
432*d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionSetUp(PetscSection);
433*d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionGetMaxDof(PetscSection, PetscInt*);
434*d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionGetStorageSize(PetscSection, PetscInt*);
435*d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionGetConstrainedStorageSize(PetscSection, PetscInt*);
436*d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionGetOffset(PetscSection, PetscInt, PetscInt*);
437*d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionSetOffset(PetscSection, PetscInt, PetscInt);
438*d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionGetFieldOffset(PetscSection, PetscInt, PetscInt, PetscInt*);
439*d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionSetFieldOffset(PetscSection, PetscInt, PetscInt, PetscInt);
440*d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionGetOffsetRange(PetscSection, PetscInt *, PetscInt *);
441*d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionView(PetscSection, PetscViewer);
442*d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionReset(PetscSection);
443*d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionDestroy(PetscSection*);
444*d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionCreateGlobalSection(PetscSection, PetscSF, PetscBool, PetscSection *);
445*d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionCreateGlobalSectionCensored(PetscSection, PetscSF, PetscBool, PetscInt, const PetscInt [], PetscSection *);
446*d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionCreateSubsection(PetscSection, PetscInt, PetscInt [], PetscSection *);
447*d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionCreateSubmeshSection(PetscSection, IS, PetscSection *);
448*d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionGetPointLayout(MPI_Comm, PetscSection, PetscLayout *);
449*d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSectionGetValueLayout(MPI_Comm, PetscSection, PetscLayout *);
450*d53a3d6fSBarry Smith 
451*d53a3d6fSBarry Smith /* PetscSF support */
452*d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSFConvertPartition(PetscSF, PetscSection, IS, ISLocalToGlobalMapping *, PetscSF *);
453*d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSFCreateRemoteOffsets(PetscSF, PetscSection, PetscSection, PetscInt **);
454*d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSFDistributeSection(PetscSF, PetscSection, PetscInt **, PetscSection);
455*d53a3d6fSBarry Smith PETSC_EXTERN PetscErrorCode PetscSFCreateSectionSF(PetscSF, PetscSection, PetscInt [], PetscSection, PetscSF *);
456*d53a3d6fSBarry Smith 
4579e03d832SJed Brown /* Reset __FUNCT__ in case the user does not define it themselves */
4589e03d832SJed Brown #undef __FUNCT__
4599e03d832SJed Brown #define __FUNCT__ "User provided function"
4609e03d832SJed Brown 
461a2ce50c7SBarry Smith #endif
462