Mercurial > mplayer.hg
comparison stream/cache2.c @ 19271:64d82a45a05d
introduce new 'stream' directory for all stream layer related components and split them from libmpdemux
author | ben |
---|---|
date | Mon, 31 Jul 2006 17:39:17 +0000 |
parents | libmpdemux/cache2.c@d2d9d011203f |
children | 84f95595f31f |
comparison
equal
deleted
inserted
replaced
19270:7d39b911f0bd | 19271:64d82a45a05d |
---|---|
1 #include "config.h" | |
2 | |
3 // Initial draft of my new cache system... | |
4 // Note it runs in 2 processes (using fork()), but doesn't requires locking!! | |
5 // TODO: seeking, data consistency checking | |
6 | |
7 #define READ_USLEEP_TIME 10000 | |
8 #define FILL_USLEEP_TIME 50000 | |
9 #define PREFILL_SLEEP_TIME 200 | |
10 | |
11 #include <stdio.h> | |
12 #include <stdlib.h> | |
13 #include <string.h> | |
14 #include <signal.h> | |
15 #include <sys/types.h> | |
16 #include <unistd.h> | |
17 | |
18 #include "osdep/timer.h" | |
19 #ifndef WIN32 | |
20 #include <sys/wait.h> | |
21 #include "osdep/shmem.h" | |
22 #else | |
23 #include <windows.h> | |
24 static DWORD WINAPI ThreadProc(void* s); | |
25 #endif | |
26 | |
27 #include "mp_msg.h" | |
28 #include "help_mp.h" | |
29 | |
30 #include "stream.h" | |
31 | |
32 int stream_fill_buffer(stream_t *s); | |
33 int stream_seek_long(stream_t *s,off_t pos); | |
34 | |
35 extern int mp_input_check_interrupt(int time); | |
36 | |
37 typedef struct { | |
38 // constats: | |
39 unsigned char *buffer; // base pointer of the alllocated buffer memory | |
40 int buffer_size; // size of the alllocated buffer memory | |
41 int sector_size; // size of a single sector (2048/2324) | |
42 int back_size; // we should keep back_size amount of old bytes for backward seek | |
43 int fill_limit; // we should fill buffer only if space>=fill_limit | |
44 int seek_limit; // keep filling cache if distanse is less that seek limit | |
45 // filler's pointers: | |
46 int eof; | |
47 off_t min_filepos; // buffer contain only a part of the file, from min-max pos | |
48 off_t max_filepos; | |
49 off_t offset; // filepos <-> bufferpos offset value (filepos of the buffer's first byte) | |
50 // reader's pointers: | |
51 off_t read_filepos; | |
52 // commands/locking: | |
53 // int seek_lock; // 1 if we will seek/reset buffer, 2 if we are ready for cmd | |
54 // int fifo_flag; // 1 if we should use FIFO to notice cache about buffer reads. | |
55 // callback | |
56 stream_t* stream; | |
57 } cache_vars_t; | |
58 | |
59 static int min_fill=0; | |
60 | |
61 int cache_fill_status=0; | |
62 | |
63 void cache_stats(cache_vars_t* s){ | |
64 int newb=s->max_filepos-s->read_filepos; // new bytes in the buffer | |
65 mp_msg(MSGT_CACHE,MSGL_INFO,"0x%06X [0x%06X] 0x%06X ",(int)s->min_filepos,(int)s->read_filepos,(int)s->max_filepos); | |
66 mp_msg(MSGT_CACHE,MSGL_INFO,"%3d %% (%3d%%)\n",100*newb/s->buffer_size,100*min_fill/s->buffer_size); | |
67 } | |
68 | |
69 int cache_read(cache_vars_t* s,unsigned char* buf,int size){ | |
70 int total=0; | |
71 while(size>0){ | |
72 int pos,newb,len; | |
73 | |
74 //printf("CACHE2_READ: 0x%X <= 0x%X <= 0x%X \n",s->min_filepos,s->read_filepos,s->max_filepos); | |
75 | |
76 if(s->read_filepos>=s->max_filepos || s->read_filepos<s->min_filepos){ | |
77 // eof? | |
78 if(s->eof) break; | |
79 // waiting for buffer fill... | |
80 usec_sleep(READ_USLEEP_TIME); // 10ms | |
81 continue; // try again... | |
82 } | |
83 | |
84 newb=s->max_filepos-s->read_filepos; // new bytes in the buffer | |
85 if(newb<min_fill) min_fill=newb; // statistics... | |
86 | |
87 // printf("*** newb: %d bytes ***\n",newb); | |
88 | |
89 pos=s->read_filepos - s->offset; | |
90 if(pos<0) pos+=s->buffer_size; else | |
91 if(pos>=s->buffer_size) pos-=s->buffer_size; | |
92 | |
93 if(newb>s->buffer_size-pos) newb=s->buffer_size-pos; // handle wrap... | |
94 if(newb>size) newb=size; | |
95 | |
96 // check: | |
97 if(s->read_filepos<s->min_filepos) mp_msg(MSGT_CACHE,MSGL_ERR,"Ehh. s->read_filepos<s->min_filepos !!! Report bug...\n"); | |
98 | |
99 // len=write(mem,newb) | |
100 //printf("Buffer read: %d bytes\n",newb); | |
101 memcpy(buf,&s->buffer[pos],newb); | |
102 buf+=newb; | |
103 len=newb; | |
104 // ... | |
105 | |
106 s->read_filepos+=len; | |
107 size-=len; | |
108 total+=len; | |
109 | |
110 } | |
111 cache_fill_status=(s->max_filepos-s->read_filepos)/(s->buffer_size / 100); | |
112 return total; | |
113 } | |
114 | |
115 int cache_fill(cache_vars_t* s){ | |
116 int back,back2,newb,space,len,pos; | |
117 off_t read=s->read_filepos; | |
118 | |
119 if(read<s->min_filepos || read>s->max_filepos){ | |
120 // seek... | |
121 mp_msg(MSGT_CACHE,MSGL_DBG2,"Out of boundaries... seeking to 0x%"PRIX64" \n",(int64_t)read); | |
122 // streaming: drop cache contents only if seeking backward or too much fwd: | |
123 if(s->stream->type!=STREAMTYPE_STREAM || | |
124 read<s->min_filepos || read>=s->max_filepos+s->seek_limit) | |
125 { | |
126 s->offset= // FIXME!? | |
127 s->min_filepos=s->max_filepos=read; // drop cache content :( | |
128 if(s->stream->eof) stream_reset(s->stream); | |
129 stream_seek(s->stream,read); | |
130 mp_msg(MSGT_CACHE,MSGL_DBG2,"Seek done. new pos: 0x%"PRIX64" \n",(int64_t)stream_tell(s->stream)); | |
131 } | |
132 } | |
133 | |
134 // calc number of back-bytes: | |
135 back=read - s->min_filepos; | |
136 if(back<0) back=0; // strange... | |
137 if(back>s->back_size) back=s->back_size; | |
138 | |
139 // calc number of new bytes: | |
140 newb=s->max_filepos - read; | |
141 if(newb<0) newb=0; // strange... | |
142 | |
143 // calc free buffer space: | |
144 space=s->buffer_size - (newb+back); | |
145 | |
146 // calc bufferpos: | |
147 pos=s->max_filepos - s->offset; | |
148 if(pos>=s->buffer_size) pos-=s->buffer_size; // wrap-around | |
149 | |
150 if(space<s->fill_limit){ | |
151 // printf("Buffer is full (%d bytes free, limit: %d)\n",space,s->fill_limit); | |
152 return 0; // no fill... | |
153 } | |
154 | |
155 // printf("### read=0x%X back=%d newb=%d space=%d pos=%d\n",read,back,newb,space,pos); | |
156 | |
157 // reduce space if needed: | |
158 if(space>s->buffer_size-pos) space=s->buffer_size-pos; | |
159 | |
160 // if(space>32768) space=32768; // limit one-time block size | |
161 if(space>4*s->sector_size) space=4*s->sector_size; | |
162 | |
163 // if(s->seek_lock) return 0; // FIXME | |
164 | |
165 #if 1 | |
166 // back+newb+space <= buffer_size | |
167 back2=s->buffer_size-(space+newb); // max back size | |
168 if(s->min_filepos<(read-back2)) s->min_filepos=read-back2; | |
169 #else | |
170 s->min_filepos=read-back; // avoid seeking-back to temp area... | |
171 #endif | |
172 | |
173 // .... | |
174 //printf("Buffer fill: %d bytes of %d\n",space,s->buffer_size); | |
175 //len=stream_fill_buffer(s->stream); | |
176 //memcpy(&s->buffer[pos],s->stream->buffer,len); // avoid this extra copy! | |
177 // .... | |
178 len=stream_read(s->stream,&s->buffer[pos],space); | |
179 if(!len) s->eof=1; | |
180 | |
181 s->max_filepos+=len; | |
182 if(pos+len>=s->buffer_size){ | |
183 // wrap... | |
184 s->offset+=s->buffer_size; | |
185 } | |
186 | |
187 return len; | |
188 | |
189 } | |
190 | |
191 cache_vars_t* cache_init(int size,int sector){ | |
192 int num; | |
193 #ifndef WIN32 | |
194 cache_vars_t* s=shmem_alloc(sizeof(cache_vars_t)); | |
195 #else | |
196 cache_vars_t* s=malloc(sizeof(cache_vars_t)); | |
197 #endif | |
198 if(s==NULL) return NULL; | |
199 | |
200 memset(s,0,sizeof(cache_vars_t)); | |
201 num=size/sector; | |
202 if(num < 16){ | |
203 num = 16; | |
204 }//32kb min_size | |
205 s->buffer_size=num*sector; | |
206 s->sector_size=sector; | |
207 #ifndef WIN32 | |
208 s->buffer=shmem_alloc(s->buffer_size); | |
209 #else | |
210 s->buffer=malloc(s->buffer_size); | |
211 #endif | |
212 | |
213 if(s->buffer == NULL){ | |
214 #ifndef WIN32 | |
215 shmem_free(s,sizeof(cache_vars_t)); | |
216 #else | |
217 free(s); | |
218 #endif | |
219 return NULL; | |
220 } | |
221 | |
222 s->fill_limit=8*sector; | |
223 s->back_size=s->buffer_size/2; | |
224 return s; | |
225 } | |
226 | |
227 void cache_uninit(stream_t *s) { | |
228 cache_vars_t* c = s->cache_data; | |
229 if(!s->cache_pid) return; | |
230 #ifndef WIN32 | |
231 kill(s->cache_pid,SIGKILL); | |
232 waitpid(s->cache_pid,NULL,0); | |
233 #else | |
234 TerminateThread((HANDLE)s->cache_pid,0); | |
235 free(c->stream); | |
236 #endif | |
237 if(!c) return; | |
238 #ifndef WIN32 | |
239 shmem_free(c->buffer,c->buffer_size); | |
240 shmem_free(s->cache_data,sizeof(cache_vars_t)); | |
241 #else | |
242 free(c->buffer); | |
243 free(s->cache_data); | |
244 #endif | |
245 } | |
246 | |
247 static void exit_sighandler(int x){ | |
248 // close stream | |
249 exit(0); | |
250 } | |
251 | |
252 int stream_enable_cache(stream_t *stream,int size,int min,int seek_limit){ | |
253 int ss=(stream->type==STREAMTYPE_VCD)?VCD_SECTOR_DATA:STREAM_BUFFER_SIZE; | |
254 cache_vars_t* s; | |
255 | |
256 if (stream->type==STREAMTYPE_STREAM && stream->fd < 0) { | |
257 // The stream has no 'fd' behind it, so is non-cacheable | |
258 mp_msg(MSGT_CACHE,MSGL_STATUS,"\rThis stream is non-cacheable\n"); | |
259 return 1; | |
260 } | |
261 | |
262 s=cache_init(size,ss); | |
263 if(s == NULL) return 0; | |
264 stream->cache_data=s; | |
265 s->stream=stream; // callback | |
266 s->seek_limit=seek_limit; | |
267 | |
268 | |
269 //make sure that we won't wait from cache_fill | |
270 //more data than it is alowed to fill | |
271 if (s->seek_limit > s->buffer_size - s->fill_limit ){ | |
272 s->seek_limit = s->buffer_size - s->fill_limit; | |
273 } | |
274 if (min > s->buffer_size - s->fill_limit) { | |
275 min = s->buffer_size - s->fill_limit; | |
276 } | |
277 | |
278 #ifndef WIN32 | |
279 if((stream->cache_pid=fork())){ | |
280 #else | |
281 { | |
282 DWORD threadId; | |
283 stream_t* stream2=malloc(sizeof(stream_t)); | |
284 memcpy(stream2,s->stream,sizeof(stream_t)); | |
285 s->stream=stream2; | |
286 stream->cache_pid = CreateThread(NULL,0,ThreadProc,s,0,&threadId); | |
287 #endif | |
288 // wait until cache is filled at least prefill_init % | |
289 mp_msg(MSGT_CACHE,MSGL_V,"CACHE_PRE_INIT: %"PRId64" [%"PRId64"] %"PRId64" pre:%d eof:%d \n", | |
290 (int64_t)s->min_filepos,(int64_t)s->read_filepos,(int64_t)s->max_filepos,min,s->eof); | |
291 while(s->read_filepos<s->min_filepos || s->max_filepos-s->read_filepos<min){ | |
292 mp_msg(MSGT_CACHE,MSGL_STATUS,MSGTR_CacheFill, | |
293 100.0*(float)(s->max_filepos-s->read_filepos)/(float)(s->buffer_size), | |
294 (int64_t)s->max_filepos-s->read_filepos | |
295 ); | |
296 if(s->eof) break; // file is smaller than prefill size | |
297 if(mp_input_check_interrupt(PREFILL_SLEEP_TIME)) | |
298 return 0; | |
299 } | |
300 mp_msg(MSGT_CACHE,MSGL_STATUS,"\n"); | |
301 return 1; // parent exits | |
302 } | |
303 | |
304 #ifdef WIN32 | |
305 } | |
306 static DWORD WINAPI ThreadProc(void*s){ | |
307 #endif | |
308 | |
309 // cache thread mainloop: | |
310 signal(SIGTERM,exit_sighandler); // kill | |
311 while(1){ | |
312 if(!cache_fill((cache_vars_t*)s)){ | |
313 usec_sleep(FILL_USLEEP_TIME); // idle | |
314 } | |
315 // cache_stats(s->cache_data); | |
316 } | |
317 } | |
318 | |
319 int cache_stream_fill_buffer(stream_t *s){ | |
320 int len; | |
321 if(s->eof){ s->buf_pos=s->buf_len=0; return 0; } | |
322 if(!s->cache_pid) return stream_fill_buffer(s); | |
323 | |
324 // cache_stats(s->cache_data); | |
325 | |
326 if(s->pos!=((cache_vars_t*)s->cache_data)->read_filepos) mp_msg(MSGT_CACHE,MSGL_ERR,"!!! read_filepos differs!!! report this bug...\n"); | |
327 | |
328 len=cache_read(s->cache_data,s->buffer, ((cache_vars_t*)s->cache_data)->sector_size); | |
329 //printf("cache_stream_fill_buffer->read -> %d\n",len); | |
330 | |
331 if(len<=0){ s->eof=1; s->buf_pos=s->buf_len=0; return 0; } | |
332 s->buf_pos=0; | |
333 s->buf_len=len; | |
334 s->pos+=len; | |
335 // printf("[%d]",len);fflush(stdout); | |
336 return len; | |
337 | |
338 } | |
339 | |
340 int cache_stream_seek_long(stream_t *stream,off_t pos){ | |
341 cache_vars_t* s; | |
342 off_t newpos; | |
343 if(!stream->cache_pid) return stream_seek_long(stream,pos); | |
344 | |
345 s=stream->cache_data; | |
346 // s->seek_lock=1; | |
347 | |
348 mp_msg(MSGT_CACHE,MSGL_DBG2,"CACHE2_SEEK: 0x%"PRIX64" <= 0x%"PRIX64" (0x%"PRIX64") <= 0x%"PRIX64" \n",s->min_filepos,pos,s->read_filepos,s->max_filepos); | |
349 | |
350 newpos=pos/s->sector_size; newpos*=s->sector_size; // align | |
351 stream->pos=s->read_filepos=newpos; | |
352 s->eof=0; // !!!!!!! | |
353 | |
354 cache_stream_fill_buffer(stream); | |
355 | |
356 pos-=newpos; | |
357 if(pos>=0 && pos<=stream->buf_len){ | |
358 stream->buf_pos=pos; // byte position in sector | |
359 return 1; | |
360 } | |
361 | |
362 // stream->buf_pos=stream->buf_len=0; | |
363 // return 1; | |
364 | |
365 mp_msg(MSGT_CACHE,MSGL_V,"cache_stream_seek: WARNING! Can't seek to 0x%"PRIX64" !\n",(int64_t)(pos+newpos)); | |
366 return 0; | |
367 } |