/*
 * Copyright 2018 Red Hat Inc.
 *
 * Permission is hereby granted, free of charge, to any person obtaining a
 * copy of this software and associated documentation files (the "Software"),
 * to deal in the Software without restriction, including without limitation
 * the rights to use, copy, modify, merge, publish, distribute, sublicense,
 * and/or sell copies of the Software, and to permit persons to whom the
 * Software is furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
 * THE COPYRIGHT HOLDER(S) OR AUTHOR(S) BE LIABLE FOR ANY CLAIM, DAMAGES OR
 * OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
 * ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
 * OTHER DEALINGS IN THE SOFTWARE.
 */
#include "wndw.h"
#include "atom.h"

#include <drm/drm_atomic_helper.h>
#include <nouveau_bo.h>

#include <nvif/pushc37b.h>

#include <nvhw/class/clc57e.h>

static int
wndwc57e_image_set(struct nv50_wndw *wndw, struct nv50_wndw_atom *asyw)
{
 struct nvif_push *push = &wndw->wndw.push;
 int ret;

 if ((ret = PUSH_WAIT(push, 17)))
  return ret;

 PUSH_MTHD(push, NVC57E, SET_PRESENT_CONTROL,
    NVVAL(NVC57E, SET_PRESENT_CONTROL, MIN_PRESENT_INTERVAL, asyw->image.interval) |
    NVVAL(NVC57E, SET_PRESENT_CONTROL, BEGIN_MODE, asyw->image.mode) |
    NVDEF(NVC57E, SET_PRESENT_CONTROL, TIMESTAMP_MODE, DISABLE));

 PUSH_MTHD(push, NVC57E, SET_SIZE,
    NVVAL(NVC57E, SET_SIZE, WIDTH, asyw->image.w) |
    NVVAL(NVC57E, SET_SIZE, HEIGHT, asyw->image.h),

    SET_STORAGE,
    NVVAL(NVC57E, SET_STORAGE, BLOCK_HEIGHT, asyw->image.blockh) |
    NVVAL(NVC57E, SET_STORAGE, MEMORY_LAYOUT, asyw->image.layout),

    SET_PARAMS,
    NVVAL(NVC57E, SET_PARAMS, FORMAT, asyw->image.format) |
    NVDEF(NVC57E, SET_PARAMS, CLAMP_BEFORE_BLEND, DISABLE) |
    NVDEF(NVC57E, SET_PARAMS, SWAP_UV, DISABLE) |
    NVDEF(NVC57E, SET_PARAMS, FMT_ROUNDING_MODE, ROUND_TO_NEAREST),

    SET_PLANAR_STORAGE(0),
    NVVAL(NVC57E, SET_PLANAR_STORAGE, PITCH, asyw->image.blocks[0]) |
    NVVAL(NVC57E, SET_PLANAR_STORAGE, PITCH, asyw->image.pitch[0] >> 6));

 PUSH_MTHD(push, NVC57E, SET_CONTEXT_DMA_ISO(0), asyw->image.handle, 1);
 PUSH_MTHD(push, NVC57E, SET_OFFSET(0), asyw->image.offset[0] >> 8);

 PUSH_MTHD(push, NVC57E, SET_POINT_IN(0),
    NVVAL(NVC57E, SET_POINT_IN, X, asyw->state.src_x >> 16) |
    NVVAL(NVC57E, SET_POINT_IN, Y, asyw->state.src_y >> 16));

 PUSH_MTHD(push, NVC57E, SET_SIZE_IN,
    NVVAL(NVC57E, SET_SIZE_IN, WIDTH, asyw->state.src_w >> 16) |
    NVVAL(NVC57E, SET_SIZE_IN, HEIGHT, asyw->state.src_h >> 16));

 PUSH_MTHD(push, NVC57E, SET_SIZE_OUT,
    NVVAL(NVC57E, SET_SIZE_OUT, WIDTH, asyw->state.crtc_w) |
    NVVAL(NVC57E, SET_SIZE_OUT, HEIGHT, asyw->state.crtc_h));
 return 0;
}

int
wndwc57e_csc_clr(struct nv50_wndw *wndw)
{
 struct nvif_push *push = &wndw->wndw.push;
 const u32 identity[12] = {
  0x00010000, 0x00000000, 0x00000000, 0x00000000,
  0x00000000, 0x00010000, 0x00000000, 0x00000000,
  0x00000000, 0x00000000, 0x00010000, 0x00000000,
 };
 int ret;

 if ((ret = PUSH_WAIT(push, 1 + ARRAY_SIZE(identity))))
  return ret;

 PUSH_MTHD(push, NVC57E, SET_FMT_COEFFICIENT_C00, identity, ARRAY_SIZE(identity));
 return 0;
}

int
wndwc57e_csc_set(struct nv50_wndw *wndw, struct nv50_wndw_atom *asyw)
{
 struct nvif_push *push = &wndw->wndw.push;
 int ret;

 if ((ret = PUSH_WAIT(push, 13)))
  return ret;

 PUSH_MTHD(push, NVC57E, SET_FMT_COEFFICIENT_C00, asyw->csc.matrix, 12);
 return 0;
}

int
wndwc57e_ilut_clr(struct nv50_wndw *wndw)
{
 struct nvif_push *push = &wndw->wndw.push;
 int ret;

 if ((ret = PUSH_WAIT(push, 2)))
  return ret;

 PUSH_MTHD(push, NVC57E, SET_CONTEXT_DMA_ILUT, 0x00000000);
 return 0;
}

