/*

 * 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();

}