forked from mia/Aegisub
305 lines
8.5 KiB
C++
305 lines
8.5 KiB
C++
// Copyright (c) 2007, 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/
|
|
//
|
|
// $Id$
|
|
|
|
/// @file subtitle_format_ttxt.cpp
|
|
/// @brief Reading/writing MPEG-4 Timed Text subtitles in TTXT XML format
|
|
/// @ingroup subtitle_io
|
|
///
|
|
|
|
#include "config.h"
|
|
|
|
#include "subtitle_format_ttxt.h"
|
|
|
|
#include "ass_dialogue.h"
|
|
#include "ass_file.h"
|
|
#include "ass_time.h"
|
|
#include "compat.h"
|
|
#include "main.h"
|
|
|
|
DEFINE_SIMPLE_EXCEPTION(TTXTParseError, SubtitleFormatParseError, "subtitle_io/parse/ttxt")
|
|
|
|
TTXTSubtitleFormat::TTXTSubtitleFormat()
|
|
: SubtitleFormat("MPEG-4 Streaming Text")
|
|
{
|
|
}
|
|
|
|
wxArrayString TTXTSubtitleFormat::GetReadWildcards() const {
|
|
wxArrayString formats;
|
|
formats.Add("ttxt");
|
|
return formats;
|
|
}
|
|
|
|
wxArrayString TTXTSubtitleFormat::GetWriteWildcards() const {
|
|
return GetReadWildcards();
|
|
}
|
|
|
|
void TTXTSubtitleFormat::ReadFile(wxString const& filename, wxString const& forceEncoding) {
|
|
LoadDefault(false);
|
|
|
|
// Load XML document
|
|
wxXmlDocument doc;
|
|
if (!doc.Load(filename)) throw TTXTParseError("Failed loading TTXT XML file.", 0);
|
|
|
|
// Check root node name
|
|
if (doc.GetRoot()->GetName() != "TextStream") throw TTXTParseError("Invalid TTXT file.", 0);
|
|
|
|
// Check version
|
|
wxString verStr = doc.GetRoot()->GetAttribute("version", "");
|
|
version = -1;
|
|
if (verStr == "1.0") version = 0;
|
|
else if (verStr == "1.1") version = 1;
|
|
else throw TTXTParseError("Unknown TTXT version: " + STD_STR(verStr), 0);
|
|
|
|
// Get children
|
|
diag = NULL;
|
|
wxXmlNode *child = doc.GetRoot()->GetChildren();
|
|
int lines = 0;
|
|
while (child) {
|
|
// Line
|
|
if (child->GetName() == "TextSample") {
|
|
if (ProcessLine(child)) lines++;
|
|
}
|
|
|
|
// Header
|
|
else if (child->GetName() == "TextStreamHeader") {
|
|
ProcessHeader(child);
|
|
}
|
|
|
|
// Proceed to next child
|
|
child = child->GetNext();
|
|
}
|
|
|
|
// No lines?
|
|
if (lines == 0) {
|
|
AssDialogue *line = new AssDialogue();
|
|
line->group = "[Events]";
|
|
line->Style = "Default";
|
|
line->Start = 0;
|
|
line->End = 5000;
|
|
Line->push_back(line);
|
|
}
|
|
}
|
|
|
|
bool TTXTSubtitleFormat::ProcessLine(wxXmlNode *node) {
|
|
// Get time
|
|
wxString sampleTime = node->GetAttribute("sampleTime", "00:00:00.000");
|
|
AssTime time;
|
|
time.ParseASS(sampleTime);
|
|
|
|
// Set end time of last line
|
|
if (diag) diag->End = time;
|
|
diag = NULL;
|
|
|
|
// Get text
|
|
wxString text;
|
|
if (version == 0) text = node->GetAttribute("text", "");
|
|
else text = node->GetNodeContent();
|
|
|
|
// Create line
|
|
if (!text.IsEmpty()) {
|
|
// Create dialogue
|
|
diag = new AssDialogue();
|
|
diag->Start = time;
|
|
diag->End = 36000000-10;
|
|
diag->group = "[Events]";
|
|
diag->Style = "Default";
|
|
diag->Comment = false;
|
|
|
|
// Process text for 1.0
|
|
if (version == 0) {
|
|
wxString finalText;
|
|
finalText.Alloc(text.Length());
|
|
bool in = false;
|
|
bool first = true;
|
|
for (size_t i=0;i<text.Length();i++) {
|
|
if (text[i] == '\'') {
|
|
if (!in && !first) finalText += "\\N";
|
|
first = false;
|
|
in = !in;
|
|
}
|
|
else if (in) finalText += text[i];
|
|
}
|
|
diag->Text = finalText;
|
|
}
|
|
|
|
// Process text for 1.1
|
|
else {
|
|
text.Replace("\r", "");
|
|
text.Replace("\n", "\\N");
|
|
diag->Text = text;
|
|
}
|
|
|
|
// Insert dialogue
|
|
Line->push_back(diag);
|
|
return true;
|
|
}
|
|
|
|
else return false;
|
|
}
|
|
|
|
void TTXTSubtitleFormat::ProcessHeader(wxXmlNode *node) {
|
|
// TODO
|
|
}
|
|
|
|
void TTXTSubtitleFormat::WriteFile(wxString const& filename, wxString const& encoding) {
|
|
// Convert to TTXT
|
|
CreateCopy();
|
|
ConvertToTTXT();
|
|
|
|
// Create XML structure
|
|
wxXmlDocument doc;
|
|
wxXmlNode *root = new wxXmlNode(NULL, wxXML_ELEMENT_NODE, "TextStream");
|
|
root->AddAttribute("version", "1.1");
|
|
doc.SetRoot(root);
|
|
|
|
// Create header
|
|
WriteHeader(root);
|
|
|
|
// Create lines
|
|
int i=1;
|
|
using std::list;
|
|
prev = NULL;
|
|
for (list<AssEntry*>::iterator cur=Line->begin();cur!=Line->end();cur++) {
|
|
AssDialogue *current = dynamic_cast<AssDialogue*>(*cur);
|
|
if (current && !current->Comment) {
|
|
WriteLine(root, current);
|
|
i++;
|
|
}
|
|
else
|
|
throw TTXTParseError("Unexpected line type in TTXT file", 0);
|
|
}
|
|
|
|
// Save XML
|
|
//prevNode->SetNext(NULL);
|
|
doc.Save(filename);
|
|
|
|
// Clear
|
|
ClearCopy();
|
|
}
|
|
|
|
void TTXTSubtitleFormat::WriteHeader(wxXmlNode *root) {
|
|
// Write stream header
|
|
wxXmlNode *node = new wxXmlNode(wxXML_ELEMENT_NODE, "TextStreamHeader");
|
|
node->AddAttribute("width", "400");
|
|
node->AddAttribute("height", "60");
|
|
node->AddAttribute("layer", "0");
|
|
node->AddAttribute("translation_x", "0");
|
|
node->AddAttribute("translation_y", "0");
|
|
root->AddChild(node);
|
|
root = node;
|
|
|
|
// Write sample description
|
|
node = new wxXmlNode(wxXML_ELEMENT_NODE, "TextSampleDescription");
|
|
node->AddAttribute("horizontalJustification", "center");
|
|
node->AddAttribute("verticalJustification", "bottom");
|
|
node->AddAttribute("backColor", "0 0 0 0");
|
|
node->AddAttribute("verticalText", "no");
|
|
node->AddAttribute("fillTextRegion", "no");
|
|
node->AddAttribute("continuousKaraoke", "no");
|
|
node->AddAttribute("scroll", "None");
|
|
root->AddChild(node);
|
|
root = node;
|
|
|
|
// Write font table
|
|
node = new wxXmlNode(wxXML_ELEMENT_NODE, "FontTable");
|
|
wxXmlNode *subNode = new wxXmlNode(wxXML_ELEMENT_NODE, "FontTableEntry");
|
|
subNode->AddAttribute("fontName", "Sans");
|
|
subNode->AddAttribute("fontID", "1");
|
|
node->AddChild(subNode);
|
|
root->AddChild(node);
|
|
|
|
// Write text box
|
|
node = new wxXmlNode(wxXML_ELEMENT_NODE, "TextBox");
|
|
node->AddAttribute("top", "0");
|
|
node->AddAttribute("left", "0");
|
|
node->AddAttribute("bottom", "60");
|
|
node->AddAttribute("right", "400");
|
|
root->AddChild(node);
|
|
|
|
// Write style
|
|
node = new wxXmlNode(wxXML_ELEMENT_NODE, "Style");
|
|
node->AddAttribute("styles", "Normal");
|
|
node->AddAttribute("fontID", "1");
|
|
node->AddAttribute("fontSize", "18");
|
|
node->AddAttribute("color", "ff ff ff ff");
|
|
root->AddChild(node);
|
|
}
|
|
|
|
void TTXTSubtitleFormat::WriteLine(wxXmlNode *root, AssDialogue *line) {
|
|
// If it doesn't start at the end of previous, add blank
|
|
wxXmlNode *node, *subNode;
|
|
if (prev && prev->End != line->Start) {
|
|
node = new wxXmlNode(wxXML_ELEMENT_NODE, "TextSample");
|
|
node->AddAttribute("sampleTime", "0" + prev->End.GetASSFormated(true));
|
|
node->AddAttribute("xml:space", "preserve");
|
|
subNode = new wxXmlNode(wxXML_TEXT_NODE, "", "");
|
|
node->AddChild(subNode);
|
|
root->AddChild(node);
|
|
}
|
|
|
|
// Generate and insert node
|
|
node = new wxXmlNode(wxXML_ELEMENT_NODE, "TextSample");
|
|
node->AddAttribute("sampleTime", "0" + line->Start.GetASSFormated(true));
|
|
node->AddAttribute("xml:space", "preserve");
|
|
subNode = new wxXmlNode(wxXML_TEXT_NODE, "", line->Text);
|
|
node->AddChild(subNode);
|
|
root->AddChild(node);
|
|
|
|
// Set as previous
|
|
prev = line;
|
|
}
|
|
|
|
void TTXTSubtitleFormat::ConvertToTTXT () {
|
|
SortLines();
|
|
StripComments();
|
|
RecombineOverlaps();
|
|
MergeIdentical();
|
|
StripTags();
|
|
ConvertNewlines("\r\n");
|
|
|
|
// Find last line
|
|
AssTime lastTime;
|
|
for (std::list<AssEntry*>::reverse_iterator cur=Line->rbegin();cur!=Line->rend();cur++) {
|
|
if (AssDialogue *prev = dynamic_cast<AssDialogue*>(*cur)) {
|
|
lastTime = prev->End;
|
|
break;
|
|
}
|
|
}
|
|
|
|
// Insert blank line at the end
|
|
AssDialogue *diag = new AssDialogue();
|
|
diag->Start = lastTime;
|
|
diag->End = lastTime+OPT_GET("Timing/Default Duration")->GetInt();
|
|
diag->group = "[Events]";
|
|
diag->Style = "Default";
|
|
diag->Comment = false;
|
|
Line->push_back(diag);
|
|
}
|