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;
|
|
26 extern float avi_video_ftime;
|
|
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 int skip_audio_bytes=0;
|
|
42 float skip_audio_secs=0;
|
|
43
|
1405
|
44 if(demuxer->file_format==DEMUXER_TYPE_AVI && demuxer->idx_size<=0){
|
|
45 printf("Can't seek in raw .AVI streams! (index required, try with the -idx switch!) \n");
|
|
46 return 0;
|
|
47 }
|
|
48
|
|
49 current_module="seek";
|
|
50
|
1403
|
51 // clear demux buffers:
|
|
52 if(sh_audio){ ds_free_packs(d_audio);sh_audio->a_buffer_len=0;}
|
|
53 ds_free_packs(d_video);
|
|
54
|
|
55 // printf("sh_audio->a_buffer_len=%d \n",sh_audio->a_buffer_len);
|
|
56
|
|
57
|
1405
|
58 switch(demuxer->file_format){
|
1403
|
59
|
|
60 case DEMUXER_TYPE_AVI: {
|
|
61 //================= seek in AVI ==========================
|
|
62 int rel_seek_frames=rel_seek_secs*sh_video->fps;
|
|
63 int curr_audio_pos=0;
|
|
64 int audio_chunk_pos=-1;
|
|
65 int video_chunk_pos=d_video->pos;
|
1405
|
66 int i;
|
1403
|
67
|
|
68 skip_video_frames=0;
|
|
69 avi_audio_pts=0;
|
|
70
|
|
71 // find nearest video keyframe chunk pos:
|
|
72 if(rel_seek_frames>0){
|
|
73 // seek forward
|
|
74 while(video_chunk_pos<demuxer->idx_size){
|
|
75 int id=((AVIINDEXENTRY *)demuxer->idx)[video_chunk_pos].ckid;
|
|
76 if(avi_stream_id(id)==d_video->id){ // video frame
|
|
77 if((--rel_seek_frames)<0 && ((AVIINDEXENTRY *)demuxer->idx)[video_chunk_pos].dwFlags&AVIIF_KEYFRAME) break;
|
|
78 ++skip_audio_bytes;
|
|
79 }
|
|
80 ++video_chunk_pos;
|
|
81 }
|
|
82 } else {
|
|
83 // seek backward
|
|
84 while(video_chunk_pos>=0){
|
|
85 int id=((AVIINDEXENTRY *)demuxer->idx)[video_chunk_pos].ckid;
|
|
86 if(avi_stream_id(id)==d_video->id){ // video frame
|
|
87 if((++rel_seek_frames)>0 && ((AVIINDEXENTRY *)demuxer->idx)[video_chunk_pos].dwFlags&AVIIF_KEYFRAME) break;
|
|
88 --skip_audio_bytes;
|
|
89 }
|
|
90 --video_chunk_pos;
|
|
91 }
|
|
92 }
|
|
93 demuxer->idx_pos_a=demuxer->idx_pos_v=demuxer->idx_pos=video_chunk_pos;
|
|
94 // printf("%d frames skipped\n",skip_audio_bytes);
|
|
95
|
|
96 // re-calc video pts:
|
|
97 d_video->pack_no=0;
|
|
98 for(i=0;i<video_chunk_pos;i++){
|
|
99 int id=((AVIINDEXENTRY *)demuxer->idx)[i].ckid;
|
|
100 if(avi_stream_id(id)==d_video->id) ++d_video->pack_no;
|
|
101 }
|
|
102 sh_video->num_frames=d_video->pack_no;
|
|
103 avi_video_pts=d_video->pack_no*(float)sh_video->video.dwScale/(float)sh_video->video.dwRate;
|
|
104
|
|
105 if(sh_audio){
|
|
106 int i;
|
|
107 int apos=0;
|
|
108 int last=0;
|
|
109 int len=0;
|
|
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;
|
|
173
|
|
174 }
|
|
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);
|
|
179
|
|
180 }
|
|
181 break;
|
|
182
|
|
183 case DEMUXER_TYPE_ASF: {
|
|
184 //================= seek in ASF ==========================
|
|
185 float p_rate=10; // packets / sec
|
|
186 int rel_seek_packs=rel_seek_secs*p_rate;
|
|
187 int rel_seek_bytes=rel_seek_packs*asf_packetsize;
|
|
188 int newpos;
|
|
189 //printf("ASF: packs: %d duration: %d \n",(int)fileh.packets,*((int*)&fileh.duration));
|
|
190 // printf("ASF_seek: %d secs -> %d packs -> %d bytes \n",
|
|
191 // rel_seek_secs,rel_seek_packs,rel_seek_bytes);
|
|
192 newpos=demuxer->filepos+rel_seek_bytes;
|
|
193 if(newpos<0 || newpos<demuxer->movi_start) newpos=demuxer->movi_start;
|
|
194 // printf("\r -- asf: newpos=%d -- \n",newpos);
|
|
195 stream_seek(demuxer->stream,newpos);
|
|
196
|
|
197 ds_fill_buffer(d_video);
|
|
198 if(sh_audio) ds_fill_buffer(d_audio);
|
|
199
|
|
200 while(1){
|
|
201 if(sh_audio){
|
|
202 // sync audio:
|
|
203 if (d_video->pts > d_audio->pts){
|
|
204 if(!ds_fill_buffer(d_audio)) sh_audio=NULL; // skip audio. EOF?
|
|
205 continue;
|
|
206 }
|
|
207 }
|
|
208 if(d_video->flags&1) break; // found a keyframe!
|
|
209 if(!ds_fill_buffer(d_video)) break; // skip frame. EOF?
|
|
210 }
|
|
211
|
|
212 }
|
|
213 break;
|
|
214
|
|
215 case DEMUXER_TYPE_MPEG_ES:
|
|
216 case DEMUXER_TYPE_MPEG_PS: {
|
|
217 //================= seek in MPEG ==========================
|
|
218 int newpos;
|
|
219 if(!sh_video->i_bps) // unspecified?
|
|
220 newpos=demuxer->filepos+2324*75*rel_seek_secs; // 174.3 kbyte/sec
|
|
221 else
|
|
222 newpos=demuxer->filepos+(sh_video->i_bps)*rel_seek_secs;
|
|
223
|
|
224 if(newpos<seek_to_byte) newpos=seek_to_byte;
|
|
225 newpos&=~(STREAM_BUFFER_SIZE-1); /* sector boundary */
|
|
226 stream_seek(demuxer->stream,newpos);
|
|
227 // re-sync video:
|
|
228 videobuf_code_len=0; // reset ES stream buffer
|
|
229 while(1){
|
|
230 int i=sync_video_packet(d_video);
|
|
231 if(i==0x1B3 || i==0x1B8) break; // found it!
|
1405
|
232 if(!i || !skip_video_packet(d_video)) break; // EOF?
|
1403
|
233 }
|
|
234 }
|
|
235 break;
|
|
236
|
1405
|
237 } // switch(demuxer->file_format)
|
1403
|
238
|
|
239 //====================== re-sync audio: =====================
|
|
240 if(sh_audio){
|
|
241
|
|
242 if(skip_audio_bytes){
|
|
243 demux_read_data(d_audio,NULL,skip_audio_bytes);
|
|
244 //d_audio->pts=0; // PTS is outdated because of the raw data skipping
|
|
245 }
|
|
246
|
|
247 current_module="resync_audio";
|
|
248 resync_audio_stream(sh_audio);
|
|
249
|
|
250 // re-sync PTS (MPEG-PS only!!!)
|
1405
|
251 if(demuxer->file_format==DEMUXER_TYPE_MPEG_PS)
|
1403
|
252 if(d_video->pts && d_audio->pts){
|
|
253 if (d_video->pts < d_audio->pts){
|
|
254
|
|
255 } else {
|
|
256 while(d_video->pts > d_audio->pts){
|
|
257 skip_audio_frame(sh_audio);
|
|
258 }
|
|
259 }
|
|
260 }
|
|
261
|
1405
|
262 printf("A:%6.1f V:%6.1f A-V:%7.3f ct: ? \r",d_audio->pts,d_video->pts,0.0f);
|
1403
|
263 } else {
|
1405
|
264 printf("A: --- V:%6.1f \r",d_video->pts);
|
1403
|
265 }
|
1405
|
266 fflush(stdout);
|
1403
|
267
|
|
268 if(sh_audio) sh_audio->timer=-skip_audio_secs;
|
|
269 sh_video->timer=0; // !!!!!!
|
|
270
|
1405
|
271 return 1;
|
|
272 }
|