mirror of https://github.com/mpv-player/mpv
496 lines
14 KiB
C
496 lines
14 KiB
C
#include "config.h"
|
|
|
|
// Initial draft of my new cache system...
|
|
// Note it runs in 2 processes (using fork()), but doesn't requires locking!!
|
|
// TODO: seeking, data consistency checking
|
|
|
|
#define READ_USLEEP_TIME 10000
|
|
#define FILL_USLEEP_TIME 50000
|
|
#define PREFILL_SLEEP_TIME 200
|
|
#define CONTROL_SLEEP_TIME 0
|
|
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <signal.h>
|
|
#include <sys/types.h>
|
|
#include <unistd.h>
|
|
|
|
#include "osdep/shmem.h"
|
|
#include "osdep/timer.h"
|
|
#if defined(__MINGW32__)
|
|
#include <windows.h>
|
|
static void ThreadProc( void *s );
|
|
#elif defined(__OS2__)
|
|
#define INCL_DOS
|
|
#include <os2.h>
|
|
static void ThreadProc( void *s );
|
|
#elif defined(PTHREAD_CACHE)
|
|
#include <pthread.h>
|
|
static void *ThreadProc(void *s);
|
|
#else
|
|
#include <sys/wait.h>
|
|
#endif
|
|
|
|
#include "mp_msg.h"
|
|
#include "help_mp.h"
|
|
|
|
#include "stream.h"
|
|
#include "cache2.h"
|
|
extern int use_gui;
|
|
|
|
typedef struct {
|
|
// constats:
|
|
unsigned char *buffer; // base pointer of the alllocated buffer memory
|
|
int buffer_size; // size of the alllocated buffer memory
|
|
int sector_size; // size of a single sector (2048/2324)
|
|
int back_size; // we should keep back_size amount of old bytes for backward seek
|
|
int fill_limit; // we should fill buffer only if space>=fill_limit
|
|
int seek_limit; // keep filling cache if distanse is less that seek limit
|
|
// filler's pointers:
|
|
int eof;
|
|
off_t min_filepos; // buffer contain only a part of the file, from min-max pos
|
|
off_t max_filepos;
|
|
off_t offset; // filepos <-> bufferpos offset value (filepos of the buffer's first byte)
|
|
// reader's pointers:
|
|
off_t read_filepos;
|
|
// commands/locking:
|
|
// int seek_lock; // 1 if we will seek/reset buffer, 2 if we are ready for cmd
|
|
// int fifo_flag; // 1 if we should use FIFO to notice cache about buffer reads.
|
|
// callback
|
|
stream_t* stream;
|
|
volatile int control;
|
|
volatile unsigned control_uint_arg;
|
|
volatile double control_double_arg;
|
|
volatile int control_res;
|
|
volatile off_t control_new_pos;
|
|
volatile double stream_time_length;
|
|
} cache_vars_t;
|
|
|
|
static int min_fill=0;
|
|
|
|
int cache_fill_status=0;
|
|
|
|
static void cache_stats(cache_vars_t* s){
|
|
int newb=s->max_filepos-s->read_filepos; // new bytes in the buffer
|
|
mp_msg(MSGT_CACHE,MSGL_INFO,"0x%06X [0x%06X] 0x%06X ",(int)s->min_filepos,(int)s->read_filepos,(int)s->max_filepos);
|
|
mp_msg(MSGT_CACHE,MSGL_INFO,"%3d %% (%3d%%)\n",100*newb/s->buffer_size,100*min_fill/s->buffer_size);
|
|
}
|
|
|
|
static int cache_read(cache_vars_t* s,unsigned char* buf,int size){
|
|
int total=0;
|
|
while(size>0){
|
|
int pos,newb,len;
|
|
|
|
//printf("CACHE2_READ: 0x%X <= 0x%X <= 0x%X \n",s->min_filepos,s->read_filepos,s->max_filepos);
|
|
|
|
if(s->read_filepos>=s->max_filepos || s->read_filepos<s->min_filepos){
|
|
// eof?
|
|
if(s->eof) break;
|
|
// waiting for buffer fill...
|
|
usec_sleep(READ_USLEEP_TIME); // 10ms
|
|
continue; // try again...
|
|
}
|
|
|
|
newb=s->max_filepos-s->read_filepos; // new bytes in the buffer
|
|
if(newb<min_fill) min_fill=newb; // statistics...
|
|
|
|
// printf("*** newb: %d bytes ***\n",newb);
|
|
|
|
pos=s->read_filepos - s->offset;
|
|
if(pos<0) pos+=s->buffer_size; else
|
|
if(pos>=s->buffer_size) pos-=s->buffer_size;
|
|
|
|
if(newb>s->buffer_size-pos) newb=s->buffer_size-pos; // handle wrap...
|
|
if(newb>size) newb=size;
|
|
|
|
// check:
|
|
if(s->read_filepos<s->min_filepos) mp_msg(MSGT_CACHE,MSGL_ERR,"Ehh. s->read_filepos<s->min_filepos !!! Report bug...\n");
|
|
|
|
// len=write(mem,newb)
|
|
//printf("Buffer read: %d bytes\n",newb);
|
|
memcpy(buf,&s->buffer[pos],newb);
|
|
buf+=newb;
|
|
len=newb;
|
|
// ...
|
|
|
|
s->read_filepos+=len;
|
|
size-=len;
|
|
total+=len;
|
|
|
|
}
|
|
cache_fill_status=(s->max_filepos-s->read_filepos)/(s->buffer_size / 100);
|
|
return total;
|
|
}
|
|
|
|
static int cache_fill(cache_vars_t* s){
|
|
int back,back2,newb,space,len,pos;
|
|
off_t read=s->read_filepos;
|
|
|
|
if(read<s->min_filepos || read>s->max_filepos){
|
|
// seek...
|
|
mp_msg(MSGT_CACHE,MSGL_DBG2,"Out of boundaries... seeking to 0x%"PRIX64" \n",(int64_t)read);
|
|
// streaming: drop cache contents only if seeking backward or too much fwd:
|
|
if(s->stream->type!=STREAMTYPE_STREAM ||
|
|
read<s->min_filepos || read>=s->max_filepos+s->seek_limit)
|
|
{
|
|
s->offset= // FIXME!?
|
|
s->min_filepos=s->max_filepos=read; // drop cache content :(
|
|
if(s->stream->eof) stream_reset(s->stream);
|
|
stream_seek(s->stream,read);
|
|
mp_msg(MSGT_CACHE,MSGL_DBG2,"Seek done. new pos: 0x%"PRIX64" \n",(int64_t)stream_tell(s->stream));
|
|
}
|
|
}
|
|
|
|
// calc number of back-bytes:
|
|
back=read - s->min_filepos;
|
|
if(back<0) back=0; // strange...
|
|
if(back>s->back_size) back=s->back_size;
|
|
|
|
// calc number of new bytes:
|
|
newb=s->max_filepos - read;
|
|
if(newb<0) newb=0; // strange...
|
|
|
|
// calc free buffer space:
|
|
space=s->buffer_size - (newb+back);
|
|
|
|
// calc bufferpos:
|
|
pos=s->max_filepos - s->offset;
|
|
if(pos>=s->buffer_size) pos-=s->buffer_size; // wrap-around
|
|
|
|
if(space<s->fill_limit){
|
|
// printf("Buffer is full (%d bytes free, limit: %d)\n",space,s->fill_limit);
|
|
return 0; // no fill...
|
|
}
|
|
|
|
// printf("### read=0x%X back=%d newb=%d space=%d pos=%d\n",read,back,newb,space,pos);
|
|
|
|
// reduce space if needed:
|
|
if(space>s->buffer_size-pos) space=s->buffer_size-pos;
|
|
|
|
// if(space>32768) space=32768; // limit one-time block size
|
|
if(space>4*s->sector_size) space=4*s->sector_size;
|
|
|
|
// if(s->seek_lock) return 0; // FIXME
|
|
|
|
#if 1
|
|
// back+newb+space <= buffer_size
|
|
back2=s->buffer_size-(space+newb); // max back size
|
|
if(s->min_filepos<(read-back2)) s->min_filepos=read-back2;
|
|
#else
|
|
s->min_filepos=read-back; // avoid seeking-back to temp area...
|
|
#endif
|
|
|
|
// ....
|
|
//printf("Buffer fill: %d bytes of %d\n",space,s->buffer_size);
|
|
//len=stream_fill_buffer(s->stream);
|
|
//memcpy(&s->buffer[pos],s->stream->buffer,len); // avoid this extra copy!
|
|
// ....
|
|
len=stream_read(s->stream,&s->buffer[pos],space);
|
|
if(!len) s->eof=1;
|
|
|
|
s->max_filepos+=len;
|
|
if(pos+len>=s->buffer_size){
|
|
// wrap...
|
|
s->offset+=s->buffer_size;
|
|
}
|
|
|
|
return len;
|
|
|
|
}
|
|
|
|
static int cache_execute_control(cache_vars_t *s) {
|
|
int res = 1;
|
|
static unsigned last;
|
|
if (!s->stream->control) {
|
|
s->stream_time_length = 0;
|
|
s->control_new_pos = 0;
|
|
s->control_res = STREAM_UNSUPPORTED;
|
|
s->control = -1;
|
|
return res;
|
|
}
|
|
if (GetTimerMS() - last > 99) {
|
|
double len;
|
|
if (s->stream->control(s->stream, STREAM_CTRL_GET_TIME_LENGTH, &len) == STREAM_OK)
|
|
s->stream_time_length = len;
|
|
else
|
|
s->stream_time_length = 0;
|
|
last = GetTimerMS();
|
|
}
|
|
if (s->control == -1) return res;
|
|
switch (s->control) {
|
|
case STREAM_CTRL_GET_CURRENT_TIME:
|
|
case STREAM_CTRL_SEEK_TO_TIME:
|
|
case STREAM_CTRL_GET_ASPECT_RATIO:
|
|
s->control_res = s->stream->control(s->stream, s->control, &s->control_double_arg);
|
|
break;
|
|
case STREAM_CTRL_SEEK_TO_CHAPTER:
|
|
case STREAM_CTRL_GET_NUM_CHAPTERS:
|
|
case STREAM_CTRL_GET_CURRENT_CHAPTER:
|
|
case STREAM_CTRL_GET_NUM_ANGLES:
|
|
case STREAM_CTRL_GET_ANGLE:
|
|
case STREAM_CTRL_SET_ANGLE:
|
|
s->control_res = s->stream->control(s->stream, s->control, &s->control_uint_arg);
|
|
break;
|
|
case -2:
|
|
res = 0;
|
|
default:
|
|
s->control_res = STREAM_UNSUPPORTED;
|
|
break;
|
|
}
|
|
s->control_new_pos = s->stream->pos;
|
|
s->control = -1;
|
|
return res;
|
|
}
|
|
|
|
static cache_vars_t* cache_init(int size,int sector){
|
|
int num;
|
|
#if !defined(__MINGW32__) && !defined(PTHREAD_CACHE) && !defined(__OS2__)
|
|
cache_vars_t* s=shmem_alloc(sizeof(cache_vars_t));
|
|
#else
|
|
cache_vars_t* s=malloc(sizeof(cache_vars_t));
|
|
#endif
|
|
if(s==NULL) return NULL;
|
|
|
|
memset(s,0,sizeof(cache_vars_t));
|
|
num=size/sector;
|
|
if(num < 16){
|
|
num = 16;
|
|
}//32kb min_size
|
|
s->buffer_size=num*sector;
|
|
s->sector_size=sector;
|
|
#if !defined(__MINGW32__) && !defined(PTHREAD_CACHE) && !defined(__OS2__)
|
|
s->buffer=shmem_alloc(s->buffer_size);
|
|
#else
|
|
s->buffer=malloc(s->buffer_size);
|
|
#endif
|
|
|
|
if(s->buffer == NULL){
|
|
#if !defined(__MINGW32__) && !defined(PTHREAD_CACHE) && !defined(__OS2__)
|
|
shmem_free(s,sizeof(cache_vars_t));
|
|
#else
|
|
free(s);
|
|
#endif
|
|
return NULL;
|
|
}
|
|
|
|
s->fill_limit=8*sector;
|
|
s->back_size=s->buffer_size/2;
|
|
return s;
|
|
}
|
|
|
|
void cache_uninit(stream_t *s) {
|
|
cache_vars_t* c = s->cache_data;
|
|
if(!s->cache_pid) return;
|
|
#if defined(__MINGW32__) || defined(PTHREAD_CACHE) || defined(__OS2__)
|
|
cache_do_control(s, -2, NULL);
|
|
#else
|
|
kill(s->cache_pid,SIGKILL);
|
|
waitpid(s->cache_pid,NULL,0);
|
|
#endif
|
|
if(!c) return;
|
|
#if defined(__MINGW32__) || defined(PTHREAD_CACHE) || defined(__OS2__)
|
|
free(c->stream);
|
|
free(c->buffer);
|
|
free(s->cache_data);
|
|
#else
|
|
shmem_free(c->buffer,c->buffer_size);
|
|
shmem_free(s->cache_data,sizeof(cache_vars_t));
|
|
#endif
|
|
}
|
|
|
|
static void exit_sighandler(int x){
|
|
// close stream
|
|
exit(0);
|
|
}
|
|
|
|
int stream_enable_cache(stream_t *stream,int size,int min,int seek_limit){
|
|
int ss = stream->sector_size ? stream->sector_size : STREAM_BUFFER_SIZE;
|
|
cache_vars_t* s;
|
|
|
|
if (stream->type==STREAMTYPE_STREAM && stream->fd < 0) {
|
|
// The stream has no 'fd' behind it, so is non-cacheable
|
|
mp_msg(MSGT_CACHE,MSGL_STATUS,"\rThis stream is non-cacheable\n");
|
|
return 1;
|
|
}
|
|
|
|
s=cache_init(size,ss);
|
|
if(s == NULL) return 0;
|
|
stream->cache_data=s;
|
|
s->stream=stream; // callback
|
|
s->seek_limit=seek_limit;
|
|
|
|
|
|
//make sure that we won't wait from cache_fill
|
|
//more data than it is alowed to fill
|
|
if (s->seek_limit > s->buffer_size - s->fill_limit ){
|
|
s->seek_limit = s->buffer_size - s->fill_limit;
|
|
}
|
|
if (min > s->buffer_size - s->fill_limit) {
|
|
min = s->buffer_size - s->fill_limit;
|
|
}
|
|
|
|
#if !defined(__MINGW32__) && !defined(PTHREAD_CACHE) && !defined(__OS2__)
|
|
if((stream->cache_pid=fork())){
|
|
#else
|
|
{
|
|
stream_t* stream2=malloc(sizeof(stream_t));
|
|
memcpy(stream2,s->stream,sizeof(stream_t));
|
|
s->stream=stream2;
|
|
#if defined(__MINGW32__)
|
|
stream->cache_pid = _beginthread( ThreadProc, 0, s );
|
|
#elif defined(__OS2__)
|
|
stream->cache_pid = _beginthread( ThreadProc, NULL, 256 * 1024, s );
|
|
#else
|
|
{
|
|
pthread_t tid;
|
|
pthread_create(&tid, NULL, ThreadProc, s);
|
|
stream->cache_pid = 1;
|
|
}
|
|
#endif
|
|
#endif
|
|
// wait until cache is filled at least prefill_init %
|
|
mp_msg(MSGT_CACHE,MSGL_V,"CACHE_PRE_INIT: %"PRId64" [%"PRId64"] %"PRId64" pre:%d eof:%d \n",
|
|
(int64_t)s->min_filepos,(int64_t)s->read_filepos,(int64_t)s->max_filepos,min,s->eof);
|
|
while(s->read_filepos<s->min_filepos || s->max_filepos-s->read_filepos<min){
|
|
mp_tmsg(MSGT_CACHE,MSGL_STATUS,MSGTR_CacheFill,
|
|
100.0*(float)(s->max_filepos-s->read_filepos)/(float)(s->buffer_size),
|
|
(int64_t)s->max_filepos-s->read_filepos
|
|
);
|
|
if(s->eof) break; // file is smaller than prefill size
|
|
if(stream_check_interrupt(PREFILL_SLEEP_TIME))
|
|
return 0;
|
|
}
|
|
mp_msg(MSGT_CACHE,MSGL_STATUS,"\n");
|
|
return 1; // parent exits
|
|
}
|
|
|
|
#if defined(__MINGW32__) || defined(PTHREAD_CACHE) || defined(__OS2__)
|
|
}
|
|
#ifdef PTHREAD_CACHE
|
|
static void *ThreadProc( void *s ){
|
|
#else
|
|
static void ThreadProc( void *s ){
|
|
#endif
|
|
#endif
|
|
|
|
#ifdef CONFIG_GUI
|
|
use_gui = 0; // mp_msg may not use gui stuff in forked code
|
|
#endif
|
|
// cache thread mainloop:
|
|
signal(SIGTERM,exit_sighandler); // kill
|
|
do {
|
|
if(!cache_fill(s)){
|
|
usec_sleep(FILL_USLEEP_TIME); // idle
|
|
}
|
|
// cache_stats(s->cache_data);
|
|
} while (cache_execute_control(s));
|
|
#if defined(__MINGW32__) || defined(__OS2__)
|
|
_endthread();
|
|
#endif
|
|
#ifdef PTHREAD_CACHE
|
|
return NULL;
|
|
#endif
|
|
}
|
|
|
|
int cache_stream_fill_buffer(stream_t *s){
|
|
int len;
|
|
if(s->eof){ s->buf_pos=s->buf_len=0; return 0; }
|
|
if(!s->cache_pid) return stream_fill_buffer(s);
|
|
|
|
// cache_stats(s->cache_data);
|
|
|
|
if(s->pos!=((cache_vars_t*)s->cache_data)->read_filepos) mp_msg(MSGT_CACHE,MSGL_ERR,"!!! read_filepos differs!!! report this bug...\n");
|
|
|
|
len=cache_read(s->cache_data,s->buffer, ((cache_vars_t*)s->cache_data)->sector_size);
|
|
//printf("cache_stream_fill_buffer->read -> %d\n",len);
|
|
|
|
if(len<=0){ s->eof=1; s->buf_pos=s->buf_len=0; return 0; }
|
|
s->buf_pos=0;
|
|
s->buf_len=len;
|
|
s->pos+=len;
|
|
// printf("[%d]",len);fflush(stdout);
|
|
return len;
|
|
|
|
}
|
|
|
|
int cache_stream_seek_long(stream_t *stream,off_t pos){
|
|
cache_vars_t* s;
|
|
off_t newpos;
|
|
if(!stream->cache_pid) return stream_seek_long(stream,pos);
|
|
|
|
s=stream->cache_data;
|
|
// s->seek_lock=1;
|
|
|
|
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);
|
|
|
|
newpos=pos/s->sector_size; newpos*=s->sector_size; // align
|
|
stream->pos=s->read_filepos=newpos;
|
|
s->eof=0; // !!!!!!!
|
|
|
|
cache_stream_fill_buffer(stream);
|
|
|
|
pos-=newpos;
|
|
if(pos>=0 && pos<=stream->buf_len){
|
|
stream->buf_pos=pos; // byte position in sector
|
|
return 1;
|
|
}
|
|
|
|
// stream->buf_pos=stream->buf_len=0;
|
|
// return 1;
|
|
|
|
mp_msg(MSGT_CACHE,MSGL_V,"cache_stream_seek: WARNING! Can't seek to 0x%"PRIX64" !\n",(int64_t)(pos+newpos));
|
|
return 0;
|
|
}
|
|
|
|
int cache_do_control(stream_t *stream, int cmd, void *arg) {
|
|
cache_vars_t* s = stream->cache_data;
|
|
switch (cmd) {
|
|
case STREAM_CTRL_SEEK_TO_TIME:
|
|
s->control_double_arg = *(double *)arg;
|
|
s->control = cmd;
|
|
break;
|
|
case STREAM_CTRL_SEEK_TO_CHAPTER:
|
|
case STREAM_CTRL_SET_ANGLE:
|
|
s->control_uint_arg = *(unsigned *)arg;
|
|
s->control = cmd;
|
|
break;
|
|
// the core might call these every frame, they are too slow for this...
|
|
case STREAM_CTRL_GET_TIME_LENGTH:
|
|
// case STREAM_CTRL_GET_CURRENT_TIME:
|
|
*(double *)arg = s->stream_time_length;
|
|
return s->stream_time_length ? STREAM_OK : STREAM_UNSUPPORTED;
|
|
case STREAM_CTRL_GET_NUM_CHAPTERS:
|
|
case STREAM_CTRL_GET_CURRENT_CHAPTER:
|
|
case STREAM_CTRL_GET_ASPECT_RATIO:
|
|
case STREAM_CTRL_GET_NUM_ANGLES:
|
|
case STREAM_CTRL_GET_ANGLE:
|
|
case -2:
|
|
s->control = cmd;
|
|
break;
|
|
default:
|
|
return STREAM_UNSUPPORTED;
|
|
}
|
|
while (s->control != -1)
|
|
usec_sleep(CONTROL_SLEEP_TIME);
|
|
switch (cmd) {
|
|
case STREAM_CTRL_GET_TIME_LENGTH:
|
|
case STREAM_CTRL_GET_CURRENT_TIME:
|
|
case STREAM_CTRL_GET_ASPECT_RATIO:
|
|
*(double *)arg = s->control_double_arg;
|
|
break;
|
|
case STREAM_CTRL_GET_NUM_CHAPTERS:
|
|
case STREAM_CTRL_GET_CURRENT_CHAPTER:
|
|
case STREAM_CTRL_GET_NUM_ANGLES:
|
|
case STREAM_CTRL_GET_ANGLE:
|
|
*(unsigned *)arg = s->control_uint_arg;
|
|
break;
|
|
case STREAM_CTRL_SEEK_TO_CHAPTER:
|
|
case STREAM_CTRL_SEEK_TO_TIME:
|
|
case STREAM_CTRL_SET_ANGLE:
|
|
stream->pos = s->read_filepos = s->control_new_pos;
|
|
break;
|
|
}
|
|
return s->control_res;
|
|
}
|