From 24a5ccee4b90ecf4b6b866472489ed9c5eb01e7a Mon Sep 17 00:00:00 2001
From: =?utf8?q?=EB=B0=95=EC=A2=85=ED=98=84/=EB=8F=99=EC=9E=91=EC=A0=9C?=
 =?utf8?q?=EC=96=B4Lab=28SR=29/Staff=20Engineer/=EC=82=BC=EC=84=B1?=
 =?utf8?q?=EC=A0=84=EC=9E=90?= <jh1302.park@samsung.com>
Date: Thu, 30 Aug 2018 14:32:19 +0900
Subject: [PATCH] [enco] Emit identifyInputsAndOutputs call (#1244)

Generated C++ code will invoke ANeuralNetworksModel_identifyInputsAndOutputs
with this commit.

Signed-off-by: Jonghyun Park <jh1302.park@samsung.com>
---
 contrib/enco/core/src/CppGen/Subnet.cpp | 49 ++++++++++++++++++++++++++++++++-
 1 file changed, 48 insertions(+), 1 deletion(-)

diff --git a/contrib/enco/core/src/CppGen/Subnet.cpp b/contrib/enco/core/src/CppGen/Subnet.cpp
index 876b78e..5087dc3 100644
--- a/contrib/enco/core/src/CppGen/Subnet.cpp
+++ b/contrib/enco/core/src/CppGen/Subnet.cpp
@@ -272,6 +272,51 @@ private:
   const ann::Operation *_op;
 };
 
+/**
+ * @brief Code fragment that calls ANeuralNetworksModel_identifyInputsAndOutputs
+ */
+class ArgumentDecl final : public CodeFragment
+{
+public:
+  ArgumentDecl(const std::string &mname, const ANNBinder *binder) : _mname{mname}, _binder{binder}
+  {
+    // DO NOTHING
+  }
+
+public:
+  void dump(pp::LinearDocument *doc) const override
+  {
+    doc->append("{");
+    doc->indent();
+
+    auto module = _binder->module();
+    const uint32_t input_count = module->input()->size();
+
+    doc->append("uint32_t inputs[", input_count, "];");
+    for (uint32_t n = 0; n < input_count; ++n)
+    {
+      doc->append("inputs[", n, "] = ", module->input()->at(n), ";");
+    }
+
+    const uint32_t output_count = module->output()->size();
+
+    doc->append("uint32_t outputs[", output_count, "];");
+    for (uint32_t n = 0; n < output_count; ++n)
+    {
+      doc->append("outputs[", n, "] = ", module->output()->at(n), ";");
+    }
+
+    doc->append("ANeuralNetworksModel_identifyInputsAndOutputs(", _mname, ", ", input_count,
+                ", inputs, ", output_count, ", outputs);");
+    doc->unindent();
+    doc->append("}");
+  }
+
+private:
+  std::string _mname;
+  const ANNBinder *_binder;
+};
+
 } // namespace
 
 namespace enco
@@ -322,7 +367,9 @@ std::unique_ptr<SubnetStruct> SubnetStructBuilder::build(const ANNBinder *binder
     res->ctor() << OperationDecl{mname, op};
   }
 
-  // TODO Emit ANeuralNetworksModel_identifyInputsAndOutputs call
+  // Emit ANeuralNetworksModel_identifyInputsAndOutputs call
+  res->ctor() << ArgumentDecl{mname, binder};
+
   // TODO Emit ANeuralNetworksModel_finish call
   res->ctor()->append(S(assert("NYI");));
 
-- 
2.7.4