Commit b68a6c5f authored by gbalduzz's avatar gbalduzz
Browse files

Merge remote-tracking branch 'origin/master' into single_precision_ctaux

parents e1f93abd f269ab50

cmake/dca_config.cmake

+28 −12
Original line number Diff line number Diff line 
################################################################################

# Author: Urs R. Haehner (haehneru@itp.phys.ethz.ch)

#         Giovanni Badlduzzi (gbalduzz@itp.phys.ethz.ch)

#

# Build options for DCA++.
@@ -53,8 +54,8 @@ endif() 
# TODO: Add more point groups and lattices.



# Point group

set(DCA_POINT_GROUP "D4" CACHE STRING "Point group symmetry, options are: C6 | D4.")

set_property(CACHE DCA_POINT_GROUP PROPERTY STRINGS C6 D4)

set(DCA_POINT_GROUP "D4" CACHE STRING "Point group symmetry, options are: C6 | D4 | no_symmetry<2>.")

set_property(CACHE DCA_POINT_GROUP PROPERTY STRINGS C6 D4 no_symmetry<2>)



if (DCA_POINT_GROUP STREQUAL "C6")

  set(DCA_POINT_GROUP_INCLUDE
@@ -63,14 +64,18 @@ if (DCA_POINT_GROUP STREQUAL "C6") 
elseif (DCA_POINT_GROUP STREQUAL "D4")

  set(DCA_POINT_GROUP_INCLUDE "dca/phys/domains/cluster/symmetries/point_groups/2d/2d_square.hpp")



elseif (DCA_POINT_GROUP STREQUAL "no_symmetry<2>")

  set(DCA_POINT_GROUP_INCLUDE "dca/phys/domains/cluster/symmetries/point_groups/no_symmetry.hpp")



else()

  message(FATAL_ERROR "Please set DCA_POINT_GROUP to a valid option: C6 | D4.")

endif()



# Lattice type

set(DCA_LATTICE "square" CACHE STRING

    "Lattice type, options are: bilayer | square | triangular | twoband_chain | singleband_chain.")

set_property(CACHE DCA_LATTICE PROPERTY STRINGS bilayer square triangular twoband_chain singleband_chain)

set(DCA_LATTICE "square" CACHE STRING "Lattice type, options are: bilayer | square | triangular |

    hund | twoband_Cu | threeband | FeAs.")

set_property(CACHE DCA_LATTICE PROPERTY STRINGS bilayer square triangular hund twoband_Cu threeband

             FeAs)



if (DCA_LATTICE STREQUAL "bilayer")

  set(DCA_LATTICE_TYPE dca::phys::models::bilayer_lattice<PointGroup>)
@@ -86,19 +91,30 @@ elseif (DCA_LATTICE STREQUAL "triangular") 
  set(DCA_LATTICE_TYPE dca::phys::models::triangular_lattice<PointGroup>)

  set(DCA_LATTICE_INCLUDE

    "dca/phys/models/analytic_hamiltonians/triangular_lattice.hpp")

elseif (DCA_LATTICE STREQUAL "hund")

  set(DCA_LATTICE_TYPE dca::phys::models::HundLattice<PointGroup>)



elseif (DCA_LATTICE STREQUAL "threeband")

  set(DCA_LATTICE_TYPE dca::phys::models::ThreebandHubbard<PointGroup>)

  set(DCA_LATTICE_INCLUDE

    "dca/phys/models/analytic_hamiltonians/threeband_hubbard.hpp")



elseif (DCA_LATTICE STREQUAL "twoband_chain")

  set(DCA_LATTICE_TYPE dca::phys::models::twoband_chain<dca::phys::domains::no_symmetry<1>>)

  set(DCA_LATTICE_INCLUDE

      "dca/phys/models/analytic_hamiltonians/twoband_chain.hpp")



elseif (DCA_LATTICE STREQUAL "singleband_chain")

  set(DCA_LATTICE_TYPE dca::phys::models::singleband_chain<dca::phys::domains::no_symmetry<1>>)

      "dca/phys/models/analytic_hamiltonians/hund_lattice.hpp")

elseif (DCA_LATTICE STREQUAL "FeAs")

  set(DCA_LATTICE_TYPE dca::phys::models::FeAsLattice<PointGroup>)

  set(DCA_LATTICE_INCLUDE

      "dca/phys/models/analytic_hamiltonians/singleband_chain.hpp")

      "dca/phys/models/analytic_hamiltonians/fe_as_lattice.hpp")

