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