// SPDX-License-Identifier: GPL-2.0 OR MIT
/* Copyright (c) 2026 Imagination Technologies Ltd. */

#include "pvr_device.h"
#include "pvr_dump.h"
#include "pvr_rogue_fwif.h"

#include <drm/drm_print.h>
#include <linux/types.h>

static const char *
get_reset_reason_desc(enum rogue_context_reset_reason reason)
{
	switch (reason) {
	case ROGUE_CONTEXT_RESET_REASON_NONE:
		return "None";
	case ROGUE_CONTEXT_RESET_REASON_GUILTY_LOCKUP:
		return "Guilty lockup";
	case ROGUE_CONTEXT_RESET_REASON_INNOCENT_LOCKUP:
		return "Innocent lockup";
	case ROGUE_CONTEXT_RESET_REASON_GUILTY_OVERRUNING:
		return "Guilty overrunning";
	case ROGUE_CONTEXT_RESET_REASON_INNOCENT_OVERRUNING:
		return "Innocent overrunning";
	case ROGUE_CONTEXT_RESET_REASON_HARD_CONTEXT_SWITCH:
		return "Hard context switch";
	case ROGUE_CONTEXT_RESET_REASON_WGP_CHECKSUM:
		return "CDM Mission/safety checksum mismatch";
	case ROGUE_CONTEXT_RESET_REASON_TRP_CHECKSUM:
		return "TRP checksum mismatch";
	case ROGUE_CONTEXT_RESET_REASON_GPU_ECC_OK:
		return "GPU ECC error (corrected, OK)";
	case ROGUE_CONTEXT_RESET_REASON_GPU_ECC_HWR:
		return "GPU ECC error (uncorrected, HWR)";
	case ROGUE_CONTEXT_RESET_REASON_FW_ECC_OK:
		return "Firmware ECC error (corrected, OK)";
	case ROGUE_CONTEXT_RESET_REASON_FW_ECC_ERR:
		return "Firmware ECC error (uncorrected, ERR)";
	case ROGUE_CONTEXT_RESET_REASON_FW_WATCHDOG:
		return "Firmware watchdog";
	case ROGUE_CONTEXT_RESET_REASON_FW_PAGEFAULT:
		return "Firmware pagefault";
	case ROGUE_CONTEXT_RESET_REASON_FW_EXEC_ERR:
		return "Firmware execution error";
	case ROGUE_CONTEXT_RESET_REASON_HOST_WDG_FW_ERR:
		return "Host watchdog";
	case ROGUE_CONTEXT_GEOM_OOM_DISABLED:
		return "Geometry OOM disabled";

	default:
		return "Unknown";
	}
}

static const char *
get_dm_name(u32 dm)
{
	switch (dm) {
	case PVR_FWIF_DM_GP:
		return "General purpose";
	/* PVR_FWIF_DM_TDM has the same index, but is discriminated by a device feature */
	case PVR_FWIF_DM_2D:
		return "2D or TDM";
	case PVR_FWIF_DM_GEOM:
		return "Geometry";
	case PVR_FWIF_DM_FRAG:
		return "Fragment";
	case PVR_FWIF_DM_CDM:
		return "Compute";
	case PVR_FWIF_DM_RAY:
		return "Raytracing";
	case PVR_FWIF_DM_GEOM2:
		return "Geometry 2";
	case PVR_FWIF_DM_GEOM3:
		return "Geometry 3";
	case PVR_FWIF_DM_GEOM4:
		return "Geometry 4";

	default:
		return "Unknown";
	}
}

/**
 * pvr_dump_context_reset_notification() - Handle context reset notification from FW
 * @pvr_dev: Device pointer.
 * @data: Data provided by FW.
 *
 * This will decode the data structure provided by FW and print the results via drm_info().
 */
void
pvr_dump_context_reset_notification(struct pvr_device *pvr_dev,
				    struct rogue_fwif_fwccb_cmd_context_reset_data *data)
{
	struct drm_device *drm_dev = from_pvr_device(pvr_dev);

	if (data->flags & ROGUE_FWIF_FWCCB_CMD_CONTEXT_RESET_FLAG_ALL_CTXS) {
		drm_info(drm_dev, "Received context reset notification for all contexts\n");
	} else {
		drm_info(drm_dev, "Received context reset notification on context %u\n",
			 data->server_common_context_id);
	}

	drm_info(drm_dev, "  Reset reason=%u (%s)\n", data->reset_reason,
		 get_reset_reason_desc((enum rogue_context_reset_reason)data->reset_reason));
	drm_info(drm_dev, "  Data Master=%u (%s)\n", data->dm, get_dm_name(data->dm));
	drm_info(drm_dev, "  Job ref=%u\n", data->reset_job_ref);

	if (data->flags & ROGUE_FWIF_FWCCB_CMD_CONTEXT_RESET_FLAG_PF) {
		drm_info(drm_dev, "  Page fault occurred, fault address=%llx\n",
			 data->fault_address);
	}
}