/*
 * Copyright 2013 Advanced Micro Devices, Inc.
 * All Rights Reserved.
 *
 * 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, sub license, 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 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 NON-INFRINGEMENT. IN NO EVENT SHALL
 * THE COPYRIGHT HOLDERS, AUTHORS AND/OR ITS SUPPLIERS 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.
 *
 * The above copyright notice and this permission notice (including the
 * next paragraph) shall be included in all copies or substantial portions
 * of the Software.
 *
 * Authors: Christian König <christian.koenig@amd.com>
 */

#include <linux/firmware.h>

#include "radeon.h"
#include "radeon_asic.h"
#include "cikd.h"
#include "vce.h"

#define VCE_V2_0_FW_SIZE (256 * 1024)
#define VCE_V2_0_STACK_SIZE (64 * 1024)
#define VCE_V2_0_DATA_SIZE (23552 * RADEON_MAX_VCE_HANDLES)

static void vce_v2_0_set_sw_cg(struct radeon_device *rdev, bool gated)
{
 u32 tmp;

 if (gated) {
  tmp = RREG32(VCE_CLOCK_GATING_B);
  tmp |= 0xe70000;
  WREG32(VCE_CLOCK_GATING_B, tmp);

  tmp = RREG32(VCE_UENC_CLOCK_GATING);
  tmp |= 0xff000000;
  WREG32(VCE_UENC_CLOCK_GATING, tmp);

  tmp = RREG32(VCE_UENC_REG_CLOCK_GATING);
  tmp &= ~0x3fc;
  WREG32(VCE_UENC_REG_CLOCK_GATING, tmp);

  WREG32(VCE_CGTT_CLK_OVERRIDE, 0);
 } else {
  tmp = RREG32(VCE_CLOCK_GATING_B);
  tmp |= 0xe7;
  tmp &= ~0xe70000;
  WREG32(VCE_CLOCK_GATING_B, tmp);

  tmp = RREG32(VCE_UENC_CLOCK_GATING);
  tmp |= 0x1fe000;
  tmp &= ~0xff000000;
  WREG32(VCE_UENC_CLOCK_GATING, tmp);

  tmp = RREG32(VCE_UENC_REG_CLOCK_GATING);
  tmp |= 0x3fc;
  WREG32(VCE_UENC_REG_CLOCK_GATING, tmp);
 }
}

static void vce_v2_0_set_dyn_cg(struct radeon_device *rdev, bool gated)
{
 u32 orig, tmp;

 tmp = RREG32(VCE_CLOCK_GATING_B);
 tmp &= ~0x00060006;
 if (gated) {
  tmp |= 0xe10000;
 } else {
  tmp |= 0xe1;
  tmp &= ~0xe10000;
 }
 WREG32(VCE_CLOCK_GATING_B, tmp);

 orig = tmp = RREG32(VCE_UENC_CLOCK_GATING);
 tmp &= ~0x1fe000;
 tmp &= ~0xff000000;
 if (tmp != orig)
  WREG32(VCE_UENC_CLOCK_GATING, tmp);

 orig = tmp = RREG32(VCE_UENC_REG_CLOCK_GATING);
 tmp &= ~0x3fc;
 if (tmp != orig)
  WREG32(VCE_UENC_REG_CLOCK_GATING, tmp);

 if (gated)
  WREG32(VCE_CGTT_CLK_OVERRIDE, 0);
}

static void vce_v2_0_disable_cg(struct radeon_device *rdev)
{
 WREG32(VCE_CGTT_CLK_OVERRIDE, 7);
}

/*
 * Local variable sw_cg is used for debugging purposes, in case we
 * ran into problems with dynamic clock gating. Don't remove it.
 */
