// SPDX-License-Identifier: GPL-2.0-only
/*
 * Hisilicon Kirin SoCs drm master driver
 *
 * Copyright (c) 2016 Linaro Limited.
 * Copyright (c) 2014-2016 HiSilicon Limited.
 *
 * Author:
 * Xinliang Liu <z.liuxinliang@hisilicon.com>
 * Xinliang Liu <xinliang.liu@linaro.org>
 * Xinwei Kong <kong.kongxinwei@hisilicon.com>
 */

#include <linux/component.h>
#include <linux/module.h>
#include <linux/of.h>
#include <linux/of_graph.h>
#include <linux/platform_device.h>

#include <drm/clients/drm_client_setup.h>
#include <drm/drm_atomic_helper.h>
#include <drm/drm_drv.h>
#include <drm/drm_gem_dma_helper.h>
#include <drm/drm_gem_framebuffer_helper.h>
#include <drm/drm_module.h>
#include <drm/drm_of.h>
#include <drm/drm_probe_helper.h>
#include <drm/drm_vblank.h>

#include "kirin_drm_drv.h"

#define KIRIN_MAX_PLANE 2

struct kirin_drm_private {
 struct kirin_crtc crtc;
 struct kirin_plane planes[KIRIN_MAX_PLANE];
 void *hw_ctx;
};

static int kirin_drm_crtc_init(struct drm_device *dev, struct drm_crtc *crtc,
          struct drm_plane *plane,
          const struct kirin_drm_data *driver_data)
{
 struct device_node *port;
 int ret;

 /* set crtc port so that
  * drm_of_find_possible_crtcs call works
 */
 port = of_get_child_by_name(dev->dev->of_node, "port");
 if (!port) {
  DRM_ERROR("no port node found in %pOF\n", dev->dev->of_node);
  return -EINVAL;
 }
 of_node_put(port);
 crtc->port = port;

 ret = drm_crtc_init_with_planes(dev, crtc, plane, NULL,
     driver_data->crtc_funcs, NULL);
 if (ret) {
  DRM_ERROR("failed to init crtc.\n");
  return ret;
 }

 drm_crtc_helper_add(crtc, driver_data->crtc_helper_funcs);

 return 0;
}

static int kirin_drm_plane_init(struct drm_device *dev, struct drm_plane *plane,
    enum drm_plane_type type,
    const struct kirin_drm_data *data)
{
 int ret = 0;

 ret = drm_universal_plane_init(dev, plane, 1, data->plane_funcs,
           data->channel_formats,
           data->channel_formats_cnt,
           NULL, type, NULL);
 if (ret) {
  DRM_ERROR("fail to init plane, ch=%d\n", 0);
  return ret;
 }

 drm_plane_helper_add(plane, data->plane_helper_funcs);

 return 0;
}

static void kirin_drm_private_cleanup(struct drm_device *dev)
{
 struct kirin_drm_private *kirin_priv = dev->dev_private;
 struct kirin_drm_data *data;

 data = (struct kirin_drm_data *)of_device_get_match_data(dev->dev);
 if (data->cleanup_hw_ctx)
  data->cleanup_hw_ctx(kirin_priv->hw_ctx);

 devm_kfree(dev->dev, kirin_priv);
 dev->dev_private = NULL;
}

static int kirin_drm_private_init(struct drm_device *dev,
      const struct kirin_drm_data *driver_data)
{
 struct platform_device *pdev = to_platform_device(dev->dev);
 struct kirin_drm_private *kirin_priv;
 struct drm_plane *prim_plane;
 enum drm_plane_type type;
 void *ctx;
 int ret;
 u32 ch;

 kirin_priv = devm_kzalloc(dev->dev, sizeof(*kirin_priv), GFP_KERNEL);
 if (!kirin_priv) {
  DRM_ERROR("failed to alloc kirin_drm_private\n");
  return -ENOMEM;
 }

 ctx = driver_data->alloc_hw_ctx(pdev, &kirin_priv->crtc.base);
 if (IS_ERR(ctx)) {
  DRM_ERROR("failed to initialize kirin_priv hw ctx\n");
  return -EINVAL;
 }
 kirin_priv->hw_ctx = ctx;

 /*
  * plane init
  * TODO: Now only support primary plane, overlay planes
  * need to do.
 */
 for (ch = 0; ch < driver_data->num_planes; ch++) {
  if (ch == driver_data->prim_plane)
   type = DRM_PLANE_TYPE_PRIMARY;
  else
   type = DRM_PLANE_TYPE_OVERLAY;
  ret = kirin_drm_plane_init(dev, &kirin_priv->planes[ch].base,
        type, driver_data);
  if (ret)
   return ret;
  kirin_priv->planes[ch].ch = ch;
  kirin_priv->planes[ch].hw_ctx = ctx;
 }

 /* crtc init */
 prim_plane = &kirin_priv->planes[driver_data->prim_plane].base;
 ret = kirin_drm_crtc_init(dev, &kirin_priv->crtc.base,
      prim_plane, driver_data);
 if (ret)
  return ret;
 kirin_priv->crtc.hw_ctx = ctx;
 dev->dev_private = kirin_priv;

 return 0;
}

