// SPDX-License-Identifier: GPL-2.0-only
/*
 * Copyright (C) 2011 Texas Instruments Incorporated - https://www.ti.com/
 * Author: Rob Clark <rob@ti.com>
 */

#include <linux/dma-mapping.h>
#include <linux/platform_device.h>
#include <linux/of.h>
#include <linux/sort.h>
#include <linux/sys_soc.h>

#include <drm/drm_atomic.h>
#include <drm/drm_atomic_helper.h>
#include <drm/drm_bridge.h>
#include <drm/drm_bridge_connector.h>
#include <drm/drm_drv.h>
#include <drm/drm_file.h>
#include <drm/drm_ioctl.h>
#include <drm/drm_panel.h>
#include <drm/drm_prime.h>
#include <drm/drm_print.h>
#include <drm/drm_probe_helper.h>
#include <drm/drm_vblank.h>

#include "omap_dmm_tiler.h"
#include "omap_drv.h"
#include "omap_fbdev.h"

#define DRIVER_NAME             MODULE_NAME
#define DRIVER_DESC             "OMAP DRM"
#define DRIVER_MAJOR            1
#define DRIVER_MINOR            0
#define DRIVER_PATCHLEVEL       0

/*
 * mode config funcs
 */

/* Notes about mapping DSS and DRM entities:
 *    CRTC:        overlay
 *    encoder:     manager.. with some extension to allow one primary CRTC
 *                 and zero or more video CRTC's to be mapped to one encoder?
 *    connector:   dssdev.. manager can be attached/detached from different
 *                 devices
 */

static void omap_atomic_wait_for_completion(struct drm_device *dev,
                                            struct drm_atomic_state *old_state)
{
        struct drm_crtc_state *new_crtc_state;
        struct drm_crtc *crtc;
        unsigned int i;
        int ret;

        for_each_new_crtc_in_state(old_state, crtc, new_crtc_state, i) {
                if (!new_crtc_state->active)
                        continue;

                ret = omap_crtc_wait_pending(crtc);

                if (!ret)
                        dev_warn(dev->dev,
                                 "atomic complete timeout (pipe %u)!\n", i);
        }
}

static void omap_atomic_commit_tail(struct drm_atomic_state *old_state)
{
        struct drm_device *dev = old_state->dev;
        struct omap_drm_private *priv = dev->dev_private;

        dispc_runtime_get(priv->dispc);

        /* Apply the atomic update. */
        drm_atomic_helper_commit_modeset_disables(dev, old_state);

        if (priv->omaprev != 0x3430) {
                /* With the current dss dispc implementation we have to enable
                 * the new modeset before we can commit planes. The dispc ovl
                 * configuration relies on the video mode configuration been
                 * written into the HW when the ovl configuration is
                 * calculated.
                 *
                 * This approach is not ideal because after a mode change the
                 * plane update is executed only after the first vblank
                 * interrupt. The dispc implementation should be fixed so that
                 * it is able use uncommitted drm state information.
                 */
                drm_atomic_helper_commit_modeset_enables(dev, old_state);
                omap_atomic_wait_for_completion(dev, old_state);

                drm_atomic_helper_commit_planes(dev, old_state, 0);

                drm_atomic_helper_commit_hw_done(old_state);
        } else {
                /*
                 * OMAP3 DSS seems to have issues with the work-around above,
                 * resulting in endless sync losts if a crtc is enabled without
                 * a plane. For now, skip the WA for OMAP3.
                 */
                drm_atomic_helper_commit_planes(dev, old_state, 0);

                drm_atomic_helper_commit_modeset_enables(dev, old_state);

                drm_atomic_helper_commit_hw_done(old_state);
        }

        /*
         * Wait for completion of the page flips to ensure that old buffers
         * can't be touched by the hardware anymore before cleaning up planes.
         */
        omap_atomic_wait_for_completion(dev, old_state);

        drm_atomic_helper_cleanup_planes(dev, old_state);

        dispc_runtime_put(priv->dispc);
}

