/* SPDX-License-Identifier: GPL-2.0 */
/*
 * Copyright (c) 2024 Raspberry Pi Ltd.
 * Copyright (c) 2024 Ideas on Board Oy
 */

#undef TRACE_SYSTEM
#define TRACE_SYSTEM cfe

#if !defined(_CFE_TRACE_H) || defined(TRACE_HEADER_MULTI_READ)
#define _CFE_TRACE_H

#include <linux/tracepoint.h>
#include <media/videobuf2-v4l2.h>

TRACE_EVENT(cfe_return_buffer,
	TP_PROTO(u32 node_id, u32 buf_idx, u32 queue_id),
	TP_ARGS(node_id, buf_idx, queue_id),
	TP_STRUCT__entry(
		__field(u32, node_id)
		__field(u32, buf_idx)
		__field(u32, queue_id)
	),
	TP_fast_assign(
		__entry->node_id = node_id;
		__entry->buf_idx = buf_idx;
		__entry->queue_id = queue_id;
	),
	TP_printk("node=%u buf=%u, queue=%u", __entry->node_id,
		  __entry->buf_idx, __entry->queue_id)
);

DECLARE_EVENT_CLASS(cfe_buffer_template,
	TP_PROTO(u32 node_id, struct vb2_buffer *buf),
	TP_ARGS(node_id, buf),
	TP_STRUCT__entry(
		__field(u32, node_id)
		__field(u32, buf_idx)
	),
	TP_fast_assign(
		__entry->node_id = node_id;
		__entry->buf_idx = buf->index;
	),
	TP_printk("node=%u buf=%u", __entry->node_id, __entry->buf_idx)
);

DEFINE_EVENT(cfe_buffer_template, cfe_buffer_prepare,
	TP_PROTO(u32 node_id, struct vb2_buffer *buf),
	TP_ARGS(node_id, buf));

TRACE_EVENT(cfe_buffer_queue,
	TP_PROTO(u32 node_id, struct vb2_buffer *buf, bool schedule_now),
	TP_ARGS(node_id, buf, schedule_now),
	TP_STRUCT__entry(
		__field(u32, node_id)
		__field(u32, buf_idx)
		__field(bool, schedule_now)
	),
	TP_fast_assign(
		__entry->node_id = node_id;
		__entry->buf_idx = buf->index;
		__entry->schedule_now = schedule_now;
	),
	TP_printk("node=%u buf=%u%s", __entry->node_id, __entry->buf_idx,
		  __entry->schedule_now ? " schedule immediately" : "")
);

DEFINE_EVENT(cfe_buffer_template, cfe_csi2_schedule,
	TP_PROTO(u32 node_id, struct vb2_buffer *buf),
	TP_ARGS(node_id, buf));

DEFINE_EVENT(cfe_buffer_template, cfe_fe_schedule,
	TP_PROTO(u32 node_id, struct vb2_buffer *buf),
	TP_ARGS(node_id, buf));

TRACE_EVENT(cfe_buffer_complete,
	TP_PROTO(u32 node_id, struct vb2_v4l2_buffer *buf),
	TP_ARGS(node_id, buf),
	TP_STRUCT__entry(
		__field(u32, node_id)
		__field(u32, buf_idx)
		__field(u32, seq)
		__field(u64, ts)
	),
	TP_fast_assign(
		__entry->node_id = node_id;
		__entry->buf_idx = buf->vb2_buf.index;
		__entry->seq = buf->sequence;
		__entry->ts = buf->vb2_buf.timestamp;
	),
	TP_printk("node=%u buf=%u seq=%u ts=%llu", __entry->node_id,
		  __entry->buf_idx, __entry->seq, __entry->ts)
);

TRACE_EVENT(cfe_frame_start,
	TP_PROTO(u32 node_id, u32 fs_count),
	TP_ARGS(node_id, fs_count),
	TP_STRUCT__entry(
		__field(u32, node_id)
		__field(u32, fs_count)
	),
	TP_fast_assign(
		__entry->node_id = node_id;
		__entry->fs_count = fs_count;
	),
	TP_printk("node=%u fs_count=%u", __entry->node_id, __entry->fs_count)
);

