1 /*****************************************************************************
2 * schroedinger.c: Dirac decoder module making use of libschroedinger.
3 * (http://www.bbc.co.uk/rd/projects/dirac/index.shtml)
4 * (http://diracvideo.org)
5 *****************************************************************************
6 * Copyright (C) 2008 the VideoLAN team
9 * Authors: Jonathan Rosser <jonathan.rosser@gmail.com>
10 * David Flynn <davidf at rd dot bbc.co.uk>
12 * This program is free software; you can redistribute it and/or modify
13 * it under the terms of the GNU General Public License as published by
14 * the Free Software Foundation; either version 2 of the License, or
15 * (at your option) any later version.
17 * This program is distributed in the hope that it will be useful,
18 * but WITHOUT ANY WARRANTY; without even the implied warranty of
19 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
20 * GNU General Public License for more details.
22 * You should have received a copy of the GNU General Public License
23 * along with this program; if not, write to the Free Software
24 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
25 *****************************************************************************/
27 /*****************************************************************************
29 *****************************************************************************/
34 #include <vlc_common.h>
35 #include <vlc_plugin.h>
36 #include <vlc_codec.h>
40 #include <schroedinger/schro.h>
42 /*****************************************************************************
44 *****************************************************************************/
45 static int OpenDecoder ( vlc_object_t * );
46 static void CloseDecoder ( vlc_object_t * );
49 set_category( CAT_INPUT )
50 set_subcategory( SUBCAT_INPUT_VCODEC )
51 set_description( N_("Schroedinger video decoder") )
52 set_capability( "decoder", 200 )
53 set_callbacks( OpenDecoder, CloseDecoder )
54 add_shortcut( "schroedinger" )
57 /*****************************************************************************
59 *****************************************************************************/
60 static picture_t *DecodeBlock ( decoder_t *p_dec, block_t **pp_block );
68 /*****************************************************************************
69 * decoder_sys_t : Schroedinger decoder descriptor
70 *****************************************************************************/
77 mtime_t i_frame_pts_delta;
78 SchroDecoder *p_schro;
79 SchroVideoFormat *p_format;
82 /*****************************************************************************
83 * OpenDecoder: probe the decoder and return score
84 *****************************************************************************/
85 static int OpenDecoder( vlc_object_t *p_this )
87 decoder_t *p_dec = (decoder_t*)p_this;
89 SchroDecoder *p_schro;
91 if( p_dec->fmt_in.i_codec != VLC_FOURCC('d','r','a','c') )
96 /* Allocate the memory needed to store the decoder's structure */
97 p_sys = malloc(sizeof(decoder_sys_t));
101 /* Initialise the schroedinger (and hence liboil libraries */
102 /* This does no allocation and is safe to call */
105 /* Initialise the schroedinger decoder */
106 if( !(p_schro = schro_decoder_new()) )
112 p_dec->p_sys = p_sys;
113 p_sys->p_schro = p_schro;
114 p_sys->p_format = NULL;
115 p_sys->i_lastpts = -1;
116 p_sys->i_frame_pts_delta = 0;
118 /* request packetizer */
119 p_dec->b_need_packetized = true;
121 /* Set output properties */
122 p_dec->fmt_out.i_cat = VIDEO_ES;
123 p_dec->fmt_out.i_codec = VLC_FOURCC('I','4','2','0');
126 p_dec->pf_decode_video = DecodeBlock;
131 /*****************************************************************************
132 * SetPictureFormat: Set the decoded picture params to the ones from the stream
133 *****************************************************************************/
134 static void SetVideoFormat( decoder_t *p_dec )
136 decoder_sys_t *p_sys = p_dec->p_sys;
139 p_sys->p_format = schro_decoder_get_video_format(p_sys->p_schro);
140 if( p_sys->p_format == NULL ) return;
142 p_sys->i_frame_pts_delta = INT64_C(1000000)
143 * p_sys->p_format->frame_rate_denominator
144 / p_sys->p_format->frame_rate_numerator;
146 switch( p_sys->p_format->chroma_format )
148 case SCHRO_CHROMA_420: p_dec->fmt_out.i_codec = VLC_FOURCC('I','4','2','0'); break;
149 case SCHRO_CHROMA_422: p_dec->fmt_out.i_codec = VLC_FOURCC('I','4','2','2'); break;
150 case SCHRO_CHROMA_444: p_dec->fmt_out.i_codec = VLC_FOURCC('I','4','4','4'); break;
152 p_dec->fmt_out.i_codec = 0;
156 p_dec->fmt_out.video.i_visible_width =
157 p_dec->fmt_out.video.i_width = p_sys->p_format->width;
159 p_dec->fmt_out.video.i_visible_height =
160 p_dec->fmt_out.video.i_height = p_sys->p_format->height;
162 /* aspect_ratio_[numerator|denominator] describes the pixel aspect ratio */
164 ( p_sys->p_format->aspect_ratio_numerator * p_sys->p_format->width ) /
165 ( p_sys->p_format->aspect_ratio_denominator * p_sys->p_format->height);
167 p_dec->fmt_out.video.i_aspect = VOUT_ASPECT_FACTOR * f_aspect;
169 p_dec->fmt_out.video.i_frame_rate =
170 p_sys->p_format->frame_rate_numerator;
171 p_dec->fmt_out.video.i_frame_rate_base =
172 p_sys->p_format->frame_rate_denominator;
175 /*****************************************************************************
176 * SchroFrameFree: schro_frame callback to release the associated picture_t
177 * When schro_decoder_reset() is called there will be pictures in the
178 * decoding pipeline that need to be released rather than displayed.
179 *****************************************************************************/
180 static void SchroFrameFree( SchroFrame *frame, void *priv)
182 struct picture_free_t *p_free = priv;
187 decoder_DeletePicture( p_free->p_dec, p_free->p_pic );
192 /*****************************************************************************
193 * CreateSchroFrameFromPic: wrap a picture_t in a SchroFrame
194 *****************************************************************************/
195 static SchroFrame *CreateSchroFrameFromPic( decoder_t *p_dec )
197 decoder_sys_t *p_sys = p_dec->p_sys;
198 SchroFrame *p_schroframe = schro_frame_new();
199 picture_t *p_pic = NULL;
200 struct picture_free_t *p_free;
205 p_pic = decoder_NewPicture( p_dec );
210 p_schroframe->format = SCHRO_FRAME_FORMAT_U8_420;
211 if( p_sys->p_format->chroma_format == SCHRO_CHROMA_422 )
213 p_schroframe->format = SCHRO_FRAME_FORMAT_U8_422;
215 else if( p_sys->p_format->chroma_format == SCHRO_CHROMA_444 )
217 p_schroframe->format = SCHRO_FRAME_FORMAT_U8_444;
220 p_schroframe->width = p_sys->p_format->width;
221 p_schroframe->height = p_sys->p_format->height;
223 p_free = malloc( sizeof( *p_free ) );
224 p_free->p_pic = p_pic;
225 p_free->p_dec = p_dec;
226 schro_frame_set_free_callback( p_schroframe, SchroFrameFree, p_free );
228 for( int i=0; i<3; i++ )
230 p_schroframe->components[i].width = p_pic->p[i].i_visible_pitch;
231 p_schroframe->components[i].stride = p_pic->p[i].i_pitch;
232 p_schroframe->components[i].height = p_pic->p[i].i_visible_lines;
233 p_schroframe->components[i].length =
234 p_pic->p[i].i_pitch * p_pic->p[i].i_lines;
235 p_schroframe->components[i].data = p_pic->p[i].p_pixels;
239 p_schroframe->components[i].v_shift =
240 SCHRO_FRAME_FORMAT_V_SHIFT( p_schroframe->format );
241 p_schroframe->components[i].h_shift =
242 SCHRO_FRAME_FORMAT_H_SHIFT( p_schroframe->format );
246 p_pic->b_progressive = !p_sys->p_format->interlaced;
247 p_pic->b_top_field_first = p_sys->p_format->top_field_first;
248 p_pic->i_nb_fields = 2;
253 /*****************************************************************************
254 * SchroBufferFree: schro_buffer callback to release the associated block_t
255 *****************************************************************************/
256 static void SchroBufferFree( SchroBuffer *buf, void *priv )
258 block_t *p_block = priv;
263 block_Release( p_block );
267 /*****************************************************************************
268 * CloseDecoder: decoder destruction
269 *****************************************************************************/
270 static void CloseDecoder( vlc_object_t *p_this )
272 decoder_t *p_dec = (decoder_t *)p_this;
273 decoder_sys_t *p_sys = p_dec->p_sys;
275 schro_decoder_free( p_sys->p_schro );
279 /****************************************************************************
280 * DecodeBlock: the whole thing
281 ****************************************************************************
282 * Blocks need not be Dirac dataunit aligned.
283 * If a block has a PTS signaled, it applies to the first picture at or after p_block
285 * If this function returns a picture (!NULL), it is called again and the
286 * same block is resubmitted. To avoid this, set *pp_block to NULL;
287 * If this function returns NULL, the *pp_block is lost (and leaked).
288 * This function must free all blocks when finished with them.
289 ****************************************************************************/
290 static picture_t *DecodeBlock( decoder_t *p_dec, block_t **pp_block )
292 decoder_sys_t *p_sys = p_dec->p_sys;
295 if( !pp_block ) return NULL;
298 block_t *p_block = *pp_block;
300 /* reset the decoder when seeking as the decode in progress is invalid */
301 /* discard the block as it is just a null magic block */
302 if( p_block->i_flags & (BLOCK_FLAG_DISCONTINUITY|BLOCK_FLAG_CORRUPTED) ) {
303 schro_decoder_reset( p_sys->p_schro );
305 p_sys->i_lastpts = -1;
307 block_Release( p_block );
312 SchroBuffer *p_schrobuffer;
313 p_schrobuffer = schro_buffer_new_with_data( p_block->p_buffer, p_block->i_buffer );
314 p_schrobuffer->free = SchroBufferFree;
315 p_schrobuffer->priv = p_block;
316 mtime_t *p_pts = malloc( sizeof(*p_pts) );
318 *p_pts = p_block->i_pts;
319 /* if this call fails, p_pts is freed automatically */
320 p_schrobuffer->tag = schro_tag_new( p_pts, free );
323 /* this stops the same block being fed back into this function if
324 * we were on the next iteration of this loop to output a picture */
326 schro_decoder_autoparse_push( p_sys->p_schro, p_schrobuffer );
327 /* DO NOT refer to p_block after this point, it may have been freed */
332 SchroFrame *p_schroframe;
334 int state = schro_decoder_autoparse_wait( p_sys->p_schro );
338 case SCHRO_DECODER_FIRST_ACCESS_UNIT:
339 SetVideoFormat( p_dec );
342 case SCHRO_DECODER_NEED_BITS:
345 case SCHRO_DECODER_NEED_FRAME:
346 p_schroframe = CreateSchroFrameFromPic( p_dec );
350 msg_Err( p_dec, "Could not allocate picture for decoder");
354 schro_decoder_add_output_picture( p_sys->p_schro, p_schroframe);
357 case SCHRO_DECODER_OK: {
358 SchroTag *p_tag = schro_decoder_get_picture_tag( p_sys->p_schro );
359 u_pnum = schro_decoder_get_picture_number( p_sys->p_schro );
360 p_schroframe = schro_decoder_pull( p_sys->p_schro );
361 if( !p_schroframe->priv )
363 /* frame can't be one that was allocated by us
364 * -- no private data: discard */
365 if( p_tag ) schro_tag_free( p_tag );
366 schro_frame_unref( p_schroframe );
369 p_pic = ((struct picture_free_t*) p_schroframe->priv)->p_pic;
370 p_schroframe->priv = NULL;
372 /* solve presentation time stamp for picture. If this picture
373 * was not tagged with a pts when presented to decoder, interpolate
375 * This means no need to set p_pic->b_force, as we have a pts on
379 /* free is handled by schro_frame_unref */
380 p_pic->date = *(mtime_t*) p_tag->value;
381 schro_tag_free( p_tag );
382 msg_Err(p_dec, "pts out: %"PRId64, p_pic->date);
384 else if( p_sys->i_lastpts >= 0 )
386 p_pic->date = p_sys->i_lastpts + p_sys->i_frame_pts_delta;
388 msg_Err(p_dec, "no pts");
390 p_sys->i_lastpts = p_pic->date;
392 schro_frame_unref( p_schroframe );
395 case SCHRO_DECODER_EOS:
396 /* NB, the new api will not emit _EOS, it handles the reset internally */
399 case SCHRO_DECODER_ERROR:
400 msg_Err( p_dec, "SCHRO_DECODER_ERROR");