]> git.sesse.net Git - ffmpeg/blobdiff - libavformat/dv1394.c
align
[ffmpeg] / libavformat / dv1394.c
index 7208277a1d87bcce41bbb6c20bb0ea5ef42a666e..b0892a3ef4c6b399bf63f8293a9000d243b171a4 100644 (file)
@@ -19,6 +19,7 @@
 
 #include <unistd.h>
 #include <fcntl.h>
+#include <errno.h>
 #include <sys/ioctl.h>
 #include <sys/mman.h>
 #include <sys/poll.h>
 #undef DV1394_DEBUG
 
 #include "dv1394.h"
+#include "dv.h"
 
 struct dv1394_data {
     int fd;
     int channel;
-    int width, height;
-    int frame_rate;
-    int frame_size;
     int format;
 
     void *ring; /* Ring buffer */
@@ -44,10 +43,17 @@ struct dv1394_data {
     int avail;  /* Number of frames available for reading */
     int done;   /* Number of completed frames */
 
-    int stream; /* Current stream. 0 - video, 1 - audio */
-    INT64 pts;  /* Current timestamp */
+    int64_t pts;  /* Current timestamp */
+
+    void* dv_demux; /* Generic DV muxing/demuxing context */
 };
 
+/* 
+ * The trick here is to kludge around well known problem with kernel Ooopsing
+ * when you try to capture PAL on a device node configure for NTSC. That's 
+ * why we have to configure the device node for PAL, and then read only NTSC
+ * amount of data.
+ */
 static int dv1394_reset(struct dv1394_data *dv)
 {
     struct dv1394_init init;
@@ -55,12 +61,12 @@ static int dv1394_reset(struct dv1394_data *dv)
     init.channel     = dv->channel;
     init.api_version = DV1394_API_VERSION;
     init.n_frames    = DV1394_RING_FRAMES;
-    init.format      = dv->format;
+    init.format      = DV1394_PAL;
 
     if (ioctl(dv->fd, DV1394_INIT, &init) < 0)
         return -1;
 
-    dv->avail = 0;
+    dv->avail  = dv->done = 0;
     return 0;
 }
 
@@ -77,37 +83,22 @@ static int dv1394_start(struct dv1394_data *dv)
 static int dv1394_read_header(AVFormatContext * context, AVFormatParameters * ap)
 {
     struct dv1394_data *dv = context->priv_data;
-    AVStream *vst, *ast;
     const char *video_device;
 
-    vst = av_new_stream(context, 0);
-    if (!vst)
-        return -ENOMEM;
-    ast = av_new_stream(context, 1);
-    if (!ast) {
-       av_free(vst);
-        return -ENOMEM;
-    }
+    dv->dv_demux = dv_init_demux(context, 0, 1);
+    if (!dv->dv_demux)
+        goto failed;
 
-    dv->width   = DV1394_WIDTH;
-    dv->height  = DV1394_HEIGHT;
+    if (ap->standard && !strcasecmp(ap->standard, "pal"))
+       dv->format = DV1394_PAL;
+    else
+       dv->format = DV1394_NTSC;
 
     if (ap->channel)
         dv->channel = ap->channel;
     else
         dv->channel = DV1394_DEFAULT_CHANNEL;
 
-    /* FIXME: Need a format change parameter */
-    dv->format = DV1394_NTSC;
-
-    if (dv->format == DV1394_NTSC) {
-        dv->frame_size = DV1394_NTSC_FRAME_SIZE;
-        dv->frame_rate = 30;
-    } else {
-        dv->frame_size = DV1394_PAL_FRAME_SIZE;
-        dv->frame_rate = 25;
-    }
-
     /* Open and initialize DV1394 device */
     video_device = ap->device;
     if (!video_device)
@@ -123,27 +114,13 @@ static int dv1394_read_header(AVFormatContext * context, AVFormatParameters * ap
         goto failed;
     }
 
-    dv->ring = mmap(NULL, DV1394_NTSC_FRAME_SIZE * DV1394_RING_FRAMES,
+    dv->ring = mmap(NULL, DV1394_PAL_FRAME_SIZE * DV1394_RING_FRAMES,
                     PROT_READ, MAP_PRIVATE, dv->fd, 0);
-    if (!dv->ring) {
+    if (dv->ring == MAP_FAILED) {
         perror("Failed to mmap DV ring buffer");
         goto failed;
     }
 
-    dv->stream = 0;
-
-    vst->codec.codec_type = CODEC_TYPE_VIDEO;
-    vst->codec.codec_id   = CODEC_ID_DVVIDEO;
-    vst->codec.width      = dv->width;
-    vst->codec.height     = dv->height;
-    vst->codec.frame_rate = dv->frame_rate * FRAME_RATE_BASE;
-    vst->codec.bit_rate   = 25000000;  /* Consumer DV is 25Mbps */
-
-    ast->codec.codec_type = CODEC_TYPE_AUDIO;
-    ast->codec.codec_id   = CODEC_ID_DVAUDIO;
-    ast->codec.channels   = 2;
-    ast->codec.sample_rate= 48000;
-
     av_set_pts_info(context, 48, 1, 1000000);
 
     if (dv1394_start(dv) < 0)
@@ -153,44 +130,44 @@ static int dv1394_read_header(AVFormatContext * context, AVFormatParameters * ap
 
 failed:
     close(dv->fd);
-    av_free(vst);
-    av_free(ast);
     return -EIO;
 }
 
-static inline int __copy_frame(struct dv1394_data *dv, AVPacket *pkt)
-{
-    char *ptr = dv->ring + (dv->index * dv->frame_size);
-
-    if (dv->stream) {
-        dv->index = (dv->index + 1) % DV1394_RING_FRAMES;
-        dv->done++; dv->avail--;
-    } else {
-       dv->pts = av_gettime() & ((1LL << 48) - 1);
-    }
-
-    memcpy(pkt->data, ptr, dv->frame_size);
-    pkt->stream_index = dv->stream;
-    pkt->pts = dv->pts;
-
-    dv->stream ^= 1;
-
-    return dv->frame_size;
-}
-
 static int dv1394_read_packet(AVFormatContext *context, AVPacket *pkt)
 {
     struct dv1394_data *dv = context->priv_data;
-    int len;
+    int size;
+
+    size = dv_get_packet(dv->dv_demux, pkt);
+    if (size > 0)
+        goto out;
 
     if (!dv->avail) {
         struct dv1394_status s;
         struct pollfd p;
-        p.fd = dv->fd;
-        p.events = POLLIN | POLLERR | POLLHUP;
+
+        if (dv->done) {
+            /* Request more frames */
+            if (ioctl(dv->fd, DV1394_RECEIVE_FRAMES, dv->done) < 0) {
+                /* This usually means that ring buffer overflowed.
+                 * We have to reset :(.
+                 */
+  
+                fprintf(stderr, "DV1394: Ring buffer overflow. Reseting ..\n");
+                dv1394_reset(dv);
+                dv1394_start(dv);
+            }
+            dv->done = 0;
+        }
 
         /* Wait until more frames are available */
+restart_poll:
+        p.fd = dv->fd;
+        p.events = POLLIN | POLLERR | POLLHUP;
         if (poll(&p, 1, -1) < 0) {
+            if (errno == EAGAIN || errno == EINTR)
+                goto restart_poll;
             perror("Poll failed");
             return -EIO;
         }
@@ -211,7 +188,7 @@ static int dv1394_read_packet(AVFormatContext *context, AVPacket *pkt)
 
         dv->avail = s.n_clear_frames;
         dv->index = s.first_clear_frame;
-        dv->done = 0;
+        dv->done  = 0;
 
         if (s.dropped_frames) {
             fprintf(stderr, "DV1394: Frame drop detected (%d). Reseting ..\n",
@@ -222,31 +199,21 @@ static int dv1394_read_packet(AVFormatContext *context, AVPacket *pkt)
         }
     }
 
-    if (av_new_packet(pkt, dv->frame_size) < 0)
-        return -EIO;
-
 #ifdef DV1394_DEBUG
     fprintf(stderr, "index %d, avail %d, done %d\n", dv->index, dv->avail,
             dv->done);
 #endif
 
-    len = __copy_frame(dv, pkt);
-
-    if (!dv->avail && dv->done) {
-        /* Request more frames */
-        if (ioctl(dv->fd, DV1394_RECEIVE_FRAMES, dv->done) < 0) {
-            /* This usually means that ring buffer overflowed.
-             * We have to reset :(.
-             */
-
-            fprintf(stderr, "DV1394: Ring buffer overflow. Reseting ..\n");
-
-            dv1394_reset(dv);
-            dv1394_start(dv);
-        }
-    }
-
-    return len;
+    size = dv_produce_packet(dv->dv_demux, pkt, 
+                             dv->ring + (dv->index * DV1394_PAL_FRAME_SIZE), 
+                            DV1394_PAL_FRAME_SIZE);
+    dv->index = (dv->index + 1) % DV1394_RING_FRAMES;
+    dv->done++; dv->avail--;
+    dv->pts = av_gettime() & ((1LL << 48) - 1);
+    
+out:
+    pkt->pts = dv->pts;
+    return size;
 }
 
 static int dv1394_close(AVFormatContext * context)
@@ -262,6 +229,7 @@ static int dv1394_close(AVFormatContext * context)
         perror("Failed to munmap DV1394 ring buffer");
 
     close(dv->fd);
+    av_free(dv->dv_demux);
 
     return 0;
 }