X-Git-Url: https://gitweb.ps.run/toc/blobdiff_plain/8aeae09e74b46ca52866f22b48f55fecdf27b849..dbc4a22d3c8c4189459f0361cb9da06415ec2dc9:/src/main.cpp

diff --git a/src/main.cpp b/src/main.cpp
index 199f04d..19b35f3 100644
--- a/src/main.cpp
+++ b/src/main.cpp
@@ -12,8 +12,9 @@
 using namespace antlr4;
 
 
-int main(int argc, const char * argv[]) {
-  std::ifstream ifs("test.toc");
+int main(int argc, const char * argv[])
+{
+  std::ifstream ifs("test/test2.toc");
 
   ANTLRInputStream input(ifs);
 
@@ -24,7 +25,8 @@ int main(int argc, const char * argv[]) {
   TocParser::ProgContext * prog = parser.prog();
   tree::ParseTree * tree = prog;
 
-  if (parser.getNumberOfSyntaxErrors() > 0) {
+  if (parser.getNumberOfSyntaxErrors() > 0)
+  {
     std::cerr << "Parsing error" << std::endl;
     return 1;
   }
@@ -35,14 +37,21 @@ int main(int argc, const char * argv[]) {
 
   Program prg = getProgram(prog);
 
-  tocProgram(std::cout, prg);
+  try
+  {
+    tocProgram(std::cout, prg);
 
-  if (!checkProgram(prg))
-    std::cerr << "Error" << std::endl;
+    if (!checkProgram(prg))
+      std::cerr << "Error" << std::endl;
 
-  //std::ofstream ofs("output.c");
-  //tocProg(ofs, prg);
-  //ofs.close();
+    std::ofstream ofs("output.c");
+    tocProgram(ofs, prg);
+    ofs.close();
+  }
+  catch (const std::string & e)
+  {
+    std::cerr << "Error: " << e << std::endl;
+  }
 
   return 0;
 }
\ No newline at end of file