static int drm_atomic_state_normalized_zpos_cmp(const void *a, const void *b)
{
        const struct drm_plane_state *sa = *(struct drm_plane_state **)a;
        const struct drm_plane_state *sb = *(struct drm_plane_state **)b;

        if (sa->normalized_zpos != sb->normalized_zpos)
                return sa->normalized_zpos - sb->normalized_zpos;
        else
                return sa->plane->base.id - sb->plane->base.id;
}

/*
 * This replaces the drm_atomic_normalize_zpos to handle the dual overlay case.
 *
 * Since both halves need to be 'appear' side by side the zpos is
 * recalculated when dealing with dual overlay cases so that the other
 * planes zpos is consistent.
 */
static int omap_atomic_update_normalize_zpos(struct drm_device *dev,
                                             struct drm_atomic_state *state)
{
        struct drm_crtc *crtc;
        struct drm_crtc_state *old_state, *new_state;
        struct drm_plane *plane;
        int c, i, n, inc;
        int total_planes = dev->mode_config.num_total_plane;
        struct drm_plane_state **states;
        int ret = 0;

        states = kmalloc_objs(*states, total_planes);
        if (!states)
                return -ENOMEM;

        for_each_oldnew_crtc_in_state(state, crtc, old_state, new_state, c) {
                if (old_state->plane_mask == new_state->plane_mask &&
                    !new_state->zpos_changed)
                        continue;

                /* Reset plane increment and index value for every crtc */
                n = 0;

                /*
                 * Normalization process might create new states for planes
                 * which normalized_zpos has to be recalculated.
                 */
                drm_for_each_plane_mask(plane, dev, new_state->plane_mask) {
                        struct drm_plane_state *plane_state =
                                drm_atomic_get_plane_state(new_state->state,
                                                           plane);
                        if (IS_ERR(plane_state)) {
                                ret = PTR_ERR(plane_state);
                                goto done;
                        }
                        states[n++] = plane_state;
                }

                sort(states, n, sizeof(*states),
                     drm_atomic_state_normalized_zpos_cmp, NULL);

                for (i = 0, inc = 0; i < n; i++) {
                        plane = states[i]->plane;

                        states[i]->normalized_zpos = i + inc;
                        DRM_DEBUG_ATOMIC("[PLANE:%d:%s] updated normalized zpos value %d\n",
                                         plane->base.id, plane->name,
                                         states[i]->normalized_zpos);

                        if (is_omap_plane_dual_overlay(states[i]))
                                inc++;
                }
                new_state->zpos_changed = true;
        }

done:
        kfree(states);
        return ret;
}

static int omap_atomic_check(struct drm_device *dev,
                             struct drm_atomic_state *state)
{
        int ret;

        ret = drm_atomic_helper_check(dev, state);
        if (ret)
                return ret;

        if (dev->mode_config.normalize_zpos) {
                ret = omap_atomic_update_normalize_zpos(dev, state);
                if (ret)
                        return ret;
        }

        return 0;
}

static const struct drm_mode_config_helper_funcs omap_mode_config_helper_funcs = {
        .atomic_commit_tail = omap_atomic_commit_tail,
};

static const struct drm_mode_config_funcs omap_mode_config_funcs = {
        .fb_create = omap_framebuffer_create,
        .atomic_check = omap_atomic_check,
        .atomic_commit = drm_atomic_helper_commit,
};

/* Global/shared object state funcs */

/*
 * This is a helper that returns the private state currently in operation.
 * Note that this would return the "old_state" if called in the atomic check
 * path, and the "new_state" after the atomic swap has been done.
 */
struct omap_global_state *
omap_get_existing_global_state(struct omap_drm_private *priv)
{
        return to_omap_global_state(priv->glob_obj.state);
}

/*
 * This acquires the modeset lock set aside for global state, creates
 * a new duplicated private object state.
 */
struct omap_global_state *__must_check
omap_get_global_state(struct drm_atomic_state *s)
{
        struct omap_drm_private *priv = s->dev->dev_private;
        struct drm_private_state *priv_state;

        priv_state = drm_atomic_get_private_obj_state(s, &priv->glob_obj);
        if (IS_ERR(priv_state))
                return ERR_CAST(priv_state);

        return to_omap_global_state(priv_state);
}

