/**************************************************************************
 * 
 * Copyright 2006 Tungsten Graphics, Inc., Bismarck, ND., USA
 * 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: Thomas Hellström <thomas-at-tungstengraphics-dot-com>
 */

#include "drmP.h"
#include "i915_drm.h"
#include "i915_drv.h"

drm_ttm_backend_t *i915_create_ttm_backend_entry(drm_device_t * dev)
{
	return drm_agp_init_ttm(dev);
}

int i915_fence_types(drm_buffer_object_t *bo, uint32_t * class, uint32_t * type)
{
	*class = 0;
	if (bo->mem.flags & (DRM_BO_FLAG_READ | DRM_BO_FLAG_WRITE))
		*type = 3;
	else
		*type = 1;
	return 0;
}

int i915_invalidate_caches(drm_device_t * dev, uint32_t flags)
{
	/*
	 * FIXME: Only emit once per batchbuffer submission.
	 */

	uint32_t flush_cmd = MI_NO_WRITE_FLUSH;

	if (flags & DRM_BO_FLAG_READ)
		flush_cmd |= MI_READ_FLUSH;
	if (flags & DRM_BO_FLAG_EXE)
		flush_cmd |= MI_EXE_FLUSH;

	return i915_emit_mi_flush(dev, flush_cmd);
}

#define PRIV1_ORDER 8
#define PRIV1_SIZE ((1 << PRIV1_ORDER) * PAGE_SIZE)

int i915_init_priv1(drm_device_t * dev)
{
	drm_i915_private_t *dev_priv = dev->dev_private;
	int ret;

	if (dev_priv->priv1_addr)
		return 0;

	dev_priv->priv1_addr = __get_free_pages(__GFP_NOFAIL | __GFP_HIGH,
						PRIV1_ORDER);

	if (!dev_priv->priv1_addr) {
		DRM_ERROR("Failed to get %lu bytes of physically "
			  "contiguous memory for PRIV1 type\n", PRIV1_SIZE);
		return DRM_ERR(ENOMEM);
	} else
		DRM_INFO("PRIV1 virtual 0x%lx physical 0x%lx\n",
			 dev_priv->priv1_addr,
			 virt_to_phys((void*)dev_priv->priv1_addr));

	dev_priv->priv1_order = PRIV1_ORDER;

	ret = drm_bo_init_mm(dev, DRM_BO_MEM_PRIV1,
			     virt_to_phys((void*)dev_priv->priv1_addr)
			     >> PAGE_SHIFT, (1 << dev_priv->priv1_order));

	if (ret) {
		DRM_ERROR("Failed to initialize PRIV1 memory manager\n");
		free_pages(dev_priv->priv1_addr, dev_priv->priv1_order);
		dev_priv->priv1_addr = 0;
		return ret;
	}

	return 0;
}

int i915_init_mem_type(drm_device_t * dev, uint32_t type,
		       drm_mem_type_manager_t * man)
{
	drm_i915_private_t *dev_priv = dev->dev_private;

	switch (type) {
	case DRM_BO_MEM_LOCAL:
		man->flags = _DRM_FLAG_MEMTYPE_MAPPABLE |
		    _DRM_FLAG_MEMTYPE_CACHED;
		man->drm_bus_maptype = 0;
		break;
	case DRM_BO_MEM_TT:
		if (!(drm_core_has_AGP(dev) && dev->agp)) {
			DRM_ERROR("AGP is not enabled for memory type %u\n",
				  (unsigned)type);
			return -EINVAL;
		}
		man->io_offset = dev->agp->agp_info.aper_base;
		man->io_size = dev->agp->agp_info.aper_size * 1024 * 1024;
		man->io_addr = NULL;
		man->flags = _DRM_FLAG_MEMTYPE_MAPPABLE |
		    _DRM_FLAG_MEMTYPE_CSELECT | _DRM_FLAG_NEEDS_IOREMAP;
		man->drm_bus_maptype = _DRM_AGP;
		i915_init_priv1(dev);
		break;
	case DRM_BO_MEM_PRIV0:
		if (!(drm_core_has_AGP(dev) && dev->agp)) {
			DRM_ERROR("AGP is not enabled for memory type %u\n",
				  (unsigned)type);
			return -EINVAL;
		}
		man->io_offset = dev->agp->agp_info.aper_base;
		man->io_size = dev->agp->agp_info.aper_size * 1024 * 1024;
		man->io_addr = NULL;
		man->flags =  _DRM_FLAG_MEMTYPE_MAPPABLE |
		    _DRM_FLAG_MEMTYPE_FIXED | _DRM_FLAG_NEEDS_IOREMAP;
		man->drm_bus_maptype = _DRM_AGP;
		break;
	case DRM_BO_MEM_PRIV1:
		if (!dev_priv) {
			DRM_ERROR("called without initialization\n");
			return DRM_ERR(EINVAL);
		}
		man->io_offset = 0;
		man->io_size = (1 << dev_priv->priv1_order) << PAGE_SHIFT;
		man->io_addr = (void*)dev_priv->priv1_addr;
		man->io_addr -= virt_to_phys(man->io_addr);
		man->flags =  _DRM_FLAG_MEMTYPE_MAPPABLE |
			_DRM_FLAG_MEMTYPE_FIXED;
		man->drm_bus_maptype = _DRM_TTM;
		break;
	default:
		DRM_ERROR("Unsupported memory type %u\n", (unsigned)type);
		return -EINVAL;
	}
	return 0;
}

