1 /*****************************************************************************
2 * nsv.c: NullSoft Video demuxer.
3 *****************************************************************************
4 * Copyright (C) 2004 the VideoLAN team
7 * Authors: Laurent Aimar <fenrir@via.ecp.fr>
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.
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.
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., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
22 *****************************************************************************/
24 /*****************************************************************************
26 *****************************************************************************/
27 #include <stdlib.h> /* malloc(), free() */
30 #include <vlc_demux.h>
33 * - implement NSVf parsing (to get meta data)
34 * - implement missing Control (and in the right way)
38 /*****************************************************************************
40 *****************************************************************************/
41 static int Open ( vlc_object_t
* );
42 static void Close ( vlc_object_t
* );
45 set_description( _("NullSoft demuxer" ) );
46 set_capability( "demux2", 10 );
47 set_category( CAT_INPUT
);
48 set_subcategory( SUBCAT_INPUT_DEMUX
);
49 set_callbacks( Open
, Close
);
50 add_shortcut( "nsv" );
53 /*****************************************************************************
55 *****************************************************************************/
59 es_format_t fmt_audio
;
62 es_format_t fmt_video
;
73 static int Demux ( demux_t
*p_demux
);
74 static int Control( demux_t
*p_demux
, int i_query
, va_list args
);
76 static int ReSynch( demux_t
*p_demux
);
78 static int ReadNSVf( demux_t
*p_demux
);
79 static int ReadNSVs( demux_t
*p_demux
);
81 /*****************************************************************************
83 *****************************************************************************/
84 static int Open( vlc_object_t
*p_this
)
86 demux_t
*p_demux
= (demux_t
*)p_this
;
91 if( stream_Peek( p_demux
->s
, &p_peek
, 8 ) < 8 )
94 if( strncmp( (char *)p_peek
, "NSVf", 4 )
95 && strncmp( (char *)p_peek
, "NSVs", 4 ))
97 /* In case we had force this demuxer we try to resynch */
98 if( strcmp( p_demux
->psz_demux
, "nsv" ) || ReSynch( p_demux
) )
104 /* Fill p_demux field */
105 p_demux
->pf_demux
= Demux
;
106 p_demux
->pf_control
= Control
;
107 p_demux
->p_sys
= p_sys
= malloc( sizeof( demux_sys_t
) );
109 es_format_Init( &p_sys
->fmt_audio
, AUDIO_ES
, 0 );
110 p_sys
->p_audio
= NULL
;
112 es_format_Init( &p_sys
->fmt_video
, VIDEO_ES
, 0 );
113 p_sys
->p_video
= NULL
;
115 es_format_Init( &p_sys
->fmt_sub
, SPU_ES
, 0 );
120 p_sys
->i_pcr_inc
= 0;
125 /*****************************************************************************
127 *****************************************************************************/
128 static void Close( vlc_object_t
*p_this
)
130 demux_t
*p_demux
= (demux_t
*)p_this
;
131 demux_sys_t
*p_sys
= p_demux
->p_sys
;
137 /*****************************************************************************
139 *****************************************************************************/
140 static int Demux( demux_t
*p_demux
)
142 demux_sys_t
*p_sys
= p_demux
->p_sys
;
152 if( stream_Peek( p_demux
->s
, &p_peek
, 8 ) < 8 )
154 msg_Warn( p_demux
, "cannot peek" );
158 if( !strncmp( (char *)p_peek
, "NSVf", 4 ) )
160 if( ReadNSVf( p_demux
) )
165 else if( !strncmp( (char *)p_peek
, "NSVs", 4 ) )
167 if( ReadNSVs( p_demux
) )
173 else if( GetWLE( p_peek
) == 0xbeef )
175 /* Next frame of the current NSVs chunk */
176 if( stream_Read( p_demux
->s
, NULL
, 2 ) < 2 )
178 msg_Warn( p_demux
, "cannot read" );
185 msg_Err( p_demux
, "invalid signature 0x%x (%4.4s)", *(uint32_t*)p_peek
, (char*)p_peek
);
186 if( ReSynch( p_demux
) )
193 if( stream_Read( p_demux
->s
, header
, 5 ) < 5 )
195 msg_Warn( p_demux
, "cannot read" );
200 es_out_Control( p_demux
->out
, ES_OUT_SET_PCR
, (int64_t)p_sys
->i_pcr
);
203 i_size
= ( header
[0] >> 4 ) | ( header
[1] << 4 ) | ( header
[2] << 12 );
207 if( (header
[0]&0x0f) != 0x0 )
212 if( stream_Read( p_demux
->s
, aux
, 6 ) < 6 )
214 msg_Warn( p_demux
, "cannot read" );
217 i_aux
= GetWLE( aux
);
218 fcc
= VLC_FOURCC( aux
[2], aux
[3], aux
[4], aux
[5] );
220 msg_Dbg( p_demux
, "Belekas: %d - size=%d fcc=%4.4s",
221 header
[0]&0xf, i_aux
, (char*)&fcc
);
223 if( fcc
== VLC_FOURCC( 'S', 'U', 'B', 'T' ) && i_aux
> 2 )
225 if( p_sys
->p_sub
== NULL
)
227 p_sys
->fmt_sub
.i_codec
= VLC_FOURCC( 's', 'u', 'b', 't' );
228 p_sys
->p_sub
= es_out_Add( p_demux
->out
, &p_sys
->fmt_sub
);
229 es_out_Control( p_demux
->out
, ES_OUT_SET_ES
, p_sys
->p_sub
);
231 stream_Read( p_demux
->s
, NULL
, 2 );
233 if( ( p_frame
= stream_Block( p_demux
->s
, i_aux
- 2 ) ) )
235 uint8_t *p
= p_frame
->p_buffer
;
237 while( p
< &p_frame
->p_buffer
[p_frame
->i_buffer
] && *p
!= 0 )
241 if( *p
== 0 && p
+ 1 < &p_frame
->p_buffer
[p_frame
->i_buffer
] )
243 p_frame
->i_buffer
-= p
+ 1 - p_frame
->p_buffer
;
244 p_frame
->p_buffer
= p
+ 1;
247 /* Skip the first part (it is the language name) */
248 p_frame
->i_pts
= p_sys
->i_pcr
;
249 p_frame
->i_dts
= p_sys
->i_pcr
+ 4000000; /* 4s */
251 es_out_Send( p_demux
->out
, p_sys
->p_sub
, p_frame
);
256 /* We skip this extra data */
257 if( stream_Read( p_demux
->s
, NULL
, i_aux
) < i_aux
)
259 msg_Warn( p_demux
, "cannot read" );
266 /* msg_Dbg( p_demux, "frame video size=%d", i_size ); */
267 if( i_size
> 0 && ( p_frame
= stream_Block( p_demux
->s
, i_size
) ) )
269 p_frame
->i_dts
= p_sys
->i_pcr
;
270 es_out_Send( p_demux
->out
, p_sys
->p_video
, p_frame
);
275 i_size
= header
[3] | ( header
[4] << 8 );
278 /* msg_Dbg( p_demux, "frame audio size=%d", i_size ); */
279 if( p_sys
->fmt_audio
.i_codec
== VLC_FOURCC( 'a', 'r', 'a', 'w' ) )
282 stream_Read( p_demux
->s
, h
, 4 );
284 p_sys
->fmt_audio
.audio
.i_channels
= h
[1];
285 p_sys
->fmt_audio
.audio
.i_rate
= GetWLE( &h
[2] );
289 if( p_sys
->p_audio
== NULL
)
291 p_sys
->p_audio
= es_out_Add( p_demux
->out
, &p_sys
->fmt_audio
);
294 if( ( p_frame
= stream_Block( p_demux
->s
, i_size
) ) )
297 p_frame
->i_pts
= p_sys
->i_pcr
;
298 es_out_Send( p_demux
->out
, p_sys
->p_audio
, p_frame
);
302 p_sys
->i_pcr
+= p_sys
->i_pcr_inc
;
303 if( p_sys
->i_time
>= 0 )
305 p_sys
->i_time
+= p_sys
->i_pcr_inc
;
311 /*****************************************************************************
313 *****************************************************************************/
314 static int Control( demux_t
*p_demux
, int i_query
, va_list args
)
316 demux_sys_t
*p_sys
= p_demux
->p_sys
;
322 case DEMUX_GET_POSITION
:
323 pf
= (double*) va_arg( args
, double* );
324 i64
= stream_Size( p_demux
->s
);
327 *pf
= (double)stream_Tell( p_demux
->s
) / (double)i64
;
335 case DEMUX_SET_POSITION
:
336 f
= (double) va_arg( args
, double );
337 i64
= stream_Size( p_demux
->s
);
339 es_out_Control( p_demux
->out
, ES_OUT_RESET_PCR
);
340 if( stream_Seek( p_demux
->s
, (int64_t)(i64
* f
) ) || ReSynch( p_demux
) )
344 p_sys
->i_time
= -1; /* Invalidate time display */
348 pi64
= (int64_t*)va_arg( args
, int64_t * );
349 if( p_sys
->i_time
< 0 )
354 *pi64
= p_sys
->i_time
;
358 case DEMUX_GET_LENGTH
:
359 pi64
= (int64_t*)va_arg( args
, int64_t * );
360 if( p_sys
->i_mux_rate
> 0 )
362 *pi64
= (int64_t)1000000 * ( stream_Size( p_demux
->s
) / 50 ) / p_sys
->i_mux_rate
;
370 pf
= (double*)va_arg( args
, double * );
371 *pf
= (double)1000000.0 / (double)p_sys
->i_pcr_inc
;
380 /*****************************************************************************
382 *****************************************************************************/
383 static int ReSynch( demux_t
*p_demux
)
389 while( !p_demux
->b_die
)
391 if( ( i_peek
= stream_Peek( p_demux
->s
, &p_peek
, 1024 ) ) < 8 )
397 while( i_skip
< i_peek
- 4 )
399 if( !strncmp( (char *)p_peek
, "NSVf", 4 )
400 || !strncmp( (char *)p_peek
, "NSVs", 4 ) )
404 stream_Read( p_demux
->s
, NULL
, i_skip
);
412 stream_Read( p_demux
->s
, NULL
, i_skip
);
417 /*****************************************************************************
419 *****************************************************************************/
420 static int ReadNSVf( demux_t
*p_demux
)
422 /* demux_sys_t *p_sys = p_demux->p_sys; */
426 msg_Dbg( p_demux
, "new NSVf chunk" );
427 if( stream_Peek( p_demux
->s
, &p
, 8 ) < 8 )
432 i_size
= GetDWLE( &p
[4] );
433 msg_Dbg( p_demux
, " - size=%d", i_size
);
435 return stream_Read( p_demux
->s
, NULL
, i_size
) == i_size
? VLC_SUCCESS
: VLC_EGENERIC
;
437 /*****************************************************************************
439 *****************************************************************************/
440 static int ReadNSVs( demux_t
*p_demux
)
442 demux_sys_t
*p_sys
= p_demux
->p_sys
;
446 if( stream_Read( p_demux
->s
, header
, 19 ) < 19 )
448 msg_Warn( p_demux
, "cannot read" );
453 switch( ( fcc
= VLC_FOURCC( header
[4], header
[5], header
[6], header
[7] ) ) )
455 case VLC_FOURCC( 'V', 'P', '3', ' ' ):
456 case VLC_FOURCC( 'V', 'P', '3', '1' ):
457 fcc
= VLC_FOURCC( 'V', 'P', '3', '1' );
459 case VLC_FOURCC( 'V', 'P', '6', '0' ):
460 case VLC_FOURCC( 'V', 'P', '6', '1' ):
461 case VLC_FOURCC( 'V', 'P', '6', '2' ):
462 case VLC_FOURCC( 'N', 'O', 'N', 'E' ):
465 msg_Warn( p_demux
, "unknown codec %4.4s", (char *)&fcc
);
468 if( fcc
!= VLC_FOURCC( 'N', 'O', 'N', 'E' ) && fcc
!= p_sys
->fmt_video
.i_codec
)
470 es_format_Init( &p_sys
->fmt_video
, VIDEO_ES
, fcc
);
471 p_sys
->fmt_video
.video
.i_width
= GetWLE( &header
[12] );
472 p_sys
->fmt_video
.video
.i_height
= GetWLE( &header
[14] );
475 es_out_Del( p_demux
->out
, p_sys
->p_video
);
477 p_sys
->p_video
= es_out_Add( p_demux
->out
, &p_sys
->fmt_video
);
479 msg_Dbg( p_demux
, " - video `%4.4s' %dx%d",
481 p_sys
->fmt_video
.video
.i_width
,
482 p_sys
->fmt_video
.video
.i_height
);
486 switch( ( fcc
= VLC_FOURCC( header
[8], header
[9], header
[10], header
[11] ) ) )
488 case VLC_FOURCC( 'M', 'P', '3', ' ' ):
489 fcc
= VLC_FOURCC( 'm', 'p', 'g', 'a' );
491 case VLC_FOURCC( 'P', 'C', 'M', ' ' ):
492 fcc
= VLC_FOURCC( 'a', 'r', 'a', 'w' );
494 case VLC_FOURCC( 'A', 'A', 'C', ' ' ):
495 case VLC_FOURCC( 'A', 'A', 'C', 'P' ):
496 fcc
= VLC_FOURCC( 'm', 'p', '4', 'a' );
498 case VLC_FOURCC( 'N', 'O', 'N', 'E' ):
501 msg_Warn( p_demux
, "unknown codec %4.4s", (char *)&fcc
);
505 if( fcc
!= VLC_FOURCC( 'N', 'O', 'N', 'E' ) && fcc
!= p_sys
->fmt_audio
.i_codec
)
507 msg_Dbg( p_demux
, " - audio `%4.4s'", (char*)&fcc
);
511 es_out_Del( p_demux
->out
, p_sys
->p_audio
);
512 p_sys
->p_audio
= NULL
;
514 es_format_Init( &p_sys
->fmt_audio
, AUDIO_ES
, fcc
);
517 if( header
[16]&0x80 )
519 /* Fractional frame rate */
520 switch( header
[16]&0x03 )
523 p_sys
->i_pcr_inc
= 33333; /* 300000/9 */
525 case 1: /* 29.97 fps */
526 p_sys
->i_pcr_inc
= 33367; /* 300300/9 */
529 p_sys
->i_pcr_inc
= 40000; /* 360000/9 */
531 case 3: /* 23.98 fps */
532 p_sys
->i_pcr_inc
= 41700; /* 375300/9 */
536 if( header
[16] < 0xc0 )
537 p_sys
->i_pcr_inc
= p_sys
->i_pcr_inc
* (((header
[16] ^ 0x80) >> 2 ) +1 );
539 p_sys
->i_pcr_inc
= p_sys
->i_pcr_inc
/ (((header
[16] ^ 0xc0) >> 2 ) +1 );
541 else if( header
[16] != 0 )
543 /* Integer frame rate */
544 p_sys
->i_pcr_inc
= 1000000 / header
[16];
548 msg_Dbg( p_demux
, "invalid fps (0x00)" );
549 p_sys
->i_pcr_inc
= 40000;
551 //msg_Dbg( p_demux, " - fps=%.3f", 1000000.0 / (double)p_sys->i_pcr_inc );