static struct drm_private_state *
omap_global_duplicate_state(struct drm_private_obj *obj)
{
        struct omap_global_state *state;

        state = kmemdup(obj->state, sizeof(*state), GFP_KERNEL);
        if (!state)
                return NULL;

        __drm_atomic_helper_private_obj_duplicate_state(obj, &state->base);

        return &state->base;
}

static void omap_global_destroy_state(struct drm_private_obj *obj,
                                      struct drm_private_state *state)
{
        struct omap_global_state *omap_state = to_omap_global_state(state);

        kfree(omap_state);
}

static const struct drm_private_state_funcs omap_global_state_funcs = {
        .atomic_duplicate_state = omap_global_duplicate_state,
        .atomic_destroy_state = omap_global_destroy_state,
};

static int omap_global_obj_init(struct drm_device *dev)
{
        struct omap_drm_private *priv = dev->dev_private;
        struct omap_global_state *state;

        state = kzalloc_obj(*state);
        if (!state)
                return -ENOMEM;

        drm_atomic_private_obj_init(dev, &priv->glob_obj, &state->base,
                                    &omap_global_state_funcs);
        return 0;
}

static void omap_global_obj_fini(struct omap_drm_private *priv)
{
        drm_atomic_private_obj_fini(&priv->glob_obj);
}

static void omap_disconnect_pipelines(struct drm_device *ddev)
{
        struct omap_drm_private *priv = ddev->dev_private;
        unsigned int i;

        for (i = 0; i < priv->num_pipes; i++) {
                struct omap_drm_pipeline *pipe = &priv->pipes[i];

                omapdss_device_disconnect(priv->dss, pipe->output);

                omapdss_device_put(pipe->output);
                pipe->output = NULL;
        }

        memset(&priv->channels, 0, sizeof(priv->channels));

        priv->num_pipes = 0;
}

static int omap_connect_pipelines(struct drm_device *ddev)
{
        struct omap_drm_private *priv = ddev->dev_private;
        struct omap_dss_device *output = NULL;
        int r;

        for_each_dss_output(output) {
                r = omapdss_device_connect(priv->dss, output);
                if (r == -EPROBE_DEFER) {
                        omapdss_device_put(output);
                        return r;
                } else if (r) {
                        dev_warn(output->dev, "could not connect output %s\n",
                                 output->name);
                } else {
                        struct omap_drm_pipeline *pipe;

                        pipe = &priv->pipes[priv->num_pipes++];
                        pipe->output = omapdss_device_get(output);

                        if (priv->num_pipes == ARRAY_SIZE(priv->pipes)) {
                                /* To balance the 'for_each_dss_output' loop */
                                omapdss_device_put(output);
                                break;
                        }
                }
        }

        return 0;
}

static int omap_compare_pipelines(const void *a, const void *b)
{
        const struct omap_drm_pipeline *pipe1 = a;
        const struct omap_drm_pipeline *pipe2 = b;

        if (pipe1->alias_id > pipe2->alias_id)
                return 1;
        else if (pipe1->alias_id < pipe2->alias_id)
                return -1;
        return 0;
}

static int omap_modeset_init_properties(struct drm_device *dev)
{
        struct omap_drm_private *priv = dev->dev_private;
        unsigned int num_planes = dispc_get_num_ovls(priv->dispc);

        priv->zorder_prop = drm_property_create_range(dev, 0, "zorder", 0,
                                                      num_planes - 1);
        if (!priv->zorder_prop)
                return -ENOMEM;

        return 0;
}

static int omap_display_id(struct omap_dss_device *output)
{
        struct device_node *node = NULL;

        if (output->bridge) {
                struct drm_bridge *bridge __free(drm_bridge_put) =
                        drm_bridge_chain_get_last_bridge(output->bridge->encoder);

                node = bridge->of_node;
        }

        return node ? of_alias_get_id(node, "display") : -ENODEV;
}

