From ce716f9c063a91adf0c6322ba3c3111640e3c714 Mon Sep 17 00:00:00 2001
From: jiangjiajun <jiangjiajun@baidu.com>
Date: Tue, 16 Jul 2019 15:45:29 +0800
Subject: [PATCH] add codes

---
 x2paddle/core/emitter.py       | 16 +++++++++++++++-
 x2paddle/emitter/tf_emitter.py |  2 +-
 2 files changed, 16 insertions(+), 2 deletions(-)

diff --git a/x2paddle/core/emitter.py b/x2paddle/core/emitter.py
index 613454b..81a5b0b 100644
--- a/x2paddle/core/emitter.py
+++ b/x2paddle/core/emitter.py
@@ -15,7 +15,21 @@
 
 class Emitter(object):
     def __init__(self):
-        print("Nothing done")
+        self.paddle_codes = ""
+        self.tab = "    "
+
+    def add_codes(self, codes, indent=0):
+        if isinstance(codes, list):
+            for code in codes:
+                self.paddle_codes += (self.tab * indent + code + '\n')
+        elif isinstance(codes, str):
+            self.paddle_codes += (self.tab * indent + codes + '\n')
+        else:
+            raise Exception("Unknown type of codes")
+
+    def add_heads(self):
+        self.add_codes("import paddle.fluid as fluid")
+        self.add_codes("")
 
     def save_inference_model(self):
         print("Not Implement")
diff --git a/x2paddle/emitter/tf_emitter.py b/x2paddle/emitter/tf_emitter.py
index 0e343ae..2071b15 100644
--- a/x2paddle/emitter/tf_emitter.py
+++ b/x2paddle/emitter/tf_emitter.py
@@ -17,4 +17,4 @@ from x2paddle.core.emitter import Emitter
 
 class TFEmitter(Emitter):
     def __init__(self):
-        super(TFEmitter, self
+        super(TFEmitter, self).__init__()
-- 
GitLab