libdc1394.c 14 KB
Newer Older
1 2 3
/*
 * IIDC1394 grab interface (uses libdc1394 and libraw1394)
 * Copyright (c) 2004 Roman Shaposhnik
4
 * Copyright (c) 2008 Alessandro Sappia
5
 *
6 7 8
 * This file is part of FFmpeg.
 *
 * FFmpeg is free software; you can redistribute it and/or
9 10
 * modify it under the terms of the GNU Lesser General Public
 * 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
 * FFmpeg is distributed in the hope that it will be useful,
14 15 16 17 18
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
19
 * License along with FFmpeg; 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 "libavformat/internal.h"
26
#include "libavutil/log.h"
27
#include "libavutil/mathematics.h"
28
#include "libavutil/opt.h"
29
#include "libavutil/parseutils.h"
30
#include "libavutil/pixdesc.h"
31

32
#if HAVE_LIBDC1394_2
33
#include <dc1394/dc1394.h>
34
#elif HAVE_LIBDC1394_1
35 36 37
#include <libraw1394/raw1394.h>
#include <libdc1394/dc1394_control.h>

38 39 40 41 42 43 44 45 46
#define DC1394_VIDEO_MODE_320x240_YUV422 MODE_320x240_YUV422
#define DC1394_VIDEO_MODE_640x480_YUV411 MODE_640x480_YUV411
#define DC1394_VIDEO_MODE_640x480_YUV422 MODE_640x480_YUV422
#define DC1394_FRAMERATE_1_875 FRAMERATE_1_875
#define DC1394_FRAMERATE_3_75  FRAMERATE_3_75
#define DC1394_FRAMERATE_7_5   FRAMERATE_7_5
#define DC1394_FRAMERATE_15    FRAMERATE_15
#define DC1394_FRAMERATE_30    FRAMERATE_30
#define DC1394_FRAMERATE_60    FRAMERATE_60
47 48
#define DC1394_FRAMERATE_120   FRAMERATE_120
#define DC1394_FRAMERATE_240   FRAMERATE_240
49 50
#endif

51
typedef struct dc1394_data {
52
    AVClass *class;
53
#if HAVE_LIBDC1394_1
54 55
    raw1394handle_t handle;
    dc1394_cameracapture camera;
56
    int channel;
57
#elif HAVE_LIBDC1394_2
58 59 60 61
    dc1394_t *d;
    dc1394camera_t *camera;
    dc1394video_frame_t *frame;
#endif
62
    int current_frame;
63
    int  frame_rate;        /**< frames per 1000 seconds (fps * 1000) */
64
    char *video_size;       /**< String describing video size, set by a private option. */
65
    char *pixel_format;     /**< Set by a private option. */
66
    char *framerate;        /**< Set by a private option. */
67 68 69 70 71 72 73

    AVPacket packet;
} dc1394_data;

struct dc1394_frame_format {
    int width;
    int height;
74
    enum AVPixelFormat pix_fmt;
75 76
    int frame_size_id;
} dc1394_frame_formats[] = {
77 78 79
    { 320, 240, AV_PIX_FMT_UYVY422,   DC1394_VIDEO_MODE_320x240_YUV422 },
    { 640, 480, AV_PIX_FMT_UYYVYY411, DC1394_VIDEO_MODE_640x480_YUV411 },
    { 640, 480, AV_PIX_FMT_UYVY422,   DC1394_VIDEO_MODE_640x480_YUV422 },
80
    { 0, 0, 0, 0 } /* gotta be the last one */
81 82 83 84 85 86
};

