1405
|
1 #include <stdio.h>
|
|
2 #include <stdlib.h>
|
1403
|
3
|
1405
|
4 extern int verbose; // defined in mplayer.c
|
|
5
|
|
6 #include "config.h"
|
|
7
|
|
8 #include "stream.h"
|
|
9 #include "demuxer.h"
|
|
10 #include "parse_es.h"
|
|
11
|
|
12 #include "wine/mmreg.h"
|
|
13 #include "wine/avifmt.h"
|
|
14 #include "wine/vfw.h"
|
|
15
|
|
16 #include "codec-cfg.h"
|
|
17 #include "stheader.h"
|
|
18
|
|
19 extern void resync_audio_stream(sh_audio_t *sh_audio);
|
|
20 extern void skip_audio_frame(sh_audio_t *sh_audio);
|
|
21
|
|
22 extern int asf_packetsize; // for seeking
|
|
23
|
|
24 extern float avi_audio_pts;
|
|
25 extern float avi_video_pts;
|
1408
|
26 //extern float avi_video_ftime;
|
1405
|
27 extern int skip_video_frames;
|
|
28 extern float initial_pts_delay;
|
|
29 extern int seek_to_byte;
|
|
30 extern char* current_module; // for debugging
|
|
31
|
|
32 // flags:
|
|
33 // 0x1 - absolute/relative
|
|
34 // 0x2 - keyframe/hard
|
|
35
|
|
36 int demux_seek(demuxer_t *demuxer,float rel_seek_secs,int flags){
|
|
37 demux_stream_t *d_audio=demuxer->audio;
|
|
38 demux_stream_t *d_video=demuxer->video;
|
|
39 sh_audio_t *sh_audio=d_audio->sh;
|
|
40 sh_video_t *sh_video=d_video->sh;
|
1403
|
41 float skip_audio_secs=0;
|
|
42
|
1405
|
43 if(demuxer->file_format==DEMUXER_TYPE_AVI && demuxer->idx_size<=0){
|
|
44 printf("Can't seek in raw .AVI streams! (index required, try with the -idx switch!) \n");
|
|
45 return 0;
|
|
46 }
|
|
47
|
|
48 current_module="seek";
|
|
49
|
1403
|
50 // clear demux buffers:
|
|
51 if(sh_audio){ ds_free_packs(d_audio);sh_audio->a_buffer_len=0;}
|
|
52 ds_free_packs(d_video);
|
|
53
|
1421
|
54 demuxer->stream->eof=0; // clear eof flag
|
|
55
|
1403
|
56 // printf("sh_audio->a_buffer_len=%d \n",sh_audio->a_buffer_len);
|
|
57
|
|
58
|
1405
|
59 switch(demuxer->file_format){
|
1403
|
60
|
|
61 case DEMUXER_TYPE_AVI: {
|
|
62 //================= seek in AVI ==========================
|
|
63 int rel_seek_frames=rel_seek_secs*sh_video->fps;
|
|
64 int curr_audio_pos=0;
|
|
65 int audio_chunk_pos=-1;
|
|
66 int video_chunk_pos=d_video->pos;
|
1405
|
67 int i;
|
1403
|
68
|
|
69 skip_video_frames=0;
|
|
70 avi_audio_pts=0;
|
|
71
|
|
72 // find nearest video keyframe chunk pos:
|
|
73 if(rel_seek_frames>0){
|
|
74 // seek forward
|
|
75 while(video_chunk_pos<demuxer->idx_size){
|
|
76 int id=((AVIINDEXENTRY *)demuxer->idx)[video_chunk_pos].ckid;
|
|
77 if(avi_stream_id(id)==d_video->id){ // video frame
|
|
78 if((--rel_seek_frames)<0 && ((AVIINDEXENTRY *)demuxer->idx)[video_chunk_pos].dwFlags&AVIIF_KEYFRAME) break;
|
1408
|
79 // ++skip_audio_bytes;
|
1403
|
80 }
|
|
81 ++video_chunk_pos;
|
|
82 }
|
|
83 } else {
|
|
84 // seek backward
|
|
85 while(video_chunk_pos>=0){
|
|
86 int id=((AVIINDEXENTRY *)demuxer->idx)[video_chunk_pos].ckid;
|
|
87 if(avi_stream_id(id)==d_video->id){ // video frame
|
|
88 if((++rel_seek_frames)>0 && ((AVIINDEXENTRY *)demuxer->idx)[video_chunk_pos].dwFlags&AVIIF_KEYFRAME) break;
|
1408
|
89 // --skip_audio_bytes;
|
1403
|
90 }
|
|
91 --video_chunk_pos;
|
|
92 }
|
|
93 }
|
|
94 demuxer->idx_pos_a=demuxer->idx_pos_v=demuxer->idx_pos=video_chunk_pos;
|
|
95 // printf("%d frames skipped\n",skip_audio_bytes);
|
|
96
|
|
97 // re-calc video pts:
|
|
98 d_video->pack_no=0;
|
|
99 for(i=0;i<video_chunk_pos;i++){
|
|
100 int id=((AVIINDEXENTRY *)demuxer->idx)[i].ckid;
|
|
101 if(avi_stream_id(id)==d_video->id) ++d_video->pack_no;
|
|
102 }
|
|
103 sh_video->num_frames=d_video->pack_no;
|
|
104 avi_video_pts=d_video->pack_no*(float)sh_video->video.dwScale/(float)sh_video->video.dwRate;
|
|
105
|
|
106 if(sh_audio){
|
|
107 int i;
|
|
108 int apos=0;
|
|
109 int last=0;
|
|
110 int len=0;
|
1408
|
111 int skip_audio_bytes=0;
|
1403
|
112
|
|
113 // calc new audio position in audio stream: (using avg.bps value)
|
|
114 curr_audio_pos=(avi_video_pts) * sh_audio->wf->nAvgBytesPerSec;
|
|
115 if(curr_audio_pos<0)curr_audio_pos=0;
|
|
116 #if 1
|
|
117 curr_audio_pos&=~15; // requires for PCM formats!!!
|
|
118 #else
|
|
119 curr_audio_pos/=sh_audio->wf->nBlockAlign;
|
|
120 curr_audio_pos*=sh_audio->wf->nBlockAlign;
|
|
121 demuxer->audio_seekable=1;
|
|
122 #endif
|
|
123
|
|
124 // find audio chunk pos:
|
|
125 for(i=0;i<video_chunk_pos;i++){
|
|
126 int id=((AVIINDEXENTRY *)demuxer->idx)[i].ckid;
|
|
127 if(avi_stream_id(id)==d_audio->id){
|
|
128 len=((AVIINDEXENTRY *)demuxer->idx)[i].dwChunkLength;
|
|
129 last=i;
|
|
130 if(apos<=curr_audio_pos && curr_audio_pos<(apos+len)){
|
|
131 if(verbose)printf("break;\n");
|
|
132 break;
|
|
133 }
|
|
134 apos+=len;
|
|
135 }
|
|
136 }
|
|
137 if(verbose)printf("XXX i=%d last=%d apos=%d curr_audio_pos=%d \n",
|
|
138 i,last,apos,curr_audio_pos);
|
|
139 // audio_chunk_pos=last; // maybe wrong (if not break; )
|
|
140 audio_chunk_pos=i; // maybe wrong (if not break; )
|
|
141 skip_audio_bytes=curr_audio_pos-apos;
|
|
142
|
|
143 // update stream position:
|
|
144 d_audio->pos=audio_chunk_pos;
|
|
145 d_audio->dpos=apos;
|
|
146 d_audio->pts=initial_pts_delay+(float)apos/(float)sh_audio->wf->nAvgBytesPerSec;
|
|
147 demuxer->idx_pos_a=demuxer->idx_pos_v=demuxer->idx_pos=audio_chunk_pos;
|
|
148
|
|
149 if(!(sh_audio->codec->flags&CODECS_FLAG_SEEKABLE)){
|
|
150 #if 0
|
|
151 // curr_audio_pos=apos; // selected audio codec can't seek in chunk
|
|
152 skip_audio_secs=(float)skip_audio_bytes/(float)sh_audio->wf->nAvgBytesPerSec;
|
|
153 //printf("Seek_AUDIO: %d bytes --> %5.3f secs\n",skip_audio_bytes,skip_audio_secs);
|
|
154 skip_audio_bytes=0;
|
|
155 #else
|
|
156 int d=skip_audio_bytes % sh_audio->wf->nBlockAlign;
|
|
157 skip_audio_bytes-=d;
|
|
158 // curr_audio_pos-=d;
|
|
159 skip_audio_secs=(float)d/(float)sh_audio->wf->nAvgBytesPerSec;
|
|
160 //printf("Seek_AUDIO: %d bytes --> %5.3f secs\n",d,skip_audio_secs);
|
|
161 #endif
|
|
162 }
|
|
163 // now: audio_chunk_pos=pos in index
|
|
164 // skip_audio_bytes=bytes to skip from that chunk
|
|
165 // skip_audio_secs=time to play audio before video (if can't skip)
|
|
166
|
|
167 // calc skip_video_frames & adjust video pts counter:
|
|
168 // i=last;
|
|
169 for(i=demuxer->idx_pos;i<video_chunk_pos;i++){
|
|
170 int id=((AVIINDEXENTRY *)demuxer->idx)[i].ckid;
|
|
171 if(avi_stream_id(id)==d_video->id) ++skip_video_frames;
|
|
172 }
|
|
173 // requires for correct audio pts calculation (demuxer):
|
|
174 avi_video_pts-=skip_video_frames*(float)sh_video->video.dwScale/(float)sh_video->video.dwRate;
|
1408
|
175
|
|
176 if(verbose) printf("SEEK: idx=%d (a:%d v:%d) v.skip=%d a.skip=%d/%4.3f \n",
|
|
177 demuxer->idx_pos,audio_chunk_pos,video_chunk_pos,
|
|
178 skip_video_frames,skip_audio_bytes,skip_audio_secs);
|
1403
|
179
|
1408
|
180 if(skip_audio_bytes){
|
|
181 demux_read_data(d_audio,NULL,skip_audio_bytes);
|
|
182 //d_audio->pts=0; // PTS is outdated because of the raw data skipping
|
|
183 }
|
|
184 resync_audio_stream(sh_audio);
|
|
185 }
|
1403
|
186
|
|
187 }
|
|
188 break;
|
|
189
|
|
190 case DEMUXER_TYPE_ASF: {
|
|
191 //================= seek in ASF ==========================
|
|
192 float p_rate=10; // packets / sec
|
|
193 int rel_seek_packs=rel_seek_secs*p_rate;
|
|
194 int rel_seek_bytes=rel_seek_packs*asf_packetsize;
|
|
195 int newpos;
|
|
196 //printf("ASF: packs: %d duration: %d \n",(int)fileh.packets,*((int*)&fileh.duration));
|
|
197 // printf("ASF_seek: %d secs -> %d packs -> %d bytes \n",
|
|
198 // rel_seek_secs,rel_seek_packs,rel_seek_bytes);
|
|
199 newpos=demuxer->filepos+rel_seek_bytes;
|
|
200 if(newpos<0 || newpos<demuxer->movi_start) newpos=demuxer->movi_start;
|
|
201 // printf("\r -- asf: newpos=%d -- \n",newpos);
|
|
202 stream_seek(demuxer->stream,newpos);
|
|
203
|
|
204 ds_fill_buffer(d_video);
|
1408
|
205 if(sh_audio){
|
|
206 ds_fill_buffer(d_audio);
|
|
207 resync_audio_stream(sh_audio);
|
|
208 }
|
1403
|
209
|
|
210 while(1){
|
1408
|
211 if(sh_audio && !d_audio->eof){
|
|
212 float a_pts=d_audio->pts;
|
|
213 a_pts+=(ds_tell_pts(d_audio)-sh_audio->a_in_buffer_len)/(float)sh_audio->i_bps;
|
1403
|
214 // sync audio:
|
1408
|
215 if (d_video->pts > a_pts){
|
|
216 skip_audio_frame(sh_audio);
|
|
217 // if(!ds_fill_buffer(d_audio)) sh_audio=NULL; // skip audio. EOF?
|
1403
|
218 continue;
|
|
219 }
|
|
220 }
|
|
221 if(d_video->flags&1) break; // found a keyframe!
|
|
222 if(!ds_fill_buffer(d_video)) break; // skip frame. EOF?
|
|
223 }
|
|
224
|
|
225 }
|
|
226 break;
|
|
227
|
|
228 case DEMUXER_TYPE_MPEG_ES:
|
|
229 case DEMUXER_TYPE_MPEG_PS: {
|
|
230 //================= seek in MPEG ==========================
|
|
231 int newpos;
|
|
232 if(!sh_video->i_bps) // unspecified?
|
|
233 newpos=demuxer->filepos+2324*75*rel_seek_secs; // 174.3 kbyte/sec
|
|
234 else
|
|
235 newpos=demuxer->filepos+(sh_video->i_bps)*rel_seek_secs;
|
|
236
|
|
237 if(newpos<seek_to_byte) newpos=seek_to_byte;
|
|
238 newpos&=~(STREAM_BUFFER_SIZE-1); /* sector boundary */
|
|
239 stream_seek(demuxer->stream,newpos);
|
1408
|
240
|
1403
|
241 // re-sync video:
|
|
242 videobuf_code_len=0; // reset ES stream buffer
|
1408
|
243
|
|
244 ds_fill_buffer(d_video);
|
|
245 if(sh_audio){
|
|
246 ds_fill_buffer(d_audio);
|
|
247 resync_audio_stream(sh_audio);
|
|
248 }
|
|
249
|
|
250 while(1){
|
|
251 int i;
|
|
252 if(sh_audio && !d_audio->eof && d_video->pts && d_audio->pts){
|
|
253 float a_pts=d_audio->pts;
|
|
254 a_pts+=(ds_tell_pts(d_audio)-sh_audio->a_in_buffer_len)/(float)sh_audio->i_bps;
|
|
255 if(d_video->pts>a_pts){
|
|
256 skip_audio_frame(sh_audio); // sync audio
|
|
257 continue;
|
|
258 }
|
|
259 }
|
|
260 i=sync_video_packet(d_video);
|
1403
|
261 if(i==0x1B3 || i==0x1B8) break; // found it!
|
1405
|
262 if(!i || !skip_video_packet(d_video)) break; // EOF?
|
1403
|
263 }
|
1408
|
264
|
1403
|
265 }
|
|
266 break;
|
|
267
|
1405
|
268 } // switch(demuxer->file_format)
|
1403
|
269
|
|
270 //====================== re-sync audio: =====================
|
|
271 if(sh_audio){
|
1408
|
272 if(verbose){
|
|
273 float a_pts=d_audio->pts;
|
|
274 a_pts+=(ds_tell_pts(d_audio)-sh_audio->a_in_buffer_len)/(float)sh_audio->i_bps;
|
|
275 printf("SEEK: A: %5.3f V: %5.3f A-V: %5.3f \n",a_pts,d_video->pts,a_pts-d_video->pts);
|
|
276 }
|
1405
|
277 printf("A:%6.1f V:%6.1f A-V:%7.3f ct: ? \r",d_audio->pts,d_video->pts,0.0f);
|
1403
|
278 } else {
|
1405
|
279 printf("A: --- V:%6.1f \r",d_video->pts);
|
1403
|
280 }
|
1405
|
281 fflush(stdout);
|
1403
|
282
|
|
283 if(sh_audio) sh_audio->timer=-skip_audio_secs;
|
|
284 sh_video->timer=0; // !!!!!!
|
|
285
|
1405
|
286 return 1;
|
|
287 }
|