[media] gspca_kinect: add support for the depth stream

Add support for the depth stream at 10bpp, for now use a 'depth_mode'
command line parameter to switch between video and depth mode.

Signed-off-by: Alexander Sosna <alexander@xxor.de>
Signed-off-by: Antonio Ospite <ao2@ao2.it>
Signed-off-by: Hans de Goede <hdegoede@redhat.com>
Signed-off-by: Mauro Carvalho Chehab <m.chehab@samsung.com>

1 files changed, 87 insertions, 11 deletions

diff --git a/drivers/media/usb/gspca/kinect.c b/drivers/media/usb/gspca/kinect.c
index 081f05162809..45bc1f51c5d8 100644
--- a/drivers/media/usb/gspca/kinect.c
+++ b/drivers/media/usb/gspca/kinect.c
@@ -36,6 +36,8 @@ MODULE_AUTHOR("Antonio Ospite <ospite@studenti.unina.it>");
 MODULE_DESCRIPTION("GSPCA/Kinect Sensor Device USB Camera Driver");
 MODULE_LICENSE("GPL");
 
+static bool depth_mode;
+
 struct pkt_hdr {
         uint8_t magic[2];
         uint8_t pad;
@@ -73,6 +75,14 @@ struct sd {
 
 #define FPS_HIGH       0x0100
 
+static const struct v4l2_pix_format depth_camera_mode[] = {
+        {640, 480, V4L2_PIX_FMT_Y10BPACK, V4L2_FIELD_NONE,
+         .bytesperline = 640 * 10 / 8,
+         .sizeimage =  640 * 480 * 10 / 8,
+         .colorspace = V4L2_COLORSPACE_SRGB,
+         .priv = MODE_640x488 | FORMAT_Y10B},
+};
+
 static const struct v4l2_pix_format video_camera_mode[] = {
         {640, 480, V4L2_PIX_FMT_SGRBG8, V4L2_FIELD_NONE,
          .bytesperline = 640,
@@ -219,7 +229,7 @@ static int write_register(struct gspca_dev *gspca_dev, uint16_t reg,
 }
 
 /* this function is called at probe time */
-static int sd_config(struct gspca_dev *gspca_dev,
+static int sd_config_video(struct gspca_dev *gspca_dev,
                      const struct usb_device_id *id)
 {
         struct sd *sd = (struct sd *) gspca_dev;
@@ -227,8 +237,6 @@ static int sd_config(struct gspca_dev *gspca_dev,
 
         sd->cam_tag = 0;
 
-        /* Only video stream is supported for now,
-         * which has stream flag = 0x80 */
         sd->stream_flag = 0x80;
 
         cam = &gspca_dev->cam;
@@ -236,6 +244,8 @@ static int sd_config(struct gspca_dev *gspca_dev,
         cam->cam_mode = video_camera_mode;
         cam->nmodes = ARRAY_SIZE(video_camera_mode);
 
+        gspca_dev->xfer_ep = 0x81;
+
 #if 0
         /* Setting those values is not needed for video stream */
         cam->npkt = 15;
@@ -245,6 +255,26 @@ static int sd_config(struct gspca_dev *gspca_dev,
         return 0;
 }
 
+static int sd_config_depth(struct gspca_dev *gspca_dev,
+                     const struct usb_device_id *id)
+{
+        struct sd *sd = (struct sd *) gspca_dev;
+        struct cam *cam;
+
+        sd->cam_tag = 0;
+
+        sd->stream_flag = 0x70;
+
+        cam = &gspca_dev->cam;
+
+        cam->cam_mode = depth_camera_mode;
+        cam->nmodes = ARRAY_SIZE(depth_camera_mode);
+
+        gspca_dev->xfer_ep = 0x82;
+
+        return 0;
+}
+
 /* this function is called at probe and resume time */
 static int sd_init(struct gspca_dev *gspca_dev)
 {
@@ -253,7 +283,7 @@ static int sd_init(struct gspca_dev *gspca_dev)
         return 0;
 }
 
-static int sd_start(struct gspca_dev *gspca_dev)
+static int sd_start_video(struct gspca_dev *gspca_dev)
 {
         int mode;
         uint8_t fmt_reg, fmt_val;
@@ -325,12 +355,39 @@ static int sd_start(struct gspca_dev *gspca_dev)
         return 0;
 }
 
-static void sd_stopN(struct gspca_dev *gspca_dev)
+static int sd_start_depth(struct gspca_dev *gspca_dev)
+{
+        /* turn off IR-reset function */
+        write_register(gspca_dev, 0x105, 0x00);
+
+        /* reset depth stream */
+        write_register(gspca_dev, 0x06, 0x00);
+        /* Depth Stream Format 0x03: 11 bit stream | 0x02: 10 bit */
+        write_register(gspca_dev, 0x12, 0x02);
+        /* Depth Stream Resolution 1: standard (640x480) */
+        write_register(gspca_dev, 0x13, 0x01);
+        /* Depth Framerate / 0x1e (30): 30 fps */
+        write_register(gspca_dev, 0x14, 0x1e);
+        /* Depth Stream Control  / 2: Open Depth Stream */
+        write_register(gspca_dev, 0x06, 0x02);
+        /* disable depth hflip / LSB = 0: Smoothing Disabled */
+        write_register(gspca_dev, 0x17, 0x00);
+
+        return 0;
+}
+
+static void sd_stopN_video(struct gspca_dev *gspca_dev)
 {
         /* reset video stream */
         write_register(gspca_dev, 0x05, 0x00);
 }
 
+static void sd_stopN_depth(struct gspca_dev *gspca_dev)
+{
+        /* reset depth stream */
+        write_register(gspca_dev, 0x06, 0x00);
+}
+
 static void sd_pkt_scan(struct gspca_dev *gspca_dev, u8 *__data, int len)
 {
         struct sd *sd = (struct sd *) gspca_dev;
@@ -366,12 +423,24 @@ static void sd_pkt_scan(struct gspca_dev *gspca_dev, u8 *__data, int len)
 }
 
 /* sub-driver description */
-static const struct sd_desc sd_desc = {
+static const struct sd_desc sd_desc_video = {
         .name      = MODULE_NAME,
-        .config    = sd_config,
+        .config    = sd_config_video,
         .init      = sd_init,
-        .start     = sd_start,
-        .stopN     = sd_stopN,
+        .start     = sd_start_video,
+        .stopN     = sd_stopN_video,
+        .pkt_scan  = sd_pkt_scan,
+        /*
+        .get_streamparm = sd_get_streamparm,
+        .set_streamparm = sd_set_streamparm,
+        */
+};
+static const struct sd_desc sd_desc_depth = {
+        .name      = MODULE_NAME,
+        .config    = sd_config_depth,
+        .init      = sd_init,
+        .start     = sd_start_depth,
+        .stopN     = sd_stopN_depth,
         .pkt_scan  = sd_pkt_scan,
         /*
         .get_streamparm = sd_get_streamparm,
@@ -391,8 +460,12 @@ MODULE_DEVICE_TABLE(usb, device_table);
 /* -- device connect -- */
 static int sd_probe(struct usb_interface *intf, const struct usb_device_id *id)
 {
-        return gspca_dev_probe(intf, id, &sd_desc, sizeof(struct sd),
-                                THIS_MODULE);
+        if (depth_mode)
+                return gspca_dev_probe(intf, id, &sd_desc_depth,
+                                       sizeof(struct sd), THIS_MODULE);
+        else
+                return gspca_dev_probe(intf, id, &sd_desc_video,
+                                       sizeof(struct sd), THIS_MODULE);
 }
 
 static struct usb_driver sd_driver = {
@@ -408,3 +481,6 @@ static struct usb_driver sd_driver = {
 };
 
 module_usb_driver(sd_driver);
+
+module_param(depth_mode, bool, 0644);
+MODULE_PARM_DESC(depth_mode, "0=video 1=depth");