/* SPDX-License-Identifier: MIT */
/*
 * Copyright © 2020 Intel Corporation
 */

#ifndef __INTEL_ENGINE_STATS_H__
#define __INTEL_ENGINE_STATS_H__

#include <linux/atomic.h>
#include <linux/ktime.h>
#include <linux/seqlock.h>

#include "i915_gem.h" /* GEM_BUG_ON */
#include "intel_engine.h"

static inline void intel_engine_context_in(struct intel_engine_cs *engine)
{
        struct intel_engine_execlists_stats *stats = &engine->stats.execlists;
        unsigned long flags;

        if (stats->active) {
                stats->active++;
                return;
        }

        /* The writer is serialised; but the pmu reader may be from hardirq */
        local_irq_save(flags);
        write_seqcount_begin(&stats->lock);

        stats->start = ktime_get();
        stats->active++;

        write_seqcount_end(&stats->lock);
        local_irq_restore(flags);

        GEM_BUG_ON(!stats->active);
}

static inline void intel_engine_context_out(struct intel_engine_cs *engine)
{
        struct intel_engine_execlists_stats *stats = &engine->stats.execlists;
        unsigned long flags;

        GEM_BUG_ON(!stats->active);
        if (stats->active > 1) {
                stats->active--;
                return;
        }

        local_irq_save(flags);
        write_seqcount_begin(&stats->lock);

        stats->active--;
        stats->total = ktime_add(stats->total,
                                 ktime_sub(ktime_get(), stats->start));

        write_seqcount_end(&stats->lock);
        local_irq_restore(flags);
}

#endif /* __INTEL_ENGINE_STATS_H__ */