mirror of
https://github.com/triqs/dft_tools
synced 2024-11-01 11:43:47 +01:00
f1c32012a6
- for all functions, except when GIL option (not implemented) is here we use a triqs::py_stream, which redirect the stream to python sys.write - so that a simple C++ code with a std::cout print its output in the notebook... - NB : it only works for the code within the cell. If it calls another function on the lib which uses cout, print is not redirected. Designed for simple tests.... - cerr not (yet) redirected. (useful ?).
98 lines
3.2 KiB
C++
98 lines
3.2 KiB
C++
/*******************************************************************************
|
|
*
|
|
* TRIQS: a Toolbox for Research in Interacting Quantum Systems
|
|
*
|
|
* Copyright (C) 2012 by M. Ferrero, 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 <hdf5.h>
|
|
#include <triqs/h5.hpp>
|
|
#include <triqs/utility/view_tools.hpp>
|
|
|
|
//#define CHECK_OR_THROW(Cond, Mess)
|
|
#define CHECK_OR_THROW(Cond, Mess) \
|
|
if (!(Cond)) TRIQS_RUNTIME_ERROR << "Error in h5 (de)serialization " << Mess;
|
|
|
|
namespace triqs {
|
|
namespace h5 {
|
|
|
|
template <typename T> std::string serialize(T const &x) {
|
|
|
|
hid_t fapl_id = H5Pcreate(H5P_FILE_ACCESS);
|
|
CHECK_OR_THROW((fapl_id >= 0), "creating fapl");
|
|
|
|
auto err = H5Pset_fapl_core(fapl_id, (size_t)(64 * 1024), false);
|
|
CHECK_OR_THROW((err >= 0), "setting core file driver in fapl.");
|
|
|
|
hid_t file_id = H5Fcreate("serialization", 0, H5P_DEFAULT, fapl_id);
|
|
CHECK_OR_THROW((file_id >= 0), "created core file");
|
|
|
|
auto gr = triqs::h5::group(file_id, false);
|
|
h5_write(gr, "object", x);
|
|
|
|
err = H5Fflush(file_id, H5F_SCOPE_GLOBAL);
|
|
CHECK_OR_THROW((err >= 0), "flushed core file.");
|
|
|
|
ssize_t image_len = H5Fget_file_image(file_id, NULL, (size_t)0);
|
|
CHECK_OR_THROW((image_len > 0), "got image file size");
|
|
|
|
std::string buf(image_len, 0);
|
|
|
|
ssize_t bytes_read = H5Fget_file_image(file_id, (void *)&buf[0], (size_t)image_len);
|
|
CHECK_OR_THROW(bytes_read == image_len, "wrote file into image buffer");
|
|
|
|
err = H5Fclose(file_id);
|
|
CHECK_OR_THROW((err >= 0), "closed core file(1).");
|
|
|
|
err = H5Pclose(fapl_id);
|
|
CHECK_OR_THROW((err >= 0), "closed fapl(1).");
|
|
|
|
return buf;
|
|
}
|
|
|
|
template <typename T> T deserialize(std::string const &buf) {
|
|
T res;
|
|
|
|
hid_t fapl_id = H5Pcreate(H5P_FILE_ACCESS);
|
|
CHECK_OR_THROW((fapl_id >= 0), "creating fapl");
|
|
|
|
auto err = H5Pset_fapl_core(fapl_id, (size_t)(64 * 1024), false);
|
|
CHECK_OR_THROW((err >= 0), "setting core file driver in fapl.");
|
|
|
|
err = H5Pset_file_image(fapl_id, (void *)&buf[0], buf.size());
|
|
CHECK_OR_THROW((err >= 0), "set file image in fapl.");
|
|
|
|
hid_t file_id = H5Fopen("serialization", H5F_ACC_RDONLY, fapl_id);
|
|
CHECK_OR_THROW((file_id >= 0), "opened received file image file");
|
|
|
|
auto gr = triqs::h5::group(file_id, false);
|
|
h5_read(gr, "object", res);
|
|
|
|
err = H5Fclose(file_id);
|
|
CHECK_OR_THROW((err >= 0), "closed core file(1).");
|
|
|
|
err = H5Pclose(fapl_id);
|
|
CHECK_OR_THROW((err >= 0), "closed fapl(1).");
|
|
|
|
return res;
|
|
}
|
|
}
|
|
|
|
#undef CHECK_OR_THROW
|
|
}
|
|
|