git.sesse.net Git - ffmpeg/blob - libavdevice/libdc1394.c

   1 /*
   2  * IIDC1394 grab interface (uses libdc1394 and libraw1394)
   3  * Copyright (c) 2004 Roman Shaposhnik
   4  * Copyright (c) 2008 Alessandro Sappia
   5  *
   6  * This file is part of Libav.
   7  *
   8  * Libav is free software; you can redistribute it and/or
   9  * modify it under the terms of the GNU Lesser General Public
  10  * License as published by the Free Software Foundation; either
  11  * version 2.1 of the License, or (at your option) any later version.
  12  *
  13  * Libav is distributed in the hope that it will be useful,
  14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  16  * Lesser General Public License for more details.
  17  *
  18  * You should have received a copy of the GNU Lesser General Public
  19  * License along with Libav; if not, write to the Free Software
  20  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  21  */
  22
  23 #include "config.h"
  24 #include "libavformat/avformat.h"
  25 #include "libavutil/log.h"
  26 #include "libavutil/opt.h"
  27 #include "libavutil/parseutils.h"
  28
  29 #if HAVE_LIBDC1394_2
  30 #include <dc1394/dc1394.h>
  31 #elif HAVE_LIBDC1394_1
  32 #include <libraw1394/raw1394.h>
  33 #include <libdc1394/dc1394_control.h>
  34
  35 #define DC1394_VIDEO_MODE_320x240_YUV422 MODE_320x240_YUV422
  36 #define DC1394_VIDEO_MODE_640x480_YUV411 MODE_640x480_YUV411
  37 #define DC1394_VIDEO_MODE_640x480_YUV422 MODE_640x480_YUV422
  38 #define DC1394_FRAMERATE_1_875 FRAMERATE_1_875
  39 #define DC1394_FRAMERATE_3_75  FRAMERATE_3_75
  40 #define DC1394_FRAMERATE_7_5   FRAMERATE_7_5
  41 #define DC1394_FRAMERATE_15    FRAMERATE_15
  42 #define DC1394_FRAMERATE_30    FRAMERATE_30
  43 #define DC1394_FRAMERATE_60    FRAMERATE_60
  44 #define DC1394_FRAMERATE_120   FRAMERATE_120
  45 #define DC1394_FRAMERATE_240   FRAMERATE_240
  46 #endif
  47
  48 #undef free
  49
  50 typedef struct dc1394_data {
  51     AVClass *class;
  52 #if HAVE_LIBDC1394_1
  53     raw1394handle_t handle;
  54     dc1394_cameracapture camera;
  55     int channel;
  56 #elif HAVE_LIBDC1394_2
  57     dc1394_t *d;
  58     dc1394camera_t *camera;
  59     dc1394video_frame_t *frame;
  60 #endif
  61     int current_frame;
  62     int fps;
  63     char *video_size;       /**< String describing video size, set by a private option. */
  64
  65     AVPacket packet;
  66 } dc1394_data;
  67
  68 struct dc1394_frame_format {
  69     int width;
  70     int height;
  71     enum PixelFormat pix_fmt;
  72     int frame_size_id;
  73 } dc1394_frame_formats[] = {
  74     { 320, 240, PIX_FMT_UYVY422, DC1394_VIDEO_MODE_320x240_YUV422 },
  75     { 640, 480, PIX_FMT_UYYVYY411, DC1394_VIDEO_MODE_640x480_YUV411 },
  76     { 640, 480, PIX_FMT_UYVY422, DC1394_VIDEO_MODE_640x480_YUV422 },
  77     { 0, 0, 0, 0 } /* gotta be the last one */
  78 };
  79
  80 struct dc1394_frame_rate {
  81     int frame_rate;
  82     int frame_rate_id;
  83 } dc1394_frame_rates[] = {
  84     {  1875, DC1394_FRAMERATE_1_875 },
  85     {  3750, DC1394_FRAMERATE_3_75  },
  86     {  7500, DC1394_FRAMERATE_7_5   },
  87     { 15000, DC1394_FRAMERATE_15    },
  88     { 30000, DC1394_FRAMERATE_30    },
  89     { 60000, DC1394_FRAMERATE_60    },
  90     {120000, DC1394_FRAMERATE_120   },
  91     {240000, DC1394_FRAMERATE_240    },
  92     { 0, 0 } /* gotta be the last one */
  93 };
  94
  95 #define OFFSET(x) offsetof(dc1394_data, x)
  96 #define DEC AV_OPT_FLAG_DECODING_PARAM
  97 static const AVOption options[] = {
  98 #if HAVE_LIBDC1394_1
  99     { "channel", "", offsetof(dc1394_data, channel), FF_OPT_TYPE_INT, {.dbl = 0}, 0, INT_MAX, AV_OPT_FLAG_DECODING_PARAM },
 100 #endif
 101     { "video_size", "A string describing frame size, such as 640x480 or hd720.", OFFSET(video_size), FF_OPT_TYPE_STRING, {.str = "qvga"}, 0, 0, DEC },
 102     { NULL },
 103 };
 104
 105 static const AVClass libdc1394_class = {
 106     .class_name = "libdc1394 indev",
 107     .item_name  = av_default_item_name,
 108     .option     = options,
 109     .version    = LIBAVUTIL_VERSION_INT,
 110 };
 111
 112
 113 static inline int dc1394_read_common(AVFormatContext *c, AVFormatParameters *ap,
 114                                      struct dc1394_frame_format **select_fmt, struct dc1394_frame_rate **select_fps)
 115 {
 116     dc1394_data* dc1394 = c->priv_data;
 117     AVStream* vst;
 118     struct dc1394_frame_format *fmt;
 119     struct dc1394_frame_rate *fps;
 120     enum PixelFormat pix_fmt = ap->pix_fmt == PIX_FMT_NONE ? PIX_FMT_UYVY422 : ap->pix_fmt; /* defaults */
 121     int width, height;
 122     int frame_rate           = !ap->time_base.num ? 30000 : av_rescale(1000, ap->time_base.den, ap->time_base.num);
 123     int ret = 0;
 124
 125     if ((ret = av_parse_video_size(&width, &height, dc1394->video_size)) < 0) {
 126         av_log(c, AV_LOG_ERROR, "Couldn't parse video size.\n");
 127         goto out;
 128     }
 129 #if FF_API_FORMAT_PARAMETERS
 130     if (ap->width > 0)
 131         width = ap->width;
 132     if (ap->height > 0)
 133         height = ap->height;
 134 #endif
 135
 136     for (fmt = dc1394_frame_formats; fmt->width; fmt++)
 137          if (fmt->pix_fmt == pix_fmt && fmt->width == width && fmt->height == height)
 138              break;
 139
 140     for (fps = dc1394_frame_rates; fps->frame_rate; fps++)
 141          if (fps->frame_rate == frame_rate)
 142              break;
 143
 144     if (!fps->frame_rate || !fmt->width) {
 145         av_log(c, AV_LOG_ERROR, "Can't find matching camera format for %s, %dx%d@%d:1000fps\n", avcodec_get_pix_fmt_name(pix_fmt),
 146                                                                                                 width, height, frame_rate);
 147         ret = AVERROR(EINVAL);
 148         goto out;
 149     }
 150
 151     /* create a video stream */
 152     vst = av_new_stream(c, 0);
 153     if (!vst) {
 154         ret = AVERROR(ENOMEM);
 155         goto out;
 156     }
 157     av_set_pts_info(vst, 64, 1, 1000);
 158     vst->codec->codec_type = AVMEDIA_TYPE_VIDEO;
 159     vst->codec->codec_id = CODEC_ID_RAWVIDEO;
 160     vst->codec->time_base.den = fps->frame_rate;
 161     vst->codec->time_base.num = 1000;
 162     vst->codec->width = fmt->width;
 163     vst->codec->height = fmt->height;
 164     vst->codec->pix_fmt = fmt->pix_fmt;
 165
 166     /* packet init */
 167     av_init_packet(&dc1394->packet);
 168     dc1394->packet.size = avpicture_get_size(fmt->pix_fmt, fmt->width, fmt->height);
 169     dc1394->packet.stream_index = vst->index;
 170     dc1394->packet.flags |= AV_PKT_FLAG_KEY;
 171
 172     dc1394->current_frame = 0;
 173     dc1394->fps = fps->frame_rate;
 174
 175     vst->codec->bit_rate = av_rescale(dc1394->packet.size * 8, fps->frame_rate, 1000);
 176     *select_fps = fps;
 177     *select_fmt = fmt;
 178 out:
 179     av_freep(&dc1394->video_size);
 180     return ret;
 181 }
 182
 183 #if HAVE_LIBDC1394_1
 184 static int dc1394_v1_read_header(AVFormatContext *c, AVFormatParameters * ap)
 185 {
 186     dc1394_data* dc1394 = c->priv_data;
 187     AVStream* vst;
 188     nodeid_t* camera_nodes;
 189     int res;
 190     struct dc1394_frame_format *fmt = NULL;
 191     struct dc1394_frame_rate *fps = NULL;
 192
 193     if (dc1394_read_common(c,ap,&fmt,&fps) != 0)
 194         return -1;
 195
 196 #if FF_API_FORMAT_PARAMETERS
 197     if (ap->channel)
 198         dc1394->channel = ap->channel;
 199 #endif
 200
 201     /* Now let us prep the hardware. */
 202     dc1394->handle = dc1394_create_handle(0); /* FIXME: gotta have ap->port */
 203     if (!dc1394->handle) {
 204         av_log(c, AV_LOG_ERROR, "Can't acquire dc1394 handle on port %d\n", 0 /* ap->port */);
 205         goto out;
 206     }
 207     camera_nodes = dc1394_get_camera_nodes(dc1394->handle, &res, 1);
 208     if (!camera_nodes || camera_nodes[dc1394->channel] == DC1394_NO_CAMERA) {
 209         av_log(c, AV_LOG_ERROR, "There's no IIDC camera on the channel %d\n", dc1394->channel);
 210         goto out_handle;
 211     }
 212     res = dc1394_dma_setup_capture(dc1394->handle, camera_nodes[dc1394->channel],
 213                                    0,
 214                                    FORMAT_VGA_NONCOMPRESSED,
 215                                    fmt->frame_size_id,
 216                                    SPEED_400,
 217                                    fps->frame_rate_id, 8, 1,
 218                                    c->filename,
 219                                    &dc1394->camera);
 220     dc1394_free_camera_nodes(camera_nodes);
 221     if (res != DC1394_SUCCESS) {
 222         av_log(c, AV_LOG_ERROR, "Can't prepare camera for the DMA capture\n");
 223         goto out_handle;
 224     }
 225
 226     res = dc1394_start_iso_transmission(dc1394->handle, dc1394->camera.node);
 227     if (res != DC1394_SUCCESS) {
 228         av_log(c, AV_LOG_ERROR, "Can't start isochronous transmission\n");
 229         goto out_handle_dma;
 230     }
 231
 232     return 0;
 233
 234 out_handle_dma:
 235     dc1394_dma_unlisten(dc1394->handle, &dc1394->camera);
 236     dc1394_dma_release_camera(dc1394->handle, &dc1394->camera);
 237 out_handle:
 238     dc1394_destroy_handle(dc1394->handle);
 239 out:
 240     return -1;
 241 }
 242
 243 static int dc1394_v1_read_packet(AVFormatContext *c, AVPacket *pkt)
 244 {
 245     struct dc1394_data *dc1394 = c->priv_data;
 246     int res;
 247
 248     /* discard stale frame */
 249     if (dc1394->current_frame++) {
 250         if (dc1394_dma_done_with_buffer(&dc1394->camera) != DC1394_SUCCESS)
 251             av_log(c, AV_LOG_ERROR, "failed to release %d frame\n", dc1394->current_frame);
 252     }
 253
 254     res = dc1394_dma_single_capture(&dc1394->camera);
 255
 256     if (res == DC1394_SUCCESS) {
 257         dc1394->packet.data = (uint8_t *)(dc1394->camera.capture_buffer);
 258         dc1394->packet.pts = (dc1394->current_frame * 1000000) / dc1394->fps;
 259         res = dc1394->packet.size;
 260     } else {
 261         av_log(c, AV_LOG_ERROR, "DMA capture failed\n");
 262         dc1394->packet.data = NULL;
 263         res = -1;
 264     }
 265
 266     *pkt = dc1394->packet;
 267     return res;
 268 }
 269
 270 static int dc1394_v1_close(AVFormatContext * context)
 271 {
 272     struct dc1394_data *dc1394 = context->priv_data;
 273
 274     dc1394_stop_iso_transmission(dc1394->handle, dc1394->camera.node);
 275     dc1394_dma_unlisten(dc1394->handle, &dc1394->camera);
 276     dc1394_dma_release_camera(dc1394->handle, &dc1394->camera);
 277     dc1394_destroy_handle(dc1394->handle);
 278
 279     return 0;
 280 }
 281
 282 #elif HAVE_LIBDC1394_2
 283 static int dc1394_v2_read_header(AVFormatContext *c, AVFormatParameters * ap)
 284 {
 285     dc1394_data* dc1394 = c->priv_data;
 286     dc1394camera_list_t *list;
 287     int res, i;
 288     struct dc1394_frame_format *fmt = NULL;
 289     struct dc1394_frame_rate *fps = NULL;
 290
 291     if (dc1394_read_common(c,ap,&fmt,&fps) != 0)
 292        return -1;
 293
 294     /* Now let us prep the hardware. */
 295     dc1394->d = dc1394_new();
 296     dc1394_camera_enumerate (dc1394->d, &list);
 297     if ( !list || list->num == 0) {
 298         av_log(c, AV_LOG_ERROR, "Unable to look for an IIDC camera\n\n");
 299         goto out;
 300     }
 301
 302     /* FIXME: To select a specific camera I need to search in list its guid */
 303     dc1394->camera = dc1394_camera_new (dc1394->d, list->ids[0].guid);
 304     if (list->num > 1) {
 305         av_log(c, AV_LOG_INFO, "Working with the first camera found\n");
 306     }
 307
 308     /* Freeing list of cameras */
 309     dc1394_camera_free_list (list);
 310
 311     /* Select MAX Speed possible from the cam */
 312     if (dc1394->camera->bmode_capable>0) {
 313        dc1394_video_set_operation_mode(dc1394->camera, DC1394_OPERATION_MODE_1394B);
 314        i = DC1394_ISO_SPEED_800;
 315     } else {
 316        i = DC1394_ISO_SPEED_400;
 317     }
 318
 319     for (res = DC1394_FAILURE; i >= DC1394_ISO_SPEED_MIN && res != DC1394_SUCCESS; i--) {
 320             res=dc1394_video_set_iso_speed(dc1394->camera, i);
 321     }
 322     if (res != DC1394_SUCCESS) {
 323         av_log(c, AV_LOG_ERROR, "Couldn't set ISO Speed\n");
 324         goto out_camera;
 325     }
 326
 327     if (dc1394_video_set_mode(dc1394->camera, fmt->frame_size_id) != DC1394_SUCCESS) {
 328         av_log(c, AV_LOG_ERROR, "Couldn't set video format\n");
 329         goto out_camera;
 330     }
 331
 332     if (dc1394_video_set_framerate(dc1394->camera,fps->frame_rate_id) != DC1394_SUCCESS) {
 333         av_log(c, AV_LOG_ERROR, "Couldn't set framerate %d \n",fps->frame_rate);
 334         goto out_camera;
 335     }
 336     if (dc1394_capture_setup(dc1394->camera, 10, DC1394_CAPTURE_FLAGS_DEFAULT)!=DC1394_SUCCESS) {
 337         av_log(c, AV_LOG_ERROR, "Cannot setup camera \n");
 338         goto out_camera;
 339     }
 340
 341     if (dc1394_video_set_transmission(dc1394->camera, DC1394_ON) !=DC1394_SUCCESS) {
 342         av_log(c, AV_LOG_ERROR, "Cannot start capture\n");
 343         goto out_camera;
 344     }
 345     return 0;
 346
 347 out_camera:
 348     dc1394_capture_stop(dc1394->camera);
 349     dc1394_video_set_transmission(dc1394->camera, DC1394_OFF);
 350     dc1394_camera_free (dc1394->camera);
 351 out:
 352     dc1394_free(dc1394->d);
 353     return -1;
 354 }
 355
 356 static int dc1394_v2_read_packet(AVFormatContext *c, AVPacket *pkt)
 357 {
 358     struct dc1394_data *dc1394 = c->priv_data;
 359     int res;
 360
 361     /* discard stale frame */
 362     if (dc1394->current_frame++) {
 363         if (dc1394_capture_enqueue(dc1394->camera, dc1394->frame) != DC1394_SUCCESS)
 364             av_log(c, AV_LOG_ERROR, "failed to release %d frame\n", dc1394->current_frame);
 365     }
 366
 367     res = dc1394_capture_dequeue(dc1394->camera, DC1394_CAPTURE_POLICY_WAIT, &dc1394->frame);
 368     if (res == DC1394_SUCCESS) {
 369         dc1394->packet.data = (uint8_t *)(dc1394->frame->image);
 370         dc1394->packet.pts = (dc1394->current_frame  * 1000000) / (dc1394->fps);
 371         res = dc1394->frame->image_bytes;
 372     } else {
 373         av_log(c, AV_LOG_ERROR, "DMA capture failed\n");
 374         dc1394->packet.data = NULL;
 375         res = -1;
 376     }
 377
 378     *pkt = dc1394->packet;
 379     return res;
 380 }
 381
 382 static int dc1394_v2_close(AVFormatContext * context)
 383 {
 384     struct dc1394_data *dc1394 = context->priv_data;
 385
 386     dc1394_video_set_transmission(dc1394->camera, DC1394_OFF);
 387     dc1394_capture_stop(dc1394->camera);
 388     dc1394_camera_free(dc1394->camera);
 389     dc1394_free(dc1394->d);
 390
 391     return 0;
 392 }
 393
 394 AVInputFormat ff_libdc1394_demuxer = {
 395     .name           = "libdc1394",
 396     .long_name      = NULL_IF_CONFIG_SMALL("dc1394 v.2 A/V grab"),
 397     .priv_data_size = sizeof(struct dc1394_data),
 398     .read_header    = dc1394_v2_read_header,
 399     .read_packet    = dc1394_v2_read_packet,
 400     .read_close     = dc1394_v2_close,
 401     .flags          = AVFMT_NOFILE,
 402     .priv_class     = &libdc1394_class,
 403 };
 404
 405 #endif
 406 #if HAVE_LIBDC1394_1
 407 AVInputFormat ff_libdc1394_demuxer = {
 408     .name           = "libdc1394",
 409     .long_name      = NULL_IF_CONFIG_SMALL("dc1394 v.1 A/V grab"),
 410     .priv_data_size = sizeof(struct dc1394_data),
 411     .read_header    = dc1394_v1_read_header,
 412     .read_packet    = dc1394_v1_read_packet,
 413     .read_close     = dc1394_v1_close,
 414     .flags          = AVFMT_NOFILE,
 415     .priv_class     = &libdc1394_class,
 416 };
 417 #endif