/*
 * Copyright © 2009
 *
 * 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 (including the next
 * paragraph) 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 AUTHORS OR COPYRIGHT HOLDERS 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.
 *
 * Authors:
 *    Daniel Vetter <daniel@ffwll.ch>
 *
 * Derived from Xorg ddx, xf86-video-intel, src/i830_video.c
 */

#include <drm/drm_fourcc.h>
#include <drm/drm_gem.h>
#include <drm/drm_print.h>

#include "intel_color_regs.h"
#include "intel_de.h"
#include "intel_display_regs.h"
#include "intel_display_types.h"
#include "intel_frontbuffer.h"
#include "intel_overlay.h"
#include "intel_parent.h"
#include "intel_pfit_regs.h"

/* Limits for overlay size. According to intel doc, the real limits are:
 * Y width: 4095, UV width (planar): 2047, Y height: 2047,
 * UV width (planar): * 1023. But the xorg thinks 2048 for height and width. Use
 * the minimum of both.
 */
#define IMAGE_MAX_WIDTH         2048
#define IMAGE_MAX_HEIGHT        2046 /* 2 * 1023 */
/* on 830 and 845 these large limits result in the card hanging */
#define IMAGE_MAX_WIDTH_LEGACY  1024
#define IMAGE_MAX_HEIGHT_LEGACY 1088

/* overlay register definitions */
/* OCMD register */
#define OCMD_TILED_SURFACE      (0x1<<19)
#define OCMD_MIRROR_MASK        (0x3<<17)
#define OCMD_MIRROR_MODE        (0x3<<17)
#define OCMD_MIRROR_HORIZONTAL  (0x1<<17)
#define OCMD_MIRROR_VERTICAL    (0x2<<17)
#define OCMD_MIRROR_BOTH        (0x3<<17)
#define OCMD_BYTEORDER_MASK     (0x3<<14) /* zero for YUYV or FOURCC YUY2 */
#define OCMD_UV_SWAP            (0x1<<14) /* YVYU */
#define OCMD_Y_SWAP             (0x2<<14) /* UYVY or FOURCC UYVY */
#define OCMD_Y_AND_UV_SWAP      (0x3<<14) /* VYUY */
#define OCMD_SOURCE_FORMAT_MASK (0xf<<10)
#define OCMD_RGB_888            (0x1<<10) /* not in i965 Intel docs */
#define OCMD_RGB_555            (0x2<<10) /* not in i965 Intel docs */
#define OCMD_RGB_565            (0x3<<10) /* not in i965 Intel docs */
#define OCMD_YUV_422_PACKED     (0x8<<10)
#define OCMD_YUV_411_PACKED     (0x9<<10) /* not in i965 Intel docs */
#define OCMD_YUV_420_PLANAR     (0xc<<10)
#define OCMD_YUV_422_PLANAR     (0xd<<10)
#define OCMD_YUV_410_PLANAR     (0xe<<10) /* also 411 */
#define OCMD_TVSYNCFLIP_PARITY  (0x1<<9)
#define OCMD_TVSYNCFLIP_ENABLE  (0x1<<7)
#define OCMD_BUF_TYPE_MASK      (0x1<<5)
#define OCMD_BUF_TYPE_FRAME     (0x0<<5)
#define OCMD_BUF_TYPE_FIELD     (0x1<<5)
#define OCMD_TEST_MODE          (0x1<<4)
#define OCMD_BUFFER_SELECT      (0x3<<2)
#define OCMD_BUFFER0            (0x0<<2)
#define OCMD_BUFFER1            (0x1<<2)
#define OCMD_FIELD_SELECT       (0x1<<2)
#define OCMD_FIELD0             (0x0<<1)
#define OCMD_FIELD1             (0x1<<1)
#define OCMD_ENABLE             (0x1<<0)

/* OCONFIG register */
#define OCONF_PIPE_MASK         (0x1<<18)
#define OCONF_PIPE_A            (0x0<<18)
#define OCONF_PIPE_B            (0x1<<18)
#define OCONF_GAMMA2_ENABLE     (0x1<<16)
#define OCONF_CSC_MODE_BT601    (0x0<<5)
#define OCONF_CSC_MODE_BT709    (0x1<<5)
#define OCONF_CSC_BYPASS        (0x1<<4)
#define OCONF_CC_OUT_8BIT       (0x1<<3)
#define OCONF_TEST_MODE         (0x1<<2)
#define OCONF_THREE_LINE_BUFFER (0x1<<0)
#define OCONF_TWO_LINE_BUFFER   (0x0<<0)

/* DCLRKM (dst-key) register */
#define DST_KEY_ENABLE          (0x1<<31)
#define CLK_RGB24_MASK          0x0
#define CLK_RGB16_MASK          0x070307
#define CLK_RGB15_MASK          0x070707

#define RGB30_TO_COLORKEY(c) \
        ((((c) & 0x3fc00000) >> 6) | (((c) & 0x000ff000) >> 4) | (((c) & 0x000003fc) >> 2))
