RasterProcessTool/RasterMainWidgetGUI/RasterMainWidget/tmsprovider.cpp

#include <QImage>
#include <QPainter>
#include <QThreadPool>
#include <QDir>
#include <QSqlQuery>
#include <QSqlError>

#include <tmsprovider.h>
#include <network.h>
#pragma execution_character_set("utf-8")

namespace LAMPMainWidget
{
	TileDownloadTask::TileDownloadTask(TileInfo tile, QObject* parent)
		: QObject(parent),
		mTile(std::move(tile))
	{
	}

	void
		TileDownloadTask::run()
	{
		Network web{};
		QByteArray data = web.httpsRequest(mTile.url);
		mTile.data = data;
		tileReady(mTile);
	}

	TmsProvider::TmsProvider(QObject* parent)
		: LayerProvider(parent),
		mImage(nullptr)
	{
		if (!QFile::exists(mDbName)) {
			QFile dbFile{ mDbName };
			dbFile.open(QIODevice::ReadWrite);
			dbFile.close();
		}
	}

	TmsProvider::~TmsProvider()
	{
		delete mImage;
	}

	bool
		TmsProvider::initCache()
	{
		mDbConn = QSqlDatabase::addDatabase("QSQLITE", id());
		mDbConn.setDatabaseName(mDbName);
		if (!mDbConn.open()) {
			qCritical() << "缓存数据库打开失败";
			return false;
		}

		QSqlQuery sql{ mDbConn };
		if (!mDbConn.tables().contains(mTableName)) {
			sql.prepare(QString("create table %1 ("
				"id integer primary key autoincrement,"
				"provider text not null,"
				"zoom integer not null,"
				"position text not null,"
				"data blob not null"
				")").arg(mTableName));
			if (!sql.exec()) {
				qCritical() << "缓存表创建失败" << sql.lastError().text();
				return false;
			}
		}

		mDbConn.close();
		return true;
	}

	QByteArray
		TmsProvider::getCache(const QPoint& pos, int zoom)
	{
		QByteArray res{};
		mDbConn.open();
		QSqlQuery select{ mDbConn };
		select.prepare(QString("select data from %1 "
			"where provider = :provider "
			"and zoom = :zoom "
			"and position = :position "
			"order by id desc limit 1"
		).arg(mTableName));
		select.bindValue(":provider", id());
		select.bindValue(":zoom", zoom);
		select.bindValue(":position", QString("%1:%2").arg(pos.x()).arg(pos.y()));
		if (!select.exec()) {
			qDebug() << pos << "查询失败=>" << select.lastError().text();
			mDbConn.close();
			return res;
		}
		if (!select.seek(0)) {
			mDbConn.close();
			return res;
		}

		res = select.value(0).toByteArray();
		mDbConn.close();
		return res;
	}

	bool
		TmsProvider::addCache(const QPoint& pos, int zoom, QByteArray data)
	{
		if (data.isEmpty()) {
			qWarning() << "瓦片数据为空";
			return false;
		}
		mDbConn.open();
		QSqlQuery sql{ mDbConn };
		sql.prepare(QString{
		  "insert into %1 (provider, zoom, position, data)"
		  "values( :provider, :zoom, :position, :data)"
			}.arg(mTableName));
		sql.bindValue(":tbname", mTableName);
		sql.bindValue(":provider", id());
		sql.bindValue(":zoom", zoom);
		sql.bindValue(":position", QString("%1:%2").arg(pos.x()).arg(pos.y()));
		sql.bindValue(":data", data);

		if (!sql.exec()) {
			qCritical() << pos << "=>" << sql.lastError().text();
			mDbConn.close();
			return false;
		}

		mDbConn.close();
		return true;
	}

	bool
		TmsProvider::cacheContains(const QPoint& pos, int zoom)
	{
		QByteArray res = getCache(pos, zoom);
		return !res.isEmpty();
	}

	void
		TmsProvider::createTask(const QRectF& rect, int zoom)
	{
		newImage(rect);

		/// 创建下载瓦片任务
		const QSize sz = tileSize();
		int xMin = qFloor(rect.topLeft().x() / sz.width());
		double xOffset = rect.topLeft().x() / sz.width() - xMin;
		int xMax = qFloor(rect.bottomRight().x() / sz.width());
		int yMin = qFloor(rect.topLeft().y() / sz.height());
		double yOffset = rect.topLeft().y() / sz.height() - yMin;
		int yMax = qFloor(rect.bottomRight().y() / sz.height());
		for (int i = xMin; i <= xMax; ++i) {
			for (int j = yMin; j <= yMax; ++j) {
				TileInfo tile{};
				tile.index = QPointF{ i - xMin - xOffset, j - yMin - yOffset };
				tile.position = QPoint{ i, j };
				tile.zoom = zoom;
				tile.coord = QPoint{ i - xMin, j - yMin };
				tile.url = tileUrl(PointXY{ static_cast<double>(i), static_cast<double>(j) }, zoom);
				auto tileData = getCache(tile.position, tile.zoom);
				if (!tileData.isEmpty()) {
					tile.data = tileData;
					tileReady(tile);
					continue;
				}

				auto* task = new TileDownloadTask(tile);
				QObject::connect(task, &TileDownloadTask::tileReady, this, &TmsProvider::tileReady);
				QThreadPool::globalInstance()->start(task);
			}
		}
	}

	void
		TmsProvider::tileReady(TileInfo tile)
	{
		if (!cacheContains(tile.position, tile.zoom)) {
			addCache(tile.position, tile.zoom, tile.data);
		}
		QPainter painter{ mImage };
		QImage img = QImage::fromData(tile.data);
		if (img.isNull()) {
			return;
		}

		double xPos = tile.index.x() * tileSize().width();
		double yPos = tile.index.y() * tileSize().height();

		painter.drawImage(QPointF(xPos, yPos), img);
	}

	void
		TmsProvider::newImage(const QRectF& rect)
	{
		QSize imgSize{ int(rect.width()), int(rect.height()) };
		if (!mImage || imgSize != mImage->size()) {
			mImage = new QImage(imgSize, QImage::Format_RGB32);
		}
		mImage->fill(Qt::white);
	}
}