// SPDX-License-Identifier: GPL-2.0 #ifndef DIVETRIPMODEL_H #define DIVETRIPMODEL_H #include "core/dive.h" #include "core/subsurface-qt/divelistnotifier.h" #include #include #include class DiveFilter; // There are two different representations of the dive list: // 1) Tree view: two-level model where dives are grouped by trips // 2) List view: one-level model where dives are sorted by one out // of many keys (e.g. date, depth, etc.). // // These two representations are realized by two classes, viz. // DiveTripModelTree and DiveTripModelList. Both classes derive // from DiveTripModelBase, which implements common features (e.g. // definition of the column types, access of data from the core // structures) and a common interface. class DiveTripModelBase : public QAbstractItemModel { Q_OBJECT public: enum Column { NR, DATE, RATING, DEPTH, DURATION, TEMPERATURE, TOTALWEIGHT, SUIT, CYLINDER, GAS, SAC, OTU, MAXCNS, TAGS, PHOTOS, BUDDIES, DIVEGUIDE, COUNTRY, LOCATION, COLUMNS }; enum ExtraRoles { STAR_ROLE = Qt::UserRole + 1, IS_TRIP_ROLE, DIVE_ROLE, TRIP_ROLE, DIVE_IDX, SELECTED_ROLE, CURRENT_ROLE, TRIP_HAS_CURRENT_ROLE, // Returns true if this is a trip and it contains the current dive LAST_ROLE }; enum Layout { TREE, LIST, }; // Call after having set the model to be informed of the current selection. void initSelection(); Qt::ItemFlags flags(const QModelIndex &index) const; QVariant headerData(int section, Qt::Orientation orientation, int role = Qt::DisplayRole) const override; DiveTripModelBase(QObject *parent = 0); int columnCount(const QModelIndex&) const; // Used for sorting. This is a bit of a layering violation, as sorting should be performed // by the higher-up QSortFilterProxyModel, but it makes things so much easier! virtual bool lessThan(const QModelIndex &i1, const QModelIndex &i2) const = 0; protected slots: void reset(); signals: // The propagation of selection changes is complex. // The control flow of dive-selection goes: // Commands/DiveListNotifier ---(dive */dive_trip *)---> DiveTripModel ---(QModelIndex)---> DiveListView // i.e. The command objects send changes in terms of pointer-to-dives, which the DiveTripModel transforms // into QModelIndexes according to the current view (tree/list). Finally, the DiveListView transforms these // indices into local indices according to current sorting/filtering and instructs the QSelectionModel to // perform the appropriate actions. void selectionChanged(const QVector &indices); void currentDiveChanged(QModelIndex index); protected: dive *oldCurrent; QBrush invalidForeground; QFont invalidFont; // Access trip and dive data QVariant diveData(const struct dive *d, int column, int role) const; // Not static because we have to access invalidFont static QVariant tripData(const dive_trip *trip, int column, int role); static QString tripTitle(const dive_trip *trip); static QString tripShortDate(const dive_trip *trip); static QString getDescription(int column); void currentChanged(); virtual dive *diveOrNull(const QModelIndex &index) const = 0; // Returns a dive if this index represents a dive, null otherwise virtual void clearData() = 0; virtual void populate() = 0; virtual QModelIndex diveToIdx(const dive *d) const = 0; }; class DiveTripModelTree final : public DiveTripModelBase { Q_OBJECT public slots: void divesAdded(dive_trip *trip, bool addTrip, const QVector &dives); void divesDeleted(dive_trip *trip, bool deleteTrip, const QVector &dives); void divesMovedBetweenTrips(dive_trip *from, dive_trip *to, bool deleteFrom, bool createTo, const QVector &dives); void diveSiteChanged(dive_site *ds, int field); void divesChanged(const QVector &dives); void diveChanged(dive *d); void divesTimeChanged(timestamp_t delta, const QVector &dives); void divesSelected(const QVector &dives); void tripChanged(dive_trip *trip, TripField); void filterReset(); public: DiveTripModelTree(QObject *parent = nullptr); int tripInDirection(const struct dive *d, int direction) const; private: int rowCount(const QModelIndex &parent) const override; void clearData() override; void populate() override; QModelIndex index(int row, int column, const QModelIndex &parent) const override; QModelIndex parent(const QModelIndex &index) const override; QVariant data(const QModelIndex &index, int role) const override; bool lessThan(const QModelIndex &i1, const QModelIndex &i2) const override; void divesSelectedTrip(dive_trip *trip, const QVector &dives, QVector &); dive *diveOrNull(const QModelIndex &index) const override; void divesChangedTrip(dive_trip *trip, const QVector &dives); void divesShown(dive_trip *trip, const QVector &dives); void divesHidden(dive_trip *trip, const QVector &dives); void divesTimeChangedTrip(dive_trip *trip, timestamp_t delta, const QVector &dives); void divesDeletedInternal(dive_trip *trip, bool deleteTrip, const QVector &dives); // The tree model has two levels. At the top level, we have either trips or dives // that do not belong to trips. Such a top-level item is represented by the "Item" // struct, which is based on the dive_or_trip structure. // If it is a trip, additionally, the dives are collected in a vector. // The items are ordered chronologically according to the dive_or_trip_less_than() // function, which guarantees a stable ordering even in the case of equal timestamps. // For dives and trips, it place dives chronologically after trips, so that in // the default-descending view they are shown before trips. struct Item { dive_or_trip d_or_t; std::vector dives; // std::vector<> instead of QVector for insert() with three iterators Item(dive_trip *t, const QVector &dives); Item(dive_trip *t, dive *d); // Initialize a trip with one dive Item(dive *d); // Initialize a top-level dive bool isDive(const dive *) const; // Helper function: is this the give dive? dive *getDive() const; // Helper function: returns top-level-dive or null timestamp_t when() const; // Helper function: start time of dive *or* trip }; std::vector items; // Use std::vector for convenience of emplace_back() dive_or_trip tripOrDive(const QModelIndex &index) const; // Returns either a pointer to a trip or a dive, or twice null of index is invalid // null, something is really wrong // Addition and deletion of dives and trips void addTrip(dive_trip *trip, const QVector &dives); void addDivesToTrip(int idx, const QVector &dives); void addDivesTopLevel(const QVector &dives); void removeDivesFromTrip(int idx, const QVector &dives); void removeDivesTopLevel(const QVector &dives); void removeTrip(int idx); void topLevelChanged(int idx); // Access trips and dives int findTripIdx(const dive_trip *trip) const; int findDiveIdx(const dive *d) const; // Find _top_level_ dive QModelIndex diveToIdx(const dive *d) const; // Find _any_ dive int findDiveInTrip(int tripIdx, const dive *d) const; // Find dive inside trip. Second parameter is index of trip int findInsertionIndex(const dive_trip *trip) const; // Where to insert trip // Comparison function between dive and arbitrary entry static bool dive_before_entry(const dive *d, const Item &entry); }; class DiveTripModelList final : public DiveTripModelBase { Q_OBJECT public slots: void divesAdded(dive_trip *trip, bool addTrip, const QVector &dives); void divesDeleted(dive_trip *trip, bool deleteTrip, const QVector &dives); void diveSiteChanged(dive_site *ds, int field); void divesChanged(const QVector &dives); void diveChanged(dive *d); void divesTimeChanged(timestamp_t delta, const QVector &dives); // Does nothing in list view. //void divesMovedBetweenTrips(dive_trip *from, dive_trip *to, bool deleteFrom, bool createTo, const QVector &dives); void divesSelected(const QVector &dives); void filterReset(); public: DiveTripModelList(QObject *parent = nullptr); private: int rowCount(const QModelIndex &parent) const override; void clearData() override; void populate() override; QModelIndex index(int row, int column, const QModelIndex &parent) const override; QModelIndex parent(const QModelIndex &index) const override; QVariant data(const QModelIndex &index, int role) const override; bool lessThan(const QModelIndex &i1, const QModelIndex &i2) const override; dive *diveOrNull(const QModelIndex &index) const override; void addDives(QVector &dives); void removeDives(QVector dives); QModelIndex diveToIdx(const dive *d) const; void divesDeletedInternal(const QVector &dives); std::vector items; // TODO: access core data directly }; #endif