mirror of https://github.com/libAthena/athena.git
122 lines
3.3 KiB
C++
122 lines
3.3 KiB
C++
#pragma once
|
|
|
|
#include <string>
|
|
|
|
#include "athena/DNA.hpp"
|
|
#include "athena/FileReader.hpp"
|
|
#include "athena/FileWriter.hpp"
|
|
#include "athena/YAMLDocReader.hpp"
|
|
#include "athena/YAMLDocWriter.hpp"
|
|
|
|
namespace athena::io {
|
|
|
|
template <class T>
|
|
std::string_view __GetDNAName(const T& dna) {
|
|
if constexpr (__IsDNAVRecord_v<T>) {
|
|
return dna.DNATypeV();
|
|
} else {
|
|
return dna.DNAType();
|
|
}
|
|
}
|
|
|
|
template <class T>
|
|
std::string ToYAMLString(const T& dna) {
|
|
YAMLDocWriter docWriter(__GetDNAName(dna));
|
|
|
|
std::string res;
|
|
yaml_emitter_set_output(docWriter.getEmitter(), (yaml_write_handler_t*)YAMLStdStringWriter, &res);
|
|
yaml_emitter_set_unicode(docWriter.getEmitter(), true);
|
|
yaml_emitter_set_width(docWriter.getEmitter(), -1);
|
|
|
|
dna.write(docWriter);
|
|
if (!docWriter.finish(nullptr))
|
|
return std::string();
|
|
|
|
return res;
|
|
}
|
|
|
|
template <class T>
|
|
bool FromYAMLString(T& dna, std::string_view str) {
|
|
YAMLStdStringViewReaderState reader(str);
|
|
YAMLDocReader docReader;
|
|
yaml_parser_set_input(docReader.getParser(), (yaml_read_handler_t*)YAMLStdStringReader, &reader);
|
|
if (!docReader.parse(nullptr))
|
|
return false;
|
|
dna.read(docReader);
|
|
return true;
|
|
}
|
|
|
|
template <class DNASubtype>
|
|
bool ValidateFromYAMLString(std::string_view str) {
|
|
YAMLStdStringViewReaderState reader(str);
|
|
YAMLDocReader docReader;
|
|
yaml_parser_set_input(docReader.getParser(), (yaml_read_handler_t*)YAMLStdStringReader, &reader);
|
|
bool retval = docReader.ValidateClassType(DNASubtype::DNAType());
|
|
return retval;
|
|
}
|
|
|
|
template <class T>
|
|
bool ToYAMLStream(const T& dna, athena::io::IStreamWriter& fout) {
|
|
YAMLDocWriter docWriter(__GetDNAName(dna));
|
|
|
|
yaml_emitter_set_unicode(docWriter.getEmitter(), true);
|
|
yaml_emitter_set_width(docWriter.getEmitter(), -1);
|
|
|
|
dna.write(docWriter);
|
|
return docWriter.finish(&fout);
|
|
}
|
|
|
|
template <class T>
|
|
bool ToYAMLStream(const T& dna, athena::io::IStreamWriter& fout, void (T::*fn)(YAMLDocWriter& out) const) {
|
|
YAMLDocWriter docWriter(__GetDNAName(dna));
|
|
|
|
yaml_emitter_set_unicode(docWriter.getEmitter(), true);
|
|
yaml_emitter_set_width(docWriter.getEmitter(), -1);
|
|
|
|
(dna.*fn)(docWriter);
|
|
return docWriter.finish(&fout);
|
|
}
|
|
|
|
template <class T>
|
|
bool FromYAMLStream(T& dna, athena::io::IStreamReader& fin) {
|
|
YAMLDocReader docReader;
|
|
if (!docReader.parse(&fin))
|
|
return false;
|
|
dna.read(docReader);
|
|
return true;
|
|
}
|
|
|
|
template <class T>
|
|
bool FromYAMLStream(T& dna, athena::io::IStreamReader& fin, void (T::*fn)(YAMLDocReader& in)) {
|
|
YAMLDocReader docReader;
|
|
if (!docReader.parse(&fin))
|
|
return false;
|
|
(dna.*fn)(docReader);
|
|
return true;
|
|
}
|
|
|
|
template <class T, typename NameT>
|
|
bool MergeToYAMLFile(const T& dna, const NameT& filename) {
|
|
athena::io::FileReader r(filename);
|
|
YAMLDocWriter docWriter(__GetDNAName(dna), r.isOpen() ? &r : nullptr);
|
|
r.close();
|
|
|
|
dna.write(docWriter);
|
|
athena::io::FileWriter w(filename);
|
|
if (!w.isOpen())
|
|
return false;
|
|
return docWriter.finish(&w);
|
|
}
|
|
|
|
template <class DNASubtype>
|
|
bool ValidateFromYAMLStream(athena::io::IStreamReader& fin) {
|
|
YAMLDocReader reader;
|
|
atUint64 pos = fin.position();
|
|
yaml_parser_set_input(reader.getParser(), (yaml_read_handler_t*)YAMLAthenaReader, &fin);
|
|
bool retval = reader.ValidateClassType(DNASubtype::DNAType());
|
|
fin.seek(pos, athena::SeekOrigin::Begin);
|
|
return retval;
|
|
}
|
|
|
|
} // namespace athena::io
|