/* 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__ */