TRACE_EVENT(cfe_frame_end,
	TP_PROTO(u32 node_id, u32 fs_count),
	TP_ARGS(node_id, fs_count),
	TP_STRUCT__entry(
		__field(u32, node_id)
		__field(u32, fs_count)
	),
	TP_fast_assign(
		__entry->node_id = node_id;
		__entry->fs_count = fs_count;
	),
	TP_printk("node=%u fs_count=%u", __entry->node_id, __entry->fs_count)
);

TRACE_EVENT(cfe_prepare_next_job,
	TP_PROTO(bool fe_enabled),
	TP_ARGS(fe_enabled),
	TP_STRUCT__entry(
		__field(bool, fe_enabled)
	),
	TP_fast_assign(
		__entry->fe_enabled = fe_enabled;
	),
	TP_printk("fe_enabled=%u", __entry->fe_enabled)
);

/* These are copied from csi2.c */
#define CSI2_STATUS_IRQ_FS(x)			(BIT(0) << (x))
#define CSI2_STATUS_IRQ_FE(x)			(BIT(4) << (x))
#define CSI2_STATUS_IRQ_FE_ACK(x)		(BIT(8) << (x))
#define CSI2_STATUS_IRQ_LE(x)			(BIT(12) << (x))
#define CSI2_STATUS_IRQ_LE_ACK(x)		(BIT(16) << (x))

TRACE_EVENT(csi2_irq,
	TP_PROTO(u32 channel, u32 status, u32 dbg),
	TP_ARGS(channel, status, dbg),
	TP_STRUCT__entry(
		__field(u32, channel)
		__field(u32, status)
		__field(u32, dbg)
	),
	TP_fast_assign(
		__entry->channel = channel;
		__entry->status = status;
		__entry->dbg = dbg;
	),
	TP_printk("ch=%u flags=[ %s%s%s%s%s] frame=%u line=%u\n",
		  __entry->channel,
		  (__entry->status & CSI2_STATUS_IRQ_FS(__entry->channel)) ?
			"FS " : "",
		  (__entry->status & CSI2_STATUS_IRQ_FE(__entry->channel)) ?
			"FE " : "",
		  (__entry->status & CSI2_STATUS_IRQ_FE_ACK(__entry->channel)) ?
			"FE_ACK " : "",
		  (__entry->status & CSI2_STATUS_IRQ_LE(__entry->channel)) ?
			"LE " : "",
		  (__entry->status & CSI2_STATUS_IRQ_LE_ACK(__entry->channel)) ?
			"LE_ACK " : "",
		  __entry->dbg >> 16, __entry->dbg & 0xffff)
);

TRACE_EVENT(fe_irq,
	TP_PROTO(u32 status, u32 output_status, u32 frame_status,
		 u32 error_status, u32 int_status),
	TP_ARGS(status, output_status, frame_status, error_status, int_status),
	TP_STRUCT__entry(
		__field(u32, status)
		__field(u32, output_status)
		__field(u32, frame_status)
		__field(u32, error_status)
		__field(u32, int_status)
	),
	TP_fast_assign(
		__entry->status = status;
		__entry->output_status = output_status;
		__entry->frame_status = frame_status;
		__entry->error_status = error_status;
		__entry->int_status = int_status;
	),
	TP_printk("status 0x%x out_status 0x%x frame_status 0x%x error_status 0x%x int_status 0x%x",
		  __entry->status,
		  __entry->output_status,
		  __entry->frame_status,
		  __entry->error_status,
		  __entry->int_status)
);

#endif /* _CFE_TRACE_H */

/* This part must be outside protection */
#undef TRACE_INCLUDE_PATH
#define TRACE_INCLUDE_PATH .
#define TRACE_INCLUDE_FILE ../../drivers/media/platform/raspberrypi/rp1-cfe/cfe-trace
#include <trace/define_trace.h>