1 /*****************************************************************************
2 * rawvideo.c: Pseudo video decoder/packetizer for raw video data
3 *****************************************************************************
4 * Copyright (C) 2001, 2002 VLC authors and VideoLAN
7 * Authors: Laurent Aimar <fenrir@via.ecp.fr>
9 * This program is free software; you can redistribute it and/or modify it
10 * under the terms of the GNU Lesser General Public License as published by
11 * the Free Software Foundation; either version 2.1 of the License, or
12 * (at your option) any later version.
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 Lesser General Public License for more details.
19 * You should have received a copy of the GNU Lesser General Public License
20 * along with this program; if not, write to the Free Software Foundation,
21 * Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
22 *****************************************************************************/
24 /*****************************************************************************
26 *****************************************************************************/
31 #include <vlc_common.h>
32 #include <vlc_plugin.h>
33 #include <vlc_codec.h>
35 /*****************************************************************************
36 * decoder_sys_t : raw video decoder descriptor
37 *****************************************************************************/
48 plane_t planes
[PICTURE_PLANE_MAX
];
56 /****************************************************************************
58 ****************************************************************************/
59 static int OpenDecoder ( vlc_object_t
* );
60 static int OpenPacketizer( vlc_object_t
* );
61 static void CloseDecoder ( vlc_object_t
* );
63 static void *DecodeBlock ( decoder_t
*, block_t
** );
65 static picture_t
*DecodeFrame( decoder_t
*, block_t
* );
66 static block_t
*SendFrame ( decoder_t
*, block_t
* );
68 /*****************************************************************************
70 *****************************************************************************/
72 set_description( N_("Pseudo raw video decoder") )
73 set_capability( "decoder", 50 )
74 set_category( CAT_INPUT
)
75 set_subcategory( SUBCAT_INPUT_VCODEC
)
76 set_callbacks( OpenDecoder
, CloseDecoder
)
79 set_description( N_("Pseudo raw video packetizer") )
80 set_capability( "packetizer", 100 )
81 set_callbacks( OpenPacketizer
, CloseDecoder
)
84 /*****************************************************************************
85 * OpenDecoder: probe the decoder and return score
86 *****************************************************************************/
87 static int OpenDecoder( vlc_object_t
*p_this
)
89 decoder_t
*p_dec
= (decoder_t
*)p_this
;
92 switch( p_dec
->fmt_in
.i_codec
)
111 case VLC_CODEC_I422_10L
:
112 case VLC_CODEC_I422_10B
:
121 case VLC_CODEC_RGB32
:
122 case VLC_CODEC_RGB24
:
123 case VLC_CODEC_RGB16
:
124 case VLC_CODEC_RGB15
:
134 /* Allocate the memory needed to store the decoder's structure */
135 if( ( p_dec
->p_sys
= p_sys
=
136 (decoder_sys_t
*)malloc(sizeof(decoder_sys_t
)) ) == NULL
)
139 p_dec
->p_sys
->b_packetizer
= false;
140 p_sys
->b_invert
= false;
142 if( (int)p_dec
->fmt_in
.video
.i_height
< 0 )
144 /* Frames are coded from bottom to top */
145 p_dec
->fmt_in
.video
.i_height
=
146 (unsigned int)(-(int)p_dec
->fmt_in
.video
.i_height
);
147 p_sys
->b_invert
= true;
150 if( p_dec
->fmt_in
.video
.i_width
<= 0 || p_dec
->fmt_in
.video
.i_height
<= 0 )
152 msg_Err( p_dec
, "invalid display size %dx%d",
153 p_dec
->fmt_in
.video
.i_width
, p_dec
->fmt_in
.video
.i_height
);
157 es_format_Copy( &p_dec
->fmt_out
, &p_dec
->fmt_in
);
159 date_Init( &p_sys
->pts
, p_dec
->fmt_out
.video
.i_frame_rate
,
160 p_dec
->fmt_out
.video
.i_frame_rate_base
);
161 if( p_dec
->fmt_out
.video
.i_frame_rate
== 0 ||
162 p_dec
->fmt_out
.video
.i_frame_rate_base
== 0)
164 msg_Warn( p_dec
, "invalid frame rate %d/%d, using 25 fps instead",
165 p_dec
->fmt_out
.video
.i_frame_rate
,
166 p_dec
->fmt_out
.video
.i_frame_rate_base
);
167 date_Init( &p_sys
->pts
, 25, 1 );
170 /* Find out p_vdec->i_raw_size */
171 video_format_Setup( &p_dec
->fmt_out
.video
, p_dec
->fmt_in
.i_codec
,
172 p_dec
->fmt_in
.video
.i_width
,
173 p_dec
->fmt_in
.video
.i_height
,
174 p_dec
->fmt_in
.video
.i_sar_num
,
175 p_dec
->fmt_in
.video
.i_sar_den
);
177 picture_Setup( &picture
, p_dec
->fmt_out
.i_codec
,
178 p_dec
->fmt_in
.video
.i_width
,
179 p_dec
->fmt_in
.video
.i_height
, 0, 1 );
180 p_sys
->i_raw_size
= 0;
181 for( int i
= 0; i
< picture
.i_planes
; i
++ )
183 p_sys
->i_raw_size
+= picture
.p
[i
].i_visible_pitch
*
184 picture
.p
[i
].i_visible_lines
;
185 p_sys
->planes
[i
] = picture
.p
[i
];
188 if( !p_dec
->fmt_in
.video
.i_sar_num
|| !p_dec
->fmt_in
.video
.i_sar_den
)
190 p_dec
->fmt_out
.video
.i_sar_num
= 1;
191 p_dec
->fmt_out
.video
.i_sar_den
= 1;
195 p_dec
->pf_decode_video
= (picture_t
*(*)(decoder_t
*, block_t
**))
197 p_dec
->pf_packetize
= (block_t
*(*)(decoder_t
*, block_t
**))
203 static int OpenPacketizer( vlc_object_t
*p_this
)
205 decoder_t
*p_dec
= (decoder_t
*)p_this
;
207 int i_ret
= OpenDecoder( p_this
);
209 if( i_ret
== VLC_SUCCESS
) p_dec
->p_sys
->b_packetizer
= true;
214 /****************************************************************************
215 * DecodeBlock: the whole thing
216 ****************************************************************************
217 * This function must be fed with complete frames.
218 ****************************************************************************/
219 static void *DecodeBlock( decoder_t
*p_dec
, block_t
**pp_block
)
221 decoder_sys_t
*p_sys
= p_dec
->p_sys
;
225 if( !pp_block
|| !*pp_block
) return NULL
;
230 if( p_block
->i_pts
<= VLC_TS_INVALID
&& p_block
->i_dts
<= VLC_TS_INVALID
&&
231 !date_Get( &p_sys
->pts
) )
233 /* We've just started the stream, wait for the first PTS. */
234 block_Release( p_block
);
238 /* Date management: If there is a pts avaliable, use that. */
239 if( p_block
->i_pts
> VLC_TS_INVALID
)
241 date_Set( &p_sys
->pts
, p_block
->i_pts
);
243 else if( p_block
->i_dts
> VLC_TS_INVALID
)
245 /* NB, davidf doesn't quite agree with this in general, it is ok
246 * for rawvideo since it is in order (ie pts=dts), however, it
247 * may not be ok for an out-of-order codec, so don't copy this
248 * without thinking */
249 date_Set( &p_sys
->pts
, p_block
->i_dts
);
252 if( p_block
->i_buffer
< p_sys
->i_raw_size
)
254 msg_Warn( p_dec
, "invalid frame size (%zu < %zu)",
255 p_block
->i_buffer
, p_sys
->i_raw_size
);
257 block_Release( p_block
);
261 if( p_sys
->b_packetizer
)
263 p_buf
= SendFrame( p_dec
, p_block
);
267 p_buf
= DecodeFrame( p_dec
, p_block
);
270 /* Date management: 1 frame per packet */
271 date_Increment( &p_sys
->pts
, 1 );
277 /*****************************************************************************
279 *****************************************************************************/
280 static void FillPicture( decoder_t
*p_dec
, block_t
*p_block
, picture_t
*p_pic
)
283 decoder_sys_t
*p_sys
= p_dec
->p_sys
;
284 uint8_t *p_src
= p_block
->p_buffer
;
286 for( i_plane
= 0; i_plane
< p_pic
->i_planes
; i_plane
++ )
288 int i_pitch
= p_pic
->p
[i_plane
].i_pitch
;
289 int i_visible_pitch
= p_sys
->planes
[i_plane
].i_visible_pitch
;
290 int i_visible_lines
= p_sys
->planes
[i_plane
].i_visible_lines
;
291 uint8_t *p_dst
= p_pic
->p
[i_plane
].p_pixels
;
292 uint8_t *p_dst_end
= p_dst
+i_pitch
*i_visible_lines
;
294 if( p_sys
->b_invert
)
295 for( p_dst_end
-= i_pitch
; p_dst
<= p_dst_end
;
296 p_dst_end
-= i_pitch
, p_src
+= i_visible_pitch
)
297 memcpy( p_dst_end
, p_src
, i_visible_pitch
);
299 for( ; p_dst
< p_dst_end
;
300 p_dst
+= i_pitch
, p_src
+= i_visible_pitch
)
301 memcpy( p_dst
, p_src
, i_visible_pitch
);
305 /*****************************************************************************
306 * DecodeFrame: decodes a video frame.
307 *****************************************************************************/
308 static picture_t
*DecodeFrame( decoder_t
*p_dec
, block_t
*p_block
)
310 decoder_sys_t
*p_sys
= p_dec
->p_sys
;
313 /* Get a new picture */
314 p_pic
= decoder_NewPicture( p_dec
);
317 block_Release( p_block
);
321 FillPicture( p_dec
, p_block
, p_pic
);
323 p_pic
->date
= date_Get( &p_sys
->pts
);
324 if( p_block
->i_flags
& BLOCK_FLAG_INTERLACED_MASK
)
326 p_pic
->b_progressive
= false;
327 p_pic
->i_nb_fields
= 2;
328 if( p_block
->i_flags
& BLOCK_FLAG_TOP_FIELD_FIRST
)
329 p_pic
->b_top_field_first
= true;
331 p_pic
->b_top_field_first
= false;
334 p_pic
->b_progressive
= true;
336 block_Release( p_block
);
340 /*****************************************************************************
341 * SendFrame: send a video frame to the stream output.
342 *****************************************************************************/
343 static block_t
*SendFrame( decoder_t
*p_dec
, block_t
*p_block
)
345 decoder_sys_t
*p_sys
= p_dec
->p_sys
;
347 p_block
->i_dts
= p_block
->i_pts
= date_Get( &p_sys
->pts
);
349 if( p_sys
->b_invert
)
352 uint8_t *p_tmp
, *p_pixels
;
355 /* Fill in picture_t fields */
356 picture_Setup( &pic
, p_dec
->fmt_out
.i_codec
,
357 p_dec
->fmt_out
.video
.i_width
,
358 p_dec
->fmt_out
.video
.i_height
, 0, 1 );
362 msg_Err( p_dec
, "unsupported chroma" );
366 p_tmp
= malloc( pic
.p
[0].i_pitch
);
369 p_pixels
= p_block
->p_buffer
;
370 for( i
= 0; i
< pic
.i_planes
; i
++ )
372 int i_pitch
= pic
.p
[i
].i_pitch
;
373 uint8_t *p_top
= p_pixels
;
374 uint8_t *p_bottom
= p_pixels
+ i_pitch
*
375 (pic
.p
[i
].i_visible_lines
- 1);
377 for( j
= 0; j
< pic
.p
[i
].i_visible_lines
/ 2; j
++ )
379 memcpy( p_tmp
, p_bottom
, pic
.p
[i
].i_visible_pitch
);
380 memcpy( p_bottom
, p_top
, pic
.p
[i
].i_visible_pitch
);
381 memcpy( p_top
, p_tmp
, pic
.p
[i
].i_visible_pitch
);
386 p_pixels
+= i_pitch
* pic
.p
[i
].i_lines
;
394 /*****************************************************************************
395 * CloseDecoder: decoder destruction
396 *****************************************************************************/
397 static void CloseDecoder( vlc_object_t
*p_this
)
399 decoder_t
*p_dec
= (decoder_t
*)p_this
;
400 free( p_dec
->p_sys
);