#include "project.h"
#include "ui_project.h"
#include "io/media.h"

#include "panels/panels.h"
#include "panels/timeline.h"
#include "panels/viewer.h"
#include "playback/playback.h"
#include "effects/effects.h"
#include "panels/timeline.h"
#include "project/sequence.h"
#include "project/effect.h"
#include "effects/transition.h"
#include "io/previewgenerator.h"

#include <QFileDialog>
#include <QString>
#include <QVariant>
#include <QDebug>
#include <QCharRef>
#include <QMessageBox>
#include <QDropEvent>
#include <QMimeData>
#include <QPushButton>
#include <QXmlStreamReader>
#include <QXmlStreamWriter>

extern "C" {
	#include <libavformat/avformat.h>
	#include <libavcodec/avcodec.h>
}

bool project_changed = false;
QString project_url = "";

Project::Project(QWidget *parent) :
	QDockWidget(parent),
	ui(new Ui::Project)
{
    ui->setupUi(this);
    source_table = ui->treeWidget;
    connect(ui->treeWidget, SIGNAL(itemChanged(QTreeWidgetItem*,int)), this, SLOT(rename_media(QTreeWidgetItem*,int)));
}

Project::~Project()
{
	delete ui;
}

QString Project::get_next_sequence_name() {
	int n = 1;
	bool found = true;
	QString name;
	while (found) {
		found = false;
		name = "Sequence ";
		if (n < 10) {
			name += "0";
		}
		name += QString::number(n);
		for (int i=0;i<ui->treeWidget->topLevelItemCount();i++) {
			if (QString::compare(ui->treeWidget->topLevelItem(i)->text(0), name, Qt::CaseInsensitive) == 0) {
				found = true;
				n++;
				break;
			}
		}
	}
	return name;
}

void Project::rename_media(QTreeWidgetItem* item, int column) {
    int type = get_type_from_tree(item);
    QString n = item->text(column);
    switch (type) {
    case MEDIA_TYPE_FOOTAGE: get_media_from_tree(item)->name = n; break;
    case MEDIA_TYPE_SEQUENCE: get_sequence_from_tree(item)->name = n; break;
    }
}

void Project::duplicate_selected() {
    QList<QTreeWidgetItem*> items = ui->treeWidget->selectedItems();
    for (int i=0;i<items.size();i++) {
        int type = get_type_from_tree(items.at(i));
        if (type == MEDIA_TYPE_SEQUENCE) {
            new_sequence(get_sequence_from_tree(items.at(i))->copy(), false);
        }
    }
}

void Project::new_sequence(Sequence *s, bool open) {
    QTreeWidgetItem* item = new_item();
    item->setText(0, s->name);
    set_sequence_of_tree(item, s);

	ui->treeWidget->addTopLevelItem(item);

    project_changed = true;

    if (open) set_sequence(s);
}

