hdf_pete.h

Go to the documentation of this file.

//////////////////////////////////////////////////////////////////////////////////////
// This file is distributed under the University of Illinois/NCSA Open Source License.
// See LICENSE file in top directory for details.
//
// Copyright (c) 2016 Jeongnim Kim and QMCPACK developers.
//
// File developed by: Jeremy McMinnis, jmcminis@gmail.com, University of Illinois at Urbana-Champaign
//
// File created by: Jeongnim Kim, jeongnim.kim@gmail.com, University of Illinois at Urbana-Champaign
//////////////////////////////////////////////////////////////////////////////////////


#ifndef QMCPLUSPLUS_HDF_PETE_TRAITS_H
#define QMCPLUSPLUS_HDF_PETE_TRAITS_H

#include "hdf_dataproxy.h"
#include "OhmmsPETE/OhmmsVector.h"
#include "OhmmsPETE/OhmmsMatrix.h"
#include "OhmmsPETE/OhmmsArray.h"

namespace qmcplusplus
{
/** specialization for Vector<T>
 *
 * Used with any T with a proper h5_space_type, e.g., intrinsic, TinyVector<T,D>, Tensor<T,D>
 */
template<typename T>
struct h5data_proxy<Vector<T>> : public h5_space_type<T, 1>
{
  using FileSpace = h5_space_type<T, 1>;
  using FileSpace::dims;
  using FileSpace::get_address;
  using data_type = Vector<T>;

  inline h5data_proxy(const data_type& a) { dims[0] = a.size(); }

  inline bool read(data_type& ref, hid_t grp, const std::string& aname, hid_t xfer_plist = H5P_DEFAULT)
  {
    if (!checkShapeConsistency<T>(grp, aname, FileSpace::rank, dims))
      ref.resize(dims[0]);
    return h5d_read(grp, aname, get_address(ref.data()), xfer_plist);
  }

  inline bool write(const data_type& ref, hid_t grp, const std::string& aname, hid_t xfer_plist = H5P_DEFAULT) const
  {
    return h5d_write(grp, aname.c_str(), FileSpace::rank, dims, get_address(ref.data()), xfer_plist);
  }
};


template<typename T>
struct h5data_proxy<Matrix<T>> : public h5_space_type<T, 2>
{
  using FileSpace = h5_space_type<T, 2>;
  using FileSpace::dims;
  using FileSpace::get_address;
  using data_type = Matrix<T>;

  inline h5data_proxy(const data_type& a)
  {
    dims[0] = a.rows();
    dims[1] = a.cols();
  }

  inline bool read(data_type& ref, hid_t grp, const std::string& aname, hid_t xfer_plist = H5P_DEFAULT)
  {
    if (!checkShapeConsistency<T>(grp, aname, FileSpace::rank, dims))
      ref.resize(dims[0], dims[1]);
    return h5d_read(grp, aname, get_address(ref.data()), xfer_plist);
  }

  inline bool write(const data_type& ref, hid_t grp, const std::string& aname, hid_t xfer_plist = H5P_DEFAULT) const
  {
    return h5d_write(grp, aname.c_str(), FileSpace::rank, dims, get_address(ref.data()), xfer_plist);
  }
};


template<typename T, unsigned D>
struct h5data_proxy<Array<T, D>> : public h5_space_type<T, D>
{
  using FileSpace = h5_space_type<T, D>;
  using FileSpace::dims;
  using FileSpace::get_address;
  using data_type = Array<T, D>;

  inline h5data_proxy(const data_type& a)
  {
    for (int i = 0; i < D; ++i)
      dims[i] = a.size(i);
  }

  inline bool read(data_type& ref, hid_t grp, const std::string& aname, hid_t xfer_plist = H5P_DEFAULT)
  {
    if (!checkShapeConsistency<T>(grp, aname, FileSpace::rank, dims))
    {
      std::array<size_t, D> dims_array;
      std::copy(std::cbegin(dims), std::cend(dims), dims_array.begin());
      ref.resize(dims_array);
    }
    return h5d_read(grp, aname, get_address(ref.data()), xfer_plist);
  }

  inline bool write(const data_type& ref, hid_t grp, const std::string& aname, hid_t xfer_plist = H5P_DEFAULT) const
  {
    return h5d_write(grp, aname.c_str(), FileSpace::rank, dims, get_address(ref.data()), xfer_plist);
  }
};

} // namespace qmcplusplus
#endif