Aegisub/aegisub/subtitle_format_ttxt.cpp

// Copyright (c) 2006, 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
//
// Website: http://aegisub.cellosoft.com
// Contact: mailto:zeratul@cellosoft.com
//


///////////
// Headers
#include "subtitle_format_ttxt.h"
#include "ass_time.h"
#include "ass_file.h"


///////////////////
// Get format name
wxString TTXTSubtitleFormat::GetName() {
	return _T("MPEG-4 Streaming Text");
}


//////////////////////
// Get read wildcards
wxArrayString TTXTSubtitleFormat::GetReadWildcards() {
	wxArrayString formats;
	formats.Add(_T("ttxt"));
	return formats;
}


///////////////////////
// Get write wildcards
wxArrayString TTXTSubtitleFormat::GetWriteWildcards() {
	return GetReadWildcards();
	//return wxArrayString();
}


////////////////////
// Can read a file?
bool TTXTSubtitleFormat::CanReadFile(wxString filename) {
	return (filename.Right(5).Lower() == _T(".ttxt"));
}


/////////////////////
// Can write a file?
bool TTXTSubtitleFormat::CanWriteFile(wxString filename) {
	//return false;
	return (filename.Right(5).Lower() == _T(".ttxt"));
}


///////////////
// Read a file
void TTXTSubtitleFormat::ReadFile(wxString filename,wxString forceEncoding) {
	// Load default
	LoadDefault(false);

	// Load XML document
	wxXmlDocument doc;
	if (!doc.Load(filename)) throw _T("Failed loading TTXT XML file.");

	// Check root node name
	if (doc.GetRoot()->GetName() != _T("TextStream")) throw _T("Invalid TTXT file.");

	// Check version
	wxString verStr = doc.GetRoot()->GetPropVal(_T("version"),_T(""));
	version = -1;
	if (verStr == _T("1.0")) version = 0;
	else if (verStr == _T("1.1")) version = 1;
	else throw wxString(_T("Unknown TTXT version: ") + verStr);

	// Get children
	diag = NULL;
	wxXmlNode *child = doc.GetRoot()->GetChildren();
	int lines = 0;
	while (child) {
		// Line
		if (child->GetName() == _T("TextSample")) {
			if (ProcessLine(child)) lines++;
		}

		// Header
		else if (child->GetName() == _T("TextStreamHeader")) {
			ProcessHeader(child);
		}

		// Proceed to next child
		child = child->GetNext();
	}

	// No lines?
	if (lines == 0) {
		AssDialogue *line = new AssDialogue();
		line->group = _T("[Events]");
		line->Style = _T("Default");
		line->StartMS = 0;
		line->Start.SetMS(0);
		line->End.SetMS(5000);
		Line->push_back(line);
	}
}


///////////////////////////
// Process a dialogue line
bool TTXTSubtitleFormat::ProcessLine(wxXmlNode *node) {
	// Get properties
	wxString sampleTime = node->GetPropVal(_T("sampleTime"),_T("00:00:00.000"));
	wxString text;
	if (version == 0) text = node->GetPropVal(_T("text"),_T(""));
	else text = node->GetNodeContent();

	// Parse time
	AssTime time;
	time.ParseASS(sampleTime);

	// Set end time of last line
	if (diag) diag->End = time;
	diag = NULL;

	// Create line
	if (!text.IsEmpty()) {
		// Create dialogue
		diag = new AssDialogue();
		diag->Start = time;
		diag->End.SetMS(time.GetMS()+5000);
		diag->group = _T("[Events]");
		diag->Style = _T("Default");
		diag->Comment = false;
		diag->StartMS = diag->Start.GetMS();

		// 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] == _T('\'')) {
					if (!in && !first) finalText += _T("\\N");
					first = false;
					in = !in;
				}
				else if (in) finalText += text[i];
			}
			diag->Text = finalText;
		}

		// Process text for 1.1
		else {
			text.Replace(_T("\r"),_T(""));
			text.Replace(_T("\n"),_T("\\N"));
			diag->Text = text;
		}

		// Insert dialogue
		diag->UpdateData();
		Line->push_back(diag);
		return true;
	}

	else return false;
}


//////////////////////
// Process the header
void TTXTSubtitleFormat::ProcessHeader(wxXmlNode *node) {
	// TODO
}


