mirror of
https://github.com/triqs/dft_tools
synced 2024-12-24 13:23:37 +01:00
Add map<string,T> to h5
This commit is contained in:
parent
e891474544
commit
dfa27554f5
2
test/triqs/h5/CMakeLists.txt
Normal file
2
test/triqs/h5/CMakeLists.txt
Normal file
@ -0,0 +1,2 @@
|
||||
all_tests()
|
||||
|
58
test/triqs/h5/map_h5.cpp
Normal file
58
test/triqs/h5/map_h5.cpp
Normal file
@ -0,0 +1,58 @@
|
||||
/*******************************************************************************
|
||||
*
|
||||
* TRIQS: a Toolbox for Research in Interacting Quantum Systems
|
||||
*
|
||||
* Copyright (C) 2011 by O. Parcollet
|
||||
*
|
||||
* TRIQS 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.
|
||||
*
|
||||
* TRIQS 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
|
||||
* TRIQS. If not, see <http://www.gnu.org/licenses/>.
|
||||
*
|
||||
******************************************************************************/
|
||||
#include <iostream>
|
||||
#include <map>
|
||||
#include <triqs/h5.hpp>
|
||||
|
||||
namespace h5 = triqs::h5;
|
||||
|
||||
int main(int argc, char **argv) {
|
||||
|
||||
try {
|
||||
|
||||
// write
|
||||
std::map<std::string, int> m = { {"a",1}, {"b",2} };
|
||||
std::map<std::string, std::vector<double>> mv = { {"a",{1.0, 2.0}}, {"b",{2.0, 3.0, 4.0}} };
|
||||
|
||||
H5::H5File file1("test_map.h5", H5F_ACC_TRUNC);
|
||||
h5::group top1(file1);
|
||||
h5_write(top1, "map_int", m);
|
||||
h5_write(top1, "map_vec", mv);
|
||||
|
||||
// read
|
||||
std::map<std::string, int> mm = { {"c",1} };
|
||||
std::map<std::string, std::vector<double>> mmv = { {"c",{1.0}} };
|
||||
|
||||
H5::H5File file2("test_map.h5", H5F_ACC_RDONLY);
|
||||
h5::group top2(file2);
|
||||
h5_read(top2, "map_int", mm);
|
||||
h5_read(top2, "map_vec", mmv);
|
||||
|
||||
for (auto const & val: mm) std::cout << val.first << " " << val.second << std::endl;
|
||||
for (auto const & val: mmv) {
|
||||
std::cout << val.first << std::endl;
|
||||
for (auto const & x: val.second) std::cout << x << std::endl;
|
||||
}
|
||||
|
||||
}
|
||||
TRIQS_CATCH_AND_ABORT;
|
||||
}
|
||||
|
@ -23,6 +23,7 @@
|
||||
#include "./h5/group.hpp"
|
||||
#include "./h5/scalar.hpp"
|
||||
#include "./h5/vector.hpp"
|
||||
#include "./h5/map.hpp"
|
||||
#include "./h5/string.hpp"
|
||||
//#include "./h5/make_h5_read_write.hpp"
|
||||
|
||||
|
56
triqs/h5/map.hpp
Normal file
56
triqs/h5/map.hpp
Normal file
@ -0,0 +1,56 @@
|
||||
/*******************************************************************************
|
||||
*
|
||||
* TRIQS: a Toolbox for Research in Interacting Quantum Systems
|
||||
*
|
||||
* Copyright (C) 2011 by O. Parcollet
|
||||
*
|
||||
* TRIQS 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.
|
||||
*
|
||||
* TRIQS 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
|
||||
* TRIQS. If not, see <http://www.gnu.org/licenses/>.
|
||||
*
|
||||
******************************************************************************/
|
||||
#pragma once
|
||||
|
||||
#include "group.hpp"
|
||||
#include "string.hpp"
|
||||
#include <map>
|
||||
|
||||
namespace triqs {
|
||||
|
||||
template <typename T>
|
||||
std::string get_triqs_hdf5_data_scheme(std::map<std::string,T> const&) {
|
||||
using triqs::get_triqs_hdf5_data_scheme;
|
||||
return "std::map<string," + get_triqs_hdf5_data_scheme(T()) + ">";
|
||||
}
|
||||
|
||||
namespace h5 {
|
||||
|
||||
template<typename T>
|
||||
void h5_write (group f, std::string const & name, std::map<std::string,T> const & M) {
|
||||
auto gr = f.create_group(name);
|
||||
for (auto& pvp : M) h5_write(gr, pvp.first, pvp.second);
|
||||
}
|
||||
|
||||
template<typename T>
|
||||
void h5_read (group f, std::string const & name, std::map<std::string,T> & M) {
|
||||
auto gr = f.open_group(name);
|
||||
M.clear();
|
||||
T value;
|
||||
for (auto const & x: gr.get_all_dataset_names()) {
|
||||
h5_read(gr, x, value);
|
||||
M.emplace(x, value);
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
}
|
||||
|
Loading…
Reference in New Issue
Block a user