[gcc/devel/rust/master] borrowck: Dump: simplify cfg

[Thomas Schwinge via Gcc-cvs]

https://gcc.gnu.org/g:c92e110f06c49931a169319e4fd679c8e76ffcde

commit c92e110f06c49931a169319e4fd679c8e76ffcde
Author: Jakub Dupak <d...@jakubdupak.com>
Date:   Wed Oct 18 22:50:02 2023 +0200

    borrowck: Dump: simplify cfg
    
    Add option to simplify cfg for better readability. Off by default.
    
    gcc/rust/ChangeLog:
    
            * checks/errors/borrowck/rust-bir-dump.cc (simplify_cfg): Simplify 
cfg logic.
            (Dump::go): Run simplify cfg.
            * checks/errors/borrowck/rust-bir-dump.h: Option to simplify cfg.
    
    Signed-off-by: Jakub Dupak <d...@jakubdupak.com>

Diff:
---
 gcc/rust/checks/errors/borrowck/rust-bir-dump.cc | 40 +++++++++++++++++++++---
 gcc/rust/checks/errors/borrowck/rust-bir-dump.h  |  7 +++--
 2 files changed, 41 insertions(+), 6 deletions(-)

diff --git a/gcc/rust/checks/errors/borrowck/rust-bir-dump.cc 
b/gcc/rust/checks/errors/borrowck/rust-bir-dump.cc
index 4571b2fe8577..709d36a7cd33 100644
--- a/gcc/rust/checks/errors/borrowck/rust-bir-dump.cc
+++ b/gcc/rust/checks/errors/borrowck/rust-bir-dump.cc
@@ -1,3 +1,4 @@
+#include <numeric>
 #include "rust-bir-dump.h"
 
 namespace Rust {
@@ -43,11 +44,39 @@ print_comma_separated (std::ostream &stream, const 
std::vector<T> &collection,
     }
 }
 
-static constexpr bool FOLD_CFG = true;
+void
+simplify_cfg (Function &func, std::vector<BasicBlockId> &bb_fold_map)
+{
+  // The BIR builder can generate many useless basic blocks, which contain only
+  // a goto.
+  // For actual borrow-checking, the folding has little value.
+
+  bool stabilized = false;
+  while (!stabilized)
+    {
+      stabilized = true;
+      // BB0 cannot be folded as it is an entry block.
+      for (size_t i = 1; i < func.basic_blocks.size (); ++i)
+       {
+         const BasicBlock &bb = func.basic_blocks[bb_fold_map[i]];
+         if (bb.statements.empty () && bb.is_goto_terminated ())
+           {
+             bb_fold_map[i] = bb.successors.at (0);
+             stabilized = false;
+           }
+       }
+    }
+}
 
 void
-Dump::go ()
+Dump::go (bool enable_simplify_cfg)
 {
+  // To avoid mutation of the BIR, we use indirection through bb_fold_map.
+  std::iota (bb_fold_map.begin (), bb_fold_map.end (), 0);
+
+  if (enable_simplify_cfg)
+    simplify_cfg (func, bb_fold_map);
+
   stream << "fn " << name << "(";
   print_comma_separated (stream, func.arguments, [this] (PlaceId place_id) {
     stream << "_" << get_place_name (place_id) << ": "
@@ -71,13 +100,16 @@ Dump::go ()
 
   for (BasicBlockId id = 0; id < func.basic_blocks.size (); ++id)
     {
+      if (bb_fold_map[id] != id)
+       continue; // This BB was folded.
+
       if (func.basic_blocks[id].statements.empty ()
          && func.basic_blocks[id].successors.empty ())
        continue;
 
       BasicBlock &bb = func.basic_blocks[id];
       stream << "\n";
-      stream << indentation << "bb" << id << ": {\n";
+      stream << indentation << "bb" << bb_fold_map[id] << ": {\n";
       for (auto &stmt : bb.statements)
        {
          stream << indentation << indentation;
@@ -86,7 +118,7 @@ Dump::go ()
        }
       stream << indentation << "} -> [";
       for (auto succ : bb.successors)
-       stream << "bb" << succ << ", ";
+       stream << "bb" << bb_fold_map[succ] << ", ";
       stream << "]\n";
     }
 
diff --git a/gcc/rust/checks/errors/borrowck/rust-bir-dump.h 
b/gcc/rust/checks/errors/borrowck/rust-bir-dump.h
index 7c24749fed55..1f8de045c22c 100644
--- a/gcc/rust/checks/errors/borrowck/rust-bir-dump.h
+++ b/gcc/rust/checks/errors/borrowck/rust-bir-dump.h
@@ -35,13 +35,16 @@ class Dump : public Visitor
   Function &func;
   const std::string &name;
 
+  std::vector<BasicBlockId> bb_fold_map;
+
   PlaceId node_place = INVALID_PLACE;
 
 public:
   Dump (std::ostream &os, Function &func, const std::string &name)
-    : stream (os), place_db (func.place_db), func (func), name (name)
+    : stream (os), place_db (func.place_db), func (func), name (name),
+      bb_fold_map (func.basic_blocks.size ())
   {}
-  void go ();
+  void go (bool enable_simplify_cfg = false);
 
 protected:
   void visit (Node &node) override;