1 files changed, 22 insertions, 0 deletions

diff --git a/tools/perf/builtin-diff.c b/tools/perf/builtin-diff.c
index 73d8bffdd484..a77e31246c00 100644
--- a/tools/perf/builtin-diff.c
+++ b/tools/perf/builtin-diff.c
@@ -778,6 +778,7 @@ static int __hpp__color_compare(struct perf_hpp_fmt *fmt,
                 container_of(fmt, struct diff_hpp_fmt, fmt);
         struct hist_entry *pair = get_pair_fmt(he, dfmt);
         double diff;
+        s64 wdiff;
         char pfmt[20] = " ";
 
         if (!pair)
@@ -806,6 +807,18 @@ static int __hpp__color_compare(struct perf_hpp_fmt *fmt,
                 scnprintf(pfmt, 20, "%%%d.6f", dfmt->header_width);
                 return value_color_snprintf(hpp->buf, hpp->size,
                                         pfmt, diff);
+        case COMPUTE_WEIGHTED_DIFF:
+                if (he->dummy)
+                        goto dummy_print;
+                if (pair->diff.computed)
+                        wdiff = pair->diff.wdiff;
+                else
+                        wdiff = compute_wdiff(he, pair);
+
+                scnprintf(pfmt, 20, "%%14ld", dfmt->header_width);
+                return color_snprintf(hpp->buf, hpp->size,
+                                get_percent_color(wdiff),
+                                pfmt, wdiff);
         default:
                 BUG_ON(1);
         }
@@ -826,6 +839,12 @@ static int hpp__color_ratio(struct perf_hpp_fmt *fmt,
         return __hpp__color_compare(fmt, hpp, he, COMPUTE_RATIO);
 }
 
+static int hpp__color_wdiff(struct perf_hpp_fmt *fmt,
+                        struct perf_hpp *hpp, struct hist_entry *he)
+{
+        return __hpp__color_compare(fmt, hpp, he, COMPUTE_WEIGHTED_DIFF);
+}
+
 static void
 hpp__entry_unpair(struct hist_entry *he, int idx, char *buf, size_t size)
 {
@@ -1007,6 +1026,9 @@ static void data__hpp_register(struct data__file *d, int idx)
         case PERF_HPP_DIFF__RATIO:
                 fmt->color = hpp__color_ratio;
                 break;
+        case PERF_HPP_DIFF__WEIGHTED_DIFF:
+                fmt->color = hpp__color_wdiff;
+                break;
         default:
                 break;
         }