Media* Project::import_file(QString file) {
    QByteArray ba = file.toLatin1();
    char* filename = new char[ba.size()+1];
    strcpy(filename, ba.data());

    Media* m = NULL;

    AVFormatContext* pFormatCtx = NULL;
    int errCode = avformat_open_input(&pFormatCtx, filename, NULL, NULL);
    if(errCode != 0) {
        char err[1024];
        av_strerror(errCode, err, 1024);
        qDebug() << "[ERROR] Could not open" << filename << "-" << err;
    } else {
        errCode = avformat_find_stream_info(pFormatCtx, NULL);
        if (errCode < 0) {
            char err[1024];
            av_strerror(errCode, err, 1024);
            fprintf(stderr, "[ERROR] Could not find stream information. %s\n", err);
        } else {
            av_dump_format(pFormatCtx, 0, filename, 0);

            m = new Media();
            m->url = file;
            m->name = file.mid(file.lastIndexOf('/')+1);

            // detect video/audio streams in file
            for (int i=0;i<(int)pFormatCtx->nb_streams;i++) {
                // Find the decoder for the video stream
                if (avcodec_find_decoder(pFormatCtx->streams[i]->codecpar->codec_id) == NULL) {
                    qDebug() << "[ERROR] Unsupported codec in stream %d.\n";
                } else {
                    MediaStream* ms = new MediaStream();
                    ms->preview_done = false;
                    ms->file_index = i;
                    if (pFormatCtx->streams[i]->codecpar->codec_type == AVMEDIA_TYPE_VIDEO) {
                        bool infinite_length = (pFormatCtx->streams[i]->avg_frame_rate.den == 0);
                        ms->video_width = pFormatCtx->streams[i]->codecpar->width;
                        ms->video_height = pFormatCtx->streams[i]->codecpar->height;
                        ms->infinite_length = infinite_length;
                        m->video_tracks.append(ms);
                    } else if (pFormatCtx->streams[i]->codecpar->codec_type == AVMEDIA_TYPE_AUDIO) {
                        ms->audio_channels = pFormatCtx->streams[i]->codecpar->channels;
                        m->audio_tracks.append(ms);
                    }
                }
            }
            m->length = pFormatCtx->duration;

            QTreeWidgetItem* item = new_item();
            if (m->video_tracks.size() == 0) {
                item->setIcon(0, QIcon(":/icons/audiosource.png"));
            } else {
                item->setIcon(0, QIcon(":/icons/videosource.png"));
            }
            item->setText(0, m->name);
            item->setText(1, QString::number(m->length));
            set_media_of_tree(item, m);

            ui->treeWidget->addTopLevelItem(item);

            project_changed = true;

            // generate waveform/thumbnail in another thread
            PreviewGenerator* pg = new PreviewGenerator();
            pg->fmt_ctx = pFormatCtx; // cleaned up in PG
            pg->media = m;
            connect(pg, SIGNAL(finished()), pg, SLOT(deleteLater()));
            pg->start(QThread::LowPriority);
        }
    }

    delete [] filename;
    return m;
}

QTreeWidgetItem* Project::new_item() {
    QTreeWidgetItem* item = new QTreeWidgetItem();
    item->setFlags(item->flags() | Qt::ItemIsEditable);
    return item;
}

bool Project::is_focused() {
    return ui->treeWidget->hasFocus();
}

void Project::new_folder() {
    QTreeWidgetItem* item = new_item();
    item->setChildIndicatorPolicy(QTreeWidgetItem::ShowIndicator);
    item->setText(0, "New Folder");
    set_item_to_folder(item);
    ui->treeWidget->addTopLevelItem(item);
    project_changed = true;
}

void Project::delete_selected_media() {
    QList<QTreeWidgetItem*> items = ui->treeWidget->selectedItems();
    bool remove = true;
    bool redraw = false;

    // check if media is in use
    for (int i=0;i<items.size();i++) {
        QTreeWidgetItem* item = items.at(i);
        Media* m = get_media_from_tree(item);
        if (get_type_from_tree(item) == MEDIA_TYPE_FOOTAGE && sequence != NULL) {
            for (int j=0;j<sequence->clip_count();j++) {
                Clip* c = sequence->get_clip(j);
                if (c != NULL && c->media == m) {
                    remove = false;
                    break;
                }
            }
            if (!remove) {
                QMessageBox confirm(this);
                confirm.setWindowTitle("Delete media in use?");
                confirm.setText("The media '" + m->name + "' is currently used in the sequence. Deleting it will remove all instances in the sequence. Are you sure you want to do this?");
                QAbstractButton* yes_button = confirm.addButton(QMessageBox::Yes);
                QAbstractButton* skip_button = confirm.addButton("Skip", QMessageBox::NoRole);
                QAbstractButton* abort_button = confirm.addButton(QMessageBox::Abort);
                confirm.exec();
                if (confirm.clickedButton() == yes_button) {
                    // remove all clips referencing this media
                    for (int j=0;j<sequence->clip_count();j++) {
                        if (sequence->get_clip(j)->media == m) {
                            sequence->delete_clip(j);
                        }
                    }
                    remove = true;
                    redraw = true;
                } else if (confirm.clickedButton() == skip_button) {
                    items.removeAt(i);
                    i--;
                    remove = true;
                } else if (confirm.clickedButton() == abort_button) {
                    break;
                }
            }
        }
    }

    // remove
    if (remove) {
        for (int i=0;i<items.size();i++) {
            QTreeWidgetItem* item = items.at(i);
            delete_media(item);
            delete item;
        }
    }

    // redraw clips
    if (redraw) {
        panel_timeline->redraw_all_clips(true);
    }
}

