summaryrefslogtreecommitdiff
path: root/src/lib/ffmpeg_subtitle_stream.cc
blob: 466032b37235b4a48c744f42fdd7541cc9f77155 (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
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
/*
    Copyright (C) 2013-2016 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 "ffmpeg_subtitle_stream.h"
#include "raw_convert.h"
#include <libxml++/libxml++.h>
#include <boost/foreach.hpp>
#include <iostream>

using std::string;
using std::map;
using std::list;
using std::cout;

/** Construct a SubtitleStream from a value returned from to_string().
 *  @param t String returned from to_string().
 *  @param v State file version.
 */
FFmpegSubtitleStream::FFmpegSubtitleStream (cxml::ConstNodePtr node, int version)
	: FFmpegStream (node)
{
	if (version == 32) {
		BOOST_FOREACH (cxml::NodePtr i, node->node_children ("Period")) {
			/* In version 32 we assumed that from times were unique, so they weer
			   used as identifiers.
			*/
			add_subtitle (
				raw_convert<string> (i->string_child ("From")),
				ContentTimePeriod (
					ContentTime (i->number_child<ContentTime::Type> ("From")),
					ContentTime (i->number_child<ContentTime::Type> ("To"))
					)
				);
		}
	} else {
		/* In version 33 we use a hash of various parts of the subtitle as the id */
		BOOST_FOREACH (cxml::NodePtr i, node->node_children ("Subtitle")) {
			add_subtitle (
				raw_convert<string> (i->string_child ("Id")),
				ContentTimePeriod (
					ContentTime (i->number_child<ContentTime::Type> ("From")),
					ContentTime (i->number_child<ContentTime::Type> ("To"))
					)
				);
		}
	}
}

void
FFmpegSubtitleStream::as_xml (xmlpp::Node* root) const
{
	FFmpegStream::as_xml (root);

	for (map<string, ContentTimePeriod>::const_iterator i = _subtitles.begin(); i != _subtitles.end(); ++i) {
		xmlpp::Node* node = root->add_child ("Subtitle");
		node->add_child("Id")->add_child_text (i->first);
		node->add_child("From")->add_child_text (raw_convert<string> (i->second.from.get ()));
		node->add_child("To")->add_child_text (raw_convert<string> (i->second.to.get ()));
	}
}

void
FFmpegSubtitleStream::add_subtitle (string id, ContentTimePeriod period)
{
	DCPOMATIC_ASSERT (_subtitles.find (id) == _subtitles.end ());
	_subtitles[id] = period;
}

list<ContentTimePeriod>
FFmpegSubtitleStream::subtitles_during (ContentTimePeriod period, bool starting) const
{
	list<ContentTimePeriod> d;

	/* XXX: inefficient */
	for (map<string, ContentTimePeriod>::const_iterator i = _subtitles.begin(); i != _subtitles.end(); ++i) {
		if ((starting && period.contains (i->second.from)) || (!starting && period.overlaps (i->second))) {
			d.push_back (i->second);
		}
	}

	return d;
}

ContentTime
FFmpegSubtitleStream::find_subtitle_to (string id) const
{
	map<string, ContentTimePeriod>::const_iterator i = _subtitles.find (id);
	DCPOMATIC_ASSERT (i != _subtitles.end ());
	return i->second.to;
}

/** Add some offset to all the times in the stream */
void
FFmpegSubtitleStream::add_offset (ContentTime offset)
{
	for (map<string, ContentTimePeriod>::iterator i = _subtitles.begin(); i != _subtitles.end(); ++i) {
		i->second.from += offset;
		i->second.to += offset;
	}
}