////////////////
// Write a file
void TTXTSubtitleFormat::WriteFile(wxString filename,wxString encoding) {
	// Convert to TTXT
	CreateCopy();
	ConvertToTTXT();

	// Create XML structure
	wxXmlDocument doc;
	wxXmlNode *root = new wxXmlNode(NULL,wxXML_ELEMENT_NODE,_T("TextStream"));
	root->AddProperty(_T("version"),_T("1.1"));
	doc.SetRoot(root);
	wxXmlNode *node,*subNode;

	// Create header
	node = new wxXmlNode(wxXML_ELEMENT_NODE,_T("TextStreamHeader"));
	node->AddProperty(_T("width"),_T("400"));
	node->AddProperty(_T("height"),_T("60"));
	root->AddChild(node);

	// Create lines
	int i=1;
	using std::list;
	AssDialogue *prev = NULL;
	for (list<AssEntry*>::iterator cur=Line->begin();cur!=Line->end();cur++) {
		AssDialogue *current = AssEntry::GetAsDialogue(*cur);
		if (current) {
			// Get line
			if (current->Comment) throw _T("Unexpected line type (comment)");

			// If it doesn't start at the end of previous, add blank
			if (prev && prev->End != current->Start) {
				node = new wxXmlNode(wxXML_ELEMENT_NODE,_T("TextSample"));
				node->AddProperty(_T("startTime"),_T("0") + prev->End.GetASSFormated(true));
				node->AddProperty(_T("xml:space"),_T("preserve"));
				subNode = new wxXmlNode(wxXML_TEXT_NODE,_T(""),_T(""));
				node->AddChild(subNode);
				root->AddChild(node);
			}

			// Generate and insert node
			node = new wxXmlNode(wxXML_ELEMENT_NODE,_T("TextSample"));
			node->AddProperty(_T("sampleTime"),_T("0") + current->Start.GetASSFormated(true));
			node->AddProperty(_T("xml:space"),_T("preserve"));
			subNode = new wxXmlNode(wxXML_TEXT_NODE,_T(""),current->Text);
			node->AddChild(subNode);
			root->AddChild(node);

			// Set as previous
			prev = current;

			i++;
		}
		else throw _T("Unexpected line type");
	}

	// Save XML
	//prevNode->SetNext(NULL);
	doc.Save(filename);

	// Clear
	ClearCopy();
}


///////////////////////
// Convert line to TTXT
void TTXTSubtitleFormat::DialogueToTTXT(AssDialogue *current,std::list<AssEntry*>::iterator prev) {
	using std::list;
	AssDialogue *previous;
	if (prev != Line->end()) previous = AssEntry::GetAsDialogue(*prev);
	else previous = NULL;

	// Strip ASS tags
	current->StripTags();

	// Join equal lines
	if (previous != NULL) {
		if (previous->Text == current->Text) {
			if (abs(current->Start.GetMS() - previous->End.GetMS()) < 20) {
				current->Start = (current->Start < previous->Start ? current->Start : previous->Start);
				current->End = (current->End > previous->End ? current->End : previous->End);
				delete *prev;
				Line->erase(prev);
			}
		}
	}

	// Fix line breaks
	current->Text.Replace(_T("\\n"),_T("\r\n"),true);
	current->Text.Replace(_T("\\N"),_T("\r\n"),true);
	while (current->Text.Replace(_T("\r\n\r\n"),_T("\r\n"),true));
}


//////////////////////////////
// Converts whole file to TTXT
void TTXTSubtitleFormat::ConvertToTTXT () {
	using std::list;

	// Sort lines
	Line->sort(LessByPointedToValue<AssEntry>());

	// Prepare processing
	list<AssEntry*>::iterator next;
	list<AssEntry*>::iterator prev = Line->end();
	AssTime lastTime;

	// Process lines
	bool notfirst = false;
	for (list<AssEntry*>::iterator cur=Line->begin();cur!=Line->end();cur=next) {
		next = cur;
		next++;

		// Dialogue line (not comment)
		AssDialogue *current = AssEntry::GetAsDialogue(*cur);
		if (current && !current->Comment) {
			DialogueToTTXT(current,prev);
			notfirst = true;
			prev = cur;
			lastTime = current->End;
		}

		// Other line, delete it
		else {
			delete *cur;
			Line->erase(cur);
		}
	}

	// Insert blank line at the end
	AssDialogue *diag = new AssDialogue();
	diag->Start = lastTime;
	diag->End.SetMS(lastTime.GetMS()+5000);
	diag->group = _T("[Events]");
	diag->Style = _T("Default");
	diag->Comment = false;
	diag->StartMS = diag->Start.GetMS();
	Line->push_back(diag);
}