aboutsummaryrefslogtreecommitdiff
path: root/libs/installer_bsplugins/src/TESData/FormParser.h
diff options
context:
space:
mode:
Diffstat (limited to 'libs/installer_bsplugins/src/TESData/FormParser.h')
-rw-r--r--libs/installer_bsplugins/src/TESData/FormParser.h115
1 files changed, 115 insertions, 0 deletions
diff --git a/libs/installer_bsplugins/src/TESData/FormParser.h b/libs/installer_bsplugins/src/TESData/FormParser.h
new file mode 100644
index 0000000..b0220a0
--- /dev/null
+++ b/libs/installer_bsplugins/src/TESData/FormParser.h
@@ -0,0 +1,115 @@
+#ifndef TESDATA_FORMPARSER_H
+#define TESDATA_FORMPARSER_H
+
+#include "DataItem.h"
+#include "TESFile/Stream.h"
+#include "TESFile/Type.h"
+
+#include <coroutine>
+#include <istream>
+#include <map>
+#include <memory>
+#include <span>
+#include <string>
+
+using namespace Qt::Literals::StringLiterals;
+
+namespace TESData
+{
+
+namespace Parse
+{
+ struct Promise;
+
+ struct Task : std::coroutine_handle<Promise>
+ {
+ using promise_type = Promise;
+
+ Task(std::coroutine_handle<Promise>&& promise)
+ : std::coroutine_handle<Promise>::coroutine_handle(promise)
+ {}
+ };
+
+ struct Promise
+ {
+ Task get_return_object() { return {Task::from_promise(*this)}; }
+ std::suspend_always initial_suspend() noexcept { return {}; }
+ std::suspend_always final_suspend() noexcept { return {}; }
+ void return_void() {}
+ void unhandled_exception() {}
+ };
+} // namespace Parse
+
+using ParseTask = Parse::Task;
+
+class IFormParser
+{
+public:
+ virtual void parseFlags(DataItem* root, int fileIndex, std::uint32_t flags) const = 0;
+
+ virtual ParseTask parseForm(DataItem* root, int fileIndex, bool localized,
+ std::span<const std::string> masters,
+ const std::string& plugin, const TESFile::Type& signature,
+ std::istream* const& stream) const = 0;
+};
+
+class FormParserManager final
+{
+ template <typename T, TESFile::Type Type>
+ friend struct RegisterFormParser;
+
+public:
+ enum class Game
+ {
+ TES4,
+ FO3,
+ FNV,
+ TES5,
+ FO4,
+ SSE,
+
+ Unknown
+ };
+
+ FormParserManager() = delete;
+
+ [[nodiscard]] static std::shared_ptr<const IFormParser> getParser(Game game,
+ TESFile::Type type);
+
+private:
+ using RegistrationMap = std::map<TESFile::Type, std::shared_ptr<IFormParser>>;
+
+ static auto registrationMap() -> RegistrationMap&;
+};
+
+template <typename T, TESFile::Type Type>
+struct [[maybe_unused]] RegisterFormParser
+{
+ explicit RegisterFormParser()
+ {
+ FormParserManager::registrationMap().emplace(Type, std::make_shared<T>());
+ }
+};
+
+template <TESFile::Type Type = {}>
+class FormParser : public IFormParser
+{
+ inline static RegisterFormParser<FormParser<Type>, Type> reg{};
+
+public:
+ void parseFlags(DataItem* root, int fileIndex, std::uint32_t flags) const override;
+
+ ParseTask parseForm(DataItem* root, int fileIndex, bool localized,
+ std::span<const std::string> masters, const std::string& plugin,
+ const TESFile::Type& signature,
+ std::istream* const& stream) const override;
+};
+
+QString readBytes(std::istream& stream, int size);
+QString readLstring(bool localized, std::istream& stream);
+QString readFormId(std::span<const std::string> masters, const std::string& plugin,
+ std::istream& stream);
+
+} // namespace TESData
+
+#endif // TESDATA_FORMPARSER_H