vpu_drv.c

  1// SPDX-License-Identifier: GPL-2.0
  2/*
  3 * Copyright 2020-2021 NXP
  4 */
  5
  6#include <linux/init.h>
  7#include <linux/interconnect.h>
  8#include <linux/ioctl.h>
  9#include <linux/list.h>
 10#include <linux/kernel.h>
 11#include <linux/module.h>
 12#include <linux/dma-map-ops.h>
 13#include <linux/of.h>
 14#include <linux/of_platform.h>
 15#include <linux/platform_device.h>
 16#include <linux/slab.h>
 17#include <linux/types.h>
 18#include <linux/pm_runtime.h>
 19#include <linux/videodev2.h>
 20#include <linux/of_reserved_mem.h>
 21#include <media/v4l2-device.h>
 22#include <media/videobuf2-v4l2.h>
 23#include <media/v4l2-mem2mem.h>
 24#include <media/v4l2-ioctl.h>
 25#include <linux/debugfs.h>
 26#include "vpu.h"
 27#include "vpu_imx8q.h"
 28
 29bool debug;
 30module_param(debug, bool, 0644);
 31
 32void vpu_writel(struct vpu_dev *vpu, u32 reg, u32 val)
 33{
 34	writel(val, vpu->base + reg);
 35}
 36
 37u32 vpu_readl(struct vpu_dev *vpu, u32 reg)
 38{
 39	return readl(vpu->base + reg);
 40}
 41
 42static void vpu_dev_get(struct vpu_dev *vpu)
 43{
 44	if (atomic_inc_return(&vpu->ref_vpu) == 1 && vpu->res->setup)
 45		vpu->res->setup(vpu);
 46}
 47
 48static void vpu_dev_put(struct vpu_dev *vpu)
 49{
 50	atomic_dec(&vpu->ref_vpu);
 51}
 52
 53static void vpu_enc_get(struct vpu_dev *vpu)
 54{
 55	if (atomic_inc_return(&vpu->ref_enc) == 1 && vpu->res->setup_encoder)
 56		vpu->res->setup_encoder(vpu);
 57}
 58
 59static void vpu_enc_put(struct vpu_dev *vpu)
 60{
 61	atomic_dec(&vpu->ref_enc);
 62}
 63
 64static void vpu_dec_get(struct vpu_dev *vpu)
 65{
 66	if (atomic_inc_return(&vpu->ref_dec) == 1 && vpu->res->setup_decoder)
 67		vpu->res->setup_decoder(vpu);
 68}
 69
 70static void vpu_dec_put(struct vpu_dev *vpu)
 71{
 72	atomic_dec(&vpu->ref_dec);
 73}
 74
 75static int vpu_init_media_device(struct vpu_dev *vpu)
 76{
 77	vpu->mdev.dev = vpu->dev;
 78	strscpy(vpu->mdev.model, "amphion-vpu", sizeof(vpu->mdev.model));
 79	strscpy(vpu->mdev.bus_info, "platform: amphion-vpu", sizeof(vpu->mdev.bus_info));
 80	media_device_init(&vpu->mdev);
 81	vpu->v4l2_dev.mdev = &vpu->mdev;
 82
 83	return 0;
 84}
 85
 86static int vpu_probe(struct platform_device *pdev)
 87{
 88	struct device *dev = &pdev->dev;
 89	struct vpu_dev *vpu;
 90	int ret;
 91
 92	dev_dbg(dev, "probe\n");
 93	vpu = devm_kzalloc(dev, sizeof(*vpu), GFP_KERNEL);
 94	if (!vpu)
 95		return -ENOMEM;
 96
 97	vpu->pdev = pdev;
 98	vpu->dev = dev;
 99	mutex_init(&vpu->lock);
100	INIT_LIST_HEAD(&vpu->cores);
101	platform_set_drvdata(pdev, vpu);
102	atomic_set(&vpu->ref_vpu, 0);
103	atomic_set(&vpu->ref_enc, 0);
104	atomic_set(&vpu->ref_dec, 0);
105	vpu->get_vpu = vpu_dev_get;
106	vpu->put_vpu = vpu_dev_put;
107	vpu->get_enc = vpu_enc_get;
108	vpu->put_enc = vpu_enc_put;
109	vpu->get_dec = vpu_dec_get;
110	vpu->put_dec = vpu_dec_put;
111
112	vpu->base = devm_platform_ioremap_resource(pdev, 0);
113	if (IS_ERR(vpu->base))
114		return PTR_ERR(vpu->base);
115
116	vpu->res = of_device_get_match_data(dev);
117	if (!vpu->res)
118		return -ENODEV;
119
120	pm_runtime_enable(dev);
121
122	ret = v4l2_device_register(dev, &vpu->v4l2_dev);
123	if (ret)
124		goto err_vpu_deinit;
125
126	vpu_init_media_device(vpu);
127	vpu->encoder.type = VPU_CORE_TYPE_ENC;
128	vpu->encoder.function = MEDIA_ENT_F_PROC_VIDEO_ENCODER;
129	vpu->decoder.type = VPU_CORE_TYPE_DEC;
130	vpu->decoder.function = MEDIA_ENT_F_PROC_VIDEO_DECODER;
131	ret = vpu_add_func(vpu, &vpu->decoder);
132	if (ret)
133		goto err_add_decoder;
134	ret = vpu_add_func(vpu, &vpu->encoder);
135	if (ret)
136		goto err_add_encoder;
137	ret = media_device_register(&vpu->mdev);
138	if (ret)
139		goto err_vpu_media;
140	vpu->debugfs = debugfs_create_dir("amphion_vpu", NULL);
141
142	of_platform_populate(dev->of_node, NULL, NULL, dev);
143
144	return 0;
145
146err_vpu_media:
147	vpu_remove_func(&vpu->encoder);
148err_add_encoder:
149	vpu_remove_func(&vpu->decoder);
150err_add_decoder:
151	media_device_cleanup(&vpu->mdev);
152	v4l2_device_unregister(&vpu->v4l2_dev);
153err_vpu_deinit:
154	pm_runtime_disable(dev);
155	pm_runtime_set_suspended(dev);
156
157	return ret;
158}
159
160static void vpu_remove(struct platform_device *pdev)
161{
162	struct vpu_dev *vpu = platform_get_drvdata(pdev);
163	struct device *dev = &pdev->dev;
164
165	debugfs_remove_recursive(vpu->debugfs);
166	vpu->debugfs = NULL;
167
168	pm_runtime_disable(dev);
169
170	media_device_unregister(&vpu->mdev);
171	vpu_remove_func(&vpu->decoder);
172	vpu_remove_func(&vpu->encoder);
173	media_device_cleanup(&vpu->mdev);
174	v4l2_device_unregister(&vpu->v4l2_dev);
175	mutex_destroy(&vpu->lock);
176}
177
178static struct vpu_resources imx8qxp_res = {
179	.plat_type = IMX8QXP,
180	.mreg_base = 0x40000000,
181	.setup = vpu_imx8q_setup,
182	.setup_encoder = vpu_imx8q_setup_enc,
183	.setup_decoder = vpu_imx8q_setup_dec,
184	.reset = vpu_imx8q_reset
185};
186
187static struct vpu_resources imx8qm_res = {
188	.plat_type = IMX8QM,
189	.mreg_base = 0x40000000,
190	.setup = vpu_imx8q_setup,
191	.setup_encoder = vpu_imx8q_setup_enc,
192	.setup_decoder = vpu_imx8q_setup_dec,
193	.reset = vpu_imx8q_reset
194};
195
196static const struct of_device_id vpu_dt_match[] = {
197	{ .compatible = "nxp,imx8qxp-vpu", .data = &imx8qxp_res },
198	{ .compatible = "nxp,imx8qm-vpu", .data = &imx8qm_res },
199	{}
200};
201MODULE_DEVICE_TABLE(of, vpu_dt_match);
202
203static struct platform_driver amphion_vpu_driver = {
204	.probe = vpu_probe,
205	.remove = vpu_remove,
206	.driver = {
207		.name = "amphion-vpu",
208		.of_match_table = vpu_dt_match,
209	},
210};
211
212static int __init vpu_driver_init(void)
213{
214	int ret;
215
216	ret = platform_driver_register(&amphion_vpu_driver);
217	if (ret)
218		return ret;
219
220	ret = vpu_core_driver_init();
221	if (ret)
222		platform_driver_unregister(&amphion_vpu_driver);
223
224	return ret;
225}
226
227static void __exit vpu_driver_exit(void)
228{
229	vpu_core_driver_exit();
230	platform_driver_unregister(&amphion_vpu_driver);
231}
232module_init(vpu_driver_init);
233module_exit(vpu_driver_exit);
234
235MODULE_AUTHOR("Freescale Semiconductor, Inc.");
236MODULE_DESCRIPTION("Linux VPU driver for Freescale i.MX8Q");
237MODULE_LICENSE("GPL v2");