xref: /petsc/src/dm/impls/plex/tests/ex6.c (revision b122ec5aa1bd4469eb4e0673542fb7de3f411254)

static char help[] = "Tests for DMLabel lookup\n\n";

#include <petscdmplex.h>

typedef struct {
  PetscInt  debug;        /* The debugging level */
  PetscInt  pStart, pEnd; /* The label chart */
  PetscInt  numStrata;    /* The number of label strata */
  PetscReal fill;         /* Percentage of label to fill */
  PetscInt  size;         /* The number of set values */
} AppCtx;

PetscErrorCode ProcessOptions(MPI_Comm comm, AppCtx *options)
{
  PetscErrorCode ierr;

  PetscFunctionBegin;
  options->debug     = 0;
  options->pStart    = 0;
  options->pEnd      = 1000;
  options->numStrata = 5;
  options->fill      = 0.10;

  ierr = PetscOptionsBegin(comm, "", "Meshing Problem Options", "DMPLEX");CHKERRQ(ierr);
  CHKERRQ(PetscOptionsBoundedInt("-debug", "The debugging level", "ex6.c", options->debug, &options->debug, NULL,0));
  CHKERRQ(PetscOptionsBoundedInt("-num_strata", "The number of label values", "ex6.c", options->numStrata, &options->numStrata, NULL,0));
  CHKERRQ(PetscOptionsBoundedInt("-pend", "The label point limit", "ex6.c", options->pEnd, &options->pEnd, NULL,0));
  CHKERRQ(PetscOptionsReal("-fill", "The percentage of label chart to set", "ex6.c", options->fill, &options->fill, NULL));
  ierr = PetscOptionsEnd();CHKERRQ(ierr);
  PetscFunctionReturn(0);
}

PetscErrorCode TestSetup(DMLabel label, AppCtx *user)
{
  PetscRandom    r;
  PetscInt       n = (PetscInt) (user->fill*(user->pEnd - user->pStart)), i;

  PetscFunctionBegin;
  CHKERRQ(PetscRandomCreate(PETSC_COMM_SELF, &r));
  CHKERRQ(PetscRandomSetFromOptions(r));/* -random_type <> */
  CHKERRQ(PetscRandomSetInterval(r, user->pStart, user->pEnd));
  CHKERRQ(PetscRandomSetSeed(r, 123456789L));
  CHKERRQ(PetscRandomSeed(r));
  user->size = 0;
  for (i = 0; i < n; ++i) {
    PetscReal p;
    PetscInt  val;

    CHKERRQ(PetscRandomGetValueReal(r, &p));
    CHKERRQ(DMLabelGetValue(label, (PetscInt) p, &val));
    if (val < 0) {
      ++user->size;
      CHKERRQ(DMLabelSetValue(label, (PetscInt) p, i % user->numStrata));
    }
  }
  CHKERRQ(PetscRandomDestroy(&r));
  CHKERRQ(DMLabelCreateIndex(label, user->pStart, user->pEnd));
  CHKERRQ(PetscPrintf(PETSC_COMM_SELF, "Created label with chart [%D, %D) and set %D values\n", user->pStart, user->pEnd, user->size));
  PetscFunctionReturn(0);
}

PetscErrorCode TestLookup(DMLabel label, AppCtx *user)
{
  const PetscInt pStart = user->pStart;
  const PetscInt pEnd   = user->pEnd;
  PetscInt       p, n = 0;

  PetscFunctionBegin;
  for (p = pStart; p < pEnd; ++p) {
    PetscInt  val;
    PetscBool has;

    CHKERRQ(DMLabelGetValue(label, p, &val));
    CHKERRQ(DMLabelHasPoint(label, p, &has));
    PetscCheckFalse(((val >= 0) && !has) || ((val < 0) && has),PETSC_COMM_SELF, PETSC_ERR_PLIB, "Label value %D does not match contains check %D for point %D", val, (PetscInt) has, p);
    if (has) ++n;
  }
  PetscCheckFalse(n != user->size,PETSC_COMM_SELF, PETSC_ERR_PLIB, "Invalid number of label points detected %D does not match number set %D", n, user->size);
  /* Also put in timing code */
  PetscFunctionReturn(0);
}

PetscErrorCode TestClear(DMLabel label, AppCtx *user)
{
  PetscInt       pStart = user->pStart, pEnd = user->pEnd, p;
  PetscInt       defaultValue;

  PetscFunctionBegin;
  CHKERRQ(DMLabelGetDefaultValue(label,&defaultValue));
  for (p = pStart; p < pEnd; p++) {
    PetscInt  val;
    PetscBool hasPoint;

    CHKERRQ(DMLabelGetValue(label,p,&val));
    if (val != defaultValue) {
      CHKERRQ(DMLabelClearValue(label,p,val));
    }
    CHKERRQ(DMLabelGetValue(label,p,&val));
    CHKERRQ(DMLabelHasPoint(label,p,&hasPoint));
    PetscCheckFalse(val != defaultValue,PETSC_COMM_SELF,PETSC_ERR_PLIB,"Expected default value %D after clearing point %D, got %D",defaultValue,p,val);
    PetscCheck(!hasPoint,PETSC_COMM_SELF,PETSC_ERR_PLIB,"Label contains %D after clearing",p);
  }
  PetscFunctionReturn(0);
}

int main(int argc, char **argv)
{
  DMLabel        label;
  AppCtx         user;                 /* user-defined work context */

  CHKERRQ(PetscInitialize(&argc, &argv, NULL,help));
  CHKERRQ(ProcessOptions(PETSC_COMM_WORLD, &user));
  CHKERRQ(DMLabelCreate(PETSC_COMM_SELF, "Test Label", &label));
  CHKERRQ(TestSetup(label, &user));
  CHKERRQ(TestLookup(label, &user));
  CHKERRQ(TestClear(label,&user));
  CHKERRQ(DMLabelDestroy(&label));
  CHKERRQ(PetscFinalize());
  return 0;
}

/*TEST

  test:
    suffix: 0
    args: -malloc_dump
  test:
    suffix: 1
    args: -malloc_dump -pend 10000
  test:
    suffix: 2
    args: -malloc_dump -pend 10000 -fill 0.05
  test:
    suffix: 3
    args: -malloc_dump -pend 10000 -fill 0.25

TEST*/