From 646161af2dfde8c2410bcbd3d34e1da943e422c9 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/Senior=20Engineer/=EC=82=BC=EC=84=B1?=
 =?utf8?q?=EC=A0=84=EC=9E=90?= <jh1302.park@samsung.com>
Date: Wed, 25 Apr 2018 10:31:24 +0900
Subject: [PATCH] [caffegen] Allow Command to read command-line (#125)

This commit revises 'Command' interface to take command-line arguments
from main.

Signed-off-by: Jonghyun Park <jh1302.park@samsung.com>
---
 contrib/caffegen/include/Command.h     | 2 +-
 contrib/caffegen/src/Driver.cpp        | 2 +-
 contrib/caffegen/src/EncodeCommand.cpp | 2 +-
 contrib/caffegen/src/EncodeCommand.h   | 2 +-
 contrib/caffegen/src/FillCommand.cpp   | 2 +-
 contrib/caffegen/src/FillCommand.h     | 2 +-
 6 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/contrib/caffegen/include/Command.h b/contrib/caffegen/include/Command.h
index bc1c77e..1ca806a 100644
--- a/contrib/caffegen/include/Command.h
+++ b/contrib/caffegen/include/Command.h
@@ -5,7 +5,7 @@ struct Command
 {
   virtual ~Command() = default;
 
-  virtual int run(void) const = 0;
+  virtual int run(int argc, char **argv) const = 0;
 };
 
 #endif // __COMMAND_H__
diff --git a/contrib/caffegen/src/Driver.cpp b/contrib/caffegen/src/Driver.cpp
index 05899a6..76f7344 100644
--- a/contrib/caffegen/src/Driver.cpp
+++ b/contrib/caffegen/src/Driver.cpp
@@ -13,5 +13,5 @@ int main(int argc, char **argv)
   commands["fill"] = nncc::foundation::make_unique<FillCommand>();
   commands["encode"] = nncc::foundation::make_unique<EncodeCommand>();
 
-  return commands.at(argv[1])->run();
+  return commands.at(argv[1])->run(argc - 2, argv + 2);
 }
diff --git a/contrib/caffegen/src/EncodeCommand.cpp b/contrib/caffegen/src/EncodeCommand.cpp
index 382a270..3f7428d 100644
--- a/contrib/caffegen/src/EncodeCommand.cpp
+++ b/contrib/caffegen/src/EncodeCommand.cpp
@@ -8,7 +8,7 @@
 
 #include <iostream>
 
-int EncodeCommand::run(void) const
+int EncodeCommand::run(int, char **) const
 {
   caffe::NetParameter param;
 
diff --git a/contrib/caffegen/src/EncodeCommand.h b/contrib/caffegen/src/EncodeCommand.h
index 4371fd1..2188297 100644
--- a/contrib/caffegen/src/EncodeCommand.h
+++ b/contrib/caffegen/src/EncodeCommand.h
@@ -5,7 +5,7 @@
 
 struct EncodeCommand final : public Command
 {
-  int run(void) const override;
+  int run(int argc, char **argv) const override;
 };
 
 #endif // __ENCODE_COMMAND_H__
diff --git a/contrib/caffegen/src/FillCommand.cpp b/contrib/caffegen/src/FillCommand.cpp
index 0c366f7..266b205 100644
--- a/contrib/caffegen/src/FillCommand.cpp
+++ b/contrib/caffegen/src/FillCommand.cpp
@@ -13,7 +13,7 @@
 #include <random>
 #include <iostream>
 
-int FillCommand::run(void) const
+int FillCommand::run(int, char **) const
 {
   caffe::NetParameter param;
 
diff --git a/contrib/caffegen/src/FillCommand.h b/contrib/caffegen/src/FillCommand.h
index d9ce04e..1300cef 100644
--- a/contrib/caffegen/src/FillCommand.h
+++ b/contrib/caffegen/src/FillCommand.h
@@ -5,7 +5,7 @@
 
 struct FillCommand final : public Command
 {
-  int run(void) const override;
+  int run(int argc, char **argv) const override;
 };
 
 #endif // __FILL_COMMAND_H__
-- 
2.7.4