1 files changed, 80 insertions, 0 deletions
diff --git a/src/armnn/JsonUtils.hpp b/src/armnn/JsonUtils.hpp
new file mode 100644
index 0000000000..44fa7edc85
--- /dev/null
+++ b/src/armnn/JsonUtils.hpp
@@ -0,0 +1,80 @@
+//
+// Copyright © 2021 Arm Ltd and Contributors. All rights reserved.
+// SPDX-License-Identifier: MIT
+//
+
+#pragma once
+
+#include <iomanip>
+
+#include "armnn/Types.hpp"
+#include "armnn/backends/WorkloadInfo.hpp"
+
+namespace armnn
+{
+
+class JsonUtils
+{
+public:
+    JsonUtils(std::ostream& outputStream)
+        : m_NumTabs(0), m_OutputStream(outputStream)
+    {}
+
+    void PrintTabs()
+    {
+        unsigned int numTabs = m_NumTabs;
+        while ( numTabs-- > 0 )
+        {
+            m_OutputStream << "\t";
+        }
+    }
+
+    void DecrementNumberOfTabs()
+    {
+        if ( m_NumTabs == 0 )
+        {
+            return;
+        }
+        --m_NumTabs;
+    }
+
+    void IncrementNumberOfTabs()
+    {
+        ++m_NumTabs;
+    }
+
+    void PrintNewLine()
+    {
+        m_OutputStream << std::endl;
+    }
+
+    void PrintFooter()
+    {
+        DecrementNumberOfTabs();
+        PrintTabs();
+        m_OutputStream << "}";
+    }
+
+    void PrintHeader()
+    {
+        m_OutputStream << "{" << std::endl;
+        IncrementNumberOfTabs();
+    }
+
+    void PrintArmNNHeader()
+    {
+        PrintTabs();
+        m_OutputStream << R"("ArmNN": {)" << std::endl;
+        IncrementNumberOfTabs();
+    }
+    void PrintSeparator()
+    {
+        m_OutputStream << ",";
+    }
+
+private:
+    unsigned int m_NumTabs;
+    std::ostream& m_OutputStream;
+};
+
+} // namespace armnn
+\ No newline at end of file