69 lines
1.9 KiB
C++
69 lines
1.9 KiB
C++
#include "image_forever.h"
|
|
#include <QMutex>
|
|
#include <QMutexLocker>
|
|
#include <QtCore/qbuffer.h>
|
|
#include <map>
|
|
|
|
struct ImageForeverItem {
|
|
QImage* image;
|
|
dust3d::Uuid id;
|
|
QByteArray* imageByteArray;
|
|
};
|
|
static std::map<dust3d::Uuid, ImageForeverItem> g_foreverMap;
|
|
static QMutex g_mapMutex;
|
|
|
|
const QImage* ImageForever::get(const dust3d::Uuid& id)
|
|
{
|
|
QMutexLocker locker(&g_mapMutex);
|
|
auto findResult = g_foreverMap.find(id);
|
|
if (findResult == g_foreverMap.end())
|
|
return nullptr;
|
|
return findResult->second.image;
|
|
}
|
|
|
|
void ImageForever::copy(const dust3d::Uuid& id, QImage& image)
|
|
{
|
|
QMutexLocker locker(&g_mapMutex);
|
|
auto findResult = g_foreverMap.find(id);
|
|
if (findResult == g_foreverMap.end())
|
|
return;
|
|
image = *findResult->second.image;
|
|
}
|
|
|
|
const QByteArray* ImageForever::getPngByteArray(const dust3d::Uuid& id)
|
|
{
|
|
QMutexLocker locker(&g_mapMutex);
|
|
auto findResult = g_foreverMap.find(id);
|
|
if (findResult == g_foreverMap.end())
|
|
return nullptr;
|
|
return findResult->second.imageByteArray;
|
|
}
|
|
|
|
dust3d::Uuid ImageForever::add(const QImage* image, dust3d::Uuid toId)
|
|
{
|
|
QMutexLocker locker(&g_mapMutex);
|
|
if (nullptr == image)
|
|
return dust3d::Uuid();
|
|
dust3d::Uuid newId = toId.isNull() ? dust3d::Uuid::createUuid() : toId;
|
|
if (g_foreverMap.find(newId) != g_foreverMap.end())
|
|
return newId;
|
|
QImage* newImage = new QImage(*image);
|
|
QByteArray* imageByteArray = new QByteArray();
|
|
QBuffer pngBuffer(imageByteArray);
|
|
pngBuffer.open(QIODevice::WriteOnly);
|
|
newImage->save(&pngBuffer, "PNG");
|
|
g_foreverMap[newId] = { newImage, newId, imageByteArray };
|
|
return newId;
|
|
}
|
|
|
|
void ImageForever::remove(const dust3d::Uuid& id)
|
|
{
|
|
QMutexLocker locker(&g_mapMutex);
|
|
auto findImage = g_foreverMap.find(id);
|
|
if (findImage == g_foreverMap.end())
|
|
return;
|
|
delete findImage->second.image;
|
|
delete findImage->second.imageByteArray;
|
|
g_foreverMap.erase(id);
|
|
}
|