4 * Copyright (C) 2008 Adam Williams <broadcast at earthling dot net>
6 * This program is free software; you can redistribute it and/or modify
7 * it under the terms of the GNU General Public License as published by
8 * the Free Software Foundation; either version 2 of the License, or
9 * (at your option) any later version.
11 * This program is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 * GNU General Public License for more details.
16 * You should have received a copy of the GNU General Public License
17 * along with this program; if not, write to the Free Software
18 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
24 #include "condition.h"
25 #include "device1394input.h"
26 #include "ieee1394-ioctl.h"
29 #include "video1394.h"
33 #include <sys/ioctl.h>
39 #define INPUT_SAMPLES 131072
40 #define BUFFER_TIMEOUT 500000
43 Device1394Input::Device1394Input()
52 current_outbuffer = 0;
63 Device1394Input::~Device1394Input()
65 // Driver crashes if it isn't stopped before cancelling the thread.
66 // May still crash during the cancel though.
77 for(int i = 0; i < total_buffers; i++)
80 delete [] buffer_valid;
84 munmap(input_buffer, total_buffers * buffer_size);
88 delete [] audio_buffer;
96 if(video_lock) delete video_lock;
97 if(audio_lock) delete audio_lock;
98 if(buffer_lock) delete buffer_lock;
105 int Device1394Input::open(const char *path,
116 this->channel = channel;
117 this->length = length;
118 this->channels = channels;
119 this->samplerate = samplerate;
124 buffer_size = is_pal ? DV_PAL_SIZE : DV_NTSC_SIZE;
125 total_buffers = length;
128 // Initialize grabbing
131 if((fd = ::open(path, O_RDWR)) < 0)
133 printf("Device1394Input::open %s: %s\n", path, strerror(errno));
137 #define CIP_N_NTSC 68000000
138 #define CIP_D_NTSC 1068000000
143 struct dv1394_init init =
145 api_version: DV1394_API_VERSION,
146 channel: (unsigned int)channel,
147 n_frames: (unsigned int)length,
148 format: is_pal ? DV1394_PAL: DV1394_NTSC,
153 if(ioctl(fd, DV1394_IOC_INIT, &init) < 0)
155 printf("Device1394Input::open DV1394_IOC_INIT: %s\n", strerror(errno));
158 input_buffer = (unsigned char*)mmap(0,
159 length * buffer_size,
160 PROT_READ | PROT_WRITE,
165 if(ioctl(fd, DV1394_IOC_START_RECEIVE, 0) < 0)
167 perror("Device1394Input::open DV1394_START_RECEIVE");
173 buffer = new char*[total_buffers];
174 buffer_valid = new int[total_buffers];
175 bzero(buffer_valid, sizeof(int) * total_buffers);
176 for(int i = 0; i < total_buffers; i++)
178 buffer[i] = new char[DV_PAL_SIZE];
182 audio_buffer = new char[INPUT_SAMPLES * 2 * channels];
183 audio_lock = new Condition(0, "Device1394Input::audio_lock");
184 video_lock = new Condition(0, "Device1394Input::video_lock");
185 buffer_lock = new Mutex("Device1394Input::buffer_lock");
194 void Device1394Input::run()
198 // Wait for frame to arrive
199 struct dv1394_status status;
200 printf("Device1394Input::run %d done=%d\n", __LINE__, done);
202 Thread::enable_cancel();
203 if(ioctl(fd, DV1394_IOC_WAIT_FRAMES, 1))
205 perror("Device1394Input::run DV1394_IOC_WAIT_FRAMES");
209 if(ioctl(fd, DV1394_IOC_GET_STATUS, &status))
211 perror("Device1394Input::run DV1394_IOC_GET_STATUS");
213 Thread::disable_cancel();
217 buffer_lock->lock("Device1394Input::run 1");
219 int nframes = status.n_clear_frames;
220 for(int i = 0; i < nframes; i++)
222 // Get a buffer to transfer to
225 if(!buffer_valid[current_inbuffer])
226 dst = buffer[current_inbuffer];
230 char *src = (char*)(input_buffer + buffer_size * status.first_clear_frame);
231 // static FILE *test = 0;
232 // if(!test) test = fopen("/tmp/test", "w");
233 // fwrite(src, buffer_size, 1, test);
238 memcpy(dst, src, buffer_size);
239 buffer_valid[current_inbuffer] = 1;
240 video_lock->unlock();
245 if(audio_samples < INPUT_SAMPLES - 2048)
247 int audio_result = dv_read_audio(decoder,
248 (unsigned char*)audio_buffer +
249 audio_samples * 2 * 2,
254 int real_freq = decoder->decoder->audio->frequency;
255 if (real_freq == 32000)
257 // do in-place _FAST_ && _SIMPLE_ upsampling to 48khz
258 // i also think user should get a warning that his material is effectively 32khz
259 // we take 16bit samples for both channels in one 32bit int
260 int *twosample = (int*) (audio_buffer + audio_samples * 2 * 2);
261 int from = audio_result - 1;
262 int new_result = audio_result * 48000 / real_freq;
263 for (int to = new_result - 1; to >=0; to--)
265 if ((to % 3) == 0 || (to % 3) == 1) from --;
266 twosample[to] = twosample[from];
268 audio_result = new_result;
272 audio_samples += audio_result;
275 audio_lock->unlock();
280 increment_counter(¤t_inbuffer);
283 Thread::enable_cancel();
284 if(ioctl(fd, DV1394_IOC_RECEIVE_FRAMES, 1))
286 perror("Device1394Input::run DV1394_IOC_RECEIVE_FRAMES");
289 if(ioctl(fd, DV1394_IOC_GET_STATUS, &status))
291 perror("Device1394Input::run DV1394_IOC_GET_STATUS");
293 Thread::disable_cancel();
296 buffer_lock->unlock();
300 void Device1394Input::increment_counter(int *counter)
303 if(*counter >= total_buffers) *counter = 0;
306 void Device1394Input::decrement_counter(int *counter)
309 if(*counter < 0) *counter = total_buffers - 1;
314 int Device1394Input::read_video(VFrame *data)
318 // Take over buffer table
319 buffer_lock->lock("Device1394Input::read_video 1");
320 // Wait for buffer with timeout
321 while(!buffer_valid[current_outbuffer] && !result)
323 buffer_lock->unlock();
324 result = video_lock->timed_lock(BUFFER_TIMEOUT, "Device1394Input::read_video 2");
325 buffer_lock->lock("Device1394Input::read_video 3");
329 if(buffer_valid[current_outbuffer])
331 data->allocate_compressed_data(buffer_size);
332 data->set_compressed_size(buffer_size);
333 memcpy(data->get_data(), buffer[current_outbuffer], buffer_size);
334 buffer_valid[current_outbuffer] = 0;
335 increment_counter(¤t_outbuffer);
338 buffer_lock->unlock();
345 int Device1394Input::read_audio(char *data, int samples)
348 int timeout = (int64_t)samples * (int64_t)1000000 * (int64_t)2 / (int64_t)samplerate;
349 if(timeout < 500000) timeout = 500000;
351 // Take over buffer table
352 buffer_lock->lock("Device1394Input::read_audio 1");
353 // Wait for buffer with timeout
354 while(audio_samples < samples && !result)
356 buffer_lock->unlock();
357 result = audio_lock->timed_lock(timeout, "Device1394Input::read_audio 2");
358 buffer_lock->lock("Device1394Input::read_audio 3");
360 //printf("Device1394Input::read_audio 1 %d %d\n", result, timeout);
362 if(audio_samples >= samples)
364 memcpy(data, audio_buffer, samples * bits * channels / 8);
366 audio_buffer + samples * bits * channels / 8,
367 (audio_samples - samples) * bits * channels / 8);
368 audio_samples -= samples;
370 //printf("Device1394Input::read_audio 100\n");
371 buffer_lock->unlock();