/*
 * Copyright © 2014 Intel Corporation
 *
 * 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.vetter@ffwll.ch>
 */

/**
 * DOC: frontbuffer tracking
 *
 * Many features require us to track changes to the currently active
 * frontbuffer, especially rendering targeted at the frontbuffer.
 *
 * To be able to do so we track frontbuffers using a bitmask for all possible
 * frontbuffer slots through intel_frontbuffer_track(). The functions in this
 * file are then called when the contents of the frontbuffer are invalidated,
 * when frontbuffer rendering has stopped again to flush out all the changes
 * and when the frontbuffer is exchanged with a flip. Subsystems interested in
 * frontbuffer changes (e.g. PSR, FBC, DRRS) should directly put their callbacks
 * into the relevant places and filter for the frontbuffer slots that they are
 * interested int.
 *
 * On a high level there are two types of powersaving features. The first one
 * work like a special cache (FBC and PSR) and are interested when they should
 * stop caching and when to restart caching. This is done by placing callbacks
 * into the invalidate and the flush functions: At invalidate the caching must
 * be stopped and at flush time it can be restarted. And maybe they need to know
 * when the frontbuffer changes (e.g. when the hw doesn't initiate an invalidate
 * and flush on its own) which can be achieved with placing callbacks into the
 * flip functions.
 *
 * The other type of display power saving feature only cares about busyness
 * (e.g. DRRS). In that case all three (invalidate, flush and flip) indicate
 * busyness. There is no direct way to detect idleness. Instead an idle timer
 * work delayed work should be started from the flush and flip functions and
 * cancelled as soon as busyness is detected.
 */

#include <drm/drm_gem.h>

#include "i915_active.h"
#include "i915_vma.h"
#include "intel_bo.h"
#include "intel_display_trace.h"
#include "intel_display_types.h"
#include "intel_dp.h"
#include "intel_drrs.h"
#include "intel_fbc.h"
#include "intel_frontbuffer.h"
#include "intel_psr.h"
#include "intel_tdf.h"

/**
 * frontbuffer_flush - flush frontbuffer
 * @display: display device
 * @frontbuffer_bits: frontbuffer plane tracking bits
 * @origin: which operation caused the flush
 *
 * This function gets called every time rendering on the given planes has
 * completed and frontbuffer caching can be started again. Flushes will get
 * delayed if they're blocked by some outstanding asynchronous rendering.
 *
 * Can be called without any locks held.
 */
static void frontbuffer_flush(struct intel_display *display,
                              unsigned int frontbuffer_bits,
                              enum fb_op_origin origin)
{
        /* Delay flushing when rings are still busy.*/
        spin_lock(&display->fb_tracking.lock);
        frontbuffer_bits &= ~display->fb_tracking.busy_bits;
        spin_unlock(&display->fb_tracking.lock);

        if (!frontbuffer_bits)
                return;

        trace_intel_frontbuffer_flush(display, frontbuffer_bits, origin);

        might_sleep();
        intel_td_flush(display);
        intel_drrs_flush(display, frontbuffer_bits);
        intel_psr_flush(display, frontbuffer_bits, origin);
        intel_fbc_flush(display, frontbuffer_bits, origin);
}

/**
 * intel_frontbuffer_flip_prepare - prepare asynchronous frontbuffer flip
 * @display: display device
 * @frontbuffer_bits: frontbuffer plane tracking bits
 *
 * This function gets called after scheduling a flip on @obj. The actual
 * frontbuffer flushing will be delayed until completion is signalled with
 * intel_frontbuffer_flip_complete. If an invalidate happens in between this
 * flush will be cancelled.
 *
 * Can be called without any locks held.
 */
void intel_frontbuffer_flip_prepare(struct intel_display *display,
                                    unsigned frontbuffer_bits)
{
        spin_lock(&display->fb_tracking.lock);
        display->fb_tracking.flip_bits |= frontbuffer_bits;
        /* Remove stale busy bits due to the old buffer. */
        display->fb_tracking.busy_bits &= ~frontbuffer_bits;
        spin_unlock(&display->fb_tracking.lock);
}

