Aegisub/aegisub/src/ass_parser.cpp

175 lines
5.1 KiB
C++
Raw Normal View History

2012-10-12 04:12:42 +02:00
// Copyright (c) 2012, 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.
#include "config.h"
#include "ass_parser.h"
#include "ass_attachment.h"
#include "ass_dialogue.h"
#include "ass_file.h"
#include "ass_style.h"
#include "subtitle_format.h"
2012-10-12 04:12:42 +02:00
2012-10-19 17:57:56 +02:00
#ifndef AGI_PRE
#include <wx/log.h>
#endif
2012-10-12 04:12:42 +02:00
AssParser::AssParser(AssFile *target, int version)
: target(target)
, version(version)
, attach(0)
, state(&AssParser::ParseScriptInfoLine)
2012-10-12 04:12:42 +02:00
{
}
AssParser::~AssParser() {
}
void AssParser::ParseAttachmentLine(wxString const& data) {
bool is_filename = data.StartsWith("fontname: ") || data.StartsWith("filename: ");
2012-10-12 04:12:42 +02:00
bool valid_data = data.size() > 0 && data.size() <= 80;
for (size_t i = 0; i < data.size(); ++i) {
if (data[i] < 33 || data[i] >= 97) {
valid_data = false;
break;
2012-10-12 04:12:42 +02:00
}
}
// Data is over, add attachment to the file
if (!valid_data || is_filename) {
attach->Finish();
target->Line.push_back(attach.release());
AddLine(data);
}
else {
attach->AddData(data);
2012-10-12 04:12:42 +02:00
// Done building
if (data.Length() < 80) {
2012-10-12 04:12:42 +02:00
attach->Finish();
target->Line.push_back(attach.release());
}
}
}
void AssParser::ParseScriptInfoLine(wxString const& data) {
// If the first nonblank line isn't a header pretend it starts with [Script Info]
if (target->Line.empty())
target->Line.push_back(new AssEntry("[Script Info]", "[Script Info]"));
if (data.StartsWith(";")) {
// Skip stupid comments added by other programs
// Of course, we'll add our own in place later... ;)
return;
}
if (data.StartsWith("ScriptType:")) {
wxString versionString = data.Mid(11).Trim(true).Trim(false).Lower();
int trueVersion;
if (versionString == "v4.00")
trueVersion = 0;
else if (versionString == "v4.00+")
trueVersion = 1;
else
throw SubtitleFormatParseError("Unknown SSA file format version", 0);
if (trueVersion != version) {
wxLogMessage("Warning: File has the wrong extension.");
version = trueVersion;
2012-10-12 04:12:42 +02:00
}
}
target->Line.push_back(new AssEntry(data, "[Script Info]"));
}
void AssParser::ParseEventLine(wxString const& data) {
if (data.StartsWith("Dialogue:") || data.StartsWith("Comment:"))
target->Line.push_back(new AssDialogue(data));
else if (data.StartsWith("Format:"))
target->Line.push_back(new AssEntry("Format: Layer, Start, End, Style, Name, MarginL, MarginR, MarginV, Effect, Text", "[Events]"));
}
void AssParser::ParseStyleLine(wxString const& data) {
if (data.StartsWith("Style:"))
target->Line.push_back(new AssStyle(data, version));
else if (data.StartsWith("Format:"))
target->Line.push_back(new AssEntry("Format: Name, Fontname, Fontsize, PrimaryColour, SecondaryColour, OutlineColour, BackColour, Bold, Italic, Underline, StrikeOut, ScaleX, ScaleY, Spacing, Angle, BorderStyle, Outline, Shadow, Alignment, MarginL, MarginR, MarginV, Encoding", "[V4+ Styles]"));
}
void AssParser::ParseFontLine(wxString const& data) {
if (data.StartsWith("fontname: ")) {
attach.reset(new AssAttachment(data.Mid(10), "[Fonts]"));
}
}
void AssParser::ParseGraphicsLine(wxString const& data) {
if (data.StartsWith("filename: ")) {
attach.reset(new AssAttachment(data.Mid(10), "[Graphics]"));
}
}
void AssParser::AppendUnknownLine(wxString const& data) {
target->Line.push_back(new AssEntry(data, target->Line.back()->group));
}
void AssParser::AddLine(wxString const& data) {
// Special-case for attachments since a line could theoretically be both a
// valid attachment data line and a valid section header, and if an
// attachment is in progress it needs to be treated as that
if (attach.get()) {
ParseAttachmentLine(data);
return;
}
2012-10-12 04:12:42 +02:00
if (data.empty()) return;
// Section header
if (data[0] == '[' && data.Last() == ']') {
// Ugly hacks to allow intermixed v4 and v4+ style sections
const wxString low = data.Lower();
wxString header = data;
if (low == "[v4 styles]") {
header = "[V4+ Styles]";
version = 0;
state = &AssParser::ParseStyleLine;
2012-10-12 04:12:42 +02:00
}
else if (low == "[v4+ styles]") {
header = "[V4+ Styles]";
version = 1;
state = &AssParser::ParseStyleLine;
2012-10-12 04:12:42 +02:00
}
else if (low == "[events]") {
state = &AssParser::ParseEventLine;
2012-10-12 04:12:42 +02:00
}
else if (low == "[script info]") {
state = &AssParser::ParseScriptInfoLine;
2012-10-12 04:12:42 +02:00
}
else if (low == "[graphics]") {
state = &AssParser::ParseGraphicsLine;
}
else if (low == "[fonts]") {
state = &AssParser::ParseFontLine;
}
else {
state = &AssParser::AppendUnknownLine;
2012-10-12 04:12:42 +02:00
}
target->Line.push_back(new AssEntry(header, header));
2012-10-12 04:12:42 +02:00
return;
}
(this->*state)(data);
2012-10-12 04:12:42 +02:00
}