// SPDX-License-Identifier: GPL-2.0
#include <inttypes.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <errno.h>
#include <linux/zalloc.h>

#include "values.h"
#include "debug.h"
#include "evsel.h"

int perf_read_values_init(struct perf_read_values *values)
{
        values->threads_max = 16;
        values->pid = malloc(values->threads_max * sizeof(*values->pid));
        values->tid = malloc(values->threads_max * sizeof(*values->tid));
        values->value = zalloc(values->threads_max * sizeof(*values->value));
        if (!values->pid || !values->tid || !values->value) {
                pr_debug("failed to allocate read_values threads arrays");
                goto out_free_pid;
        }
        values->threads = 0;

        values->counters_max = 16;
        values->counters = malloc(values->counters_max * sizeof(*values->counters));
        if (!values->counters) {
                pr_debug("failed to allocate read_values counters array");
                goto out_free_counter;
        }
        values->num_counters = 0;

        return 0;

out_free_counter:
        zfree(&values->counters);
out_free_pid:
        zfree(&values->pid);
        zfree(&values->tid);
        zfree(&values->value);
        return -ENOMEM;
}

void perf_read_values_destroy(struct perf_read_values *values)
{
        int i;

        if (!values->threads_max || !values->counters_max)
                return;

        for (i = 0; i < values->threads; i++)
                zfree(&values->value[i]);
        zfree(&values->value);
        zfree(&values->pid);
        zfree(&values->tid);
        zfree(&values->counters);
}

static int perf_read_values__enlarge_threads(struct perf_read_values *values)
{
        int nthreads_max = values->threads_max * 2;
        void *npid = realloc(values->pid, nthreads_max * sizeof(*values->pid)),
             *ntid = realloc(values->tid, nthreads_max * sizeof(*values->tid)),
             *nvalue = realloc(values->value, nthreads_max * sizeof(*values->value));

        if (!npid || !ntid || !nvalue)
                goto out_err;

        values->threads_max = nthreads_max;
        values->pid = npid;
        values->tid = ntid;
        values->value = nvalue;
        return 0;
out_err:
        free(npid);
        free(ntid);
        free(nvalue);
        pr_debug("failed to enlarge read_values threads arrays");
        return -ENOMEM;
}

static int perf_read_values__findnew_thread(struct perf_read_values *values,
                                            u32 pid, u32 tid)
{
        int i;

        for (i = 0; i < values->threads; i++)
                if (values->pid[i] == pid && values->tid[i] == tid)
                        return i;

        if (values->threads == values->threads_max) {
                i = perf_read_values__enlarge_threads(values);
                if (i < 0)
                        return i;
        }

        i = values->threads;

        values->value[i] = zalloc(values->counters_max * sizeof(**values->value));
        if (!values->value[i]) {
                pr_debug("failed to allocate read_values counters array");
                return -ENOMEM;
        }
        values->pid[i] = pid;
        values->tid[i] = tid;
        values->threads = i + 1;

        return i;
}

static int perf_read_values__enlarge_counters(struct perf_read_values *values)
{
        int counters_max = values->counters_max * 2;
        struct evsel **new_counters = realloc(values->counters,
                                             counters_max * sizeof(*values->counters));

        if (!new_counters) {
                pr_debug("failed to enlarge read_values counters array");
                goto out_enomem;
        }

        for (int i = 0; i < values->threads; i++) {
                u64 *value = realloc(values->value[i], counters_max * sizeof(**values->value));

                if (!value) {
                        pr_debug("failed to enlarge read_values ->values array");
                        goto out_free_counters;
                }

                for (int j = values->counters_max; j < counters_max; j++)
                        value[j] = 0;

                values->value[i] = value;
        }

        values->counters_max = counters_max;
        values->counters = new_counters;

        return 0;
out_free_counters:
        free(new_counters);
out_enomem:
        return -ENOMEM;
}

static int perf_read_values__findnew_counter(struct perf_read_values *values,
                                             struct evsel *evsel)
{
        int i;

        for (i = 0; i < values->num_counters; i++)
                if (values->counters[i] == evsel)
                        return i;

        if (values->num_counters == values->counters_max) {
                int err = perf_read_values__enlarge_counters(values);

                if (err)
                        return err;
        }

        i = values->num_counters++;
        values->counters[i] = evsel;

        return i;
}

