From 654c3f95919e5935632eb2450d52662cf49249d1 Mon Sep 17 00:00:00 2001
From: luffy06 <534427411@qq.com>
Date: Wed, 17 Jan 2024 13:26:06 +0800
Subject: [PATCH] update analyze codes

---
 analyze_cgf.py | 17 +++++++++++++----
 1 file changed, 13 insertions(+), 4 deletions(-)

diff --git a/analyze_cgf.py b/analyze_cgf.py
index 2c78d723e..40e8f85a8 100644
--- a/analyze_cgf.py
+++ b/analyze_cgf.py
@@ -93,6 +93,9 @@ def travel_in_topology(nodes, show_path=False):
         if degree == 0:
             queue.append(name)
 
+    if show_path:
+        print("Graph:")
+
     while len(queue) > 0:
         cur_node = queue.pop(0)
         for next_node in nodes[cur_node].next:
@@ -102,18 +105,24 @@ def travel_in_topology(nodes, show_path=False):
                 queue.append(next_node)
                 if show_path:
                     for prev_node in nodes[next_node].prev:
-                        print(f"{prev_node} -> {next_node}")
+                        print(f"\t{prev_node} -> {next_node}")
     
-    return concur, orders
+    return orders
 
 if __name__ == "__main__":
     parser = argparse.ArgumentParser()
     parser.add_argument("--log_file", required=True, type=str)
     parser.add_argument("--show_path", action=argparse.BooleanOptionalAction)
+    parser.add_argument("--show_detailed_orders", action=argparse.BooleanOptionalAction)
     args = parser.parse_args()
 
     gf = read_graph(args.log_file, skip_pattens=[".weight"])
 
-    concur, orders = travel_in_topology(gf, show_path=args.show_path)
-    print(f"max concurrency: {concur}, max order {np.max(list(orders.values()))}")
+    orders = travel_in_topology(gf, show_path=args.show_path)
+    order_values = list(orders.values())
+    print(f"Max concurrency: {np.max([order_values.count(x) if x > 3 else 0 for x in np.unique(order_values)])}\nMax order {np.max(order_values)}")
+    if args.show_detailed_orders:
+        print(f"Detailed orders:")
+        for name, order in sorted(list(orders.items()), key=lambda x: x[1]):
+            print(f"\t{name}: {order}")
             
\ No newline at end of file