elseif (DCA_LATTICE STREQUAL "twoband_Cu")

  set(DCA_LATTICE_TYPE dca::phys::models::TwoBandCu<PointGroup>)

  set(DCA_LATTICE_INCLUDE

      "dca/phys/models/analytic_hamiltonians/twoband_Cu.hpp")



else()

  message(FATAL_ERROR

          "Please set DCA_LATTICE to a valid option: bilayer | square | triangular | twoband_chain | singleband_chain.")

  message(FATAL_ERROR "Please set DCA_LATTICE to a valid option: bilayer | square | triangular |

          hund | twoband_Cu | threeband | FeAs.")

endif()



# Model type

include/dca/function/domains/dmn_0.hpp

+3 −3
Original line number Diff line number Diff line
@@ -37,7 +37,7 @@ public: 
    return parameters::get_size();

  }



  static std::vector<element_type>& get_elements() {

  static const std::vector<element_type>& get_elements() {

    return parameters::get_elements();

  }
@@ -75,7 +75,7 @@ void dmn_0<parameters>::reset() { 
  dmn_0::initialize();

}



}  // func

}  // dca

}  // namespace func

}  // namespace dca



#endif  // DCA_FUNCTION_DOMAINS_DMN_0_HPP

include/dca/io/hdf5/hdf5_reader.hpp

+143 −158
Original line number Diff line number Diff line
@@ -15,7 +15,6 @@ 
#include <complex>

#include <string>

#include <vector>

#include <type_traits>



#include "H5Cpp.h"
@@ -35,7 +34,7 @@ public: 
  typedef H5::H5File file_type;



  // In: verbose. If true, the reader outputs a short log whenever it is executed.

  HDF5Reader(bool verbose = true) : my_file(NULL), my_paths(0), verbose_(verbose) {}

  HDF5Reader(bool verbose = true) : verbose_(verbose) {}



  ~HDF5Reader();
@@ -50,10 +49,10 @@ public: 
  void close_file();



  void open_group(std::string name) {

    my_paths.push_back(name);

    paths_.push_back(name);

  }

  void close_group() {

    my_paths.pop_back();

    paths_.pop_back();

  }



  std::string get_path();
@@ -63,18 +62,21 @@ public: 


  // `execute` returns true if the object is read correctly.



  template <typename scalartype>

  bool execute(std::string name, scalartype& value);

  template <typename Scalartype>

  bool execute(const std::string& name, Scalartype& value);



  template <typename scalar_type>

  bool execute(std::string name, std::vector<scalar_type>& value);

  template <typename Scalar>

  bool execute(const std::string& name, std::vector<Scalar>& value);



  template <typename scalar_type>

  bool execute(std::string name, std::vector<std::complex<scalar_type>>& value);

  template <typename Scalar>

  bool execute(const std::string& name, std::vector<std::vector<Scalar>>& value);



  bool execute(std::string name, std::string& value);

  template <typename Scalar, std::size_t n>

  bool execute(const std::string& name, std::vector<std::array<Scalar, n>>& value);



  bool execute(std::string name, std::vector<std::string>& value);

  bool execute(const std::string& name, std::string& value);



  bool execute(const std::string& name, std::vector<std::string>& value);



  // TODO: Remove? (only thing that depends on domains.hpp)

  template <typename domain_type>
@@ -82,30 +84,33 @@ public: 
    return false;

  }



  template <typename scalartype, typename domain_type>

  bool execute(func::function<scalartype, domain_type>& f);

  template <typename Scalartype, typename domain_type>

  bool execute(func::function<Scalartype, domain_type>& f);



  template <typename scalartype, typename domain_type>

  bool execute(std::string name, func::function<scalartype, domain_type>& f);

  template <typename Scalartype, typename domain_type>

  bool execute(const std::string& name, func::function<Scalartype, domain_type>& f);



  template <typename scalar_type>

  bool execute(std::string name, dca::linalg::Vector<scalar_type, dca::linalg::CPU>& A);

  template <typename Scalar>

  bool execute(const std::string& name, dca::linalg::Vector<Scalar, dca::linalg::CPU>& A);



  template <typename scalar_type>

  bool execute(std::string name, dca::linalg::Vector<std::complex<scalar_type>, dca::linalg::CPU>& A);

  template <typename Scalar>

  bool execute(const std::string& name, dca::linalg::Matrix<Scalar, dca::linalg::CPU>& A);



  template <typename scalar_type>

  bool execute(std::string name, dca::linalg::Matrix<scalar_type, dca::linalg::CPU>& A);

  template <typename Scalar>

  bool execute(dca::linalg::Matrix<Scalar, dca::linalg::CPU>& A);



  bool execute(std::string name, io::Buffer& buff) {

  bool execute(const std::string& name, io::Buffer& buff) {

    return execute(name, static_cast<io::Buffer::Container&>(buff));

  }



private:

  bool fexists(const char* filename);

  bool exists(const std::string& name) const;



  void read(const std::string& name, H5::DataType type, void* data) const;

  std::vector<hsize_t> readSize(const std::string& name) const;



  H5::H5File* my_file;

  std::vector<std::string> my_paths;

  std::unique_ptr<H5::H5File> file_;

  std::vector<std::string> paths_;



  bool verbose_;

};
@@ -118,188 +123,168 @@ void HDF5Reader::from_file(arbitrary_struct_t& arbitrary_struct, std::string fil 
  reader_obj.close_file();

}



