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 29 bool debug; 30 module_param(debug, bool, 0644); 31 32 void vpu_writel(struct vpu_dev *vpu, u32 reg, u32 val) 33 { 34 writel(val, vpu->base + reg); 35 } 36 37 u32 vpu_readl(struct vpu_dev *vpu, u32 reg) 38 { 39 return readl(vpu->base + reg); 40 } 41 42 static 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 48 static void vpu_dev_put(struct vpu_dev *vpu) 49 { 50 atomic_dec(&vpu->ref_vpu); 51 } 52 53 static 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 59 static void vpu_enc_put(struct vpu_dev *vpu) 60 { 61 atomic_dec(&vpu->ref_enc); 62 } 63 64 static 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 70 static void vpu_dec_put(struct vpu_dev *vpu) 71 { 72 atomic_dec(&vpu->ref_dec); 73 } 74 75 static 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 86 static 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 146 err_vpu_media: 147 vpu_remove_func(&vpu->encoder); 148 err_add_encoder: 149 vpu_remove_func(&vpu->decoder); 150 err_add_decoder: 151 media_device_cleanup(&vpu->mdev); 152 v4l2_device_unregister(&vpu->v4l2_dev); 153 err_vpu_deinit: 154 pm_runtime_disable(dev); 155 pm_runtime_set_suspended(dev); 156 157 return ret; 158 } 159 160 static 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 178 static 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 187 static 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 196 static 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 }; 201 MODULE_DEVICE_TABLE(of, vpu_dt_match); 202 203 static 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 212 static int __init vpu_driver_init(void) 213 { 214 int ret; 215 216 ret = platform_driver_register(&hion_vpu_driver); 217 if (ret) 218 return ret; 219 220 ret = vpu_core_driver_init(); 221 if (ret) 222 platform_driver_unregister(&hion_vpu_driver); 223 224 return ret; 225 } 226 227 static void __exit vpu_driver_exit(void) 228 { 229 vpu_core_driver_exit(); 230 platform_driver_unregister(&hion_vpu_driver); 231 } 232 module_init(vpu_driver_init); 233 module_exit(vpu_driver_exit); 234 235 MODULE_AUTHOR("Freescale Semiconductor, Inc."); 236 MODULE_DESCRIPTION("Linux VPU driver for Freescale i.MX8Q"); 237 MODULE_LICENSE("GPL v2"); 238