Mercurial > libavformat.hg
annotate dc1394.c @ 1282:faf64f0d3445 libavformat
AVISynth support, patch by Steve Lhomme % slhomme A divxcorp P com %
Original Thread:
Date: Aug 16, 2006 11:45 PM
Subject: [Ffmpeg-devel] [PATCH] AVISynth support
author | gpoirier |
---|---|
date | Tue, 29 Aug 2006 07:34:58 +0000 |
parents | d18cc9a1fd02 |
children | 0899bfe4105c |
rev | line source |
---|---|
571 | 1 /* |
2 * IIDC1394 grab interface (uses libdc1394 and libraw1394) | |
3 * Copyright (c) 2004 Roman Shaposhnik | |
4 * | |
5 * This library is free software; you can redistribute it and/or | |
6 * modify it under the terms of the GNU Lesser General Public | |
7 * License as published by the Free Software Foundation; either | |
8 * version 2 of the License, or (at your option) any later version. | |
9 * | |
10 * This library is distributed in the hope that it will be useful, | |
11 * but WITHOUT ANY WARRANTY; without even the implied warranty of | |
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | |
13 * Lesser General Public License for more details. | |
14 * | |
15 * You should have received a copy of the GNU Lesser General Public | |
16 * License along with this library; if not, write to the Free Software | |
896
edbe5c3717f9
Update licensing information: The FSF changed postal address.
diego
parents:
887
diff
changeset
|
17 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
571 | 18 */ |
19 | |
20 #include "avformat.h" | |
21 | |
22 #include <libraw1394/raw1394.h> | |
23 #include <libdc1394/dc1394_control.h> | |
24 | |
25 #undef free | |
26 | |
27 typedef struct dc1394_data { | |
28 raw1394handle_t handle; | |
29 dc1394_cameracapture camera; | |
30 int current_frame; | |
31 int fps; | |
32 | |
33 AVPacket packet; | |
34 } dc1394_data; | |
35 | |
36 struct dc1394_frame_format { | |
37 int width; | |
38 int height; | |
39 enum PixelFormat pix_fmt; | |
40 int frame_size_id; | |
41 } dc1394_frame_formats[] = { | |
42 { 320, 240, PIX_FMT_UYVY422, MODE_320x240_YUV422 }, | |
43 { 640, 480, PIX_FMT_UYVY411, MODE_640x480_YUV411 }, | |
44 { 640, 480, PIX_FMT_UYVY422, MODE_640x480_YUV422 }, | |
45 { 0, 0, 0, MODE_320x240_YUV422 } /* default -- gotta be the last one */ | |
46 }; | |
47 | |
48 struct dc1394_frame_rate { | |
49 int frame_rate; | |
50 int frame_rate_id; | |
51 } dc1394_frame_rates[] = { | |
52 { 1875, FRAMERATE_1_875 }, | |
53 { 3750, FRAMERATE_3_75 }, | |
54 { 7500, FRAMERATE_7_5 }, | |
55 { 15000, FRAMERATE_15 }, | |
56 { 30000, FRAMERATE_30 }, | |
57 { 60000, FRAMERATE_60 }, | |
58 { 0, FRAMERATE_30 } /* default -- gotta be the last one */ | |
59 }; | |
60 | |
61 static int dc1394_read_header(AVFormatContext *c, AVFormatParameters * ap) | |
62 { | |
63 dc1394_data* dc1394 = c->priv_data; | |
64 AVStream* vst; | |
65 nodeid_t* camera_nodes; | |
66 int res; | |
67 struct dc1394_frame_format *fmt; | |
68 struct dc1394_frame_rate *fps; | |
69 | |
70 for (fmt = dc1394_frame_formats; fmt->width; fmt++) | |
71 if (fmt->pix_fmt == ap->pix_fmt && fmt->width == ap->width && fmt->height == ap->height) | |
887 | 72 break; |
885 | 73 |
571 | 74 for (fps = dc1394_frame_rates; fps->frame_rate; fps++) |
743 | 75 if (fps->frame_rate == av_rescale(1000, ap->time_base.den, ap->time_base.num)) |
887 | 76 break; |
885 | 77 |
571 | 78 /* create a video stream */ |
79 vst = av_new_stream(c, 0); | |
80 if (!vst) | |
887 | 81 return -1; |
571 | 82 av_set_pts_info(vst, 64, 1, 1000); |
833 | 83 vst->codec->codec_type = CODEC_TYPE_VIDEO; |
84 vst->codec->codec_id = CODEC_ID_RAWVIDEO; | |
85 vst->codec->time_base.den = fps->frame_rate; | |
86 vst->codec->time_base.num = 1000; | |
87 vst->codec->width = fmt->width; | |
88 vst->codec->height = fmt->height; | |
89 vst->codec->pix_fmt = fmt->pix_fmt; | |
571 | 90 |
91 /* packet init */ | |
92 av_init_packet(&dc1394->packet); | |
93 dc1394->packet.size = avpicture_get_size(fmt->pix_fmt, fmt->width, fmt->height); | |
94 dc1394->packet.stream_index = vst->index; | |
95 dc1394->packet.flags |= PKT_FLAG_KEY; | |
885 | 96 |
571 | 97 dc1394->current_frame = 0; |
98 dc1394->fps = fps->frame_rate; | |
99 | |
833 | 100 vst->codec->bit_rate = av_rescale(dc1394->packet.size * 8, fps->frame_rate, 1000); |
885 | 101 |
571 | 102 /* Now lets prep the hardware */ |
103 dc1394->handle = dc1394_create_handle(0); /* FIXME: gotta have ap->port */ | |
104 if (!dc1394->handle) { | |
887 | 105 av_log(c, AV_LOG_ERROR, "Can't acquire dc1394 handle on port %d\n", 0 /* ap->port */); |
571 | 106 goto out; |
107 } | |
108 camera_nodes = dc1394_get_camera_nodes(dc1394->handle, &res, 1); | |
109 if (!camera_nodes || camera_nodes[ap->channel] == DC1394_NO_CAMERA) { | |
110 av_log(c, AV_LOG_ERROR, "There's no IIDC camera on the channel %d\n", ap->channel); | |
885 | 111 goto out_handle; |
571 | 112 } |
885 | 113 res = dc1394_dma_setup_capture(dc1394->handle, camera_nodes[ap->channel], |
887 | 114 0, |
115 FORMAT_VGA_NONCOMPRESSED, | |
116 fmt->frame_size_id, | |
117 SPEED_400, | |
118 fps->frame_rate_id, 8, 1, | |
119 ap->device, | |
120 &dc1394->camera); | |
571 | 121 dc1394_free_camera_nodes(camera_nodes); |
122 if (res != DC1394_SUCCESS) { | |
123 av_log(c, AV_LOG_ERROR, "Can't prepare camera for the DMA capture\n"); | |
887 | 124 goto out_handle; |
571 | 125 } |
126 | |
127 res = dc1394_start_iso_transmission(dc1394->handle, dc1394->camera.node); | |
128 if (res != DC1394_SUCCESS) { | |
129 av_log(c, AV_LOG_ERROR, "Can't start isochronous transmission\n"); | |
887 | 130 goto out_handle_dma; |
571 | 131 } |
885 | 132 |
571 | 133 return 0; |
134 | |
135 out_handle_dma: | |
136 dc1394_dma_unlisten(dc1394->handle, &dc1394->camera); | |
137 dc1394_dma_release_camera(dc1394->handle, &dc1394->camera); | |
138 out_handle: | |
139 dc1394_destroy_handle(dc1394->handle); | |
140 out: | |
141 return -1; | |
142 } | |
143 | |
144 static int dc1394_read_packet(AVFormatContext *c, AVPacket *pkt) | |
145 { | |
146 struct dc1394_data *dc1394 = c->priv_data; | |
147 int res; | |
885 | 148 |
571 | 149 /* discard stale frame */ |
150 if (dc1394->current_frame++) { | |
887 | 151 if (dc1394_dma_done_with_buffer(&dc1394->camera) != DC1394_SUCCESS) |
571 | 152 av_log(c, AV_LOG_ERROR, "failed to release %d frame\n", dc1394->current_frame); |
153 } | |
885 | 154 |
571 | 155 res = dc1394_dma_single_capture(&dc1394->camera); |
156 | |
157 if (res == DC1394_SUCCESS) { | |
885 | 158 dc1394->packet.data = (uint8_t *)(dc1394->camera.capture_buffer); |
159 dc1394->packet.pts = (dc1394->current_frame * 1000000) / dc1394->fps; | |
887 | 160 res = dc1394->packet.size; |
571 | 161 } else { |
162 av_log(c, AV_LOG_ERROR, "DMA capture failed\n"); | |
163 dc1394->packet.data = NULL; | |
885 | 164 res = -1; |
571 | 165 } |
885 | 166 |
571 | 167 *pkt = dc1394->packet; |
168 return res; | |
169 } | |
170 | |
171 static int dc1394_close(AVFormatContext * context) | |
172 { | |
173 struct dc1394_data *dc1394 = context->priv_data; | |
174 | |
175 dc1394_stop_iso_transmission(dc1394->handle, dc1394->camera.node); | |
176 dc1394_dma_unlisten(dc1394->handle, &dc1394->camera); | |
177 dc1394_dma_release_camera(dc1394->handle, &dc1394->camera); | |
178 dc1394_destroy_handle(dc1394->handle); | |
885 | 179 |
571 | 180 return 0; |
181 } | |
182 | |
1169 | 183 AVInputFormat dc1394_demuxer = { |
571 | 184 .name = "dc1394", |
185 .long_name = "dc1394 A/V grab", | |
186 .priv_data_size = sizeof(struct dc1394_data), | |
187 .read_header = dc1394_read_header, | |
188 .read_packet = dc1394_read_packet, | |
189 .read_close = dc1394_close, | |
190 .flags = AVFMT_NOFILE | |
191 }; |