wrapper_i_hdf.h

Go to the documentation of this file.

//Copyright 2009 Thomas A Caswell
//tcaswell@uchicago.edu
//http://jfi.uchicago.edu/~tcaswell
//
//This program is free software; you can redistribute it and/or modify
//it under the terms of the GNU General Public License as published by
//the Free Software Foundation; either version 3 of the License, or (at
//your option) any later version.
//
//This program is distributed in the hope that it will be useful, but
//WITHOUT ANY WARRANTY; without even the implied warranty of
//MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
//General Public License for more details.
//
//You should have received a copy of the GNU General Public License
//along with this program; if not, see <http://www.gnu.org/licenses>.
//
//Additional permission under GNU GPL version 3 section 7
//
//If you modify this Program, or any covered work, by linking or
//combining it with MATLAB (or a modified version of that library),
//containing parts covered by the terms of MATLAB User License, the
//licensors of this Program grant you additional permission to convey
//the resulting work.

#ifndef WRAPPER_I_HDF
#define WRAPPER_I_HDF

#include "H5Cpp.h"

#include <string>
#include <vector>
#include <map>

#include "wrapper_i.h"
#include "data_map.h"
#include "part_def.h"

namespace utilities{
class Wrapper_i_hdf:public Wrapper_in{
private:
  std::string file_name_;
  
  static std::string format_name(int in);
  
  std::set<std::pair<utilities::D_TYPE,int> > data_types_;
  std::set<utilities::D_TYPE> data_types_set_;
  
  Data_map d_mapi_;
  Data_map d_mapf_;
  Data_map d_mapc_;
  
  std::vector<std::vector<int*> > data_i_;
  std::vector<std::vector<float*> > data_f_;
  std::vector<std::vector<std::complex<float>*> > data_c_;

  std::vector<int> frame_c_;
  unsigned int  frame_count_;

  unsigned int total_part_count_;
  
  bool two_d_data_;
  
  std::vector<float> frame_zdata_;

  unsigned int start_;
  
  bool locked_;
  

  void clean_data();
  
  bool priv_init(int f_count =0);
  void make_dtype_pairs(int comp_nuber);
  void make_dtype_set();
  
public:
  typedef enum DATA_DIMS
    {
      TWO_D = 0,
      THREE_D
    } DATA_DIMS;
    
    
  
  int                 get_value(int& out,
                                  int ind,D_TYPE type, int frame) const ;
  float               get_value(float& out,
                                int ind,D_TYPE type, int frame) const ;
  std::complex<float> get_value(std::complex<float>& out,
                                int ind,D_TYPE type, int frame) const ;


  std::set<D_TYPE>    get_data_types() const ;



  int get_num_entries(unsigned int frame) const ;
  int get_num_entries() const ;

  int get_num_frames() const ;

  bool contains_type(D_TYPE type) const ;

  

  Tuplef get_dims() const;


  
  ~Wrapper_i_hdf();

  Wrapper_i_hdf();
  
  
  bool initialize(const std::string & fname,
                const std::set<utilities::D_TYPE>& dtypes,
                int comp_number,
                unsigned int start=0,
                int frames =0);
  bool initialize(const std::string & fname,
                const std::set<utilities::D_TYPE> &dtypes,
                int comp_number,
                unsigned int start,
                int f_count,
                bool two_d_data);
  
  bool initialize(const std::string & fname,
                const std::set<std::pair<utilities::D_TYPE,int > > &dtypes,
                unsigned int start,
                int f_count,
                bool two_d_data);
  

  bool initialize(int f_count);
  
  
  float get_xy_scale() const;
  
  unsigned int get_start_offset() const {return start_;}
  
  bool set_file_name(const std::string & fname);
  
  bool add_dtype(utilities::D_TYPE dtype,int comp_key);
  

  bool set_twoD(bool twod_data);
  
  
};

}

#endif