add audio to proxy, and minor bug fixes
[goodguy/history.git] / cinelerra-5.1 / libzmpeg3 / video / seek.C
1 #include "../libzmpeg3.h"
2
3 static void
4 toc_error()
5 {
6   zerr( "frame accurate seeking without a table of contents \n"
7     "is no longer supported.  Use mpeg3toc <mpeg file> <table of contents>\n"
8     "to generate a table of contents and load the table of contents instead.\n");
9 }
10
11 void zvideo_t::
12 cache_frame()
13 {
14   if( output_src[0] ) {
15     int coded_size = coded_picture_width * coded_picture_height;
16     int chrom_size = chrom_width * chrom_height;
17     track->frame_cache->put_frame(framenum-1,
18       output_src[0], output_src[1], output_src[2],
19       coded_size, chrom_size, chrom_size);
20   }
21 }
22
23 int zvideo_t::
24 drop_frames(long frames, int cache_it)
25 {
26   int result = 0;
27   long frame_number = framenum + frames;
28 //zmsgs("drop frames framenum %d frames %ld\n", framenum, frames);
29
30 /* Read the selected number of frames and skip b-frames */
31   while(!result && frame_number > framenum ) {
32     if( cache_it ) {
33       result = read_frame_backend(0);
34       if( !result && ref_frames > 1 )
35         cache_frame();
36     }
37     else {
38 //zmsgs("framenum %d frame_offset %jx last_packet_start %jx repeat %d/%d",
39 //  framenum,track->frame_offsets[framenum+1],vstream->demuxer->last_packet_start,
40 //  repeat_fields, current_field);
41       result = read_frame_backend(frame_number - framenum);
42 //zmsgs(" type %c\n","XIPBD"[pict_type]);
43     }
44   }
45
46   return result;
47 }
48
49 int zbits_t::
50 next_start_code()
51 {
52 //zmsg("1\n");
53   /* search forwards */
54   prev_byte_align();
55   int64_t stream_end = demuxer->stream_end;
56   if( demuxer->src->is_transport_stream() )
57     demuxer->stream_end = demuxer->program_byte + MAX_TS_PROBE;
58   else if( demuxer->src->is_program_stream() )
59     demuxer->stream_end = demuxer->program_byte + MAX_PGM_PROBE;
60   int zcode = -1, result = -1;
61
62   while( !eof() ) {
63     int bits = (uint8_t)zcode!=0 ? 24 : 8;
64     zcode = (zcode<<bits) | get_bits_noptr(bits);
65     if( (zcode&0xffffff) == PACKET_START_CODE_PREFIX ) {
66       zcode = (zcode<<8) + get_byte_noptr();
67       reset(result = zcode);
68 //zmsgs("5 %04x\n",zcode);
69       break;
70     }
71   }
72
73   demuxer->stream_end = stream_end;
74   return result;
75 }
76
77 /* Line up on the beginning of the next code. */
78 int zbits_t::
79 next_code(uint32_t zcode)
80 {
81   while(!eof() && show_bits_noptr(32) != zcode ) {
82     get_byte_noptr();
83   }
84   return eof();
85 }
86
87 /* Line up on the beginning of the previous code. */
88 int zdemuxer_t::
89 prev_code(uint32_t zcode)
90 {
91   uint32_t current_code = 0;
92 #define PREV_CODE_MACRO { current_code >>= 8; \
93   current_code |= ((uint32_t)read_prev_char()) << 24; }
94
95   PREV_CODE_MACRO
96   PREV_CODE_MACRO
97   PREV_CODE_MACRO
98   PREV_CODE_MACRO
99
100   while( !bof() && current_code != zcode ) {
101     PREV_CODE_MACRO
102   }
103   return bof();
104 }
105
106 int zvideo_t::
107 seek_byte(int64_t byte)
108 {
109   byte_seek = byte;
110   return 0;
111 }
112
113 int zvideo_t::
114 seek_frame(long frame)
115 {
116   frame_seek = frame;
117   return 0;
118 }
119
120 int zvideo_t::
121 rewind_video(int preload)
122 {
123   vstream->reset();
124   framenum = 0;
125   if( track->frame_offsets )
126     vstream->seek_byte(track->frame_offsets[0]);
127   else
128     vstream->seek_byte(0);
129   track->reset_pts();
130   repeat_data = ref_frames = 0;
131   last_number = framenum = -1;
132   return preload ? read_frame_backend(0) : 0;
133 }
134
135 int zvideo_t::
136 seek()
137 {
138   int result = 0;
139   int64_t byte;
140   long frame_number;
141   int debug = 0;
142   demuxer_t *demux = vstream->demuxer;
143
144 /* Must do seeking here so files which don't use video don't seek. */
145 /* Seeking is done in the demuxer */
146 /* Seek to absolute byte */
147   if( byte_seek >= 0 ) {
148     byte = byte_seek;
149     byte_seek = -1;
150     if( byte != demux->tell_byte() ) {
151       demux->seek_byte(byte);
152       track->reset_pts();
153       if( byte > 0 ) {
154   //zmsg("1\n");
155         demux->start_reverse();
156   //zmsgs("1 %jd\n", demux->tell_byte());
157         for( int i=0; i<2; ++i ) {     /* Rewind 2 I-frames */
158           if( has_gops ?
159             demux->prev_code(GOP_START_CODE) :
160             demux->prev_code(SEQUENCE_START_CODE) ) break;
161         }
162   //zmsgs("2 %jd\n", demux->tell_byte());
163         demux->start_forward();
164         if( track->frame_offsets ) {
165           int64_t offset = demux->tell_byte();
166           for( int k=0; k<track->total_keyframe_numbers; ++k ) {
167             frame_number = track->keyframe_numbers[k];
168             if( track->frame_offsets[frame_number] >= offset ) {
169               /* first entry is special */
170               framenum = frame_number>0 ? frame_number-1 : 0;
171               break;
172             }
173           }
174         }
175         else {
176           /* framenum no longer accurate, clear subtitles */
177           reset_subtitles();
178         }
179       }
180       else { /* Read first frame */
181         rewind_video();
182       }
183       vstream->reset();
184       repeat_data = ref_frames = 0;
185   //zmsgs("4 %jd\n", demux->tell_byte());
186       while( !result && !demux->eof() && demux->tell_byte() < byte ) {
187         result = read_frame_backend(0);
188       }
189     }
190   }
191   else if( frame_seek >= 0 ) {
192     /* Seek to a frame number */
193     frame_number = frame_seek;
194     frame_seek = -1;
195     if( frame_number != framenum ) {
196       if( debug ) zmsgs("%d\n", __LINE__);
197       if( frame_number < 0 ) frame_number = 0;
198       if( frame_number > maxframe ) frame_number = maxframe;
199       if(debug) zmsgs("%d %ld %d\n", __LINE__, frame_number, framenum);
200       /* Seek to I frame in table of contents. */
201       /* Determine time between seek position and previous subtitle. */
202       /* Subtract time difference from subtitle display time. */
203       if( frame_number > 0 && track->frame_offsets ) {
204         if( debug ) zmsgs("%d\n", __LINE__);
205         track->frame_cache->reset();
206         if( debug ) zmsgs("%d\n", __LINE__);
207         if( frame_number < framenum || 
208             frame_number-framenum > SEEK_THRESHOLD ) {
209           int idx = track->find_keyframe_index(frame_number);
210           /* back up 2 I frames, data may reference old refframes */
211           if( --idx < 0 ) idx = 0;
212           idx = track->keyframe_numbers[idx];
213           byte = track->frame_offsets[idx];
214           /* maybe lots of frames in one block, move is to earliest frame */
215           while( idx > 0 && track->frame_offsets[idx-1] == byte ) --idx;
216           if( debug ) zmsgs("%ld idx=%d, byte=%jd\n", frame_number,idx,byte);
217           framenum = idx-1;
218           vstream->seek_byte(byte);
219           track->reset_pts();
220           repeat_data = ref_frames = 0;
221           /* get the first I-frame.  It does not count in the frame sequence */
222           /*   phys order is IBBPBBPBB... presentation order is BBIBBPBB... */
223           result = read_frame_backend(0);
224         }
225         if( debug ) zmsgs("%d %ld %d\n", __LINE__,frame_number,framenum);
226         if( !result && frame_number > framenum ) {
227           /* Read up to current frame */
228           int n = frame_number - framenum;
229           result = drop_frames(n, n < MAX_CACHE_FRAMES ? 1 : 0);
230           if( debug ) zmsgs("%d\n", __LINE__);
231         }
232       }
233       else if( frame_number == 0 ) {
234         result = rewind_video();
235       }
236       else {
237         /* No support for seeking without table of contents */
238         toc_error();
239         result = 1;
240       }
241     }
242   }
243   else if( framenum < 0 ) {   /* preload */
244     result = rewind_video(1);
245   }
246   else
247     return 0;
248
249   seek_time = 0;
250   return result;
251 }
252
253 int zvideo_t::
254 previous_frame()
255 {
256   int result = 0;
257   int64_t target_byte = 0;
258   demuxer_t *demux = vstream->demuxer;
259   if( demux->tell_byte() <= 0 ) return 1;
260   /* Get location of end of previous picture */
261   demux->start_reverse();
262   result = demux->prev_code(PICTURE_START_CODE);
263   if( !result ) result = demux->prev_code(PICTURE_START_CODE);
264   if( !result ) result = demux->prev_code(PICTURE_START_CODE);
265   if( !result ) target_byte = demux->tell_byte();
266   if( !result ) { /* Rewind 2 I-frames */
267     if( has_gops )
268       result = demux->prev_code(GOP_START_CODE);
269     else
270       result = demux->prev_code(SEQUENCE_START_CODE);
271   }
272
273   if( !result )
274   {
275     if( has_gops )
276       result = demux->prev_code(GOP_START_CODE);
277     else
278       result = demux->prev_code(SEQUENCE_START_CODE);
279   }
280
281   demux->start_forward();
282   vstream->reset();
283
284   result = 0;    /* Read up to correct byte */
285   repeat_data = ref_frames = 0;
286   while( !result && !demux->eof() && demux->tell_byte() < target_byte ) {
287     result = read_frame_backend(0);
288   }
289   repeat_data = 0;
290
291   return 0;
292 }
293