/*
 * Hedgewars, a free turn based strategy game
 * Copyright (c) 2004-2018 Andrey Korotaev <unC0Rr@gmail.com>
 *
 * 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; version 2 of the License
 *
 * 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 St, Fifth Floor, Boston, MA  02110-1301  USA
 */

/**
 * @file
 * @brief ThemeFilterProxyModel class implementation
 */

#include "ThemeModel.h"
#include "ThemeFilterProxyModel.h"

ThemeFilterProxyModel::ThemeFilterProxyModel(QObject *parent)
    : QSortFilterProxyModel(parent)
{
    isFilteringDLC = false;
    isFilteringHidden = false;
}

bool ThemeFilterProxyModel::filterAcceptsRow(int sourceRow, const QModelIndex & sourceParent) const
{
    if(isFilteringDLC || isFilteringHidden)
    {
        QModelIndex index = sourceModel()->index(sourceRow, 0, sourceParent);
        bool isDLC = index.data(ThemeModel::IsDlcRole).toBool();
        bool isHidden = index.data(ThemeModel::IsHiddenRole).toBool();
        if(
            ((isFilteringDLC && !isDLC) || !isFilteringDLC) &&
            ((isFilteringHidden && !isHidden) || !isFilteringHidden))
        {
            if(!filterRegExp().isEmpty())
            {
                // Also check regular expression set by the theme chooser search
                QString name = index.data(ThemeModel::ActualNameRole).toString();
                int index = filterRegExp().indexIn(name);
                return index != -1;
            }
            else
                return true;
        }
    }
    else
    {
        return true;
    }
}

void ThemeFilterProxyModel::setFilterDLC(bool enable)
{
    isFilteringDLC = enable;
    invalidateFilter();
}

void ThemeFilterProxyModel::setFilterHidden(bool enable)
{
    isFilteringHidden = enable;
    invalidateFilter();
}