// SPDX-License-Identifier: GPL-2.0-or-later
/* Hisilicon Hibmc SoC drm driver
 *
 * Based on the bochs drm driver.
 *
 * Copyright (c) 2016 Huawei Limited.
 *
 * Author:
 * Rongrong Zou <zourongrong@huawei.com>
 * Rongrong Zou <zourongrong@gmail.com>
 * Jianhua Li <lijianhua@huawei.com>
 */

#include <linux/delay.h>

#include <drm/drm_atomic.h>
#include <drm/drm_atomic_helper.h>
#include <drm/drm_fourcc.h>
#include <drm/drm_gem_vram_helper.h>
#include <drm/drm_vblank.h>

#include "hibmc_drm_drv.h"
#include "hibmc_drm_regs.h"

struct hibmc_display_panel_pll {
 u64 M;
 u64 N;
 u64 OD;
 u64 POD;
};

struct hibmc_dislay_pll_config {
 u64 hdisplay;
 u64 vdisplay;
 u32 pll1_config_value;
 u32 pll2_config_value;
};

static const struct hibmc_dislay_pll_config hibmc_pll_table[] = {
 {640, 480, CRT_PLL1_HS_25MHZ, CRT_PLL2_HS_25MHZ},
 {800, 600, CRT_PLL1_HS_40MHZ, CRT_PLL2_HS_40MHZ},
 {1024, 768, CRT_PLL1_HS_65MHZ, CRT_PLL2_HS_65MHZ},
 {1152, 864, CRT_PLL1_HS_80MHZ_1152, CRT_PLL2_HS_80MHZ},
 {1280, 768, CRT_PLL1_HS_80MHZ, CRT_PLL2_HS_80MHZ},
 {1280, 720, CRT_PLL1_HS_74MHZ, CRT_PLL2_HS_74MHZ},
 {1280, 960, CRT_PLL1_HS_108MHZ, CRT_PLL2_HS_108MHZ},
 {1280, 1024, CRT_PLL1_HS_108MHZ, CRT_PLL2_HS_108MHZ},
 {1440, 900, CRT_PLL1_HS_106MHZ, CRT_PLL2_HS_106MHZ},
 {1600, 900, CRT_PLL1_HS_108MHZ, CRT_PLL2_HS_108MHZ},
 {1600, 1200, CRT_PLL1_HS_162MHZ, CRT_PLL2_HS_162MHZ},
 {1920, 1080, CRT_PLL1_HS_148MHZ, CRT_PLL2_HS_148MHZ},
 {1920, 1200, CRT_PLL1_HS_193MHZ, CRT_PLL2_HS_193MHZ},
};

static int hibmc_plane_atomic_check(struct drm_plane *plane,
        struct drm_atomic_state *state)
{
 struct drm_plane_state *new_plane_state = drm_atomic_get_new_plane_state(state,
           plane);
 struct drm_framebuffer *fb = new_plane_state->fb;
 struct drm_crtc *crtc = new_plane_state->crtc;
 struct drm_crtc_state *crtc_state;
 u32 src_w = new_plane_state->src_w >> 16;
 u32 src_h = new_plane_state->src_h >> 16;

 if (!crtc || !fb)
  return 0;

 crtc_state = drm_atomic_get_crtc_state(state, crtc);
 if (IS_ERR(crtc_state))
  return PTR_ERR(crtc_state);

 if (src_w != new_plane_state->crtc_w || src_h != new_plane_state->crtc_h) {
  drm_dbg_atomic(plane->dev, "scale not support\n");
  return -EINVAL;
 }

 if (new_plane_state->crtc_x < 0 || new_plane_state->crtc_y < 0) {
  drm_dbg_atomic(plane->dev, "crtc_x/y of drm_plane state is invalid\n");
  return -EINVAL;
 }

 if (!crtc_state->enable)
  return 0;

 if (new_plane_state->crtc_x + new_plane_state->crtc_w >
     crtc_state->adjusted_mode.hdisplay ||
     new_plane_state->crtc_y + new_plane_state->crtc_h >
     crtc_state->adjusted_mode.vdisplay) {
  drm_dbg_atomic(plane->dev, "visible portion of plane is invalid\n");
  return -EINVAL;
 }

