xref: /petsc/src/dm/impls/composite/packimpl.h (revision bebe2cf65d55febe21a5af8db2bd2e168caaa2e7)
1a12302e2SJed Brown #if !defined(_packimpl_h)
2a12302e2SJed Brown #define _packimpl_h
3a12302e2SJed Brown 
43c48a1e8SJed Brown #include <petscdmcomposite.h>    /*I "petscdmcomposite.h" I*/
5af0996ceSBarry Smith #include <petsc/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 */
269ae5db72SJed Brown   PetscInt               nghost;               /* number of all local entries (includes DMDA ghost points) */
27*bebe2cf6SSatish Balay   PetscInt               nDM,nmine;            /* how many DM's and separate 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 
34b412c318SBarry Smith PETSC_INTERN PetscErrorCode DMCreateMatrix_Composite(DM,Mat*);
35a12302e2SJed Brown 
36a12302e2SJed Brown #endif
37