struct dc1394_frame_rate {
    int frame_rate;
    int frame_rate_id;
} dc1394_frame_rates[] = {
87 88 89 90 91 92
    {  1875, DC1394_FRAMERATE_1_875 },
    {  3750, DC1394_FRAMERATE_3_75  },
    {  7500, DC1394_FRAMERATE_7_5   },
    { 15000, DC1394_FRAMERATE_15    },
    { 30000, DC1394_FRAMERATE_30    },
    { 60000, DC1394_FRAMERATE_60    },
93 94
    {120000, DC1394_FRAMERATE_120   },
    {240000, DC1394_FRAMERATE_240    },
95
    { 0, 0 } /* gotta be the last one */
96 97
};

98 99
#define OFFSET(x) offsetof(dc1394_data, x)
#define DEC AV_OPT_FLAG_DECODING_PARAM
100 101
static const AVOption options[] = {
#if HAVE_LIBDC1394_1
102
    { "channel", "", offsetof(dc1394_data, channel), AV_OPT_TYPE_INT, {.i64 = 0}, 0, INT_MAX, AV_OPT_FLAG_DECODING_PARAM },
103
#endif
104 105 106
    { "video_size", "A string describing frame size, such as 640x480 or hd720.", OFFSET(video_size), AV_OPT_TYPE_STRING, {.str = "qvga"}, 0, 0, DEC },
    { "pixel_format", "", OFFSET(pixel_format), AV_OPT_TYPE_STRING, {.str = "uyvy422"}, 0, 0, DEC },
    { "framerate", "", OFFSET(framerate), AV_OPT_TYPE_STRING, {.str = "ntsc"}, 0, 0, DEC },
107 108 109 110 111 112 113 114 115 116 117
    { NULL },
};

static const AVClass libdc1394_class = {
    .class_name = "libdc1394 indev",
    .item_name  = av_default_item_name,
    .option     = options,
    .version    = LIBAVUTIL_VERSION_INT,
};


118
static inline int dc1394_read_common(AVFormatContext *c,
119 120 121
                                     struct dc1394_frame_format **select_fmt, struct dc1394_frame_rate **select_fps)
{
    dc1394_data* dc1394 = c->priv_data;
122 123 124
    AVStream* vst;
    struct dc1394_frame_format *fmt;
    struct dc1394_frame_rate *fps;
125
    enum AVPixelFormat pix_fmt;
126
    int width, height;
127
    AVRational framerate;
128
    int ret = 0;
129

130
    if ((pix_fmt = av_get_pix_fmt(dc1394->pixel_format)) == AV_PIX_FMT_NONE) {
131 132
        av_log(c, AV_LOG_ERROR, "No such pixel format: %s.\n", dc1394->pixel_format);
        ret = AVERROR(EINVAL);
133 134 135
        goto out;
    }

136
    if ((ret = av_parse_video_size(&width, &height, dc1394->video_size)) < 0) {
137
        av_log(c, AV_LOG_ERROR, "Could not parse video size '%s'.\n", dc1394->video_size);
138
        goto out;
139
    }
140
    if ((ret = av_parse_video_rate(&framerate, dc1394->framerate)) < 0) {
141
        av_log(c, AV_LOG_ERROR, "Could not parse framerate '%s'.\n", dc1394->framerate);
142
        goto out;
143
    }
144
    dc1394->frame_rate = av_rescale(1000, framerate.num, framerate.den);
145

146
    for (fmt = dc1394_frame_formats; fmt->width; fmt++)
147
         if (fmt->pix_fmt == pix_fmt && fmt->width == width && fmt->height == height)
148
             break;
149

150
    for (fps = dc1394_frame_rates; fps->frame_rate; fps++)
151
         if (fps->frame_rate == dc1394->frame_rate)
152
             break;
153

154
    if (!fps->frame_rate || !fmt->width) {
155
        av_log(c, AV_LOG_ERROR, "Can't find matching camera format for %s, %dx%d@%d:1000fps\n", av_get_pix_fmt_name(pix_fmt),
156
                                                                                                width, height, dc1394->frame_rate);
157
        ret = AVERROR(EINVAL);
158
        goto out;
159 160 161
    }