int perf_read_values_add_value(struct perf_read_values *values,
                                u32 pid, u32 tid,
                                struct evsel *evsel, u64 value)
{
        int tindex, cindex;

        tindex = perf_read_values__findnew_thread(values, pid, tid);
        if (tindex < 0)
                return tindex;
        cindex = perf_read_values__findnew_counter(values, evsel);
        if (cindex < 0)
                return cindex;

        values->value[tindex][cindex] += value;
        return 0;
}

static void perf_read_values__display_pretty(FILE *fp,
                                             struct perf_read_values *values)
{
        int i, j;
        int pidwidth, tidwidth;
        int *counterwidth;

        counterwidth = malloc(values->num_counters * sizeof(*counterwidth));
        if (!counterwidth) {
                fprintf(fp, "INTERNAL ERROR: Failed to allocate counterwidth array\n");
                return;
        }
        tidwidth = 3;
        pidwidth = 3;
        for (j = 0; j < values->num_counters; j++)
                counterwidth[j] = strlen(evsel__name(values->counters[j]));
        for (i = 0; i < values->threads; i++) {
                int width;

                width = snprintf(NULL, 0, "%d", values->pid[i]);
                if (width > pidwidth)
                        pidwidth = width;
                width = snprintf(NULL, 0, "%d", values->tid[i]);
                if (width > tidwidth)
                        tidwidth = width;
                for (j = 0; j < values->num_counters; j++) {
                        width = snprintf(NULL, 0, "%" PRIu64, values->value[i][j]);
                        if (width > counterwidth[j])
                                counterwidth[j] = width;
                }
        }

        fprintf(fp, "# %*s  %*s", pidwidth, "PID", tidwidth, "TID");
        for (j = 0; j < values->num_counters; j++)
                fprintf(fp, "  %*s", counterwidth[j], evsel__name(values->counters[j]));
        fprintf(fp, "\n");

        for (i = 0; i < values->threads; i++) {
                fprintf(fp, "  %*d  %*d", pidwidth, values->pid[i],
                        tidwidth, values->tid[i]);
                for (j = 0; j < values->num_counters; j++)
                        fprintf(fp, "  %*" PRIu64,
                                counterwidth[j], values->value[i][j]);
                fprintf(fp, "\n");
        }
        free(counterwidth);
}

static void perf_read_values__display_raw(FILE *fp,
                                          struct perf_read_values *values)
{
        int width, pidwidth, tidwidth, namewidth, rawwidth, countwidth;
        int i, j;

        tidwidth = 3; /* TID */
        pidwidth = 3; /* PID */
        namewidth = 4; /* "Name" */
        rawwidth = 3; /* "Raw" */
        countwidth = 5; /* "Count" */

        for (i = 0; i < values->threads; i++) {
                width = snprintf(NULL, 0, "%d", values->pid[i]);
                if (width > pidwidth)
                        pidwidth = width;
                width = snprintf(NULL, 0, "%d", values->tid[i]);
                if (width > tidwidth)
                        tidwidth = width;
        }
        for (j = 0; j < values->num_counters; j++) {
                width = strlen(evsel__name(values->counters[j]));
                if (width > namewidth)
                        namewidth = width;
                width = snprintf(NULL, 0, "%x", values->counters[j]->core.idx);
                if (width > rawwidth)
                        rawwidth = width;
        }
        for (i = 0; i < values->threads; i++) {
                for (j = 0; j < values->num_counters; j++) {
                        width = snprintf(NULL, 0, "%" PRIu64, values->value[i][j]);
                        if (width > countwidth)
                                countwidth = width;
                }
        }

        fprintf(fp, "# %*s  %*s  %*s  %*s  %*s\n",
                pidwidth, "PID", tidwidth, "TID",
                namewidth, "Name", rawwidth, "Raw",
                countwidth, "Count");
        for (i = 0; i < values->threads; i++)
                for (j = 0; j < values->num_counters; j++)
                        fprintf(fp, "  %*d  %*d  %*s  %*x  %*" PRIu64,
                                pidwidth, values->pid[i],
                                tidwidth, values->tid[i],
                                namewidth, evsel__name(values->counters[j]),
                                rawwidth, values->counters[j]->core.idx,
                                countwidth, values->value[i][j]);
}

void perf_read_values_display(FILE *fp, struct perf_read_values *values, int raw)
{
        if (raw)
                perf_read_values__display_raw(fp, values);
        else
                perf_read_values__display_pretty(fp, values);
}