forked from mia/Aegisub
108 lines
3.6 KiB
C++
108 lines
3.6 KiB
C++
// Copyright (c) 2014, Thomas Goyne <plorkyeran@aegisub.org>
|
|
//
|
|
// 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.
|
|
//
|
|
// 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/
|
|
|
|
#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"
|
|
#include "factory_manager.h"
|
|
#include "options.h"
|
|
#include "subtitles_provider_csri.h"
|
|
#include "subtitles_provider_libass.h"
|
|
|
|
namespace {
|
|
struct factory {
|
|
std::string name;
|
|
std::string subtype;
|
|
std::unique_ptr<SubtitlesProvider> (*create)(std::string const& subtype, agi::BackgroundRunner *br);
|
|
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) {
|
|
auto preferred = OPT_GET("Subtitle/Provider")->GetString();
|
|
auto sorted = GetSorted(factories(), preferred);
|
|
|
|
std::string error;
|
|
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"; }
|
|
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());
|
|
}
|