uint32_t i915_evict_mask(drm_buffer_object_t *bo)
{
	switch (bo->mem.mem_type) {
	case DRM_BO_MEM_PRIV1:
	case DRM_BO_MEM_LOCAL:
	case DRM_BO_MEM_TT:
		return DRM_BO_FLAG_MEM_LOCAL;
	default:
		return DRM_BO_FLAG_MEM_TT | DRM_BO_FLAG_CACHED;
	}
}

static void i915_emit_copy_blit(drm_device_t * dev,
				uint32_t src_offset,
				uint32_t dst_offset,
				uint32_t pages, int direction)
{
	uint32_t cur_pages;
	uint32_t stride = PAGE_SIZE;
	drm_i915_private_t *dev_priv = dev->dev_private;
	RING_LOCALS;

	if (!dev_priv)
		return;

	i915_kernel_lost_context(dev_priv, &dev_priv->ring);
	while (pages > 0) {
		cur_pages = pages;
		if (cur_pages > 2048)
			cur_pages = 2048;
		pages -= cur_pages;

		BEGIN_RING(&dev_priv->ring, 6);
		OUT_RING(SRC_COPY_BLT_CMD | XY_SRC_COPY_BLT_WRITE_ALPHA |
			 XY_SRC_COPY_BLT_WRITE_RGB);
		OUT_RING((stride & 0xffff) | (0xcc << 16) | (1 << 24) |
			 (1 << 25) | (direction ? (1 << 30) : 0));
		OUT_RING((cur_pages << 16) | PAGE_SIZE);
		OUT_RING(dst_offset);
		OUT_RING(stride & 0xffff);
		OUT_RING(src_offset);
		ADVANCE_RING();
	}
	return;
}

static int i915_move_blit(drm_buffer_object_t * bo,
			  int evict, int no_wait, drm_bo_mem_reg_t * new_mem)
{
	drm_bo_mem_reg_t *old_mem = &bo->mem;
	int dir = 0;

	if (new_mem->mem_type == DRM_BO_MEM_PRIV1)
		return DRM_ERR(EINVAL);

	if ((old_mem->mem_type == new_mem->mem_type) &&
	    (new_mem->mm_node->start <
	     old_mem->mm_node->start + old_mem->mm_node->size)) {
		dir = 1;
	}

	i915_emit_copy_blit(bo->dev,
			    old_mem->mm_node->start << PAGE_SHIFT,
			    new_mem->mm_node->start << PAGE_SHIFT,
			    new_mem->num_pages, dir);

	i915_emit_mi_flush(bo->dev, MI_READ_FLUSH | MI_EXE_FLUSH);

	return drm_bo_move_accel_cleanup(bo, evict, no_wait, 0,
					 DRM_FENCE_TYPE_EXE |
					 DRM_I915_FENCE_TYPE_RW,
					 DRM_I915_FENCE_FLAG_FLUSHED, new_mem);
}

/*
 * Flip destination ttm into cached-coherent AGP, 
 * then blit and subsequently move out again.
 */

static int i915_move_flip(drm_buffer_object_t * bo,
			  int evict, int no_wait, drm_bo_mem_reg_t * new_mem)
{
	drm_device_t *dev = bo->dev;
	drm_bo_mem_reg_t tmp_mem;
	int ret;

	tmp_mem = *new_mem;
	tmp_mem.mm_node = NULL;
	tmp_mem.mask = DRM_BO_FLAG_MEM_TT |
	    DRM_BO_FLAG_CACHED | DRM_BO_FLAG_FORCE_CACHING;

	ret = drm_bo_mem_space(bo, &tmp_mem, no_wait);
	if (ret)
		return ret;

	ret = drm_bind_ttm(bo->ttm, 1, tmp_mem.mm_node->start);
	if (ret)
		goto out_cleanup;

	ret = i915_move_blit(bo, 1, no_wait, &tmp_mem);
	if (ret)
		goto out_cleanup;

	ret = drm_bo_move_ttm(bo, evict, no_wait, new_mem);
out_cleanup:
	if (tmp_mem.mm_node) {
		mutex_lock(&dev->struct_mutex);
		if (tmp_mem.mm_node != bo->pinned_node)
			drm_mm_put_block(tmp_mem.mm_node);
		tmp_mem.mm_node = NULL;
		mutex_unlock(&dev->struct_mutex);
	}
	return ret;
}

int i915_move(drm_buffer_object_t * bo,
	      int evict, int no_wait, drm_bo_mem_reg_t * new_mem)
{
	drm_bo_mem_reg_t *old_mem = &bo->mem;

	if (old_mem->mem_type == DRM_BO_MEM_LOCAL || old_mem->mem_type ==
	    DRM_BO_MEM_PRIV1) {
		return drm_bo_move_memcpy(bo, evict, no_wait, new_mem);
	} else if (new_mem->mem_type == DRM_BO_MEM_LOCAL) {
		if (i915_move_flip(bo, evict, no_wait, new_mem))
			return drm_bo_move_memcpy(bo, evict, no_wait, new_mem);
	} else {
		if (i915_move_blit(bo, evict, no_wait, new_mem))
			return drm_bo_move_memcpy(bo, evict, no_wait, new_mem);
	}
	return 0;
}