static int kirin_drm_kms_init(struct drm_device *dev,
         const struct kirin_drm_data *driver_data)
{
 int ret;

 /* dev->mode_config initialization */
 drm_mode_config_init(dev);
 dev->mode_config.min_width = 0;
 dev->mode_config.min_height = 0;
 dev->mode_config.max_width = driver_data->config_max_width;
 dev->mode_config.max_height = driver_data->config_max_width;
 dev->mode_config.funcs = driver_data->mode_config_funcs;

 /* display controller init */
 ret = kirin_drm_private_init(dev, driver_data);
 if (ret)
  goto err_mode_config_cleanup;

 /* bind and init sub drivers */
 ret = component_bind_all(dev->dev, dev);
 if (ret) {
  DRM_ERROR("failed to bind all component.\n");
  goto err_private_cleanup;
 }

 /* vblank init */
 ret = drm_vblank_init(dev, dev->mode_config.num_crtc);
 if (ret) {
  DRM_ERROR("failed to initialize vblank.\n");
  goto err_unbind_all;
 }

 /* reset all the states of crtc/plane/encoder/connector */
 drm_mode_config_reset(dev);

 /* init kms poll for handling hpd */
 drm_kms_helper_poll_init(dev);

 return 0;

err_unbind_all:
 component_unbind_all(dev->dev, dev);
err_private_cleanup:
 kirin_drm_private_cleanup(dev);
err_mode_config_cleanup:
 drm_mode_config_cleanup(dev);
 return ret;
}

static int kirin_drm_kms_cleanup(struct drm_device *dev)
{
 drm_kms_helper_poll_fini(dev);
 drm_atomic_helper_shutdown(dev);
 kirin_drm_private_cleanup(dev);
 drm_mode_config_cleanup(dev);

 return 0;
}

static int kirin_drm_bind(struct device *dev)
{
 struct kirin_drm_data *driver_data;
 struct drm_device *drm_dev;
 int ret;

 driver_data = (struct kirin_drm_data *)of_device_get_match_data(dev);
 if (!driver_data)
  return -EINVAL;

 drm_dev = drm_dev_alloc(driver_data->driver, dev);
 if (IS_ERR(drm_dev))
  return PTR_ERR(drm_dev);
 dev_set_drvdata(dev, drm_dev);

 /* display controller init */
 ret = kirin_drm_kms_init(drm_dev, driver_data);
 if (ret)
  goto err_drm_dev_put;

 ret = drm_dev_register(drm_dev, 0);
 if (ret)
  goto err_kms_cleanup;

 drm_client_setup(drm_dev, NULL);

 return 0;

err_kms_cleanup:
 kirin_drm_kms_cleanup(drm_dev);
err_drm_dev_put:
 drm_dev_put(drm_dev);
 dev_set_drvdata(dev, NULL);

 return ret;
}

static void kirin_drm_unbind(struct device *dev)
{
 struct drm_device *drm_dev = dev_get_drvdata(dev);

 drm_dev_unregister(drm_dev);
 kirin_drm_kms_cleanup(drm_dev);
 drm_dev_put(drm_dev);
 dev_set_drvdata(dev, NULL);
}

static const struct component_master_ops kirin_drm_ops = {
 .bind = kirin_drm_bind,
 .unbind = kirin_drm_unbind,
};

static int kirin_drm_platform_probe(struct platform_device *pdev)
{
 struct device *dev = &pdev->dev;
 struct device_node *np = dev->of_node;
 struct component_match *match = NULL;
 struct device_node *remote;

 remote = of_graph_get_remote_node(np, 0, 0);
 if (!remote)
  return -ENODEV;

 drm_of_component_match_add(dev, &match, component_compare_of, remote);
 of_node_put(remote);

 return component_master_add_with_match(dev, &kirin_drm_ops, match);
}

static void kirin_drm_platform_remove(struct platform_device *pdev)
{
 component_master_del(&pdev->dev, &kirin_drm_ops);
}

static void kirin_drm_platform_shutdown(struct platform_device *pdev)
{
 drm_atomic_helper_shutdown(platform_get_drvdata(pdev));
}

static const struct of_device_id kirin_drm_dt_ids[] = {
 { .compatible = "hisilicon,hi6220-ade",
   .data = &ade_driver_data,
 },
 { /* end node */ },
};
MODULE_DEVICE_TABLE(of, kirin_drm_dt_ids);

static struct platform_driver kirin_drm_platform_driver = {
 .probe = kirin_drm_platform_probe,
 .remove = kirin_drm_platform_remove,
 .shutdown = kirin_drm_platform_shutdown,
 .driver = {
  .name = "kirin-drm",
  .of_match_table = kirin_drm_dt_ids,
 },
};

drm_module_platform_driver(kirin_drm_platform_driver);

MODULE_AUTHOR("Xinliang Liu <xinliang.liu@linaro.org>");
MODULE_AUTHOR("Xinliang Liu <z.liuxinliang@hisilicon.com>");
MODULE_AUTHOR("Xinwei Kong <kong.kongxinwei@hisilicon.com>");
MODULE_DESCRIPTION("hisilicon Kirin SoCs' DRM master driver");
MODULE_LICENSE("GPL v2");