MeshRefiner.cpp

Go to the documentation of this file.

#include "MeshRefiner.hpp"

#include "EdgeSizeEvaluator.hpp"
#include "EntityRefiner.hpp"
#include "moab/Interface.hpp"
#include "RefinerTagManager.hpp"
#include "MeshOutputFunctor.hpp"

#ifdef USE_MPI
#include "moab/ParallelComm.hpp"
#include "moab_mpi.h"
#else // USE_MPI
typedef int MPI_Comm;
#endif // USE_MPI

namespace moab {

MeshRefiner::MeshRefiner( Interface* imesh, Interface* omesh )
{  
  this->mesh_in = imesh;
  this->mesh_out = omesh;
  this->tag_manager = new RefinerTagManager( this->mesh_in, this->mesh_out );
  this->output_functor = new MeshOutputFunctor( this->tag_manager );
  this->entity_refiner = 0;
  this->comm = ParallelComm::get_pcomm( this->mesh_out, 0 );
}

MeshRefiner::~MeshRefiner()
{
  delete this->tag_manager;
  delete this->output_functor;
  if ( this->entity_refiner )
    delete this->entity_refiner;
}

bool MeshRefiner::set_entity_refiner( EntityRefiner* er )
{
  if ( ! er || er == this->entity_refiner )
    return false;

  this->entity_refiner = er;
  return true;
}

void MeshRefiner::reset_vertex_tags()
{
  this->tag_manager->reset_vertex_tags();
}

int MeshRefiner::add_vertex_tag( Tag tag_handle )
{
  return this->tag_manager->add_vertex_tag( tag_handle );
}

struct MeshRefinerIterator {
  Range subset;
  EntityHandle destination_set;
};

bool MeshRefiner::refine( Range& range )
{
  this->tag_manager->create_output_tags();
  if ( ! this->entity_refiner->prepare( this->tag_manager, this->output_functor ) )
    { // Oops, perhaps the edge_size_evaluator was not set?
    return false;
    }

  MeshRefinerIterator entry;
  std::vector<MeshRefinerIterator> work;

  entry.subset = range;
  entry.destination_set = 0;
  work.push_back( entry );

  while ( ! work.empty() )
    {
    entry = work.back();
    work.pop_back();
    this->output_functor->destination_set = entry.destination_set;
    for ( Range::const_iterator it = entry.subset.begin(); it != entry.subset.end(); ++ it )
      {
      EntityType etyp = this->mesh_in->type_from_handle( *it );
      if ( etyp == MBENTITYSET )
        {
        Range set_ents;
        if ( this->mesh_in->get_entities_by_handle( *it, set_ents, false ) == MB_SUCCESS )
          {
          // Create a matching set on the output mesh.
          MeshRefinerIterator set_work;
          unsigned int set_work_opts;
          this->mesh_in->get_meshset_options( *it, set_work_opts );
          this->mesh_out->create_meshset( set_work_opts, set_work.destination_set );
          set_work.subset = set_ents;
          work.push_back( set_work );
          // Copy any per-element tag values the user has requested to the output set.
          this->tag_manager->set_element_tags_from_ent( *it );
          this->tag_manager->assign_element_tags( set_work.destination_set );
          // Copy the global ID to the new set (assuming it exists).
          this->tag_manager->copy_gid( *it, set_work.destination_set );
          }
        }
      else
        {
        this->tag_manager->set_element_tags_from_ent( *it );
        this->tag_manager->set_element_procs_from_ent( *it );
        this->entity_refiner->refine_entity( etyp, *it );
        }
      }
    }
  this->output_functor->assign_global_ids( this->comm );

  return true;
}

} // namespace moab