libsub
collect.h
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 #ifndef LIBSUB_COLLECT_H
21 #define LIBSUB_COLLECT_H
22 
23 #include "subtitle.h"
24 #include "raw_subtitle.h"
25 #include <algorithm>
26 
27 namespace sub {
28 
32 template <class T>
33 T
34 collect (std::vector<RawSubtitle> raw)
35 {
36  std::stable_sort (raw.begin(), raw.end());
37 
38  T out;
39 
40  boost::optional<Subtitle> current;
41  for (auto const& i: raw) {
42  if (current && current->same_metadata(i)) {
43  /* This RawSubtitle can be added to current... */
44  if (!current->lines.empty() && current->lines.back().same_metadata(i)) {
45  /* ... and indeed to its last line */
46  current->lines.back().blocks.push_back(Block(i));
47  } else {
48  /* ... as a new line */
49  current->lines.push_back(Line(i));
50  }
51  } else {
52  /* We must start a new Subtitle */
53  if (current) {
54  out.push_back (current.get ());
55  }
56  current = Subtitle (i);
57  }
58  }
59 
60  if (current) {
61  out.push_back (current.get ());
62  }
63 
64  return out;
65 }
66 
67 }
68 
69 #endif