video.c: don't display pre-rolled pictures
[vlc.git] / src / input / mem_stream.c
1 /*****************************************************************************
2  * mem_stream.c: stream_t wrapper around memory buffer
3  *****************************************************************************
4  * Copyright (C) 1999-2004 VideoLAN
5  * $Id: stream.c 9390 2004-11-22 09:56:48Z fenrir $
6  *
7  * Authors: Sigmund Augdal <sigmunau@idi.ntnu.no>
8  *
9  * This program is free software; you can redistribute it and/or modify
10  * it under the terms of the GNU General Public License as published by
11  * the Free Software Foundation; either version 2 of the License, or
12  * (at your option) any later version.
13  *
14  * This program is distributed in the hope that it will be useful,
15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
17  * GNU General Public License for more details.
18  *
19  * You should have received a copy of the GNU General Public License
20  * along with this program; if not, write to the Free Software
21  * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111, USA.
22  *****************************************************************************/
23
24 #include <stdlib.h>
25 #include <vlc/vlc.h>
26 #include <vlc/input.h>
27
28 #include "input_internal.h"
29
30 struct stream_sys_t
31 {
32     int64_t     i_pos;      /* Current reading offset */
33     int64_t     i_size;
34     uint8_t    *p_buffer;
35
36 };
37
38 static int AStreamReadMem( stream_t *, void *p_read, int i_read );
39 static int AStreamPeekMem( stream_t *, uint8_t **pp_peek, int i_read );
40 static int AStreamControl( stream_t *, int i_query, va_list );
41
42 /****************************************************************************
43  * stream_MemoryNew: create a stream from a buffer
44  ****************************************************************************/
45 stream_t *__stream_MemoryNew( vlc_object_t *p_this, uint8_t *p_buffer,
46                               int64_t i_size )
47 {
48     stream_t *s = vlc_object_create( p_this, VLC_OBJECT_STREAM );
49     stream_sys_t *p_sys;
50
51     if( !s ) return NULL;
52
53     s->p_sys = p_sys = malloc( sizeof( stream_sys_t ) );
54     p_sys->i_pos = 0;
55     p_sys->i_size = i_size;
56     p_sys->p_buffer = p_buffer;
57
58     s->pf_block   = NULL;
59     s->pf_read    = AStreamReadMem;    /* Set up later */
60     s->pf_peek    = AStreamPeekMem;
61     s->pf_control = AStreamControl;
62     vlc_object_attach( s, p_this );
63
64     return s;
65 }
66
67 void stream_MemoryDelete( stream_t *s, vlc_bool_t b_free_buffer )
68 {
69     if( b_free_buffer ) free( s->p_sys->p_buffer );
70     free( s->p_sys );
71     vlc_object_detach( s );
72     vlc_object_destroy( s );
73 }
74
75 /****************************************************************************
76  * AStreamControl:
77  ****************************************************************************/
78 static int AStreamControl( stream_t *s, int i_query, va_list args )
79 {
80     stream_sys_t *p_sys = s->p_sys;
81
82     vlc_bool_t *p_bool;
83     int64_t    *pi_64, i_64;
84     int        i_int;
85
86     switch( i_query )
87     {
88         case STREAM_GET_SIZE:
89             pi_64 = (int64_t*)va_arg( args, int64_t * );
90             *pi_64 = p_sys->i_size;
91             break;
92
93         case STREAM_CAN_SEEK:
94             p_bool = (vlc_bool_t*)va_arg( args, vlc_bool_t * );
95             *p_bool = VLC_TRUE;
96             break;
97
98         case STREAM_CAN_FASTSEEK:
99             p_bool = (vlc_bool_t*)va_arg( args, vlc_bool_t * );
100             *p_bool = VLC_TRUE;
101             break;
102
103         case STREAM_GET_POSITION:
104             pi_64 = (int64_t*)va_arg( args, int64_t * );
105             *pi_64 = p_sys->i_pos;
106             break;
107
108         case STREAM_SET_POSITION:
109             i_64 = (int64_t)va_arg( args, int64_t );
110             i_64 = __MAX( i_64, 0 );
111             i_64 = __MIN( i_64, s->p_sys->i_size ); 
112             p_sys->i_pos = i_64;
113             break;
114
115         case STREAM_GET_MTU:
116             return VLC_EGENERIC;
117
118         case STREAM_CONTROL_ACCESS:
119             i_int = (int) va_arg( args, int );
120             msg_Err( s, "Hey, what are you thinking ?"
121                      "DON'T USE STREAM_CONTROL_ACCESS !!!" );
122             return VLC_EGENERIC;
123
124         default:
125             msg_Err( s, "invalid stream_vaControl query=0x%x", i_query );
126             return VLC_EGENERIC;
127     }
128     return VLC_SUCCESS;
129 }
130
131 static int AStreamReadMem( stream_t *s, void *p_read, int i_read )
132 {
133     stream_sys_t *p_sys = s->p_sys;
134     int i_res = __MIN( i_read, p_sys->i_size - p_sys->i_pos );
135     memcpy( p_read, p_sys->p_buffer + p_sys->i_pos, i_res );
136     p_sys->i_pos += i_res;
137     return i_res;
138 }
139
140 static int AStreamPeekMem( stream_t *s, uint8_t **pp_peek, int i_read )
141 {
142     stream_sys_t *p_sys = s->p_sys;
143     int i_res = __MIN( i_read, p_sys->i_size - p_sys->i_pos );
144     *pp_peek = p_sys->p_buffer + p_sys->i_pos;
145     return i_res;
146 }