/*
 * 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/class.h>
#include <nvif/if0014.h>
#include <nvif/pushc37b.h>

#include <nvhw/class/clc37e.h>

static int
wndwc37e_csc_clr(struct nv50_wndw *wndw)
{
        return 0;
}

static int
wndwc37e_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, NVC37E, SET_CSC_RED2RED, asyw->csc.matrix, 12);
        return 0;
}

static int
wndwc37e_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, NVC37E, SET_CONTEXT_DMA_INPUT_LUT, 0x00000000);
        return 0;
}

static int
wndwc37e_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, NVC37E, SET_CONTROL_INPUT_LUT,
                  NVVAL(NVC37E, SET_CONTROL_INPUT_LUT, OUTPUT_MODE, asyw->xlut.i.output_mode) |
                  NVVAL(NVC37E, SET_CONTROL_INPUT_LUT, RANGE, asyw->xlut.i.range) |
                  NVVAL(NVC37E, SET_CONTROL_INPUT_LUT, SIZE, asyw->xlut.i.size),

                                SET_OFFSET_INPUT_LUT, asyw->xlut.i.offset >> 8,
                                SET_CONTEXT_DMA_INPUT_LUT, asyw->xlut.handle);
        return 0;
}

static void
wndwc37e_ilut(struct nv50_wndw *wndw, struct nv50_wndw_atom *asyw, int size)
{
        asyw->xlut.i.size = size == 1024 ? NVC37E_SET_CONTROL_INPUT_LUT_SIZE_SIZE_1025 :
                                           NVC37E_SET_CONTROL_INPUT_LUT_SIZE_SIZE_257;
        asyw->xlut.i.range = NVC37E_SET_CONTROL_INPUT_LUT_RANGE_UNITY;
        asyw->xlut.i.output_mode = NVC37E_SET_CONTROL_INPUT_LUT_OUTPUT_MODE_INTERPOLATE;
        asyw->xlut.i.load = head907d_olut_load;
}

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

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

        PUSH_MTHD(push, NVC37E, SET_COMPOSITION_CONTROL,
                  NVDEF(NVC37E, SET_COMPOSITION_CONTROL, COLOR_KEY_SELECT, DISABLE) |
                  NVVAL(NVC37E, SET_COMPOSITION_CONTROL, DEPTH, asyw->blend.depth),

                                SET_COMPOSITION_CONSTANT_ALPHA,
                  NVVAL(NVC37E, SET_COMPOSITION_CONSTANT_ALPHA, K1, asyw->blend.k1) |
                  NVVAL(NVC37E, SET_COMPOSITION_CONSTANT_ALPHA, K2, 0),

                                SET_COMPOSITION_FACTOR_SELECT,
                  NVVAL(NVC37E, SET_COMPOSITION_FACTOR_SELECT, SRC_COLOR_FACTOR_MATCH_SELECT,
                                                               asyw->blend.src_color) |
                  NVVAL(NVC37E, SET_COMPOSITION_FACTOR_SELECT, SRC_COLOR_FACTOR_NO_MATCH_SELECT,
                                                               asyw->blend.src_color) |
                  NVVAL(NVC37E, SET_COMPOSITION_FACTOR_SELECT, DST_COLOR_FACTOR_MATCH_SELECT,
                                                               asyw->blend.dst_color) |
                  NVVAL(NVC37E, SET_COMPOSITION_FACTOR_SELECT, DST_COLOR_FACTOR_NO_MATCH_SELECT,
                                                               asyw->blend.dst_color),

                                SET_KEY_ALPHA,
                  NVVAL(NVC37E, SET_KEY_ALPHA, MIN, 0x0000) |
                  NVVAL(NVC37E, SET_KEY_ALPHA, MAX, 0xffff),

                                SET_KEY_RED_CR,
                  NVVAL(NVC37E, SET_KEY_RED_CR, MIN, 0x0000) |
                  NVVAL(NVC37E, SET_KEY_RED_CR, MAX, 0xffff),

                                SET_KEY_GREEN_Y,
                  NVVAL(NVC37E, SET_KEY_GREEN_Y, MIN, 0x0000) |
                  NVVAL(NVC37E, SET_KEY_GREEN_Y, MAX, 0xffff),

                                SET_KEY_BLUE_CB,
                  NVVAL(NVC37E, SET_KEY_BLUE_CB, MIN, 0x0000) |
                  NVVAL(NVC37E, SET_KEY_BLUE_CB, MAX, 0xffff));
        return 0;
}

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

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

        PUSH_MTHD(push, NVC37E, SET_PRESENT_CONTROL,
                  NVVAL(NVC37E, SET_PRESENT_CONTROL, MIN_PRESENT_INTERVAL, 0) |
                  NVDEF(NVC37E, SET_PRESENT_CONTROL, BEGIN_MODE, NON_TEARING));

        PUSH_MTHD(push, NVC37E, SET_CONTEXT_DMA_ISO(0), 0x00000000);
        return 0;
}

static int
wndwc37e_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, NVC37E, SET_PRESENT_CONTROL,
                  NVVAL(NVC37E, SET_PRESENT_CONTROL, MIN_PRESENT_INTERVAL, asyw->image.interval) |
                  NVVAL(NVC37E, SET_PRESENT_CONTROL, BEGIN_MODE, asyw->image.mode) |
                  NVDEF(NVC37E, SET_PRESENT_CONTROL, TIMESTAMP_MODE, DISABLE));

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

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

                                SET_PARAMS,
                  NVVAL(NVC37E, SET_PARAMS, FORMAT, asyw->image.format) |
                  NVVAL(NVC37E, SET_PARAMS, COLOR_SPACE, asyw->image.colorspace) |
                  NVDEF(NVC37E, SET_PARAMS, INPUT_RANGE, BYPASS) |
                  NVDEF(NVC37E, SET_PARAMS, UNDERREPLICATE, DISABLE) |
                  NVDEF(NVC37E, SET_PARAMS, DE_GAMMA, NONE) |
                  NVVAL(NVC37E, SET_PARAMS, CSC, asyw->csc.valid) |
                  NVDEF(NVC37E, SET_PARAMS, CLAMP_BEFORE_BLEND, DISABLE) |
                  NVDEF(NVC37E, SET_PARAMS, SWAP_UV, DISABLE),

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

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

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

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

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

int
wndwc37e_ntfy_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, NVC37E, SET_CONTEXT_DMA_NOTIFIER, 0x00000000);
        return 0;
}

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

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

        PUSH_MTHD(push, NVC37E, SET_CONTEXT_DMA_NOTIFIER, asyw->ntfy.handle,

                                SET_NOTIFIER_CONTROL,
                  NVVAL(NVC37E, SET_NOTIFIER_CONTROL, MODE, asyw->ntfy.awaken) |
                  NVVAL(NVC37E, SET_NOTIFIER_CONTROL, OFFSET, asyw->ntfy.offset >> 4));
        return 0;
}

int
wndwc37e_sema_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, NVC37E, SET_CONTEXT_DMA_SEMAPHORE, 0x00000000);
        return 0;
}

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

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

        PUSH_MTHD(push, NVC37E, SET_SEMAPHORE_CONTROL, asyw->sema.offset,
                                SET_SEMAPHORE_ACQUIRE, asyw->sema.acquire,
                                SET_SEMAPHORE_RELEASE, asyw->sema.release,
                                SET_CONTEXT_DMA_SEMAPHORE, asyw->sema.handle);
        return 0;
}

int
wndwc37e_update(struct nv50_wndw *wndw, u32 *interlock)
{
        struct nvif_push *push = &wndw->wndw.push;
        int ret;

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

        PUSH_MTHD(push, NVC37E, SET_INTERLOCK_FLAGS, interlock[NV50_DISP_INTERLOCK_CURS] << 1 |
                                                     interlock[NV50_DISP_INTERLOCK_CORE],
                                SET_WINDOW_INTERLOCK_FLAGS, interlock[NV50_DISP_INTERLOCK_WNDW]);

        PUSH_MTHD(push, NVC37E, UPDATE, 0x00000001 |
                  NVVAL(NVC37E, UPDATE, INTERLOCK_WITH_WIN_IMM,
                          !!(interlock[NV50_DISP_INTERLOCK_WIMM] & wndw->interlock.data)));

        return PUSH_KICK(push);
}

void
wndwc37e_release(struct nv50_wndw *wndw, struct nv50_wndw_atom *asyw,
                 struct nv50_head_atom *asyh)
{
}

int
wndwc37e_acquire(struct nv50_wndw *wndw, struct nv50_wndw_atom *asyw,
                 struct nv50_head_atom *asyh)
{
        return drm_atomic_helper_check_plane_state(&asyw->state, &asyh->state,
                                                   DRM_PLANE_NO_SCALING,
                                                   DRM_PLANE_NO_SCALING,
                                                   true, true);
}

static const u32
wndwc37e_format[] = {
        DRM_FORMAT_C8,
        DRM_FORMAT_YUYV,
        DRM_FORMAT_UYVY,
        DRM_FORMAT_XRGB8888,
        DRM_FORMAT_ARGB8888,
        DRM_FORMAT_RGB565,
        DRM_FORMAT_XRGB1555,
        DRM_FORMAT_ARGB1555,
        DRM_FORMAT_XBGR2101010,
        DRM_FORMAT_ABGR2101010,
        DRM_FORMAT_XBGR8888,
        DRM_FORMAT_ABGR8888,
        DRM_FORMAT_XRGB2101010,
        DRM_FORMAT_ARGB2101010,
        DRM_FORMAT_XBGR16161616F,
        DRM_FORMAT_ABGR16161616F,
        0
};

static const struct nv50_wndw_func
wndwc37e = {
        .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 = wndwc37e_ilut,
        .ilut_size = 1024,
        .xlut_set = wndwc37e_ilut_set,
        .xlut_clr = wndwc37e_ilut_clr,
        .csc = base907c_csc,
        .csc_set = wndwc37e_csc_set,
        .csc_clr = wndwc37e_csc_clr,
        .image_set = wndwc37e_image_set,
        .image_clr = wndwc37e_image_clr,
        .blend_set = wndwc37e_blend_set,
        .update = wndwc37e_update,
};

int
wndwc37e_new_(const struct nv50_wndw_func *func, struct nouveau_drm *drm,
              enum drm_plane_type type, int index, s32 oclass, u32 heads,
              struct nv50_wndw **pwndw)
{
        struct nvif_disp_chan_v0 args = {
                .id = index,
        };
        struct nv50_disp *disp = nv50_disp(drm->dev);
        struct nv50_wndw *wndw;
        int ret;

        ret = nv50_wndw_new_(func, drm->dev, type, "wndw", index,
                             wndwc37e_format, heads, NV50_DISP_INTERLOCK_WNDW,
                             BIT(index), &wndw);
        if (*pwndw = wndw, ret)
                return ret;

        ret = nv50_dmac_create(drm,
                               &oclass, 0, &args, sizeof(args),
                               disp->sync->offset, &wndw->wndw);
        if (ret) {
                NV_ERROR(drm, "qndw%04x allocation failed: %d\n", oclass, ret);
                return ret;
        }

        wndw->ntfy = NV50_DISP_WNDW_NTFY(wndw->id);
        wndw->sema = NV50_DISP_WNDW_SEM0(wndw->id);
        wndw->data = 0x00000000;
        return 0;
}

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