    /* create a video stream */
162
    vst = avformat_new_stream(c, NULL);
163 164
    if (!vst) {
        ret = AVERROR(ENOMEM);
165
        goto out;
166
    }
167
    avpriv_set_pts_info(vst, 64, 1, 1000);
168
    vst->codec->codec_type = AVMEDIA_TYPE_VIDEO;
169
    vst->codec->codec_id = AV_CODEC_ID_RAWVIDEO;
170 171
    vst->codec->time_base.den = framerate.num;
    vst->codec->time_base.num = framerate.den;
Sam Hocevar's avatar
Sam Hocevar committed
172 173 174
    vst->codec->width = fmt->width;
    vst->codec->height = fmt->height;
    vst->codec->pix_fmt = fmt->pix_fmt;
175 176 177

    /* packet init */
    av_init_packet(&dc1394->packet);
178
    dc1394->packet.size = avpicture_get_size(fmt->pix_fmt, fmt->width, fmt->height);
179 180 181 182 183
    dc1394->packet.stream_index = vst->index;
    dc1394->packet.flags |= AV_PKT_FLAG_KEY;

    dc1394->current_frame = 0;

Sam Hocevar's avatar
Sam Hocevar committed
184
    vst->codec->bit_rate = av_rescale(dc1394->packet.size * 8, fps->frame_rate, 1000);
185 186 187
    *select_fps = fps;
    *select_fmt = fmt;
out:
188
    return ret;
189 190
}

191
#if HAVE_LIBDC1394_1
192
static int dc1394_v1_read_header(AVFormatContext *c)
193 194 195 196 197 198 199 200
{
    dc1394_data* dc1394 = c->priv_data;
    AVStream* vst;
    nodeid_t* camera_nodes;
    int res;
    struct dc1394_frame_format *fmt = NULL;
    struct dc1394_frame_rate *fps = NULL;

201
    if (dc1394_read_common(c, &fmt, &fps) != 0)
202
        return -1;
203

204
    /* Now let us prep the hardware. */
205 206
    dc1394->handle = dc1394_create_handle(0); /* FIXME: gotta have ap->port */
    if (!dc1394->handle) {
207
        av_log(c, AV_LOG_ERROR, "Can't acquire dc1394 handle on port %d\n", 0 /* ap->port */);
208 209 210
        goto out;
    }
    camera_nodes = dc1394_get_camera_nodes(dc1394->handle, &res, 1);
211 212
    if (!camera_nodes || camera_nodes[dc1394->channel] == DC1394_NO_CAMERA) {
        av_log(c, AV_LOG_ERROR, "There's no IIDC camera on the channel %d\n", dc1394->channel);
213
        goto out_handle;
214
    }
215
    res = dc1394_dma_setup_capture(dc1394->handle, camera_nodes[dc1394->channel],
216 217 218 219 220
                                   0,
                                   FORMAT_VGA_NONCOMPRESSED,
                                   fmt->frame_size_id,
                                   SPEED_400,
                                   fps->frame_rate_id, 8, 1,
221
                                   c->filename,
222
                                   &dc1394->camera);
223 224 225
    dc1394_free_camera_nodes(camera_nodes);
    if (res != DC1394_SUCCESS) {
        av_log(c, AV_LOG_ERROR, "Can't prepare camera for the DMA capture\n");
226
        goto out_handle;
227 228 229 230 231
    }

    res = dc1394_start_iso_transmission(dc1394->handle, dc1394->camera.node);
    if (res != DC1394_SUCCESS) {
        av_log(c, AV_LOG_ERROR, "Can't start isochronous transmission\n");
232
        goto out_handle_dma;
233
    }
234

235 236 237 238 239 240 241 242 243 244 245
    return 0;

out_handle_dma:
    dc1394_dma_unlisten(dc1394->handle, &dc1394->camera);
    dc1394_dma_release_camera(dc1394->handle, &dc1394->camera);
out_handle:
    dc1394_destroy_handle(dc1394->handle);
out:
    return -1;
}