 if (new_plane_state->fb->pitches[0] % 128 != 0) {
  drm_dbg_atomic(plane->dev, "wrong stride with 128-byte aligned\n");
  return -EINVAL;
 }
 return 0;
}

static void hibmc_plane_atomic_update(struct drm_plane *plane,
          struct drm_atomic_state *state)
{
 struct drm_plane_state *new_state = drm_atomic_get_new_plane_state(state,
            plane);
 u32 reg;
 s64 gpu_addr = 0;
 u32 line_l;
 struct hibmc_drm_private *priv = to_hibmc_drm_private(plane->dev);
 struct drm_gem_vram_object *gbo;

 if (!new_state->fb)
  return;

 gbo = drm_gem_vram_of_gem(new_state->fb->obj[0]);

 gpu_addr = drm_gem_vram_offset(gbo);
 if (WARN_ON_ONCE(gpu_addr < 0))
  return; /* Bug: we didn't pin the BO to VRAM in prepare_fb. */

 writel(gpu_addr, priv->mmio + HIBMC_CRT_FB_ADDRESS);

 reg = new_state->fb->width * (new_state->fb->format->cpp[0]);

 line_l = new_state->fb->pitches[0];
 writel(HIBMC_FIELD(HIBMC_CRT_FB_WIDTH_WIDTH, reg) |
        HIBMC_FIELD(HIBMC_CRT_FB_WIDTH_OFFS, line_l),
        priv->mmio + HIBMC_CRT_FB_WIDTH);

 /* SET PIXEL FORMAT */
 reg = readl(priv->mmio + HIBMC_CRT_DISP_CTL);
 reg &= ~HIBMC_CRT_DISP_CTL_FORMAT_MASK;
 reg |= HIBMC_FIELD(HIBMC_CRT_DISP_CTL_FORMAT,
      new_state->fb->format->cpp[0] * 8 / 16);
 writel(reg, priv->mmio + HIBMC_CRT_DISP_CTL);
}

static const u32 channel_formats1[] = {
 DRM_FORMAT_RGB565, DRM_FORMAT_BGR565, DRM_FORMAT_RGB888,
 DRM_FORMAT_BGR888, DRM_FORMAT_XRGB8888, DRM_FORMAT_XBGR8888,
 DRM_FORMAT_RGBA8888, DRM_FORMAT_BGRA8888, DRM_FORMAT_ARGB8888,
 DRM_FORMAT_ABGR8888
};

static const struct drm_plane_funcs hibmc_plane_funcs = {
 .update_plane = drm_atomic_helper_update_plane,
 .disable_plane = drm_atomic_helper_disable_plane,
 .destroy = drm_plane_cleanup,
 .reset = drm_atomic_helper_plane_reset,
 .atomic_duplicate_state = drm_atomic_helper_plane_duplicate_state,
 .atomic_destroy_state = drm_atomic_helper_plane_destroy_state,
};

static const struct drm_plane_helper_funcs hibmc_plane_helper_funcs = {
 DRM_GEM_VRAM_PLANE_HELPER_FUNCS,
 .atomic_check = hibmc_plane_atomic_check,
 .atomic_update = hibmc_plane_atomic_update,
};

static void hibmc_crtc_dpms(struct drm_crtc *crtc, u32 dpms)
{
 struct hibmc_drm_private *priv = to_hibmc_drm_private(crtc->dev);
 u32 reg;

 reg = readl(priv->mmio + HIBMC_CRT_DISP_CTL);
 reg &= ~HIBMC_CRT_DISP_CTL_DPMS_MASK;
 reg |= HIBMC_FIELD(HIBMC_CRT_DISP_CTL_DPMS, dpms);
 reg &= ~HIBMC_CRT_DISP_CTL_TIMING_MASK;
 if (dpms == HIBMC_CRT_DPMS_ON)
  reg |= HIBMC_CRT_DISP_CTL_TIMING(1);
 writel(reg, priv->mmio + HIBMC_CRT_DISP_CTL);
}