static int omap_modeset_init(struct drm_device *dev)
{
        struct omap_drm_private *priv = dev->dev_private;
        int num_ovls = dispc_get_num_ovls(priv->dispc);
        int num_mgrs = dispc_get_num_mgrs(priv->dispc);
        unsigned int i;
        int ret;
        u32 plane_crtc_mask;

        if (!omapdss_stack_is_ready())
                return -EPROBE_DEFER;

        ret = omap_modeset_init_properties(dev);
        if (ret < 0)
                return ret;

        /*
         * This function creates exactly one connector, encoder, crtc,
         * and primary plane per each connected dss-device. Each
         * connector->encoder->crtc chain is expected to be separate
         * and each crtc is connect to a single dss-channel. If the
         * configuration does not match the expectations or exceeds
         * the available resources, the configuration is rejected.
         */
        ret = omap_connect_pipelines(dev);
        if (ret < 0)
                return ret;

        if (priv->num_pipes > num_mgrs || priv->num_pipes > num_ovls) {
                dev_err(dev->dev, "%s(): Too many connected displays\n",
                        __func__);
                return -EINVAL;
        }

        /* Create all planes first. They can all be put to any CRTC. */
        plane_crtc_mask = (1 << priv->num_pipes) - 1;

        for (i = 0; i < num_ovls; i++) {
                enum drm_plane_type type = i < priv->num_pipes
                                         ? DRM_PLANE_TYPE_PRIMARY
                                         : DRM_PLANE_TYPE_OVERLAY;
                struct drm_plane *plane;

                if (WARN_ON(priv->num_planes >= ARRAY_SIZE(priv->planes)))
                        return -EINVAL;

                plane = omap_plane_init(dev, i, type, plane_crtc_mask);
                if (IS_ERR(plane))
                        return PTR_ERR(plane);

                priv->planes[priv->num_planes++] = plane;
        }

        /*
         * Create the encoders, attach the bridges and get the pipeline alias
         * IDs.
         */
        for (i = 0; i < priv->num_pipes; i++) {
                struct omap_drm_pipeline *pipe = &priv->pipes[i];
                int id;

                pipe->encoder = omap_encoder_init(dev, pipe->output);
                if (!pipe->encoder)
                        return -ENOMEM;

                if (pipe->output->bridge) {
                        ret = drm_bridge_attach(pipe->encoder,
                                                pipe->output->bridge, NULL,
                                                DRM_BRIDGE_ATTACH_NO_CONNECTOR);
                        if (ret < 0)
                                return ret;
                }

                id = omap_display_id(pipe->output);
                pipe->alias_id = id >= 0 ? id : i;
        }

        /* Sort the pipelines by DT aliases. */
        sort(priv->pipes, priv->num_pipes, sizeof(priv->pipes[0]),
             omap_compare_pipelines, NULL);

        /*
         * Populate the pipeline lookup table by DISPC channel. Only one display
         * is allowed per channel.
         */
        for (i = 0; i < priv->num_pipes; ++i) {
                struct omap_drm_pipeline *pipe = &priv->pipes[i];
                enum omap_channel channel = pipe->output->dispc_channel;

                if (WARN_ON(priv->channels[channel] != NULL))
                        return -EINVAL;

                priv->channels[channel] = pipe;
        }

        /* Create the connectors and CRTCs. */
        for (i = 0; i < priv->num_pipes; i++) {
                struct omap_drm_pipeline *pipe = &priv->pipes[i];
                struct drm_encoder *encoder = pipe->encoder;
                struct drm_crtc *crtc;

                pipe->connector = drm_bridge_connector_init(dev, encoder);
                if (IS_ERR(pipe->connector)) {
                        dev_err(priv->dev,
                                "unable to create bridge connector for %s\n",
                                pipe->output->name);
                        return PTR_ERR(pipe->connector);
                }

                drm_connector_attach_encoder(pipe->connector, encoder);

                crtc = omap_crtc_init(dev, pipe, priv->planes[i]);
                if (IS_ERR(crtc))
                        return PTR_ERR(crtc);

                encoder->possible_crtcs = 1 << i;
                pipe->crtc = crtc;
        }

        DBG("registered %u planes, %u crtcs/encoders/connectors\n",
            priv->num_planes, priv->num_pipes);

        dev->mode_config.min_width = 8;
        dev->mode_config.min_height = 2;

        /*
         * Note: these values are used for multiple independent things:
         * connector mode filtering, buffer sizes, crtc sizes...
         * Use big enough values here to cover all use cases, and do more
         * specific checking in the respective code paths.
         */
        dev->mode_config.max_width = 8192;
        dev->mode_config.max_height = 8192;

        /* We want the zpos to be normalized */
        dev->mode_config.normalize_zpos = true;

        dev->mode_config.funcs = &omap_mode_config_funcs;
        dev->mode_config.helper_private = &omap_mode_config_helper_funcs;

        drm_mode_config_reset(dev);

        omap_drm_irq_install(dev);

        return 0;
}

