chn/ufo
1
0
Fork 0
mirror of https://github.com/CHN-beta/ufo.git synced 2024-10-22 19:58:44 +08:00
Code Issues Packages Projects Releases Wiki Activity

add hdf5 output support

Browse Source
This commit is contained in:
陈浩南 2023-10-05 19:38:27 +08:00
parent 73cefd917c
commit 507338cca2
1 changed files with 33 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

35
src/unfold.cpp
View File

@ -70,12 +70,13 @@ namespace ufo
throw std::runtime_error("QPointDataOutputFile.SameAsConfigFile should not be set.");
if
(
!std::set<std::string>{"yaml", "yaml-human-readable", "zpp"}
!std::set<std::string>{"yaml", "yaml-human-readable", "zpp", "hdf5"}
.contains(QPointDataOutputFile[i].Format)
)
throw std::runtime_error(fmt::format
(
"Unknown QPointDataOutputFile[{}].Format: {}, should be \"yaml\", \"yaml-human-readable\" or \"zpp\".",
"Unknown QPointDataOutputFile[{}].Format: {}, should be "
"\"yaml\", \"yaml-human-readable\", \"zpp\" or \"hdf5\".",
i, QPointDataOutputFile[i].Format
));
}
@ -240,6 +241,36 @@ namespace ufo
file.exceptions(std::ios::badbit | std::ios::failbit);
file.write(reinterpret_cast<const char*>(data.data()), data.size());
}
else if (format == "hdf5")
{
std::vector<std::vector<double>> Qpoint;
std::vector<std::vector<double>> Source;
std::vector<std::vector<double>> Frequency;
std::vector<std::vector<double>> Weight;
for (auto& qpoint : QPointData)
{
Qpoint.emplace_back(qpoint.QPoint.data(), qpoint.QPoint.data() + 3);
Source.emplace_back(qpoint.Source.data(), qpoint.Source.data() + 3);
Frequency.emplace_back();
Weight.emplace_back();
for (auto& mode : qpoint.ModeData)
{
Frequency.back().push_back(mode.Frequency);
Weight.back().push_back(mode.Weight);
}
}
HighFive::File file(filename,
HighFive::File::ReadWrite | HighFive::File::Create | HighFive::File::Truncate);
file.createDataSet<double>("QPoint",
HighFive::DataSpace::From(Qpoint)).write(Qpoint);
file.createDataSet<double>("Source",
HighFive::DataSpace::From(Source)).write(Source);
file.createDataSet<double>("Frequency",
HighFive::DataSpace::From(Frequency)).write(Frequency);
file.createDataSet<double>("Weight",
HighFive::DataSpace::From(Weight)).write(Weight);
}
}
UnfoldSolver::UnfoldSolver(std::string config_file) : Input_([&]