static void hibmc_crtc_atomic_enable(struct drm_crtc *crtc,
         struct drm_atomic_state *state)
{
 u32 reg;
 struct hibmc_drm_private *priv = to_hibmc_drm_private(crtc->dev);

 hibmc_set_power_mode(priv, HIBMC_PW_MODE_CTL_MODE_MODE0);

 /* Enable display power gate & LOCALMEM power gate*/
 reg = readl(priv->mmio + HIBMC_CURRENT_GATE);
 reg &= ~HIBMC_CURR_GATE_LOCALMEM_MASK;
 reg &= ~HIBMC_CURR_GATE_DISPLAY_MASK;
 reg |= HIBMC_CURR_GATE_LOCALMEM(1);
 reg |= HIBMC_CURR_GATE_DISPLAY(1);
 hibmc_set_current_gate(priv, reg);
 drm_crtc_vblank_on(crtc);
 hibmc_crtc_dpms(crtc, HIBMC_CRT_DPMS_ON);
}

static void hibmc_crtc_atomic_disable(struct drm_crtc *crtc,
          struct drm_atomic_state *state)
{
 u32 reg;
 struct hibmc_drm_private *priv = to_hibmc_drm_private(crtc->dev);

 hibmc_crtc_dpms(crtc, HIBMC_CRT_DPMS_OFF);
 drm_crtc_vblank_off(crtc);

 hibmc_set_power_mode(priv, HIBMC_PW_MODE_CTL_MODE_SLEEP);

 /* Enable display power gate & LOCALMEM power gate*/
 reg = readl(priv->mmio + HIBMC_CURRENT_GATE);
 reg &= ~HIBMC_CURR_GATE_LOCALMEM_MASK;
 reg &= ~HIBMC_CURR_GATE_DISPLAY_MASK;
 reg |= HIBMC_CURR_GATE_LOCALMEM(0);
 reg |= HIBMC_CURR_GATE_DISPLAY(0);
 hibmc_set_current_gate(priv, reg);
}

static enum drm_mode_status
hibmc_crtc_mode_valid(struct drm_crtc *crtc,
        const struct drm_display_mode *mode)
{
 size_t i = 0;
 int vrefresh = drm_mode_vrefresh(mode);

 if (vrefresh < 59 || vrefresh > 61)
  return MODE_NOCLOCK;

 for (i = 0; i < ARRAY_SIZE(hibmc_pll_table); i++) {
  if (hibmc_pll_table[i].hdisplay == mode->hdisplay &&
      hibmc_pll_table[i].vdisplay == mode->vdisplay)
   return MODE_OK;
 }

 return MODE_BAD;
}

static u32 format_pll_reg(void)
{
 u32 pllreg = 0;
 struct hibmc_display_panel_pll pll = {0};

 /*
  * Note that all PLL's have the same format. Here,
  * we just use Panel PLL parameter to work out the bit
  * fields in the register.On returning a 32 bit number, the value can
  * be applied to any PLL in the calling function.
 */
 pllreg |= HIBMC_FIELD(HIBMC_PLL_CTRL_BYPASS, 0);
 pllreg |= HIBMC_FIELD(HIBMC_PLL_CTRL_POWER, 1);
 pllreg |= HIBMC_FIELD(HIBMC_PLL_CTRL_INPUT, 0);
 pllreg |= HIBMC_FIELD(HIBMC_PLL_CTRL_POD, pll.POD);
 pllreg |= HIBMC_FIELD(HIBMC_PLL_CTRL_OD, pll.OD);
 pllreg |= HIBMC_FIELD(HIBMC_PLL_CTRL_N, pll.N);
 pllreg |= HIBMC_FIELD(HIBMC_PLL_CTRL_M, pll.M);

 return pllreg;
}