static void omap_modeset_fini(struct drm_device *ddev)
{
        omap_drm_irq_uninstall(ddev);

        drm_mode_config_cleanup(ddev);
}

/*
 * drm ioctl funcs
 */


static int ioctl_get_param(struct drm_device *dev, void *data,
                struct drm_file *file_priv)
{
        struct omap_drm_private *priv = dev->dev_private;
        struct drm_omap_param *args = data;

        DBG("%p: param=%llu", dev, args->param);

        switch (args->param) {
        case OMAP_PARAM_CHIPSET_ID:
                args->value = priv->omaprev;
                break;
        default:
                DBG("unknown parameter %lld", args->param);
                return -EINVAL;
        }

        return 0;
}

#define OMAP_BO_USER_MASK       0x00ffffff      /* flags settable by userspace */

static int ioctl_gem_new(struct drm_device *dev, void *data,
                struct drm_file *file_priv)
{
        struct drm_omap_gem_new *args = data;
        u32 flags = args->flags & OMAP_BO_USER_MASK;

        VERB("%p:%p: size=0x%08x, flags=%08x", dev, file_priv,
             args->size.bytes, flags);

        return omap_gem_new_handle(dev, file_priv, args->size, flags,
                                   &args->handle);
}

static int ioctl_gem_info(struct drm_device *dev, void *data,
                struct drm_file *file_priv)
{
        struct drm_omap_gem_info *args = data;
        struct drm_gem_object *obj;
        int ret = 0;

        VERB("%p:%p: handle=%d", dev, file_priv, args->handle);

        obj = drm_gem_object_lookup(file_priv, args->handle);
        if (!obj)
                return -ENOENT;

        args->size = omap_gem_mmap_size(obj);
        args->offset = omap_gem_mmap_offset(obj);

        drm_gem_object_put(obj);

        return ret;
}

static const struct drm_ioctl_desc ioctls[DRM_COMMAND_END - DRM_COMMAND_BASE] = {
        DRM_IOCTL_DEF_DRV(OMAP_GET_PARAM, ioctl_get_param,
                          DRM_RENDER_ALLOW),
        DRM_IOCTL_DEF_DRV(OMAP_SET_PARAM, drm_invalid_op,
                          DRM_AUTH | DRM_MASTER | DRM_ROOT_ONLY),
        DRM_IOCTL_DEF_DRV(OMAP_GEM_NEW, ioctl_gem_new,
                          DRM_RENDER_ALLOW),
        /* Deprecated, to be removed. */
        DRM_IOCTL_DEF_DRV(OMAP_GEM_CPU_PREP, drm_noop,
                          DRM_RENDER_ALLOW),
        /* Deprecated, to be removed. */
        DRM_IOCTL_DEF_DRV(OMAP_GEM_CPU_FINI, drm_noop,
                          DRM_RENDER_ALLOW),
        DRM_IOCTL_DEF_DRV(OMAP_GEM_INFO, ioctl_gem_info,
                          DRM_RENDER_ALLOW),
};

/*
 * drm driver funcs
 */

