Quelle  vsp1_rwpf.c   Sprache: C

// SPDX-License-Identifier: GPL-2.0+
/*
 * vsp1_rwpf.c  --  R-Car VSP1 Read and Write Pixel Formatters
 *
 * Copyright (C) 2013-2014 Renesas Electronics Corporation
 *
 * Contact: Laurent Pinchart (laurent.pinchart@ideasonboard.com)
 */

#include <media/v4l2-subdev.h>

#include "vsp1.h"
#include "vsp1_entity.h"
#include "vsp1_rwpf.h"
#include "vsp1_video.h"

#define RWPF_MIN_WIDTH    1
#define RWPF_MIN_HEIGHT    1

/* -----------------------------------------------------------------------------
 * V4L2 Subdevice Operations
 */

static int vsp1_rwpf_enum_mbus_code(struct v4l2_subdev *subdev,
        struct v4l2_subdev_state *sd_state,
        struct v4l2_subdev_mbus_code_enum *code)
{
 static const unsigned int codes[] = {
  MEDIA_BUS_FMT_ARGB8888_1X32,
  MEDIA_BUS_FMT_AHSV8888_1X32,
  MEDIA_BUS_FMT_AYUV8_1X32,
 };

 if (code->index >= ARRAY_SIZE(codes))
  return -EINVAL;

 code->code = codes[code->index];

 if (code->pad == RWPF_PAD_SOURCE &&
     code->code == MEDIA_BUS_FMT_AYUV8_1X32)
  code->flags = V4L2_SUBDEV_MBUS_CODE_CSC_YCBCR_ENC
       | V4L2_SUBDEV_MBUS_CODE_CSC_QUANTIZATION;

 return 0;
}

static int vsp1_rwpf_enum_frame_size(struct v4l2_subdev *subdev,
         struct v4l2_subdev_state *sd_state,
         struct v4l2_subdev_frame_size_enum *fse)
{
 struct vsp1_rwpf *rwpf = to_rwpf(subdev);

 return vsp1_subdev_enum_frame_size(subdev, sd_state, fse,
        RWPF_MIN_WIDTH,
        RWPF_MIN_HEIGHT, rwpf->max_width,
        rwpf->max_height);
}

static int vsp1_rwpf_set_format(struct v4l2_subdev *subdev,
    struct v4l2_subdev_state *sd_state,
    struct v4l2_subdev_format *fmt)
{
 struct vsp1_rwpf *rwpf = to_rwpf(subdev);
 struct v4l2_subdev_state *state;
 struct v4l2_mbus_framefmt *format;
 int ret = 0;

 mutex_lock(&rwpf->entity.lock);

 state = vsp1_entity_get_state(&rwpf->entity, sd_state, fmt->which);
 if (!state) {
  ret = -EINVAL;
  goto done;
 }

 /* Default to YUV if the requested format is not supported. */
 if (fmt->format.code != MEDIA_BUS_FMT_ARGB8888_1X32 &&
     fmt->format.code != MEDIA_BUS_FMT_AHSV8888_1X32 &&
     fmt->format.code != MEDIA_BUS_FMT_AYUV8_1X32)
  fmt->format.code = MEDIA_BUS_FMT_AYUV8_1X32;

 format = v4l2_subdev_state_get_format(state, fmt->pad);

