src/lib/subrip.cc

   1 /*
   2     Copyright (C) 2014 Carl Hetherington <cth@carlh.net>
   3
   4     This program is free software; you can redistribute it and/or modify
   5     it under the terms of the GNU General Public License as published by
   6     the Free Software Foundation; either version 2 of the License, or
   7     (at your option) any later version.
   8
   9     This program is distributed in the hope that it will be useful,
  10     but WITHOUT ANY WARRANTY; without even the implied warranty of
  11     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12     GNU General Public License for more details.
  13
  14     You should have received a copy of the GNU General Public License
  15     along with this program; if not, write to the Free Software
  16     Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  17
  18 */
  19
  20 #include "subrip.h"
  21 #include "cross.h"
  22 #include "exceptions.h"
  23 #include "subrip_content.h"
  24 #include <sub/subrip_reader.h>
  25 #include <sub/collect.h>
  26 #include <uchardet/uchardet.h>
  27 #include <iostream>
  28
  29 #include "i18n.h"
  30
  31 using std::vector;
  32 using std::cout;
  33 using std::string;
  34 using boost::shared_ptr;
  35
  36 SubRip::SubRip (shared_ptr<const SubRipContent> content)
  37 {
  38         FILE* f = fopen_boost (content->path (0), "r");
  39         if (!f) {
  40                 throw OpenFileError (content->path (0));
  41         }
  42
  43         /* Guess the encoding */
  44         uchardet_t det = uchardet_new ();
  45         char buffer[1024];
  46         while (!feof (f)) {
  47                 int const n = fread (buffer, 1, sizeof (buffer), f);
  48                 if (uchardet_handle_data (det, buffer, n)) {
  49                         break;
  50                 }
  51         }
  52
  53         uchardet_data_end (det);
  54         string charset = uchardet_get_charset (det);
  55         uchardet_delete (det);
  56
  57         if (charset != "UTF-8") {
  58                 throw TextEncodingError (_("unrecognised character set; please use files encoded in UTF-8"));
  59         }
  60
  61         rewind (f);
  62         sub::SubripReader reader (f);
  63         _subtitles = sub::collect<vector<sub::Subtitle> > (reader.subtitles ());
  64 }
  65
  66 ContentTime
  67 SubRip::length () const
  68 {
  69         if (_subtitles.empty ()) {
  70                 return ContentTime ();
  71         }
  72
  73         return ContentTime::from_seconds (_subtitles.back().to.all_as_seconds ());
  74 }