#define RGB16_TO_COLORKEY(c) \
        ((((c) & 0xf800) << 8) | (((c) & 0x07e0) << 5) | (((c) & 0x001f) << 3))
#define RGB15_TO_COLORKEY(c) \
        ((((c) & 0x7c00) << 9) | (((c) & 0x03e0) << 6) | (((c) & 0x001f) << 3))
#define RGB8I_TO_COLORKEY(c) \
        ((((c) & 0xff) << 16) | (((c) & 0xff) << 8) | (((c) & 0xff) << 0))

/* polyphase filter coefficients */
#define N_HORIZ_Y_TAPS          5
#define N_VERT_Y_TAPS           3
#define N_HORIZ_UV_TAPS         3
#define N_VERT_UV_TAPS          3
#define N_PHASES                17
#define MAX_TAPS                5

/* memory bufferd overlay registers */
struct overlay_registers {
        u32 OBUF_0Y;
        u32 OBUF_1Y;
        u32 OBUF_0U;
        u32 OBUF_0V;
        u32 OBUF_1U;
        u32 OBUF_1V;
        u32 OSTRIDE;
        u32 YRGB_VPH;
        u32 UV_VPH;
        u32 HORZ_PH;
        u32 INIT_PHS;
        u32 DWINPOS;
        u32 DWINSZ;
        u32 SWIDTH;
        u32 SWIDTHSW;
        u32 SHEIGHT;
        u32 YRGBSCALE;
        u32 UVSCALE;
        u32 OCLRC0;
        u32 OCLRC1;
        u32 DCLRKV;
        u32 DCLRKM;
        u32 SCLRKVH;
        u32 SCLRKVL;
        u32 SCLRKEN;
        u32 OCONFIG;
        u32 OCMD;
        u32 RESERVED1; /* 0x6C */
        u32 OSTART_0Y;
        u32 OSTART_1Y;
        u32 OSTART_0U;
        u32 OSTART_0V;
        u32 OSTART_1U;
        u32 OSTART_1V;
        u32 OTILEOFF_0Y;
        u32 OTILEOFF_1Y;
        u32 OTILEOFF_0U;
        u32 OTILEOFF_0V;
        u32 OTILEOFF_1U;
        u32 OTILEOFF_1V;
        u32 FASTHSCALE; /* 0xA0 */
        u32 UVSCALEV; /* 0xA4 */
        u32 RESERVEDC[(0x200 - 0xA8) / 4]; /* 0xA8 - 0x1FC */
        u16 Y_VCOEFS[N_VERT_Y_TAPS * N_PHASES]; /* 0x200 */
        u16 RESERVEDD[0x100 / 2 - N_VERT_Y_TAPS * N_PHASES];
        u16 Y_HCOEFS[N_HORIZ_Y_TAPS * N_PHASES]; /* 0x300 */
        u16 RESERVEDE[0x200 / 2 - N_HORIZ_Y_TAPS * N_PHASES];
        u16 UV_VCOEFS[N_VERT_UV_TAPS * N_PHASES]; /* 0x500 */
        u16 RESERVEDF[0x100 / 2 - N_VERT_UV_TAPS * N_PHASES];
        u16 UV_HCOEFS[N_HORIZ_UV_TAPS * N_PHASES]; /* 0x600 */
        u16 RESERVEDG[0x100 / 2 - N_HORIZ_UV_TAPS * N_PHASES];
};

struct intel_overlay {
        struct intel_display *display;
        struct intel_crtc *crtc;
        bool pfit_active;
        u32 pfit_vscale_ratio; /* shifted-point number, (1<<12) == 1.0 */
        u32 color_key:24;
        u32 color_key_enabled:1;
        u32 brightness, contrast, saturation;
        u32 old_xscale, old_yscale;
        struct overlay_registers __iomem *regs;
};

void intel_overlay_reset(struct intel_display *display)
{
        struct intel_overlay *overlay = display->overlay;

        if (!overlay)
                return;

        overlay->old_xscale = 0;
        overlay->old_yscale = 0;
        overlay->crtc = NULL;

        intel_parent_overlay_reset(display);
}

static int packed_depth_bytes(u32 format)
{
        switch (format & I915_OVERLAY_DEPTH_MASK) {
        case I915_OVERLAY_YUV422:
                return 4;
        case I915_OVERLAY_YUV411:
                /* return 6; not implemented */
        default:
                return -EINVAL;
        }
}

static int packed_width_bytes(u32 format, short width)
{
        switch (format & I915_OVERLAY_DEPTH_MASK) {
        case I915_OVERLAY_YUV422:
                return width << 1;
        default:
                return -EINVAL;
        }
}

static int uv_hsubsampling(u32 format)
{
        switch (format & I915_OVERLAY_DEPTH_MASK) {
        case I915_OVERLAY_YUV422:
        case I915_OVERLAY_YUV420:
                return 2;
        case I915_OVERLAY_YUV411:
        case I915_OVERLAY_YUV410:
                return 4;
        default:
                return -EINVAL;
        }
}