246
static int dc1394_v1_read_packet(AVFormatContext *c, AVPacket *pkt)
247 248 249
{
    struct dc1394_data *dc1394 = c->priv_data;
    int res;
250

251 252
    /* discard stale frame */
    if (dc1394->current_frame++) {
253
        if (dc1394_dma_done_with_buffer(&dc1394->camera) != DC1394_SUCCESS)
254 255
            av_log(c, AV_LOG_ERROR, "failed to release %d frame\n", dc1394->current_frame);
    }
256

257 258 259
    res = dc1394_dma_single_capture(&dc1394->camera);

    if (res == DC1394_SUCCESS) {
260
        dc1394->packet.data = (uint8_t *)(dc1394->camera.capture_buffer);
261
        dc1394->packet.pts = (dc1394->current_frame * 1000000) / dc1394->frame_rate;
262
        res = dc1394->packet.size;
263 264 265
    } else {
        av_log(c, AV_LOG_ERROR, "DMA capture failed\n");
        dc1394->packet.data = NULL;
266
        res = -1;
267
    }
268

269 270 271 272
    *pkt = dc1394->packet;
    return res;
}

273
static int dc1394_v1_close(AVFormatContext * context)
274 275 276 277 278 279 280
{
    struct dc1394_data *dc1394 = context->priv_data;

    dc1394_stop_iso_transmission(dc1394->handle, dc1394->camera.node);
    dc1394_dma_unlisten(dc1394->handle, &dc1394->camera);
    dc1394_dma_release_camera(dc1394->handle, &dc1394->camera);
    dc1394_destroy_handle(dc1394->handle);
281

282 283 284
    return 0;
}

285
#elif HAVE_LIBDC1394_2
286
static int dc1394_v2_read_header(AVFormatContext *c)
287 288 289 290 291 292 293
{
    dc1394_data* dc1394 = c->priv_data;
    dc1394camera_list_t *list;
    int res, i;
    struct dc1394_frame_format *fmt = NULL;
    struct dc1394_frame_rate *fps = NULL;

294
    if (dc1394_read_common(c, &fmt, &fps) != 0)
295 296
       return -1;

297
    /* Now let us prep the hardware. */
298 299 300 301 302 303 304 305 306 307 308 309 310 311 312
    dc1394->d = dc1394_new();
    dc1394_camera_enumerate (dc1394->d, &list);
    if ( !list || list->num == 0) {
        av_log(c, AV_LOG_ERROR, "Unable to look for an IIDC camera\n\n");
        goto out;
    }

    /* FIXME: To select a specific camera I need to search in list its guid */
    dc1394->camera = dc1394_camera_new (dc1394->d, list->ids[0].guid);
    if (list->num > 1) {
        av_log(c, AV_LOG_INFO, "Working with the first camera found\n");
    }

    /* Freeing list of cameras */
    dc1394_camera_free_list (list);
313

314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344 345 346 347 348 349 350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 368 369 370 371
    /* Select MAX Speed possible from the cam */
    if (dc1394->camera->bmode_capable>0) {
       dc1394_video_set_operation_mode(dc1394->camera, DC1394_OPERATION_MODE_1394B);
       i = DC1394_ISO_SPEED_800;
    } else {
       i = DC1394_ISO_SPEED_400;
    }

    for (res = DC1394_FAILURE; i >= DC1394_ISO_SPEED_MIN && res != DC1394_SUCCESS; i--) {
            res=dc1394_video_set_iso_speed(dc1394->camera, i);
    }
    if (res != DC1394_SUCCESS) {
        av_log(c, AV_LOG_ERROR, "Couldn't set ISO Speed\n");
        goto out_camera;
    }

    if (dc1394_video_set_mode(dc1394->camera, fmt->frame_size_id) != DC1394_SUCCESS) {
        av_log(c, AV_LOG_ERROR, "Couldn't set video format\n");
        goto out_camera;
    }

    if (dc1394_video_set_framerate(dc1394->camera,fps->frame_rate_id) != DC1394_SUCCESS) {
        av_log(c, AV_LOG_ERROR, "Couldn't set framerate %d \n",fps->frame_rate);
        goto out_camera;
    }
    if (dc1394_capture_setup(dc1394->camera, 10, DC1394_CAPTURE_FLAGS_DEFAULT)!=DC1394_SUCCESS) {
        av_log(c, AV_LOG_ERROR, "Cannot setup camera \n");
        goto out_camera;
    }

    if (dc1394_video_set_transmission(dc1394->camera, DC1394_ON) !=DC1394_SUCCESS) {
        av_log(c, AV_LOG_ERROR, "Cannot start capture\n");
        goto out_camera;
    }
    return 0;

out_camera:
    dc1394_capture_stop(dc1394->camera);
    dc1394_video_set_transmission(dc1394->camera, DC1394_OFF);
    dc1394_camera_free (dc1394->camera);
out:
    dc1394_free(dc1394->d);
    return -1;
}

