SelectorFixture.cpp

/*------------------------------------------------------------------------*/
/*                 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.                                             */
/*------------------------------------------------------------------------*/


#include <sstream>
#include <stk_mesh/fixtures/SelectorFixture.hpp>

namespace stk {
namespace mesh {
namespace fixtures {

  SelectorFixture::~SelectorFixture() {}

  SelectorFixture::SelectorFixture()
    : m_MetaData( std::vector<std::string>(1, std::string("MyEntityRank")) )
    , m_BulkData( m_MetaData , MPI_COMM_WORLD )
    , m_partA( m_MetaData.declare_part( "PartA" , 0 ) )
    , m_partB( m_MetaData.declare_part( "PartB" , 0 ) )
    , m_partC( m_MetaData.declare_part( "PartC" , 0 ) )
    , m_partD( m_MetaData.declare_part( "PartD" , 0 ) )
    , m_entity1( NULL )
    , m_entity2( NULL )
    , m_entity3( NULL )
    , m_entity4( NULL )
    , m_entity5( NULL )
  {
    m_MetaData.commit();

    m_BulkData.modification_begin();

    const unsigned entity_count = 5 ;

    stk::mesh::EntityRank ent_type = 0; // rank

    // Create Entities and assign to parts:
    stk::mesh::EntityId ent_id =
      1 + entity_count * m_BulkData.parallel_rank(); // Unique ID

    std::vector<stk::mesh::Part*> partMembership;

    // Entity1 is contained in PartA
    partMembership.clear();
    partMembership.push_back( & m_partA );
    m_entity1 = & m_BulkData.declare_entity(ent_type, ent_id, partMembership);
    ++ent_id;

    // Entity2 is contained in PartA and PartB
    partMembership.clear();
    partMembership.push_back( & m_partA );
    partMembership.push_back( & m_partB );
    m_entity2 = & m_BulkData.declare_entity(ent_type, ent_id, partMembership);
    ++ent_id;

    // Entity3 is contained in PartB and PartC
    partMembership.clear();
    partMembership.push_back( & m_partB );
    partMembership.push_back( & m_partC );
    m_entity3 = & m_BulkData.declare_entity(ent_type, ent_id, partMembership);
    ++ent_id;

    // Entity4 is contained in PartC
    partMembership.clear();
    partMembership.push_back( & m_partC );
    m_entity4 = & m_BulkData.declare_entity(ent_type, ent_id, partMembership);
    ++ent_id;

    // Entity5 is not contained in any Part
    partMembership.clear();
    m_entity5 = & m_BulkData.declare_entity(ent_type, ent_id, partMembership);

    m_BulkData.modification_end();
  }

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

  VariableSelectorFixture::~VariableSelectorFixture() {}

  VariableSelectorFixture::VariableSelectorFixture(int NumParts)
    : m_MetaData( std::vector<std::string>(1, std::string("MyEntityRank")) )
    , m_BulkData( m_MetaData , MPI_COMM_WORLD )
    , m_declared_part_vector()
  {
    // Create Parts and commit:
    std::string myPartName;
    stk::mesh::EntityRank myRank = 0;

    std::string partName = "Part_";
    for (int part_i=0 ; part_i<NumParts; ++part_i) {
      std::ostringstream localPartName(partName);
      localPartName << part_i;
      stk::mesh::Part * part =
        & m_MetaData.declare_part(localPartName.str(),myRank);
      m_declared_part_vector.push_back( part );
    }

    m_MetaData.commit();

    // Create Entities and assign to parts:

    m_BulkData.modification_begin();

    stk::mesh::EntityRank ent_type = 0; // rank
    stk::mesh::EntityId ent_id =
      1 + NumParts * m_BulkData.parallel_rank(); // Unique ID

    for (int part_i = 0 ; part_i < NumParts ; ++part_i) {
      std::vector<stk::mesh::Part*> partMembership;
      partMembership.push_back(m_declared_part_vector[part_i]);
      stk::mesh::Entity * e =
        & m_BulkData.declare_entity(ent_type, ent_id, partMembership);
      m_entities.push_back( e );
      ++ent_id;
    }

    m_BulkData.modification_end();
  }

} // fixtures
} // mesh
} // stk