tools/dumpsubs.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 "reader_factory.h"
  21 #include "reader.h"
  22 #include "collect.h"
  23 #include <getopt.h>
  24 #include <boost/filesystem.hpp>
  25 #include <map>
  26
  27 using std::string;
  28 using std::cerr;
  29 using std::cout;
  30 using std::map;
  31 using std::list;
  32 using boost::shared_ptr;
  33 using namespace sub;
  34
  35 static void
  36 help (string n)
  37 {
  38         cerr << "Syntax: " << n << " <file>\n";
  39 }
  40
  41 int
  42 main (int argc, char* argv[])
  43 {
  44         int option_index = 0;
  45         while (1) {
  46                 static struct option long_options[] = {
  47                         { "help", no_argument, 0, 'h'},
  48                         { 0, 0, 0, 0 }
  49                 };
  50
  51                 int c = getopt_long (argc, argv, "h", long_options, &option_index);
  52
  53                 if (c == -1) {
  54                         break;
  55                 }
  56
  57                 switch (c) {
  58                 case 'h':
  59                         help (argv[0]);
  60                         exit (EXIT_SUCCESS);
  61                 }
  62         }
  63
  64         if (argc <= optind || argc > (optind + 1)) {
  65                 help (argv[0]);
  66                 exit (EXIT_FAILURE);
  67         }
  68
  69         if (!boost::filesystem::exists (argv[optind])) {
  70                 cerr << argv[0] << ": file " << argv[optind] << " not found.\n";
  71                 exit (EXIT_FAILURE);
  72         }
  73
  74         shared_ptr<Reader> reader = reader_factory (argv[optind]);
  75         if (!reader) {
  76                 cerr << argv[0] << ": could not read subtitle file " << argv[optind] << "\n";
  77                 exit (EXIT_FAILURE);
  78         }
  79
  80         map<string, string> metadata = reader->metadata ();
  81         for (map<string, string>::const_iterator i = metadata.begin(); i != metadata.end(); ++i) {
  82                 cout << i->first << ": " << i->second << "\n";
  83         }
  84
  85         list<sub::Subtitle> subs = collect<list<sub::Subtitle> > (reader->subtitles ());
  86         int n = 0;
  87         for (list<sub::Subtitle>::const_iterator i = subs.begin(); i != subs.end(); ++i) {
  88                 cout << "Subtitle " << n << " at " << i->from << " -> " << i->to << "\n";
  89                 for (list<sub::Line>::const_iterator j = i->lines.begin(); j != i->lines.end(); ++j) {
  90                         cout << "\t";
  91                         bool italic = false;
  92                         bool underline = false;
  93                         for (list<sub::Block>::const_iterator k = j->blocks.begin(); k != j->blocks.end(); ++k) {
  94                                 if (k->italic && !italic) {
  95                                         cout << "<i>";
  96                                 } else if (italic && !k->italic) {
  97                                         cout << "</i>";
  98                                 }
  99                                 if (k->underline && !underline) {
 100                                         cout << "<u>";
 101                                 } else if (underline && !k->underline) {
 102                                         cout << "</u>";
 103                                 }
 104
 105                                 italic = k->italic;
 106                                 underline = k->underline;
 107
 108                                 cout << k->text;
 109                         }
 110
 111                         if (italic) {
 112                                 cout << "</i>";
 113                         }
 114                         if (underline) {
 115                                 cout << "</u>";
 116                         }
 117                         cout << "\n";
 118                 }
 119
 120                 ++n;
 121         }
 122
 123         return 0;
 124 }