static int dc1394_v2_read_packet(AVFormatContext *c, AVPacket *pkt)
{
    struct dc1394_data *dc1394 = c->priv_data;
    int res;

    /* discard stale frame */
    if (dc1394->current_frame++) {
        if (dc1394_capture_enqueue(dc1394->camera, dc1394->frame) != DC1394_SUCCESS)
            av_log(c, AV_LOG_ERROR, "failed to release %d frame\n", dc1394->current_frame);
    }

    res = dc1394_capture_dequeue(dc1394->camera, DC1394_CAPTURE_POLICY_WAIT, &dc1394->frame);
    if (res == DC1394_SUCCESS) {
372 373
        dc1394->packet.data = (uint8_t *) dc1394->frame->image;
        dc1394->packet.pts  = dc1394->current_frame * 1000000 / dc1394->frame_rate;
374 375 376 377 378 379 380 381 382 383 384 385 386 387 388 389 390 391 392 393 394 395 396
        res = dc1394->frame->image_bytes;
    } else {
        av_log(c, AV_LOG_ERROR, "DMA capture failed\n");
        dc1394->packet.data = NULL;
        res = -1;
    }

    *pkt = dc1394->packet;
    return res;
}

static int dc1394_v2_close(AVFormatContext * context)
{
    struct dc1394_data *dc1394 = context->priv_data;

    dc1394_video_set_transmission(dc1394->camera, DC1394_OFF);
    dc1394_capture_stop(dc1394->camera);
    dc1394_camera_free(dc1394->camera);
    dc1394_free(dc1394->d);

    return 0;
}

397
AVInputFormat ff_libdc1394_demuxer = {
398
    .name           = "libdc1394",
399
    .long_name      = NULL_IF_CONFIG_SMALL("dc1394 v.2 A/V grab"),
400 401 402 403
    .priv_data_size = sizeof(struct dc1394_data),
    .read_header    = dc1394_v2_read_header,
    .read_packet    = dc1394_v2_read_packet,
    .read_close     = dc1394_v2_close,
404
    .flags          = AVFMT_NOFILE,
405
    .priv_class     = &libdc1394_class,
406 407 408
};

#endif
409
#if HAVE_LIBDC1394_1
410
AVInputFormat ff_libdc1394_demuxer = {
411
    .name           = "libdc1394",
412
    .long_name      = NULL_IF_CONFIG_SMALL("dc1394 v.1 A/V grab"),
413 414 415 416
    .priv_data_size = sizeof(struct dc1394_data),
    .read_header    = dc1394_v1_read_header,
    .read_packet    = dc1394_v1_read_packet,
    .read_close     = dc1394_v1_close,
417
    .flags          = AVFMT_NOFILE,
418
    .priv_class     = &libdc1394_class,
419 420
};
#endif