void Project::process_file_list(QStringList& files) {
    for (int i=0;i<files.count();i++) {
        QString file(files.at(i));

        // heuristic to determine whether file is part of an image sequence
        int lastcharindex = file.lastIndexOf(".")-1;
        if (lastcharindex == -2) {
            lastcharindex = file.length() - 1;
        }
        if (file[lastcharindex].isDigit()) {
            bool is_img_sequence = false;
            QString sequence_test(file);
            char lastchar = sequence_test.at(lastcharindex).toLatin1();

            sequence_test[lastcharindex] = lastchar + 1;
            if (QFileInfo::exists(sequence_test)) is_img_sequence = true;

            sequence_test[lastcharindex] = lastchar - 1;
            if (QFileInfo::exists(sequence_test)) is_img_sequence = true;

            if (is_img_sequence &&
                    QMessageBox::question(this, "Image sequence detected", "The file '" + file + "' appears to be part of an image sequence. Would you like to import it as such?", QMessageBox::Yes | QMessageBox::No, QMessageBox::Yes) == QMessageBox::Yes) {
                // change url to an image sequence instead
                int digit_count = 0;
                QString ext = file.mid(lastcharindex+1);
                while (file[lastcharindex].isDigit()) {
                    digit_count++;
                    lastcharindex--;
                }
                QString new_filename = file.left(lastcharindex+1) + "%";
                if (digit_count < 10) {
                    new_filename += "0";
                }
                new_filename += QString::number(digit_count) + "d" + ext;
                file = new_filename;
            }
        }

        import_file(file);
    }
}

void Project::import_dialog() {
	QStringList files = QFileDialog::getOpenFileNames(this, "Import media...", "", "All Files (*.*)");
    process_file_list(files);
}

void Project::set_item_to_folder(QTreeWidgetItem* item) {
    item->setData(0, Qt::UserRole + 1, MEDIA_TYPE_FOLDER);
}

Media* Project::get_media_from_tree(QTreeWidgetItem* item) {
    return reinterpret_cast<Media*>(item->data(0, Qt::UserRole + 2).value<quintptr>());
}

void Project::set_media_of_tree(QTreeWidgetItem* item, Media* media) {
    item->setData(0, Qt::UserRole + 1, MEDIA_TYPE_FOOTAGE);
    item->setData(0, Qt::UserRole + 2, QVariant::fromValue(reinterpret_cast<quintptr>(media)));
}

Sequence* Project::get_sequence_from_tree(QTreeWidgetItem* item) {
    return reinterpret_cast<Sequence*>(item->data(0, Qt::UserRole + 2).value<quintptr>());
}

void Project::set_sequence_of_tree(QTreeWidgetItem* item, Sequence* sequence) {
    item->setData(0, Qt::UserRole + 1, MEDIA_TYPE_SEQUENCE);
    item->setData(0, Qt::UserRole + 2, QVariant::fromValue(reinterpret_cast<quintptr>(sequence)));
}

int Project::get_type_from_tree(QTreeWidgetItem* item) {
    return item->data(0, Qt::UserRole + 1).toInt();
}

void Project::delete_media(QTreeWidgetItem* item) {
    int type = get_type_from_tree(item);
    switch (type) {
    case MEDIA_TYPE_FOOTAGE:
        delete get_media_from_tree(item);
        break;
    case MEDIA_TYPE_SEQUENCE:
        Sequence* s = get_sequence_from_tree(item);
        if (sequence == s) {
            set_sequence(NULL);
        }
        delete s;
        break;
    }

    project_changed = true;
}

void Project::clear() {
    while (ui->treeWidget->topLevelItemCount() > 0) {
        QTreeWidgetItem* item = ui->treeWidget->topLevelItem(0);
        delete_media(item);
        delete item;
    }
}

void Project::new_project() {
    // clear existing project
    set_sequence(NULL);
    clear();
    project_changed = false;
}

