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