255 lines
9.3 KiB
C++
255 lines
9.3 KiB
C++
#include "main.h"
|
|
#include "retain_vars.hpp"
|
|
#include "utils/logger.h"
|
|
#include <coreinit/cache.h>
|
|
#include <coreinit/title.h>
|
|
#include <malloc.h>
|
|
#include <nn/acp.h>
|
|
#include <string>
|
|
#include <vpad/input.h>
|
|
#include <wups.h>
|
|
#include <wups/config/WUPSConfigItemBoolean.h>
|
|
#include <wups/config/WUPSConfigItemIntegerRange.h>
|
|
#include <wups/config/WUPSConfigItemMultipleValues.h>
|
|
|
|
// Mandatory plugin information.
|
|
WUPS_PLUGIN_NAME("Screenshot plugin");
|
|
WUPS_PLUGIN_DESCRIPTION("This plugin allows you to make screenshots that will be saved to the sd card");
|
|
WUPS_PLUGIN_VERSION(VERSION_FULL);
|
|
WUPS_PLUGIN_AUTHOR("Maschell");
|
|
WUPS_PLUGIN_LICENSE("GPL");
|
|
|
|
// FS Access
|
|
WUPS_USE_WUT_DEVOPTAB();
|
|
|
|
WUPS_USE_STORAGE("screenshot_plugin");
|
|
|
|
#define ENABLED_CONFIG_STRING "enabled"
|
|
#define FORMAT_CONFIG_STRING "format"
|
|
#define QUALITY_CONFIG_STRING "quality"
|
|
#define SCREEN_CONFIG_STRING "screen"
|
|
|
|
// Gets called once the loader exists.
|
|
INITIALIZE_PLUGIN() {
|
|
initLogging();
|
|
gButtonCombo = VPAD_BUTTON_R | VPAD_BUTTON_L | VPAD_BUTTON_ZR | VPAD_BUTTON_ZL;
|
|
OSMemoryBarrier();
|
|
|
|
// Open storage to read values
|
|
WUPSStorageError storageRes = WUPS_OpenStorage();
|
|
if (storageRes != WUPS_STORAGE_ERROR_SUCCESS) {
|
|
DEBUG_FUNCTION_LINE_ERR("Failed to open storage %s (%d)", WUPS_GetStorageStatusStr(storageRes), storageRes);
|
|
} else {
|
|
// Try to get value from storage
|
|
if ((storageRes = WUPS_GetInt(nullptr, ENABLED_CONFIG_STRING, reinterpret_cast<int32_t *>(&gEnabled))) == WUPS_STORAGE_ERROR_NOT_FOUND) {
|
|
// Add the value to the storage if it's missing.
|
|
if (WUPS_StoreBool(nullptr, ENABLED_CONFIG_STRING, gEnabled) != WUPS_STORAGE_ERROR_SUCCESS) {
|
|
DEBUG_FUNCTION_LINE_ERR("Failed to store value");
|
|
}
|
|
} else if (storageRes != WUPS_STORAGE_ERROR_SUCCESS) {
|
|
DEBUG_FUNCTION_LINE_ERR("Failed to get value %s (%d)", WUPS_GetStorageStatusStr(storageRes), storageRes);
|
|
}
|
|
// Try to get value from storage
|
|
if ((storageRes = WUPS_GetInt(nullptr, FORMAT_CONFIG_STRING, reinterpret_cast<int32_t *>(&gOutputFormat))) == WUPS_STORAGE_ERROR_NOT_FOUND) {
|
|
// Add the value to the storage if it's missing.
|
|
if (WUPS_StoreBool(nullptr, FORMAT_CONFIG_STRING, gOutputFormat) != WUPS_STORAGE_ERROR_SUCCESS) {
|
|
DEBUG_FUNCTION_LINE_ERR("Failed to store value");
|
|
}
|
|
} else if (storageRes != WUPS_STORAGE_ERROR_SUCCESS) {
|
|
DEBUG_FUNCTION_LINE_ERR("Failed to get value %s (%d)", WUPS_GetStorageStatusStr(storageRes), storageRes);
|
|
}
|
|
|
|
// Try to get value from storage
|
|
if ((storageRes = WUPS_GetInt(nullptr, QUALITY_CONFIG_STRING, reinterpret_cast<int32_t *>(&gQuality))) == WUPS_STORAGE_ERROR_NOT_FOUND) {
|
|
// Add the value to the storage if it's missing.
|
|
if (WUPS_StoreInt(nullptr, QUALITY_CONFIG_STRING, (int32_t) gQuality) != WUPS_STORAGE_ERROR_SUCCESS) {
|
|
DEBUG_FUNCTION_LINE_ERR("Failed to store value");
|
|
}
|
|
} else if (storageRes != WUPS_STORAGE_ERROR_SUCCESS) {
|
|
DEBUG_FUNCTION_LINE_ERR("Failed to get value %s (%d)", WUPS_GetStorageStatusStr(storageRes), storageRes);
|
|
}
|
|
|
|
// Try to get value from storage
|
|
if ((storageRes = WUPS_GetInt(nullptr, SCREEN_CONFIG_STRING, reinterpret_cast<int32_t *>(&gImageSource))) == WUPS_STORAGE_ERROR_NOT_FOUND) {
|
|
// Add the value to the storage if it's missing.
|
|
if (WUPS_StoreInt(nullptr, SCREEN_CONFIG_STRING, (int32_t) gImageSource) != WUPS_STORAGE_ERROR_SUCCESS) {
|
|
DEBUG_FUNCTION_LINE_ERR("Failed to store value");
|
|
}
|
|
} else if (storageRes != WUPS_STORAGE_ERROR_SUCCESS) {
|
|
DEBUG_FUNCTION_LINE_ERR("Failed to get value %s (%d)", WUPS_GetStorageStatusStr(storageRes), storageRes);
|
|
}
|
|
|
|
// Close storage
|
|
if (WUPS_CloseStorage() != WUPS_STORAGE_ERROR_SUCCESS) {
|
|
DEBUG_FUNCTION_LINE_ERR("Failed to close storage");
|
|
}
|
|
}
|
|
if (gOutputFormat >= 3) {
|
|
gOutputFormat = IMAGE_OUTPUT_FORMAT_JPEG;
|
|
}
|
|
if (gQuality < 10) {
|
|
gQuality = 10;
|
|
} else if (gQuality > 100) {
|
|
gQuality = 100;
|
|
}
|
|
}
|
|
|
|
void formatChanged(ConfigItemMultipleValues *item, uint32_t newValue) {
|
|
DEBUG_FUNCTION_LINE("New value in %s changed: %d", item->configID, newValue);
|
|
gOutputFormat = (ImageOutputFormatEnum) newValue;
|
|
|
|
if (gOutputFormat >= 3) {
|
|
gOutputFormat = IMAGE_OUTPUT_FORMAT_JPEG;
|
|
}
|
|
|
|
WUPS_StoreInt(nullptr, item->configID, (int32_t) newValue);
|
|
}
|
|
|
|
void imageSourceChanged(ConfigItemMultipleValues *item, uint32_t newValue) {
|
|
DEBUG_FUNCTION_LINE("New value in %s changed: %d", item->configID, newValue);
|
|
gImageSource = (ImageSourceEnum) newValue;
|
|
|
|
if (gImageSource >= 3) {
|
|
gImageSource = IMAGE_SOURCE_TV_AND_DRC;
|
|
}
|
|
|
|
WUPS_StoreInt(nullptr, item->configID, (int32_t) newValue);
|
|
}
|
|
|
|
void qualityChanged(ConfigItemIntegerRange *item, int32_t newValue) {
|
|
DEBUG_FUNCTION_LINE("New quality: %d", newValue);
|
|
gQuality = (ImageOutputFormatEnum) newValue;
|
|
|
|
if (gQuality < 10) {
|
|
gQuality = 10;
|
|
} else if (gQuality > 100) {
|
|
gQuality = 100;
|
|
}
|
|
|
|
WUPS_StoreInt(nullptr, QUALITY_CONFIG_STRING, (int32_t) gQuality);
|
|
}
|
|
|
|
void enabledChanged(ConfigItemBoolean *item, bool newValue) {
|
|
DEBUG_FUNCTION_LINE("gEnabled new value: %d", newValue);
|
|
gEnabled = (ImageOutputFormatEnum) newValue;
|
|
|
|
WUPS_StoreBool(nullptr, ENABLED_CONFIG_STRING, gEnabled);
|
|
}
|
|
|
|
WUPS_GET_CONFIG() {
|
|
// We open the storage, so we can persist the configuration the user did.
|
|
if (WUPS_OpenStorage() != WUPS_STORAGE_ERROR_SUCCESS) {
|
|
DEBUG_FUNCTION_LINE_ERR("Failed to open storage");
|
|
return 0;
|
|
}
|
|
|
|
WUPSConfigHandle config;
|
|
WUPSConfig_CreateHandled(&config, "Screenshot plugin");
|
|
|
|
WUPSConfigCategoryHandle setting;
|
|
WUPSConfig_AddCategoryByNameHandled(config, "Settings", &setting);
|
|
|
|
|
|
WUPSConfigItemBoolean_AddToCategoryHandled(config, setting, ENABLED_CONFIG_STRING, "Enabled", gEnabled, &enabledChanged);
|
|
|
|
ConfigItemMultipleValuesPair fileFormat[3];
|
|
fileFormat[0].value = IMAGE_OUTPUT_FORMAT_JPEG;
|
|
fileFormat[0].valueName = (char *) "JPEG";
|
|
|
|
fileFormat[1].value = IMAGE_OUTPUT_FORMAT_PNG;
|
|
fileFormat[1].valueName = (char *) "PNG";
|
|
|
|
fileFormat[2].value = IMAGE_OUTPUT_FORMAT_BMP;
|
|
fileFormat[2].valueName = (char *) "BMP";
|
|
|
|
uint32_t defaultIndex = 0;
|
|
uint32_t curIndex = 0;
|
|
for (auto &cur : fileFormat) {
|
|
if (cur.value == gOutputFormat) {
|
|
defaultIndex = curIndex;
|
|
break;
|
|
}
|
|
curIndex++;
|
|
}
|
|
|
|
WUPSConfigItemMultipleValues_AddToCategoryHandled(config, setting, FORMAT_CONFIG_STRING, "Output format", defaultIndex, fileFormat,
|
|
sizeof(fileFormat) / sizeof(fileFormat[0]), &formatChanged);
|
|
|
|
|
|
ConfigItemMultipleValuesPair source[3];
|
|
source[0].value = IMAGE_SOURCE_TV_AND_DRC;
|
|
source[0].valueName = (char *) "TV & Gamepad";
|
|
|
|
source[1].value = IMAGE_SOURCE_TV;
|
|
source[1].valueName = (char *) "TV only";
|
|
|
|
source[2].value = IMAGE_SOURCE_DRC;
|
|
source[2].valueName = (char *) "Gamepad only";
|
|
|
|
defaultIndex = 0;
|
|
curIndex = 0;
|
|
for (auto &cur : source) {
|
|
if (cur.value == gImageSource) {
|
|
defaultIndex = curIndex;
|
|
break;
|
|
}
|
|
curIndex++;
|
|
}
|
|
|
|
WUPSConfigItemMultipleValues_AddToCategoryHandled(config, setting, SCREEN_CONFIG_STRING, "Screen", defaultIndex, source,
|
|
sizeof(source) / sizeof(source[0]), &imageSourceChanged);
|
|
|
|
|
|
WUPSConfigItemIntegerRange_AddToCategoryHandled(config, setting, QUALITY_CONFIG_STRING, "JPEG quality", gQuality, 10, 100, &qualityChanged);
|
|
|
|
return config;
|
|
}
|
|
|
|
WUPS_CONFIG_CLOSED() {
|
|
// Save all changes
|
|
if (WUPS_CloseStorage() != WUPS_STORAGE_ERROR_SUCCESS) {
|
|
DEBUG_FUNCTION_LINE_ERR("Failed to close storage");
|
|
}
|
|
}
|
|
|
|
// Called whenever an application was started.
|
|
ON_APPLICATION_START() {
|
|
initLogging();
|
|
ACPInitialize();
|
|
auto *metaXml = (ACPMetaXml *) memalign(0x40, sizeof(ACPMetaXml));
|
|
if (ACPGetTitleMetaXml(OSGetTitleID(), metaXml) == ACP_RESULT_SUCCESS) {
|
|
gShortNameEn = metaXml->shortname_en;
|
|
std::string illegalChars = "\\/:?\"<>|@=;`_^][";
|
|
for (auto it = gShortNameEn.begin(); it < gShortNameEn.end(); ++it) {
|
|
if (*it < '0' || *it > 'z') {
|
|
*it = ' ';
|
|
}
|
|
}
|
|
for (auto it = gShortNameEn.begin(); it < gShortNameEn.end(); ++it) {
|
|
bool found = illegalChars.find(*it) != std::string::npos;
|
|
if (found) {
|
|
*it = ' ';
|
|
}
|
|
}
|
|
uint32_t length = gShortNameEn.length();
|
|
for (uint32_t i = 1; i < length; ++i) {
|
|
if (gShortNameEn[i - 1] == ' ' && gShortNameEn[i] == ' ') {
|
|
gShortNameEn.erase(i, 1);
|
|
i--;
|
|
length--;
|
|
}
|
|
}
|
|
if (gShortNameEn.size() == 1 && gShortNameEn[0] == ' ') {
|
|
gShortNameEn.clear();
|
|
} else {
|
|
DEBUG_FUNCTION_LINE("Detected name as \"%s\"", gShortNameEn.c_str());
|
|
}
|
|
} else {
|
|
gShortNameEn.clear();
|
|
}
|
|
}
|
|
|
|
ON_APPLICATION_REQUESTS_EXIT() {
|
|
deinitLogging();
|
|
}
|