summaryrefslogtreecommitdiff
path: root/src/stl_writer.cc
blob: 22667325ba85bbfe6ee56e416e037837adbf4204 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
/*
    Copyright (C) 2014 Carl Hetherington <cth@carlh.net>

    This program is free software; you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation; either version 2 of the License, or
    (at your option) any later version.

    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with this program; if not, write to the Free Software
    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.

*/

#include "stl_writer.h"
#include <boost/optional.hpp>

using std::list;
using std::ostream;
using std::string;
using boost::optional;
using namespace sub;

STLWriter::STLWriter (list<Subtitle> subtitles, ostream& out)
	: Writer (subtitles)
{
	optional<string> font;
	optional<int> font_size;
	bool bold = false;
	bool italic = false;
	bool underline = false;
	int line = 0;
	optional<FrameTime> from;
	optional<FrameTime> to;
	
	for (list<Subtitle>::const_iterator i = subtitles.begin(); i != subtitles.end(); ++i) {
		bool started_new = false;
		if (!font || font.get() != i->font) {
			out << "\n$FontName = " << i->font;
			font = i->font;
			started_new = true;
		}
		if (!font_size || font_size.get() != i->font_size.points.get()) {
			out << "\n$FontSize = " << i->font_size.points.get();
			font_size = i->font_size.points.get();
			started_new = true;
		}
		string text;
		if (bold != i->bold) {
			text += "^B";
			bold = i->bold;
		}
		if (italic != i->italic) {
			text += "^I";
			italic = i->italic;
		}
		if (underline != i->underline) {
			text += "^U";
			underline = i->underline;
		}

		text += i->text;

		if (from && from.get() == i->from.frame.get() && to && to.get() == i->to.frame.get() && !started_new) {
		        for (int j = line; j < i->line; ++j) {
				out << "|";
			}
			out << text;
			line = i->line;
		} else {
			out << "\n" << i->from.frame.get().timecode() << "," << i->to.frame.get().timecode() << "," << text;
			from = i->from.frame.get();
			to = i->to.frame.get();
			line = 0;
		}
	}

	out << "\n";
}