static void set_vclock_hisilicon(struct drm_device *dev, u64 pll)
{
 u32 val;
 struct hibmc_drm_private *priv = to_hibmc_drm_private(dev);

 val = readl(priv->mmio + CRT_PLL1_HS);
 val &= ~(CRT_PLL1_HS_OUTER_BYPASS(1));
 writel(val, priv->mmio + CRT_PLL1_HS);

 val = CRT_PLL1_HS_INTER_BYPASS(1) | CRT_PLL1_HS_POWERON(1);
 writel(val, priv->mmio + CRT_PLL1_HS);

 writel(pll, priv->mmio + CRT_PLL1_HS);

 usleep_range(1000, 2000);

 val = pll & ~(CRT_PLL1_HS_POWERON(1));
 writel(val, priv->mmio + CRT_PLL1_HS);

 usleep_range(1000, 2000);

 val &= ~(CRT_PLL1_HS_INTER_BYPASS(1));
 writel(val, priv->mmio + CRT_PLL1_HS);

 usleep_range(1000, 2000);

 val |= CRT_PLL1_HS_OUTER_BYPASS(1);
 writel(val, priv->mmio + CRT_PLL1_HS);
}

static void get_pll_config(u64 x, u64 y, u32 *pll1, u32 *pll2)
{
 size_t i;
 size_t count = ARRAY_SIZE(hibmc_pll_table);

 for (i = 0; i < count; i++) {
  if (hibmc_pll_table[i].hdisplay == x &&
      hibmc_pll_table[i].vdisplay == y) {
   *pll1 = hibmc_pll_table[i].pll1_config_value;
   *pll2 = hibmc_pll_table[i].pll2_config_value;
   return;
  }
 }

 /* if found none, we use default value */
 *pll1 = CRT_PLL1_HS_25MHZ;
 *pll2 = CRT_PLL2_HS_25MHZ;
}

/*
 * This function takes care the extra registers and bit fields required to
 * setup a mode in board.
 * Explanation about Display Control register:
 * FPGA only supports 7 predefined pixel clocks, and clock select is
 * in bit 4:0 of new register 0x802a8.
 */
static u32 display_ctrl_adjust(struct drm_device *dev,
          struct drm_display_mode *mode,
          u32 ctrl)
{
 u64 x, y;
 u32 pll1; /* bit[31:0] of PLL */
 u32 pll2; /* bit[63:32] of PLL */
 struct hibmc_drm_private *priv = to_hibmc_drm_private(dev);

 x = mode->hdisplay;
 y = mode->vdisplay;

 get_pll_config(x, y, &pll1, &pll2);
 writel(pll2, priv->mmio + CRT_PLL2_HS);
 set_vclock_hisilicon(dev, pll1);

 /*
  * Hisilicon has to set up the top-left and bottom-right
  * registers as well.
  * Note that normal chip only use those two register for
  * auto-centering mode.
 */
 writel(HIBMC_FIELD(HIBMC_CRT_AUTO_CENTERING_TL_TOP, 0) |
        HIBMC_FIELD(HIBMC_CRT_AUTO_CENTERING_TL_LEFT, 0),
        priv->mmio + HIBMC_CRT_AUTO_CENTERING_TL);

 writel(HIBMC_FIELD(HIBMC_CRT_AUTO_CENTERING_BR_BOTTOM, y - 1) |
        HIBMC_FIELD(HIBMC_CRT_AUTO_CENTERING_BR_RIGHT, x - 1),
        priv->mmio + HIBMC_CRT_AUTO_CENTERING_BR);

 /*
  * Assume common fields in ctrl have been properly set before
  * calling this function.
  * This function only sets the extra fields in ctrl.
 */

 /* Set bit 25 of display controller: Select CRT or VGA clock */
 ctrl &= ~HIBMC_CRT_DISP_CTL_CRTSELECT_MASK;
 ctrl &= ~HIBMC_CRT_DISP_CTL_CLOCK_PHASE_MASK;

 ctrl |= HIBMC_CRT_DISP_CTL_CRTSELECT(HIBMC_CRTSELECT_CRT);

 /* clock_phase_polarity is 0 */
 ctrl |= HIBMC_CRT_DISP_CTL_CLOCK_PHASE(0);

 writel(ctrl, priv->mmio + HIBMC_CRT_DISP_CTL);

 return ctrl;
}