 if (fmt->pad == RWPF_PAD_SOURCE) {
  const struct v4l2_mbus_framefmt *sink_format =
   v4l2_subdev_state_get_format(state, RWPF_PAD_SINK);
  u16 flags = fmt->format.flags & V4L2_MBUS_FRAMEFMT_SET_CSC;
  bool csc;

  /*
 * The RWPF performs format conversion but can't scale, only the
 * format code, encoding and quantization can be changed on the
 * source pad when converting between RGB and YUV.
 */
  if (sink_format->code != MEDIA_BUS_FMT_AHSV8888_1X32 &&
      fmt->format.code != MEDIA_BUS_FMT_AHSV8888_1X32)
   format->code = fmt->format.code;
  else
   format->code = sink_format->code;

  /*
 * Encoding and quantization can only be configured when YCbCr
 * <-> RGB is enabled. The V4L2 API requires userspace to set
 * the V4L2_MBUS_FRAMEFMT_SET_CSC flag. If either of these
 * conditions is not met, use the encoding and quantization
 * values from the sink pad.
 */
  csc = (format->code == MEDIA_BUS_FMT_AYUV8_1X32) !=
        (sink_format->code == MEDIA_BUS_FMT_AYUV8_1X32);

  if (csc && (flags & V4L2_MBUS_FRAMEFMT_SET_CSC)) {
   format->ycbcr_enc = fmt->format.ycbcr_enc;
   format->quantization = fmt->format.quantization;
  } else {
   format->ycbcr_enc = sink_format->ycbcr_enc;
   format->quantization = sink_format->quantization;
  }

  vsp1_entity_adjust_color_space(format);

  fmt->format = *format;
  fmt->format.flags = flags;

  goto done;
 }

 format->code = fmt->format.code;
 format->width = clamp_t(unsigned int, fmt->format.width,
    RWPF_MIN_WIDTH, rwpf->max_width);
 format->height = clamp_t(unsigned int, fmt->format.height,
     RWPF_MIN_HEIGHT, rwpf->max_height);
 format->field = V4L2_FIELD_NONE;

 format->colorspace = fmt->format.colorspace;
 format->xfer_func = fmt->format.xfer_func;
 format->ycbcr_enc = fmt->format.ycbcr_enc;
 format->quantization = fmt->format.quantization;

 vsp1_entity_adjust_color_space(format);

 fmt->format = *format;

 if (rwpf->entity.type == VSP1_ENTITY_RPF) {
  struct v4l2_rect *crop;

  /* Update the sink crop rectangle. */
  crop = v4l2_subdev_state_get_crop(state, RWPF_PAD_SINK);
  crop->left = 0;
  crop->top = 0;
  crop->width = fmt->format.width;
  crop->height = fmt->format.height;
 }

 /* Propagate the format to the source pad. */
 format = v4l2_subdev_state_get_format(state, RWPF_PAD_SOURCE);
 *format = fmt->format;

 if (rwpf->flip.rotate) {
  format->width = fmt->format.height;
  format->height = fmt->format.width;
 }

done:
 mutex_unlock(&rwpf->entity.lock);
 return ret;
}

static int vsp1_rwpf_get_selection(struct v4l2_subdev *subdev,
       struct v4l2_subdev_state *sd_state,
       struct v4l2_subdev_selection *sel)
{
 struct vsp1_rwpf *rwpf = to_rwpf(subdev);
 struct v4l2_subdev_state *state;
 struct v4l2_mbus_framefmt *format;
 int ret = 0;

 /*
 * Cropping is only supported on the RPF and is implemented on the sink
 * pad.
 */
 if (rwpf->entity.type == VSP1_ENTITY_WPF || sel->pad != RWPF_PAD_SINK)
  return -EINVAL;

 mutex_lock(&rwpf->entity.lock);

 state = vsp1_entity_get_state(&rwpf->entity, sd_state, sel->which);
 if (!state) {
  ret = -EINVAL;
  goto done;
 }

 switch (sel->target) {
 case V4L2_SEL_TGT_CROP:
  sel->r = *v4l2_subdev_state_get_crop(state, RWPF_PAD_SINK);
  break;

 case V4L2_SEL_TGT_CROP_BOUNDS:
  format = v4l2_subdev_state_get_format(state, RWPF_PAD_SINK);
  sel->r.left = 0;
  sel->r.top = 0;
  sel->r.width = format->width;
  sel->r.height = format->height;
  break;

 default:
  ret = -EINVAL;
  break;
 }

done:
 mutex_unlock(&rwpf->entity.lock);
 return ret;
}

