gta5view/SnapmaticPicture.cpp

648 lines
19 KiB
C++
Raw Normal View History

2016-02-26 22:22:34 +01:00
/*****************************************************************************
* gta5sync GRAND THEFT AUTO V SYNC
* Copyright (C) 2016 Syping
2016-02-26 22:22:34 +01:00
*
2016-03-21 07:54:14 +01:00
* 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.
2016-02-26 22:22:34 +01:00
*
2016-03-21 07:54:14 +01:00
* 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.
2016-02-26 22:22:34 +01:00
*
2016-03-21 07:54:14 +01:00
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
2016-02-26 22:22:34 +01:00
*****************************************************************************/
2016-03-20 21:57:18 +01:00
#include "SnapmaticPicture.h"
2016-03-31 23:41:53 +02:00
#include "StringParser.h"
#include <QJsonDocument>
#include <QJsonObject>
#include <QStringList>
#include <QVariantMap>
#include <QJsonArray>
#include <QString>
2016-07-22 16:06:36 +02:00
#include <QBuffer>
#include <QDebug>
#include <QImage>
#include <QFile>
2016-02-26 22:22:34 +01:00
SnapmaticPicture::SnapmaticPicture(const QString &fileName, QObject *parent) : QObject(parent), picFileName(fileName)
2016-02-26 22:22:34 +01:00
{
2016-03-21 21:46:36 +01:00
// PARSE INT INIT - DO NOT CHANGE THIS VALUES
snapmaticHeaderLength = 278;
2016-03-23 01:09:27 +01:00
snapmaticUsefulLength = 260;
2016-07-22 16:06:36 +02:00
snapmaticFileMaxSize = 528192;
2016-03-21 21:46:36 +01:00
jpegHeaderLineDifStr = 2;
jpegPreHeaderLength = 14;
jpegPicStreamLength = 524288;
jsonStreamLength = 3076;
tideStreamLength = 260;
2016-03-21 21:46:36 +01:00
2016-07-22 16:06:36 +02:00
// PARSE EDITOR INIT
jpegStreamEditorBegin = 292;
jsonStreamEditorBegin = 524588;
jsonStreamEditorLength = 3072;
rawPicContent = "";
2016-03-21 21:46:36 +01:00
// INIT PIC
cachePicture = QImage(0, 0, QImage::Format_RGB32);
picExportFileName = "";
pictureStr = "";
lastStep = "";
sortStr = "";
titlStr = "";
descStr = "";
2016-03-21 22:29:57 +01:00
picOk = 0;
2016-03-21 21:46:36 +01:00
// INIT JSON
2016-03-21 22:29:57 +01:00
jsonOk = 0;
jsonStr = "";
}
SnapmaticPicture::~SnapmaticPicture()
{
}
bool SnapmaticPicture::readingPicture(bool writeEnabled_, bool cacheEnabled_)
{
// Start opening file
// lastStep is like currentStep
// Set boolean values
writeEnabled = writeEnabled_;
cacheEnabled = cacheEnabled_;
QFile *picFile = new QFile(picFileName);
2016-07-22 16:06:36 +02:00
QIODevice *picStream;
if (!picFile->open(QFile::ReadOnly))
{
lastStep = "1;/1,OpenFile," + StringParser::convertDrawStringForLog(picFileName);
picFile->deleteLater();
delete picFile;
return false;
}
2016-07-22 16:06:36 +02:00
rawPicContent = picFile->read(snapmaticFileMaxSize);
picFile->close();
delete picFile;
2016-07-22 16:06:36 +02:00
picStream = new QBuffer(&rawPicContent);
picStream->open(QIODevice::ReadWrite);
// Reading Snapmatic Header
2016-07-22 16:06:36 +02:00
if (!picStream->isReadable())
{
lastStep = "2;/3,ReadingFile," + StringParser::convertDrawStringForLog(picFileName) + ",1,NOHEADER";
2016-07-22 16:06:36 +02:00
picStream->close();
picStream->deleteLater();
delete picStream;
return false;
}
2016-07-22 16:06:36 +02:00
QByteArray snapmaticHeaderLine = picStream->read(snapmaticHeaderLength);
pictureStr = getSnapmaticPictureString(snapmaticHeaderLine);
// Reading JPEG Header Line
2016-07-22 16:06:36 +02:00
if (!picStream->isReadable())
{
lastStep = "2;/3,ReadingFile," + StringParser::convertDrawStringForLog(picFileName) + ",2,NOHEADER";
2016-07-22 16:06:36 +02:00
picStream->close();
picStream->deleteLater();
delete picStream;
return false;
}
2016-07-22 16:06:36 +02:00
QByteArray jpegHeaderLine = picStream->read(jpegPreHeaderLength);
// Checking for JPEG
2016-03-21 21:46:36 +01:00
jpegHeaderLine.remove(0, jpegHeaderLineDifStr);
if (jpegHeaderLine.left(4) != "JPEG")
{
lastStep = "2;/3,ReadingFile," + StringParser::convertDrawStringForLog(picFileName) + ",2,NOJPEG";
2016-07-22 16:06:36 +02:00
picStream->close();
picStream->deleteLater();
delete picStream;
return false;
}
2016-03-21 07:54:14 +01:00
// Read JPEG Stream
2016-07-22 16:06:36 +02:00
if (!picStream->isReadable())
2016-03-21 07:54:14 +01:00
{
lastStep = "2;/3,ReadingFile," + StringParser::convertDrawStringForLog(picFileName) + ",2,NOPIC";
2016-07-22 16:06:36 +02:00
picStream->close();
picStream->deleteLater();
delete picStream;
2016-03-21 07:54:14 +01:00
return false;
}
2016-07-22 16:06:36 +02:00
QByteArray jpegRawContent = picStream->read(jpegPicStreamLength);
if (cacheEnabled) picOk = cachePicture.loadFromData(jpegRawContent, "JPEG");
if (!cacheEnabled)
{
QImage tempPicture;
picOk = tempPicture.loadFromData(jpegRawContent, "JPEG");
}
2016-03-21 07:54:14 +01:00
// Read JSON Stream
2016-07-22 16:06:36 +02:00
if (!picStream->isReadable())
2016-03-21 07:54:14 +01:00
{
lastStep = "2;/3,ReadingFile," + StringParser::convertDrawStringForLog(picFileName) + ",3,NOJSON";
2016-07-22 16:06:36 +02:00
picStream->close();
picStream->deleteLater();
delete picStream;
2016-03-21 22:29:57 +01:00
return picOk;
2016-03-21 07:54:14 +01:00
}
2016-07-22 16:06:36 +02:00
else if (picStream->read(4) != "JSON")
2016-03-21 07:54:14 +01:00
{
lastStep = "2;/3,ReadingFile," + StringParser::convertDrawStringForLog(picFileName) + ",3,CTJSON";
2016-07-22 16:06:36 +02:00
picStream->close();
picStream->deleteLater();
delete picStream;
2016-03-21 22:29:57 +01:00
return picOk;
2016-03-21 07:54:14 +01:00
}
2016-07-22 16:06:36 +02:00
QByteArray jsonRawContent = picStream->read(jsonStreamLength);
2016-03-21 07:54:14 +01:00
jsonStr = getSnapmaticJSONString(jsonRawContent);
parseJsonContent(); // JSON parsing is own function
2016-03-21 07:54:14 +01:00
2016-07-22 16:06:36 +02:00
if (!picStream->isReadable())
{
lastStep = "2;/3,ReadingFile," + StringParser::convertDrawStringForLog(picFileName) + ",4,NOTITL";
2016-07-22 16:06:36 +02:00
picStream->close();
picStream->deleteLater();
delete picStream;
return picOk;
}
2016-07-22 16:06:36 +02:00
else if (picStream->read(4) != "TITL")
{
lastStep = "2;/3,ReadingFile," + StringParser::convertDrawStringForLog(picFileName) + ",4,CTTITL";
2016-07-22 16:06:36 +02:00
picStream->close();
picStream->deleteLater();
delete picStream;
return picOk;
}
2016-07-22 16:06:36 +02:00
QByteArray titlRawContent = picStream->read(tideStreamLength);
titlStr = getSnapmaticTIDEString(titlRawContent);
2016-07-22 16:06:36 +02:00
if (!picStream->isReadable())
{
lastStep = "2;/3,ReadingFile," + StringParser::convertDrawStringForLog(picFileName) + ",5,NODESC";
2016-07-22 16:06:36 +02:00
picStream->close();
picStream->deleteLater();
delete picStream;
return picOk;
}
2016-07-22 16:06:36 +02:00
else if (picStream->read(4) != "DESC")
{
lastStep = "2;/3,ReadingFile," + StringParser::convertDrawStringForLog(picFileName) + ",5,CTDESC";
2016-07-22 16:06:36 +02:00
picStream->close();
picStream->deleteLater();
delete picStream;
return picOk;
}
2016-07-22 16:06:36 +02:00
QByteArray descRawContent = picStream->read(tideStreamLength);
descStr = getSnapmaticTIDEString(descRawContent);
parseSnapmaticExportAndSortString();
2016-07-22 16:06:36 +02:00
picStream->close();
picStream->deleteLater();
delete picStream;
if (!writeEnabled) { rawPicContent.clear(); }
2016-03-21 22:29:57 +01:00
return picOk;
}
QString SnapmaticPicture::getSnapmaticPictureString(const QByteArray &snapmaticHeader)
{
2016-03-31 23:41:53 +02:00
QByteArray snapmaticBytes = snapmaticHeader.left(snapmaticUsefulLength);
QList<QByteArray> snapmaticBytesList = snapmaticBytes.split(char(0x01));
snapmaticBytes = snapmaticBytesList.at(1);
snapmaticBytesList.clear();
return StringParser::parseTitleString(snapmaticBytes, snapmaticBytes.length());
}
QString SnapmaticPicture::getSnapmaticJSONString(const QByteArray &jsonBytes)
2016-03-21 07:54:14 +01:00
{
QByteArray jsonUsefulBytes = jsonBytes;
jsonUsefulBytes.replace((char)0x00, "");
jsonUsefulBytes.replace((char)0x0c, "");
return QString::fromUtf8(jsonUsefulBytes).trimmed();
2016-03-21 07:54:14 +01:00
}
QString SnapmaticPicture::getSnapmaticTIDEString(const QByteArray &tideBytes)
{
QByteArray tideUsefulBytes = tideBytes;
tideUsefulBytes.remove(0,4);
2016-03-29 13:29:14 +02:00
QList<QByteArray> tideUsefulBytesList = tideUsefulBytes.split(char(0x00));
return QString::fromUtf8(tideUsefulBytesList.at(0)).trimmed();
}
void SnapmaticPicture::parseSnapmaticExportAndSortString()
{
QStringList pictureStrList = pictureStr.split(" - ");
if (pictureStrList.length() <= 2)
{
QString dtStr = pictureStrList.at(1);
QStringList dtStrList = dtStr.split(" ");
if (dtStrList.length() <= 2)
{
QString dayStr;
QString yearStr;
QString monthStr;
QString dateStr = dtStrList.at(0);
QString timeStr = dtStrList.at(1);
timeStr.replace(":","");
QStringList dateStrList = dateStr.split("/");
if (dateStrList.length() <= 3)
{
dayStr = dateStrList.at(1);
yearStr = dateStrList.at(2);
monthStr = dateStrList.at(0);
}
QString cmpPicTitl = titlStr;
2016-04-20 12:14:19 +02:00
cmpPicTitl.replace("\"", "''");
cmpPicTitl.replace(" ", "_");
2016-04-20 12:14:19 +02:00
cmpPicTitl.replace(":", "-");
cmpPicTitl.replace("\\", "");
cmpPicTitl.replace("/", "");
cmpPicTitl.replace("<", "");
cmpPicTitl.replace(">", "");
cmpPicTitl.replace("*", "");
cmpPicTitl.replace("?", "");
cmpPicTitl.replace(".", "");
sortStr = yearStr + monthStr + dayStr + timeStr;
picExportFileName = sortStr + "_" + cmpPicTitl + ".jpg";
}
}
}
bool SnapmaticPicture::readingPictureFromFile(const QString &fileName, bool writeEnabled_, bool cacheEnabled_)
{
if (fileName != "")
{
picFileName = fileName;
return readingPicture(writeEnabled_, cacheEnabled_);
}
else
{
return false;
}
}
bool SnapmaticPicture::setPicture(const QImage &picture)
{
if (writeEnabled)
{
QByteArray picByteArray;
QBuffer snapmaticStream(&rawPicContent);
snapmaticStream.open(QIODevice::ReadWrite);
if (snapmaticStream.seek(jpegStreamEditorBegin))
{
bool saveSuccess;
2016-12-14 00:43:31 +01:00
Q_UNUSED(saveSuccess)
QByteArray picByteArray1;
QBuffer picStream1(&picByteArray1);
picStream1.open(QIODevice::WriteOnly);
2016-12-10 05:56:09 +01:00
saveSuccess = picture.save(&picStream1, "JPEG", 95);
picStream1.close();
if (picByteArray1.length() > jpegPicStreamLength)
{
QByteArray picByteArray2;
QBuffer picStream2(&picByteArray2);
picStream2.open(QIODevice::WriteOnly);
2016-12-10 05:56:09 +01:00
saveSuccess = picture.save(&picStream2, "JPEG", 80);
picStream2.close();
if (picByteArray2.length() > jpegPicStreamLength)
{
snapmaticStream.close();
return false;
}
picByteArray = picByteArray2;
}
else
{
picByteArray = picByteArray1;
}
2016-12-10 05:56:09 +01:00
}
while (picByteArray.length() != jpegPicStreamLength)
{
picByteArray.append((char)0x00);
}
int result = snapmaticStream.write(picByteArray);
if (result != 0)
{
if (cacheEnabled)
{
cachePicture = picture;
}
return true;
}
return false;
}
return false;
}
bool SnapmaticPicture::exportPicture(const QString &fileName)
{
QFile *picFile = new QFile(fileName);
if (picFile->open(QIODevice::WriteOnly))
{
picFile->write(rawPicContent);
picFile->close();
picFile->deleteLater();
return true;
}
else
{
return false;
}
}
QString SnapmaticPicture::getExportPictureFileName()
{
return picExportFileName;
}
2016-04-03 10:17:01 +02:00
QString SnapmaticPicture::getPictureFileName()
{
return picFileName;
}
QString SnapmaticPicture::getPictureSortStr()
{
return sortStr;
}
QString SnapmaticPicture::getPictureDesc()
{
return descStr;
}
QString SnapmaticPicture::getPictureTitl()
{
return titlStr;
}
QString SnapmaticPicture::getPictureStr()
{
return pictureStr;
}
QString SnapmaticPicture::getLastStep()
{
return lastStep;
}
QImage SnapmaticPicture::getPicture()
{
if (cacheEnabled)
{
return cachePicture;
}
else if (writeEnabled)
{
2016-12-14 00:43:31 +01:00
bool returnOk = 0;
QImage returnPicture;
QBuffer snapmaticStream(&rawPicContent);
snapmaticStream.open(QIODevice::ReadOnly);
if (snapmaticStream.seek(jpegStreamEditorBegin))
{
QByteArray jpegRawContent = snapmaticStream.read(jpegPicStreamLength);
returnOk = returnPicture.loadFromData(jpegRawContent, "JPEG");
}
snapmaticStream.close();
if (returnOk)
{
return returnPicture;
}
}
else
{
2016-12-14 00:43:31 +01:00
bool returnOk = 0;
QImage returnPicture;
QIODevice *picStream;
QFile *picFile = new QFile(picFileName);
if (!picFile->open(QFile::ReadOnly))
{
lastStep = "1;/1,OpenFile," + StringParser::convertDrawStringForLog(picFileName);
picFile->deleteLater();
delete picFile;
return QImage(0, 0, QImage::Format_RGB32);
}
rawPicContent = picFile->read(snapmaticFileMaxSize);
picFile->close();
delete picFile;
picStream = new QBuffer(&rawPicContent);
picStream->open(QIODevice::ReadWrite);
if (picStream->seek(jpegStreamEditorBegin))
{
QByteArray jpegRawContent = picStream->read(jpegPicStreamLength);
returnOk = returnPicture.loadFromData(jpegRawContent, "JPEG");
}
picStream->close();
delete picStream;
if (returnOk)
{
return returnPicture;
}
}
return QImage(0, 0, QImage::Format_RGB32);
}
2016-03-21 22:29:57 +01:00
bool SnapmaticPicture::isPicOk()
{
return picOk;
}
2016-04-13 00:32:14 +02:00
void SnapmaticPicture::setPicFileName(QString picFileName_)
{
picFileName = picFileName_;
}
void SnapmaticPicture::clearCache()
{
cacheEnabled = false;
cachePicture = QImage(0, 0, QImage::Format_RGB32);
}
// JSON part
void SnapmaticPicture::parseJsonContent()
{
2016-12-10 05:56:09 +01:00
QJsonDocument jsonDocument = QJsonDocument::fromJson(jsonStr.toUtf8());
QJsonObject jsonObject = jsonDocument.object();
2016-12-14 00:43:31 +01:00
QVariantMap jsonMap = jsonObject.toVariantMap(); // backward compatibility
2016-12-10 04:35:24 +01:00
if (jsonObject.contains("loc"))
{
QJsonObject locObject = jsonObject["loc"].toObject();
2016-12-10 04:35:24 +01:00
if (locObject.contains("x")) { localSpJson.location.x = locObject["x"].toDouble(); }
if (locObject.contains("y")) { localSpJson.location.y = locObject["y"].toDouble(); }
if (locObject.contains("z")) { localSpJson.location.z = locObject["z"].toDouble(); }
}
2016-12-10 04:35:24 +01:00
if (jsonObject.contains("area"))
{
2016-12-10 04:35:24 +01:00
localSpJson.area = jsonObject["area"].toString();
}
2016-12-10 04:35:24 +01:00
if (jsonObject.contains("crewid"))
2016-03-21 21:46:36 +01:00
{
2016-12-10 04:35:24 +01:00
localSpJson.crewID = jsonObject["crewid"].toInt();
2016-03-21 21:46:36 +01:00
}
2016-12-10 04:35:24 +01:00
if (jsonObject.contains("creat"))
2016-05-08 19:07:55 +02:00
{
QDateTime createdTimestamp;
2016-12-14 00:43:31 +01:00
localSpJson.createdTimestamp = jsonMap["creat"].toUInt();
2016-12-10 04:35:24 +01:00
createdTimestamp.setTime_t(localSpJson.createdTimestamp);
localSpJson.createdDateTime = createdTimestamp;
2016-05-08 19:07:55 +02:00
}
2016-12-10 04:35:24 +01:00
if (jsonObject.contains("plyrs"))
{
2016-12-14 00:43:31 +01:00
localSpJson.playersList = jsonMap["plyrs"].toStringList();
}
2016-12-10 04:35:24 +01:00
if (jsonObject.contains("meme"))
{
2016-12-10 04:35:24 +01:00
localSpJson.isMeme = jsonObject["meme"].toBool();
}
2016-12-10 04:35:24 +01:00
if (jsonObject.contains("mug"))
{
2016-12-10 04:35:24 +01:00
localSpJson.isMug = jsonObject["mug"].toBool();
}
2016-12-10 04:35:24 +01:00
if (jsonObject.contains("slf"))
{
2016-12-10 04:35:24 +01:00
localSpJson.isSelfie = jsonObject["slf"].toBool();
}
2016-12-10 04:35:24 +01:00
if (jsonObject.contains("drctr"))
{
2016-12-10 04:35:24 +01:00
localSpJson.isFromDirector = jsonObject["drctr"].toBool();
}
2016-12-10 04:35:24 +01:00
if (jsonObject.contains("rsedtr"))
{
2016-12-10 04:35:24 +01:00
localSpJson.isFromRSEditor = jsonObject["rsedtr"].toBool();
}
2016-03-21 22:29:57 +01:00
jsonOk = true;
}
bool SnapmaticPicture::isJsonOk()
{
return jsonOk;
}
QString SnapmaticPicture::getJsonStr()
{
return jsonStr;
}
2016-12-10 04:35:24 +01:00
SnapmaticProperties SnapmaticPicture::getSnapmaticProperties()
2016-05-08 19:07:55 +02:00
{
2016-12-10 04:35:24 +01:00
return localSpJson;
2016-05-08 19:07:55 +02:00
}
2016-12-10 05:56:09 +01:00
bool SnapmaticPicture::setSnapmaticProperties(SnapmaticProperties newSpJson)
{
QJsonDocument jsonDocument = QJsonDocument::fromJson(jsonStr.toUtf8());
QJsonObject jsonObject = jsonDocument.object();
QJsonObject locObject;
locObject["x"] = newSpJson.location.x;
locObject["y"] = newSpJson.location.y;
locObject["z"] = newSpJson.location.z;
jsonObject["loc"] = locObject;
jsonObject["area"] = newSpJson.area;
jsonObject["crewid"] = newSpJson.crewID;
jsonObject["creat"] = QJsonValue::fromVariant(newSpJson.createdTimestamp);
jsonObject["plyrs"] = QJsonValue::fromVariant(newSpJson.playersList);
jsonObject["meme"] = newSpJson.isMeme;
jsonObject["mug"] = newSpJson.isMug;
jsonObject["slf"] = newSpJson.isSelfie;
jsonObject["drctr"] = newSpJson.isFromDirector;
jsonObject["rsedtr"] = newSpJson.isFromRSEditor;
jsonDocument.setObject(jsonObject);
QString newJsonStr = QString::fromUtf8(jsonDocument.toJson(QJsonDocument::Compact));
if (newJsonStr.length() < jsonStreamEditorLength)
{
if (writeEnabled)
{
QByteArray jsonByteArray = newJsonStr.toUtf8();
while (jsonByteArray.length() != jsonStreamEditorLength)
{
jsonByteArray.append((char)0x00);
}
QBuffer snapmaticStream(&rawPicContent);
snapmaticStream.open(QIODevice::ReadWrite);
if (!snapmaticStream.seek(jsonStreamEditorBegin))
{
snapmaticStream.close();
return false;
}
int result = snapmaticStream.write(jsonByteArray);
snapmaticStream.close();
if (result != 0)
{
localSpJson = newSpJson;
2016-12-10 05:56:09 +01:00
jsonStr = newJsonStr;
return true;
}
}
else
{
return false;
}
}
else
{
return false;
}
return true;
}
// VISIBILITY
bool SnapmaticPicture::isHidden()
{
if (picFileName.right(7) == ".hidden")
{
return true;
}
return false;
}
bool SnapmaticPicture::setPictureHidden()
{
if (!isHidden())
{
QString newPicFileName = QString(picFileName + ".hidden");
if (QFile::rename(picFileName, newPicFileName))
{
picFileName = newPicFileName;
return true;
}
return false;
}
return true;
}
bool SnapmaticPicture::setPictureVisible()
{
if (isHidden())
{
QString newPicFileName = QString(picFileName).remove(picFileName.length() - 7, 7);
if (QFile::rename(picFileName, newPicFileName))
{
picFileName = newPicFileName;
return true;
}
return false;
}
return true;
}