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

#include <drm/drm_device.h>

#include "i9xx_display_sr.h"
#include "i9xx_wm_regs.h"
#include "intel_de.h"
#include "intel_display_regs.h"
#include "intel_gmbus.h"
#include "intel_pci_config.h"

static void i9xx_display_save_swf(struct intel_display *display)
{
        int i;

        /* Scratch space */
        if (DISPLAY_VER(display) == 2 && display->platform.mobile) {
                for (i = 0; i < 7; i++) {
                        display->restore.saveSWF0[i] = intel_de_read(display, SWF0(display, i));
                        display->restore.saveSWF1[i] = intel_de_read(display, SWF1(display, i));
                }
                for (i = 0; i < 3; i++)
                        display->restore.saveSWF3[i] = intel_de_read(display, SWF3(display, i));
        } else if (DISPLAY_VER(display) == 2) {
                for (i = 0; i < 7; i++)
                        display->restore.saveSWF1[i] = intel_de_read(display, SWF1(display, i));
        } else if (HAS_GMCH(display)) {
                for (i = 0; i < 16; i++) {
                        display->restore.saveSWF0[i] = intel_de_read(display, SWF0(display, i));
                        display->restore.saveSWF1[i] = intel_de_read(display, SWF1(display, i));
                }
                for (i = 0; i < 3; i++)
                        display->restore.saveSWF3[i] = intel_de_read(display, SWF3(display, i));
        }
}

static void i9xx_display_restore_swf(struct intel_display *display)
{
        int i;

        /* Scratch space */
        if (DISPLAY_VER(display) == 2 && display->platform.mobile) {
                for (i = 0; i < 7; i++) {
                        intel_de_write(display, SWF0(display, i), display->restore.saveSWF0[i]);
                        intel_de_write(display, SWF1(display, i), display->restore.saveSWF1[i]);
                }
                for (i = 0; i < 3; i++)
                        intel_de_write(display, SWF3(display, i), display->restore.saveSWF3[i]);
        } else if (DISPLAY_VER(display) == 2) {
                for (i = 0; i < 7; i++)
                        intel_de_write(display, SWF1(display, i), display->restore.saveSWF1[i]);
        } else if (HAS_GMCH(display)) {
                for (i = 0; i < 16; i++) {
                        intel_de_write(display, SWF0(display, i), display->restore.saveSWF0[i]);
                        intel_de_write(display, SWF1(display, i), display->restore.saveSWF1[i]);
                }
                for (i = 0; i < 3; i++)
                        intel_de_write(display, SWF3(display, i), display->restore.saveSWF3[i]);
        }
}

void i9xx_display_sr_save(struct intel_display *display)
{
        struct pci_dev *pdev = to_pci_dev(display->drm->dev);

        if (!HAS_DISPLAY(display))
                return;

        /* Display arbitration control */
        if (DISPLAY_VER(display) <= 4)
                display->restore.saveDSPARB = intel_de_read(display, DSPARB(display));

        if (DISPLAY_VER(display) == 4)
                pci_read_config_word(pdev, GCDGMBUS, &display->restore.saveGCDGMBUS);

        i9xx_display_save_swf(display);
}

void i9xx_display_sr_restore(struct intel_display *display)
{
        struct pci_dev *pdev = to_pci_dev(display->drm->dev);

        if (!HAS_DISPLAY(display))
                return;

        i9xx_display_restore_swf(display);

        if (DISPLAY_VER(display) == 4)
                pci_write_config_word(pdev, GCDGMBUS, display->restore.saveGCDGMBUS);

        /* Display arbitration */
        if (DISPLAY_VER(display) <= 4)
                intel_de_write(display, DSPARB(display), display->restore.saveDSPARB);
}