// 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);
}