1: #pragma once

  3: #include <petscviewer.h>
  4: #include <petscpartitioner.h>
  5: #include <petsc/private/petscimpl.h>

  7: PETSC_EXTERN PetscBool      PetscPartitionerRegisterAllCalled;
  8: PETSC_EXTERN PetscErrorCode PetscPartitionerRegisterAll(void);

 10: typedef struct _PetscPartitionerOps *PetscPartitionerOps;
 11: struct _PetscPartitionerOps {
 12:   PetscErrorCode (*setfromoptions)(PetscPartitioner, PetscOptionItems *);
 13:   PetscErrorCode (*setup)(PetscPartitioner);
 14:   PetscErrorCode (*reset)(PetscPartitioner);
 15:   PetscErrorCode (*view)(PetscPartitioner, PetscViewer);
 16:   PetscErrorCode (*destroy)(PetscPartitioner);
 17:   PetscErrorCode (*partition)(PetscPartitioner, PetscInt, PetscInt, PetscInt[], PetscInt[], PetscSection, PetscSection, PetscSection, PetscSection, IS *);
 18: };

 20: struct _p_PetscPartitioner {
 21:   PETSCHEADER(struct _PetscPartitionerOps);
 22:   void       *data;    /* Implementation object */
 23:   PetscInt    height;  /* Height of points to partition into non-overlapping subsets */
 24:   PetscInt    edgeCut; /* The number of edge cut by the partition */
 25:   PetscReal   balance; /* The maximum partition size divided by the minimum size */
 26:   PetscViewer viewer;
 27:   PetscViewer viewerGraph;
 28:   PetscBool   viewGraph;
 29:   PetscBool   noGraph; /* if true, the partitioner does not need the connectivity graph, only the number of local vertices */
 30:   PetscBool   usevwgt; /* if true, the partitioner looks at the local section vertSection to weight the vertices of the graph */
 31:   PetscBool   useewgt; /* if true, the partitioner looks at the topology to weight the edges of the graph */
 32: };