#define LOAD_STATE_IDLE 0
#define LOAD_STATE_MEDIA 1
#define LOAD_STATE_FOOTAGE 2
#define LOAD_STATE_TIMELINE 3
#define LOAD_STATE_SEQUENCE 4
#define LOAD_STATE_CLIP 5
#define LOAD_STATE_CLIP_EFFECTS 6
#define LOAD_STATE_EFFECT 7
#define LOAD_STATE_CLIP_LINKS 8

void Project::load_project() {
    new_project();

    QFile file(project_url);
    if (!file.open(QIODevice::ReadOnly/* | QIODevice::Text*/)) {
        qDebug() << "[ERROR] Could not open file";
        return;
    }

    QXmlStreamReader stream(&file);

    // temp variables for loading
    QVector<Media*> temp_media_list;
    QString temp_name;
    QString temp_url;
    int temp_media_id;
    Sequence* temp_seq;
    Clip* temp_clip;

    int state = LOAD_STATE_IDLE;
    bool error = false;
    while (!error && !stream.atEnd()) {
        stream.readNext();
        switch (state) {
        case LOAD_STATE_IDLE:
            if (stream.isStartElement()) {
                if (stream.name() == "footage") {
                    for (int j=0;j<stream.attributes().size();j++) {
                        const QXmlStreamAttribute& attr = stream.attributes().at(j);
                        if (attr.name() == "id") {
                            temp_media_id = attr.value().toInt();
                        }
                    }
                    state = LOAD_STATE_FOOTAGE;
                } else if (stream.name() == "sequence") {
                    temp_seq = new Sequence();
                    state = LOAD_STATE_SEQUENCE;
                }
            }
            break;
        case LOAD_STATE_FOOTAGE:
            if (stream.isEndElement() && stream.name() == "footage") {
                Media* m = import_file(temp_url);
                if (m == NULL) {
                    error = (QMessageBox::warning(this, "Source load error", "Couldn't load source file '" + temp_url + "'. This is unsupported and may lead to crashes. Do you wish to continue?", QMessageBox::Yes, QMessageBox::No) == QMessageBox::No);
                } else {
                    m->save_id = temp_media_id;
                    m->name = temp_name;
                    temp_media_list.append(m);
                }
                state = LOAD_STATE_IDLE;
            } else if (stream.isStartElement()) {
                if (stream.name() == "name") {
                    stream.readNext();
                    temp_name = stream.text().toString();
                } else if (stream.name() == "url") {
                    stream.readNext();
                    temp_url = stream.text().toString();
                }
            }
            break;
        case LOAD_STATE_SEQUENCE:
            if (stream.isEndElement() && stream.name() == "sequence") {
                // correct IDs
                for (int i=0;i<temp_seq->clip_count();i++) {
                    Clip* clip = temp_seq->get_clip(i);

                    // correct IDs in linked clips
                    for (int j=0;j<clip->linked.size();j++) {
                        bool found = false;
                        for (int k=0;k<temp_seq->clip_count();k++) {
                            if (temp_seq->get_clip(k)->load_id == clip->linked.at(j)) {
                                clip->linked[j] = k;
                                found = true;
                                break;
                            }
                        }
                        if (!found) {                            
                            if (QMessageBox::warning(this, "Corrupt clip link", "Project contains a non-existent clip link. It may be corrupt. Would you like to load it anyway?", QMessageBox::Yes, QMessageBox::No) == QMessageBox::Yes) {
                                clip->linked.removeAt(j);
                            } else {
                                error = true;
                            }
                        }
                    }
                }

                temp_seq->reset_undo();
                new_sequence(temp_seq, false);
                state = LOAD_STATE_IDLE;
            } else if (stream.isStartElement()) {
                if (stream.name() == "name") {
                    stream.readNext();
                    temp_seq->name = stream.text().toString();
                } else if (stream.name() == "width") {
                    stream.readNext();
                    temp_seq->width = stream.text().toInt();
                } else if (stream.name() == "height") {
                    stream.readNext();
                    temp_seq->height = stream.text().toInt();
                } else if (stream.name() == "framerate") {
                    stream.readNext();
                    temp_seq->frame_rate = stream.text().toFloat();
                } else if (stream.name() == "afreq") {
                    stream.readNext();
                    temp_seq->audio_frequency = stream.text().toInt();
                } else if (stream.name() == "alayout") {
                    stream.readNext();
                    temp_seq->audio_layout = stream.text().toInt();
                } else if (stream.name() == "clip") {
                    temp_clip = new Clip();
                    temp_clip->sequence = temp_seq;
                    state = LOAD_STATE_CLIP;
                }
            }
            break;
        case LOAD_STATE_CLIP:
            if (stream.isEndElement() && stream.name() == "clip") {
                temp_seq->add_clip(temp_clip);
                state = LOAD_STATE_SEQUENCE;
            } else if (stream.isStartElement()) {
                if (stream.name() == "name") {
                    stream.readNext();
                    temp_clip->name = stream.text().toString();
                } else if (stream.name() == "id") {
                    stream.readNext();
                    temp_clip->load_id = stream.text().toInt();
                } else if (stream.name() == "clipin") {
                    stream.readNext();
                    temp_clip->clip_in = stream.text().toInt();
                } else if (stream.name() == "in") {
                    stream.readNext();
                    temp_clip->timeline_in = stream.text().toInt();
                } else if (stream.name() == "out") {
                    stream.readNext();
                    temp_clip->timeline_out = stream.text().toInt();
                } else if (stream.name() == "track") {
                    stream.readNext();
                    temp_clip->track = stream.text().toInt();
                } else if (stream.name() == "opening") {
                    stream.readNext();
                    temp_clip->opening_transition = create_transition(stream.text().toInt(), temp_clip);
                } else if (stream.name() == "closing") {
                    stream.readNext();
                    temp_clip->closing_transition = create_transition(stream.text().toInt(), temp_clip);
                } else if (stream.name() == "color") {
                    for (int j=0;j<stream.attributes().size();j++) {
                        const QXmlStreamAttribute& attr = stream.attributes().at(j);
                        if (attr.name().toString() == QLatin1String("r")) {
                            temp_clip->color_r = attr.value().toInt();
                        } else if (attr.name().toString() == QLatin1String("g")) {
                            temp_clip->color_g = attr.value().toInt();
                        } else if (attr.name().toString() == QLatin1String("b")) {
                            temp_clip->color_b = attr.value().toInt();
                        }
                    }
                } else if (stream.name() == "media") {
                    stream.readNext();
                    for (int i=0;i<temp_media_list.size();i++) {
                        Media* m = temp_media_list.at(i);
                        if (m->save_id == stream.text().toInt()) {
                            temp_clip->media = m;
                            break;
                        }
                    }
                } else if (stream.name() == "stream") {
                    stream.readNext();
                    int stream_index = stream.text().toInt();
                    if (temp_clip->media != NULL) {
                        temp_clip->media_stream = temp_clip->media->get_stream_from_file_index(stream_index);
                        if (temp_clip->media_stream == NULL) {
                            if (QMessageBox::warning(this, "Source changed", "Source file '" + temp_clip->media->url + "' appears to have changed since last saving the project. This is unsupported and may lead to crashes. Do you wish to continue?", QMessageBox::Yes, QMessageBox::No) == QMessageBox::No) {
                                error = true;
                            }
                        }
                    }
                } else if (stream.name() == "linked") {
                    state = LOAD_STATE_CLIP_LINKS;
                } else if (stream.name() == "effect") {
                    int effect_id = -1;
                    for (int j=0;j<stream.attributes().size();j++) {
                        const QXmlStreamAttribute& attr = stream.attributes().at(j);
                        if (attr.name().toString() == QLatin1String("id")) {
                            effect_id = attr.value().toInt();
                        }
                    }
                    if (effect_id != -1) {
                        Effect* e = create_effect(effect_id, temp_clip);
                        e->load(&stream);
                        temp_clip->effects.append(e);
                    }
                }
            }
            break;
        case LOAD_STATE_CLIP_LINKS:
            if (stream.name() == "linked" && stream.isEndElement()) {
                state = LOAD_STATE_CLIP;
            } else if (stream.isStartElement() && stream.name() == "link") {
                stream.readNext();
                temp_clip->linked.append(stream.text().toInt());
            }
            break;
        }
    }
    if (stream.hasError()) {
        qDebug() << "[ERROR] Error parsing XML." << stream.error();
        QMessageBox::critical(this, "XML Parsing Error", "Couldn't load '" + project_url + "'. " + stream.error(), QMessageBox::Ok);
        error = true;
    }

    if (error) {
        new_project();
    } else {
        panel_timeline->redraw_all_clips(false);
        project_changed = false;
    }
}