static int uv_vsubsampling(u32 format)
{
        switch (format & I915_OVERLAY_DEPTH_MASK) {
        case I915_OVERLAY_YUV420:
        case I915_OVERLAY_YUV410:
                return 2;
        case I915_OVERLAY_YUV422:
        case I915_OVERLAY_YUV411:
                return 1;
        default:
                return -EINVAL;
        }
}

static u32 calc_swidthsw(struct intel_display *display, u32 offset, u32 width)
{
        u32 sw;

        if (DISPLAY_VER(display) == 2)
                sw = ALIGN((offset & 31) + width, 32);
        else
                sw = ALIGN((offset & 63) + width, 64);

        if (sw == 0)
                return 0;

        return (sw - 32) >> 3;
}

static const u16 y_static_hcoeffs[N_PHASES][N_HORIZ_Y_TAPS] = {
        [ 0] = { 0x3000, 0xb4a0, 0x1930, 0x1920, 0xb4a0, },
        [ 1] = { 0x3000, 0xb500, 0x19d0, 0x1880, 0xb440, },
        [ 2] = { 0x3000, 0xb540, 0x1a88, 0x2f80, 0xb3e0, },
        [ 3] = { 0x3000, 0xb580, 0x1b30, 0x2e20, 0xb380, },
        [ 4] = { 0x3000, 0xb5c0, 0x1bd8, 0x2cc0, 0xb320, },
        [ 5] = { 0x3020, 0xb5e0, 0x1c60, 0x2b80, 0xb2c0, },
        [ 6] = { 0x3020, 0xb5e0, 0x1cf8, 0x2a20, 0xb260, },
        [ 7] = { 0x3020, 0xb5e0, 0x1d80, 0x28e0, 0xb200, },
        [ 8] = { 0x3020, 0xb5c0, 0x1e08, 0x3f40, 0xb1c0, },
        [ 9] = { 0x3020, 0xb580, 0x1e78, 0x3ce0, 0xb160, },
        [10] = { 0x3040, 0xb520, 0x1ed8, 0x3aa0, 0xb120, },
        [11] = { 0x3040, 0xb4a0, 0x1f30, 0x3880, 0xb0e0, },
        [12] = { 0x3040, 0xb400, 0x1f78, 0x3680, 0xb0a0, },
        [13] = { 0x3020, 0xb340, 0x1fb8, 0x34a0, 0xb060, },
        [14] = { 0x3020, 0xb240, 0x1fe0, 0x32e0, 0xb040, },
        [15] = { 0x3020, 0xb140, 0x1ff8, 0x3160, 0xb020, },
        [16] = { 0xb000, 0x3000, 0x0800, 0x3000, 0xb000, },
};

static const u16 uv_static_hcoeffs[N_PHASES][N_HORIZ_UV_TAPS] = {
        [ 0] = { 0x3000, 0x1800, 0x1800, },
        [ 1] = { 0xb000, 0x18d0, 0x2e60, },
        [ 2] = { 0xb000, 0x1990, 0x2ce0, },
        [ 3] = { 0xb020, 0x1a68, 0x2b40, },
        [ 4] = { 0xb040, 0x1b20, 0x29e0, },
        [ 5] = { 0xb060, 0x1bd8, 0x2880, },
        [ 6] = { 0xb080, 0x1c88, 0x3e60, },
        [ 7] = { 0xb0a0, 0x1d28, 0x3c00, },
        [ 8] = { 0xb0c0, 0x1db8, 0x39e0, },
        [ 9] = { 0xb0e0, 0x1e40, 0x37e0, },
        [10] = { 0xb100, 0x1eb8, 0x3620, },
        [11] = { 0xb100, 0x1f18, 0x34a0, },
        [12] = { 0xb100, 0x1f68, 0x3360, },
        [13] = { 0xb0e0, 0x1fa8, 0x3240, },
        [14] = { 0xb0c0, 0x1fe0, 0x3140, },
        [15] = { 0xb060, 0x1ff0, 0x30a0, },
        [16] = { 0x3000, 0x0800, 0x3000, },
};

static void update_polyphase_filter(struct overlay_registers __iomem *regs)
{
        memcpy_toio(regs->Y_HCOEFS, y_static_hcoeffs, sizeof(y_static_hcoeffs));
        memcpy_toio(regs->UV_HCOEFS, uv_static_hcoeffs,
                    sizeof(uv_static_hcoeffs));
}

static bool update_scaling_factors(struct intel_overlay *overlay,
                                   struct overlay_registers __iomem *regs,
                                   struct drm_intel_overlay_put_image *params)
{
        /* fixed point with a 12 bit shift */
        u32 xscale, yscale, xscale_UV, yscale_UV;
#define FP_SHIFT 12
#define FRACT_MASK 0xfff
        bool scale_changed = false;
        int uv_hscale = uv_hsubsampling(params->flags);
        int uv_vscale = uv_vsubsampling(params->flags);

