#ifndef SUBTITLE_H
#define SUBTITLE_H

/**
 * Copyright (C) 2007-2009 Sergio Pistone <sergio_pistone@yahoo.com.ar>
 * Copyright (C) 2010-2015 Mladen Milinkovic <max@smoothware.net>
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the
 * Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
 * Boston, MA 02110-1301, USA.
 */


#ifdef HAVE_CONFIG_H
#include <config.h>
#endif

#include "range.h"
#include "rangelist.h"
#include "time.h"
#include "sstring.h"
#include "subtitleline.h"
#include "actionmanager.h"
#include "formatdata.h"

#include <QObject>
#include <QString>
#include <QStringList>
#include <QList>

namespace SubtitleComposer {
class CompositeAction;

class Subtitle : public QObject
{
	Q_OBJECT

	friend class SubtitleLine;
	friend class SubtitleIterator;

	friend class SubtitleAction;
	friend class SetFramesPerSecondAction;
	friend class InsertLinesAction;
	friend class RemoveLinesAction;
	friend class MoveLineAction;

	friend class SubtitleLineAction;
	friend class SetLinePrimaryTextAction;
	friend class SetLineSecondaryTextAction;
	friend class SetLineTextsAction;
	friend class SetLineShowTimeAction;
	friend class SetLineHideTimeAction;
	friend class SetLineTimesAction;
	friend class SetLineErrorsAction;
	friend class ToggleLineMarkedAction;

	friend class SubtitleCompositeActionExecutor;

	friend class Format;
	friend class InputFormat;

public:
	typedef enum {
		Primary = SubtitleLine::Primary,
		Secondary = SubtitleLine::Secondary,
		Both = SubtitleLine::Both,
		TextTargetSIZE = SubtitleLine::TextTargetSIZE
	} TextTarget;

	static double defaultFramesPerSecond();
	static void setDefaultFramesPerSecond(double framesPerSecond);

	Subtitle(double framesPerSecond = defaultFramesPerSecond());
	virtual ~Subtitle();

/// primary data includes primary text, timing information, format data and all errors except secondary only errors
	void setPrimaryData(const Subtitle &from, bool usePrimaryData);
	void clearPrimaryTextData();

/// secondary data includes secondary text and secondary only errors
	void setSecondaryData(const Subtitle &from, bool usePrimaryData);
	void clearSecondaryTextData();

	bool isPrimaryDirty() const;
	void clearPrimaryDirty();

	bool isSecondaryDirty() const;
	void clearSecondaryDirty();

	ActionManager & actionManager();
	const ActionManager & actionManager() const;

	double framesPerSecond() const;
	void setFramesPerSecond(double framesPerSecond);
	void changeFramesPerSecond(double toFramesPerSecond, double fromFramesPerSecond = -1.0);

	bool isEmpty() const;
	int linesCount() const;
	int lastIndex() const;

	SubtitleLine * line(int index);
	const SubtitleLine * line(int index) const;
	SubtitleLine * firstLine();
	const SubtitleLine * firstLine() const;
	SubtitleLine * lastLine();
	const SubtitleLine * lastLine() const;

	inline const QList<SubtitleLine *> & allLines() const { return m_lines; }

	inline const QList<const SubtitleLine *> & anchoredLines() const { return m_anchoredLines; }

	bool isLineAnchored(int index);
	bool isLineAnchored(const SubtitleLine *line);
	void toggleLineAnchor(int index);
	void toggleLineAnchor(const SubtitleLine *line);
	void removeAllAnchors();

	void insertLine(SubtitleLine *line, int index = -1);
	void insertLines(const QList<SubtitleLine *> &lines, int index = -1);
	SubtitleLine * insertNewLine(int index, bool timeAfter, TextTarget target);
	void removeLines(const RangeList &ranges, TextTarget target);

	void swapTexts(const RangeList &ranges);

	void splitLines(const RangeList &ranges);
	void joinLines(const RangeList &ranges);

	void shiftAnchoredLine(SubtitleLine *anchoredLine, const Time &newShowTime);

	void shiftLines(const RangeList &ranges, long msecs);
	void adjustLines(const Range &range, long firstTime, long lastTime);
	void sortLines(const Range &range);

	void applyDurationLimits(const RangeList &ranges, const Time &minDuration, const Time &maxDuration, bool canOverlap);
	void setMaximumDurations(const RangeList &ranges);
	void setAutoDurations(const RangeList &ranges, int msecsPerChar, int msecsPerWord, int msecsPerLine, bool canOverlap, TextTarget calculationTarget);