static int vsp1_rwpf_set_selection(struct v4l2_subdev *subdev,
       struct v4l2_subdev_state *sd_state,
       struct v4l2_subdev_selection *sel)
{
 struct vsp1_rwpf *rwpf = to_rwpf(subdev);
 struct v4l2_subdev_state *state;
 struct v4l2_mbus_framefmt *format;
 struct v4l2_rect *crop;
 int ret = 0;

 /*
 * Cropping is only supported on the RPF and is implemented on the sink
 * pad.
 */
 if (rwpf->entity.type == VSP1_ENTITY_WPF || sel->pad != RWPF_PAD_SINK)
  return -EINVAL;

 if (sel->target != V4L2_SEL_TGT_CROP)
  return -EINVAL;

 mutex_lock(&rwpf->entity.lock);

 state = vsp1_entity_get_state(&rwpf->entity, sd_state, sel->which);
 if (!state) {
  ret = -EINVAL;
  goto done;
 }

 /* Make sure the crop rectangle is entirely contained in the image. */
 format = v4l2_subdev_state_get_format(state, RWPF_PAD_SINK);

 /*
 * Restrict the crop rectangle coordinates to multiples of 2 to avoid
 * shifting the color plane.
 */
 if (format->code == MEDIA_BUS_FMT_AYUV8_1X32) {
  sel->r.left = ALIGN(sel->r.left, 2);
  sel->r.top = ALIGN(sel->r.top, 2);
  sel->r.width = round_down(sel->r.width, 2);
  sel->r.height = round_down(sel->r.height, 2);
 }

 sel->r.left = min_t(unsigned int, sel->r.left, format->width - 2);
 sel->r.top = min_t(unsigned int, sel->r.top, format->height - 2);
 sel->r.width = min_t(unsigned int, sel->r.width,
        format->width - sel->r.left);
 sel->r.height = min_t(unsigned int, sel->r.height,
         format->height - sel->r.top);

 crop = v4l2_subdev_state_get_crop(state, RWPF_PAD_SINK);
 *crop = sel->r;

 /* Propagate the format to the source pad. */
 format = v4l2_subdev_state_get_format(state, RWPF_PAD_SOURCE);
 format->width = crop->width;
 format->height = crop->height;

done:
 mutex_unlock(&rwpf->entity.lock);
 return ret;
}

static const struct v4l2_subdev_pad_ops vsp1_rwpf_pad_ops = {
 .enum_mbus_code = vsp1_rwpf_enum_mbus_code,
 .enum_frame_size = vsp1_rwpf_enum_frame_size,
 .get_fmt = vsp1_subdev_get_pad_format,
 .set_fmt = vsp1_rwpf_set_format,
 .get_selection = vsp1_rwpf_get_selection,
 .set_selection = vsp1_rwpf_set_selection,
};

const struct v4l2_subdev_ops vsp1_rwpf_subdev_ops = {
 .pad    = &vsp1_rwpf_pad_ops,
};

/* -----------------------------------------------------------------------------
 * Controls
 */

static int vsp1_rwpf_s_ctrl(struct v4l2_ctrl *ctrl)
{
 struct vsp1_rwpf *rwpf =
  container_of(ctrl->handler, struct vsp1_rwpf, ctrls);

 switch (ctrl->id) {
 case V4L2_CID_ALPHA_COMPONENT:
  rwpf->alpha = ctrl->val;
  break;
 }

 return 0;
}

static const struct v4l2_ctrl_ops vsp1_rwpf_ctrl_ops = {
 .s_ctrl = vsp1_rwpf_s_ctrl,
};

int vsp1_rwpf_init_ctrls(struct vsp1_rwpf *rwpf, unsigned int ncontrols)
{
 v4l2_ctrl_handler_init(&rwpf->ctrls, ncontrols + 1);
 v4l2_ctrl_new_std(&rwpf->ctrls, &vsp1_rwpf_ctrl_ops,
     V4L2_CID_ALPHA_COMPONENT, 0, 255, 1, 255);

 rwpf->entity.subdev.ctrl_handler = &rwpf->ctrls;

 return rwpf->ctrls.error;
}