        if (params->dst_width > 1)
                xscale = ((params->src_scan_width - 1) << FP_SHIFT) /
                        params->dst_width;
        else
                xscale = 1 << FP_SHIFT;

        if (params->dst_height > 1)
                yscale = ((params->src_scan_height - 1) << FP_SHIFT) /
                        params->dst_height;
        else
                yscale = 1 << FP_SHIFT;

        /*if (params->format & I915_OVERLAY_YUV_PLANAR) {*/
        xscale_UV = xscale/uv_hscale;
        yscale_UV = yscale/uv_vscale;
        /* make the Y scale to UV scale ratio an exact multiply */
        xscale = xscale_UV * uv_hscale;
        yscale = yscale_UV * uv_vscale;
        /*} else {
          xscale_UV = 0;
          yscale_UV = 0;
          }*/

        if (xscale != overlay->old_xscale || yscale != overlay->old_yscale)
                scale_changed = true;
        overlay->old_xscale = xscale;
        overlay->old_yscale = yscale;

        iowrite32(((yscale & FRACT_MASK) << 20) |
                  ((xscale >> FP_SHIFT)  << 16) |
                  ((xscale & FRACT_MASK) << 3),
                 &regs->YRGBSCALE);

        iowrite32(((yscale_UV & FRACT_MASK) << 20) |
                  ((xscale_UV >> FP_SHIFT)  << 16) |
                  ((xscale_UV & FRACT_MASK) << 3),
                 &regs->UVSCALE);

        iowrite32((((yscale    >> FP_SHIFT) << 16) |
                   ((yscale_UV >> FP_SHIFT) << 0)),
                 &regs->UVSCALEV);

        if (scale_changed)
                update_polyphase_filter(regs);

        return scale_changed;
}

static void update_colorkey(struct intel_overlay *overlay,
                            struct overlay_registers __iomem *regs)
{
        const struct intel_plane_state *state =
                to_intel_plane_state(overlay->crtc->base.primary->state);
        u32 key = overlay->color_key;
        u32 format = 0;
        u32 flags = 0;

        if (overlay->color_key_enabled)
                flags |= DST_KEY_ENABLE;

        if (state->uapi.visible)
                format = state->hw.fb->format->format;

        switch (format) {
        case DRM_FORMAT_C8:
                key = RGB8I_TO_COLORKEY(key);
                flags |= CLK_RGB24_MASK;
                break;
        case DRM_FORMAT_XRGB1555:
                key = RGB15_TO_COLORKEY(key);
                flags |= CLK_RGB15_MASK;
                break;
        case DRM_FORMAT_RGB565:
                key = RGB16_TO_COLORKEY(key);
                flags |= CLK_RGB16_MASK;
                break;
        case DRM_FORMAT_XRGB2101010:
        case DRM_FORMAT_XBGR2101010:
                key = RGB30_TO_COLORKEY(key);
                flags |= CLK_RGB24_MASK;
                break;
        default:
                flags |= CLK_RGB24_MASK;
                break;
        }

        iowrite32(key, &regs->DCLRKV);
        iowrite32(flags, &regs->DCLRKM);
}

static u32 overlay_cmd_reg(struct drm_intel_overlay_put_image *params)
{
        u32 cmd = OCMD_ENABLE | OCMD_BUF_TYPE_FRAME | OCMD_BUFFER0;

        if (params->flags & I915_OVERLAY_YUV_PLANAR) {
                switch (params->flags & I915_OVERLAY_DEPTH_MASK) {
                case I915_OVERLAY_YUV422:
                        cmd |= OCMD_YUV_422_PLANAR;
                        break;
                case I915_OVERLAY_YUV420:
                        cmd |= OCMD_YUV_420_PLANAR;
                        break;
                case I915_OVERLAY_YUV411:
                case I915_OVERLAY_YUV410:
                        cmd |= OCMD_YUV_410_PLANAR;
                        break;
                }
        } else { /* YUV packed */
                switch (params->flags & I915_OVERLAY_DEPTH_MASK) {
                case I915_OVERLAY_YUV422:
                        cmd |= OCMD_YUV_422_PACKED;
                        break;
                case I915_OVERLAY_YUV411:
                        cmd |= OCMD_YUV_411_PACKED;
                        break;
                }

                switch (params->flags & I915_OVERLAY_SWAP_MASK) {
                case I915_OVERLAY_NO_SWAP:
                        break;
                case I915_OVERLAY_UV_SWAP:
                        cmd |= OCMD_UV_SWAP;
                        break;
                case I915_OVERLAY_Y_SWAP:
                        cmd |= OCMD_Y_SWAP;
                        break;
                case I915_OVERLAY_Y_AND_UV_SWAP:
                        cmd |= OCMD_Y_AND_UV_SWAP;
                        break;
                }
        }

        return cmd;
}

