V4L/DVB (10073): mt9m111: Add automatic white balance control

Signed-off-by: Robert Jarzmik <robert.jarzmik@free.fr>
Signed-off-by: Guennadi Liakhovetski <g.liakhovetski@gmx.de>
Signed-off-by: Mauro Carvalho Chehab <mchehab@redhat.com>

1 files changed, 27 insertions, 1 deletions

diff --git a/drivers/media/video/mt9m111.c b/drivers/media/video/mt9m111.c
index b0e6046ea967..2ab1e0f4925d 100644
--- a/drivers/media/video/mt9m111.c
+++ b/drivers/media/video/mt9m111.c
@@ -90,7 +90,7 @@
 #define MT9M111_OUTPUT_FORMAT_CTRL2_B   0x19b
 
 #define MT9M111_OPMODE_AUTOEXPO_EN      (1 << 14)
-
+#define MT9M111_OPMODE_AUTOWHITEBAL_EN  (1 << 1)
 
 #define MT9M111_OUTFMT_PROCESSED_BAYER  (1 << 14)
 #define MT9M111_OUTFMT_BYPASS_IFP       (1 << 10)
@@ -163,6 +163,7 @@ struct mt9m111 {
         unsigned int swap_rgb_red_blue:1;
         unsigned int swap_yuv_y_chromas:1;
         unsigned int swap_yuv_cb_cr:1;
+        unsigned int autowhitebalance:1;
 };
 
 static int reg_page_map_set(struct i2c_client *client, const u16 reg)
@@ -702,6 +703,23 @@ static int mt9m111_set_autoexposure(struct soc_camera_device *icd, int on)
 
         return ret;
 }
+
+static int mt9m111_set_autowhitebalance(struct soc_camera_device *icd, int on)
+{
+        struct mt9m111 *mt9m111 = container_of(icd, struct mt9m111, icd);
+        int ret;
+
+        if (on)
+                ret = reg_set(OPER_MODE_CTRL, MT9M111_OPMODE_AUTOWHITEBAL_EN);
+        else
+                ret = reg_clear(OPER_MODE_CTRL, MT9M111_OPMODE_AUTOWHITEBAL_EN);
+
+        if (!ret)
+                mt9m111->autowhitebalance = on;
+
+        return ret;
+}
+
 static int mt9m111_get_control(struct soc_camera_device *icd,
                                struct v4l2_control *ctrl)
 {
@@ -738,6 +756,9 @@ static int mt9m111_get_control(struct soc_camera_device *icd,
         case V4L2_CID_EXPOSURE_AUTO:
                 ctrl->value = mt9m111->autoexposure;
                 break;
+        case V4L2_CID_AUTO_WHITE_BALANCE:
+                ctrl->value = mt9m111->autowhitebalance;
+                break;
         }
         return 0;
 }
@@ -771,6 +792,9 @@ static int mt9m111_set_control(struct soc_camera_device *icd,
         case V4L2_CID_EXPOSURE_AUTO:
                 ret =  mt9m111_set_autoexposure(icd, ctrl->value);
                 break;
+        case V4L2_CID_AUTO_WHITE_BALANCE:
+                ret =  mt9m111_set_autowhitebalance(icd, ctrl->value);
+                break;
         default:
                 ret = -EINVAL;
         }
@@ -789,6 +813,7 @@ static int mt9m111_restore_state(struct soc_camera_device *icd)
         mt9m111_set_flip(icd, mt9m111->vflip, MT9M111_RMB_MIRROR_ROWS);
         mt9m111_set_global_gain(icd, icd->gain);
         mt9m111_set_autoexposure(icd, mt9m111->autoexposure);
+        mt9m111_set_autowhitebalance(icd, mt9m111->autowhitebalance);
         return 0;
 }
 
@@ -883,6 +908,7 @@ static int mt9m111_video_probe(struct soc_camera_device *icd)
                 goto eisis;
 
         mt9m111->autoexposure = 1;
+        mt9m111->autowhitebalance = 1;
 
         mt9m111->swap_rgb_even_odd = 1;
         mt9m111->swap_rgb_red_blue = 1;