static void hibmc_crtc_mode_set_nofb(struct drm_crtc *crtc)
{
 u32 val;
 struct drm_display_mode *mode = &crtc->state->mode;
 struct drm_device *dev = crtc->dev;
 struct hibmc_drm_private *priv = to_hibmc_drm_private(dev);
 u32 width = mode->hsync_end - mode->hsync_start;
 u32 height = mode->vsync_end - mode->vsync_start;

 writel(format_pll_reg(), priv->mmio + HIBMC_CRT_PLL_CTRL);
 writel(HIBMC_FIELD(HIBMC_CRT_HORZ_TOTAL_TOTAL, mode->htotal - 1) |
        HIBMC_FIELD(HIBMC_CRT_HORZ_TOTAL_DISP_END, mode->hdisplay - 1),
        priv->mmio + HIBMC_CRT_HORZ_TOTAL);

 writel(HIBMC_FIELD(HIBMC_CRT_HORZ_SYNC_WIDTH, width) |
        HIBMC_FIELD(HIBMC_CRT_HORZ_SYNC_START, mode->hsync_start - 1),
        priv->mmio + HIBMC_CRT_HORZ_SYNC);

 writel(HIBMC_FIELD(HIBMC_CRT_VERT_TOTAL_TOTAL, mode->vtotal - 1) |
        HIBMC_FIELD(HIBMC_CRT_VERT_TOTAL_DISP_END, mode->vdisplay - 1),
        priv->mmio + HIBMC_CRT_VERT_TOTAL);

 writel(HIBMC_FIELD(HIBMC_CRT_VERT_SYNC_HEIGHT, height) |
        HIBMC_FIELD(HIBMC_CRT_VERT_SYNC_START, mode->vsync_start - 1),
        priv->mmio + HIBMC_CRT_VERT_SYNC);

 val = HIBMC_FIELD(HIBMC_CRT_DISP_CTL_VSYNC_PHASE, 0);
 val |= HIBMC_FIELD(HIBMC_CRT_DISP_CTL_HSYNC_PHASE, 0);
 val |= HIBMC_CRT_DISP_CTL_TIMING(1);
 val |= HIBMC_CRT_DISP_CTL_PLANE(1);

 display_ctrl_adjust(dev, mode, val);
}

static void hibmc_crtc_atomic_begin(struct drm_crtc *crtc,
        struct drm_atomic_state *state)
{
 u32 reg;
 struct drm_device *dev = crtc->dev;
 struct hibmc_drm_private *priv = to_hibmc_drm_private(dev);

 hibmc_set_power_mode(priv, HIBMC_PW_MODE_CTL_MODE_MODE0);

 /* Enable display power gate & LOCALMEM power gate*/
 reg = readl(priv->mmio + HIBMC_CURRENT_GATE);
 reg &= ~HIBMC_CURR_GATE_DISPLAY_MASK;
 reg &= ~HIBMC_CURR_GATE_LOCALMEM_MASK;
 reg |= HIBMC_CURR_GATE_DISPLAY(1);
 reg |= HIBMC_CURR_GATE_LOCALMEM(1);
 hibmc_set_current_gate(priv, reg);

 /* We can add more initialization as needed. */
}

static void hibmc_crtc_atomic_flush(struct drm_crtc *crtc,
        struct drm_atomic_state *state)

{
 unsigned long flags;

 spin_lock_irqsave(&crtc->dev->event_lock, flags);
 if (crtc->state->event)
  drm_crtc_send_vblank_event(crtc, crtc->state->event);
 crtc->state->event = NULL;
 spin_unlock_irqrestore(&crtc->dev->event_lock, flags);
}

static int hibmc_crtc_enable_vblank(struct drm_crtc *crtc)
{
 struct hibmc_drm_private *priv = to_hibmc_drm_private(crtc->dev);

 writel(HIBMC_RAW_INTERRUPT_EN_VBLANK(1),
        priv->mmio + HIBMC_RAW_INTERRUPT_EN);

 return 0;
}