static int intel_overlay_do_put_image(struct intel_overlay *overlay,
                                      struct drm_gem_object *obj,
                                      struct drm_intel_overlay_put_image *params)
{
        struct intel_display *display = overlay->display;
        struct overlay_registers __iomem *regs = overlay->regs;
        u32 swidth, swidthsw, sheight, ostride;
        enum pipe pipe = overlay->crtc->pipe;
        bool scale_changed = false;
        struct i915_vma *vma;
        int ret, tmp_width;
        u32 tmp, offset;

        drm_WARN_ON(display->drm,
                    !drm_modeset_is_locked(&display->drm->mode_config.connection_mutex));

        ret = intel_parent_overlay_release_old_vid(display);
        if (ret != 0)
                return ret;

        vma = intel_parent_overlay_pin_fb(display, obj, &offset);
        if (IS_ERR(vma))
                return PTR_ERR(vma);

        if (!intel_parent_overlay_is_active(display)) {
                const struct intel_crtc_state *crtc_state =
                        overlay->crtc->config;
                u32 oconfig = 0;

                if (crtc_state->gamma_enable &&
                    crtc_state->gamma_mode == GAMMA_MODE_MODE_8BIT)
                        oconfig |= OCONF_CC_OUT_8BIT;
                if (crtc_state->gamma_enable)
                        oconfig |= OCONF_GAMMA2_ENABLE;
                if (DISPLAY_VER(display) == 4)
                        oconfig |= OCONF_CSC_MODE_BT709;
                oconfig |= pipe == 0 ?
                        OCONF_PIPE_A : OCONF_PIPE_B;
                iowrite32(oconfig, &regs->OCONFIG);

                ret = intel_parent_overlay_on(display, INTEL_FRONTBUFFER_OVERLAY(pipe));
                if (ret != 0)
                        goto out_unpin;
        }

        iowrite32(params->dst_y << 16 | params->dst_x, &regs->DWINPOS);
        iowrite32(params->dst_height << 16 | params->dst_width, &regs->DWINSZ);

        if (params->flags & I915_OVERLAY_YUV_PACKED)
                tmp_width = packed_width_bytes(params->flags,
                                               params->src_width);
        else
                tmp_width = params->src_width;

        swidth = params->src_width;
        swidthsw = calc_swidthsw(display, params->offset_Y, tmp_width);
        sheight = params->src_height;
        iowrite32(offset + params->offset_Y, &regs->OBUF_0Y);
        ostride = params->stride_Y;

        if (params->flags & I915_OVERLAY_YUV_PLANAR) {
                int uv_hscale = uv_hsubsampling(params->flags);
                int uv_vscale = uv_vsubsampling(params->flags);
                u32 tmp_U, tmp_V;

                swidth |= (params->src_width / uv_hscale) << 16;
                sheight |= (params->src_height / uv_vscale) << 16;

                tmp_U = calc_swidthsw(display, params->offset_U,
                                      params->src_width / uv_hscale);
                tmp_V = calc_swidthsw(display, params->offset_V,
                                      params->src_width / uv_hscale);
                swidthsw |= max(tmp_U, tmp_V) << 16;

                iowrite32(offset + params->offset_U,
                          &regs->OBUF_0U);
                iowrite32(offset + params->offset_V,
                          &regs->OBUF_0V);

                ostride |= params->stride_UV << 16;
        }

        iowrite32(swidth, &regs->SWIDTH);
        iowrite32(swidthsw, &regs->SWIDTHSW);
        iowrite32(sheight, &regs->SHEIGHT);
        iowrite32(ostride, &regs->OSTRIDE);

        scale_changed = update_scaling_factors(overlay, regs, params);

        update_colorkey(overlay, regs);

        iowrite32(overlay_cmd_reg(params), &regs->OCMD);

        /* check for underruns */
        tmp = intel_de_read(display, DOVSTA);
        if (tmp & (1 << 17))
                drm_dbg(display->drm, "overlay underrun, DOVSTA: %x\n", tmp);

        ret = intel_parent_overlay_continue(display, vma, scale_changed);
        if (ret)
                goto out_unpin;

        return 0;

out_unpin:
        intel_parent_overlay_unpin_fb(display, vma);

        return ret;
}

int intel_overlay_switch_off(struct intel_overlay *overlay)
{
        struct intel_display *display = overlay->display;
        int ret;

        drm_WARN_ON(display->drm,
                    !drm_modeset_is_locked(&display->drm->mode_config.connection_mutex));

        ret = intel_parent_overlay_recover_from_interrupt(display);
        if (ret != 0)
                return ret;

        if (!intel_parent_overlay_is_active(display))
                return 0;

        ret = intel_parent_overlay_release_old_vid(display);
        if (ret != 0)
                return ret;

        iowrite32(0, &overlay->regs->OCMD);

        overlay->crtc->overlay = NULL;
        overlay->crtc = NULL;

        return intel_parent_overlay_off(display);
}

static int check_overlay_possible_on_crtc(struct intel_overlay *overlay,
                                          struct intel_crtc *crtc)
{
        if (!crtc->active)
                return -EINVAL;

        /* can't use the overlay with double wide pipe */
        if (crtc->config->double_wide)
                return -EINVAL;

