EntityRefiner.hpp

Go to the documentation of this file.

/*
 * MOAB, a Mesh-Oriented datABase, is a software component for creating,
 * storing and accessing finite element mesh data.
 * 
 * Copyright 2007 Sandia Corporation.  Under the terms of Contract
 * DE-AC04-94AL85000 with Sandia Coroporation, the U.S. Government
 * retains certain rights in this software.
 * 
 * This library is free software; you can 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.
 * 
 */

#ifndef MOAB_ENTITY_REFINER_HPP
#define MOAB_ENTITY_REFINER_HPP

#include "moab/Types.hpp"
#include "moab/Compiler.hpp" // for MB_DLL_EXPORT

#include <vector>

namespace moab {

class Interface;
class EdgeSizeEvaluator;
class RefinerTagManager;

class MB_DLL_EXPORT EntityRefinerOutputFunctor
{
public:
  virtual ~EntityRefinerOutputFunctor() { }
  virtual EntityHandle map_vertex( EntityHandle vhash, const double* vcoords, const void* vtags ) = 0;
  EntityHandle operator () ( EntityHandle h0, EntityHandle h1, const double* vcoords, const void* vtags )
    {
    EntityHandle harr[2];
    harr[0] = h0;
    harr[1] = h1;
    return (*this)( 2, harr, vcoords, vtags );
    }
  virtual EntityHandle operator () ( EntityHandle h0, EntityHandle h1, EntityHandle h2, const double* vcoords, const void* vtags )
    {
    EntityHandle harr[3];
    harr[0] = h0;
    harr[1] = h1;
    harr[2] = h2;
    return (*this)( 3, harr, vcoords, vtags );
    }
  virtual EntityHandle operator () ( int nhash, EntityHandle* hash, const double* vcoords, const void* vtags ) = 0;
  virtual void operator () ( EntityHandle vhash ) = 0;
  virtual void operator () ( EntityType etyp ) = 0;
};

class MB_DLL_EXPORT EntityRefiner
{
public:
  EntityRefiner();
  virtual ~EntityRefiner();

  virtual bool prepare( RefinerTagManager* tmgr, EntityRefinerOutputFunctor* ofunc );
  virtual bool refine_entity( EntityType typ, EntityHandle ent ) = 0;
  virtual unsigned long get_heap_size_bound( int max_recursions ) const = 0;

  virtual bool set_edge_size_evaluator( EdgeSizeEvaluator* );
  EdgeSizeEvaluator* get_edge_size_evaluator() { return this->edge_size_evaluator; }

  virtual bool set_output_functor( EntityRefinerOutputFunctor* func_obj );
  EntityRefinerOutputFunctor* get_output_functor() { return this->output_functor; }

  virtual bool set_minimum_number_of_subdivisions( int mn );
  int get_minimum_number_of_subdivisions() const { return this->minimum_number_of_subdivisions; }

  virtual bool set_maximum_number_of_subdivisions( int mx );
  int get_maximum_number_of_subdivisions() const { return this->maximum_number_of_subdivisions; }

protected:
  Interface* mesh_in;
  EdgeSizeEvaluator* edge_size_evaluator;
  EntityRefinerOutputFunctor* output_functor;
  int minimum_number_of_subdivisions;
  int maximum_number_of_subdivisions;
  std::vector<double> coord_heap;
  std::vector<double>::iterator current_coord;
  std::vector<char> tag_heap;
  std::vector<char>::iterator current_tag;

  void update_heap_size();
  void reset_heap_pointers();
  double* heap_coord_storage();
  void* heap_tag_storage();
};

} // namespace moab 

#endif // MOAB_ENTITY_REFINER_HPP