Subversion Repositories Kolibri OS

Rev

Go to most recent revision | Blame | Last modification | View Log | RSS feed

  1. /*
  2.  * Copyright (c) 2012 Clément Bœsch
  3.  *
  4.  * This file is part of FFmpeg.
  5.  *
  6.  * FFmpeg is free software; you can redistribute it and/or
  7.  * modify it under the terms of the GNU Lesser General Public
  8.  * License as published by the Free Software Foundation; either
  9.  * version 2.1 of the License, or (at your option) any later version.
  10.  *
  11.  * FFmpeg is distributed in the hope that it will be useful,
  12.  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13.  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14.  * Lesser General Public License for more details.
  15.  *
  16.  * You should have received a copy of the GNU Lesser General Public
  17.  * License along with FFmpeg; if not, write to the Free Software
  18.  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  19.  */
  20.  
  21. /**
  22.  * @file
  23.  * SubViewer v1 subtitle demuxer
  24.  */
  25.  
  26. #include "avformat.h"
  27. #include "internal.h"
  28. #include "subtitles.h"
  29.  
  30. typedef struct {
  31.     FFDemuxSubtitlesQueue q;
  32. } SubViewer1Context;
  33.  
  34. static int subviewer1_probe(AVProbeData *p)
  35. {
  36.     const unsigned char *ptr = p->buf;
  37.  
  38.     if (strstr(ptr, "******** START SCRIPT ********"))
  39.         return AVPROBE_SCORE_EXTENSION;
  40.     return 0;
  41. }
  42.  
  43. static int subviewer1_read_header(AVFormatContext *s)
  44. {
  45.     int delay = 0;
  46.     AVPacket *sub = NULL;
  47.     SubViewer1Context *subviewer1 = s->priv_data;
  48.     AVStream *st = avformat_new_stream(s, NULL);
  49.  
  50.     if (!st)
  51.         return AVERROR(ENOMEM);
  52.     avpriv_set_pts_info(st, 64, 1, 1);
  53.     st->codec->codec_type = AVMEDIA_TYPE_SUBTITLE;
  54.     st->codec->codec_id   = AV_CODEC_ID_SUBVIEWER1;
  55.  
  56.     while (!url_feof(s->pb)) {
  57.         char line[4096];
  58.         int len = ff_get_line(s->pb, line, sizeof(line));
  59.         int hh, mm, ss;
  60.  
  61.         if (!len)
  62.             break;
  63.  
  64.         if (!strncmp(line, "[DELAY]", 7)) {
  65.             ff_get_line(s->pb, line, sizeof(line));
  66.             sscanf(line, "%d", &delay);
  67.         }
  68.  
  69.         if (sscanf(line, "[%d:%d:%d]", &hh, &mm, &ss) == 3) {
  70.             const int64_t pos = avio_tell(s->pb);
  71.             int64_t pts_start = hh*3600LL + mm*60LL + ss + delay;
  72.  
  73.             len = ff_get_line(s->pb, line, sizeof(line));
  74.             line[strcspn(line, "\r\n")] = 0;
  75.             if (!*line) {
  76.                 if (sub)
  77.                     sub->duration = pts_start - sub->pts;
  78.             } else {
  79.                 sub = ff_subtitles_queue_insert(&subviewer1->q, line, len, 0);
  80.                 if (!sub)
  81.                     return AVERROR(ENOMEM);
  82.                 sub->pos = pos;
  83.                 sub->pts = pts_start;
  84.                 sub->duration = -1;
  85.             }
  86.         }
  87.     }
  88.  
  89.     ff_subtitles_queue_finalize(&subviewer1->q);
  90.     return 0;
  91. }
  92.  
  93. static int subviewer1_read_packet(AVFormatContext *s, AVPacket *pkt)
  94. {
  95.     SubViewer1Context *subviewer1 = s->priv_data;
  96.     return ff_subtitles_queue_read_packet(&subviewer1->q, pkt);
  97. }
  98.  
  99. static int subviewer1_read_seek(AVFormatContext *s, int stream_index,
  100.                                int64_t min_ts, int64_t ts, int64_t max_ts, int flags)
  101. {
  102.     SubViewer1Context *subviewer1 = s->priv_data;
  103.     return ff_subtitles_queue_seek(&subviewer1->q, s, stream_index,
  104.                                    min_ts, ts, max_ts, flags);
  105. }
  106.  
  107. static int subviewer1_read_close(AVFormatContext *s)
  108. {
  109.     SubViewer1Context *subviewer1 = s->priv_data;
  110.     ff_subtitles_queue_clean(&subviewer1->q);
  111.     return 0;
  112. }
  113.  
  114. AVInputFormat ff_subviewer1_demuxer = {
  115.     .name           = "subviewer1",
  116.     .long_name      = NULL_IF_CONFIG_SMALL("SubViewer v1 subtitle format"),
  117.     .priv_data_size = sizeof(SubViewer1Context),
  118.     .read_probe     = subviewer1_probe,
  119.     .read_header    = subviewer1_read_header,
  120.     .read_packet    = subviewer1_read_packet,
  121.     .read_seek2     = subviewer1_read_seek,
  122.     .read_close     = subviewer1_read_close,
  123.     .extensions     = "sub",
  124. };
  125.