void vce_v2_0_enable_mgcg(struct radeon_device *rdev, bool enable)
{
 bool sw_cg = false;

 if (enable && (rdev->cg_flags & RADEON_CG_SUPPORT_VCE_MGCG)) {
  if (sw_cg)
   vce_v2_0_set_sw_cg(rdev, true);
  else
   vce_v2_0_set_dyn_cg(rdev, true);
 } else {
  vce_v2_0_disable_cg(rdev);

  if (sw_cg)
   vce_v2_0_set_sw_cg(rdev, false);
  else
   vce_v2_0_set_dyn_cg(rdev, false);
 }
}

static void vce_v2_0_init_cg(struct radeon_device *rdev)
{
 u32 tmp;

 tmp = RREG32(VCE_CLOCK_GATING_A);
 tmp &= ~(CGC_CLK_GATE_DLY_TIMER_MASK | CGC_CLK_GATER_OFF_DLY_TIMER_MASK);
 tmp |= (CGC_CLK_GATE_DLY_TIMER(0) | CGC_CLK_GATER_OFF_DLY_TIMER(4));
 tmp |= CGC_UENC_WAIT_AWAKE;
 WREG32(VCE_CLOCK_GATING_A, tmp);

 tmp = RREG32(VCE_UENC_CLOCK_GATING);
 tmp &= ~(CLOCK_ON_DELAY_MASK | CLOCK_OFF_DELAY_MASK);
 tmp |= (CLOCK_ON_DELAY(0) | CLOCK_OFF_DELAY(4));
 WREG32(VCE_UENC_CLOCK_GATING, tmp);

 tmp = RREG32(VCE_CLOCK_GATING_B);
 tmp |= 0x10;
 tmp &= ~0x100000;
 WREG32(VCE_CLOCK_GATING_B, tmp);
}

unsigned vce_v2_0_bo_size(struct radeon_device *rdev)
{
 WARN_ON(rdev->vce_fw->size > VCE_V2_0_FW_SIZE);
 return VCE_V2_0_FW_SIZE + VCE_V2_0_STACK_SIZE + VCE_V2_0_DATA_SIZE;
}

int vce_v2_0_resume(struct radeon_device *rdev)
{
 uint64_t addr = rdev->vce.gpu_addr;
 uint32_t size;

 WREG32_P(VCE_CLOCK_GATING_A, 0, ~(1 << 16));
 WREG32_P(VCE_UENC_CLOCK_GATING, 0x1FF000, ~0xFF9FF000);
 WREG32_P(VCE_UENC_REG_CLOCK_GATING, 0x3F, ~0x3F);
 WREG32(VCE_CLOCK_GATING_B, 0xf7);

 WREG32(VCE_LMI_CTRL, 0x00398000);
 WREG32_P(VCE_LMI_CACHE_CTRL, 0x0, ~0x1);
 WREG32(VCE_LMI_SWAP_CNTL, 0);
 WREG32(VCE_LMI_SWAP_CNTL1, 0);
 WREG32(VCE_LMI_VM_CTRL, 0);

 WREG32(VCE_LMI_VCPU_CACHE_40BIT_BAR, addr >> 8);

 addr &= 0xff;
 size = VCE_V2_0_FW_SIZE;
 WREG32(VCE_VCPU_CACHE_OFFSET0, addr & 0x7fffffff);
 WREG32(VCE_VCPU_CACHE_SIZE0, size);

 addr += size;
 size = VCE_V2_0_STACK_SIZE;
 WREG32(VCE_VCPU_CACHE_OFFSET1, addr & 0x7fffffff);
 WREG32(VCE_VCPU_CACHE_SIZE1, size);

 addr += size;
 size = VCE_V2_0_DATA_SIZE;
 WREG32(VCE_VCPU_CACHE_OFFSET2, addr & 0x7fffffff);
 WREG32(VCE_VCPU_CACHE_SIZE2, size);

 WREG32_P(VCE_LMI_CTRL2, 0x0, ~0x100);

 WREG32_P(VCE_SYS_INT_EN, VCE_SYS_INT_TRAP_INTERRUPT_EN,
   ~VCE_SYS_INT_TRAP_INTERRUPT_EN);

 vce_v2_0_init_cg(rdev);

 return 0;
}