// SPDX-License-Identifier: GPL-2.0
#include <errno.h>
#include <stdlib.h>
#include <string.h>
#include "evsel.h"
#include "counts.h"
#include <perf/threadmap.h>
#include <linux/zalloc.h>

struct perf_counts *perf_counts__new(int ncpus, int nthreads)
{
        struct perf_counts *counts = zalloc(sizeof(*counts));

        if (counts) {
                struct xyarray *values;

                values = xyarray__new(ncpus, nthreads, sizeof(struct perf_counts_values));
                if (!values) {
                        free(counts);
                        return NULL;
                }

                counts->values = values;

                values = xyarray__new(ncpus, nthreads, sizeof(bool));
                if (!values) {
                        xyarray__delete(counts->values);
                        free(counts);
                        return NULL;
                }

                counts->loaded = values;
        }

        return counts;
}

void perf_counts__delete(struct perf_counts *counts)
{
        if (counts) {
                xyarray__delete(counts->loaded);
                xyarray__delete(counts->values);
                free(counts);
        }
}

void perf_counts__reset(struct perf_counts *counts)
{
        xyarray__reset(counts->loaded);
        xyarray__reset(counts->values);
}

void evsel__reset_counts(struct evsel *evsel)
{
        perf_counts__reset(evsel->counts);
}

int evsel__alloc_counts(struct evsel *evsel)
{
        struct perf_cpu_map *cpus = evsel__cpus(evsel);
        int nthreads = perf_thread_map__nr(evsel->core.threads);

        evsel->counts = perf_counts__new(perf_cpu_map__nr(cpus), nthreads);
        return evsel->counts != NULL ? 0 : -ENOMEM;
}

void evsel__free_counts(struct evsel *evsel)
{
        perf_counts__delete(evsel->counts);
        evsel->counts = NULL;
}