MeshReadWriteUtils.hpp

/*------------------------------------------------------------------------*/
/*                 Copyright 2010, 2011 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_IO_MESHREADWRITEUTILS_HPP
#define STK_IO_MESHREADWRITEUTILS_HPP
#include <string>
#include <vector>
#include <set>
#include <stk_util/parallel/Parallel.hpp>
#include <Ioss_PropertyManager.h>

namespace Ioss {
  class Region;
}

namespace stk {
  namespace mesh {
    class Part;
    class BulkData;
    class Selector;
    namespace fem {
      class FEMMetaData;
    }
  }
  namespace io {
    class MeshData {
      // Used to maintain state between the meta data and bulk data
      // portions of the mesh generation process for use cases.
    public:
      MeshData() : m_input_region(NULL), m_output_region(NULL),
       m_anded_selector(NULL)
      {}

      ~MeshData();

      Ioss::PropertyManager m_property_manager;

      Ioss::Region *m_input_region;
      Ioss::Region *m_output_region;

      stk::mesh::Selector *m_anded_selector;

    private:
      MeshData(const MeshData&); // Do not implement
      MeshData& operator=(const MeshData&); // Do not implement

    };

    void show_mesh_help();

    void create_input_mesh(const std::string &type,
         const std::string &filename,
         MPI_Comm comm,
         stk::mesh::fem::FEMMetaData &metadata,
         MeshData &mesh_data);

    void populate_bulk_data(stk::mesh::BulkData &bulk_data, stk::io::MeshData &mesh_data);

    void process_mesh_bulk_data(Ioss::Region *region, stk::mesh::BulkData &bulk_data);

    void define_input_fields(MeshData &mesh_data, stk::mesh::fem::FEMMetaData &meta_data);

    void process_input_request(MeshData &mesh_data, stk::mesh::BulkData &bulk, int step);

    void process_input_request(MeshData &mesh_data, stk::mesh::BulkData &bulk, double time);

    void input_mesh_fields(Ioss::Region *region, stk::mesh::BulkData &bulk_data, int step);

    void create_output_mesh(const std::string &filename,
          MPI_Comm comm,
          stk::mesh::BulkData &bulk_data,
          MeshData &mesh_data);

    void define_output_fields(const MeshData &mesh_data,
            const stk::mesh::fem::FEMMetaData &fem_meta,
            bool add_all_fields = false);

    int process_output_request(MeshData &mesh_data,
             stk::mesh::BulkData &bulk,
             double time,
                               const std::set<const stk::mesh::Part*> &exclude=std::set<const stk::mesh::Part*>());
    template <typename INT>
    void get_element_block_sizes(MeshData &mesh_data,
                                 std::vector<INT>& el_blocks);
  }
}
#endif