Aegisub/src/ass_dialogue.cpp

289 lines
8.4 KiB
C++
Raw Normal View History

2006-01-16 22:02:54 +01:00
// Copyright (c) 2005, Rodrigo Braz Monteiro
// All rights reserved.
//
// Redistribution and use in source and binary forms, with or without
// modification, are permitted provided that the following conditions are met:
//
// * Redistributions of source code must retain the above copyright notice,
// this list of conditions and the following disclaimer.
// * Redistributions in binary form must reproduce the above copyright notice,
// this list of conditions and the following disclaimer in the documentation
// and/or other materials provided with the distribution.
// * Neither the name of the Aegisub Group nor the names of its contributors
// may be used to endorse or promote products derived from this software
// without specific prior written permission.
//
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
// AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
// IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
// ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
// LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
// CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
// SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
// INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
// CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
// ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
// POSSIBILITY OF SUCH DAMAGE.
//
// Aegisub Project http://www.aegisub.org/
/// @file ass_dialogue.cpp
/// @brief Class for dialogue lines in subtitles
/// @ingroup subs_storage
2006-01-16 22:02:54 +01:00
#include "ass_dialogue.h"
#include "subtitle_format.h"
#include "utils.h"
#include <libaegisub/of_type_adaptor.h>
#include <libaegisub/split.h>
#include <libaegisub/make_unique.h>
#include <boost/algorithm/string/predicate.hpp>
2012-12-30 00:53:56 +01:00
#include <boost/algorithm/string/join.hpp>
#include <boost/algorithm/string/trim.hpp>
2012-12-30 00:53:56 +01:00
#include <boost/lexical_cast.hpp>
#include <boost/regex.hpp>
#include <boost/spirit/include/karma_generate.hpp>
#include <boost/spirit/include/karma_int.hpp>
2012-12-30 00:53:56 +01:00
2012-12-07 17:06:03 +01:00
using namespace boost::adaptors;
static int next_id = 0;
AssDialogue::AssDialogue() {
Id = ++next_id;
2006-01-16 22:02:54 +01:00
}
AssDialogue::AssDialogue(AssDialogue const& that)
: AssDialogueBase(that)
, AssEntryListHook(that)
{
Id = ++next_id;
}
AssDialogue::AssDialogue(AssDialogueBase const& that) : AssDialogueBase(that) { }
2006-01-16 22:02:54 +01:00
AssDialogue::AssDialogue(std::string const& data) {
Id = ++next_id;
Parse(data);
}
AssDialogue::~AssDialogue () { }
class tokenizer {
agi::StringRange str;
agi::split_iterator<agi::StringRange::const_iterator> pos;
public:
tokenizer(agi::StringRange const& str) : str(str) , pos(agi::Split(str, ',')) { }
agi::StringRange next_tok() {
if (pos.eof())
throw SubtitleFormatParseError("Failed parsing line: " + std::string(str.begin(), str.end()));
return *pos++;
}
std::string next_str() { return agi::str(next_tok()); }
std::string next_str_trim() { return agi::str(boost::trim_copy(next_tok())); }
};
2006-01-16 22:02:54 +01:00
void AssDialogue::Parse(std::string const& raw) {
agi::StringRange str;
if (boost::starts_with(raw, "Dialogue:")) {
2006-01-16 22:02:54 +01:00
Comment = false;
str = agi::StringRange(raw.begin() + 10, raw.end());
2006-01-16 22:02:54 +01:00
}
else if (boost::starts_with(raw, "Comment:")) {
2006-01-16 22:02:54 +01:00
Comment = true;
str = agi::StringRange(raw.begin() + 9, raw.end());
2006-01-16 22:02:54 +01:00
}
else
throw SubtitleFormatParseError("Failed parsing line: " + raw);
tokenizer tkn(str);
2006-01-16 22:02:54 +01:00
// Get first token and see if it has "Marked=" in it
auto tmp = tkn.next_str_trim();
bool ssa = boost::istarts_with(tmp, "marked=");
// Get layer number
if (ssa)
Layer = 0;
else
Layer = boost::lexical_cast<int>(tmp);
Start = tkn.next_str_trim();
End = tkn.next_str_trim();
Style = tkn.next_str_trim();
Actor = tkn.next_str_trim();
for (int& margin : Margin)
margin = mid(0, boost::lexical_cast<int>(tkn.next_str()), 9999);
Effect = tkn.next_str_trim();
std::string text{tkn.next_tok().begin(), str.end()};
if (text.size() > 1 && text[0] == '{' && text[1] == '=') {
static const boost::regex extradata_test("^\\{(=\\d+)+\\}");
boost::match_results<std::string::iterator> rematch;
if (boost::regex_search(text.begin(), text.end(), rematch, extradata_test)) {
std::string extradata_str = rematch.str(0);
text = rematch.suffix().str();
static const boost::regex idmatcher("=(\\d+)");
auto start = extradata_str.begin();
auto end = extradata_str.end();
std::vector<uint32_t> ids;
while (boost::regex_search(start, end, rematch, idmatcher)) {
auto id = boost::lexical_cast<uint32_t>(rematch.str(1));
ids.push_back(id);
start = rematch.suffix().first;
}
ExtradataIds = ids;
}
}
Text = text;
2006-01-16 22:02:54 +01:00
}
2014-07-09 00:16:49 +02:00
static void append_int(std::string &str, int v) {
boost::spirit::karma::generate(back_inserter(str), boost::spirit::karma::int_, v);
str += ',';
}
2014-07-09 00:16:49 +02:00
static void append_str(std::string &out, std::string const& str) {
out += str;
out += ',';
}
2014-07-09 00:16:49 +02:00
static void append_unsafe_str(std::string &out, std::string const& str) {
for (auto c : str) {
if (c == ',')
out += ';';
else
out += c;
}
out += ',';
}
std::string AssDialogue::GetEntryData() const {
std::string str = Comment ? "Comment: " : "Dialogue: ";
str.reserve(51 + Style.get().size() + Actor.get().size() + Effect.get().size() + Text.get().size());
append_int(str, Layer);
append_str(str, Start.GetAssFormatted());
append_str(str, End.GetAssFormatted());
append_unsafe_str(str, Style);
append_unsafe_str(str, Actor);
2013-11-21 18:13:36 +01:00
for (auto margin : Margin)
append_int(str, margin);
append_unsafe_str(str, Effect);
if (ExtradataIds.get().size() > 0) {
str += '{';
for (auto id : ExtradataIds.get()) {
str += '=';
boost::spirit::karma::generate(back_inserter(str), boost::spirit::karma::int_, id);
}
str += '}';
}
for (auto c : Text.get()) {
if (c != '\n' && c != '\r')
str += c;
}
return str;
}
std::vector<std::unique_ptr<AssDialogueBlock>> AssDialogue::ParseTags() const {
std::vector<std::unique_ptr<AssDialogueBlock>> Blocks;
2006-01-16 22:02:54 +01:00
// Empty line, make an empty block
if (Text.get().empty()) {
Blocks.push_back(agi::make_unique<AssDialogueBlockPlain>());
return Blocks;
}
2006-01-16 22:02:54 +01:00
int drawingLevel = 0;
std::string const& text(Text.get());
2006-01-16 22:02:54 +01:00
2012-12-30 00:53:56 +01:00
for (size_t len = text.size(), cur = 0; cur < len; ) {
2006-01-16 22:02:54 +01:00
// Overrides block
2012-12-30 00:53:56 +01:00
if (text[cur] == '{') {
size_t end = text.find('}', cur);
// VSFilter requires that override blocks be closed, while libass
// does not. We match VSFilter here.
2012-12-30 00:53:56 +01:00
if (end == std::string::npos)
goto plain;
++cur;
2006-01-16 22:02:54 +01:00
// Get contents of block
2012-12-30 00:53:56 +01:00
std::string work = text.substr(cur, end - cur);
cur = end + 1;
2012-12-30 00:53:56 +01:00
if (work.size() && work.find('\\') == std::string::npos) {
//We've found an override block with no backslashes
//We're going to assume it's a comment and not consider it an override block
Blocks.push_back(agi::make_unique<AssDialogueBlockComment>(work));
}
else {
// Create block
auto block = agi::make_unique<AssDialogueBlockOverride>(work);
block->ParseTags();
// Look for \p in block
for (auto const& tag : block->Tags) {
if (tag.Name == "\\p")
drawingLevel = tag.Params[0].Get<int>(0);
2006-01-16 22:02:54 +01:00
}
Blocks.push_back(std::move(block));
2006-01-16 22:02:54 +01:00
}
continue;
2006-01-16 22:02:54 +01:00
}
2006-01-16 22:02:54 +01:00
// Plain-text/drawing block
plain:
2012-12-30 00:53:56 +01:00
std::string work;
size_t end = text.find('{', cur + 1);
if (end == std::string::npos) {
work = text.substr(cur);
cur = len;
}
2006-01-16 22:02:54 +01:00
else {
2012-12-30 00:53:56 +01:00
work = text.substr(cur, end - cur);
cur = end;
2006-01-16 22:02:54 +01:00
}
if (drawingLevel == 0)
Blocks.push_back(agi::make_unique<AssDialogueBlockPlain>(work));
else
Blocks.push_back(agi::make_unique<AssDialogueBlockDrawing>(work, drawingLevel));
2006-01-16 22:02:54 +01:00
}
return Blocks;
}
void AssDialogue::StripTags() {
Text = GetStrippedText();
2006-01-16 22:02:54 +01:00
}
static std::string get_text(std::unique_ptr<AssDialogueBlock> &d) { return d->GetText(); }
void AssDialogue::UpdateText(std::vector<std::unique_ptr<AssDialogueBlock>>& blocks) {
if (blocks.empty()) return;
Text = join(blocks | transformed(get_text), "");
2006-01-16 22:02:54 +01:00
}
bool AssDialogue::CollidesWith(const AssDialogue *target) const {
2006-01-16 22:02:54 +01:00
if (!target) return false;
return ((Start < target->Start) ? (target->Start < End) : (Start < target->End));
2006-01-16 22:02:54 +01:00
}
2012-12-30 00:53:56 +01:00
static std::string get_text_p(AssDialogueBlock *d) { return d->GetText(); }
std::string AssDialogue::GetStrippedText() const {
auto blocks = ParseTags();
return join(blocks | agi::of_type<AssDialogueBlockPlain>() | transformed(get_text_p), "");
}