pre change

[toc] / src / main.cpp

diff --git a/src/main.cpp b/src/main.cpp
index 4c78f5b20e5cc5b757b3aeef2295fb29d19e0fb6..19b35f32ddbde4361e0b053b2a317f8172f84e68 100644 (file)
--- a/src/main.cpp
+++ b/src/main.cpp
@@ -14,7 +14,7 @@ using namespace antlr4;
 \r
 int main(int argc, const char * argv[])\r
 {\r
-  std::ifstream ifs("test.toc");\r
+  std::ifstream ifs("test/test2.toc");\r
 \r
   ANTLRInputStream input(ifs);\r
 \r
@@ -37,14 +37,21 @@ int main(int argc, const char * argv[])
 \r
   Program prg = getProgram(prog);\r
 \r
-  tocProgram(std::cout, prg);\r
+  try\r
+  {\r
+    tocProgram(std::cout, prg);\r
 \r
-  if (!checkProgram(prg))\r
-    std::cerr << "Error" << std::endl;\r
+    if (!checkProgram(prg))\r
+      std::cerr << "Error" << std::endl;\r
 \r
-  //std::ofstream ofs("output.c");\r
-  //tocProg(ofs, prg);\r
-  //ofs.close();\r
+    std::ofstream ofs("output.c");\r
+    tocProgram(ofs, prg);\r
+    ofs.close();\r
+  }\r
+  catch (const std::string & e)\r
+  {\r
+    std::cerr << "Error: " << e << std::endl;\r
+  }\r
 \r
   return 0;\r
 }
\ No newline at end of file