        return 0;
}

static void update_pfit_vscale_ratio(struct intel_overlay *overlay)
{
        struct intel_display *display = overlay->display;
        u32 ratio;

        /* XXX: This is not the same logic as in the xorg driver, but more in
         * line with the intel documentation for the i965
         */
        if (DISPLAY_VER(display) >= 4) {
                u32 tmp = intel_de_read(display, PFIT_PGM_RATIOS(display));

                /* on i965 use the PGM reg to read out the autoscaler values */
                ratio = REG_FIELD_GET(PFIT_VERT_SCALE_MASK_965, tmp);
        } else {
                u32 tmp;

                if (intel_de_read(display, PFIT_CONTROL(display)) & PFIT_VERT_AUTO_SCALE)
                        tmp = intel_de_read(display, PFIT_AUTO_RATIOS(display));
                else
                        tmp = intel_de_read(display, PFIT_PGM_RATIOS(display));

                ratio = REG_FIELD_GET(PFIT_VERT_SCALE_MASK, tmp);
        }

        overlay->pfit_vscale_ratio = ratio;
}

static int check_overlay_dst(struct intel_overlay *overlay,
                             struct drm_intel_overlay_put_image *rec)
{
        const struct intel_crtc_state *crtc_state =
                overlay->crtc->config;
        struct drm_rect req, clipped;

        drm_rect_init(&req, rec->dst_x, rec->dst_y,
                      rec->dst_width, rec->dst_height);

        clipped = req;

        if (!drm_rect_intersect(&clipped, &crtc_state->pipe_src))
                return -EINVAL;

        if (!drm_rect_equals(&clipped, &req))
                return -EINVAL;

        return 0;
}

static int check_overlay_scaling(struct drm_intel_overlay_put_image *rec)
{
        u32 tmp;

        /* downscaling limit is 8.0 */
        tmp = ((rec->src_scan_height << 16) / rec->dst_height) >> 16;
        if (tmp > 7)
                return -EINVAL;

        tmp = ((rec->src_scan_width << 16) / rec->dst_width) >> 16;
        if (tmp > 7)
                return -EINVAL;

        return 0;
}

static int check_overlay_src(struct intel_display *display,
                             struct drm_intel_overlay_put_image *rec,
                             struct drm_gem_object *obj)
{
        int uv_hscale = uv_hsubsampling(rec->flags);
        int uv_vscale = uv_vsubsampling(rec->flags);
        u32 stride_mask;
        int depth;
        u32 tmp;

        /* check src dimensions */
        if (display->platform.i845g || display->platform.i830) {
                if (rec->src_height > IMAGE_MAX_HEIGHT_LEGACY ||
                    rec->src_width  > IMAGE_MAX_WIDTH_LEGACY)
                        return -EINVAL;
        } else {
                if (rec->src_height > IMAGE_MAX_HEIGHT ||
                    rec->src_width  > IMAGE_MAX_WIDTH)
                        return -EINVAL;
        }

        /* better safe than sorry, use 4 as the maximal subsampling ratio */
        if (rec->src_height < N_VERT_Y_TAPS*4 ||
            rec->src_width  < N_HORIZ_Y_TAPS*4)
                return -EINVAL;

        /* check alignment constraints */
        switch (rec->flags & I915_OVERLAY_TYPE_MASK) {
        case I915_OVERLAY_RGB:
                /* not implemented */
                return -EINVAL;

        case I915_OVERLAY_YUV_PACKED:
                if (uv_vscale != 1)
                        return -EINVAL;

                depth = packed_depth_bytes(rec->flags);
                if (depth < 0)
                        return depth;

                /* ignore UV planes */
                rec->stride_UV = 0;
                rec->offset_U = 0;
                rec->offset_V = 0;
                /* check pixel alignment */
                if (rec->offset_Y % depth)
                        return -EINVAL;
                break;

        case I915_OVERLAY_YUV_PLANAR:
                if (uv_vscale < 0 || uv_hscale < 0)
                        return -EINVAL;
                /* no offset restrictions for planar formats */
                break;

        default:
                return -EINVAL;
        }

        if (rec->src_width % uv_hscale)
                return -EINVAL;

        /* stride checking */
        if (display->platform.i830 || display->platform.i845g)
                stride_mask = 255;
        else
                stride_mask = 63;

        if (rec->stride_Y & stride_mask || rec->stride_UV & stride_mask)
                return -EINVAL;
        if (DISPLAY_VER(display) == 4 && rec->stride_Y < 512)
                return -EINVAL;

        tmp = (rec->flags & I915_OVERLAY_TYPE_MASK) == I915_OVERLAY_YUV_PLANAR ?
                4096 : 8192;
        if (rec->stride_Y > tmp || rec->stride_UV > 2*1024)
                return -EINVAL;

