pokemonoverviewmodel.h

Go to the documentation of this file.

/*
  * Copyright 2026 Twilight
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
  * You may obtain a copy of the License at
  *
  *   http://www.apache.org/licenses/LICENSE-2.0
  *
  * Unless required by applicable law or agreed to in writing, software
  * distributed under the License is distributed on an "AS IS" BASIS,
  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  * See the License for the specific language governing permissions and
  * limitations under the License.
*/
#pragma once
#include <QObject>
#include <QString>
#include <QStringList>
#include <QVector>
#include <QCollator>
#include <QAbstractListModel>
 
class Storage;
class PokemonStorageBox;
class PlayerBasics;

class PokemonOverviewModel : public QAbstractListModel
{
  Q_OBJECT

  Q_PROPERTY(QStringList columns READ columns NOTIFY columnsChanged)
 
  
  Q_PROPERTY(int sortSelect MEMBER sortSelect NOTIFY sortSelectChanged)
  Q_PROPERTY(QString sortLabel READ sortLabel NOTIFY sortSelectChanged)
  Q_PROPERTY(QString sortIcon READ sortIcon NOTIFY sortSelectChanged)
 
public:
  enum OverviewRoles {
    NameRole = Qt::UserRole + 1, 
    CountsRole,                  
    TooltipsRole,                
  };

  enum SortSelect {
    SortBegin,
    SortDex,      
    SortName,     
    SortInternal, 
    SortEnd
  };

  PokemonOverviewModel(PokemonStorageBox* party, Storage* storage, PlayerBasics* basics);
 
  virtual int rowCount(const QModelIndex& parent) const override;           
  virtual QVariant data(const QModelIndex& index, int role) const override; 
  virtual QHash<int, QByteArray> roleNames() const override;                
 
  QStringList columns() const; 
  QString sortLabel() const;   
  QString sortIcon() const;    
 
  Q_INVOKABLE void sortCycle(); 
 
public slots:
  void rebuild(); 
 
signals:
  void columnsChanged();
  void sortSelectChanged();
 
private:
  struct Cell {
    int count = 0;
    QString tooltip;
  };

  struct Row {
    QString name; 
    int dex;      
    int id;       
    QVector<Cell> cells;
  };

  Cell buildCell(PokemonStorageBox* box, int speciesId) const;
 
  void applySort(); 
 
  PokemonStorageBox* party = nullptr; 
  Storage* storage = nullptr;         
  PlayerBasics* basics = nullptr;     
 
  QVector<PokemonStorageBox*> colBoxes; 
  QStringList colLabels;                
  QVector<Row> rows;                    
 
  int sortSelect = SortName; 
};