void Project::save_project() {
    QFile file(project_url);
    if (!file.open(QIODevice::WriteOnly/* | QIODevice::Text*/)) {
        qDebug() << "[ERROR] Could not open file";
        return;
    }

    QXmlStreamWriter stream(&file);
    stream.setAutoFormatting(true);
    stream.writeStartDocument();

    stream.writeStartElement("project");

    stream.writeTextElement("version", "180601");

    int len = ui->treeWidget->topLevelItemCount();
    stream.writeStartElement("media");
    for (int i=0;i<len;i++) {
        QTreeWidgetItem* item = ui->treeWidget->topLevelItem(i);
        if (get_type_from_tree(item) == MEDIA_TYPE_FOOTAGE) {
            Media* m = get_media_from_tree(item);
            m->save_id = i;
            stream.writeStartElement("footage");
            stream.writeAttribute("id", QString::number(i));
            stream.writeTextElement("name", m->name);
            stream.writeTextElement("url", m->url);
            stream.writeEndElement();
        }
    }
    stream.writeEndElement();

    stream.writeStartElement("timeline");
    for (int i=0;i<len;i++) {
        QTreeWidgetItem* item = ui->treeWidget->topLevelItem(i);
        if (get_type_from_tree(item) == MEDIA_TYPE_SEQUENCE) {
            Sequence* s = get_sequence_from_tree(item);
            stream.writeStartElement("sequence");
            stream.writeTextElement("name", s->name);
            stream.writeTextElement("width", QString::number(s->width));
            stream.writeTextElement("height", QString::number(s->height));
            stream.writeTextElement("framerate", QString::number(s->frame_rate));
            stream.writeTextElement("afreq", QString::number(s->audio_frequency));
            stream.writeTextElement("alayout", QString::number(s->audio_layout));
            stream.writeStartElement("clips");

            for (int i=0;i<s->clip_count();i++) {
                Clip* c = s->get_clip(i);
                if (c != NULL) {
                    stream.writeStartElement("clip");
                    stream.writeTextElement("id", QString::number(i));
                    stream.writeTextElement("name", c->name);
                    stream.writeTextElement("clipin", QString::number(c->clip_in));
                    stream.writeTextElement("in", QString::number(c->timeline_in));
                    stream.writeTextElement("out", QString::number(c->timeline_out));
                    stream.writeTextElement("track", QString::number(c->track));
                    if (c->opening_transition != NULL) {
                        stream.writeTextElement("opening", QString::number(c->opening_transition->id));
                    }
                    if (c->closing_transition != NULL) {
                        stream.writeTextElement("closing", QString::number(c->closing_transition->id));
                    }
                    stream.writeStartElement("color");
                    stream.writeAttribute("r", QString::number(c->color_r));
                    stream.writeAttribute("g", QString::number(c->color_g));
                    stream.writeAttribute("b", QString::number(c->color_b));
                    stream.writeEndElement();
                    stream.writeTextElement("media", QString::number(c->media->save_id));
                    stream.writeTextElement("stream", QString::number(c->media_stream->file_index));
                    stream.writeStartElement("linked");
                    for (int j=0;j<c->linked.size();j++) {
                        stream.writeTextElement("link", QString::number(c->linked.at(j)));
                    }
                    stream.writeEndElement();
                    stream.writeStartElement("effects");
                    for (int j=0;j<c->effects.size();j++) {
                        stream.writeStartElement("effect");
                        Effect* e = c->effects.at(j);
                        stream.writeAttribute("id", QString::number(e->id));
                        e->save(&stream);
                        stream.writeEndElement();
                    }
                    stream.writeEndElement();
                    stream.writeEndElement();
                }
            }
            stream.writeEndElement();
        }
    }
    stream.writeEndElement();

    stream.writeEndElement();

    stream.writeEndDocument();

    file.close();

    project_changed = false;
}