static void hibmc_crtc_disable_vblank(struct drm_crtc *crtc)
{
 struct hibmc_drm_private *priv = to_hibmc_drm_private(crtc->dev);

 writel(HIBMC_RAW_INTERRUPT_EN_VBLANK(0),
        priv->mmio + HIBMC_RAW_INTERRUPT_EN);
}

static void hibmc_crtc_load_lut(struct drm_crtc *crtc)
{
 struct hibmc_drm_private *priv = to_hibmc_drm_private(crtc->dev);
 void __iomem   *mmio = priv->mmio;
 u16 *r, *g, *b;
 u32 reg;
 u32 i;

 r = crtc->gamma_store;
 g = r + crtc->gamma_size;
 b = g + crtc->gamma_size;

 for (i = 0; i < crtc->gamma_size; i++) {
  u32 offset = i << 2;
  u8 red = *r++ >> 8;
  u8 green = *g++ >> 8;
  u8 blue = *b++ >> 8;
  u32 rgb = (red << 16) | (green << 8) | blue;

  writel(rgb, mmio + HIBMC_CRT_PALETTE + offset);
 }

 reg = readl(priv->mmio + HIBMC_CRT_DISP_CTL);
 reg |= HIBMC_FIELD(HIBMC_CTL_DISP_CTL_GAMMA, 1);
 writel(reg, priv->mmio + HIBMC_CRT_DISP_CTL);
}

static int hibmc_crtc_gamma_set(struct drm_crtc *crtc, u16 *red, u16 *green,
    u16 *blue, uint32_t size,
    struct drm_modeset_acquire_ctx *ctx)
{
 hibmc_crtc_load_lut(crtc);

 return 0;
}

static const struct drm_crtc_funcs hibmc_crtc_funcs = {
 .page_flip = drm_atomic_helper_page_flip,
 .set_config = drm_atomic_helper_set_config,
 .destroy = drm_crtc_cleanup,
 .reset = drm_atomic_helper_crtc_reset,
 .atomic_duplicate_state =  drm_atomic_helper_crtc_duplicate_state,
 .atomic_destroy_state = drm_atomic_helper_crtc_destroy_state,
 .enable_vblank = hibmc_crtc_enable_vblank,
 .disable_vblank = hibmc_crtc_disable_vblank,
 .gamma_set = hibmc_crtc_gamma_set,
};

static const struct drm_crtc_helper_funcs hibmc_crtc_helper_funcs = {
 .mode_set_nofb = hibmc_crtc_mode_set_nofb,
 .atomic_begin = hibmc_crtc_atomic_begin,
 .atomic_flush = hibmc_crtc_atomic_flush,
 .atomic_enable = hibmc_crtc_atomic_enable,
 .atomic_disable = hibmc_crtc_atomic_disable,
 .mode_valid = hibmc_crtc_mode_valid,
};

int hibmc_de_init(struct hibmc_drm_private *priv)
{
 struct drm_device *dev = &priv->dev;
 struct drm_crtc *crtc = &priv->crtc;
 struct drm_plane *plane = &priv->primary_plane;
 int ret;

 ret = drm_universal_plane_init(dev, plane, 1, &hibmc_plane_funcs,
           channel_formats1,
           ARRAY_SIZE(channel_formats1),
           NULL,
           DRM_PLANE_TYPE_PRIMARY,
           NULL);

 if (ret) {
  drm_err(dev, "failed to init plane: %d\n", ret);
  return ret;
 }

 drm_plane_helper_add(plane, &hibmc_plane_helper_funcs);

 ret = drm_crtc_init_with_planes(dev, crtc, plane,
     NULL, &hibmc_crtc_funcs, NULL);
 if (ret) {
  drm_err(dev, "failed to init crtc: %d\n", ret);
  return ret;
 }

 ret = drm_mode_crtc_set_gamma_size(crtc, 256);
 if (ret) {
  drm_err(dev, "failed to set gamma size: %d\n", ret);
  return ret;
 }
 drm_crtc_helper_add(crtc, &hibmc_crtc_helper_funcs);

 return 0;
}