#include "projectfilter.h"

#include "project/media.h"
#include "project/sequence.h"

#include <QDebug>

ProjectFilter::ProjectFilter(QObject *parent) :
	QSortFilterProxyModel(parent),
	show_sequences(true)
{}

bool ProjectFilter::get_show_sequences() {
	return show_sequences;
}

void ProjectFilter::set_show_sequences(bool b) {
	show_sequences = b;
    invalidateFilter();
}

void ProjectFilter::update_search_filter(const QString &s) {
    search_filter = s;
    invalidateFilter();
}

bool ProjectFilter::filterAcceptsRow(int source_row, const QModelIndex &source_parent) const {
    // retrieve media object from index
    QModelIndex index = sourceModel()->index(source_row, 0, source_parent);
    Media* media = static_cast<Media*>(index.internalPointer());

    // hide sequences if show_sequences is false
	if (!show_sequences) {
		if (media != nullptr && media->get_type() == MEDIA_TYPE_SEQUENCE) {
			return false;
		}
	}

    // filter by search filter string
    if (!search_filter.isEmpty()) {
        // search markers if media is a sequene
        bool marker_contains_search = false;

        if (media->get_type() == MEDIA_TYPE_SEQUENCE
                || media->get_type() == MEDIA_TYPE_FOOTAGE) {
            QVector<Marker>& markers = media->get_markers();
            for (int i=0;i<markers.size();i++) {
                if (markers.at(i).name.contains(search_filter, Qt::CaseInsensitive)) {
                    marker_contains_search = true;
                    break;
                }
            }
        }

        // hide any rows that don't contain the search string (unless it's a folder)
        if (!marker_contains_search
                && media->get_type() != MEDIA_TYPE_FOLDER
                && !media->get_name().contains(search_filter, Qt::CaseInsensitive)) {
            return false;
        }
    }

	return QSortFilterProxyModel::filterAcceptsRow(source_row, source_parent);
}