Aegisub/src/subtitles_provider.cpp

109 lines
3.6 KiB
C++
Raw Permalink Normal View History

2014-03-24 15:54:22 +01:00
// Copyright (c) 2014, Thomas Goyne <plorkyeran@aegisub.org>
//
2014-03-24 15:54:22 +01:00
// Permission to use, copy, modify, and distribute this software for any
// purpose with or without fee is hereby granted, provided that the above
// copyright notice and this permission notice appear in all copies.
//
2014-03-24 15:54:22 +01:00
// THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
// WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
// MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
// ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
// WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
// ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
// OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
//
// Aegisub Project http://www.aegisub.org/
2014-03-24 15:54:22 +01:00
#include "include/aegisub/subtitles_provider.h"
#include "ass_dialogue.h"
#include "ass_attachment.h"
#include "ass_file.h"
#include "ass_info.h"
#include "ass_style.h"
2014-03-24 15:54:22 +01:00
#include "factory_manager.h"
#include "options.h"
#include "subtitles_provider_csri.h"
#include "subtitles_provider_libass.h"
2014-03-24 15:54:22 +01:00
namespace {
struct factory {
std::string name;
std::string subtype;
std::unique_ptr<SubtitlesProvider> (*create)(std::string const& subtype, agi::BackgroundRunner *br);
2014-03-24 15:54:22 +01:00
bool hidden;
};
std::vector<factory> const& factories() {
static std::vector<factory> factories;
if (factories.size()) return factories;
#ifdef WITH_CSRI
for (auto const& subtype : csri::List())
factories.push_back(factory{"CSRI/" + subtype, subtype, csri::Create, false});
#endif
factories.push_back(factory{"libass", "", libass::Create, false});
return factories;
}
}
std::vector<std::string> SubtitlesProviderFactory::GetClasses() {
return ::GetClasses(factories());
}
std::unique_ptr<SubtitlesProvider> SubtitlesProviderFactory::GetProvider(agi::BackgroundRunner *br) {
2014-03-24 15:54:22 +01:00
auto preferred = OPT_GET("Subtitle/Provider")->GetString();
auto sorted = GetSorted(factories(), preferred);
std::string error;
2014-03-24 15:54:22 +01:00
for (auto factory : sorted) {
try {
auto provider = factory->create(factory->subtype, br);
if (provider) return provider;
}
catch (agi::UserCancelException const&) { throw; }
catch (agi::Exception const& err) { error += factory->name + ": " + err.GetMessage() + "\n"; }
2014-03-24 15:54:22 +01:00
catch (...) { error += factory->name + ": Unknown error\n"; }
}
throw error;
}
void SubtitlesProvider::LoadSubtitles(AssFile *subs, int time) {
buffer.clear();
auto push_header = [&](const char *str) {
buffer.insert(buffer.end(), str, str + strlen(str));
};
auto push_line = [&](std::string const& str) {
buffer.insert(buffer.end(), &str[0], &str[0] + str.size());
buffer.push_back('\n');
};
push_header("\xEF\xBB\xBF[Script Info]\n");
for (auto const& line : subs->Info)
push_line(line.GetEntryData());
push_header("[V4+ Styles]\n");
for (auto const& line : subs->Styles)
push_line(line.GetEntryData());
if (!subs->Attachments.empty()) {
// TODO: some scripts may have a lot of attachments,
// so ideally we'd want to write only those actually used on the requested video frame,
// but this would require some pre-parsing of the attached font files with FreeType,
// which isn't probably trivial.
push_header("[Fonts]\n");
for (auto const& attachment : subs->Attachments)
if (attachment.Group() == AssEntryGroup::FONT)
push_line(attachment.GetEntryData());
}
push_header("[Events]\n");
for (auto const& line : subs->Events) {
if (!line.Comment && (time < 0 || !(line.Start > time || line.End <= time)))
push_line(line.GetEntryData());
}
LoadSubtitles(&buffer[0], buffer.size());
}