1 /*****************************************************************************
2 * vobsub.c: Demux vobsub files.
3 *****************************************************************************
4 * Copyright (C) 1999-2004 VLC authors and VideoLAN
7 * Authors: Laurent Aimar <fenrir@via.ecp.fr>
8 * Derk-Jan Hartman <hartman at videolan dot org>
10 * This program is free software; you can redistribute it and/or modify it
11 * under the terms of the GNU Lesser General Public License as published by
12 * the Free Software Foundation; either version 2.1 of the License, or
13 * (at your option) any later version.
15 * This program is distributed in the hope that it will be useful,
16 * but WITHOUT ANY WARRANTY; without even the implied warranty of
17 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
18 * GNU Lesser General Public License for more details.
20 * You should have received a copy of the GNU Lesser General Public License
21 * along with this program; if not, write to the Free Software Foundation,
22 * Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
23 *****************************************************************************/
25 /*****************************************************************************
27 *****************************************************************************/
35 #include <vlc_common.h>
36 #include <vlc_plugin.h>
37 #include <vlc_demux.h>
42 #include "subtitle_helper.h"
44 /*****************************************************************************
46 *****************************************************************************/
47 static int Open ( vlc_object_t
*p_this
);
48 static void Close( vlc_object_t
*p_this
);
51 set_description( N_("Vobsub subtitles parser") )
52 set_category( CAT_INPUT
)
53 set_subcategory( SUBCAT_INPUT_DEMUX
)
54 set_capability( "demux", 1 )
56 set_callbacks( Open
, Close
)
58 add_shortcut( "vobsub", "subtitle" )
61 /*****************************************************************************
63 *****************************************************************************/
75 int i_vobsub_location
;
83 int i_current_subtitle
;
85 subtitle_t
*p_subtitles
;
92 int64_t i_next_demux_date
;
96 stream_t
*p_vobsub_stream
;
100 vobsub_track_t
*track
;
102 int i_original_frame_width
;
103 int i_original_frame_height
;
105 uint32_t palette
[16];
109 static int Demux( demux_t
* );
110 static int Control( demux_t
*, int, va_list );
112 static int TextLoad( text_t
*, stream_t
*s
);
113 static void TextUnload( text_t
* );
114 static int ParseVobSubIDX( demux_t
* );
115 static int DemuxVobSub( demux_t
*, block_t
*);
117 /*****************************************************************************
119 *****************************************************************************/
120 static int Open ( vlc_object_t
*p_this
)
122 demux_t
*p_demux
= (demux_t
*)p_this
;
124 char *psz_vobname
, *s
;
126 uint64_t i_read_offset
= 0;
128 if( ( s
= peek_Readline( p_demux
->s
, &i_read_offset
) ) != NULL
)
130 if( !strcasestr( s
, "# VobSub index file" ) )
132 msg_Dbg( p_demux
, "this doesn't seem to be a vobsub file" );
140 msg_Dbg( p_demux
, "could not read vobsub IDX file" );
145 p_demux
->p_sys
= p_sys
= malloc( sizeof( demux_sys_t
) );
146 if( unlikely( !p_sys
) )
150 p_sys
->p_vobsub_stream
= NULL
;
152 p_sys
->track
= malloc( sizeof( vobsub_track_t
) );
153 if( unlikely( !p_sys
->track
) )
155 p_sys
->i_original_frame_width
= -1;
156 p_sys
->i_original_frame_height
= -1;
157 p_sys
->b_palette
= false;
158 memset( p_sys
->palette
, 0, 16 * sizeof( uint32_t ) );
160 /* Load the whole file */
161 TextLoad( &p_sys
->txt
, p_demux
->s
);
164 ParseVobSubIDX( p_demux
);
167 TextUnload( &p_sys
->txt
);
169 /* Find the total length of the vobsubs */
170 if( p_sys
->i_tracks
> 0 )
172 for( int i
= 0; i
< p_sys
->i_tracks
; i
++ )
174 if( p_sys
->track
[i
].i_subtitles
> 1 )
176 if( p_sys
->track
[i
].p_subtitles
[p_sys
->track
[i
].i_subtitles
-1].i_start
> p_sys
->i_length
)
177 p_sys
->i_length
= (int64_t) p_sys
->track
[i
].p_subtitles
[p_sys
->track
[i
].i_subtitles
-1].i_start
+ ( 1 *1000 *1000 );
182 psz_vobname
= strdup( p_demux
->psz_url
);
183 if( psz_vobname
== NULL
)
186 i_len
= strlen( psz_vobname
);
187 if( i_len
>= 4 ) memcpy( psz_vobname
+ i_len
- 4, ".sub", 4 );
190 p_sys
->p_vobsub_stream
= vlc_stream_NewURL( p_demux
, psz_vobname
);
191 if( p_sys
->p_vobsub_stream
== NULL
)
193 msg_Err( p_demux
, "couldn't open .sub Vobsub file: %s",
200 p_demux
->pf_demux
= Demux
;
201 p_demux
->pf_control
= Control
;
206 /* Clean all subs from all tracks */
207 for( int i
= 0; i
< p_sys
->i_tracks
; i
++ )
208 free( p_sys
->track
[i
].p_subtitles
);
209 free( p_sys
->track
);
215 /*****************************************************************************
216 * Close: Close subtitle demux
217 *****************************************************************************/
218 static void Close( vlc_object_t
*p_this
)
220 demux_t
*p_demux
= (demux_t
*)p_this
;
221 demux_sys_t
*p_sys
= p_demux
->p_sys
;
223 if( p_sys
->p_vobsub_stream
)
224 vlc_stream_Delete( p_sys
->p_vobsub_stream
);
226 /* Clean all subs from all tracks */
227 for( int i
= 0; i
< p_sys
->i_tracks
; i
++ )
228 free( p_sys
->track
[i
].p_subtitles
);
229 free( p_sys
->track
);
233 /*****************************************************************************
235 *****************************************************************************/
236 static int Control( demux_t
*p_demux
, int i_query
, va_list args
)
238 demux_sys_t
*p_sys
= p_demux
->p_sys
;
246 *va_arg( args
, bool * ) = true;
249 case DEMUX_GET_LENGTH
:
250 pi64
= va_arg( args
, int64_t * );
251 *pi64
= (int64_t) p_sys
->i_length
;
255 pi64
= va_arg( args
, int64_t * );
256 for( i
= 0; i
< p_sys
->i_tracks
; i
++ )
259 /* Check the ES is selected */
260 es_out_Control( p_demux
->out
, ES_OUT_GET_ES_STATE
,
261 p_sys
->track
[i
].p_es
, &b_selected
);
262 if( b_selected
) break;
264 if( i
< p_sys
->i_tracks
&& p_sys
->track
[i
].i_current_subtitle
< p_sys
->track
[i
].i_subtitles
)
266 *pi64
= p_sys
->track
[i
].p_subtitles
[p_sys
->track
[i
].i_current_subtitle
].i_start
;
272 i64
= va_arg( args
, int64_t );
273 for( i
= 0; i
< p_sys
->i_tracks
; i
++ )
275 p_sys
->track
[i
].i_current_subtitle
= 0;
276 while( p_sys
->track
[i
].i_current_subtitle
< p_sys
->track
[i
].i_subtitles
&&
277 p_sys
->track
[i
].p_subtitles
[p_sys
->track
[i
].i_current_subtitle
].i_start
< i64
)
279 p_sys
->track
[i
].i_current_subtitle
++;
282 if( p_sys
->track
[i
].i_current_subtitle
>= p_sys
->track
[i
].i_subtitles
)
287 case DEMUX_GET_POSITION
:
288 pf
= va_arg( args
, double * );
289 for( i
= 0; i
< p_sys
->i_tracks
; i
++ )
292 /* Check the ES is selected */
293 es_out_Control( p_demux
->out
, ES_OUT_GET_ES_STATE
,
294 p_sys
->track
[i
].p_es
, &b_selected
);
295 if( b_selected
) break;
297 if( p_sys
->track
[i
].i_current_subtitle
>= p_sys
->track
[i
].i_subtitles
)
301 else if( p_sys
->track
[i
].i_subtitles
> 0 )
303 *pf
= (double)p_sys
->track
[i
].p_subtitles
[p_sys
->track
[i
].i_current_subtitle
].i_start
/
304 (double)p_sys
->i_length
;
312 case DEMUX_SET_POSITION
:
313 f
= va_arg( args
, double );
314 i64
= (int64_t) f
* p_sys
->i_length
;
316 for( i
= 0; i
< p_sys
->i_tracks
; i
++ )
318 p_sys
->track
[i
].i_current_subtitle
= 0;
319 while( p_sys
->track
[i
].i_current_subtitle
< p_sys
->track
[i
].i_subtitles
&&
320 p_sys
->track
[i
].p_subtitles
[p_sys
->track
[i
].i_current_subtitle
].i_start
< i64
)
322 p_sys
->track
[i
].i_current_subtitle
++;
324 if( p_sys
->track
[i
].i_current_subtitle
>= p_sys
->track
[i
].i_subtitles
)
329 case DEMUX_SET_NEXT_DEMUX_TIME
:
330 p_sys
->i_next_demux_date
= va_arg( args
, int64_t );
333 case DEMUX_GET_PTS_DELAY
:
336 case DEMUX_GET_TITLE_INFO
:
337 case DEMUX_HAS_UNSUPPORTED_META
:
338 case DEMUX_GET_ATTACHMENTS
:
339 case DEMUX_CAN_RECORD
:
343 msg_Warn( p_demux
, "unknown query in subtitle control" );
348 /*****************************************************************************
349 * Demux: Send subtitle to decoder
350 *****************************************************************************/
351 static int Demux( demux_t
*p_demux
)
353 demux_sys_t
*p_sys
= p_demux
->p_sys
;
357 for( int i
= 0; i
< p_sys
->i_tracks
; i
++ )
359 #define tk p_sys->track[i]
360 if( tk
.i_current_subtitle
>= tk
.i_subtitles
)
363 i_maxdate
= p_sys
->i_next_demux_date
;
364 if( i_maxdate
<= 0 && tk
.i_current_subtitle
< tk
.i_subtitles
)
366 /* Should not happen */
367 i_maxdate
= tk
.p_subtitles
[tk
.i_current_subtitle
].i_start
+ 1;
370 while( tk
.i_current_subtitle
< tk
.i_subtitles
&&
371 tk
.p_subtitles
[tk
.i_current_subtitle
].i_start
< i_maxdate
)
373 int i_pos
= tk
.p_subtitles
[tk
.i_current_subtitle
].i_vobsub_location
;
377 /* first compute SPU size */
378 if( tk
.i_current_subtitle
+ 1 < tk
.i_subtitles
)
380 i_size
= tk
.p_subtitles
[tk
.i_current_subtitle
+1].i_vobsub_location
- i_pos
;
382 if( i_size
<= 0 ) i_size
= 65535; /* Invalid or EOF */
384 /* Seek at the right place */
385 if( vlc_stream_Seek( p_sys
->p_vobsub_stream
, i_pos
) )
388 "cannot seek in the VobSub to the correct time %d", i_pos
);
389 tk
.i_current_subtitle
++;
393 /* allocate a packet */
394 if( ( p_block
= block_Alloc( i_size
) ) == NULL
)
396 tk
.i_current_subtitle
++;
401 i_read
= vlc_stream_Read( p_sys
->p_vobsub_stream
, p_block
->p_buffer
, i_size
);
404 block_Release( p_block
);
405 tk
.i_current_subtitle
++;
408 p_block
->i_buffer
= i_read
;
411 p_block
->i_pts
= VLC_TS_0
+ tk
.p_subtitles
[tk
.i_current_subtitle
].i_start
;
413 /* demux this block */
414 DemuxVobSub( p_demux
, p_block
);
416 block_Release( p_block
);
418 tk
.i_current_subtitle
++;
424 p_sys
->i_next_demux_date
= 0;
429 static int TextLoad( text_t
*txt
, stream_t
*s
)
434 /* load the complete file */
437 char *psz
= vlc_stream_ReadLine( s
);
440 if( psz
== NULL
|| (n
>= INT_MAX
/sizeof(char *)) )
446 ppsz_new
= realloc( lines
, (n
+ 1) * sizeof (char *) );
447 if( ppsz_new
== NULL
)
456 txt
->i_line_count
= n
;
463 static void TextUnload( text_t
*txt
)
465 for( int i
= 0; i
< txt
->i_line_count
; i
++ )
466 free( txt
->line
[i
] );
470 txt
->i_line_count
= 0;
473 static char *TextGetLine( text_t
*txt
)
475 if( txt
->i_line
>= txt
->i_line_count
)
478 return txt
->line
[txt
->i_line
++];
481 static int ParseVobSubIDX( demux_t
*p_demux
)
483 demux_sys_t
*p_sys
= p_demux
->p_sys
;
484 text_t
*txt
= &p_sys
->txt
;
489 if( ( line
= TextGetLine( txt
) ) == NULL
)
491 return( VLC_EGENERIC
);
494 if( *line
== 0 || *line
== '\r' || *line
== '\n' || *line
== '#' )
498 else if( !strncmp( "size:", line
, 5 ) )
500 /* Store the original size of the video */
501 if( vobsub_size_parse( line
, &p_sys
->i_original_frame_width
,
502 &p_sys
->i_original_frame_height
) == VLC_SUCCESS
)
504 msg_Dbg( p_demux
, "original frame size: %dx%d", p_sys
->i_original_frame_width
, p_sys
->i_original_frame_height
);
508 msg_Warn( p_demux
, "reading original frame size failed" );
511 else if( !strncmp( "palette:", line
, 8 ) )
513 if( vobsub_palette_parse( line
, p_sys
->palette
) == VLC_SUCCESS
)
515 p_sys
->b_palette
= true;
516 msg_Dbg( p_demux
, "vobsub palette read" );
520 msg_Warn( p_demux
, "reading original palette failed" );
523 else if( !strncmp( "id:", line
, 3 ) )
525 char language
[33]; /* Usually 2 or 3 letters, sometimes more.
526 Spec (or lack of) doesn't define any limit */
530 /* Lets start a new track */
531 if( sscanf( line
, "id: %32[^ ,], index: %d",
532 language
, &i_track_id
) != 2 )
534 if( sscanf( line
, "id: , index: %d", &i_track_id
) != 1 )
536 msg_Warn( p_demux
, "reading new track failed" );
543 p_sys
->track
= xrealloc( p_sys
->track
,
544 sizeof( vobsub_track_t
) * (p_sys
->i_tracks
+ 1 ) );
547 vobsub_track_t
*current_tk
= &p_sys
->track
[p_sys
->i_tracks
- 1];
548 memset( current_tk
, 0, sizeof( vobsub_track_t
) );
549 current_tk
->i_current_subtitle
= 0;
550 current_tk
->i_subtitles
= 0;
551 current_tk
->p_subtitles
= xmalloc( sizeof( subtitle_t
) );
552 current_tk
->i_track_id
= i_track_id
;
553 current_tk
->i_delay
= (int64_t)0;
555 es_format_Init( &fmt
, SPU_ES
, VLC_CODEC_SPU
);
556 fmt
.subs
.spu
.i_original_frame_width
= p_sys
->i_original_frame_width
;
557 fmt
.subs
.spu
.i_original_frame_height
= p_sys
->i_original_frame_height
;
558 fmt
.psz_language
= language
;
559 if( p_sys
->b_palette
)
561 fmt
.subs
.spu
.palette
[0] = SPU_PALETTE_DEFINED
;
562 memcpy( &fmt
.subs
.spu
.palette
[1], p_sys
->palette
, 16 * sizeof( uint32_t ) );
565 current_tk
->p_es
= es_out_Add( p_demux
->out
, &fmt
);
566 msg_Dbg( p_demux
, "New vobsub track detected: %i [%s]", i_track_id
, language
);
568 else if( !strncmp( line
, "timestamp:", 10 ) )
571 * timestamp: [sign]hh:mm:ss:mss, filepos: loc
572 * loc is the hex location of the spu in the .sub file
574 int h
, m
, s
, ms
, count
, loc
= 0;
576 int64_t i_start
, i_location
= 0;
578 if( p_sys
->i_tracks
> 0 &&
579 sscanf( line
, "timestamp: %d%n:%d:%d:%d, filepos: %x",
580 &h
, &count
, &m
, &s
, &ms
, &loc
) >= 5 )
582 vobsub_track_t
*current_tk
= &p_sys
->track
[p_sys
->i_tracks
- 1];
583 subtitle_t
*current_sub
;
585 if( line
[count
-3] == '-' )
590 i_start
= (int64_t) ( h
* 3600*1000 +
596 current_tk
->i_subtitles
++;
597 current_tk
->p_subtitles
=
598 xrealloc( current_tk
->p_subtitles
,
599 sizeof( subtitle_t
) * (current_tk
->i_subtitles
+ 1 ) );
600 current_sub
= ¤t_tk
->p_subtitles
[current_tk
->i_subtitles
- 1];
602 current_sub
->i_start
= i_start
* i_sign
;
603 current_sub
->i_start
+= current_tk
->i_delay
;
604 current_sub
->i_vobsub_location
= i_location
;
608 msg_Warn( p_demux
, "reading timestamp failed" );
611 else if( !strncasecmp( line
, "delay:", 6 ) )
614 * delay: [sign]hh:mm:ss:mss
616 int h
, m
, s
, ms
, count
= 0;
620 if( p_sys
->i_tracks
> 0 &&
621 sscanf( line
, "%*celay: %d%n:%d:%d:%d",
622 &h
, &count
, &m
, &s
, &ms
) >= 4 )
624 vobsub_track_t
*current_tk
= &p_sys
->track
[p_sys
->i_tracks
- 1];
625 if( line
[count
-3] == '-' )
630 i_gap
= (int64_t) ( h
* 3600*1000 +
635 current_tk
->i_delay
= current_tk
->i_delay
+ (i_gap
* i_sign
);
636 msg_Dbg( p_demux
, "sign: %+d gap: %+"PRId64
" global delay: %+"PRId64
"",
637 i_sign
, i_gap
, current_tk
->i_delay
);
641 msg_Warn( p_demux
, "reading delay failed" );
648 static int DemuxVobSub( demux_t
*p_demux
, block_t
*p_bk
)
650 demux_sys_t
*p_sys
= p_demux
->p_sys
;
651 uint8_t *p
= p_bk
->p_buffer
;
652 uint8_t *p_end
= &p_bk
->p_buffer
[p_bk
->i_buffer
];
655 while( p
+ 6 < p_end
)
657 int i_size
= ps_pkt_size( p
, p_end
- p
);
665 if( i_size
> p_end
- p
)
667 msg_Warn( p_demux
, "broken PES size" );
671 if( p
[0] != 0 || p
[1] != 0 || p
[2] != 0x01 )
673 msg_Warn( p_demux
, "invalid PES" );
679 /* msg_Dbg( p_demux, "we don't need these ps packets (id=0x1%2.2x)", p[3] ); */
685 p_pkt
= block_Alloc( i_size
);
686 if( unlikely(p_pkt
== NULL
) )
688 memcpy( p_pkt
->p_buffer
, p
, i_size
);
691 i_id
= ps_pkt_id( p_pkt
);
692 if( (i_id
&0xffe0) != 0xbd20 ||
693 ps_pkt_parse_pes( VLC_OBJECT(p_demux
), p_pkt
, 1 ) )
695 block_Release( p_pkt
);
699 /* msg_Dbg( p_demux, "SPU track %d size %d", i_spu, i_size ); */
701 for( i
= 0; i
< p_sys
->i_tracks
; i
++ )
703 vobsub_track_t
*p_tk
= &p_sys
->track
[i
];
705 p_pkt
->i_dts
= p_pkt
->i_pts
= p_bk
->i_pts
;
708 if( p_tk
->p_es
&& p_tk
->i_track_id
== i_spu
)
710 es_out_Send( p_demux
->out
, p_tk
->p_es
, p_pkt
);
711 p_bk
->i_pts
= VLC_TS_INVALID
; /*only first packet has a pts */
715 if( i
>= p_sys
->i_tracks
)
717 block_Release( p_pkt
);