Support print asr after pass.

  Now we will generate a log folder to store each asr pass log.
  And add a helper function to select file to store the asr.

Author: redbopo

src/bin/lpython.cpp

@@ -789,6 +789,7 @@ int main(int argc, char *argv[])
         bool static_link = false;
         std::string arg_backend = "llvm";
         std::string arg_kernel_f;
+        bool print_after_asr_pass = false;
         bool print_targets = false;
         bool print_rtlib_header_dir = false;
 
@@ -855,6 +856,7 @@ int main(int argc, char *argv[])
         app.add_option("--target", compiler_options.target, "Generate code for the given target")->capture_default_str();
         app.add_flag("--print-targets", print_targets, "Print the registered targets");
         app.add_flag("--get-rtlib-header-dir", print_rtlib_header_dir, "Print the path to the runtime library header file");
+        app.add_flag("--print-after-asr-pass", print_after_asr_pass, "Print ASR output after every pass");
 
         // LSP specific options
         app.add_flag("--show-errors", show_errors, "Show errors when LSP is running in the background");
@@ -864,6 +866,7 @@ int main(int argc, char *argv[])
             lpython_pass_manager.use_optimization_passes();
         }
 
+
         /*
         * Subcommands:
         */
@@ -957,6 +960,10 @@ int main(int argc, char *argv[])
             return 1;
         }
 
+        if( print_after_asr_pass ) {
+            lpython_pass_manager.print_after_asr_pass();
+        }
+
         if (arg_backend == "llvm") {
             backend = Backend::llvm;
         } else if (arg_backend == "cpp") {

src/libasr/pass/pass_manager.h

@@ -14,6 +14,10 @@
     #include <lpython/utils.h>
 #endif
 
+#if __has_include(<lpython/utils.h>)
+    #include <lpython/pickle.h>
+#endif
+
 #include <libasr/pass/do_loops.h>
 #include <libasr/pass/for_all.h>
 #include <libasr/pass/implied_do_loops.h>
@@ -37,7 +41,6 @@
 #include <libasr/pass/loop_vectorise.h>
 #include <libasr/pass/update_array_dim_intrinsic_calls.h>
 #include <libasr/pass/pass_array_by_data.h>
-
 #include <map>
 #include <vector>
 
@@ -77,18 +80,32 @@ namespace LCompilers {
 
         bool is_fast;
         bool apply_default_passes;
+        bool do_print_after_asr_pass;
 
         void _apply_passes(Allocator& al, LFortran::ASR::TranslationUnit_t* asr,
                            std::vector<std::string>& passes, PassOptions pass_options) {
             pass_options.runtime_library_dir = LFortran::get_runtime_library_dir();
+            if (do_print_after_asr_pass) {
+                std::string cmd = "mkdir asr_pass_log";
+                int err = system(cmd.c_str());
+                if (err) {
+                    std::cout << "The command '" + cmd + "' failed." << std::endl;
+                    exit(1);
+                }
+                LFortran::write_asr_to_file("./asr_pass_log/0_raw.log", *asr, false, false);
+            }
             for (size_t i = 0; i < passes.size(); i++) {
                 _passes_db[passes[i]](al, *asr, pass_options);
+                if (do_print_after_asr_pass) {
+                    std::string file = "./asr_pass_log/" + std::to_string(i+1)+"_"+passes[i]+".log";
+                    LFortran::write_asr_to_file(file,*asr, false, false);
+                }
             }
         }
 
         public:
 
-        PassManager(): is_fast{false}, apply_default_passes{false} {
+        PassManager(): is_fast{false}, apply_default_passes{false}, do_print_after_asr_pass(false) {
             _passes = {
                 "global_stmts",
                 "class_constructor",
@@ -171,6 +188,12 @@ namespace LCompilers {
             }
         }
 
+
+
+        void print_after_asr_pass() {
+            do_print_after_asr_pass = true;
+        }
+
         void use_optimization_passes() {
             is_fast = true;
         }

src/lpython/pickle.cpp

@@ -1,4 +1,5 @@
 #include <string>
+#include <fstream>
 
 #include <lpython/pickle.h>
 #include <lpython/pickle.h>
@@ -95,7 +96,7 @@ class ASRTreeVisitor :
 {
 public:
     bool show_intrinsic_modules;
-    
+
     std::string get_str() {
         return s;
     }
@@ -114,4 +115,11 @@ std::string pickle_tree(LFortran::ASR::TranslationUnit_t &asr, bool colors, bool
     return pickle_tree((ASR::asr_t &)asr, colors, show_intrinsic_modules);
 }
 
+void write_asr_to_file(std::string file, LFortran::ASR::TranslationUnit_t &asr, bool colors, bool show_intrinsic_modules) {
+    std::ofstream f;
+    f.open(file);
+    f << pickle_tree(asr, colors, show_intrinsic_modules);
+    f.close();
+}
+
 }

src/lpython/pickle.h

@@ -14,6 +14,7 @@ namespace LFortran {
     // Print the tree structure
     std::string pickle_tree(LFortran::ASR::asr_t &asr, bool colors, bool show_intrinsic_modules);
     std::string pickle_tree(LFortran::ASR::TranslationUnit_t &asr, bool colors, bool show_intrinsic_modules);
+    void write_asr_to_file(std::string file, LFortran::ASR::TranslationUnit_t &asr, bool colors, bool show_intrinsic_modules);
 
 }