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

#include <linux/slab.h>

#include <drm/drm_drv.h>

#include "intel_display_core.h"
#include "intel_display_device.h"
#include "intel_display_irq.h"
#include "intel_display_params.h"
#include "intel_display_snapshot.h"
#include "intel_dmc.h"
#include "intel_overlay.h"

struct intel_display_snapshot {
        struct intel_display *display;

        struct intel_display_device_info info;
        struct intel_display_runtime_info runtime_info;
        struct intel_display_params params;
        struct intel_overlay_snapshot *overlay;
        struct intel_dmc_snapshot *dmc;
        struct intel_display_irq_snapshot *irq;
};

struct intel_display_snapshot *intel_display_snapshot_capture(struct intel_display *display)
{
        struct intel_display_snapshot *snapshot;

        snapshot = kzalloc_obj(*snapshot, GFP_ATOMIC);
        if (!snapshot)
                return NULL;

        snapshot->display = display;

        memcpy(&snapshot->info, DISPLAY_INFO(display), sizeof(snapshot->info));
        memcpy(&snapshot->runtime_info, DISPLAY_RUNTIME_INFO(display),
               sizeof(snapshot->runtime_info));

        intel_display_params_copy(&snapshot->params);

        snapshot->irq = intel_display_irq_snapshot_capture(display);
        snapshot->overlay = intel_overlay_snapshot_capture(display);
        snapshot->dmc = intel_dmc_snapshot_capture(display);

        return snapshot;
}

void intel_display_snapshot_print(const struct intel_display_snapshot *snapshot,
                                  struct drm_printer *p)
{
        struct intel_display *display;

        if (!snapshot)
                return;

        display = snapshot->display;

        intel_display_device_info_print(&snapshot->info, &snapshot->runtime_info, p);
        intel_display_params_dump(&snapshot->params, display->drm->driver->name, p);

        intel_display_irq_snapshot_print(snapshot->irq, p);
        intel_overlay_snapshot_print(snapshot->overlay, p);
        intel_dmc_snapshot_print(snapshot->dmc, p);
}

void intel_display_snapshot_free(struct intel_display_snapshot *snapshot)
{
        if (!snapshot)
                return;

        intel_display_params_free(&snapshot->params);

        kfree(snapshot->irq);
        kfree(snapshot->overlay);
        kfree(snapshot->dmc);
        kfree(snapshot);
}