completed thesisHEADmaster

1 files changed, 58 insertions, 0 deletions

diff --git a/python/openvino/runtime/common/format_reader/register.h b/python/openvino/runtime/common/format_reader/register.h
new file mode 100644
index 0000000..781eca3
--- /dev/null
+++ b/python/openvino/runtime/common/format_reader/register.h
@@ -0,0 +1,58 @@
+// Copyright (C) 2018-2022 Intel Corporation
+// SPDX-License-Identifier: Apache-2.0
+//
+/**
+ * \brief Register for readers
+ * \file register.h
+ */
+#pragma once
+
+#include <functional>
+#include <string>
+#include <vector>
+
+#include "format_reader.h"
+
+namespace FormatReader {
+/**
+ * \class Registry
+ * \brief Create reader from fabric
+ */
+class Registry {
+private:
+    typedef std::function<Reader*(const std::string& filename)> CreatorFunction;
+    static std::vector<CreatorFunction> _data;
+
+public:
+    /**
+     * \brief Create reader
+     * @param filename - path to input data
+     * @return Reader for input data or nullptr
+     */
+    static Reader* CreateReader(const char* filename);
+
+    /**
+     * \brief Registers reader in fabric
+     * @param f - a creation function
+     */
+    static void RegisterReader(CreatorFunction f);
+};
+
+/**
+ * \class Register
+ * \brief Registers reader in fabric
+ */
+template <typename To>
+class Register {
+public:
+    /**
+     * \brief Constructor creates creation function for fabric
+     * @return Register object
+     */
+    Register() {
+        Registry::RegisterReader([](const std::string& filename) -> Reader* {
+            return new To(filename);
+        });
+    }
+};
+}  // namespace FormatReader