        /* check buffer dimensions */
        switch (rec->flags & I915_OVERLAY_TYPE_MASK) {
        case I915_OVERLAY_RGB:
        case I915_OVERLAY_YUV_PACKED:
                /* always 4 Y values per depth pixels */
                if (packed_width_bytes(rec->flags, rec->src_width) > rec->stride_Y)
                        return -EINVAL;

                tmp = rec->stride_Y*rec->src_height;
                if (rec->offset_Y + tmp > obj->size)
                        return -EINVAL;
                break;

        case I915_OVERLAY_YUV_PLANAR:
                if (rec->src_width > rec->stride_Y)
                        return -EINVAL;
                if (rec->src_width/uv_hscale > rec->stride_UV)
                        return -EINVAL;

                tmp = rec->stride_Y * rec->src_height;
                if (rec->offset_Y + tmp > obj->size)
                        return -EINVAL;

                tmp = rec->stride_UV * (rec->src_height / uv_vscale);
                if (rec->offset_U + tmp > obj->size ||
                    rec->offset_V + tmp > obj->size)
                        return -EINVAL;
                break;
        }

        return 0;
}

int intel_overlay_put_image_ioctl(struct drm_device *dev, void *data,
                                  struct drm_file *file_priv)
{
        struct intel_display *display = to_intel_display(dev);
        struct drm_intel_overlay_put_image *params = data;
        struct intel_overlay *overlay;
        struct drm_crtc *drmmode_crtc;
        struct drm_gem_object *obj;
        struct intel_crtc *crtc;
        int ret;

        overlay = display->overlay;
        if (!overlay) {
                drm_dbg(display->drm, "userspace bug: no overlay\n");
                return -ENODEV;
        }

        if (!(params->flags & I915_OVERLAY_ENABLE)) {
                drm_modeset_lock_all(dev);
                ret = intel_overlay_switch_off(overlay);
                drm_modeset_unlock_all(dev);

                return ret;
        }

        drmmode_crtc = drm_crtc_find(dev, file_priv, params->crtc_id);
        if (!drmmode_crtc)
                return -ENOENT;
        crtc = to_intel_crtc(drmmode_crtc);

        obj = intel_parent_overlay_obj_lookup(display, file_priv, params->bo_handle);
        if (IS_ERR(obj))
                return PTR_ERR(obj);

        drm_modeset_lock_all(dev);

        ret = intel_parent_overlay_recover_from_interrupt(display);
        if (ret != 0)
                goto out_unlock;

        if (overlay->crtc != crtc) {
                ret = intel_overlay_switch_off(overlay);
                if (ret != 0)
                        goto out_unlock;

                ret = check_overlay_possible_on_crtc(overlay, crtc);
                if (ret != 0)
                        goto out_unlock;

                overlay->crtc = crtc;
                crtc->overlay = overlay;

                /* line too wide, i.e. one-line-mode */
                if (drm_rect_width(&crtc->config->pipe_src) > 1024 &&
                    crtc->config->gmch_pfit.control & PFIT_ENABLE) {
                        overlay->pfit_active = true;
                        update_pfit_vscale_ratio(overlay);
                } else
                        overlay->pfit_active = false;
        }

        ret = check_overlay_dst(overlay, params);
        if (ret != 0)
                goto out_unlock;

        if (overlay->pfit_active) {
                params->dst_y = (((u32)params->dst_y << 12) /
                                 overlay->pfit_vscale_ratio);
                /* shifting right rounds downwards, so add 1 */
                params->dst_height = (((u32)params->dst_height << 12) /
                                 overlay->pfit_vscale_ratio) + 1;
        }

        if (params->src_scan_height > params->src_height ||
            params->src_scan_width > params->src_width) {
                ret = -EINVAL;
                goto out_unlock;
        }

        ret = check_overlay_src(display, params, obj);
        if (ret != 0)
                goto out_unlock;

        /* Check scaling after src size to prevent a divide-by-zero. */
        ret = check_overlay_scaling(params);
        if (ret != 0)
                goto out_unlock;

        ret = intel_overlay_do_put_image(overlay, obj, params);
        if (ret != 0)
                goto out_unlock;

        drm_modeset_unlock_all(dev);
        drm_gem_object_put(obj);

        return 0;

out_unlock:
        drm_modeset_unlock_all(dev);
        drm_gem_object_put(obj);

        return ret;
}

static void update_reg_attrs(struct intel_overlay *overlay,
                             struct overlay_registers __iomem *regs)
{
        iowrite32((overlay->contrast << 18) | (overlay->brightness & 0xff),
                  &regs->OCLRC0);
        iowrite32(overlay->saturation, &regs->OCLRC1);
}

static bool check_gamma_bounds(u32 gamma1, u32 gamma2)
{
        int i;

        if (gamma1 & 0xff000000 || gamma2 & 0xff000000)
                return false;

        for (i = 0; i < 3; i++) {
                if (((gamma1 >> i*8) & 0xff) >= ((gamma2 >> i*8) & 0xff))
                        return false;
        }

        return true;
}

static bool check_gamma5_errata(u32 gamma5)
{
        int i;

        for (i = 0; i < 3; i++) {
                if (((gamma5 >> i*8) & 0xff) == 0x80)
                        return false;
        }

        return true;
}

