GetEntities.hpp

/*------------------------------------------------------------------------*/
/*                 Copyright 2010 Sandia Corporation.                     */
/*  Under terms of Contract DE-AC04-94AL85000, there is a non-exclusive   */
/*  license for use of this work by or on behalf of the U.S. Government.  */
/*  Export of this program may require a license from the                 */
/*  United States Government.                                             */
/*------------------------------------------------------------------------*/


#ifndef stk_mesh_GetEntities_hpp
#define stk_mesh_GetEntities_hpp

//----------------------------------------------------------------------

#include <stk_mesh/base/Entity.hpp>
#include <stk_mesh/base/Selector.hpp>
#include <stk_mesh/base/BulkData.hpp>
#include <stk_mesh/base/Bucket.hpp>
#include <stk_mesh/base/Iterators.hpp>

//----------------------------------------------------------------------

namespace stk_classic {
namespace mesh {

//----------------------------------------------------------------------

void count_entities( const Selector & selector ,
                     const BulkData & mesh ,
                     std::vector<EntityRank> & count );

void get_entities( const BulkData & mesh , EntityRank entity_rank,
                   std::vector< Entity*> & entities);

// Note: Order will be different than above method because above method sorts
// TODO: remove underscore once API and STK_Mesh are in different namespaces
BucketVectorEntityIteratorRange get_entities( EntityRank entity_rank, const BulkData& mesh );

unsigned count_selected_entities( const Selector & selector ,
                                  const std::vector< Bucket * > & input_buckets );

void get_selected_entities( const Selector & selector ,
                            const std::vector< Bucket * > & input_buckets ,
                            std::vector< Entity * > & entities );

SelectedBucketRangeEntityIteratorRange get_selected_entities( const Selector & selector,
                                                              const AllBucketsRange& bucket_range );

} // namespace mesh
} // namespace stk_classic

//----------------------------------------------------------------------
//----------------------------------------------------------------------

#endif //  stk_mesh_GetEntities_hpp