static int dev_open(struct drm_device *dev, struct drm_file *file)
{
        file->driver_priv = NULL;

        DBG("open: dev=%p, file=%p", dev, file);

        return 0;
}

DEFINE_DRM_GEM_FOPS(omapdriver_fops);

static const struct drm_driver omap_drm_driver = {
        .driver_features = DRIVER_MODESET | DRIVER_GEM  |
                DRIVER_ATOMIC | DRIVER_RENDER,
        .open = dev_open,
#ifdef CONFIG_DEBUG_FS
        .debugfs_init = omap_debugfs_init,
#endif
        .gem_prime_import = omap_gem_prime_import,
        .dumb_create = omap_gem_dumb_create,
        .dumb_map_offset = omap_gem_dumb_map_offset,
        OMAP_FBDEV_DRIVER_OPS,
        .ioctls = ioctls,
        .num_ioctls = DRM_OMAP_NUM_IOCTLS,
        .fops = &omapdriver_fops,
        .name = DRIVER_NAME,
        .desc = DRIVER_DESC,
        .major = DRIVER_MAJOR,
        .minor = DRIVER_MINOR,
        .patchlevel = DRIVER_PATCHLEVEL,
};

static const struct soc_device_attribute omapdrm_soc_devices[] = {
        { .family = "OMAP3", .data = (void *)0x3430 },
        { .family = "OMAP4", .data = (void *)0x4430 },
        { .family = "OMAP5", .data = (void *)0x5430 },
        { .family = "DRA7",  .data = (void *)0x0752 },
        { /* sentinel */ }
};

static int omapdrm_init(struct omap_drm_private *priv, struct device *dev)
{
        const struct soc_device_attribute *soc;
        struct dss_pdata *pdata = dev->platform_data;
        struct drm_device *ddev;
        int ret;

        DBG("%s", dev_name(dev));

        if (drm_firmware_drivers_only())
                return -ENODEV;

        /* Allocate and initialize the DRM device. */
        ddev = drm_dev_alloc(&omap_drm_driver, dev);
        if (IS_ERR(ddev))
                return PTR_ERR(ddev);

        priv->ddev = ddev;
        ddev->dev_private = priv;

        priv->dev = dev;
        priv->dss = pdata->dss;
        priv->dispc = dispc_get_dispc(priv->dss);

        priv->dss->mgr_ops_priv = priv;

        soc = soc_device_match(omapdrm_soc_devices);
        priv->omaprev = soc ? (uintptr_t)soc->data : 0;
        priv->wq = alloc_ordered_workqueue("omapdrm", 0);
        if (!priv->wq) {
                ret = -ENOMEM;
                goto err_alloc_workqueue;
        }

        mutex_init(&priv->list_lock);
        INIT_LIST_HEAD(&priv->obj_list);

        /* Get memory bandwidth limits */
        priv->max_bandwidth = dispc_get_memory_bandwidth_limit(priv->dispc);

        omap_gem_init(ddev);

        drm_mode_config_init(ddev);

        ret = omap_global_obj_init(ddev);
        if (ret)
                goto err_gem_deinit;

        ret = omap_hwoverlays_init(priv);
        if (ret)
                goto err_free_priv_obj;

        ret = omap_modeset_init(ddev);
        if (ret) {
                dev_err(priv->dev, "omap_modeset_init failed: ret=%d\n", ret);
                goto err_free_overlays;
        }

        /* Initialize vblank handling, start with all CRTCs disabled. */
        ret = drm_vblank_init(ddev, priv->num_pipes);
        if (ret) {
                dev_err(priv->dev, "could not init vblank\n");
                goto err_cleanup_modeset;
        }

        drm_kms_helper_poll_init(ddev);

        /*
         * Register the DRM device with the core and the connectors with
         * sysfs.
         */
        ret = drm_dev_register(ddev, 0);
        if (ret)
                goto err_cleanup_helpers;

        omap_fbdev_setup(ddev);

        return 0;

err_cleanup_helpers:
        drm_kms_helper_poll_fini(ddev);
err_cleanup_modeset:
        omap_modeset_fini(ddev);
err_free_overlays:
        omap_hwoverlays_destroy(priv);
err_free_priv_obj:
        omap_global_obj_fini(priv);
err_gem_deinit:
        drm_mode_config_cleanup(ddev);
        omap_gem_deinit(ddev);
        destroy_workqueue(priv->wq);
err_alloc_workqueue:
        omap_disconnect_pipelines(ddev);
        drm_dev_put(ddev);
        return ret;
}

