4 * Copyright (C) 2009 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
23 #include "bcsignals.h"
24 #include "condition.h"
26 #include "filethread.h"
30 #include "videodevice.inc"
36 FileThreadFrame::FileThreadFrame()
42 FileThreadFrame::~FileThreadFrame()
44 if(frame) delete frame;
49 FileThread::FileThread(File *file, int do_audio, int do_video)
58 FileThread::~FileThread()
65 void FileThread::reset()
86 bzero(read_frames, sizeof(FileThreadFrame*) * MAX_READ_FRAMES);
90 void FileThread::create_objects(File *file,
95 this->do_audio = do_audio;
96 this->do_video = do_video;
97 file_lock = new Mutex("FileThread::file_lock");
98 read_wait_lock = new Condition(0, "FileThread::read_wait_lock");
99 user_wait_lock = new Condition(0, "FileThread::user_wait_lock");
100 frame_lock = new Mutex("FileThread::frame_lock");
101 for(int i = 0; i < MAX_READ_FRAMES; i++)
102 read_frames[i] = new FileThreadFrame;
106 void FileThread::delete_objects()
108 for(int i = 0; i < MAX_READ_FRAMES; i++)
109 delete read_frames[i];
113 for(int i = 0; i < ring_buffers; i++)
115 delete output_lock[i];
117 delete [] output_lock;
122 for(int i = 0; i < ring_buffers; i++)
124 delete input_lock[i];
126 delete [] input_lock;
131 delete [] last_buffer;
134 delete [] output_size;
139 delete read_wait_lock;
140 delete user_wait_lock;
146 void FileThread::run()
150 if(debug) PRINT_TRACE
154 if(debug) PRINT_TRACE
156 while(!done && !disable_read)
158 if(debug) PRINT_TRACE
159 frame_lock->lock("FileThread::run 1");
160 int local_total_frames = total_frames;
161 frame_lock->unlock();
163 if(local_total_frames >= MAX_READ_FRAMES)
165 read_wait_lock->lock("FileThread::run");
169 if(debug) PRINT_TRACE
170 if(done || disable_read) break;
172 // Make local copes of the locked parameters
173 FileThreadFrame *local_frame = 0;
174 int64_t local_position = 0;
176 if(debug) PRINT_TRACE
178 frame_lock->lock("FileThread::run 2");
179 // Get position of next frame to read
181 local_position = read_frames[total_frames - 1]->position + 1;
183 local_position = start_position;
184 //printf("FileThread::run 1 %d %jd\n", total_frames, local_position);
186 // Get first available frame
187 local_total_frames = total_frames;
188 local_frame = read_frames[local_total_frames];
190 local_frame->valid = 0;
191 frame_lock->unlock();
196 if(debug) PRINT_TRACE
197 file->set_layer(local_layer, 1);
198 file->set_video_position(local_position, 1);
199 int supported_colormodel =
200 file->get_best_colormodel(PLAYBACK_ASYNCHRONOUS, local_layer);
201 if(debug) PRINT_TRACE
205 if(local_frame->frame &&
206 !local_frame->frame->params_match(file->asset->width,
208 supported_colormodel))
210 delete local_frame->frame;
211 local_frame->frame = 0;
214 //printf("FileThread::run %d\n", __LINE__);
215 if(!local_frame->frame)
218 new VFrame(file->asset->width, file->asset->height,
219 supported_colormodel, 0);
223 // printf("FileThread::run %d w=%d h=%d supported_colormodel=%d\n",
225 // local_frame->frame->get_w(),
226 // local_frame->frame->get_h(),
227 // local_frame->frame->get_color_model());
231 printf("file=%p local_frame->frame=%p\n", file, local_frame->frame);
233 file->read_frame(local_frame->frame, 1);
234 if(debug) PRINT_TRACE
235 local_frame->position = local_position;
236 local_frame->layer = local_layer;
238 // Put frame in last position but since the last position now may be
239 // lower than it was when we got the frame, swap the current
240 // last position with the previous last position.
241 frame_lock->lock("FileThread::run 3");
242 FileThreadFrame *old_frame = read_frames[total_frames];
243 read_frames[local_total_frames] = old_frame;
244 read_frames[total_frames++] = local_frame;
245 local_frame->valid = 1;
246 if(debug) PRINT_TRACE
247 frame_lock->unlock();
250 user_wait_lock->unlock();
251 if(debug) PRINT_TRACE
259 output_lock[local_buffer]->lock("FileThread::run 1");
265 if(!last_buffer[local_buffer])
267 if(output_size[local_buffer])
270 file_lock->lock("FileThread::run 2");
273 result = file->write_samples(
274 audio_buffer[local_buffer],
275 output_size[local_buffer]);
280 int layers = 1, count = output_size[local_buffer];
281 VFrame ***frames = video_buffer[local_buffer];
283 layers = file->asset->layers;
284 for( j=0; j<layers && !result; ++j )
285 for( i=0; i<count && !result; ++i )
286 result = file->write_compressed_frame(frames[j][i]);
289 result = file->write_frames(frames, count);
291 for( j=0; j<layers && !result; ++j )
292 for( i=0; i<count && !result; ++i )
293 file->write_frame_done(frames[j][i]->get_number());
298 return_value = result;
303 output_size[local_buffer] = 0;
308 input_lock[local_buffer]->unlock();
310 if(local_buffer >= ring_buffers) local_buffer = 0;
317 int FileThread::stop_writing()
321 int i, buffer, layer, frame;
324 input_lock[current_buffer]->lock("FileThread::stop_writing 1");
326 last_buffer[current_buffer] = 1;
328 for(i = 0; i < ring_buffers; i++)
329 output_lock[i]->unlock();
333 // wait for thread to finish
337 file_lock->lock("FileThread::stop_writing 2");
340 for(buffer = 0; buffer < ring_buffers; buffer++)
342 for(i = 0; i < file->asset->channels; i++)
343 delete audio_buffer[buffer][i];
344 delete [] audio_buffer[buffer];
346 delete [] audio_buffer;
350 // printf("FileThread::stop_writing %d %d %d %d\n",
353 // file->asset->layers,
357 for(buffer = 0; buffer < ring_buffers; buffer++)
359 for(layer = 0; layer < file->asset->layers; layer++)
361 for(frame = 0; frame < buffer_size; frame++)
363 delete video_buffer[buffer][layer][frame];
365 delete [] video_buffer[buffer][layer];
367 delete [] video_buffer[buffer];
369 delete [] video_buffer;
378 int FileThread::start_writing(long buffer_size,
384 int buffer, layer, frame;
386 this->ring_buffers = ring_buffers;
387 this->buffer_size = buffer_size;
388 this->color_model = color_model;
389 this->compressed = compressed;
390 this->current_buffer = ring_buffers - 1;
394 file_lock->lock("FileThread::start_writing 1");
399 // Buffer is swapped before first get
400 last_buffer = new int[ring_buffers];
401 output_size = new long[ring_buffers];
404 output_lock = new Condition*[ring_buffers];
405 input_lock = new Condition*[ring_buffers];
406 for(int i = 0; i < ring_buffers; i++)
408 output_lock[i] = new Condition(0, "FileThread::output_lock");
409 input_lock[i] = new Condition(1, "FileThread::input_lock");
418 audio_buffer = new Samples**[ring_buffers];
419 for(buffer = 0; buffer < ring_buffers; buffer++)
421 audio_buffer[buffer] = new Samples*[file->asset->channels];
423 for(int channel = 0; channel < file->asset->channels; channel++)
425 audio_buffer[buffer][channel] = new Samples(buffer_size);
432 this->color_model = color_model;
433 //long bytes_per_frame = VFrame::calculate_data_size(file->asset->width,
434 // file->asset->height, -1, color_model);
436 video_buffer = new VFrame***[ring_buffers];
437 // printf("FileThread::start_writing 1 %d %d %d %p\n",
439 // file->asset->layers,
442 for(buffer = 0; buffer < ring_buffers; buffer++)
444 video_buffer[buffer] = new VFrame**[file->asset->layers];
445 for(layer = 0; layer < file->asset->layers; layer++)
447 video_buffer[buffer][layer] = new VFrame*[buffer_size];
448 for(frame = 0; frame < buffer_size; frame++)
452 video_buffer[buffer][layer][frame] = new VFrame;
453 //printf("FileThread::start_writing %d %d\n", __LINE__);
457 video_buffer[buffer][layer][frame] =
458 new VFrame( file->asset->width, file->asset->height,
460 // printf("FileThread::start_writing %d %d %d %d %p\n",
465 // video_buffer[buffer][layer]);
473 for(int i = 0; i < ring_buffers; i++)
484 int FileThread::start_reading()
495 int FileThread::stop_reading()
497 if(is_reading && Thread::running())
500 read_wait_lock->unlock();
506 int FileThread::set_video_position(int64_t position)
508 // If the new position can't be added to the buffer without restarting,
510 if((position < this->start_position ||
511 position >= this->start_position + MAX_READ_FRAMES) &&
515 read_wait_lock->unlock();
519 for(int i = 0; i < MAX_READ_FRAMES; i++)
520 read_frames[i]->valid = 0;
521 this->start_position = position;
524 // If a sequential read, enable reading
525 if(this->start_position + 1 == position && disable_read)
527 this->start_position = position;
534 this->start_position = position;
537 this->read_position = position;
541 int FileThread::set_layer(int layer)
543 if(layer != this->layer)
546 read_wait_lock->unlock();
554 int FileThread::read_frame(VFrame *frame)
556 FileThreadFrame *local_frame = 0;
560 //printf("FileThread::read_frame %d this=%p\n", __LINE__, this);
562 // Search thread for frame
563 while(!got_it && !disable_read)
565 frame_lock->lock("FileThread::read_frame 1");
566 // printf("FileThread::read_frame: 1 read_position=%jd ", read_position);
567 // for(int i = 0; i < total_frames; i++)
568 // printf("%jd ", read_frames[i]->position);
570 for(int i = 0; i < total_frames; i++)
572 local_frame = read_frames[i];
573 if(local_frame->position == read_position &&
574 local_frame->layer == layer &&
575 local_frame->frame &&
576 local_frame->frame->equal_stacks(frame) &&
584 frame_lock->unlock();
586 // Not decoded yet but thread active
587 if(!got_it && !disable_read)
589 user_wait_lock->lock("FileThread::read_frame");
593 //printf("FileThread::read_frame %d this=%p\n", __LINE__, this);
597 // printf("FileThread::read_frame 1 color_model=%d disable_read=%d\n",
598 // frame->get_color_model(),
601 if(frame->get_color_model() != local_frame->frame->get_color_model() ||
602 frame->get_w() != local_frame->frame->get_w() ||
603 frame->get_h() != local_frame->frame->get_h())
605 // printf("FileThread::read_frame %d this=%p out cmodel=%d h=%d in cmodel=%d h=%d\n",
608 // frame->get_color_model(),
610 // local_frame->frame->get_color_model(),
611 // local_frame->frame->get_w());
612 BC_CModels::transfer(frame->get_rows(),
613 local_frame->frame->get_rows(),
617 local_frame->frame->get_y(),
618 local_frame->frame->get_u(),
619 local_frame->frame->get_v(),
622 local_frame->frame->get_w(),
623 local_frame->frame->get_h(),
628 local_frame->frame->get_color_model(),
629 frame->get_color_model(),
631 local_frame->frame->get_w(),
633 //for(int i = 0; i < 3000 * 1000 * 4; i++)
634 //((float*)frame->get_rows()[0])[i] = 1;
635 //printf("FileThread::read_frame %d this=%p\n", __LINE__, this);
639 //printf("FileThread::read_frame %d this=%p\n", __LINE__, this);
640 frame->copy_from(local_frame->frame);
641 //printf("FileThread::read_frame %d this=%p\n", __LINE__, this);
644 // Can't copy stacks because the stack is needed by the plugin requestor.
645 frame->copy_params(local_frame->frame);
646 //printf("FileThread::read_frame %d this=%p\n", __LINE__, this);
648 // Recycle all frames before current one but not including current one.
649 // This handles redrawing of a single frame but because FileThread has no
650 // notion of a still frame, it has to call read_frame for those.
651 frame_lock->lock("FileThread::read_frame 1");
652 FileThreadFrame *new_table[MAX_READ_FRAMES];
654 for(int j = number; j < total_frames; j++, k++)
656 new_table[k] = read_frames[j];
658 for(int j = 0; j < number; j++, k++)
660 new_table[k] = read_frames[j];
662 memcpy(read_frames, new_table, sizeof(FileThreadFrame*) * total_frames);
663 //printf("FileThread::read_frame %d this=%p\n", __LINE__, this);
664 total_frames -= number;
666 start_position = read_position;
668 frame_lock->unlock();
669 read_wait_lock->unlock();
674 // printf("FileThread::read_frame 2 color_model=%d disable_read=%d\n",
675 // frame->get_color_model(),
677 //printf("FileThread::read_frame %d this=%p\n", __LINE__, this);
678 // Use traditional read function
679 file->set_layer(layer, 1);
680 file->set_video_position(read_position, 1);
682 int result = file->read_frame(frame, 1);
683 //printf("FileThread::read_frame %d this=%p\n", __LINE__, this);
688 //printf("FileThread::read_frame %d this=%p\n", __LINE__, this);
691 int64_t FileThread::get_memory_usage()
693 frame_lock->lock("FileThread::get_memory_usage");
695 for(int i = 0; i < MAX_READ_FRAMES; i++)
696 if(read_frames[i] && read_frames[i]->frame)
697 result += read_frames[i]->frame->get_data_size();
698 frame_lock->unlock();
703 Samples** FileThread::get_audio_buffer()
707 input_lock[current_buffer]->lock("FileThread::get_audio_buffer");
708 return audio_buffer[current_buffer];
711 VFrame*** FileThread::get_video_buffer()
715 input_lock[current_buffer]->lock("FileThread::get_video_buffer");
716 return video_buffer[current_buffer];
719 VFrame*** FileThread::get_last_video_buffer()
721 return video_buffer[current_buffer];
724 int FileThread::write_buffer(long size)
726 output_size[current_buffer] = size;
728 // unlock the output lock
729 output_lock[current_buffer]->unlock();
734 void FileThread::swap_buffer()
737 if(current_buffer >= ring_buffers) current_buffer = 0;