Files
gcs-nf/opmap/core/pureimagecache.cpp
T
2019-12-31 18:56:34 +08:00

332 lines
12 KiB
C++

/**
******************************************************************************
*
* @file pureimagecache.cpp
* @author The OpenPilot Team, http://www.openpilot.org Copyright (C) 2012.
* @brief
* @see The GNU Public License (GPL) Version 3
* @defgroup OPMapWidget
* @{
*
*****************************************************************************/
/*
* 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 3 of the License, 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.
*
* You should have received a copy of the GNU General Public License along
* with this program; if not, write to the Free Software Foundation, Inc.,
* 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
*/
#include "pureimagecache.h"
#include <QDateTime>
#include <QSettings>
// #define DEBUG_PUREIMAGECACHE
namespace core {
qlonglong PureImageCache::ConnCounter = 0;
PureImageCache::PureImageCache()
{}
void PureImageCache::setGtileCache(const QString &value)
{
lock.lockForWrite();
gtilecache = value;
QDir d;
if (!d.exists(gtilecache)) {
d.mkdir(gtilecache);
#ifdef DEBUG_PUREIMAGECACHE
qDebug() << "Create Cache directory";
#endif // DEBUG_PUREIMAGECACHE
}
{
QString db = gtilecache + "Data.qmdb";
if (!QFileInfo(db).exists()) {
#ifdef DEBUG_PUREIMAGECACHE
qDebug() << "Try to create EmptyDB";
#endif // DEBUG_PUREIMAGECACHE
CreateEmptyDB(db);
}
}
lock.unlock();
}
QString PureImageCache::GtileCache()
{
return gtilecache;
}
bool PureImageCache::CreateEmptyDB(const QString &file)
{
#ifdef DEBUG_PUREIMAGECACHE
qDebug() << "Create database at!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!:" << file;
#endif // DEBUG_PUREIMAGECACHE
QFileInfo File(file);
QDir dir = File.absoluteDir();
QString path = dir.absolutePath();
QString filename = File.fileName();
if (File.exists()) {
QFile(filename).remove();
}
if (!dir.exists()) {
#ifdef DEBUG_PUREIMAGECACHE
qDebug() << "CreateEmptyDB: Cache path doesn't exist, try to create";
#endif // DEBUG_PUREIMAGECACHE
if (!dir.mkpath(path)) {
#ifdef DEBUG_PUREIMAGECACHE
qDebug() << "CreateEmptyDB: Could not create path";
#endif // DEBUG_PUREIMAGECACHE
return false;
}
}
QSqlDatabase db;
db = QSqlDatabase::addDatabase("QSQLITE", QLatin1String("CreateConn"));
db.setDatabaseName(file);
if (!db.open()) {
#ifdef DEBUG_PUREIMAGECACHE
qDebug() << "CreateEmptyDB: Unable to create database";
#endif // DEBUG_PUREIMAGECACHE
return false;
}
QSqlQuery query(db);
query.exec("CREATE TABLE IF NOT EXISTS Tiles (id INTEGER NOT NULL PRIMARY KEY, X INTEGER NOT NULL, Y INTEGER NOT NULL, Zoom INTEGER NOT NULL, Type INTEGER NOT NULL,Date TEXT)");
if (query.numRowsAffected() == -1) {
#ifdef DEBUG_PUREIMAGECACHE
qDebug() << "CreateEmptyDB: " << query.lastError().driverText();
#endif // DEBUG_PUREIMAGECACHE
db.close();
return false;
}
query.exec("CREATE TABLE IF NOT EXISTS TilesData (id INTEGER NOT NULL PRIMARY KEY CONSTRAINT fk_Tiles_id REFERENCES Tiles(id) ON DELETE CASCADE, Tile BLOB NULL)");
if (query.numRowsAffected() == -1) {
#ifdef DEBUG_PUREIMAGECACHE
qDebug() << "CreateEmptyDB: " << query.lastError().driverText();
#endif // DEBUG_PUREIMAGECACHE
db.close();
return false;
}
query.exec(
"CREATE TRIGGER fki_TilesData_id_Tiles_id "
"BEFORE INSERT ON [TilesData] "
"FOR EACH ROW BEGIN "
"SELECT RAISE(ROLLBACK, 'insert on table TilesData violates foreign key constraint fki_TilesData_id_Tiles_id') "
"WHERE (SELECT id FROM Tiles WHERE id = NEW.id) IS NULL; "
"END");
if (query.numRowsAffected() == -1) {
#ifdef DEBUG_PUREIMAGECACHE
qDebug() << "CreateEmptyDB: " << query.lastError().driverText();
#endif // DEBUG_PUREIMAGECACHE
db.close();
return false;
}
query.exec(
"CREATE TRIGGER fku_TilesData_id_Tiles_id "
"BEFORE UPDATE ON [TilesData] "
"FOR EACH ROW BEGIN "
"SELECT RAISE(ROLLBACK, 'update on table TilesData violates foreign key constraint fku_TilesData_id_Tiles_id') "
"WHERE (SELECT id FROM Tiles WHERE id = NEW.id) IS NULL; "
"END");
if (query.numRowsAffected() == -1) {
#ifdef DEBUG_PUREIMAGECACHE
qDebug() << "CreateEmptyDB: " << query.lastError().driverText();
#endif // DEBUG_PUREIMAGECACHE
db.close();
return false;
}
query.exec(
"CREATE TRIGGER fkdc_TilesData_id_Tiles_id "
"BEFORE DELETE ON Tiles "
"FOR EACH ROW BEGIN "
"DELETE FROM TilesData WHERE TilesData.id = OLD.id; "
"END");
if (query.numRowsAffected() == -1) {
#ifdef DEBUG_PUREIMAGECACHE
qDebug() << "CreateEmptyDB: " << query.lastError().driverText();
#endif // DEBUG_PUREIMAGECACHE
db.close();
return false;
}
db.close();
QSqlDatabase::removeDatabase(QLatin1String("CreateConn"));
return true;
}
bool PureImageCache::PutImageToCache(const QByteArray &tile, const MapType::Types &type, const Point &pos, const int &zoom)
{
if (gtilecache.isEmpty() | gtilecache.isNull()) {
return false;
}
lock.lockForRead();
#ifdef DEBUG_PUREIMAGECACHE
qDebug() << "PutImageToCache Start:"; // <<pos;
#endif // DEBUG_PUREIMAGECACHE
Mcounter.lock();
qlonglong id = ++ConnCounter;
Mcounter.unlock();
{
QSqlDatabase cn;
cn = QSqlDatabase::addDatabase("QSQLITE", QString::number(id));
QString db = gtilecache + "Data.qmdb";
cn.setDatabaseName(db);
cn.setConnectOptions("QSQLITE_ENABLE_SHARED_CACHE");
if (cn.open()) {
{
QSqlQuery query(cn);
query.prepare("INSERT INTO Tiles(X, Y, Zoom, Type,Date) VALUES(?, ?, ?, ?,?)");
query.addBindValue(pos.X());
query.addBindValue(pos.Y());
query.addBindValue(zoom);
query.addBindValue((int)type);
query.addBindValue(QDateTime::currentDateTime().toString());
query.exec();
}
{
QSqlQuery query(cn);
query.prepare("INSERT INTO TilesData(id, Tile) VALUES((SELECT last_insert_rowid()), ?)");
query.addBindValue(tile);
query.exec();
}
cn.close();
}
}
QSqlDatabase::removeDatabase(QString::number(id));
lock.unlock();
return true;
}
QByteArray PureImageCache::GetImageFromCache(MapType::Types type, Point pos, int zoom)
{
lock.lockForRead();
QByteArray ar;
if (gtilecache.isEmpty() | gtilecache.isNull()) {
return ar;
}
QString dir = gtilecache;
Mcounter.lock();
qlonglong id = ++ConnCounter;
Mcounter.unlock();
#ifdef DEBUG_PUREIMAGECACHE
qDebug() << "Cache dir=" << dir << " Try to GET:" << pos.X() + "," + pos.Y();
#endif // DEBUG_PUREIMAGECACHE
{
QString db = dir + "Data.qmdb";
{
QSqlDatabase cn;
cn = QSqlDatabase::addDatabase("QSQLITE", QString::number(id));
cn.setDatabaseName(db);
cn.setConnectOptions("QSQLITE_ENABLE_SHARED_CACHE");
if (cn.open()) {
{
QSqlQuery query(cn);
query.exec(QString("SELECT Tile FROM TilesData WHERE id = (SELECT id FROM Tiles WHERE X=%1 AND Y=%2 AND Zoom=%3 AND Type=%4)").arg(pos.X()).arg(pos.Y()).arg(zoom).arg((int)type));
query.next();
if (query.isValid()) {
ar = query.value(0).toByteArray();
}
}
cn.close();
}
}
}
QSqlDatabase::removeDatabase(QString::number(id));
lock.unlock();
return ar;
}
void PureImageCache::deleteOlderTiles(int const & days)
{
if (gtilecache.isEmpty() | gtilecache.isNull()) {
return;
}
QList<long> add;
bool ret = true;
QString dir = gtilecache;
{
QString db = dir + "Data.qmdb";
ret = QFileInfo(db).exists();
if (ret) {
QSqlDatabase cn;
Mcounter.lock();
qlonglong id = ++ConnCounter;
Mcounter.unlock();
cn = QSqlDatabase::addDatabase("QSQLITE", QString::number(id));
cn.setDatabaseName(db);
cn.setConnectOptions("QSQLITE_ENABLE_SHARED_CACHE");
if (cn.open()) {
{
QSqlQuery query(cn);
query.exec(QString("SELECT id, X, Y, Zoom, Type, Date FROM Tiles"));
while (query.next()) {
if (QDateTime::fromString(query.value(5).toString()).daysTo(QDateTime::currentDateTime()) > days) {
add.append(query.value(0).toLongLong());
}
}
foreach(long i, add) {
query.exec(QString("DELETE FROM Tiles WHERE id = %1;").arg(i));
}
}
cn.close();
}
QSqlDatabase::removeDatabase(QString::number(id));
}
}
}
// PureImageCache::ExportMapDataToDB("C:/Users/Xapo/Documents/mapcontrol/debug/mapscache/data.qmdb","C:/Users/Xapo/Documents/mapcontrol/debug/mapscache/data2.qmdb");
bool PureImageCache::ExportMapDataToDB(QString sourceFile, QString destFile)
{
bool ret = true;
QList<long> add;
if (!QFileInfo(destFile).exists()) {
#ifdef DEBUG_PUREIMAGECACHE
qDebug() << "Try to create EmptyDB";
#endif // DEBUG_PUREIMAGECACHE
ret = CreateEmptyDB(destFile);
}
if (!ret) {
return false;
}
QSqlDatabase ca = QSqlDatabase::addDatabase("QSQLITE", "ca");
ca.setDatabaseName(sourceFile);
if (ca.open()) {
QSqlDatabase cb = QSqlDatabase::addDatabase("QSQLITE", "cb");
cb.setDatabaseName(destFile);
if (cb.open()) {
QSqlQuery queryb(cb);
queryb.exec(QString("ATTACH DATABASE \"%1\" AS Source").arg(sourceFile));
QSqlQuery querya(ca);
querya.exec("SELECT id, X, Y, Zoom, Type, Date FROM Tiles");
while (querya.next()) {
long id = querya.value(0).toLongLong();
queryb.exec(QString("SELECT id FROM Tiles WHERE X=%1 AND Y=%2 AND Zoom=%3 AND Type=%4;").arg(querya.value(1).toLongLong()).arg(querya.value(2).toLongLong()).arg(querya.value(3).toLongLong()).arg(querya.value(4).toLongLong()));
if (!queryb.next()) {
add.append(id);
}
}
long f;
foreach(f, add) {
queryb.exec(QString("INSERT INTO Tiles(X, Y, Zoom, Type, Date) SELECT X, Y, Zoom, Type, Date FROM Source.Tiles WHERE id=%1").arg(f));
queryb.exec(QString("INSERT INTO TilesData(id, Tile) Values((SELECT last_insert_rowid()), (SELECT Tile FROM Source.TilesData WHERE id=%1))").arg(f));
}
add.clear();
ca.close();
cb.close();
} else { return false; }
} else { return false; }
QSqlDatabase::removeDatabase("ca");
QSqlDatabase::removeDatabase("cb");
return true;
}
}