mirror of https://gitlab.com/Syping/gta5view-cmd
491 lines
23 KiB
C++
491 lines
23 KiB
C++
/*****************************************************************************
|
|
* gta5view-cmd - gta5view Command Line
|
|
* Copyright (C) 2017-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/>.
|
|
*****************************************************************************/
|
|
|
|
#include "SnapmaticPicture.h"
|
|
#include <QCoreApplication>
|
|
#include <QImageReader>
|
|
#include <QDateTime>
|
|
#include <QPainter>
|
|
#include <QImage>
|
|
#include <QMap>
|
|
|
|
#include <iostream>
|
|
using namespace std;
|
|
|
|
int main(int argc, char *argv[])
|
|
{
|
|
QCoreApplication a(argc, argv);
|
|
QStringList args = a.arguments();
|
|
|
|
if (args.length() >= 3) {
|
|
QString format = "jpg";
|
|
bool isDefault = true;
|
|
bool avatarMode = false;
|
|
bool eXtendMode = false;
|
|
bool ignoreORes = false;
|
|
bool convertToGTA = false;
|
|
bool customFormat = false;
|
|
bool formatSwitch = false;
|
|
bool keepAspectRatio = true;
|
|
QMap<QString,QString> flags;
|
|
|
|
if (args.length() >= 4) {
|
|
if (args.at(3) == "-pgta") {
|
|
convertToGTA = true;
|
|
customFormat = false;
|
|
}
|
|
else if (args.at(3) == "-g5e") {
|
|
convertToGTA = true;
|
|
customFormat = true;
|
|
}
|
|
else if (args.at(3) == "-jpg") {
|
|
format = "jpg";
|
|
convertToGTA = false;
|
|
}
|
|
}
|
|
if (args.length() >= 5) {
|
|
if (args.at(4) == "-a") {
|
|
isDefault = false;
|
|
avatarMode = true;
|
|
}
|
|
else if (args.at(4) == "-aX") {
|
|
isDefault = false;
|
|
avatarMode = true;
|
|
eXtendMode = true;
|
|
}
|
|
else if (args.at(4) == "-p") {
|
|
isDefault = false;
|
|
avatarMode = false;
|
|
}
|
|
else if (args.at(4) == "-pX") {
|
|
isDefault = false;
|
|
avatarMode = false;
|
|
eXtendMode = true;
|
|
}
|
|
else if (args.at(4) == "-d") {
|
|
isDefault = true;
|
|
}
|
|
else if (args.at(4) == "-dX") {
|
|
isDefault = true;
|
|
eXtendMode = true;
|
|
}
|
|
else if (args.at(4) == "-fs") {
|
|
convertToGTA = false;
|
|
formatSwitch = true;
|
|
}
|
|
else if (args.at(4) == "-aiar") {
|
|
isDefault = false;
|
|
avatarMode = true;
|
|
keepAspectRatio = false;
|
|
}
|
|
else if (args.at(4) == "-aiarX") {
|
|
isDefault = false;
|
|
avatarMode = true;
|
|
eXtendMode = true;
|
|
keepAspectRatio = false;
|
|
}
|
|
else if (args.at(4) == "-akar") {
|
|
isDefault = false;
|
|
avatarMode = true;
|
|
keepAspectRatio = true;
|
|
}
|
|
else if (args.at(4) == "-akarX") {
|
|
isDefault = false;
|
|
avatarMode = true;
|
|
eXtendMode = true;
|
|
keepAspectRatio = true;
|
|
}
|
|
else if (args.at(4) == "-piar") {
|
|
isDefault = false;
|
|
avatarMode = false;
|
|
keepAspectRatio = false;
|
|
}
|
|
else if (args.at(4) == "-piarX") {
|
|
isDefault = false;
|
|
avatarMode = false;
|
|
eXtendMode = true;
|
|
keepAspectRatio = false;
|
|
}
|
|
else if (args.at(4) == "-pkar") {
|
|
isDefault = false;
|
|
avatarMode = false;
|
|
keepAspectRatio = true;
|
|
}
|
|
else if (args.at(4) == "-pkarX") {
|
|
isDefault = false;
|
|
avatarMode = false;
|
|
eXtendMode = true;
|
|
keepAspectRatio = true;
|
|
}
|
|
else if (args.at(4) == "-diar") {
|
|
isDefault = true;
|
|
keepAspectRatio = false;
|
|
}
|
|
else if (args.at(4) == "-diarX") {
|
|
isDefault = true;
|
|
eXtendMode = true;
|
|
keepAspectRatio = false;
|
|
}
|
|
else if (args.at(4) == "-dkar") {
|
|
isDefault = true;
|
|
keepAspectRatio = true;
|
|
}
|
|
else if (args.at(4) == "-dkarX") {
|
|
isDefault = true;
|
|
eXtendMode = true;
|
|
keepAspectRatio = true;
|
|
}
|
|
else if (args.at(4) == "-ior") {
|
|
isDefault = false;
|
|
ignoreORes = true;
|
|
keepAspectRatio = false;
|
|
}
|
|
else if (args.at(4) == "-iorX") {
|
|
isDefault = false;
|
|
ignoreORes = true;
|
|
eXtendMode = true;
|
|
keepAspectRatio = false;
|
|
}
|
|
else {
|
|
QStringList currentFlagArg = args.at(4).split("=");
|
|
QString currentFlag = currentFlagArg.at(0).toLower();
|
|
if (currentFlag.at(0) == '-') {
|
|
currentFlag.remove(0, 1);
|
|
currentFlagArg.removeFirst();
|
|
const QString currentValue = currentFlagArg.join(QString());
|
|
flags[currentFlag] = currentValue;
|
|
}
|
|
}
|
|
}
|
|
if (args.length() >= 6) {
|
|
QStringList flagArgs(args);
|
|
for (int argId = 0; argId == 5; argId++) {
|
|
flagArgs.removeAt(argId);
|
|
}
|
|
for (const QString &flagArg : flagArgs) {
|
|
QStringList currentFlagArg = flagArg.split("=");
|
|
QString currentFlag = currentFlagArg.at(0).toLower();
|
|
if (currentFlag.at(0) == '-') {
|
|
currentFlag.remove(0, 1);
|
|
currentFlagArg.removeFirst();
|
|
const QString currentValue = currentFlagArg.join(QString());
|
|
flags[currentFlag] = currentValue;
|
|
}
|
|
}
|
|
}
|
|
if (!convertToGTA) {
|
|
SnapmaticPicture picture(args.at(1));
|
|
if (picture.readingPicture(false)) {
|
|
QString filePath = args.at(2);
|
|
if (!formatSwitch) {
|
|
filePath.replace("<autodef>", picture.getExportPictureFileName());
|
|
filePath.replace("<autoext>", ".jpg");
|
|
if (!picture.exportPicture(filePath, SnapmaticFormat::JPEG_Format)) {
|
|
cout << "gta5view-cmd: Exporting of " << args.at(1).toStdString() << " to " << filePath.toStdString() << " is failed!" << endl;
|
|
return 1;
|
|
}
|
|
else {
|
|
if (flags.value("output", QString()).toLower() == "exported_filename") {
|
|
cout << filePath.toStdString() << endl;
|
|
}
|
|
else {
|
|
cout << "gta5view-cmd: Exporting of " << args.at(1).toStdString() << " to " << filePath.toStdString() << " is successful!" << endl;
|
|
}
|
|
}
|
|
}
|
|
else {
|
|
filePath.replace("<autodef>", picture.getPictureFileName());
|
|
if (customFormat) {
|
|
filePath.replace("<autoext>", ".g5e");
|
|
if (!picture.exportPicture(filePath, SnapmaticFormat::G5E_Format)) {
|
|
cout << "gta5view-cmd: Converting of " << args.at(1).toStdString() << " to " << filePath.toStdString() << " is failed!" << endl;
|
|
return 1;
|
|
}
|
|
else {
|
|
if (flags.value("output", QString()).toLower() == "exported_filename") {
|
|
cout << filePath.toStdString() << endl;
|
|
}
|
|
else {
|
|
cout << "gta5view-cmd: Converting of " << args.at(1).toStdString() << " to " << filePath.toStdString() << " is successful!" << endl;
|
|
}
|
|
}
|
|
}
|
|
else {
|
|
filePath.replace("<autoext>", "");
|
|
if (!picture.exportPicture(filePath, SnapmaticFormat::PGTA_Format)) {
|
|
cout << "gta5view-cmd: Converting of " << args.at(1).toStdString() << " to " << filePath.toStdString() << " is failed!" << endl;
|
|
return 1;
|
|
}
|
|
else {
|
|
if (flags.value("output", QString()).toLower() == "exported_filename") {
|
|
cout << filePath.toStdString() << endl;
|
|
}
|
|
else {
|
|
cout << "gta5view-cmd: Converting of " << args.at(1).toStdString() << " to " << filePath.toStdString() << " is successful!" << endl;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
else {
|
|
cout << "gta5view-cmd: Reading of " << args.at(1).toStdString() << " is failed!" << endl;
|
|
return 1;
|
|
}
|
|
}
|
|
else {
|
|
SnapmaticPicture picture(":/template.g5e");
|
|
if (picture.readingPicture(false)) {
|
|
QImage image;
|
|
QFile imageFile(args.at(1));
|
|
if (!imageFile.open(QFile::ReadOnly)) {
|
|
cout << "gta5view-cmd: Reading of " << args.at(1).toStdString() << " is failed!" << endl;
|
|
return 1;
|
|
}
|
|
QImageReader imageReader;
|
|
imageReader.setDecideFormatFromContent(true);
|
|
imageReader.setDevice(&imageFile);
|
|
if (!imageReader.read(&image)) {
|
|
cout << "gta5view-cmd: Parsing of " << args.at(1).toStdString() << " is failed!" << endl;
|
|
return 1;
|
|
}
|
|
if (!image.isNull()) {
|
|
if (!ignoreORes) {
|
|
QSize snapmaticRes(960, 536);
|
|
QSize avatarRes(470, 470);
|
|
int avatarZoneX = 145;
|
|
int avatarZoneY = 66;
|
|
|
|
if (isDefault) {
|
|
if (image.width() == image.height()) {
|
|
avatarMode = true;
|
|
}
|
|
else {
|
|
avatarMode = false;
|
|
}
|
|
}
|
|
if (!avatarMode) {
|
|
QImage snapmaticImage(snapmaticRes, QImage::Format_RGB888);
|
|
snapmaticImage.fill(Qt::black);
|
|
QPainter snapmaticPainter(&snapmaticImage);
|
|
|
|
// Picture mode
|
|
int diffWidth = 0;
|
|
int diffHeight = 0;
|
|
if (keepAspectRatio) {
|
|
image = image.scaled(snapmaticRes, Qt::KeepAspectRatio, Qt::SmoothTransformation);
|
|
if (image.width() != snapmaticRes.width()) {
|
|
diffWidth = snapmaticRes.width() - image.width();
|
|
diffWidth = diffWidth / 2;
|
|
}
|
|
else if (image.height() != snapmaticRes.height()) {
|
|
diffHeight = snapmaticRes.height() - image.height();
|
|
diffHeight = diffHeight / 2;
|
|
}
|
|
}
|
|
else {
|
|
image = image.scaled(snapmaticRes.width(), snapmaticRes.height(), Qt::IgnoreAspectRatio, Qt::SmoothTransformation);
|
|
}
|
|
snapmaticPainter.drawImage(0 + diffWidth, 0 + diffHeight, image);
|
|
snapmaticPainter.end();
|
|
image = snapmaticImage;
|
|
}
|
|
else {
|
|
QImage snapmaticImage(snapmaticRes, QImage::Format_RGB888);
|
|
snapmaticImage.fill(Qt::black);
|
|
QPainter snapmaticPainter(&snapmaticImage);
|
|
|
|
// Scale to Avatar Resolution if needed
|
|
if (image.width() != avatarRes.width()) {
|
|
image.scaled(avatarRes, Qt::IgnoreAspectRatio, Qt::SmoothTransformation);
|
|
}
|
|
|
|
// Avatar mode
|
|
int diffWidth = 0;
|
|
int diffHeight = 0;
|
|
if (keepAspectRatio) {
|
|
image = image.scaled(avatarRes, Qt::KeepAspectRatio, Qt::SmoothTransformation);
|
|
if (image.width() > image.height()) {
|
|
diffHeight = avatarRes.height() - image.height();
|
|
diffHeight = diffHeight / 2;
|
|
}
|
|
else if (image.width() < image.height()) {
|
|
diffWidth = avatarRes.width() - image.width();
|
|
diffWidth = diffWidth / 2;
|
|
}
|
|
}
|
|
else {
|
|
image = image.scaled(avatarRes, Qt::IgnoreAspectRatio, Qt::SmoothTransformation);
|
|
}
|
|
snapmaticPainter.drawImage(avatarZoneX + diffWidth, avatarZoneY + diffHeight, image);
|
|
snapmaticPainter.end();
|
|
image = snapmaticImage;
|
|
}
|
|
}
|
|
|
|
if (picture.setImage(image, eXtendMode)) {
|
|
int crew = 0;
|
|
double posX = 0;
|
|
double posY = 0;
|
|
double posZ = 0;
|
|
QStringList players;
|
|
if (flags.contains("crew")) {
|
|
bool crewValid;
|
|
int crewId = flags["crew"].toInt(&crewValid);
|
|
if (crewValid) {
|
|
crew = crewId;
|
|
}
|
|
}
|
|
if (flags.contains("title") && SnapmaticPicture::verifyTitle(flags["title"])) {
|
|
picture.setPictureTitle(flags["title"]);
|
|
}
|
|
else {
|
|
avatarMode ? picture.setPictureTitle("Converted Avatar") : picture.setPictureTitle("Converted Picture");
|
|
}
|
|
if (flags.contains("players")) {
|
|
bool flagValid = true;
|
|
QString playerFlag = flags["players"];
|
|
if (playerFlag.left(1) == "[" && playerFlag.right(1) == "]") {
|
|
playerFlag.remove(0, 1);
|
|
playerFlag.remove(playerFlag.length() - 1, 1);
|
|
}
|
|
const QStringList playersList = playerFlag.split(",");
|
|
for (QString player : playersList) {
|
|
if (player.left(1) == "\"" && player.right(1) == "\"") {
|
|
player.remove(0, 1);
|
|
player.remove(player.length() - 1, 1);
|
|
}
|
|
bool playerValid;
|
|
player.toInt(&playerValid);
|
|
if (!playerValid) {
|
|
flagValid = false;
|
|
}
|
|
}
|
|
if (flagValid) {
|
|
players = playersList;
|
|
}
|
|
}
|
|
if (flags.contains("position")) {
|
|
bool flagValid = true;
|
|
QString positionFlag = flags["position"];
|
|
if (positionFlag.left(1) == "[" && positionFlag.right(1) == "]") {
|
|
positionFlag.remove(0, 1);
|
|
positionFlag.remove(positionFlag.length() - 1, 1);
|
|
}
|
|
const QStringList positionsList = positionFlag.split(",");
|
|
for (QString position : positionsList) {
|
|
if (position.left(1) == "\"" && position.right(1) == "\"") {
|
|
position.remove(0, 1);
|
|
position.remove(position.length() - 1, 1);
|
|
}
|
|
bool positionValid;
|
|
position.toDouble(&positionValid);
|
|
if (!positionValid) {
|
|
flagValid = false;
|
|
}
|
|
}
|
|
if (flagValid) {
|
|
if (positionsList.length() >= 1) {
|
|
posX = positionsList.at(0).toDouble();
|
|
}
|
|
if (positionsList.length() >= 2) {
|
|
posY = positionsList.at(1).toDouble();
|
|
}
|
|
if (positionsList.length() >= 3) {
|
|
posZ = positionsList.at(2).toDouble();
|
|
}
|
|
}
|
|
}
|
|
SnapmaticProperties pictureSP = picture.getSnapmaticProperties();
|
|
pictureSP.uid = QString(QTime::currentTime().toString("HHmmss") +
|
|
QString::number(QDate::currentDate().dayOfYear())).toInt();
|
|
pictureSP.createdDateTime = QDateTime::currentDateTime();
|
|
pictureSP.createdTimestamp = pictureSP.createdDateTime.toTime_t();
|
|
pictureSP.crewID = crew;
|
|
pictureSP.playersList = players;
|
|
pictureSP.location.x = posX;
|
|
pictureSP.location.y = posY;
|
|
pictureSP.location.z = posZ;
|
|
picture.setSnapmaticProperties(pictureSP);
|
|
picture.setPictureFileName(QString("PGTA5%1").arg(QString::number(pictureSP.uid)));
|
|
QString filePath = args.at(2);
|
|
filePath.replace("<autodef>", picture.getPictureFileName());
|
|
if (!customFormat) {
|
|
filePath.replace("<autoext>", "");
|
|
if (!picture.exportPicture(filePath, SnapmaticFormat::PGTA_Format)) {
|
|
cout << "gta5view-cmd: Converting of " << args.at(1).toStdString() << " to " << filePath.toStdString() << " is failed!" << endl;
|
|
return 1;
|
|
}
|
|
else {
|
|
if (flags.value("output", QString()).toLower() == "exported_filename") {
|
|
cout << filePath.toStdString() << endl;
|
|
}
|
|
else {
|
|
cout << "gta5view-cmd: Converting of " << args.at(1).toStdString() << " to " << filePath.toStdString() << " is successful!" << endl;
|
|
}
|
|
}
|
|
}
|
|
else {
|
|
filePath.replace("<autoext>", ".g5e");
|
|
if (!picture.exportPicture(filePath, SnapmaticFormat::G5E_Format)) {
|
|
cout << "gta5view-cmd: Converting of " << args.at(1).toStdString() << " to " << filePath.toStdString() << " is failed!" << endl;
|
|
return 1;
|
|
}
|
|
else {
|
|
if (flags.value("output", QString()).toLower() == "exported_filename") {
|
|
cout << filePath.toStdString() << endl;
|
|
}
|
|
else {
|
|
cout << "gta5view-cmd: Converting of " << args.at(1).toStdString() << " to " << filePath.toStdString() << " is successful!" << endl;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
else {
|
|
cout << "gta5view-cmd: Editing of Snapmatic Image Stream failed!" << endl;
|
|
return 1;
|
|
}
|
|
}
|
|
else {
|
|
cout << "gta5view-cmd: " << args.at(1).toStdString() << " is invalid image!" << endl;
|
|
return 1;
|
|
}
|
|
}
|
|
else {
|
|
cout << "gta5view-cmd: Reading of internal template failed!" << endl;
|
|
return 1;
|
|
}
|
|
}
|
|
}
|
|
else {
|
|
cout << "gta5view Command Line 0.5.1" << endl << endl;
|
|
cout << "Usage: " << args.at(0).toStdString() << " source target -format -mode -flag=value" << endl;
|
|
if (args.length() >= 2 && (args.at(1) == "--help" || args.at(1) == "-help")) {
|
|
cout << "Convert-only: <autodef> (auto file name at convert)" << endl;
|
|
cout << "Convert-only: <autoext> (auto file extension at convert)" << endl;
|
|
cout << "Formats: jpg pgta g5e" << endl;
|
|
cout << "Snapmatic Modes: a p d fs aiar akar piar pkar diar dkar ior" << endl;
|
|
cout << "Snapmatic Modes: aX pX dX aiarX akarX piarX pkarX diarX dkarX iorX" << endl;
|
|
cout << "Snapmatic Flags: crew=int players=int[] position=double[] title=string" << endl;
|
|
cout << "Global Flags: output=string" << endl;
|
|
}
|
|
return 255;
|
|
}
|
|
|
|
return 0;
|
|
}
|