ASSFile.cpp

#include "ASSFile.h"
#include <string>
#include <vector>
#include <fstream>
#include <iostream>
#include <utility>

#define PREPROCESS_STOP_READ_LINE "Format: Layer, Start, End, Style, Name, MarginL, MarginR, MarginV, Effect, Text"

ASSFile::ASSFile()
{

}

ASSFile::ASSFile(std::string inFileName)
{
	std::ifstream fin(inFileName);

	if (!fin.good())
		throw "Input file not found!";

	this->title = convertUnderscoreToSpace(inFileName);

	readPreprocessLines(fin);
	processLyricLines(fin);

	fin.close();

	while (this->lines.rbegin()->getText() == "") this->lines.pop_back();
}

std::string ASSFile::convertUnderscoreToSpace(std::string input)
{
	std::string result;

	for (char c : input)
	{
		if (c == '_') result += ' ';
		else result += c;
	}

	return result;
}

void ASSFile::readPreprocessLines(std::ifstream &fin)
{
	this->preprocess.clear();

	std::string currLine;

	while (1)
	{
		getline(fin, currLine);
		this->preprocess.push_back(currLine);

		if (currLine == PREPROCESS_STOP_READ_LINE) break;
	}
}

void ASSFile::readPreprocessLines(std::string inFileName)
{
	std::ifstream fin(inFileName);

	if (!fin.good())
		throw "Cannot find style file!";

	readPreprocessLines(fin);
}

void ASSFile::removeAllTags()
{
	for (ASSLine line : this->lines)
		line.removeAllTags();
}

void ASSFile::removeNonKaraokeTags()
{
	for (ASSLine line : this->lines)
		line.removeNonKaraokeTags();
}

void ASSFile::processLyricLines(std::ifstream &fin)
{
	while (1)
	{
		std::string currLineString;
		getline(fin, currLineString);

		if (fin.eof()) break;

		ASSLine currLine(currLineString);

		this->lines.push_back(currLine);
	}
}

void ASSFile::printASSFile(std::string outFileName)
{
	std::ofstream fout(outFileName);

	for (std::string line : this->preprocess)
		fout << line << std::endl;

	for (ASSLine line : this->lines)
		fout << line.printASSLine();
}

int ASSFile::getNumLines()
{
	return lines.size();
}

ASSTime ASSFile::getEndTime()
{
	return lines.rbegin()->getEnd();
}

void ASSFile::insertLine(ASSLine toInsert)
{
	this->lines.push_back(toInsert);
}

ASSFileWithSwitch::ASSFileWithSwitch() : ASSFile()
{

}

ASSFileWithSwitch::ASSFileWithSwitch(std::string inFileName)
{
	std::ifstream fin(inFileName);

	if (!fin.good())
		throw "File not found!";

	try
	{
		readPreprocessLines(fin);
		processLyricLines(fin);
	}
	catch (const char* exception)
	{
		throw exception;
	}

	fin.close();

	while (this->lines.rbegin()->getText() == "") this->lines.pop_back();
}

void ASSFileWithSwitch::operator=(ASSFileWithSwitch input)
{
	this->lines = input.lines;
	this->preprocess = input.preprocess;
	this->title = input.title;
}

std::vector <ASSLineWithSwitch> ASSFileWithSwitch::getLines()
{
	return this->lines;
}

ASSLineWithSwitch ASSFileWithSwitch::getLine(int index)
{
	return this->lines[index];
}

void ASSFileWithSwitch::clearLines()
{
	this->lines.clear();
}

void ASSFileWithSwitch::processLyricLines(std::ifstream &fin)
{
	std::vector <ASSLine> currLines;

	while (1)
	{
		std::string currLineString;
		getline(fin, currLineString);

		if (fin.eof())
		{
			try
			{
				ASSLineWithSwitch currLinesWithSwitch(currLines);
				this->lines.push_back(currLinesWithSwitch);
			} 
			catch (const char* exception)
			{
				for (ASSLine errorLine : currLines)
				{
					std::cerr << errorLine.printASSLine();
				}
				throw exception;
			}
			
			break;
		}

		ASSLine currLine(currLineString);
		
		if (currLines.empty() || (currLines.rbegin()->getText() == currLine.getText() && currLine.getLayer() != currLines.rbegin()->getLayer()))
		{
			currLines.push_back(currLine);
		}
		else
		{
			try
			{
				ASSLineWithSwitch currLinesWithSwitch(currLines);
				this->lines.push_back(currLinesWithSwitch);
				currLines.clear();
				currLines.push_back(currLine);
			}
			catch (const char* exception)
			{
				for (ASSLine errorLine : currLines)
				{
					std::cerr << errorLine.printASSLine();
				}
				throw exception;
			}
		}
	}
}

void ASSFileWithSwitch::printASSFile(std::string outFileName)
{
	std::ofstream fout(outFileName);

	for (std::string line : this->preprocess)
		fout << line << std::endl;

	for (ASSLineWithSwitch line : this->lines)
		fout << line.printASSLine();
}

int ASSFileWithSwitch::getNumLines()
{
	return lines.size();
}

ASSTime ASSFileWithSwitch::getStartTime()
{
	return lines.begin()->getStart();
}

ASSTime ASSFileWithSwitch::getEndTime()
{
	return lines.rbegin()->getEnd();
}

void ASSFileWithSwitch::setLine(ASSLineWithSwitch toSet, int index)
{
	this->lines[index] = toSet;
}

void ASSFileWithSwitch::insertLine(ASSLineWithSwitch toInsert)
{
	this->lines.push_back(toInsert);
}

void ASSFileWithSwitch::deleteLine(int index)
{
	this->lines.erase(this->lines.begin() + index);
}

void ASSFileWithSwitch::swapLines(int index1, int index2)
{
	std::swap(lines[index1], lines[index2]);
}

void ASSFileWithSwitch::appendFile(ASSFileWithSwitch toAppend)
{
	for (ASSLineWithSwitch currLine : toAppend.getLines())
		this->lines.push_back(currLine);
}

void ASSFileWithSwitch::offsetFile(ASSTime offset)
{
	for (ASSLineWithSwitch& currLine : this->lines)
		currLine.offsetLine(offset);
}