/**
 * intel_frontbuffer_flip_complete - complete asynchronous frontbuffer flip
 * @display: display device
 * @frontbuffer_bits: frontbuffer plane tracking bits
 *
 * This function gets called after the flip has been latched and will complete
 * on the next vblank. It will execute the flush if it hasn't been cancelled yet.
 *
 * Can be called without any locks held.
 */
void intel_frontbuffer_flip_complete(struct intel_display *display,
                                     unsigned frontbuffer_bits)
{
        spin_lock(&display->fb_tracking.lock);
        /* Mask any cancelled flips. */
        frontbuffer_bits &= display->fb_tracking.flip_bits;
        display->fb_tracking.flip_bits &= ~frontbuffer_bits;
        spin_unlock(&display->fb_tracking.lock);

        if (frontbuffer_bits)
                frontbuffer_flush(display, frontbuffer_bits, ORIGIN_FLIP);
}

/**
 * intel_frontbuffer_flip - synchronous frontbuffer flip
 * @display: display device
 * @frontbuffer_bits: frontbuffer plane tracking bits
 *
 * This function gets called after scheduling a flip on @obj. This is for
 * synchronous plane updates which will happen on the next vblank and which will
 * not get delayed by pending gpu rendering.
 *
 * Can be called without any locks held.
 */
void intel_frontbuffer_flip(struct intel_display *display,
                            unsigned frontbuffer_bits)
{
        spin_lock(&display->fb_tracking.lock);
        /* Remove stale busy bits due to the old buffer. */
        display->fb_tracking.busy_bits &= ~frontbuffer_bits;
        spin_unlock(&display->fb_tracking.lock);

        frontbuffer_flush(display, frontbuffer_bits, ORIGIN_FLIP);
}

void __intel_fb_invalidate(struct intel_frontbuffer *front,
                           enum fb_op_origin origin,
                           unsigned int frontbuffer_bits)
{
        struct intel_display *display = to_intel_display(front->obj->dev);

        if (origin == ORIGIN_CS) {
                spin_lock(&display->fb_tracking.lock);
                display->fb_tracking.busy_bits |= frontbuffer_bits;
                display->fb_tracking.flip_bits &= ~frontbuffer_bits;
                spin_unlock(&display->fb_tracking.lock);
        }

        trace_intel_frontbuffer_invalidate(display, frontbuffer_bits, origin);

        might_sleep();
        intel_psr_invalidate(display, frontbuffer_bits, origin);
        intel_drrs_invalidate(display, frontbuffer_bits);
        intel_fbc_invalidate(display, frontbuffer_bits, origin);
}

void __intel_fb_flush(struct intel_frontbuffer *front,
                      enum fb_op_origin origin,
                      unsigned int frontbuffer_bits)
{
        struct intel_display *display = to_intel_display(front->obj->dev);

        if (origin == ORIGIN_CS) {
                spin_lock(&display->fb_tracking.lock);
                /* Filter out new bits since rendering started. */
                frontbuffer_bits &= display->fb_tracking.busy_bits;
                display->fb_tracking.busy_bits &= ~frontbuffer_bits;
                spin_unlock(&display->fb_tracking.lock);
        }

        if (frontbuffer_bits)
                frontbuffer_flush(display, frontbuffer_bits, origin);
}

static void intel_frontbuffer_flush_work(struct work_struct *work)
{
        struct intel_frontbuffer *front =
                container_of(work, struct intel_frontbuffer, flush_work);

        intel_bo_flush_if_display(front->obj);
        intel_frontbuffer_flush(front, ORIGIN_DIRTYFB);
        intel_frontbuffer_put(front);
}

/**
 * intel_frontbuffer_queue_flush - queue flushing frontbuffer object
 * @front: GEM object to flush
 *
 * This function is targeted for our dirty callback for queueing flush when
 * dma fence is signals
 */
void intel_frontbuffer_queue_flush(struct intel_frontbuffer *front)
{
        if (!front)
                return;

        kref_get(&front->ref);
        if (!schedule_work(&front->flush_work))
                intel_frontbuffer_put(front);
}

static int frontbuffer_active(struct i915_active *ref)
{
        struct intel_frontbuffer *front =
                container_of(ref, typeof(*front), write);

        kref_get(&front->ref);
        return 0;
}