static void omapdrm_cleanup(struct omap_drm_private *priv)
{
        struct drm_device *ddev = priv->ddev;

        DBG("");

        drm_dev_unregister(ddev);

        drm_kms_helper_poll_fini(ddev);

        drm_atomic_helper_shutdown(ddev);

        omap_modeset_fini(ddev);
        omap_hwoverlays_destroy(priv);
        omap_global_obj_fini(priv);
        drm_mode_config_cleanup(ddev);
        omap_gem_deinit(ddev);

        destroy_workqueue(priv->wq);

        omap_disconnect_pipelines(ddev);

        drm_dev_put(ddev);
}

static int pdev_probe(struct platform_device *pdev)
{
        struct omap_drm_private *priv;
        int ret;

        ret = dma_coerce_mask_and_coherent(&pdev->dev, DMA_BIT_MASK(32));
        if (ret) {
                dev_err(&pdev->dev, "Failed to set the DMA mask\n");
                return ret;
        }

        /* Allocate and initialize the driver private structure. */
        priv = kzalloc_obj(*priv);
        if (!priv)
                return -ENOMEM;

        platform_set_drvdata(pdev, priv);

        ret = omapdrm_init(priv, &pdev->dev);
        if (ret < 0)
                kfree(priv);

        return ret;
}

static void pdev_remove(struct platform_device *pdev)
{
        struct omap_drm_private *priv = platform_get_drvdata(pdev);

        omapdrm_cleanup(priv);
        kfree(priv);
}

static void pdev_shutdown(struct platform_device *pdev)
{
        struct omap_drm_private *priv = platform_get_drvdata(pdev);

        drm_atomic_helper_shutdown(priv->ddev);
}

#ifdef CONFIG_PM_SLEEP
static int omap_drm_suspend(struct device *dev)
{
        struct omap_drm_private *priv = dev_get_drvdata(dev);
        struct drm_device *drm_dev = priv->ddev;

        return drm_mode_config_helper_suspend(drm_dev);
}

static int omap_drm_resume(struct device *dev)
{
        struct omap_drm_private *priv = dev_get_drvdata(dev);
        struct drm_device *drm_dev = priv->ddev;

        drm_mode_config_helper_resume(drm_dev);

        return omap_gem_resume(drm_dev);
}
#endif

static SIMPLE_DEV_PM_OPS(omapdrm_pm_ops, omap_drm_suspend, omap_drm_resume);

static struct platform_driver pdev = {
        .driver = {
                .name = "omapdrm",
                .pm = &omapdrm_pm_ops,
        },
        .probe = pdev_probe,
        .remove = pdev_remove,
        .shutdown = pdev_shutdown,
};

static struct platform_driver * const drivers[] = {
        &omap_dmm_driver,
        &pdev,
};

static int __init omap_drm_init(void)
{
        int r;

        DBG("init");

        r = omap_dss_init();
        if (r)
                return r;

        r = platform_register_drivers(drivers, ARRAY_SIZE(drivers));
        if (r) {
                omap_dss_exit();
                return r;
        }

        return 0;
}

static void __exit omap_drm_fini(void)
{
        DBG("fini");

        platform_unregister_drivers(drivers, ARRAY_SIZE(drivers));

        omap_dss_exit();
}

module_init(omap_drm_init);
module_exit(omap_drm_fini);

MODULE_AUTHOR("Rob Clark <rob@ti.com>");
MODULE_AUTHOR("Tomi Valkeinen <tomi.valkeinen@ti.com>");
MODULE_DESCRIPTION("OMAP DRM Display Driver");
MODULE_ALIAS("platform:" DRIVER_NAME);
MODULE_LICENSE("GPL v2");