IO_Fixture.hpp

#ifndef STK_IO_UTIL_IO_FIXTURE_HPP
#define STK_IO_UTIL_IO_FIXTURE_HPP

#include <stk_util/parallel/Parallel.hpp>

#include <stk_mesh/base/Types.hpp>
#include <stk_mesh/base/BulkData.hpp>

#include <stk_mesh/fem/FEMMetaData.hpp>

#include <stk_io/IossBridge.hpp>
#include <stk_io/MeshReadWriteUtils.hpp>

#include <stk_util/environment/ReportHandler.hpp>

#include <Teuchos_RCP.hpp>

#include <string>

namespace stk_classic {
namespace io {
namespace util {

class IO_Fixture
{
 public:

  typedef stk_classic::mesh::Field< double, stk_classic::mesh::Cartesian> coord_field_type;

  IO_Fixture(stk_classic::ParallelMachine comm);
  ~IO_Fixture();

  void create_output_mesh(
    const std::string & base_exodus_filename,
    bool  add_transient = true,
    bool  add_all_fields = false
                          );
  void add_timestep_to_output_mesh( double time );

  void set_meta_data( Teuchos::RCP<stk_classic::mesh::fem::FEMMetaData> arg_meta_data );

  void set_bulk_data( Teuchos::RCP<stk_classic::mesh::BulkData> arg_bulk_data );

  void set_input_ioss_region( Teuchos::RCP<Ioss::Region> input_region );

  void initialize_meta_data( const std::string & base_filename,
                             const std::string & type = "exodusii" );

  void initialize_bulk_data();

  stk_classic::mesh::fem::FEMMetaData & meta_data()
  {
    ThrowRequire( !Teuchos::is_null(m_fem_meta_data)) ;
    return *m_fem_meta_data;
  }

  stk_classic::mesh::BulkData & bulk_data()
  {
    ThrowRequire( !Teuchos::is_null(m_bulk_data)) ;
    return *m_bulk_data;
  }

  stk_classic::io::MeshData & mesh_data()
  {
    return m_mesh_data;
  }

  coord_field_type & get_coordinate_field()
  {
    coord_field_type * coord_field = meta_data().get_field<coord_field_type>("coordinates");
    ThrowRequire( coord_field != NULL);
    return * coord_field;
  }

  Teuchos::RCP<Ioss::Region> input_ioss_region()  { return m_ioss_input_region; }
  Teuchos::RCP<Ioss::Region> output_ioss_region() { return m_ioss_output_region; }
  void output_ioss_region(Teuchos::RCP<Ioss::Region>);

 private:
  stk_classic::ParallelMachine                       m_comm;
  Teuchos::RCP<stk_classic::mesh::fem::FEMMetaData>  m_fem_meta_data;
  Teuchos::RCP<stk_classic::mesh::BulkData>          m_bulk_data;

  Teuchos::RCP<Ioss::Region>                 m_ioss_input_region;
  Teuchos::RCP<Ioss::Region>                 m_ioss_output_region;

  std::string                                m_mesh_type;
  stk_classic::io::MeshData                          m_mesh_data;

  //disallow copy constructor and assignment operator
  IO_Fixture( const IO_Fixture & );
  IO_Fixture & operator = ( const IO_Fixture & );
};

} // namespace util
} // namespace io
} // namespace stk_classic

#endif //STK_IO_UTIL_IO_FIXTURE_HPP