The new ->add_entry_cb() will be called after an entry was added to
the histogram.  It's used for code sharing between perf report and
perf top.  Note that ops->add_*_entry() should set iter->he properly
in order to call the ->add_entry_cb.

Also pass @arg to the callback function.  It'll be used by perf top
later.

Cc: Arun Sharma <asha...@fb.com>
Cc: Frederic Weisbecker <fweis...@gmail.com>
Signed-off-by: Namhyung Kim <namhy...@kernel.org>
---
 tools/perf/builtin-report.c | 30 +++++++++++++++++++++++++++---
 tools/perf/util/hist.c      | 45 +++++++++++++++++++++++++--------------------
 tools/perf/util/hist.h      |  6 +++++-
 3 files changed, 57 insertions(+), 24 deletions(-)

diff --git a/tools/perf/builtin-report.c b/tools/perf/builtin-report.c
index b8541ec61a00..3dbf2b83fc5e 100644
--- a/tools/perf/builtin-report.c
+++ b/tools/perf/builtin-report.c
@@ -87,6 +87,27 @@ static int report__config(const char *var, const char 
*value, void *cb)
        return perf_default_config(var, value, cb);
 }
 
+static int hist_iter_cb(struct hist_entry_iter *iter,
+                       struct addr_location *al, bool single,
+                       void *arg __maybe_unused)
+{
+       int err;
+       struct hist_entry *he = iter->he;
+       struct perf_evsel *evsel = iter->evsel;
+       struct perf_sample *sample = iter->sample;
+
+       err = hist_entry__inc_addr_samples(he, evsel->idx, al->addr);
+
+       if (!single)
+               goto out;
+
+       evsel->hists.stats.total_period += sample->period;
+       hists__inc_nr_events(&evsel->hists, PERF_RECORD_SAMPLE);
+
+out:
+       return err;
+}
+
 static int process_sample_event(struct perf_tool *tool,
                                union perf_event *event,
                                struct perf_sample *sample,
@@ -116,16 +137,19 @@ static int process_sample_event(struct perf_tool *tool,
                iter.ops = &hist_iter_branch;
        else if (rep->mem_mode == 1)
                iter.ops = &hist_iter_mem;
-       else if (symbol_conf.cumulate_callchain)
+       else if (symbol_conf.cumulate_callchain) {
                iter.ops = &hist_iter_cumulative;
-       else
+               iter.add_entry_cb = hist_iter_cb;
+       } else {
                iter.ops = &hist_iter_normal;
+               iter.add_entry_cb = hist_iter_cb;
+       }
 
        if (al.map != NULL)
                al.map->dso->hit = 1;
 
        ret = hist_entry_iter__add(&iter, &al, evsel, event, sample,
-                                  rep->max_stack);
+                                  rep->max_stack, NULL);
        if (ret < 0)
                pr_debug("problem adding hist entry, skipping event\n");
 
diff --git a/tools/perf/util/hist.c b/tools/perf/util/hist.c
index da4fbec3d0d8..b14fa143f7d7 100644
--- a/tools/perf/util/hist.c
+++ b/tools/perf/util/hist.c
@@ -683,11 +683,10 @@ iter_add_single_normal_entry(struct hist_entry_iter 
*iter, struct addr_location
 }
 
 static int
-iter_finish_normal_entry(struct hist_entry_iter *iter, struct addr_location 
*al)
+iter_finish_normal_entry(struct hist_entry_iter *iter,
+                        struct addr_location *al __maybe_unused)
 {
-       int err;
        struct hist_entry *he = iter->he;
-       struct perf_evsel *evsel = iter->evsel;
        struct perf_sample *sample = iter->sample;
 
        if (he == NULL)
@@ -695,13 +694,6 @@ iter_finish_normal_entry(struct hist_entry_iter *iter, 
struct addr_location *al)
 
        iter->he = NULL;
 
-       err = hist_entry__inc_addr_samples(he, evsel->idx, al->addr);
-       if (err)
-               return err;
-
-       evsel->hists.stats.total_period += sample->period;
-       hists__inc_nr_events(&evsel->hists, PERF_RECORD_SAMPLE);
-
        return hist_entry__append_callchain(he, sample);
 }
 static int
@@ -742,6 +734,7 @@ iter_add_single_cumulative_entry(struct hist_entry_iter 
*iter,
        if (he == NULL)
                return -ENOMEM;
 
+       iter->he = he;
        he_cache[iter->curr++] = he;
 
        callchain_append(he->callchain, &callchain_cursor, sample->period);
@@ -752,7 +745,7 @@ iter_add_single_cumulative_entry(struct hist_entry_iter 
*iter,
         */
        callchain_cursor_commit(&callchain_cursor);
 
-       return hist_entry__inc_addr_samples(he, evsel->idx, al->addr);
+       return 0;
 }
 
 static int
@@ -799,8 +792,11 @@ iter_add_next_cumulative_entry(struct hist_entry_iter 
*iter,
         * It's possible that it has cycles or recursive calls.
         */
        for (i = 0; i < iter->curr; i++) {
-               if (hist_entry__cmp(he_cache[i], &he_tmp) == 0)
+               if (hist_entry__cmp(he_cache[i], &he_tmp) == 0) {
+                       /* to avoid calling callback function */
+                       iter->he = NULL;
                        return 0;
+               }
        }
 
        he = __hists__add_entry(&evsel->hists, al, iter->parent, NULL, NULL,
@@ -809,24 +805,20 @@ iter_add_next_cumulative_entry(struct hist_entry_iter 
*iter,
        if (he == NULL)
                return -ENOMEM;
 
+       iter->he = he;
        he_cache[iter->curr++] = he;
 
        callchain_append(he->callchain, &cursor, sample->period);
 
-       return hist_entry__inc_addr_samples(he, evsel->idx, al->addr);
+       return 0;
 }
 
 static int
 iter_finish_cumulative_entry(struct hist_entry_iter *iter,
                             struct addr_location *al __maybe_unused)
 {
-       struct perf_evsel *evsel = iter->evsel;
-       struct perf_sample *sample = iter->sample;
-
-       evsel->hists.stats.total_period += sample->period;
-       hists__inc_nr_events(&evsel->hists, PERF_RECORD_SAMPLE);
-
        zfree(&iter->priv);
+       iter->he = NULL;
        return 0;
 }
 
@@ -864,7 +856,8 @@ const struct hist_iter_ops hist_iter_cumulative = {
 
 int hist_entry_iter__add(struct hist_entry_iter *iter, struct addr_location 
*al,
                         struct perf_evsel *evsel, const union perf_event 
*event,
-                        struct perf_sample *sample, int max_stack_depth)
+                        struct perf_sample *sample, int max_stack_depth,
+                        void *arg)
 {
        int err, err2;
 
@@ -885,10 +878,22 @@ int hist_entry_iter__add(struct hist_entry_iter *iter, 
struct addr_location *al,
        if (err)
                goto out;
 
+       if (iter->he && iter->add_entry_cb) {
+               err = iter->add_entry_cb(iter, al, true, arg);
+               if (err)
+                       goto out;
+       }
+
        while (iter->ops->next_entry(iter, al)) {
                err = iter->ops->add_next_entry(iter, al);
                if (err)
                        break;
+
+               if (iter->he && iter->add_entry_cb) {
+                       err = iter->add_entry_cb(iter, al, false, arg);
+                       if (err)
+                               goto out;
+               }
        }
 
 out:
diff --git a/tools/perf/util/hist.h b/tools/perf/util/hist.h
index 1a91d536e443..2f52c4dd61de 100644
--- a/tools/perf/util/hist.h
+++ b/tools/perf/util/hist.h
@@ -115,6 +115,9 @@ struct hist_entry_iter {
        void *priv;
 
        const struct hist_iter_ops *ops;
+       /* user-defined callback function (optional) */
+       int (*add_entry_cb)(struct hist_entry_iter *iter,
+                           struct addr_location *al, bool single, void *arg);
 };
 
 extern const struct hist_iter_ops hist_iter_normal;
@@ -131,7 +134,8 @@ struct hist_entry *__hists__add_entry(struct hists *hists,
                                      bool sample_self);
 int hist_entry_iter__add(struct hist_entry_iter *iter, struct addr_location 
*al,
                         struct perf_evsel *evsel, const union perf_event 
*event,
-                        struct perf_sample *sample, int max_stack_depth);
+                        struct perf_sample *sample, int max_stack_depth,
+                        void *arg);
 
 int64_t hist_entry__cmp(struct hist_entry *left, struct hist_entry *right);
 int64_t hist_entry__collapse(struct hist_entry *left, struct hist_entry 
*right);
-- 
1.7.11.7

--
To unsubscribe from this list: send the line "unsubscribe linux-kernel" in
the body of a message to majord...@vger.kernel.org
More majordomo info at  http://vger.kernel.org/majordomo-info.html
Please read the FAQ at  http://www.tux.org/lkml/

Reply via email to