testiterator.cc 2.43 KB
Newer Older
Praetorius, Simon's avatar
Praetorius, Simon committed
1 2 3 4 5 6 7 8 9 10 11
// -*- tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*-
// vi: set et ts=4 sw=2 sts=2:

#ifdef HAVE_CONFIG_H
# include "config.h"
#endif
#include <iostream>
#include <dune/common/parallel/mpihelper.hh> // An initializer of MPI
#include <dune/common/exceptions.hh> // We use exceptions
#include <dune/common/timer.hh>

Praetorius, Simon's avatar
Praetorius, Simon committed
12
#include <dune/grid/albertagrid.hh>
Praetorius, Simon's avatar
Praetorius, Simon committed
13
#include <dune/grid/yaspgrid.hh>
14 15
#include <dune/grid/uggrid.hh>
#include <dune/alugrid/grid.hh>
Praetorius, Simon's avatar
Praetorius, Simon committed
16 17

#include <dune/multimesh/multimesh.hh>
18
#include <dune/multimesh/mmgridfactory.hh>
Praetorius, Simon's avatar
Praetorius, Simon committed
19
#include <dune/multimesh/utility/structuredgridbuilder.hh>
20

21
#define GRID 1
Praetorius, Simon's avatar
Praetorius, Simon committed
22 23 24 25 26 27 28 29 30

using namespace Dune;

template <class Grid>
void printGrid(Grid const& grid)
{
  volatile std::size_t n = 0;
  Dune::Timer t;
  for (auto const& entities : elements(grid.leafGridView()))
31
    n += grid.grid(0).leafIndexSet().index(entities[0]);
Praetorius, Simon's avatar
Praetorius, Simon committed
32 33 34 35 36 37 38 39 40 41
  std::cout << n << "\n";
  std::cout << "time: " << t.elapsed() << "\n";
}

template <class Grid>
void printGrid2(Grid const& grid)
{
  volatile std::size_t n = 0;
  Dune::Timer t;
  for (auto const& entity : elements(grid.leafGridView()))
42
    n += grid.leafIndexSet().index(entity);
Praetorius, Simon's avatar
Praetorius, Simon committed
43 44 45 46 47 48 49 50 51
  std::cout << n << "\n";
  std::cout << "time: " << t.elapsed() << "\n";
}


int main(int argc, char** argv)
{
  MPIHelper::instance(argc, argv);

52
#if GRID == 1
Praetorius, Simon's avatar
Praetorius, Simon committed
53

Praetorius, Simon's avatar
Praetorius, Simon committed
54 55 56 57 58
  FieldVector<double,3> lower_left = {-1.5, -1.5, -1.5};
  FieldVector<double,3> bbox = {1.5, 1.5, 1.5};
  std::array<int,3> num_elements = {16, 16, 16};
  using HostGrid = YaspGrid<3, EquidistantOffsetCoordinates<double,3>>;
  MultiMesh<HostGrid> grid(2, lower_left, bbox, num_elements);
Praetorius, Simon's avatar
Praetorius, Simon committed
59

Praetorius, Simon's avatar
Praetorius, Simon committed
60
#elif GRID == 2
Praetorius, Simon's avatar
Praetorius, Simon committed
61

62 63 64 65
  FieldVector<double,2> lower_left = {-1.5, -1.5};
  FieldVector<double,2> bbox = {1.5, 1.5};
  std::array<unsigned int,2> num_elements = {2, 2};
  using HostGrid = Dune::ALUGrid<2, 2, Dune::simplex, Dune::conforming>;
66

Praetorius, Simon's avatar
Praetorius, Simon committed
67
  using Factory = StructuredGridBuilder<MultiMesh<HostGrid> >;
68 69 70 71
  GridFactory<MultiMesh<HostGrid> > gridFactory(2);

  auto gridPtr = Factory::createSimplexGrid(gridFactory, lower_left, bbox, num_elements);
  auto& grid = *gridPtr;
Praetorius, Simon's avatar
Praetorius, Simon committed
72

73
#elif GRID == 3
Praetorius, Simon's avatar
Praetorius, Simon committed
74

Praetorius, Simon's avatar
Praetorius, Simon committed
75 76 77 78 79 80
  assert(argc > 1);
  std::string filename = argv[1];
  using HostGrid = Dune::AlbertaGrid<3,3>;
  MultiMesh<HostGrid> grid(2, filename);
  grid.grid(0).globalRefine(8);
  grid.grid(1).globalRefine(8);
81

Praetorius, Simon's avatar
Praetorius, Simon committed
82 83
#endif

84 85
  std::cout << "number of grids = " << grid.numGrids() << "\n";

Praetorius, Simon's avatar
Praetorius, Simon committed
86 87 88 89 90 91
  printGrid2(grid.grid(0));
  printGrid(grid);
  grid.grid(0).globalRefine(1);
  printGrid2(grid.grid(0));
  printGrid(grid);
}