1 files changed, 37 insertions, 0 deletions
diff --git a/src/armnnQuantizer/CommandLineProcessor.hpp b/src/armnnQuantizer/CommandLineProcessor.hpp
new file mode 100644
index 0000000000..f55e7a213f
--- /dev/null
+++ b/src/armnnQuantizer/CommandLineProcessor.hpp
@@ -0,0 +1,37 @@
+//
+// Copyright © 2017 Arm Ltd. All rights reserved.
+// SPDX-License-Identifier: MIT
+//
+#pragma once
+
+#include <string>
+#include <iostream>
+
+namespace armnnQuantizer
+{
+
+// parses the command line to extract
+// * the input file -f containing the serialized fp32 ArmNN input graph (must exist...and be a input graph file)
+// * the directory -d to place the output file into (must already exist and be writable)
+// * the name of the file -o the quantized ArmNN input graph will be written to (must not already exist)
+// * LATER: the min and max overrides to be applied to the inputs
+//          specified as -i <int> (input id) -n <float> (minimum) -x <float> (maximum)
+//          multiple sets of -i, -n, -x can appear on the command line but they must match
+//          in number i.e. a -n and -x for each -i and the id of the input must correspond
+//          to an input layer in the fp32 graph when it is loaded.
+class CommandLineProcessor
+{
+public:
+    bool ProcessCommandLine(int argc, char* argv[]);
+
+    std::string GetInputFileName() {return m_InputFileName;}
+    std::string GetOutputDirectoryName() {return m_OutputDirectory;}
+    std::string GetOutputFileName() {return m_OutputFileName;}
+private:
+    std::string m_InputFileName;
+    std::string m_OutputDirectory;
+    std::string m_OutputFileName;
+};
+
+} // namespace armnnQuantizer
+