static void frontbuffer_retire(struct i915_active *ref)
{
        struct intel_frontbuffer *front =
                container_of(ref, typeof(*front), write);

        intel_frontbuffer_flush(front, ORIGIN_CS);
        intel_frontbuffer_put(front);
}

static void frontbuffer_release(struct kref *ref)
        __releases(&to_intel_display(front->obj->dev)->fb_tracking.lock)
{
        struct intel_frontbuffer *ret, *front =
                container_of(ref, typeof(*front), ref);
        struct drm_gem_object *obj = front->obj;
        struct intel_display *display = to_intel_display(obj->dev);

        drm_WARN_ON(display->drm, atomic_read(&front->bits));

        i915_ggtt_clear_scanout(to_intel_bo(obj));

        ret = intel_bo_set_frontbuffer(obj, NULL);
        drm_WARN_ON(display->drm, ret);
        spin_unlock(&display->fb_tracking.lock);

        i915_active_fini(&front->write);

        drm_gem_object_put(obj);
        kfree_rcu(front, rcu);
}

struct intel_frontbuffer *
intel_frontbuffer_get(struct drm_gem_object *obj)
{
        struct intel_display *display = to_intel_display(obj->dev);
        struct intel_frontbuffer *front, *cur;

        front = intel_bo_get_frontbuffer(obj);
        if (front)
                return front;

        front = kmalloc(sizeof(*front), GFP_KERNEL);
        if (!front)
                return NULL;

        drm_gem_object_get(obj);

        front->obj = obj;
        kref_init(&front->ref);
        atomic_set(&front->bits, 0);
        i915_active_init(&front->write,
                         frontbuffer_active,
                         frontbuffer_retire,
                         I915_ACTIVE_RETIRE_SLEEPS);
        INIT_WORK(&front->flush_work, intel_frontbuffer_flush_work);

        spin_lock(&display->fb_tracking.lock);
        cur = intel_bo_set_frontbuffer(obj, front);
        spin_unlock(&display->fb_tracking.lock);

        if (cur != front) {
                drm_gem_object_put(obj);
                kfree(front);
        }

        return cur;
}

void intel_frontbuffer_put(struct intel_frontbuffer *front)
{
        kref_put_lock(&front->ref,
                      frontbuffer_release,
                      &to_intel_display(front->obj->dev)->fb_tracking.lock);
}

/**
 * intel_frontbuffer_track - update frontbuffer tracking
 * @old: current buffer for the frontbuffer slots
 * @new: new buffer for the frontbuffer slots
 * @frontbuffer_bits: bitmask of frontbuffer slots
 *
 * This updates the frontbuffer tracking bits @frontbuffer_bits by clearing them
 * from @old and setting them in @new. Both @old and @new can be NULL.
 */
void intel_frontbuffer_track(struct intel_frontbuffer *old,
                             struct intel_frontbuffer *new,
                             unsigned int frontbuffer_bits)
{
        /*
         * Control of individual bits within the mask are guarded by
         * the owning plane->mutex, i.e. we can never see concurrent
         * manipulation of individual bits. But since the bitfield as a whole
         * is updated using RMW, we need to use atomics in order to update
         * the bits.
         */
        BUILD_BUG_ON(INTEL_FRONTBUFFER_BITS_PER_PIPE * I915_MAX_PIPES >
                     BITS_PER_TYPE(atomic_t));
        BUILD_BUG_ON(INTEL_FRONTBUFFER_BITS_PER_PIPE * I915_MAX_PIPES > 32);
        BUILD_BUG_ON(I915_MAX_PLANES > INTEL_FRONTBUFFER_BITS_PER_PIPE);

        if (old) {
                struct intel_display *display = to_intel_display(old->obj->dev);

                drm_WARN_ON(display->drm,
                            !(atomic_read(&old->bits) & frontbuffer_bits));
                atomic_andnot(frontbuffer_bits, &old->bits);
        }

        if (new) {
                struct intel_display *display = to_intel_display(new->obj->dev);

                drm_WARN_ON(display->drm,
                            atomic_read(&new->bits) & frontbuffer_bits);
                atomic_or(frontbuffer_bits, &new->bits);
        }
}