gta5view/RagePhoto.h

/*****************************************************************************
* gta5view Grand Theft Auto V Profile Viewer
* Copyright (C) 2020 Syping
*
* 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, see <http://www.gnu.org/licenses/>.
*****************************************************************************/

#ifndef RAGEPHOTO_H
#define RAGEPHOTO_H

#include <QJsonObject>
#include <QIODevice>
#include <QObject>

class RagePhoto : public QObject
{
    Q_OBJECT
public:
    enum class JsonFormat : quint8 {
        Original = 0,
        Compact = 1,
        Indented = 2,
    };
    enum class ExportFormat : quint32 {
        G5E1P = 0x454C0010U,
        G5E2P = 0x01000032U,
        G5E2S = 0x02000032U,
        G5E3P = 0x01000033U,
        G5E3S = 0x02000033U,
        Undefined = 0,
    };
    enum class PhotoFormat : quint32 {
        G5EX = 0x45354700U,
        GTA5 = 0x01000000U,
        RDR2 = 0x04000000U,
        Undefined = 0,
    };
    explicit RagePhoto();
    explicit RagePhoto(const QByteArray &data);
    explicit RagePhoto(const QString &filePath);
    explicit RagePhoto(QIODevice *ioDevice);
    bool isLoaded();
    bool load();
    void clear();
    void setDescription(const QString &description);
    void setFileData(const QByteArray &data);
    void setFilePath(const QString &filePath);
    void setIODevice(QIODevice *ioDevice);
    bool setJsonData(const QByteArray &data);
    bool setPhotoBuffer(quint32 size, bool moveOffsets = true);
    bool setPhotoData(const QByteArray &data);
    bool setPhotoData(const char *data, int size);
    void setPhotoFormat(PhotoFormat photoFormat);
    void setTitle(const QString &title);
    const QJsonObject jsonObject();
    const QByteArray jsonData(JsonFormat jsonFormat = JsonFormat::Original);
    const QByteArray photoData();
    const QString description();
    const QString photoString();
    const QString title();
    quint32 photoBuffer();
    quint32 photoSize();
    PhotoFormat photoFormat();
    QByteArray save(PhotoFormat photoFormat);
    void save(QIODevice *ioDevice, PhotoFormat photoFormat);
    static RagePhoto* loadFile(const QString &filePath);

private:
    inline quint32 charToUInt32BE(char *x);
    inline quint32 charToUInt32LE(char *x);
    inline void uInt32ToCharBE(quint32 *x, char *y);
    inline void uInt32ToCharLE(quint32 *x, char *y);
    inline QByteArray stringToUtf16LE(const QString &string);
    inline QString utf16LEToString(const QByteArray &data);
    inline QString utf16LEToString(const char *data, int size);
    PhotoFormat p_photoFormat;
    QJsonObject p_jsonObject;
    QByteArray p_fileData;
    QByteArray p_jsonData;
    QByteArray p_photoData;
    QIODevice *p_ioDevice;
    QString p_descriptionString;
    QString p_filePath;
    QString p_photoString;
    QString p_titleString;
    quint32 p_descBuffer;
    quint32 p_descOffset;
    quint32 p_endOfFile;
    quint32 p_headerSum;
    quint32 p_jsonBuffer;
    quint32 p_jsonOffset;
    quint32 p_photoBuffer;
    quint32 p_titlBuffer;
    quint32 p_titlOffset;
    bool p_isLoaded;
    int p_inputMode;
};

#endif // RAGEPHOTO_H