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