minor fixes
[goodguy/cinelerra.git] / cinelerra-5.1 / cinelerra / device1394input.C
1
2 /*
3  * CINELERRA
4  * Copyright (C) 2008 Adam Williams <broadcast at earthling dot net>
5  *
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.
10  *
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.
15  *
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
19  *
20  */
21
22 #ifdef HAVE_FIREWIRE
23
24 #include "condition.h"
25 #include "device1394input.h"
26 #include "ieee1394-ioctl.h"
27 #include "mutex.h"
28 #include "vframe.h"
29 #include "video1394.h"
30
31 #include <errno.h>
32 #include <fcntl.h>
33 #include <sys/ioctl.h>
34 #include <sys/mman.h>
35 #include <stdio.h>
36 #include <string.h>
37 #include <unistd.h>
38
39 #define INPUT_SAMPLES 131072
40 #define BUFFER_TIMEOUT 500000
41
42
43 Device1394Input::Device1394Input()
44  : Thread(1, 1, 0)
45 {
46         buffer = 0;
47         buffer_valid = 0;
48         input_buffer = 0;
49         done = 0;
50         total_buffers = 0;
51         current_inbuffer = 0;
52         current_outbuffer = 0;
53         buffer_size = 0;
54         audio_buffer = 0;
55         audio_samples = 0;
56         video_lock = 0;
57         audio_lock = 0;
58         buffer_lock = 0;
59         decoder = 0;
60         fd = -1;
61 }
62
63 Device1394Input::~Device1394Input()
64 {
65 // Driver crashes if it isn't stopped before cancelling the thread.
66 // May still crash during the cancel though.
67
68         if(Thread::running())
69         {
70                 done = 1;
71                 Thread::cancel();
72         }
73         Thread::join();
74
75         if(buffer)
76         {
77                 for(int i = 0; i < total_buffers; i++)
78                         delete [] buffer[i];
79                 delete [] buffer;
80                 delete [] buffer_valid;
81         }
82
83         if(input_buffer)
84                 munmap(input_buffer, total_buffers * buffer_size);
85
86         if(audio_buffer)
87         {
88                 delete [] audio_buffer;
89         }
90
91         if(decoder)
92         {
93                 dv_delete(decoder);
94         }
95
96         if(video_lock) delete video_lock;
97         if(audio_lock) delete audio_lock;
98         if(buffer_lock) delete buffer_lock;
99         if(fd > 0)
100         {
101                 close(fd);
102         }
103 }
104
105 int Device1394Input::open(const char *path,
106         int port,
107         int channel,
108         int length,
109         int channels,
110         int samplerate,
111         int bits,
112         int w,
113         int h)
114 {
115         int result = 0;
116         this->channel = channel;
117         this->length = length;
118         this->channels = channels;
119         this->samplerate = samplerate;
120         this->bits = bits;
121         this->w = w;
122         this->h = h;
123         is_pal = (h == 576);
124         buffer_size = is_pal ? DV_PAL_SIZE : DV_NTSC_SIZE;
125         total_buffers = length;
126
127
128 // Initialize grabbing
129         if(fd < 0 && (fd = ::open(path, O_RDWR)) < 0) {
130                 printf("Device1394Input::open %s: %s\n", path, strerror(errno));
131                 result = 1;
132         }
133         if( !result ) {
134 #define CIP_N_NTSC   68000000
135 #define CIP_D_NTSC 1068000000
136 #define CIP_N_PAL  1
137 #define CIP_D_PAL 16
138                 struct dv1394_init init = {
139                         api_version: DV1394_API_VERSION,
140                         channel: (unsigned int)channel,
141                         n_frames: (unsigned int)length,
142                         format: is_pal ? DV1394_PAL: DV1394_NTSC,
143                         cip_n: 0,
144                         cip_d: 0,
145                         syt_offset: 0
146                 };
147                 if(ioctl(fd, DV1394_IOC_INIT, &init) < 0) {
148                         printf("Device1394Input::open DV1394_IOC_INIT: %s\n", strerror(errno));
149                         result = 1;
150                 }
151         }
152         if( !result ) {
153                 input_buffer = (unsigned char*)mmap(0, length * buffer_size,
154                         PROT_READ | PROT_WRITE, MAP_SHARED, fd, 0);
155                 if( input_buffer == MAP_FAILED ) {
156                         perror("Device1394Input::open mmap");
157                         input_buffer = 0;
158                         result = 1;
159                 }
160         }
161         if( !result ) {
162                 if(ioctl(fd, DV1394_IOC_START_RECEIVE, 0) < 0) {
163                         perror("Device1394Input::open DV1394_START_RECEIVE");
164                         result = 1;
165                 }
166         }
167
168         if( !result ) {
169                 buffer = new char*[total_buffers];
170                 buffer_valid = new int[total_buffers];
171                 bzero(buffer_valid, sizeof(int) * total_buffers);
172                 for(int i = 0; i < total_buffers; i++) {
173                         buffer[i] = new char[DV_PAL_SIZE];
174                 }
175
176                 audio_buffer = new char[INPUT_SAMPLES * 2 * channels];
177                 audio_lock = new Condition(0, "Device1394Input::audio_lock");
178                 video_lock = new Condition(0, "Device1394Input::video_lock");
179                 buffer_lock = new Mutex("Device1394Input::buffer_lock");
180
181                 decoder = dv_new();
182
183                 Thread::start();
184         }
185         return result;
186 }
187
188 void Device1394Input::run()
189 {
190         while(!done)
191         {
192                 int ret = 0;
193 // Wait for frame to arrive
194                 struct dv1394_status status;
195 printf("Device1394Input::run %d done=%d\n", __LINE__, done);
196
197                 Thread::enable_cancel();
198                 if(ioctl(fd, DV1394_IOC_WAIT_FRAMES, 1)) {
199                         perror("Device1394Input::run DV1394_IOC_WAIT_FRAMES");
200                         ret = 1;
201                 }
202                 else if(ioctl(fd, DV1394_IOC_GET_STATUS, &status)) {
203                         perror("Device1394Input::run DV1394_IOC_GET_STATUS");
204                         ret = 1;
205                 }
206                 else if( !input_buffer ) {
207                         fprintf(stderr, "Device1394Input::run !input_buffer");
208                         ret = 1;
209                 }
210                 if( ret )
211                         sleep(1);
212                 Thread::disable_cancel();
213                 if( ret ) continue;
214
215                 buffer_lock->lock("Device1394Input::run 1");
216
217                 int nframes = status.n_clear_frames;
218                 for(int i = 0; i < nframes; i++)
219                 {
220 // Get a buffer to transfer to
221                         char *dst = 0;
222                         int is_overflow = 0;
223                         if(!buffer_valid[current_inbuffer])
224                                 dst = buffer[current_inbuffer];
225                         else
226                                 is_overflow = 1;
227
228                         char *src = (char*)(input_buffer + buffer_size * status.first_clear_frame);
229 // static FILE *test = 0;
230 // if(!test) test = fopen("/tmp/test", "w");
231 // fwrite(src, buffer_size, 1, test);
232
233 // Export the video
234                         if(dst)
235                         {
236                                 memcpy(dst, src, buffer_size);
237                                 buffer_valid[current_inbuffer] = 1;
238                                 video_lock->unlock();
239                         }
240
241
242 // Extract the audio
243                         if(audio_samples < INPUT_SAMPLES - 2048)
244                         {
245                                 int audio_result = dv_read_audio(decoder,
246                                         (unsigned char*)audio_buffer +
247                                                 audio_samples * 2 * 2,
248                                         (unsigned char*)src,
249                                         buffer_size,
250                                         channels,
251                                         bits);
252                                 int real_freq = decoder->decoder->audio->frequency;
253                                 if (real_freq == 32000)
254                                 {
255 // do in-place _FAST_ && _SIMPLE_ upsampling to 48khz
256 // i also think user should get a warning that his material is effectively 32khz
257 // we take 16bit samples for both channels in one 32bit int
258                                         int *twosample = (int*) (audio_buffer + audio_samples * 2 * 2);
259                                         int from = audio_result - 1;
260                                         int new_result = audio_result * 48000 / real_freq;
261                                         for (int to = new_result - 1; to >=0; to--)
262                                         {
263                                                 if ((to % 3) == 0 || (to % 3) == 1) from --;
264                                                 twosample[to] = twosample[from];
265                                         }
266                                         audio_result = new_result;
267                                 }
268
269
270                                 audio_samples += audio_result;
271
272 // Export the audio
273                                 audio_lock->unlock();
274                         }
275
276 // Advance buffer
277                         if(!is_overflow)
278                                 increment_counter(&current_inbuffer);
279
280
281                         Thread::enable_cancel();
282                         if(ioctl(fd, DV1394_IOC_RECEIVE_FRAMES, 1))
283                         {
284                                 perror("Device1394Input::run DV1394_IOC_RECEIVE_FRAMES");
285                         }
286
287                         if(ioctl(fd, DV1394_IOC_GET_STATUS, &status))
288                         {
289                                 perror("Device1394Input::run DV1394_IOC_GET_STATUS");
290                         }
291                         Thread::disable_cancel();
292                 }
293
294                 buffer_lock->unlock();
295         }
296 }
297
298 void Device1394Input::increment_counter(int *counter)
299 {
300         (*counter)++;
301         if(*counter >= total_buffers) *counter = 0;
302 }
303
304 void Device1394Input::decrement_counter(int *counter)
305 {
306         (*counter)--;
307         if(*counter < 0) *counter = total_buffers - 1;
308 }
309
310
311
312 int Device1394Input::read_video(VFrame *data)
313 {
314         int result = 0;
315
316 // Take over buffer table
317         buffer_lock->lock("Device1394Input::read_video 1");
318 // Wait for buffer with timeout
319         while(!buffer_valid[current_outbuffer] && !result)
320         {
321                 buffer_lock->unlock();
322                 result = video_lock->timed_lock(BUFFER_TIMEOUT, "Device1394Input::read_video 2");
323                 buffer_lock->lock("Device1394Input::read_video 3");
324         }
325
326 // Copy frame
327         if(buffer_valid[current_outbuffer])
328         {
329                 data->allocate_compressed_data(buffer_size);
330                 data->set_compressed_size(buffer_size);
331                 memcpy(data->get_data(), buffer[current_outbuffer], buffer_size);
332                 buffer_valid[current_outbuffer] = 0;
333                 increment_counter(&current_outbuffer);
334         }
335
336         buffer_lock->unlock();
337         return result;
338 }
339
340
341
342
343 int Device1394Input::read_audio(char *data, int samples)
344 {
345         int result = 0;
346         int timeout = (int64_t)samples * (int64_t)1000000 * (int64_t)2 / (int64_t)samplerate;
347         if(timeout < 500000) timeout = 500000;
348
349 // Take over buffer table
350         buffer_lock->lock("Device1394Input::read_audio 1");
351 // Wait for buffer with timeout
352         while(audio_samples < samples && !result)
353         {
354                 buffer_lock->unlock();
355                 result = audio_lock->timed_lock(timeout, "Device1394Input::read_audio 2");
356                 buffer_lock->lock("Device1394Input::read_audio 3");
357         }
358 //printf("Device1394Input::read_audio 1 %d %d\n", result, timeout);
359
360         if(audio_samples >= samples)
361         {
362                 memcpy(data, audio_buffer, samples * bits * channels / 8);
363                 memcpy(audio_buffer,
364                         audio_buffer + samples * bits * channels / 8,
365                         (audio_samples - samples) * bits * channels / 8);
366                 audio_samples -= samples;
367         }
368 //printf("Device1394Input::read_audio 100\n");
369         buffer_lock->unlock();
370         return result;
371 }
372
373 #endif