/*
 * Copyright 2012 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, 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: Dave Airlie <airlied@redhat.com>
 */

#include <linux/pci.h>

#include <drm/drm_managed.h>
#include <drm/drm_print.h>

#include "ast_drv.h"

static u32 ast_get_vram_size(struct ast_device *ast)
{
        u32 vram_size;
        u8 vgacr99, vgacraa;

        vgacraa = ast_get_index_reg(ast, AST_IO_VGACRI, 0xaa);
        switch (vgacraa & AST_IO_VGACRAA_VGAMEM_SIZE_MASK) {
        case 0:
                vram_size = SZ_8M;
                break;
        case 1:
                vram_size = SZ_16M;
                break;
        case 2:
                vram_size = SZ_32M;
                break;
        case 3:
                vram_size = SZ_64M;
                break;
        }

        vgacr99 = ast_get_index_reg(ast, AST_IO_VGACRI, 0x99);
        switch (vgacr99 & AST_IO_VGACR99_VGAMEM_RSRV_MASK) {
        case 1:
                vram_size -= SZ_1M;
                break;
        case 2:
                vram_size -= SZ_2M;
                break;
        case 3:
                vram_size -= SZ_4M;
                break;
        }

        return vram_size;
}

int ast_mm_init(struct ast_device *ast)
{
        struct drm_device *dev = &ast->base;
        struct pci_dev *pdev = to_pci_dev(dev->dev);
        resource_size_t base, size;
        u32 vram_size;

        base = pci_resource_start(pdev, 0);
        size = pci_resource_len(pdev, 0);

        /* Don't fail on errors, but performance might be reduced. */
        devm_arch_io_reserve_memtype_wc(dev->dev, base, size);
        devm_arch_phys_wc_add(dev->dev, base, size);

        vram_size = ast_get_vram_size(ast);

        ast->vram = devm_ioremap_wc(dev->dev, base, vram_size);
        if (!ast->vram)
                return -ENOMEM;

        ast->vram_base = base;
        ast->vram_size = vram_size;

        return 0;
}