ReadHDF5Dataset.hpp

Go to the documentation of this file.

#ifndef moab_READ_HDF5DATASET_HPP
#define moab_READ_HDF5DATASET_HPP

#ifdef USE_MPI
# include <moab_mpi.h>
#endif

#include <stdlib.h> // for size_t
#include <H5Ipublic.h>
#include <H5Spublic.h>

#include "moab/Range.hpp"
#include <vector>

namespace moab {

class ReadHDF5Dataset 
{
public:

#ifdef USE_MPI
  typedef MPI_Comm Comm;
#else
  typedef int Comm;
#endif

  class Exception { public: int line_no; Exception(int l) : line_no(l) {} };
  
  ReadHDF5Dataset( const char* debug_desc,
                   hid_t data_set_handle,
                   bool parallel,
                   const Comm* communicator = 0,
                   bool close_data_set_on_destruct = true );
  
  ReadHDF5Dataset( const char* debug_desc,
                   bool parallel,
                   const Comm* communicator = 0 );
  void init( hid_t data_set_handle, bool close_data_set_on_destruct = true );
  
  bool will_close_data_set() const { return closeDataSet; }
  void close_data_set_on_destruct( bool val ) { closeDataSet = val; }
  
  ~ReadHDF5Dataset();
  
  
  void set_file_ids( const Range& file_ids, 
                     EntityHandle start_id,
                     hsize_t row_cout,
                     hid_t data_type );
  
  void set_all_file_ids( hsize_t row_count, hid_t data_type );
  
  bool done() const { return (currOffset == rangeEnd) && (readCount == 0); }
  
  void read( void* buffer, size_t& rows_read );
  
  Range::const_iterator next_file_id() const { return currOffset; }
  
  void null_read();
  
  unsigned columns() const;
  void set_column( unsigned c );
  
  unsigned long get_read_count() const { return readCount; }
  const char* get_debug_desc() const { return mpeDesc.c_str(); }

  static void set_hyperslab_selection_limit( size_t val )
    { hyperslabSelectionLimit = val; }
  static void default_hyperslab_selection_limit();

  static void append_hyperslabs() { hyperslabSelectOp = H5S_SELECT_APPEND; }
  static void or_hyperslabs() { hyperslabSelectOp = H5S_SELECT_OR; }
private:

  Range::const_iterator next_end( Range::const_iterator iter );

  Range internalRange; 

  bool closeDataSet; 
  hsize_t dataSetOffset[64], dataSetCount[64];
  hid_t dataSet;       
  hid_t dataSpace;     
  hid_t dataType;      
  hid_t fileType;      
  hid_t ioProp;        
  int dataSpaceRank;   
  hsize_t rowsInTable; 
  bool doConversion;   
  bool nativeParallel; 
  
  hsize_t readCount;  
  hsize_t bufferSize; 
  const Comm* mpiComm;
  
  Range::const_iterator currOffset, rangeEnd;
  EntityHandle startID;
  
  static bool haveMPEEvents;
  static std::pair<int,int> mpeReadEvent;
  static std::pair<int,int> mpeReduceEvent;
  std::string mpeDesc;
  
  static size_t hyperslabSelectionLimit;
  static H5S_seloper_t hyperslabSelectOp;
}; 



} // namespace moab

#endif // moab_READ_HDF5DATASET_HPP