static int check_gamma(struct drm_intel_overlay_attrs *attrs)
{
        if (!check_gamma_bounds(0, attrs->gamma0) ||
            !check_gamma_bounds(attrs->gamma0, attrs->gamma1) ||
            !check_gamma_bounds(attrs->gamma1, attrs->gamma2) ||
            !check_gamma_bounds(attrs->gamma2, attrs->gamma3) ||
            !check_gamma_bounds(attrs->gamma3, attrs->gamma4) ||
            !check_gamma_bounds(attrs->gamma4, attrs->gamma5) ||
            !check_gamma_bounds(attrs->gamma5, 0x00ffffff))
                return -EINVAL;

        if (!check_gamma5_errata(attrs->gamma5))
                return -EINVAL;

        return 0;
}

int intel_overlay_attrs_ioctl(struct drm_device *dev, void *data,
                              struct drm_file *file_priv)
{
        struct intel_display *display = to_intel_display(dev);
        struct drm_intel_overlay_attrs *attrs = data;
        struct intel_overlay *overlay;
        int ret;

        overlay = display->overlay;
        if (!overlay) {
                drm_dbg(display->drm, "userspace bug: no overlay\n");
                return -ENODEV;
        }

        drm_modeset_lock_all(dev);

        ret = -EINVAL;
        if (!(attrs->flags & I915_OVERLAY_UPDATE_ATTRS)) {
                attrs->color_key  = overlay->color_key;
                attrs->brightness = overlay->brightness;
                attrs->contrast   = overlay->contrast;
                attrs->saturation = overlay->saturation;

                if (DISPLAY_VER(display) != 2) {
                        attrs->gamma0 = intel_de_read(display, OGAMC0);
                        attrs->gamma1 = intel_de_read(display, OGAMC1);
                        attrs->gamma2 = intel_de_read(display, OGAMC2);
                        attrs->gamma3 = intel_de_read(display, OGAMC3);
                        attrs->gamma4 = intel_de_read(display, OGAMC4);
                        attrs->gamma5 = intel_de_read(display, OGAMC5);
                }
        } else {
                if (attrs->brightness < -128 || attrs->brightness > 127)
                        goto out_unlock;
                if (attrs->contrast > 255)
                        goto out_unlock;
                if (attrs->saturation > 1023)
                        goto out_unlock;

                overlay->color_key  = attrs->color_key;
                overlay->brightness = attrs->brightness;
                overlay->contrast   = attrs->contrast;
                overlay->saturation = attrs->saturation;

                update_reg_attrs(overlay, overlay->regs);

                if (attrs->flags & I915_OVERLAY_UPDATE_GAMMA) {
                        if (DISPLAY_VER(display) == 2)
                                goto out_unlock;

                        if (intel_parent_overlay_is_active(display)) {
                                ret = -EBUSY;
                                goto out_unlock;
                        }

                        ret = check_gamma(attrs);
                        if (ret)
                                goto out_unlock;

                        intel_de_write(display, OGAMC0, attrs->gamma0);
                        intel_de_write(display, OGAMC1, attrs->gamma1);
                        intel_de_write(display, OGAMC2, attrs->gamma2);
                        intel_de_write(display, OGAMC3, attrs->gamma3);
                        intel_de_write(display, OGAMC4, attrs->gamma4);
                        intel_de_write(display, OGAMC5, attrs->gamma5);
                }
        }
        overlay->color_key_enabled = (attrs->flags & I915_OVERLAY_DISABLE_DEST_COLORKEY) == 0;

        ret = 0;
out_unlock:
        drm_modeset_unlock_all(dev);

        return ret;
}

void intel_overlay_setup(struct intel_display *display)
{
        struct intel_overlay *overlay;
        void __iomem *regs;

        if (!HAS_OVERLAY(display))
                return;

        overlay = kzalloc_obj(*overlay);
        if (!overlay)
                return;

        regs = intel_parent_overlay_setup(display,
                                          OVERLAY_NEEDS_PHYSICAL(display));
        if (IS_ERR(regs))
                goto out_free;

        overlay->display = display;
        overlay->regs = regs;
        overlay->color_key = 0x0101fe;
        overlay->color_key_enabled = true;
        overlay->brightness = -19;
        overlay->contrast = 75;
        overlay->saturation = 146;

        memset_io(overlay->regs, 0, sizeof(struct overlay_registers));
        update_polyphase_filter(overlay->regs);
        update_reg_attrs(overlay, overlay->regs);

        display->overlay = overlay;
        drm_info(display->drm, "Initialized overlay support.\n");
        return;

out_free:
        kfree(overlay);
}

bool intel_overlay_available(struct intel_display *display)
{
        return display->overlay;
}

void intel_overlay_cleanup(struct intel_display *display)
{
        if (!display->overlay)
                return;

        intel_parent_overlay_cleanup(display);

        kfree(display->overlay);
        display->overlay = NULL;
}