libdc1394.c 14.1 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
 * This file is part of Libav.
7
 *
8
 * Libav 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
 * Libav 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 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

25
#if HAVE_LIBDC1394_2
26
#include <dc1394/dc1394.h>
27
#elif HAVE_LIBDC1394_1
28 29 30
#include <libraw1394/raw1394.h>
#include <libdc1394/dc1394_control.h>

31 32 33 34 35 36 37 38 39
#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
40 41
#define DC1394_FRAMERATE_120   FRAMERATE_120
#define DC1394_FRAMERATE_240   FRAMERATE_240
42 43
#endif

44 45 46 47 48 49 50 51 52 53
#include "libavutil/internal.h"
#include "libavutil/log.h"
#include "libavutil/mathematics.h"
#include "libavutil/opt.h"
#include "libavutil/parseutils.h"
#include "libavutil/pixdesc.h"

#include "libavformat/avformat.h"
#include "libavformat/internal.h"

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

    AVPacket packet;
} dc1394_data;

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

struct dc1394_frame_rate {
    int frame_rate;
    int frame_rate_id;
} dc1394_frame_rates[] = {
91 92 93 94 95 96
    {  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    },
97 98
    {120000, DC1394_FRAMERATE_120   },
    {240000, DC1394_FRAMERATE_240    },
99
    { 0, 0 } /* gotta be the last one */
100 101
};

102 103
#define OFFSET(x) offsetof(dc1394_data, x)
#define DEC AV_OPT_FLAG_DECODING_PARAM
104 105
static const AVOption options[] = {
#if HAVE_LIBDC1394_1
106
    { "channel", "", offsetof(dc1394_data, channel), AV_OPT_TYPE_INT, {.i64 = 0}, 0, INT_MAX, AV_OPT_FLAG_DECODING_PARAM },
107
#endif
108 109 110
    { "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 },
111 112 113 114 115 116 117 118 119 120 121
    { NULL },
};

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


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

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

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

150
    for (fmt = dc1394_frame_formats; fmt->width; fmt++)
151
         if (fmt->pix_fmt == pix_fmt && fmt->width == width && fmt->height == height)
152
             break;
153

154
    for (fps = dc1394_frame_rates; fps->frame_rate; fps++)
155
         if (fps->frame_rate == dc1394->frame_rate)
156
             break;
157

158
    if (!fps->frame_rate || !fmt->width) {
159
        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),
160
                                                                                                width, height, dc1394->frame_rate);
161
        ret = AVERROR(EINVAL);
162 163 164
        goto out;
    }

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

    /* packet init */
    av_init_packet(&dc1394->packet);
    dc1394->packet.size = avpicture_get_size(fmt->pix_fmt, fmt->width, fmt->height);
    dc1394->packet.stream_index = vst->index;
184
    dc1394->packet.flags |= AV_PKT_FLAG_KEY;
185

186 187
    dc1394->current_frame = 0;

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

195
#if HAVE_LIBDC1394_1
196
static int dc1394_v1_read_header(AVFormatContext *c)
197 198 199 200 201 202 203 204
{
    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;

205
    if (dc1394_read_common(c, &fmt, &fps) != 0)
206
        return -1;
207

208
    /* Now let us prep the hardware. */
209 210
    dc1394->handle = dc1394_create_handle(0); /* FIXME: gotta have ap->port */
    if (!dc1394->handle) {
211
        av_log(c, AV_LOG_ERROR, "Can't acquire dc1394 handle on port %d\n", 0 /* ap->port */);
212 213 214
        goto out;
    }
    camera_nodes = dc1394_get_camera_nodes(dc1394->handle, &res, 1);
215 216
    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);
217
        goto out_handle;
218
    }
219
    res = dc1394_dma_setup_capture(dc1394->handle, camera_nodes[dc1394->channel],
220 221 222 223 224
                                   0,
                                   FORMAT_VGA_NONCOMPRESSED,
                                   fmt->frame_size_id,
                                   SPEED_400,
                                   fps->frame_rate_id, 8, 1,
225
                                   c->filename,
226
                                   &dc1394->camera);
227 228 229
    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");
230
        goto out_handle;
231 232 233 234 235
    }

    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");
236
        goto out_handle_dma;
237
    }
238

239 240 241 242 243 244 245 246 247 248 249
    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;
}

250
static int dc1394_v1_read_packet(AVFormatContext *c, AVPacket *pkt)
251 252 253
{
    struct dc1394_data *dc1394 = c->priv_data;
    int res;
254

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

261 262 263
    res = dc1394_dma_single_capture(&dc1394->camera);

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

273 274 275 276
    *pkt = dc1394->packet;
    return res;
}

277
static int dc1394_v1_close(AVFormatContext * context)
278 279 280 281 282 283 284
{
    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);
285

286 287 288
    return 0;
}

289
#elif HAVE_LIBDC1394_2
290
static int dc1394_v2_read_header(AVFormatContext *c)
291 292 293 294 295 296 297
{
    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;

298
    if (dc1394_read_common(c, &fmt, &fps) != 0)
299 300
       return -1;

301
    /* Now let us prep the hardware. */
302 303 304 305 306 307 308 309 310 311 312 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 372 373 374 375
    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);

    /* 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) {
376 377
        dc1394->packet.data = (uint8_t *) dc1394->frame->image;
        dc1394->packet.pts  = dc1394->current_frame * 1000000 / dc1394->frame_rate;
378 379 380 381 382 383 384 385 386 387 388 389 390 391 392 393 394 395 396 397 398 399 400
        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;
}

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

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