drivers/media/platform/amlogic/meson-ge2d/ge2d.c

   1 // SPDX-License-Identifier: GPL-2.0-or-later
   2 /*
   3  * Copyright (C) 2020 BayLibre, SAS
   4  * Author: Neil Armstrong <narmstrong@baylibre.com>
   5  */
   6
   7 #include <linux/clk.h>
   8 #include <linux/delay.h>
   9 #include <linux/bitfield.h>
  10 #include <linux/interrupt.h>
  11 #include <linux/module.h>
  12 #include <linux/of.h>
  13 #include <linux/reset.h>
  14 #include <linux/sched.h>
  15 #include <linux/slab.h>
  16 #include <linux/timer.h>
  17 #include <linux/regmap.h>
  18
  19 #include <linux/platform_device.h>
  20 #include <media/v4l2-device.h>
  21 #include <media/v4l2-event.h>
  22 #include <media/v4l2-ioctl.h>
  23 #include <media/v4l2-mem2mem.h>
  24 #include <media/v4l2-ctrls.h>
  25 #include <media/videobuf2-v4l2.h>
  26 #include <media/videobuf2-dma-contig.h>
  27
  28 #include "ge2d-regs.h"
  29
  30 #define GE2D_NAME       "meson-ge2d"
  31
  32 #define DEFAULT_WIDTH   128
  33 #define DEFAULT_HEIGHT  128
  34 #define DEFAULT_STRIDE  512
  35
  36 #define MAX_WIDTH       8191
  37 #define MAX_HEIGHT      8191
  38
  39 /*
  40  * Missing features:
  41  * - Scaling
  42  * - Simple 1/2 vertical scaling
  43  * - YUV input support
  44  * - Source global alpha
  45  * - Colorspace conversion
  46  */
  47
  48 struct ge2d_fmt {
  49         u32 fourcc;
  50         bool alpha;
  51         bool le;
  52         unsigned int depth;
  53         unsigned int hw_fmt;
  54         unsigned int hw_map;
  55 };
  56
  57 struct ge2d_frame {
  58         struct vb2_v4l2_buffer *buf;
  59
  60         /* Image Format */
  61         struct v4l2_pix_format pix_fmt;
  62
  63         /* Crop */
  64         struct v4l2_rect crop;
  65
  66         /* Image format */
  67         const struct ge2d_fmt *fmt;
  68 };
  69
  70 struct ge2d_ctx {
  71         struct v4l2_fh fh;
  72         struct meson_ge2d *ge2d;
  73         struct ge2d_frame in;
  74         struct ge2d_frame out;
  75         struct v4l2_ctrl_handler ctrl_handler;
  76
  77         unsigned long sequence_out, sequence_cap;
  78
  79         /* Control values */
  80         u32 hflip;
  81         u32 vflip;
  82         u32 xy_swap;
  83 };
  84
  85 struct meson_ge2d {
  86         struct v4l2_device v4l2_dev;
  87         struct v4l2_m2m_dev *m2m_dev;
  88         struct video_device *vfd;
  89
  90         struct device *dev;
  91         struct regmap *map;
  92         struct clk *clk;
  93
  94         /* vb2 queue lock */
  95         struct mutex mutex;
  96
  97         struct ge2d_ctx *curr;
  98 };
  99
 100 #define FMT(_fourcc, _alpha, _depth, _map)              \
 101 {                                                       \
 102         .fourcc = _fourcc,                              \
 103         .alpha = (_alpha),                              \
 104         .depth = (_depth),                              \
 105         .hw_fmt = GE2D_FORMAT_ ## _depth ## BIT,        \
 106         .hw_map = GE2D_COLOR_MAP_ ## _map,              \
 107 }
 108
 109 /* TOFIX Handle the YUV input formats */
 110 static const struct ge2d_fmt formats[] = {
 111         /*  FOURCC Alpha  HW FMT  HW MAP */
 112         FMT(V4L2_PIX_FMT_XRGB32, false, 32, BGRA8888),
 113         FMT(V4L2_PIX_FMT_RGB32, true, 32, BGRA8888),
 114         FMT(V4L2_PIX_FMT_ARGB32, true, 32, BGRA8888),
 115         FMT(V4L2_PIX_FMT_RGBX32, false, 32, ABGR8888),
 116         FMT(V4L2_PIX_FMT_RGBA32, true, 32, ABGR8888),
 117         FMT(V4L2_PIX_FMT_BGRX32, false, 32, RGBA8888),
 118         FMT(V4L2_PIX_FMT_BGRA32, true, 32, RGBA8888),
 119         FMT(V4L2_PIX_FMT_BGR32, true, 32, ARGB8888),
 120         FMT(V4L2_PIX_FMT_ABGR32, true, 32, ARGB8888),
 121         FMT(V4L2_PIX_FMT_XBGR32, false, 32, ARGB8888),
 122
 123         FMT(V4L2_PIX_FMT_RGB24, false, 24, BGR888),
 124         FMT(V4L2_PIX_FMT_BGR24, false, 24, RGB888),
 125
 126         FMT(V4L2_PIX_FMT_XRGB555X, false, 16, ARGB1555),
 127         FMT(V4L2_PIX_FMT_ARGB555X, true, 16, ARGB1555),
 128         FMT(V4L2_PIX_FMT_RGB565, false, 16, RGB565),
 129         FMT(V4L2_PIX_FMT_RGBX444, false, 16, RGBA4444),
 130         FMT(V4L2_PIX_FMT_RGBA444, true, 16, RGBA4444),
 131         FMT(V4L2_PIX_FMT_XRGB444, false, 16, ARGB4444),
 132         FMT(V4L2_PIX_FMT_ARGB444, true, 16, ARGB4444),
 133 };
 134
 135 #define NUM_FORMATS ARRAY_SIZE(formats)
 136
 137 static const struct ge2d_fmt *find_fmt(struct v4l2_format *f)
 138 {
 139         unsigned int i;
 140
 141         for (i = 0; i < NUM_FORMATS; i++) {
 142                 if (formats[i].fourcc == f->fmt.pix.pixelformat)
 143                         return &formats[i];
 144         }
 145
 146         /*
 147          * TRY_FMT/S_FMT should never return an error when the requested format
 148          * is not supported. Drivers should always return a valid format,
 149          * preferably a format that is as widely supported by applications as
 150          * possible.
 151          */
 152         return &formats[0];
 153 }
 154
 155 static struct ge2d_frame *get_frame(struct ge2d_ctx *ctx,
 156                                     enum v4l2_buf_type type)
 157 {
 158         switch (type) {
 159         case V4L2_BUF_TYPE_VIDEO_OUTPUT:
 160                 return &ctx->in;
 161         case V4L2_BUF_TYPE_VIDEO_CAPTURE:
 162                 return &ctx->out;
 163         default:
 164                 /* This should never happen, warn and return OUTPUT frame */
 165                 dev_warn(ctx->ge2d->dev, "%s: invalid buffer type\n", __func__);
 166                 return &ctx->in;
 167         }
 168 }
 169
 170 static void ge2d_hw_start(struct meson_ge2d *ge2d)
 171 {
 172         struct ge2d_ctx *ctx = ge2d->curr;
 173         u32 reg;
 174
 175         /* Reset */
 176         regmap_update_bits(ge2d->map, GE2D_GEN_CTRL1,
 177                            GE2D_SOFT_RST, GE2D_SOFT_RST);
 178         regmap_update_bits(ge2d->map, GE2D_GEN_CTRL1,
 179                            GE2D_SOFT_RST, 0);
 180
 181         usleep_range(100, 200);
 182
 183         /* Implement CANVAS for non-AXG */
 184         regmap_write(ge2d->map, GE2D_SRC1_BADDR_CTRL,
 185                      (vb2_dma_contig_plane_dma_addr(&ctx->in.buf->vb2_buf, 0) + 7) >> 3);
 186         regmap_write(ge2d->map, GE2D_SRC1_STRIDE_CTRL,
 187                      (ctx->in.pix_fmt.bytesperline + 7) >> 3);
 188         regmap_write(ge2d->map, GE2D_SRC2_BADDR_CTRL,
 189                      (vb2_dma_contig_plane_dma_addr(&ctx->out.buf->vb2_buf, 0) + 7) >> 3);
 190         regmap_write(ge2d->map, GE2D_SRC2_STRIDE_CTRL,
 191                      (ctx->out.pix_fmt.bytesperline + 7) >> 3);
 192         regmap_write(ge2d->map, GE2D_DST1_BADDR_CTRL,
 193                      (vb2_dma_contig_plane_dma_addr(&ctx->out.buf->vb2_buf, 0) + 7) >> 3);
 194         regmap_write(ge2d->map, GE2D_DST1_STRIDE_CTRL,
 195                      (ctx->out.pix_fmt.bytesperline + 7) >> 3);
 196
 197         regmap_write(ge2d->map, GE2D_GEN_CTRL0, 0);
 198         regmap_write(ge2d->map, GE2D_GEN_CTRL1,
 199                      FIELD_PREP(GE2D_INTERRUPT_CTRL, 2) |
 200                      FIELD_PREP(GE2D_SRC2_BURST_SIZE_CTRL, 3) |
 201                      FIELD_PREP(GE2D_SRC1_BURST_SIZE_CTRL, 0x3f));
 202
 203         regmap_write(ge2d->map, GE2D_GEN_CTRL2,
 204                      GE2D_SRC1_LITTLE_ENDIAN |
 205                      GE2D_SRC2_LITTLE_ENDIAN |
 206                      GE2D_DST_LITTLE_ENDIAN |
 207                      FIELD_PREP(GE2D_DST1_COLOR_MAP, ctx->out.fmt->hw_map) |
 208                      FIELD_PREP(GE2D_DST1_FORMAT, ctx->out.fmt->hw_fmt) |
 209                      FIELD_PREP(GE2D_SRC2_COLOR_MAP, ctx->out.fmt->hw_map) |
 210                      FIELD_PREP(GE2D_SRC2_FORMAT, ctx->out.fmt->hw_fmt) |
 211                      FIELD_PREP(GE2D_SRC1_COLOR_MAP, ctx->in.fmt->hw_map) |
 212                      FIELD_PREP(GE2D_SRC1_FORMAT, ctx->in.fmt->hw_fmt));
 213         regmap_write(ge2d->map, GE2D_GEN_CTRL3,
 214                      GE2D_DST1_ENABLE);
 215
 216         regmap_write(ge2d->map, GE2D_SRC1_CLIPY_START_END,
 217                      FIELD_PREP(GE2D_START, ctx->in.crop.top) |
 218                      FIELD_PREP(GE2D_END, ctx->in.crop.top + ctx->in.crop.height - 1));
 219         regmap_write(ge2d->map, GE2D_SRC1_CLIPX_START_END,
 220                      FIELD_PREP(GE2D_START, ctx->in.crop.left) |
 221                      FIELD_PREP(GE2D_END, ctx->in.crop.left + ctx->in.crop.width - 1));
 222         regmap_write(ge2d->map, GE2D_SRC2_CLIPY_START_END,
 223                      FIELD_PREP(GE2D_START, ctx->out.crop.top) |
 224                      FIELD_PREP(GE2D_END, ctx->out.crop.top + ctx->out.crop.height - 1));
 225         regmap_write(ge2d->map, GE2D_SRC2_CLIPX_START_END,
 226                      FIELD_PREP(GE2D_START, ctx->out.crop.left) |
 227                      FIELD_PREP(GE2D_END, ctx->out.crop.left + ctx->out.crop.width - 1));
 228         regmap_write(ge2d->map, GE2D_DST_CLIPY_START_END,
 229                      FIELD_PREP(GE2D_START, ctx->out.crop.top) |
 230                      FIELD_PREP(GE2D_END, ctx->out.crop.top + ctx->out.crop.height - 1));
 231         regmap_write(ge2d->map, GE2D_DST_CLIPX_START_END,
 232                      FIELD_PREP(GE2D_START, ctx->out.crop.left) |
 233                      FIELD_PREP(GE2D_END, ctx->out.crop.left + ctx->out.crop.width - 1));
 234
 235         regmap_write(ge2d->map, GE2D_SRC1_Y_START_END,
 236                      FIELD_PREP(GE2D_END, ctx->in.pix_fmt.height - 1));
 237         regmap_write(ge2d->map, GE2D_SRC1_X_START_END,
 238                      FIELD_PREP(GE2D_END, ctx->in.pix_fmt.width - 1));
 239         regmap_write(ge2d->map, GE2D_SRC2_Y_START_END,
 240                      FIELD_PREP(GE2D_END, ctx->out.pix_fmt.height - 1));
 241         regmap_write(ge2d->map, GE2D_SRC2_X_START_END,
 242                      FIELD_PREP(GE2D_END, ctx->out.pix_fmt.width - 1));
 243         regmap_write(ge2d->map, GE2D_DST_Y_START_END,
 244                      FIELD_PREP(GE2D_END, ctx->out.pix_fmt.height - 1));
 245         regmap_write(ge2d->map, GE2D_DST_X_START_END,
 246                      FIELD_PREP(GE2D_END, ctx->out.pix_fmt.width - 1));
 247
 248         /* Color, no blend, use source color */
 249         reg = GE2D_ALU_DO_COLOR_OPERATION_LOGIC(LOGIC_OPERATION_COPY,
 250                                                 COLOR_FACTOR_SRC_COLOR);
 251
 252         if (ctx->in.fmt->alpha && ctx->out.fmt->alpha)
 253                 /* Take source alpha */
 254                 reg |= GE2D_ALU_DO_ALPHA_OPERATION_LOGIC(LOGIC_OPERATION_COPY,
 255                                                          COLOR_FACTOR_SRC_ALPHA);
 256         else if (!ctx->out.fmt->alpha)
 257                 /* Set alpha to 0 */
 258                 reg |= GE2D_ALU_DO_ALPHA_OPERATION_LOGIC(LOGIC_OPERATION_SET,
 259                                                          COLOR_FACTOR_ZERO);
 260         else
 261                 /* Keep original alpha */
 262                 reg |= GE2D_ALU_DO_ALPHA_OPERATION_LOGIC(LOGIC_OPERATION_COPY,
 263                                                          COLOR_FACTOR_DST_ALPHA);
 264
 265         regmap_write(ge2d->map, GE2D_ALU_OP_CTRL, reg);
 266
 267         /* Start */
 268         regmap_write(ge2d->map, GE2D_CMD_CTRL,
 269                      (ctx->xy_swap ? GE2D_DST_XY_SWAP : 0) |
 270                      (ctx->hflip ? GE2D_SRC1_Y_REV : 0) |
 271                      (ctx->vflip ? GE2D_SRC1_X_REV : 0) |
 272                      GE2D_CBUS_CMD_WR);
 273 }
 274
 275 static void device_run(void *priv)
 276 {
 277         struct ge2d_ctx *ctx = priv;
 278         struct meson_ge2d *ge2d = ctx->ge2d;
 279
 280         ge2d->curr = ctx;
 281
 282         ctx->in.buf = v4l2_m2m_next_src_buf(ctx->fh.m2m_ctx);
 283         ctx->out.buf = v4l2_m2m_next_dst_buf(ctx->fh.m2m_ctx);
 284
 285         ge2d_hw_start(ge2d);
 286 }
 287
 288 static irqreturn_t ge2d_isr(int irq, void *priv)
 289 {
 290         struct meson_ge2d *ge2d = priv;
 291         u32 intr;
 292
 293         regmap_read(ge2d->map, GE2D_STATUS0, &intr);
 294
 295         if (!(intr & GE2D_GE2D_BUSY)) {
 296                 struct vb2_v4l2_buffer *src, *dst;
 297                 struct ge2d_ctx *ctx = ge2d->curr;
 298
 299                 ge2d->curr = NULL;
 300
 301                 src = v4l2_m2m_src_buf_remove(ctx->fh.m2m_ctx);
 302                 dst = v4l2_m2m_dst_buf_remove(ctx->fh.m2m_ctx);
 303
 304                 src->sequence = ctx->sequence_out++;
 305                 dst->sequence = ctx->sequence_cap++;
 306
 307                 dst->timecode = src->timecode;
 308                 dst->vb2_buf.timestamp = src->vb2_buf.timestamp;
 309                 dst->flags = src->flags;
 310
 311                 v4l2_m2m_buf_done(src, VB2_BUF_STATE_DONE);
 312                 v4l2_m2m_buf_done(dst, VB2_BUF_STATE_DONE);
 313                 v4l2_m2m_job_finish(ge2d->m2m_dev, ctx->fh.m2m_ctx);
 314         }
 315
 316         return IRQ_HANDLED;
 317 }
 318
 319 static const struct v4l2_m2m_ops ge2d_m2m_ops = {
 320         .device_run = device_run,
 321 };
 322
 323 static int ge2d_queue_setup(struct vb2_queue *vq,
 324                             unsigned int *nbuffers, unsigned int *nplanes,
 325                             unsigned int sizes[], struct device *alloc_devs[])
 326 {
 327         struct ge2d_ctx *ctx = vb2_get_drv_priv(vq);
 328         struct ge2d_frame *f = get_frame(ctx, vq->type);
 329
 330         if (*nplanes)
 331                 return sizes[0] < f->pix_fmt.sizeimage ? -EINVAL : 0;
 332
 333         sizes[0] = f->pix_fmt.sizeimage;
 334         *nplanes = 1;
 335
 336         return 0;
 337 }
 338
 339 static int ge2d_buf_prepare(struct vb2_buffer *vb)
 340 {
 341         struct ge2d_ctx *ctx = vb2_get_drv_priv(vb->vb2_queue);
 342         struct vb2_v4l2_buffer *vbuf = to_vb2_v4l2_buffer(vb);
 343         struct ge2d_frame *f = get_frame(ctx, vb->vb2_queue->type);
 344
 345         vbuf->field = V4L2_FIELD_NONE;
 346
 347         vb2_set_plane_payload(vb, 0, f->pix_fmt.sizeimage);
 348
 349         return 0;
 350 }
 351
 352 static void ge2d_buf_queue(struct vb2_buffer *vb)
 353 {
 354         struct vb2_v4l2_buffer *vbuf = to_vb2_v4l2_buffer(vb);
 355         struct ge2d_ctx *ctx = vb2_get_drv_priv(vb->vb2_queue);
 356
 357         v4l2_m2m_buf_queue(ctx->fh.m2m_ctx, vbuf);
 358 }
 359
 360 static int ge2d_start_streaming(struct vb2_queue *vq, unsigned int count)
 361 {
 362         struct ge2d_ctx *ctx = vb2_get_drv_priv(vq);
 363
 364         if (V4L2_TYPE_IS_OUTPUT(vq->type))
 365                 ctx->sequence_out = 0;
 366         else
 367                 ctx->sequence_cap = 0;
 368
 369         return 0;
 370 }
 371
 372 static void ge2d_stop_streaming(struct vb2_queue *vq)
 373 {
 374         struct ge2d_ctx *ctx = vb2_get_drv_priv(vq);
 375         struct vb2_v4l2_buffer *vbuf;
 376
 377         for (;;) {
 378                 if (V4L2_TYPE_IS_OUTPUT(vq->type))
 379                         vbuf = v4l2_m2m_src_buf_remove(ctx->fh.m2m_ctx);
 380                 else
 381                         vbuf = v4l2_m2m_dst_buf_remove(ctx->fh.m2m_ctx);
 382                 if (!vbuf)
 383                         break;
 384                 v4l2_m2m_buf_done(vbuf, VB2_BUF_STATE_ERROR);
 385         }
 386 }
 387
 388 static const struct vb2_ops ge2d_qops = {
 389         .queue_setup = ge2d_queue_setup,
 390         .buf_prepare = ge2d_buf_prepare,
 391         .buf_queue = ge2d_buf_queue,
 392         .start_streaming = ge2d_start_streaming,
 393         .stop_streaming = ge2d_stop_streaming,
 394         .wait_prepare = vb2_ops_wait_prepare,
 395         .wait_finish = vb2_ops_wait_finish,
 396 };
 397
 398 static int
 399 queue_init(void *priv, struct vb2_queue *src_vq, struct vb2_queue *dst_vq)
 400 {
 401         struct ge2d_ctx *ctx = priv;
 402         int ret;
 403
 404         src_vq->type = V4L2_BUF_TYPE_VIDEO_OUTPUT;
 405         src_vq->io_modes = VB2_MMAP | VB2_DMABUF;
 406         src_vq->drv_priv = ctx;
 407         src_vq->ops = &ge2d_qops;
 408         src_vq->mem_ops = &vb2_dma_contig_memops;
 409         src_vq->buf_struct_size = sizeof(struct v4l2_m2m_buffer);
 410         src_vq->timestamp_flags = V4L2_BUF_FLAG_TIMESTAMP_COPY;
 411         src_vq->lock = &ctx->ge2d->mutex;
 412         src_vq->dev = ctx->ge2d->v4l2_dev.dev;
 413
 414         ret = vb2_queue_init(src_vq);
 415         if (ret)
 416                 return ret;
 417
 418         dst_vq->type = V4L2_BUF_TYPE_VIDEO_CAPTURE;
 419         dst_vq->io_modes = VB2_MMAP | VB2_DMABUF;
 420         dst_vq->drv_priv = ctx;
 421         dst_vq->ops = &ge2d_qops;
 422         dst_vq->mem_ops = &vb2_dma_contig_memops;
 423         dst_vq->buf_struct_size = sizeof(struct v4l2_m2m_buffer);
 424         dst_vq->timestamp_flags = V4L2_BUF_FLAG_TIMESTAMP_COPY;
 425         dst_vq->lock = &ctx->ge2d->mutex;
 426         dst_vq->dev = ctx->ge2d->v4l2_dev.dev;
 427
 428         return vb2_queue_init(dst_vq);
 429 }
 430
 431 static int
 432 vidioc_querycap(struct file *file, void *priv, struct v4l2_capability *cap)
 433 {
 434         strscpy(cap->driver, GE2D_NAME, sizeof(cap->driver));
 435         strscpy(cap->card, GE2D_NAME, sizeof(cap->card));
 436         strscpy(cap->bus_info, "platform:" GE2D_NAME, sizeof(cap->bus_info));
 437
 438         return 0;
 439 }
 440
 441 static int vidioc_enum_fmt(struct file *file, void *priv, struct v4l2_fmtdesc *f)
 442 {
 443         const struct ge2d_fmt *fmt;
 444
 445         if (f->index >= NUM_FORMATS)
 446                 return -EINVAL;
 447
 448         fmt = &formats[f->index];
 449         f->pixelformat = fmt->fourcc;
 450
 451         return 0;
 452 }
 453
 454 static int vidioc_g_selection(struct file *file, void *priv,
 455                               struct v4l2_selection *s)
 456 {
 457         struct ge2d_ctx *ctx = priv;
 458         struct ge2d_frame *f;
 459         bool use_frame = false;
 460
 461         if (s->type != V4L2_BUF_TYPE_VIDEO_OUTPUT &&
 462             s->type != V4L2_BUF_TYPE_VIDEO_CAPTURE)
 463                 return -EINVAL;
 464
 465         f = get_frame(ctx, s->type);
 466
 467         switch (s->target) {
 468         case V4L2_SEL_TGT_COMPOSE_DEFAULT:
 469         case V4L2_SEL_TGT_COMPOSE_BOUNDS:
 470                 if (s->type != V4L2_BUF_TYPE_VIDEO_CAPTURE)
 471                         return -EINVAL;
 472                 break;
 473         case V4L2_SEL_TGT_CROP_DEFAULT:
 474         case V4L2_SEL_TGT_CROP_BOUNDS:
 475                 if (s->type != V4L2_BUF_TYPE_VIDEO_OUTPUT)
 476                         return -EINVAL;
 477                 break;
 478         case V4L2_SEL_TGT_COMPOSE:
 479                 if (s->type != V4L2_BUF_TYPE_VIDEO_CAPTURE)
 480                         return -EINVAL;
 481                 use_frame = true;
 482                 break;
 483         case V4L2_SEL_TGT_CROP:
 484                 if (s->type != V4L2_BUF_TYPE_VIDEO_OUTPUT)
 485                         return -EINVAL;
 486                 use_frame = true;
 487                 break;
 488         default:
 489                 return -EINVAL;
 490         }
 491
 492         if (use_frame) {
 493                 s->r = f->crop;
 494         } else {
 495                 s->r.left = 0;
 496                 s->r.top = 0;
 497                 s->r.width = f->pix_fmt.width;
 498                 s->r.height = f->pix_fmt.height;
 499         }
 500
 501         return 0;
 502 }
 503
 504 static int vidioc_s_selection(struct file *file, void *priv,
 505                               struct v4l2_selection *s)
 506 {
 507         struct ge2d_ctx *ctx = priv;
 508         struct meson_ge2d *ge2d = ctx->ge2d;
 509         struct ge2d_frame *f;
 510         int ret = 0;
 511
 512         if (s->type != V4L2_BUF_TYPE_VIDEO_OUTPUT &&
 513             s->type != V4L2_BUF_TYPE_VIDEO_CAPTURE)
 514                 return -EINVAL;
 515
 516         f = get_frame(ctx, s->type);
 517
 518         switch (s->target) {
 519         case V4L2_SEL_TGT_COMPOSE:
 520                 /*
 521                  * COMPOSE target is only valid for capture buffer type, return
 522                  * error for output buffer type
 523                  */
 524                 if (s->type != V4L2_BUF_TYPE_VIDEO_CAPTURE)
 525                         return -EINVAL;
 526                 break;
 527         case V4L2_SEL_TGT_CROP:
 528                 /*
 529                  * CROP target is only valid for output buffer type, return
 530                  * error for capture buffer type
 531                  */
 532                 if (s->type != V4L2_BUF_TYPE_VIDEO_OUTPUT)
 533                         return -EINVAL;
 534                 break;
 535         /*
 536          * bound and default crop/compose targets are invalid targets to
 537          * try/set
 538          */
 539         default:
 540                 return -EINVAL;
 541         }
 542
 543         if (s->r.top < 0 || s->r.left < 0) {
 544                 v4l2_err(&ge2d->v4l2_dev,
 545                          "doesn't support negative values for top & left.\n");
 546                 return -EINVAL;
 547         }
 548
 549         if (s->r.left + s->r.width > f->pix_fmt.width ||
 550             s->r.top + s->r.height > f->pix_fmt.height) {
 551                 v4l2_err(&ge2d->v4l2_dev, "unsupported rectangle value.\n");
 552                 return -EINVAL;
 553         }
 554
 555         f->crop = s->r;
 556
 557         return ret;
 558 }
 559
 560 static void vidioc_setup_cap_fmt(struct ge2d_ctx *ctx, struct v4l2_pix_format *f)
 561 {
 562         struct ge2d_frame *frm_out = get_frame(ctx, V4L2_BUF_TYPE_VIDEO_OUTPUT);
 563
 564         *f = frm_out->pix_fmt;
 565
 566         if (ctx->xy_swap) {
 567                 f->width = frm_out->pix_fmt.height;
 568                 f->height = frm_out->pix_fmt.width;
 569         }
 570 }
 571
 572 static int vidioc_try_fmt_cap(struct file *file, void *priv, struct v4l2_format *f)
 573 {
 574         const struct ge2d_fmt *fmt = find_fmt(f);
 575         struct ge2d_ctx *ctx = priv;
 576         struct v4l2_pix_format fmt_cap;
 577
 578         vidioc_setup_cap_fmt(ctx, &fmt_cap);
 579
 580         fmt_cap.pixelformat = fmt->fourcc;
 581
 582         fmt_cap.bytesperline = max(f->fmt.pix.bytesperline,
 583                                    ALIGN((fmt_cap.width * fmt->depth) >> 3, 8));
 584
 585         fmt_cap.sizeimage = max(f->fmt.pix.sizeimage,
 586                                 fmt_cap.height * fmt_cap.bytesperline);
 587
 588         f->fmt.pix = fmt_cap;
 589
 590         return 0;
 591 }
 592
 593 static int vidioc_s_fmt_cap(struct file *file, void *priv, struct v4l2_format *f)
 594 {
 595         struct ge2d_ctx *ctx = priv;
 596         struct meson_ge2d *ge2d = ctx->ge2d;
 597         struct vb2_queue *vq;
 598         struct ge2d_frame *frm;
 599         int ret = 0;
 600
 601         /* Adjust all values accordingly to the hardware capabilities
 602          * and chosen format.
 603          */
 604         ret = vidioc_try_fmt_cap(file, priv, f);
 605         if (ret)
 606                 return ret;
 607
 608         vq = v4l2_m2m_get_vq(ctx->fh.m2m_ctx, f->type);
 609         if (vb2_is_busy(vq)) {
 610                 v4l2_err(&ge2d->v4l2_dev, "queue (%d) bust\n", f->type);
 611                 return -EBUSY;
 612         }
 613
 614         frm = get_frame(ctx, V4L2_BUF_TYPE_VIDEO_CAPTURE);
 615
 616         frm->pix_fmt = f->fmt.pix;
 617         frm->fmt = find_fmt(f);
 618         f->fmt.pix.pixelformat = frm->fmt->fourcc;
 619
 620         /* Reset crop settings */
 621         frm->crop.left = 0;
 622         frm->crop.top = 0;
 623         frm->crop.width = frm->pix_fmt.width;
 624         frm->crop.height = frm->pix_fmt.height;
 625
 626         return 0;
 627 }
 628
 629 static int vidioc_g_fmt(struct file *file, void *priv, struct v4l2_format *f)
 630 {
 631         struct ge2d_ctx *ctx = priv;
 632         struct vb2_queue *vq;
 633         struct ge2d_frame *frm;
 634
 635         vq = v4l2_m2m_get_vq(ctx->fh.m2m_ctx, f->type);
 636         if (!vq)
 637                 return -EINVAL;
 638
 639         frm = get_frame(ctx, f->type);
 640
 641         f->fmt.pix = frm->pix_fmt;
 642         f->fmt.pix.pixelformat = frm->fmt->fourcc;
 643
 644         return 0;
 645 }
 646
 647 static int vidioc_try_fmt_out(struct file *file, void *priv, struct v4l2_format *f)
 648 {
 649         const struct ge2d_fmt *fmt = find_fmt(f);
 650
 651         f->fmt.pix.field = V4L2_FIELD_NONE;
 652         f->fmt.pix.pixelformat = fmt->fourcc;
 653
 654         if (f->fmt.pix.width > MAX_WIDTH)
 655                 f->fmt.pix.width = MAX_WIDTH;
 656         if (f->fmt.pix.height > MAX_HEIGHT)
 657                 f->fmt.pix.height = MAX_HEIGHT;
 658
 659         f->fmt.pix.bytesperline = max(f->fmt.pix.bytesperline,
 660                                       ALIGN((f->fmt.pix.width * fmt->depth) >> 3, 8));
 661
 662         f->fmt.pix.sizeimage = max(f->fmt.pix.sizeimage,
 663                                    f->fmt.pix.height * f->fmt.pix.bytesperline);
 664
 665         return 0;
 666 }
 667
 668 static int vidioc_s_fmt_out(struct file *file, void *priv, struct v4l2_format *f)
 669 {
 670         struct ge2d_ctx *ctx = priv;
 671         struct meson_ge2d *ge2d = ctx->ge2d;
 672         struct vb2_queue *vq;
 673         struct ge2d_frame *frm, *frm_cap;
 674         int ret = 0;
 675
 676         /* Adjust all values accordingly to the hardware capabilities
 677          * and chosen format.
 678          */
 679         ret = vidioc_try_fmt_out(file, priv, f);
 680         if (ret)
 681                 return ret;
 682
 683         vq = v4l2_m2m_get_vq(ctx->fh.m2m_ctx, f->type);
 684         if (vb2_is_busy(vq)) {
 685                 v4l2_err(&ge2d->v4l2_dev, "queue (%d) bust\n", f->type);
 686                 return -EBUSY;
 687         }
 688
 689         frm = get_frame(ctx, V4L2_BUF_TYPE_VIDEO_OUTPUT);
 690         frm_cap = get_frame(ctx, V4L2_BUF_TYPE_VIDEO_CAPTURE);
 691
 692         frm->pix_fmt = f->fmt.pix;
 693         frm->fmt = find_fmt(f);
 694         f->fmt.pix.pixelformat = frm->fmt->fourcc;
 695
 696         /* Reset crop settings */
 697         frm->crop.left = 0;
 698         frm->crop.top = 0;
 699         frm->crop.width = frm->pix_fmt.width;
 700         frm->crop.height = frm->pix_fmt.height;
 701
 702         /* Propagate settings to capture */
 703         vidioc_setup_cap_fmt(ctx, &frm_cap->pix_fmt);
 704
 705         return 0;
 706 }
 707
 708 static const struct v4l2_ioctl_ops ge2d_ioctl_ops = {
 709         .vidioc_querycap = vidioc_querycap,
 710
 711         .vidioc_enum_fmt_vid_cap = vidioc_enum_fmt,
 712         .vidioc_g_fmt_vid_cap = vidioc_g_fmt,
 713         .vidioc_try_fmt_vid_cap = vidioc_try_fmt_cap,
 714         .vidioc_s_fmt_vid_cap = vidioc_s_fmt_cap,
 715
 716         .vidioc_enum_fmt_vid_out = vidioc_enum_fmt,
 717         .vidioc_g_fmt_vid_out = vidioc_g_fmt,
 718         .vidioc_try_fmt_vid_out = vidioc_try_fmt_out,
 719         .vidioc_s_fmt_vid_out = vidioc_s_fmt_out,
 720
 721         .vidioc_reqbufs = v4l2_m2m_ioctl_reqbufs,
 722         .vidioc_querybuf = v4l2_m2m_ioctl_querybuf,
 723         .vidioc_qbuf = v4l2_m2m_ioctl_qbuf,
 724         .vidioc_dqbuf = v4l2_m2m_ioctl_dqbuf,
 725         .vidioc_prepare_buf = v4l2_m2m_ioctl_prepare_buf,
 726         .vidioc_create_bufs = v4l2_m2m_ioctl_create_bufs,
 727         .vidioc_expbuf = v4l2_m2m_ioctl_expbuf,
 728
 729         .vidioc_subscribe_event = v4l2_ctrl_subscribe_event,
 730         .vidioc_unsubscribe_event = v4l2_event_unsubscribe,
 731
 732         .vidioc_streamon = v4l2_m2m_ioctl_streamon,
 733         .vidioc_streamoff = v4l2_m2m_ioctl_streamoff,
 734
 735         .vidioc_g_selection = vidioc_g_selection,
 736         .vidioc_s_selection = vidioc_s_selection,
 737 };
 738
 739 static int ge2d_s_ctrl(struct v4l2_ctrl *ctrl)
 740 {
 741         struct ge2d_ctx *ctx = container_of(ctrl->handler, struct ge2d_ctx,
 742                                            ctrl_handler);
 743         struct v4l2_pix_format fmt;
 744         struct vb2_queue *vq;
 745
 746         switch (ctrl->id) {
 747         case V4L2_CID_HFLIP:
 748                 ctx->hflip = ctrl->val;
 749                 break;
 750         case V4L2_CID_VFLIP:
 751                 ctx->vflip = ctrl->val;
 752                 break;
 753         case V4L2_CID_ROTATE:
 754                 vq = v4l2_m2m_get_vq(ctx->fh.m2m_ctx, V4L2_BUF_TYPE_VIDEO_CAPTURE);
 755                 if (vb2_is_busy(vq))
 756                         return -EBUSY;
 757
 758                 if (ctrl->val == 90) {
 759                         ctx->hflip = 0;
 760                         ctx->vflip = 1;
 761                         ctx->xy_swap = 1;
 762                 } else if (ctrl->val == 180) {
 763                         ctx->hflip = 1;
 764                         ctx->vflip = 1;
 765                         ctx->xy_swap = 0;
 766                 } else if (ctrl->val == 270) {
 767                         ctx->hflip = 1;
 768                         ctx->vflip = 0;
 769                         ctx->xy_swap = 1;
 770                 } else {
 771                         ctx->hflip = 0;
 772                         ctx->vflip = 0;
 773                         ctx->xy_swap = 0;
 774                 }
 775
 776                 vidioc_setup_cap_fmt(ctx, &fmt);
 777
 778                 /*
 779                  * If the rotation parameter changes the OUTPUT frames
 780                  * parameters, take them in account
 781                  */
 782                 ctx->out.pix_fmt = fmt;
 783
 784                 break;
 785         }
 786
 787         return 0;
 788 }
 789
 790 static const struct v4l2_ctrl_ops ge2d_ctrl_ops = {
 791         .s_ctrl = ge2d_s_ctrl,
 792 };
 793
 794 static int ge2d_setup_ctrls(struct ge2d_ctx *ctx)
 795 {
 796         struct meson_ge2d *ge2d = ctx->ge2d;
 797
 798         v4l2_ctrl_handler_init(&ctx->ctrl_handler, 4);
 799
 800         v4l2_ctrl_new_std(&ctx->ctrl_handler, &ge2d_ctrl_ops,
 801                           V4L2_CID_HFLIP, 0, 1, 1, 0);
 802
 803         v4l2_ctrl_new_std(&ctx->ctrl_handler, &ge2d_ctrl_ops,
 804                           V4L2_CID_VFLIP, 0, 1, 1, 0);
 805
 806         v4l2_ctrl_new_std(&ctx->ctrl_handler, &ge2d_ctrl_ops,
 807                           V4L2_CID_ROTATE, 0, 270, 90, 0);
 808
 809         if (ctx->ctrl_handler.error) {
 810                 int err = ctx->ctrl_handler.error;
 811
 812                 v4l2_err(&ge2d->v4l2_dev, "%s failed\n", __func__);
 813                 v4l2_ctrl_handler_free(&ctx->ctrl_handler);
 814                 return err;
 815         }
 816
 817         return 0;
 818 }
 819
 820 static const struct ge2d_frame def_frame = {
 821         .pix_fmt = {
 822                 .width = DEFAULT_WIDTH,
 823                 .height = DEFAULT_HEIGHT,
 824                 .bytesperline = DEFAULT_STRIDE,
 825                 .sizeimage = DEFAULT_STRIDE * DEFAULT_HEIGHT,
 826                 .field = V4L2_FIELD_NONE,
 827         },
 828         .crop.width = DEFAULT_WIDTH,
 829         .crop.height = DEFAULT_HEIGHT,
 830         .fmt = &formats[0],
 831 };
 832
 833 static int ge2d_open(struct file *file)
 834 {
 835         struct meson_ge2d *ge2d = video_drvdata(file);
 836         struct ge2d_ctx *ctx = NULL;
 837         int ret = 0;
 838
 839         ctx = kzalloc(sizeof(*ctx), GFP_KERNEL);
 840         if (!ctx)
 841                 return -ENOMEM;
 842         ctx->ge2d = ge2d;
 843
 844         /* Set default formats */
 845         ctx->in = def_frame;
 846         ctx->out = def_frame;
 847
 848         if (mutex_lock_interruptible(&ge2d->mutex)) {
 849                 kfree(ctx);
 850                 return -ERESTARTSYS;
 851         }
 852         ctx->fh.m2m_ctx = v4l2_m2m_ctx_init(ge2d->m2m_dev, ctx, &queue_init);
 853         if (IS_ERR(ctx->fh.m2m_ctx)) {
 854                 ret = PTR_ERR(ctx->fh.m2m_ctx);
 855                 mutex_unlock(&ge2d->mutex);
 856                 kfree(ctx);
 857                 return ret;
 858         }
 859         v4l2_fh_init(&ctx->fh, video_devdata(file));
 860         file->private_data = &ctx->fh;
 861         v4l2_fh_add(&ctx->fh);
 862
 863         ge2d_setup_ctrls(ctx);
 864
 865         /* Write the default values to the ctx struct */
 866         v4l2_ctrl_handler_setup(&ctx->ctrl_handler);
 867
 868         ctx->fh.ctrl_handler = &ctx->ctrl_handler;
 869         mutex_unlock(&ge2d->mutex);
 870
 871         return 0;
 872 }
 873
 874 static int ge2d_release(struct file *file)
 875 {
 876         struct ge2d_ctx *ctx =
 877                 container_of(file->private_data, struct ge2d_ctx, fh);
 878         struct meson_ge2d *ge2d = ctx->ge2d;
 879
 880         mutex_lock(&ge2d->mutex);
 881
 882         v4l2_m2m_ctx_release(ctx->fh.m2m_ctx);
 883
 884         v4l2_ctrl_handler_free(&ctx->ctrl_handler);
 885         v4l2_fh_del(&ctx->fh);
 886         v4l2_fh_exit(&ctx->fh);
 887         kfree(ctx);
 888
 889         mutex_unlock(&ge2d->mutex);
 890
 891         return 0;
 892 }
 893
 894 static const struct v4l2_file_operations ge2d_fops = {
 895         .owner = THIS_MODULE,
 896         .open = ge2d_open,
 897         .release = ge2d_release,
 898         .poll = v4l2_m2m_fop_poll,
 899         .unlocked_ioctl = video_ioctl2,
 900         .mmap = v4l2_m2m_fop_mmap,
 901 };
 902
 903 static const struct video_device ge2d_videodev = {
 904         .name = "meson-ge2d",
 905         .fops = &ge2d_fops,
 906         .ioctl_ops = &ge2d_ioctl_ops,
 907         .minor = -1,
 908         .release = video_device_release,
 909         .vfl_dir = VFL_DIR_M2M,
 910         .device_caps = V4L2_CAP_VIDEO_M2M | V4L2_CAP_STREAMING,
 911 };
 912
 913 static const struct regmap_config meson_ge2d_regmap_conf = {
 914         .reg_bits = 8,
 915         .val_bits = 32,
 916         .reg_stride = 4,
 917         .max_register = GE2D_SRC2_STRIDE_CTRL,
 918 };
 919
 920 static int ge2d_probe(struct platform_device *pdev)
 921 {
 922         struct reset_control *rst;
 923         struct video_device *vfd;
 924         struct meson_ge2d *ge2d;
 925         void __iomem *regs;
 926         int ret = 0;
 927         int irq;
 928
 929         if (!pdev->dev.of_node)
 930                 return -ENODEV;
 931
 932         ge2d = devm_kzalloc(&pdev->dev, sizeof(*ge2d), GFP_KERNEL);
 933         if (!ge2d)
 934                 return -ENOMEM;
 935
 936         ge2d->dev = &pdev->dev;
 937         mutex_init(&ge2d->mutex);
 938
 939         regs = devm_platform_ioremap_resource(pdev, 0);
 940         if (IS_ERR(regs))
 941                 return PTR_ERR(regs);
 942
 943         ge2d->map = devm_regmap_init_mmio(ge2d->dev, regs,
 944                                           &meson_ge2d_regmap_conf);
 945         if (IS_ERR(ge2d->map))
 946                 return PTR_ERR(ge2d->map);
 947
 948         irq = platform_get_irq(pdev, 0);
 949         ret = devm_request_irq(ge2d->dev, irq, ge2d_isr, 0,
 950                                dev_name(ge2d->dev), ge2d);
 951         if (ret < 0) {
 952                 dev_err(ge2d->dev, "failed to request irq\n");
 953                 return ret;
 954         }
 955
 956         rst = devm_reset_control_get(ge2d->dev, NULL);
 957         if (IS_ERR(rst)) {
 958                 dev_err(ge2d->dev, "failed to get core reset controller\n");
 959                 return PTR_ERR(rst);
 960         }
 961
 962         ge2d->clk = devm_clk_get(ge2d->dev, NULL);
 963         if (IS_ERR(ge2d->clk)) {
 964                 dev_err(ge2d->dev, "failed to get clock\n");
 965                 return PTR_ERR(ge2d->clk);
 966         }
 967
 968         reset_control_assert(rst);
 969         udelay(1);
 970         reset_control_deassert(rst);
 971
 972         ret = clk_prepare_enable(ge2d->clk);
 973         if (ret) {
 974                 dev_err(ge2d->dev, "Cannot enable ge2d sclk: %d\n", ret);
 975                 return ret;
 976         }
 977
 978         ret = v4l2_device_register(&pdev->dev, &ge2d->v4l2_dev);
 979         if (ret)
 980                 goto disable_clks;
 981
 982         vfd = video_device_alloc();
 983         if (!vfd) {
 984                 v4l2_err(&ge2d->v4l2_dev, "Failed to allocate video device\n");
 985                 ret = -ENOMEM;
 986                 goto unreg_v4l2_dev;
 987         }
 988
 989         *vfd = ge2d_videodev;
 990         vfd->lock = &ge2d->mutex;
 991         vfd->v4l2_dev = &ge2d->v4l2_dev;
 992
 993         video_set_drvdata(vfd, ge2d);
 994         ge2d->vfd = vfd;
 995
 996         platform_set_drvdata(pdev, ge2d);
 997         ge2d->m2m_dev = v4l2_m2m_init(&ge2d_m2m_ops);
 998         if (IS_ERR(ge2d->m2m_dev)) {
 999                 v4l2_err(&ge2d->v4l2_dev, "Failed to init mem2mem device\n");
1000                 ret = PTR_ERR(ge2d->m2m_dev);
1001                 goto rel_vdev;
1002         }
1003
1004         ret = video_register_device(vfd, VFL_TYPE_VIDEO, -1);
1005         if (ret) {
1006                 v4l2_err(&ge2d->v4l2_dev, "Failed to register video device\n");
1007                 goto rel_m2m;
1008         }
1009
1010         v4l2_info(&ge2d->v4l2_dev, "Registered %s as /dev/%s\n",
1011                   vfd->name, video_device_node_name(vfd));
1012
1013         return 0;
1014
1015 rel_m2m:
1016         v4l2_m2m_release(ge2d->m2m_dev);
1017 rel_vdev:
1018         video_device_release(ge2d->vfd);
1019 unreg_v4l2_dev:
1020         v4l2_device_unregister(&ge2d->v4l2_dev);
1021 disable_clks:
1022         clk_disable_unprepare(ge2d->clk);
1023
1024         return ret;
1025 }
1026
1027 static void ge2d_remove(struct platform_device *pdev)
1028 {
1029         struct meson_ge2d *ge2d = platform_get_drvdata(pdev);
1030
1031         video_unregister_device(ge2d->vfd);
1032         v4l2_m2m_release(ge2d->m2m_dev);
1033         v4l2_device_unregister(&ge2d->v4l2_dev);
1034         clk_disable_unprepare(ge2d->clk);
1035 }
1036
1037 static const struct of_device_id meson_ge2d_match[] = {
1038         {
1039                 .compatible = "amlogic,axg-ge2d",
1040         },
1041         {},
1042 };
1043
1044 MODULE_DEVICE_TABLE(of, meson_ge2d_match);
1045
1046 static struct platform_driver ge2d_drv = {
1047         .probe = ge2d_probe,
1048         .remove_new = ge2d_remove,
1049         .driver = {
1050                 .name = "meson-ge2d",
1051                 .of_match_table = meson_ge2d_match,
1052         },
1053 };
1054
1055 module_platform_driver(ge2d_drv);
1056
1057 MODULE_AUTHOR("Neil Armstrong <narmstrong@baylibre.com>");
1058 MODULE_DESCRIPTION("Amlogic 2D Graphic Acceleration Unit");
1059 MODULE_LICENSE("GPL");