/*
Copyright (C) 2012 Sebastian Herbord. All rights reserved.

This file is part of Mod Organizer.

Mod Organizer 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 3 of the License, or
(at your option) any later version.

Mod Organizer 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 Mod Organizer.  If not, see <http://www.gnu.org/licenses/>.
*/

#ifndef MODLISTSORTPROXY_H
#define MODLISTSORTPROXY_H

#include <QSortFilterProxyModel>
#include <bitset>
#include "modlist.h"

class Profile;

class ModListSortProxy : public QSortFilterProxyModel
{
  Q_OBJECT

public:
  enum FilterMode
  {
    FilterAnd,
    FilterOr
  };

  enum CriteriaType {
    TypeSpecial,
    TypeCategory,
    TypeContent
  };

  enum SeparatorsMode
  {
    SeparatorFilter,
    SeparatorShow,
    SeparatorHide
  };

  struct Criteria
  {
    CriteriaType type;
    int id;
    bool inverse;

    bool operator==(const Criteria& other) const
    {
      return
        (type == other.type) &&
        (id == other.id) &&
        (inverse == other.inverse);
    }

    bool operator!=(const Criteria& other) const
    {
      return !(*this == other);
    }
  };

public:

  explicit ModListSortProxy(Profile *profile, QObject *parent = 0);

  void setProfile(Profile *profile);


  virtual Qt::ItemFlags flags(const QModelIndex &modelIndex) const;
  virtual bool dropMimeData(const QMimeData *data, Qt::DropAction action,
                            int row, int column, const QModelIndex &parent);

  virtual void setSourceModel(QAbstractItemModel *sourceModel) override;


  /**
   * @brief enable all mods visible under the current filter
   **/
  void enableAllVisible();

  /**
   * @brief disable all mods visible under the current filter
   **/
  void disableAllVisible();

  /**
   * @brief tests if a filtere matches for a mod
   * @param info mod information
   * @param enabled true if the mod is currently active
   * @return true if current active filters match for the specified mod
   */
  bool filterMatchesMod(ModInfo::Ptr info, bool enabled) const;

  /**
   * @return true if a filter is currently active
   */
  bool isFilterActive() const { return m_FilterActive; }

  void setCriteria(const std::vector<Criteria>& criteria);
  void setOptions(FilterMode mode, SeparatorsMode separators);

  /**
   * @brief tests if the specified index has child nodes
   * @param parent the node to test
   * @return true if there are child nodes
   */
  virtual bool hasChildren ( const QModelIndex & parent = QModelIndex() ) const {
    return rowCount(parent) > 0;
  }

  /**
   * @brief sets whether a column is visible
   * @param column the index of the column
   * @param visible the visibility of the column
   */
  void setColumnVisible(int column, bool visible);

public slots:

  void updateFilter(const QString &filter);

signals:

  void filterActive(bool active);

protected:

  virtual bool lessThan(const QModelIndex &left, const QModelIndex &right) const;
  virtual bool filterAcceptsRow(int row, const QModelIndex &parent) const;

private:

  unsigned long flagsId(const std::vector<ModInfo::EFlag> &flags) const;
  unsigned long conflictFlagsId(const std::vector<ModInfo::EConflictFlag>& flags) const;
  bool hasConflictFlag(const std::vector<ModInfo::EFlag> &flags) const;
  void updateFilterActive();
  bool filterMatchesModAnd(ModInfo::Ptr info, bool enabled) const;
  bool filterMatchesModOr(ModInfo::Ptr info, bool enabled) const;

private slots:

  void aboutToChangeData();
  void postDataChanged();

private:
  Profile* m_Profile;
  std::vector<Criteria> m_Criteria;
  QString m_Filter;
  std::bitset<ModList::COL_LASTCOLUMN + 1> m_EnabledColumns;

  bool m_FilterActive;
  FilterMode m_FilterMode;
  SeparatorsMode m_FilterSeparators;

  std::vector<Criteria> m_PreChangeCriteria;

  bool optionsMatchMod(ModInfo::Ptr info, bool enabled) const;
  bool criteriaMatchMod(ModInfo::Ptr info, bool enabled, const Criteria& c) const;
  bool categoryMatchesMod(ModInfo::Ptr info, bool enabled, int category) const;
  bool contentMatchesMod(ModInfo::Ptr info, bool enabled, int content) const;
};

#endif // MODLISTSORTPROXY_H