int
wndwc57e_ilut_set(struct nv50_wndw *wndw, struct nv50_wndw_atom *asyw)
{
 struct nvif_push *push = &wndw->wndw.push;
 int ret;

 if ((ret = PUSH_WAIT(push, 4)))
  return ret;

 PUSH_MTHD(push, NVC57E, SET_ILUT_CONTROL,
    NVVAL(NVC57E, SET_ILUT_CONTROL, SIZE, asyw->xlut.i.size) |
    NVVAL(NVC57E, SET_ILUT_CONTROL, MODE, asyw->xlut.i.mode) |
    NVVAL(NVC57E, SET_ILUT_CONTROL, INTERPOLATE, asyw->xlut.i.output_mode),

    SET_CONTEXT_DMA_ILUT, asyw->xlut.handle,
    SET_OFFSET_ILUT, asyw->xlut.i.offset >> 8);
 return 0;
}

static u16
fixedU0_16_FP16(u16 fixed)
{
        int sign = 0, exp = 0, man = 0;
        if (fixed) {
                while (--exp && !(fixed & 0x8000))
                        fixed <<= 1;
                man = ((fixed << 1) & 0xffc0) >> 6;
                exp += 15;
        }
        return (sign << 15) | (exp << 10) | man;
}

static void
wndwc57e_ilut_load(struct drm_color_lut *in, int size, void __iomem *mem)
{
 memset_io(mem, 0x00, 0x20); /* VSS header. */
 mem += 0x20;

 for (; size--; in++, mem += 0x08) {
  u16 r = fixedU0_16_FP16(drm_color_lut_extract(in->  red, 16));
  u16 g = fixedU0_16_FP16(drm_color_lut_extract(in->green, 16));
  u16 b = fixedU0_16_FP16(drm_color_lut_extract(in-> blue, 16));
  writew(r, mem + 0);
  writew(g, mem + 2);
  writew(b, mem + 4);
 }

 /* INTERPOLATE modes require a "next" entry to interpolate with,
  * so we replicate the last entry to deal with this for now.
 */
 writew(readw(mem - 8), mem + 0);
 writew(readw(mem - 6), mem + 2);
 writew(readw(mem - 4), mem + 4);
}

void
wndwc57e_ilut(struct nv50_wndw *wndw, struct nv50_wndw_atom *asyw, int size)
{
 if (!size)
  size = 1024;

 if (size == 256)
  asyw->xlut.i.mode = NVC57E_SET_ILUT_CONTROL_MODE_DIRECT8;
 else
  asyw->xlut.i.mode = NVC57E_SET_ILUT_CONTROL_MODE_DIRECT10;

 asyw->xlut.i.size = 4 /* VSS header. */ + size + 1 /* Entries. */;
 asyw->xlut.i.output_mode = NVC57E_SET_ILUT_CONTROL_INTERPOLATE_DISABLE;
 asyw->xlut.i.load = wndwc57e_ilut_load;
}

/****************************************************************
 *            Log2(block height) ----------------------------+  *
 *            Page Kind ----------------------------------+  |  *
 *            Gob Height/Page Kind Generation ------+     |  |  *
 *                          Sector layout -------+  |     |  |  *
 *                          Compression ------+  |  |     |  |  */
const u64 wndwc57e_modifiers[] = { /*         |  |  |     |  |  */
 DRM_FORMAT_MOD_NVIDIA_BLOCK_LINEAR_2D(0, 1, 2, 0x06, 0),
 DRM_FORMAT_MOD_NVIDIA_BLOCK_LINEAR_2D(0, 1, 2, 0x06, 1),
 DRM_FORMAT_MOD_NVIDIA_BLOCK_LINEAR_2D(0, 1, 2, 0x06, 2),
 DRM_FORMAT_MOD_NVIDIA_BLOCK_LINEAR_2D(0, 1, 2, 0x06, 3),
 DRM_FORMAT_MOD_NVIDIA_BLOCK_LINEAR_2D(0, 1, 2, 0x06, 4),
 DRM_FORMAT_MOD_NVIDIA_BLOCK_LINEAR_2D(0, 1, 2, 0x06, 5),
 DRM_FORMAT_MOD_LINEAR,
 DRM_FORMAT_MOD_INVALID
};

static const struct nv50_wndw_func
wndwc57e = {
 .acquire = wndwc37e_acquire,
 .release = wndwc37e_release,
 .sema_set = wndwc37e_sema_set,
 .sema_clr = wndwc37e_sema_clr,
 .ntfy_set = wndwc37e_ntfy_set,
 .ntfy_clr = wndwc37e_ntfy_clr,
 .ntfy_reset = corec37d_ntfy_init,
 .ntfy_wait_begun = base507c_ntfy_wait_begun,
 .ilut = wndwc57e_ilut,
 .ilut_identity = true,
 .ilut_size = 1024,
 .xlut_set = wndwc57e_ilut_set,
 .xlut_clr = wndwc57e_ilut_clr,
 .csc = base907c_csc,
 .csc_set = wndwc57e_csc_set,
 .csc_clr = wndwc57e_csc_clr,
 .image_set = wndwc57e_image_set,
 .image_clr = wndwc37e_image_clr,
 .blend_set = wndwc37e_blend_set,
 .update = wndwc37e_update,
};

int
wndwc57e_new(struct nouveau_drm *drm, enum drm_plane_type type, int index,
      s32 oclass, struct nv50_wndw **pwndw)
{
 return wndwc37e_new_(&wndwc57e, drm, type, index, oclass,
        BIT(index >> 1), pwndw);
}