	void fixOverlappingLines(const RangeList &ranges, const Time &minInterval = 100);

	void fixPunctuation(const RangeList &ranges, bool spaces, bool quotes, bool englishI, bool ellipsis, TextTarget target);

	void lowerCase(const RangeList &ranges, TextTarget target);
	void upperCase(const RangeList &ranges, TextTarget target);
	void titleCase(const RangeList &ranges, bool lowerFirst, TextTarget target);
	void sentenceCase(const RangeList &ranges, bool lowerFirst, TextTarget target);

	void breakLines(const RangeList &ranges, unsigned minLengthForLineBreak, TextTarget target);
	void unbreakTexts(const RangeList &ranges, TextTarget target);
	void simplifyTextWhiteSpace(const RangeList &ranges, TextTarget target);

	void syncWithSubtitle(const Subtitle &refSubtitle);
	void appendSubtitle(const Subtitle &srcSubtitle, long shiftMsecsBeforeAppend);
	void splitSubtitle(Subtitle &dstSubtitle, const Time &splitTime, bool shiftSplitLines);

	void setStyleFlags(const RangeList &ranges, int styleFlags);
	void setStyleFlags(const RangeList &ranges, int styleFlags, bool on);
	void toggleStyleFlag(const RangeList &ranges, SString::StyleFlag styleFlag);
	void changeTextColor(const RangeList &ranges, QRgb color);

	void setMarked(const RangeList &ranges, bool value);
	void toggleMarked(const RangeList &ranges);

	void clearErrors(const RangeList &ranges, int errorFlags);
	void checqCriticals(const RangeList &ranges, int errorFlags, int minDurationMsecs, int maxDurationMsecs, int minMsecsPerChar, int maxMsecsPerChar, int maxChars, int maxLines);
	void rechecqCriticals(const RangeList &ranges, int minDurationMsecs, int maxDurationMsecs, int minMsecsPerChar, int maxMsecsPerChar, int maxChars, int maxLines);

signals:
	void primaryChanged();
	void secondaryChanged();

	void primaryDirtyStateChanged(bool dirty);
	void secondaryDirtyStateChanged(bool dirty);

	void framesPerSecondChanged(double fps);
	void linesAboutToBeInserted(int firstIndex, int lastIndex);
	void linesInserted(int firstIndex, int lastIndex);
	void linesAboutToBeRemoved(int firstIndex, int lastIndex);
	void linesRemoved(int firstIndex, int lastIndex);

	void lineAnchorChanged(const SubtitleLine *line, bool anchored);

/// forwarded line signals
	void linePrimaryTextChanged(SubtitleLine *line, const SString &text);
	void lineSecondaryTextChanged(SubtitleLine *line, const SString &text);
	void lineShowTimeChanged(SubtitleLine *line, const Time &showTime);
	void lineHideTimeChanged(SubtitleLine *line, const Time &hideTime);
	void lineErrorFlagsChanged(SubtitleLine *line, int errorFlags);
	void lineMarkChanged(SubtitleLine *line, bool marked);

private:
	FormatData * formatData() const;
	void setFormatData(const FormatData *formatData);

	void beginCompositeAction(const QString &title, bool immediateExecution = true, bool delaySignals = true);
	void endCompositeAction();
	void processAction(Action *action);

	void incrementState(int dirtyMode);
	void decrementState(int dirtyMode);

	inline int normalizeRangeIndex(int index) const { return index >= m_lines.count() ? m_lines.count() - 1 : index; }

	void setLastValidCachedIndex(int lastValidCachedIndex);

private:
	ActionManager m_actionManager;
	int m_primaryState;
	int m_primaryCleanState;
	int m_secondaryState;
	int m_secondaryCleanState;

	CompositeAction *m_compositeAction;
	int m_compositeActionDepth;

	double m_framesPerSecond;
	mutable QList<SubtitleLine *> m_lines;
	QList<const SubtitleLine *> m_anchoredLines;

	int m_lastValidCachedIndex;             // all SubtitleLines with a m_cachedIndex greater
// than this value have to recalculate its value
// when asked about their index() value.

	FormatData *m_formatData;

	static double s_defaultFramesPerSecond;
};

class SubtitleCompositeActionExecutor
{
public:
	SubtitleCompositeActionExecutor(Subtitle &subtitle, const QString &title);
	~SubtitleCompositeActionExecutor();

private:
	Subtitle &m_subtitle;
};
}
#endif
