Write to a file if -o provided --show-c option

Author: czgdp1807

.gitignore

@@ -218,3 +218,9 @@ integration_tests/array_02_decl.c
 integration_tests/expr_12
 integration_tests/expr_12.c
 integration_tests/expr_07_generated.c
+integration_tests/expr_07.c
+expr_07.c
+tests/c_interop1.c
+tests/expr7.c
+tests/loop1.c
+tests/loop2.c

integration_tests/CMakeLists.txt

@@ -87,13 +87,13 @@ macro(RUN)
                 add_custom_command(
                     OUTPUT ${name}.c
                     COMMAND lpython --show-c ${CMAKE_CURRENT_SOURCE_DIR}/${name}.py
-                            --custom-c-postprocessor ${CMAKE_CURRENT_SOURCE_DIR}/${custom_c_postprocessor} > ${name}.c
+                            --custom-c-postprocessor ${CMAKE_CURRENT_SOURCE_DIR}/${custom_c_postprocessor} -o ${name}.c
                     DEPENDS ${CMAKE_CURRENT_SOURCE_DIR}/${name}.py
                     VERBATIM)
             else ()
                 add_custom_command(
                     OUTPUT ${name}.c
-                    COMMAND lpython --show-c ${CMAKE_CURRENT_SOURCE_DIR}/${name}.py > ${name}.c
+                    COMMAND lpython --show-c ${CMAKE_CURRENT_SOURCE_DIR}/${name}.py -o ${name}.c
                     DEPENDS ${CMAKE_CURRENT_SOURCE_DIR}/${name}.py
                     VERBATIM)
             endif()

integration_tests/test_c_custom_postprocessor.py

@@ -4,6 +4,10 @@
 
 filename = sys.argv[1]
 
-f = open(filename).read()
+file = open(filename, "r")
+f = file.read()
+file.close()
 
-open(filename, "w").write(f)
+file = open(filename, "w")
+file.write(f)
+file.close()

src/bin/lpython.cpp

@@ -295,21 +295,34 @@ int emit_c(const std::string &infile,
         LFORTRAN_ASSERT(diagnostics.has_error())
         return 3;
     }
+
+    std::string output_file;
+    bool print_to_std = true;
+    if( !compiler_options.arg_o.empty() ) {
+        output_file = compiler_options.arg_o;
+        print_to_std = false;
+    } else {
+        output_file = remove_extension(infile) + ".c";
+    }
+    std::string ccode_file = output_file;
+    std::ofstream ccode_out(ccode_file);
+    ccode_out << res.result;
+    ccode_out.close();
     if( !compiler_options.c_postprocessor_script.empty() ) {
-        std::string ccode_file = remove_extension(infile) + "_generated.c";
-        std::ofstream ccode_out(ccode_file);
-        ccode_out << res.result;
         std::string command = "python " + compiler_options.c_postprocessor_script + " " + ccode_file;
         system(command.c_str());
-        ccode_out.close();
-        std::ifstream ccode_in(ccode_file);
-        std::string modified_result = "";
-        while( std::getline(ccode_in, modified_result) ) {
-            std::cout << modified_result << "\n";
+        if( print_to_std ) {
+            std::ifstream ccode_in(ccode_file);
+            std::string modified_result = "";
+            while( std::getline(ccode_in, modified_result) ) {
+                std::cout << modified_result << "\n";
+            }
+            ccode_in.close();
         }
-        ccode_in.close();
     } else {
-        std::cout << res.result;
+        if( print_to_std ) {
+            std::cout << res.result;
+        }
     }
     return 0;
 }