xref: /petsc/src/dm/impls/composite/packimpl.h (revision 09573ac72a50d3e7ecd55a2b7f0ef28450cd0a8b)
1a12302e2SJed Brown #if !defined(_packimpl_h)
2a12302e2SJed Brown #define _packimpl_h
3a12302e2SJed Brown 
4a12302e2SJed Brown #include <private/dmimpl.h>      /*I      "petscdm.h"     I*/
5a12302e2SJed Brown 
6a12302e2SJed Brown /*
7a12302e2SJed Brown    rstart is where an array/subvector starts in the global parallel vector, so arrays
8a12302e2SJed Brown    rstarts are meaningless (and set to the previous one) except on the processor where the array lives
9a12302e2SJed Brown */
10a12302e2SJed Brown 
11a12302e2SJed Brown typedef enum {DMCOMPOSITE_ARRAY, DMCOMPOSITE_DM} DMCompositeLinkType;
12a12302e2SJed Brown 
13a12302e2SJed Brown struct DMCompositeLink {
14a12302e2SJed Brown   DMCompositeLinkType    type;
15a12302e2SJed Brown   struct DMCompositeLink *next;
16a12302e2SJed Brown   PetscInt               n;             /* number of owned */
17a12302e2SJed Brown   PetscInt               rstart;        /* rstart is relative to this process */
18a12302e2SJed Brown   PetscInt               grstart;       /* grstart is relative to all processes */
19a12302e2SJed Brown   PetscInt               nlocal;
20a12302e2SJed Brown 
21a12302e2SJed Brown   /* only used for DMCOMPOSITE_DM */
22a12302e2SJed Brown   PetscInt               *grstarts;     /* global row for first unknown of this DM on each process */
23a12302e2SJed Brown   DM                     dm;
24a12302e2SJed Brown 
25a12302e2SJed Brown   /* only used for DMCOMPOSITE_ARRAY */
26a12302e2SJed Brown   PetscMPIInt            rank;          /* process where array unknowns live */
27a12302e2SJed Brown };
28a12302e2SJed Brown 
29a12302e2SJed Brown typedef struct {
30a12302e2SJed Brown   PetscInt               n,N,rstart;           /* rstart is relative to all processors, n unknowns owned by this process, N is total unknowns */
31a12302e2SJed Brown   PetscInt               nghost;               /* number of all local entries include DMDA ghost points and any shared redundant arrays */
32a12302e2SJed Brown   PetscInt               nDM,nredundant,nmine; /* how many DM's and seperate redundant arrays used to build DM(nmine is ones on this process) */
33a12302e2SJed Brown   PetscBool              setup;                /* after this is set, cannot add new links to the DM*/
34a12302e2SJed Brown   struct DMCompositeLink *next;
35a12302e2SJed Brown 
36a12302e2SJed Brown   PetscErrorCode (*FormCoupleLocations)(DM,Mat,PetscInt*,PetscInt*,PetscInt,PetscInt,PetscInt,PetscInt);
37a12302e2SJed Brown } DM_Composite;
38a12302e2SJed Brown 
39*09573ac7SBarry Smith extern PetscErrorCode DMGetMatrix_Composite(DM,const MatType,Mat*);
40a12302e2SJed Brown 
41a12302e2SJed Brown #endif
42