HDF5Serializer: add translation unit

This commit is contained in:
Arne Morten Kvarving 2024-01-29 09:58:02 +01:00
parent 96a3a6d45a
commit 800422bd17
3 changed files with 81 additions and 38 deletions

View File

@ -629,6 +629,9 @@ if(HDF5_FOUND)
opm/simulators/utils/HDF5Serializer.hpp opm/simulators/utils/HDF5Serializer.hpp
opm/simulators/utils/HDF5File.hpp opm/simulators/utils/HDF5File.hpp
) )
list(APPEND MAIN_SOURCE_FILES
opm/simulators/utils/HDF5Serializer.cpp
)
endif() endif()
list (APPEND EXAMPLE_SOURCE_FILES list (APPEND EXAMPLE_SOURCE_FILES

View File

@ -0,0 +1,71 @@
/*
This file is part of the Open Porous Media project (OPM).
OPM 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 2 of the License, or
(at your option) any later version.
OPM 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 OPM. If not, see <http://www.gnu.org/licenses/>.
Consult the COPYING file in the top-level source directory of this
module for the precise wording of the license and the list of
copyright holders.
*/
#include <config.h>
#include <opm/simulators/utils/HDF5Serializer.hpp>
#include <algorithm>
#include <cstdlib>
namespace Opm {
void HDF5Serializer::writeHeader(const std::string& simulator_name,
const std::string& module_version,
const std::string& time_stamp,
const std::string& case_name,
const std::string& params,
int num_procs)
{
try {
this->pack(simulator_name, module_version, time_stamp,
case_name, params, num_procs);
} catch (...) {
m_packSize = std::numeric_limits<std::size_t>::max();
throw;
}
m_h5file.write("/", "simulator_info", m_buffer, HDF5File::DataSetMode::ROOT_ONLY);
}
int HDF5Serializer::lastReportStep() const
{
const auto entries = m_h5file.list("/report_step");
int last = -1;
for (const auto& entry : entries) {
int num = std::atoi(entry.c_str());
last = std::max(last, num);
}
return last;
}
std::vector<int> HDF5Serializer::reportSteps() const
{
const auto entries = m_h5file.list("/report_step");
std::vector<int> result(entries.size());
std::transform(entries.begin(), entries.end(), result.begin(),
[](const std::string& input)
{
return std::atoi(input.c_str());
});
std::sort(result.begin(), result.end());
return result;
}
}

View File

@ -24,12 +24,12 @@
#include <opm/common/utility/Serializer.hpp> #include <opm/common/utility/Serializer.hpp>
#include <opm/simulators/utils/HDF5File.hpp> #include <opm/simulators/utils/HDF5File.hpp>
#include <opm/simulators/utils/moduleVersion.hpp>
#include <opm/simulators/utils/ParallelCommunication.hpp> #include <opm/simulators/utils/ParallelCommunication.hpp>
#include <opm/simulators/utils/SerializationPackers.hpp> #include <opm/simulators/utils/SerializationPackers.hpp>
#include <algorithm> #include <cstddef>
#include <cstdlib> #include <limits>
#include <string>
namespace Opm { namespace Opm {
@ -55,7 +55,7 @@ public:
try { try {
this->pack(data); this->pack(data);
} catch (...) { } catch (...) {
m_packSize = std::numeric_limits<size_t>::max(); m_packSize = std::numeric_limits<std::size_t>::max();
throw; throw;
} }
@ -74,17 +74,7 @@ public:
const std::string& time_stamp, const std::string& time_stamp,
const std::string& case_name, const std::string& case_name,
const std::string& params, const std::string& params,
int num_procs) int num_procs);
{
try {
this->pack(simulator_name, module_version, time_stamp,
case_name, params, num_procs);
} catch (...) {
m_packSize = std::numeric_limits<size_t>::max();
throw;
}
m_h5file.write("/", "simulator_info", m_buffer, HDF5File::DataSetMode::ROOT_ONLY);
}
//! \brief Read data and deserialize from restart file. //! \brief Read data and deserialize from restart file.
//! \tparam T Type of class to read //! \tparam T Type of class to read
@ -100,31 +90,10 @@ public:
} }
//! \brief Returns the last report step stored in file. //! \brief Returns the last report step stored in file.
int lastReportStep() const int lastReportStep() const;
{
const auto entries = m_h5file.list("/report_step");
int last = -1;
for (const auto& entry : entries) {
int num = std::atoi(entry.c_str());
last = std::max(last, num);
}
return last;
}
//! \brief Returns a list of report steps stored in restart file. //! \brief Returns a list of report steps stored in restart file.
std::vector<int> reportSteps() const std::vector<int> reportSteps() const;
{
const auto entries = m_h5file.list("/report_step");
std::vector<int> result(entries.size());
std::transform(entries.begin(), entries.end(), result.begin(),
[](const std::string& input)
{
return std::atoi(input.c_str());
});
std::sort(result.begin(), result.end());
return result;
}
private: private:
const Serialization::MemPacker m_packer_priv{}; //!< Packer instance const Serialization::MemPacker m_packer_priv{}; //!< Packer instance