From af62d8b1260bb04b6963634db6e3752bb8a97ccf Mon Sep 17 00:00:00 2001
From: Slava Pestov <slava@oberon.internal.stack-effects.com>
Date: Tue, 12 Feb 2008 15:52:32 -0600
Subject: [PATCH] Add new optimization report tool

---
 .../specializers/specializers.factor          | 12 ++--
 extra/optimizer/debugger/debugger.factor      | 62 ++++++++++++++++++-
 2 files changed, 66 insertions(+), 8 deletions(-)

diff --git a/core/optimizer/specializers/specializers.factor b/core/optimizer/specializers/specializers.factor
index af8cd5b82e..a4f5aaab95 100755
--- a/core/optimizer/specializers/specializers.factor
+++ b/core/optimizer/specializers/specializers.factor
@@ -24,16 +24,18 @@ IN: optimizer.specializers
         \ dispatch ,
     ] [ ] make ;
 
+: specializer-methods ( word -- alist )
+    dup [ array? ] all? [ 1array ] unless [
+        [ make-specializer ] keep
+        [ declare ] curry pick append
+    ] { } map>assoc ;
+
 : specialized-def ( word -- quot )
     dup word-def swap "specializer" word-prop [
         dup { number } = [
             drop tag-specializer
         ] [
-            dup [ array? ] all? [ 1array ] unless [
-                [ make-specializer ] keep
-                [ declare ] curry pick append
-            ] { } map>assoc
-            alist>quot
+            specializer-methods alist>quot
         ] if
     ] when* ;
 
diff --git a/extra/optimizer/debugger/debugger.factor b/extra/optimizer/debugger/debugger.factor
index 499222073b..729281e81b 100755
--- a/extra/optimizer/debugger/debugger.factor
+++ b/extra/optimizer/debugger/debugger.factor
@@ -1,9 +1,10 @@
-! Copyright (C) 2006, 2007 Slava Pestov.
+! Copyright (C) 2006, 2008 Slava Pestov.
 ! See http://factorcode.org/license.txt for BSD license.
 USING: classes inference inference.dataflow io kernel
 kernel.private math.parser namespaces optimizer prettyprint
 prettyprint.backend sequences words arrays match macros
-assocs sequences.private ;
+assocs sequences.private optimizer.specializers generic
+combinators sorting math ;
 IN: optimizer.debugger
 
 ! A simple tool for turning dataflow IR into quotations, for
@@ -113,7 +114,62 @@ M: object node>quot dup class word-name comment, ;
 : dataflow>quot ( node ? -- quot )
     [ swap (dataflow>quot) ] [ ] make ;
 
-: print-dataflow ( quot ? -- )
+: optimized-quot. ( quot ? -- )
     #! Print dataflow IR for a quotation. Flag indicates if
     #! annotations should be printed or not.
     >r dataflow optimize r> dataflow>quot pprint nl ;
+
+: optimized-word. ( word ? -- ) >r specialized-def r> optimized-quot. ;
+
+SYMBOL: words-called
+SYMBOL: generics-called
+SYMBOL: methods-called
+SYMBOL: intrinsics-called
+SYMBOL: node-count
+
+: dataflow>report ( node -- alist )
+    [
+        H{ } clone words-called set
+        H{ } clone generics-called set
+        H{ } clone methods-called set
+        H{ } clone intrinsics-called set
+
+        0 swap [
+            >r 1+ r>
+            dup #call? [
+                node-param {
+                    { [ dup "intrinsics" word-prop over "if-intrinsics" word-prop or ] [ intrinsics-called ] }
+                    { [ dup generic? ] [ generics-called ] }
+                    { [ dup method-body? ] [ methods-called ] }
+                    { [ t ] [ words-called ] }
+                } cond 1 -rot get at+
+            ] [
+                drop
+            ] if
+        ] each-node
+        node-count set
+    ] H{ } make-assoc ;
+
+: quot-optimize-report ( quot -- report )
+    dataflow optimize dataflow>report ;
+
+: word-optimize-report ( word -- report )
+    word-def quot-optimize-report ;
+
+: report. ( report -- )
+    [
+        "==== Total number of dataflow nodes:" print
+        node-count get .
+
+        {
+            { generics-called "==== Generic word calls:" }
+            { words-called "==== Ordinary word calls:" }
+            { methods-called "==== Non-inlined method calls:" }
+            { intrinsics-called "==== Open-coded intrinsic calls:" }
+        } [
+            nl print get keys natural-sort stack.
+        ] assoc-each
+    ] bind ;
+
+: optimizer-report. ( word -- )
+    word-optimize-report report. ;