haariface_p.h

Go to the documentation of this file.

/* ============================================================
 *
 * This file is a part of digiKam project
 * https://www.digikam.org
 *
 * Date        : 2003-01-17
 * Description : Haar Database interface
 *
 * Copyright (C) 2016-2018 by Mario Frank <mario dot frank at uni minus potsdam dot de>
 * Copyright (C) 2003      by Ricardo Niederberger Cabral <nieder at mail dot ru>
 * Copyright (C) 2009-2022 by Gilles Caulier <caulier dot gilles at gmail dot com>
 * Copyright (C) 2009-2013 by Marcel Wiesweg <marcel dot wiesweg at gmx dot de>
 * Copyright (C) 2009-2011 by Andi Clemens <andi dot clemens at gmail dot 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;
 * 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_HAAR_IFACE_P_H
#define DIGIKAM_HAAR_IFACE_P_H
 
#include "haariface.h"
 
// C++ includes
 
#include <fstream>
#include <cmath>
#include <cstring>
 
// Qt includes
 
#include <QByteArray>
#include <QDataStream>
#include <QImage>
#include <QImageReader>
#include <QMap>
 
// Local includes
 
#include "digikam_debug.h"
#include "jpegutils.h"
#include "dimg.h"
#include "iteminfo.h"
#include "coredbaccess.h"
#include "coredbtransaction.h"
#include "coredb.h"
#include "coredbbackend.h"
#include "coredbsearchxml.h"
#include "dbenginesqlquery.h"
#include "similaritydb.h"
#include "similaritydbaccess.h"
 
using namespace std;
 
// TODO: Always store similarities in the similaritydb
 
namespace Digikam
{
 
typedef QMap<qlonglong, Haar::SignatureData> SignatureCache;
typedef QMap<qlonglong, int>                 AlbumCache;
 
class Q_DECL_HIDDEN DatabaseBlob
{
public:
 
    enum
    {
        Version = 1
    };
 
public:
 
    DatabaseBlob() = default;
 
    void read(const QByteArray& array, Haar::SignatureData& data);
 
    QByteArray write(Haar::SignatureData& data);
};
 
// -----------------------------------------------------------------------------------------------------
 
class Q_DECL_HIDDEN HaarIface::Private
{
public:
 
    explicit Private();
    ~Private();
 
public:
 
    void rebuildSignatureCache(const QSet<qlonglong>& imageIds = {});
    bool hasSignatureCache() const;
 
    bool retrieveSignatureFromCache(qlonglong imageId, Haar::SignatureData& data);
 
    void setImageDataFromImage(const QImage& image);
    void setImageDataFromImage(const DImg& image);
 
    SignatureCache*  signatureCache()    const;
    AlbumCache*      albumCache()        const;
    Haar::ImageData* imageData()         const;
 
    void setAlbumRootsToSearch(const QSet<int>& albumRootIds);
    const QSet<int>& albumRootsToSearch() const;
 
public:
 
    const static QString            signatureQuery;
    const static Haar::WeightBin    weightBin;
 
private:
 
    QScopedPointer<SignatureCache>  m_signatureCache;
    QScopedPointer<AlbumCache>      m_albumCache;
 
    QScopedPointer<Haar::ImageData> m_data;
 
    QSet<int>                       m_albumRootsToSearch;
};
 
} // namespace Digikam
 
#endif // DIGIKAM_HAAR_IFACE_P_H