1 files changed, 33 insertions, 30 deletions

diff --git a/drivers/media/platform/vimc/vimc-scaler.c b/drivers/media/platform/vimc/vimc-scaler.c
index 5f31c1e351a3..49ab8d9dd9c9 100644
--- a/drivers/media/platform/vimc/vimc-scaler.c
+++ b/drivers/media/platform/vimc/vimc-scaler.c
@@ -25,12 +25,6 @@ MODULE_PARM_DESC(sca_mult, " the image size multiplier");
 #define IS_SRC(pad)     (pad)
 #define MAX_ZOOM        8
 
-static const u32 vimc_sca_supported_pixfmt[] = {
-        V4L2_PIX_FMT_BGR24,
-        V4L2_PIX_FMT_RGB24,
-        V4L2_PIX_FMT_ARGB32,
-};
-
 struct vimc_sca_device {
         struct vimc_ent_device ved;
         struct v4l2_subdev sd;
@@ -53,16 +47,6 @@ static const struct v4l2_mbus_framefmt sink_fmt_default = {
         .colorspace = V4L2_COLORSPACE_DEFAULT,
 };
 
-static bool vimc_sca_is_pixfmt_supported(u32 pixelformat)
-{
-        unsigned int i;
-
-        for (i = 0; i < ARRAY_SIZE(vimc_sca_supported_pixfmt); i++)
-                if (vimc_sca_supported_pixfmt[i] == pixelformat)
-                        return true;
-        return false;
-}
-
 static int vimc_sca_init_cfg(struct v4l2_subdev *sd,
                              struct v4l2_subdev_pad_config *cfg)
 {
@@ -82,13 +66,35 @@ static int vimc_sca_init_cfg(struct v4l2_subdev *sd,
         return 0;
 }
 
+static int vimc_sca_enum_mbus_code(struct v4l2_subdev *sd,
+                                   struct v4l2_subdev_pad_config *cfg,
+                                   struct v4l2_subdev_mbus_code_enum *code)
+{
+        const struct vimc_pix_map *vpix = vimc_pix_map_by_index(code->index);
+
+        /* We don't support bayer format */
+        if (!vpix || vpix->bayer)
+                return -EINVAL;
+
+        code->code = vpix->code;
+
+        return 0;
+}
+
 static int vimc_sca_enum_frame_size(struct v4l2_subdev *sd,
                                     struct v4l2_subdev_pad_config *cfg,
                                     struct v4l2_subdev_frame_size_enum *fse)
 {
+        const struct vimc_pix_map *vpix;
+
         if (fse->index)
                 return -EINVAL;
 
+        /* Only accept code in the pix map table in non bayer format */
+        vpix = vimc_pix_map_by_code(fse->code);
+        if (!vpix || vpix->bayer)
+                return -EINVAL;
+
         fse->min_width = VIMC_FRAME_MIN_WIDTH;
         fse->min_height = VIMC_FRAME_MIN_HEIGHT;
 
@@ -125,6 +131,13 @@ static int vimc_sca_get_fmt(struct v4l2_subdev *sd,
 
 static void vimc_sca_adjust_sink_fmt(struct v4l2_mbus_framefmt *fmt)
 {
+        const struct vimc_pix_map *vpix;
+
+        /* Only accept code in the pix map table in non bayer format */
+        vpix = vimc_pix_map_by_code(fmt->code);
+        if (!vpix || vpix->bayer)
+                fmt->code = sink_fmt_default.code;
+
         fmt->width = clamp_t(u32, fmt->width, VIMC_FRAME_MIN_WIDTH,
                              VIMC_FRAME_MAX_WIDTH) & ~1;
         fmt->height = clamp_t(u32, fmt->height, VIMC_FRAME_MIN_HEIGHT,
@@ -143,9 +156,6 @@ static int vimc_sca_set_fmt(struct v4l2_subdev *sd,
         struct vimc_sca_device *vsca = v4l2_get_subdevdata(sd);
         struct v4l2_mbus_framefmt *sink_fmt;
 
-        if (!vimc_mbus_code_supported(fmt->format.code))
-                fmt->format.code = sink_fmt_default.code;
-
         if (fmt->which == V4L2_SUBDEV_FORMAT_ACTIVE) {
                 /* Do not change the format while stream is on */
                 if (vsca->src_frame)
@@ -188,7 +198,7 @@ static int vimc_sca_set_fmt(struct v4l2_subdev *sd,
 
 static const struct v4l2_subdev_pad_ops vimc_sca_pad_ops = {
         .init_cfg               = vimc_sca_init_cfg,
-        .enum_mbus_code         = vimc_enum_mbus_code,
+        .enum_mbus_code         = vimc_sca_enum_mbus_code,
         .enum_frame_size        = vimc_sca_enum_frame_size,
         .get_fmt                = vimc_sca_get_fmt,
         .set_fmt                = vimc_sca_set_fmt,
@@ -199,22 +209,15 @@ static int vimc_sca_s_stream(struct v4l2_subdev *sd, int enable)
         struct vimc_sca_device *vsca = v4l2_get_subdevdata(sd);
 
         if (enable) {
-                u32 pixelformat = vsca->ved.stream->producer_pixfmt;
-                const struct v4l2_format_info *pix_info;
+                const struct vimc_pix_map *vpix;
                 unsigned int frame_size;
 
                 if (vsca->src_frame)
                         return 0;
 
-                if (!vimc_sca_is_pixfmt_supported(pixelformat)) {
-                        dev_err(vsca->dev, "pixfmt (0x%08x) is not supported\n",
-                                pixelformat);
-                        return -EINVAL;
-                }
-
                 /* Save the bytes per pixel of the sink */
-                pix_info = v4l2_format_info(pixelformat);
-                vsca->bpp = pix_info->bpp[0];
+                vpix = vimc_pix_map_by_code(vsca->sink_fmt.code);
+                vsca->bpp = vpix->bpp;
 
                 /* Calculate the width in bytes of the src frame */
                 vsca->src_line_size = vsca->sink_fmt.width *