template <typename scalar_type>

bool HDF5Reader::execute(std::string name, scalar_type& value) {

template <typename Scalar>

bool HDF5Reader::execute(const std::string& name, Scalar& value) {

  std::string full_name = get_path() + "/" + name;



  try {

    H5::DataSet dataset = my_file->openDataSet(full_name.c_str());



    H5::DataSpace dataspace = dataset.getSpace();



    H5Dread(dataset.getId(), HDF5_TYPE<scalar_type>::get(), dataspace.getId(), H5S_ALL, H5P_DEFAULT,

            &value);



    return true;

  }

  catch (...) {

    std::cout << "\n\n\t the variable (" + name + ") does not exist in path : " + get_path() +

                     "\n\n";

  if (!exists(full_name)) {

    return false;

  }



  read(full_name, HDF5_TYPE<Scalar>::get_PredType(), &value);

  return true;

}



template <typename scalar_type>

bool HDF5Reader::execute(std::string name, std::vector<scalar_type>& value) {

template <typename Scalar>

bool HDF5Reader::execute(const std::string& name, std::vector<Scalar>& value) {

  std::string full_name = get_path() + "/" + name;



  try {

    H5::DataSet dataset = my_file->openDataSet(full_name.c_str());



    value.resize(dataset.getInMemDataSize() / sizeof(scalar_type));

  if (!exists(full_name)) {

    return false;

  }



    H5::DataSpace dataspace = dataset.getSpace();

  auto dims = readSize(full_name);

  assert(dims.size() == 1);

  value.resize(dims.at(0));



    H5Dread(dataset.getId(), HDF5_TYPE<scalar_type>::get(), dataspace.getId(), H5S_ALL, H5P_DEFAULT,

            &value[0]);

  read(full_name, HDF5_TYPE<Scalar>::get_PredType(), value.data());

  return true;

}

  catch (...) {

    std::cout << "\n\n\t the variable (" + name + ") does not exist in path : " + get_path() +

                     "\n\n";



template <typename Scalar>

bool HDF5Reader::execute(const std::string& name, std::vector<std::vector<Scalar>>& value) {

  std::string full_name = get_path() + "/" + name;

  if (!exists(full_name)) {

    return false;

  }

}



template <typename scalar_type>

bool HDF5Reader::execute(std::string name, std::vector<std::complex<scalar_type>>& value) {

  std::string full_name = get_path() + "/" + name;

  auto size = readSize(full_name)[0];

  const auto type = H5::VarLenType(HDF5_TYPE<Scalar>::get_PredType());



  try {

    H5::DataSet dataset = my_file->openDataSet(full_name.c_str());

  std::vector<hvl_t> data(size);



  H5::DataSet dataset = file_->openDataSet(name.c_str());

  dataset.read(data.data(), type);



    value.resize(dataset.getInMemDataSize() / sizeof(std::complex<scalar_type>));

  value.resize(size);

  for (int i = 0; i < size; ++i) {

    value[i].resize(data[i].len);

    std::copy_n(static_cast<Scalar*>(data[i].p), data[i].len, value[i].data());

  }



    H5::DataSpace dataspace = dataset.getSpace();

  dataset.vlenReclaim(data.data(), type, dataset.getSpace());



    H5Dread(dataset.getId(), HDF5_TYPE<scalar_type>::get(), dataspace.getId(), H5S_ALL, H5P_DEFAULT,

            &value[0]);

  return true;

}

  catch (...) {

    std::cout << "\n\n\t the variable (" + name + ") does not exist in path : " + get_path() +

                     "\n\n";



template <typename Scalar, std::size_t n>

bool HDF5Reader::execute(const std::string& name, std::vector<std::array<Scalar, n>>& value) {

  std::string full_name = get_path() + "/" + name;

  if (!exists(full_name)) {

    return false;

  }

}



template <typename scalartype, typename domain_type>

bool HDF5Reader::execute(func::function<scalartype, domain_type>& f) {

  return execute(f.get_name(), f);

  auto dims = readSize(full_name);

  assert(dims.size() == 2);

  if (dims.at(1) != n) {

    throw(std::length_error("Wrong array size"));

  }



template <typename scalartype, typename domain_type>

bool HDF5Reader::execute(std::string name, func::function<scalartype, domain_type>& f) {

  std::cout << "\n\tstart reading function : " << name;

  open_group(name);

  bool success = true;

  value.resize(dims[0]);

  read(full_name, HDF5_TYPE<Scalar>::get_PredType(), value.data());



  try {

    std::string full_name = get_path() + "/data";

  return true;

}



    H5::DataSet dataset = my_file->openDataSet(full_name.c_str());

template <typename Scalartype, typename domain_type>

bool HDF5Reader::execute(func::function<Scalartype, domain_type>& f) {

  return execute(f.get_name(), f);

}



    H5::DataSpace dataspace = dataset.getSpace();

template <typename Scalartype, typename domain_type>

bool HDF5Reader::execute(const std::string& name, func::function<Scalartype, domain_type>& f) {

  std::string full_name = get_path() + "/" + name;



    H5Dread(dataset.getId(), HDF5_TYPE<scalartype>::get(), dataspace.getId(), H5S_ALL, H5P_DEFAULT,

            &f(0));

  }

  catch (const H5::FileIException& err) {

  if (!exists(full_name)) {

    std::cout << "\n\n\t the function (" + name + ") does not exist in path : " + get_path() +

                     "\n\n";

    success = false;

    return false;

  }



  close_group();

  return success;

}

  std::cout << "\n\tstart reading function : " << name;



template <typename scalar_type>

bool HDF5Reader::execute(std::string name, dca::linalg::Vector<scalar_type, dca::linalg::CPU>& V) {

  open_group(name);

  bool success = true;

  H5::DataSet dataset = file_->openDataSet(full_name.c_str());



  try {

    std::string full_name = get_path() + "/data";



    H5::DataSet dataset = my_file->openDataSet(full_name.c_str());



    V.resize(dataset.getInMemDataSize() / sizeof(scalar_type));



    H5::DataSpace dataspace = dataset.getSpace();



    H5Dread(dataset.getId(), HDF5_TYPE<scalar_type>::get(), dataspace.getId(), H5S_ALL, H5P_DEFAULT,

            &V[0]);

    // Read sizes.

    std::vector<hsize_t> dims;

    auto domain_attribute = dataset.openAttribute("domain-sizes");

    hsize_t n_dims;

    domain_attribute.getSpace().getSimpleExtentDims(&n_dims);

    dims.resize(n_dims);

    domain_attribute.read(HDF5_TYPE<hsize_t>::get_PredType(), dims.data());



    // Check sizes.

    if (dims.size() != f.signature())

      throw(std::length_error("The number of domains is different"));

    for (int i = 0; i < f.signature(); ++i) {

      if (dims[i] != f[i])

        throw(std::length_error("The size of domain " + std::to_string(i) + " is different"));

    }

  catch (const H5::FileIException& err) {

    std::cout << "\n\n\t the vector (" + name + ") does not exist in path : " + get_path() + "\n\n";

    success = false;

  }



  close_group();

  return success;

  catch (H5::Exception& err) {

    std::cerr << "Could not perform a size check on the function  " << name << std::endl;

  }



template <typename scalar_type>

bool HDF5Reader::execute(std::string name,

                         dca::linalg::Vector<std::complex<scalar_type>, dca::linalg::CPU>& V) {

  open_group(name);

  bool success = true;



  try {

    std::string full_name = get_path() + "/data";



    H5::DataSet dataset = my_file->openDataSet(full_name.c_str());



    V.resize(dataset.getInMemDataSize() / sizeof(std::complex<scalar_type>));

  read(full_name, HDF5_TYPE<Scalartype>::get_PredType(), f.values());



    H5::DataSpace dataspace = dataset.getSpace();



    H5Dread(dataset.getId(), HDF5_TYPE<scalar_type>::get(), dataspace.getId(), H5S_ALL, H5P_DEFAULT,

            &V[0]);

  }

  catch (const H5::FileIException& err) {

    std::cout << "\n\n\t the vector (" + name + ") does not exist in path : " + get_path() + "\n\n";

    success = false;

  return true;

}



  close_group();

  return success;

template <typename Scalar>

bool HDF5Reader::execute(const std::string& name, dca::linalg::Vector<Scalar, dca::linalg::CPU>& V) {

  std::string full_name = get_path() + "/" + name;

  if (!exists(full_name)) {

    return false;

  }



template <typename scalar_type>

bool HDF5Reader::execute(std::string name, dca::linalg::Matrix<scalar_type, dca::linalg::CPU>& A) {

  open_group(name);

  bool success = true;



  try {std::vector<int> size(2);

    execute("current-size", size);

    A.resizeNoCopy(std::make_pair(size[0], size[1]));

  auto dims = readSize(full_name);

  assert(dims.size() == 1);

  V.resize(dims.at(0));



    std::vector<scalar_type> a_compressed(size[0] * size[1]);

  read(full_name, HDF5_TYPE<Scalar>::get_PredType(), V.ptr());



    std::string full_name = get_path() + "/data";

  return true;

}



    H5::DataSet dataset = my_file->openDataSet(full_name.c_str());

template <typename Scalar>

bool HDF5Reader::execute(const std::string& name, dca::linalg::Matrix<Scalar, dca::linalg::CPU>& A) {

  std::string full_name = get_path() + "/" + name;

  if (!exists(full_name)) {

    return false;

  }



    H5::DataSpace dataspace = dataset.getSpace();

  auto dims = readSize(full_name);

  assert(dims.size() == 2);



    H5Dread(dataset.getId(), HDF5_TYPE<scalar_type>::get(), dataspace.getId(), H5S_ALL, H5P_DEFAULT,

            a_compressed.data());

  std::vector<Scalar> linearized(dims[0] * dims[1]);

  read(full_name, HDF5_TYPE<Scalar>::get_PredType(), linearized.data());



    unsigned index = 0;

  // HDF5 is column major, while Matrix is row major.

  A.resizeNoCopy(std::make_pair(dims[0], dims[1]));

  for (int i = 0, linindex = 0; i < A.nrRows(); ++i) {

    for (int j = 0; j < A.nrCols(); ++j)

      for (int i = 0; i < A.nrRows(); ++i)

        A(i, j) = a_compressed[index++];

      A(i, j) = linearized[linindex++];

  }

  catch (const H5::FileIException& err) {

    std::cout << "\n\n\t the function (" + name + ") does not exist in path : " + get_path() +

                     "\n\n";

    success = false;



  A.set_name(name);



  return true;

}



  close_group();

  return success;

template <typename Scalar>

bool HDF5Reader::execute(dca::linalg::Matrix<Scalar, dca::linalg::CPU>& A) {

  return execute(A.get_name(), A);

}



}  // namespace io

include/dca/io/hdf5/hdf5_types.hpp

+17 −25
Original line number Diff line number Diff line
@@ -6,6 +6,7 @@ 
// See CITATION.md for citation guidelines, if DCA++ is used for scientific publications.

//

// Author: Peter Staar (taa@zurich.ibm.com)

//         Giovanni Balduzzi (gbalduzz@itp.phys.ethz.ch)

//

// HDF5 types.
@@ -143,42 +144,33 @@ public: 
};



template <>

class HDF5_TYPE<std::complex<float>> {

class HDF5_TYPE<unsigned char> {

public:

  static hid_t get() {

    return H5T_NATIVE_FLOAT;

    return H5T_NATIVE_UCHAR;

  }



  static H5::PredType get_PredType() {

    return H5::PredType::NATIVE_FLOAT;

    return H5::PredType::NATIVE_UCHAR;

  }

};



template <>

class HDF5_TYPE<std::complex<double>> {

template <class T>

class HDF5_TYPE<std::complex<T>> {

public:

  static hid_t get() {

    return H5T_NATIVE_DOUBLE;

  }



  static H5::PredType get_PredType() {

    return H5::PredType::NATIVE_DOUBLE;

  }

  static H5::CompType get_PredType() {

    auto build_type = []() {

      H5::CompType complex_data_type(2 * sizeof(T));

      complex_data_type.insertMember("r", 0, HDF5_TYPE<T>::get_PredType());

      complex_data_type.insertMember("i", sizeof(T), HDF5_TYPE<T>::get_PredType());

      return complex_data_type;

    };



template <>

class HDF5_TYPE<unsigned char> {

public:

  static hid_t get() {

    return H5T_NATIVE_UCHAR;

  }



  static H5::PredType get_PredType() {

    return H5::PredType::NATIVE_UCHAR;

    static H5::CompType type = build_type();

    return type;

  }

};



}  // io

}  // dca

}  // namespace io

}  // namespace dca



#endif  // DCA_IO_HDF5_HDF5_TYPES_HPP

include/dca/io/hdf5/hdf5_writer.hpp

+123 −483

File changed.

Preview size limit exceeded, changes collapsed.