/*
 * 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/console.h>
#include <linux/module.h>
#include <linux/pci.h>

#include <drm/drm_crtc_helper.h>
#include <drm/drm_drv.h>
#include <drm/drm_gem_vram_helper.h>
#include <drm/drm_pci.h>
#include <drm/drm_probe_helper.h>
#include <drm/drm_vram_mm_helper.h>

#include "ast_drv.h"

int ast_modeset = -1;

MODULE_PARM_DESC(modeset, "Disable/Enable modesetting");
module_param_named(modeset, ast_modeset, int, 0400);

#define PCI_VENDOR_ASPEED 0x1a03

static struct drm_driver driver;

#define AST_VGA_DEVICE(id, info) {              \
        .class = PCI_BASE_CLASS_DISPLAY << 16,  \
        .class_mask = 0xff0000,                 \
        .vendor = PCI_VENDOR_ASPEED,                    \
        .device = id,                           \
        .subvendor = PCI_ANY_ID,                \
        .subdevice = PCI_ANY_ID,                \
        .driver_data = (unsigned long) info }

static const struct pci_device_id pciidlist[] = {
        AST_VGA_DEVICE(PCI_CHIP_AST2000, NULL),
        AST_VGA_DEVICE(PCI_CHIP_AST2100, NULL),
        /*      AST_VGA_DEVICE(PCI_CHIP_AST1180, NULL), - don't bind to 1180 for now */
        {0, 0, 0},
};

MODULE_DEVICE_TABLE(pci, pciidlist);

static void ast_kick_out_firmware_fb(struct pci_dev *pdev)
{
        struct apertures_struct *ap;
        bool primary = false;

        ap = alloc_apertures(1);
        if (!ap)
                return;

        ap->ranges[0].base = pci_resource_start(pdev, 0);
        ap->ranges[0].size = pci_resource_len(pdev, 0);

#ifdef CONFIG_X86
        primary = pdev->resource[PCI_ROM_RESOURCE].flags & IORESOURCE_ROM_SHADOW;
#endif
        drm_fb_helper_remove_conflicting_framebuffers(ap, "astdrmfb", primary);
        kfree(ap);
}

static int ast_pci_probe(struct pci_dev *pdev, const struct pci_device_id *ent)
{
        ast_kick_out_firmware_fb(pdev);

        return drm_get_pci_dev(pdev, ent, &driver);
}

static void
ast_pci_remove(struct pci_dev *pdev)
{
        struct drm_device *dev = pci_get_drvdata(pdev);

        drm_put_dev(dev);
}



static int ast_drm_freeze(struct drm_device *dev)
{
        drm_kms_helper_poll_disable(dev);
        pci_save_state(dev->pdev);
        drm_fb_helper_set_suspend_unlocked(dev->fb_helper, true);

        return 0;
}

static int ast_drm_thaw(struct drm_device *dev)
{
        ast_post_gpu(dev);

        drm_mode_config_reset(dev);
        drm_helper_resume_force_mode(dev);
        drm_fb_helper_set_suspend_unlocked(dev->fb_helper, false);

        return 0;
}

static int ast_drm_resume(struct drm_device *dev)
{
        int ret;

        if (pci_enable_device(dev->pdev))
                return -EIO;

        ret = ast_drm_thaw(dev);
        if (ret)
                return ret;

        drm_kms_helper_poll_enable(dev);
        return 0;
}

static int ast_pm_suspend(struct device *dev)
{
        struct pci_dev *pdev = to_pci_dev(dev);
        struct drm_device *ddev = pci_get_drvdata(pdev);
        int error;

        error = ast_drm_freeze(ddev);
        if (error)
                return error;

        pci_disable_device(pdev);
        pci_set_power_state(pdev, PCI_D3hot);
        return 0;
}
static int ast_pm_resume(struct device *dev)
{
        struct pci_dev *pdev = to_pci_dev(dev);
        struct drm_device *ddev = pci_get_drvdata(pdev);
        return ast_drm_resume(ddev);
}

static int ast_pm_freeze(struct device *dev)
{
        struct pci_dev *pdev = to_pci_dev(dev);
        struct drm_device *ddev = pci_get_drvdata(pdev);

        if (!ddev || !ddev->dev_private)
                return -ENODEV;
        return ast_drm_freeze(ddev);

}

static int ast_pm_thaw(struct device *dev)
{
        struct pci_dev *pdev = to_pci_dev(dev);
        struct drm_device *ddev = pci_get_drvdata(pdev);
        return ast_drm_thaw(ddev);
}

static int ast_pm_poweroff(struct device *dev)
{
        struct pci_dev *pdev = to_pci_dev(dev);
        struct drm_device *ddev = pci_get_drvdata(pdev);

        return ast_drm_freeze(ddev);
}

static const struct dev_pm_ops ast_pm_ops = {
        .suspend = ast_pm_suspend,
        .resume = ast_pm_resume,
        .freeze = ast_pm_freeze,
        .thaw = ast_pm_thaw,
        .poweroff = ast_pm_poweroff,
        .restore = ast_pm_resume,
};

static struct pci_driver ast_pci_driver = {
        .name = DRIVER_NAME,
        .id_table = pciidlist,
        .probe = ast_pci_probe,
        .remove = ast_pci_remove,
        .driver.pm = &ast_pm_ops,
};

static const struct file_operations ast_fops = {
        .owner = THIS_MODULE,
        DRM_VRAM_MM_FILE_OPERATIONS
};

static struct drm_driver driver = {
        .driver_features = DRIVER_MODESET | DRIVER_GEM,

        .load = ast_driver_load,
        .unload = ast_driver_unload,

        .fops = &ast_fops,
        .name = DRIVER_NAME,
        .desc = DRIVER_DESC,
        .date = DRIVER_DATE,
        .major = DRIVER_MAJOR,
        .minor = DRIVER_MINOR,
        .patchlevel = DRIVER_PATCHLEVEL,

        DRM_GEM_VRAM_DRIVER
};

static int __init ast_init(void)
{
        if (vgacon_text_force() && ast_modeset == -1)
                return -EINVAL;

        if (ast_modeset == 0)
                return -EINVAL;
        return pci_register_driver(&ast_pci_driver);
}
static void __exit ast_exit(void)
{
        pci_unregister_driver(&ast_pci_driver);
}

module_init(ast_init);
module_exit(ast_exit);

MODULE_AUTHOR(DRIVER_AUTHOR);
MODULE_DESCRIPTION(DRIVER_DESC);
MODULE_LICENSE("GPL and additional rights");