parsed_grid_refinement.h

Go to the documentation of this file.

//-----------------------------------------------------------
//
//    Copyright (C) 2015 by the deal2lkit authors
//
//    This file is part of the deal2lkit library.
//
//    The deal2lkit library is free software; you can use it, redistribute
//    it, and/or modify it under the terms of the GNU Lesser General
//    Public License as published by the Free Software Foundation; either
//    version 2.1 of the License, or (at your option) any later version.
//    The full text of the license can be found in the file LICENSE at
//    the top level of the deal2lkit distribution.
//
//-----------------------------------------------------------

#ifndef _d2k_parsed_grid_refinement_h
#define _d2k_parsed_grid_refinement_h

#include <deal.II/grid/tria.h>
#include <deal.II/grid/grid_refinement.h>

#ifdef DEAL_II_WITH_MPI
#ifdef DEAL_II_WITH_P4EST
#include <deal.II/distributed/tria.h>
#include <deal.II/distributed/grid_refinement.h>
#endif
#endif


#include <deal2lkit/config.h>
#include <deal2lkit/parameter_acceptor.h>

using namespace dealii;


D2K_NAMESPACE_OPEN
class ParsedGridRefinement : public ParameterAcceptor
{
public:
  ParsedGridRefinement(const std::string &name="",
                       const std::string &strategy="fraction",
                       const double &top_parameter=.3,
                       const double &bottom_parameter=.1,
                       const unsigned int &max_cells=0,
                       const unsigned int &order=2);

  virtual void declare_parameters(ParameterHandler &prm);


  template<int dim, class Vector , int spacedim>
  void mark_cells(const Vector &criteria,
                  Triangulation< dim, spacedim > &tria) const;

#ifdef DEAL_II_WITH_MPI
#ifdef DEAL_II_WITH_P4EST

  template<int dim, class Vector , int spacedim>
  void mark_cells(const Vector &criteria,
                  parallel::distributed::Triangulation< dim, spacedim > &tria) const;
#endif
#endif

private:
  std::string strategy;
  double top_parameter;
  double bottom_parameter;
  unsigned int max_cells;
  unsigned int order;
};



#ifdef DEAL_II_WITH_MPI
#ifdef DEAL_II_WITH_P4EST
template<int dim, class Vector , int spacedim>
void ParsedGridRefinement::mark_cells(const Vector &criteria,
                                      parallel::distributed::Triangulation< dim, spacedim > &tria) const
{
  if (strategy == "number")
    parallel::distributed::GridRefinement::refine_and_coarsen_fixed_number (tria,
        criteria,
        top_parameter, bottom_parameter,
        max_cells ? max_cells :
        std::numeric_limits< unsigned int >::max());
  else if (strategy == "fraction")
    parallel::distributed::GridRefinement::refine_and_coarsen_fixed_fraction (tria,
        criteria,
        top_parameter, bottom_parameter);
  else
    Assert(false, ExcInternalError());

}
#endif
#endif

template<int dim, class Vector , int spacedim>
void ParsedGridRefinement::mark_cells(const Vector &criteria,
                                      Triangulation< dim, spacedim > &tria) const
{
  if (strategy == "number")
    GridRefinement::refine_and_coarsen_fixed_number (tria,
                                                     criteria,
                                                     top_parameter, bottom_parameter,
                                                     max_cells ? max_cells :
                                                     std::numeric_limits< unsigned int >::max());
  else if (strategy == "fraction")
    GridRefinement::refine_and_coarsen_fixed_fraction (tria,
                                                       criteria,
                                                       top_parameter, bottom_parameter,
                                                       max_cells ? max_cells :
                                                       std::numeric_limits< unsigned int >::max());
  // This one does not seem to work properly
  // else if(strategy == "optimize")
  //   GridRefinement::refine_and_coarsen_optimize (tria, order);
  else
    Assert(false, ExcInternalError());
}

D2K_NAMESPACE_CLOSE


#endif