staging: bcm2835-camera: Ensure H264 header bytes get a sensible timestamp

H264 header come from VC with 0 timestamps, which means they get a
strange timestamp when processed with VC/kernel start times,
particularly if used with the inline header option.
Remember the last frame timestamp and use that if set, or otherwise
use the kernel start time.

https://github.com/raspberrypi/linux/issues/1836

Signed-off-by: Dave Stevenson <dave.stevenson@raspberrypi.org>

diff --git a/drivers/staging/vc04_services/bcm2835-camera/bcm2835-camera.c b/drivers/staging/vc04_services/bcm2835-camera/bcm2835-camera.c
index 84ca22d..bb82f4b 100644 (file)
--- a/drivers/staging/vc04_services/bcm2835-camera/bcm2835-camera.c
+++ b/drivers/staging/vc04_services/bcm2835-camera/bcm2835-camera.c
@@ -360,8 +360,13 @@ static void buffer_cb(struct vchiq_mmal_instance *instance,
                }
        } else {
                if (dev->capture.frame_count) {
-                       if (dev->capture.vc_start_timestamp != -1 &&
-                           pts != 0) {
+                       if (dev->capture.vc_start_timestamp == -1) {
+                               buf->vb.vb2_buf.timestamp = ktime_get_ns();
+                               v4l2_dbg(1, bcm2835_v4l2_debug, &dev->v4l2_dev,
+                                        "Buffer time set as current time - %lld",
+                                        buf->vb.vb2_buf.timestamp);
+
+                       } else if (pts != 0) {
                                ktime_t timestamp;
                                s64 runtime_us = pts -
                                    dev->capture.vc_start_timestamp;
@@ -374,10 +379,28 @@ static void buffer_cb(struct vchiq_mmal_instance *instance,
                                         ktime_to_ns(timestamp));
                                buf->vb.vb2_buf.timestamp = ktime_to_ns(timestamp);
                        } else {
-                               buf->vb.vb2_buf.timestamp = ktime_get_ns();
+                               if (dev->capture.last_timestamp) {
+                                       buf->vb.vb2_buf.timestamp =
+                                               dev->capture.last_timestamp;
+                                       v4l2_dbg(1, bcm2835_v4l2_debug,
+                                                &dev->v4l2_dev,
+                                                "Buffer time set as last timestamp - %lld",
+                                                buf->vb.vb2_buf.timestamp);
+                               } else {
+                                       buf->vb.vb2_buf.timestamp =
+                                               ktime_to_ns(dev->capture.kernel_start_ts);
+                                       v4l2_dbg(1, bcm2835_v4l2_debug,
+                                                &dev->v4l2_dev,
+                                                "Buffer time set as start timestamp - %lld",
+                                                buf->vb.vb2_buf.timestamp);
+                               }
                        }
+                       dev->capture.last_timestamp = buf->vb.vb2_buf.timestamp;
 
                        vb2_set_plane_payload(&buf->vb.vb2_buf, 0, length);
+                       v4l2_dbg(1, bcm2835_v4l2_debug, &dev->v4l2_dev,
+                                "Buffer has ts %llu",
+                                dev->capture.last_timestamp);
                        vb2_buffer_done(&buf->vb.vb2_buf, VB2_BUF_STATE_DONE);
 
                        if (mmal_flags & MMAL_BUFFER_HEADER_FLAG_EOS &&
@@ -543,6 +566,7 @@ static int start_streaming(struct vb2_queue *vq, unsigned int count)
                         dev->capture.vc_start_timestamp, parameter_size);
 
        dev->capture.kernel_start_ts = ktime_get();
+       dev->capture.last_timestamp = 0;
 
        /* enable the camera port */
        dev->capture.port->cb_ctx = dev;

diff --git a/drivers/staging/vc04_services/bcm2835-camera/bcm2835-camera.h b/drivers/staging/vc04_services/bcm2835-camera/bcm2835-camera.h
index 2b5679e..09273b0 100644 (file)
--- a/drivers/staging/vc04_services/bcm2835-camera/bcm2835-camera.h
+++ b/drivers/staging/vc04_services/bcm2835-camera/bcm2835-camera.h
@@ -90,6 +90,8 @@ struct bm2835_mmal_dev {
                s64         vc_start_timestamp;
                /* Kernel start timestamp for streaming */
                ktime_t kernel_start_ts;
+               /* Timestamp of last frame */
+               u64             last_timestamp;
 
                struct vchiq_mmal_port  *port; /* port being used for capture */
                /* camera port being used for capture */