albummodel.h

Go to the documentation of this file.

/* ============================================================
 *
 * This file is a part of digiKam project
 * https://www.digikam.org
 *
 * Date        : 2008-03-22
 * Description : Qt Model for Albums
 *
 * Copyright (C) 2008-2011 by Marcel Wiesweg <marcel dot wiesweg at gmx dot de>
 *
 * 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;
 * either version 2, or (at your option)
 * any later version.
 *
 * 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.
 *
 * ============================================================ */
 
#ifndef DIGIKAM_ALBUM_MODEL_H
#define DIGIKAM_ALBUM_MODEL_H
 
// Qt includes
 
#include <QDateTime>
 
// Local includes
 
#include "abstractalbummodel.h"
#include "digikam_export.h"
 
namespace Digikam
{
 
class DIGIKAM_GUI_EXPORT AlbumModel : public AbstractCheckableAlbumModel
{
    Q_OBJECT
 
public:
 
    explicit AlbumModel(RootAlbumBehavior rootBehavior = IncludeRootAlbum, QObject* const parent = nullptr);
    ~AlbumModel() override;
 
    PAlbum* albumForIndex(const QModelIndex& index) const;
 
protected:
 
    QVariant albumData(Album* a, int role)          const override;
    QVariant decorationRoleData(Album* a)           const override;
    Album*   albumForId(int id)                     const override;
};
 
// ------------------------------------------------------------------
 
class DIGIKAM_GUI_EXPORT TagModel : public AbstractCheckableAlbumModel
{
    Q_OBJECT
 
public:
 
    explicit TagModel(RootAlbumBehavior rootBehavior = IncludeRootAlbum, QObject* const parent = nullptr);
 
    TAlbum*      albumForIndex(const QModelIndex& index) const;
    void         setColumnHeader(const QString& header);
    void         activateFaceTagModel();
 
protected:
 
    QVariant     albumData(Album* a, int role)           const override;
    QVariant     decorationRoleData(Album* a)            const override;
    Album*       albumForId(int id)                      const override;
    QVariant     fontRoleData(Album* a)                  const override;
    bool         setData(const QModelIndex& index,
                         const QVariant& value,
                         int role = Qt::EditRole)              override;
 
private:
 
    QMap<int, int> m_unconfirmedFaceCount;
};
 
// ------------------------------------------------------------------
 
class DIGIKAM_GUI_EXPORT SearchModel : public AbstractCheckableAlbumModel
{
    Q_OBJECT
 
public:
 
    explicit SearchModel(QObject* const parent = nullptr);
 
    SAlbum* albumForIndex(const QModelIndex& index) const;
 
    void setReplaceNames(const QHash<QString, QString>& replaceNames);
    void addReplaceName(const QString& technicalName, const QString& userVisibleName);
 
    void setPixmapForNormalSearches(const QPixmap& pix);
    void setDefaultPixmap(const QPixmap& pix);
    void setPixmapForTimelineSearches(const QPixmap& pix);
    void setPixmapForHaarSearches(const QPixmap& pix);
    void setPixmapForMapSearches(const QPixmap& pix);
    void setPixmapForDuplicatesSearches(const QPixmap& pix);
 
protected:
 
    QVariant albumData(Album* a, int role)          const override;
    Album*   albumForId(int id)                     const override;
 
private Q_SLOTS:
 
    void albumSettingsChanged();
 
protected:
 
    QHash<int, QPixmap>     m_pixmaps;
    QHash<QString, QString> m_replaceNames;
};
 
// ------------------------------------------------------------------
 
class DIGIKAM_GUI_EXPORT DateAlbumModel : public AbstractCountingAlbumModel
{
    Q_OBJECT
 
public:
 
    explicit DateAlbumModel(QObject* const parent = nullptr);
 
    DAlbum* albumForIndex(const QModelIndex& index)     const;
 
    QModelIndex monthIndexForDate(const QDate& date)    const;
 
    void setPixmaps(const QPixmap& forYearAlbums,
                    const QPixmap& forMonthAlbums);
 
public Q_SLOTS:
 
    void setYearMonthMap(const QMap<YearMonth, int>& yearMonthMap);
 
protected:
 
    QString  albumName(Album* a)                        const override;
    QVariant decorationRoleData(Album* a)               const override;
    QVariant sortRoleData(Album* a)                     const override;
    Album*   albumForId(int id)                         const override;
 
protected:
 
    QPixmap m_yearPixmap;
    QPixmap m_monthPixmap;
};
 
} // namespace Digikam
 
#endif // DIGIKAM_ALBUM_MODEL_H