Skip to content

Commit

Permalink
KismetLogger, PNG Screenshots, SeqActLogEnabler, StreamingLevelsHUD t…
Browse files Browse the repository at this point in the history 
…o shared repo
  • Loading branch information
@Mgamerz
Mgamerz committed Jun 1, 2022
1 parent 7a90423 commit 7f1a8d4
Show file tree
Hide file tree
Showing 9 changed files with 20 additions and 3,970 deletions.
57 changes: 4 additions & 53 deletions LE3KismetLogger/KismetLogger.cpp
View file
Original file line number Diff line number Diff line change
@@ -1,55 +1,6 @@
#include <stdio.h>
#include <io.h>
#include <string>
#include <fstream>
#include <iostream>
#include <ostream>
#include <streambuf>
#include <sstream>
#include "../LE3-SDK/Common.h"
#include "../LE3-SDK/Interface.h"
#include "../LE3-SDK/ME3TweaksHeader.h"
//#include "../LE3-SDK/SdkHeaders.h"
// Game
#define GAMELE3

#define MYHOOK "KismetLogger_"
#include "../../Shared-ASI/KismetLogger/KismetLogger.h"

SPI_PLUGINSIDE_SUPPORT(L"KismetLogger", L"1.0.0", L"HenBagle", SPI_GAME_LE3, SPI_VERSION_ANY);
SPI_PLUGINSIDE_POSTLOAD;
SPI_PLUGINSIDE_ASYNCATTACH;

ME3TweaksASILogger logger("Kismet Logger v1", "KismetLog.txt");

// ProcessEvent hook
// ======================================================================
typedef void (*tProcessEvent)(UObject* Context, UFunction* Function, void* Parms, void* Result);
tProcessEvent ProcessEvent = nullptr;
tProcessEvent ProcessEvent_orig = nullptr;
void ProcessEvent_hook(UObject* Context, UFunction* Function, void* Parms, void* Result)
{
if (!strcmp(Function->GetFullName(), "Function Engine.SequenceOp.Activated"))
{
const auto op = reinterpret_cast<USequenceOp*>(Context);
char* fullInstancedPath = op->GetFullPath();
char* className = op->Class->Name.GetName();
logger.writeToLog(string_format("%s %s\n", className, fullInstancedPath), true);
}
ProcessEvent_orig(Context, Function, Parms, Result);
}


SPI_IMPLEMENT_ATTACH
{
Common::OpenConsole();

INIT_CHECK_SDK();
INIT_FIND_PATTERN_POSTHOOK(ProcessEvent, LE_PATTERN_POSTHOOK_PROCESSEVENT);
INIT_HOOK_PATTERN(ProcessEvent);

return true;
}

SPI_IMPLEMENT_DETACH
{
Common::CloseConsole();
return true;
}
// ASI code is all defined in KismetLogger.h in the LEASIMods repo
159 changes: 3 additions & 156 deletions LE3PNGScreenshots/LE3PNGScreenshots.cpp
View file
Original file line number Diff line number Diff line change
@@ -1,159 +1,6 @@
//#define GAMELE1
//#define GAMELE2
// Game
#define GAMELE3
#define MYHOOK "PNGScreenShots_"

#include "../../Shared-ASI/PNGScreenshots/PNGScreenshots.h"

#include <filesystem>
#include <stdio.h>
#include <fstream>
#include <Shlwapi.h>
#include "fpng/fpng.h"

#ifdef GAMELE1
#include "../LE1-SDK/SdkInitializer.h"
#include "../LE1-SDK/SdkHeaders.h"
#include "../LE1-SDK/Interface.h"
#include "../LE1-SDK/Common.h"
SPI_PLUGINSIDE_SUPPORT(L"PNGScreenShots", L"1.0.0", L"ME3Tweaks", SPI_GAME_LE1, SPI_VERSION_ANY);
#elif defined(GAMELE2)
#include "../LE2-SDK/SdkInitializer.h"
#include "../LE2-SDK/SdkHeaders.h"
#include "../LE2-SDK/Interface.h"
#include "../LE2-SDK/Common.h"
SPI_PLUGINSIDE_SUPPORT(L"PNGScreenShots", L"1.0.0", L"ME3Tweaks", SPI_GAME_LE2, SPI_VERSION_ANY);
#elif defined(GAMELE3)
#include "../LE3-SDK/SdkInitializer.h"
#include "../LE3-SDK/SdkHeaders.h"
#include "../LE3-SDK/Interface.h"
#include "../LE3-SDK/Common.h"
SPI_PLUGINSIDE_SUPPORT(L"PNGScreenShots", L"1.0.0", L"ME3Tweaks", SPI_GAME_LE3, SPI_VERSION_ANY);
#endif

SPI_PLUGINSIDE_POSTLOAD;
SPI_PLUGINSIDE_ASYNCATTACH;

// This project doesn't use C++ 20 (due to it making it not compilable) so we use c++ 11
// Taken from https://stackoverflow.com/a/26221725/800318
template<typename ... Args>
std::wstring wstring_format(const std::wstring& format, Args ... args)
{
int size_s = _snwprintf(nullptr, 0, format.c_str(), args ...) + 1; // Extra space for '\0'
if (size_s <= 0) { throw std::runtime_error("Error during formatting."); }
auto size = static_cast<size_t>(size_s);
auto buf = std::make_unique<wchar_t[]>(size);
_snwprintf(buf.get(), size, format.c_str(), args ...);
return std::wstring(buf.get(), buf.get() + size - 1); // We don't want the '\0' inside
}

// ======================================================================
// appCreateBitmap hook
// ======================================================================

// Prototypes for appCreateBitmap
typedef void (*tAppCreateBitmap)(wchar_t* pattern, int width, int height, FColor* data, void* fileManager);
tAppCreateBitmap appCreateBitmap = nullptr;
tAppCreateBitmap appCreateBitmap_orig = nullptr;

bool fpngInitialized = false;

// The initial screenshot index we should check against. As we take screenshots we know the previous images will always exist
// and as such we can just increment this on auto-generated screenshots to match the game's incremental system
int cachedScreenshotIndex = 1;

// Hopefully user never gets here but this is so it doesn't make an infinite loop
int maxScreenshotIndex = 99999;

// data is an array of FColor structs (BGRA) of size width * height
void appCreateBitmap_hook(wchar_t* inputBaseName, int width, int height, FColor* data, void* fileManager)
{
if (!fpngInitialized)
{
// Initialize the library before use
fpng::fpng_init();
fpngInitialized = true;
}

long byteCount = width * height * 3;

// Color order needs swapped around for FPNG to access data
// since nothing supports BRGA.
std::vector<unsigned char> newImageData(byteCount);
int pixelIndex = 0; // Which pixel we are one
long totalCount = width * height; // how many pixels
long position = 0; // Where to place into vector
for (long i = 0; i < totalCount; i++)
{
auto color = &data[pixelIndex];
newImageData[position] = color->R;
newImageData[position + 1L] = color->G;
newImageData[position + 2L] = color->B;

pixelIndex++;
position = pixelIndex * 3;
}

// Determine output filename.
auto path = std::filesystem::path(inputBaseName);
auto extension = path.extension();
if (extension != ".png")
{
// Calculate a new name
auto outPath = std::filesystem::path(inputBaseName);
std::wstring newFname;
while (cachedScreenshotIndex < maxScreenshotIndex)
{
#ifdef GAMELE1
newFname = wstring_format(L"PNGLE1ScreenShot%05i.png", cachedScreenshotIndex);
#elif defined(GAMELE2)
newFname = wstring_format(L"PNGLE2ScreenShot%05i.png", cachedScreenshotIndex);
#elif defined(GAMELE3)
newFname = wstring_format(L"PNGLE3ScreenShot%05i.png", cachedScreenshotIndex);
#endif
cachedScreenshotIndex++; // Increment the cached index immediately.
outPath.replace_filename(newFname);

// Check if file exists
if (!std::filesystem::exists(outPath))
{
// File doesn't exist. Use this one
path = outPath;
break;
}
// File exists, go to next one
}

if (cachedScreenshotIndex == maxScreenshotIndex)
return; // Can't take any more screenshots
}

// Save the image data using the fpng library.
fpng::fpng_encode_image_to_wfile(path.c_str(), newImageData.data(), width, height, 3, 0U); // 3bpp, no flags
}


SPI_IMPLEMENT_ATTACH
{
// Hook appCreateBitmap and provide our own implementation of saving the bitmap data to disk.
auto _ = SDKInitializer::Instance();
#ifdef GAMELE1
INIT_FIND_PATTERN_POSTHOOK(appCreateBitmap, /*40 55 53 56 57*/ "41 54 41 55 41 56 41 57 48 8d ac 24 38 f8 ff ff 48 81 ec c8 08 00 00 48 c7 44 24 70 fe ff ff ff 48 8b 05 3c b6 55 01 48 33 c4 48 89 85 b0 07 00 00 4c 89 4c 24 48 44 89 44 24 58");
#elif defined(GAMELE2)
INIT_FIND_PATTERN_POSTHOOK(appCreateBitmap, /*40 55 53 56 57 */ "41 54 41 55 41 56 41 57 48 8d ac 24 38 f8 ff ff 48 81 ec c8 08 00 00 48 c7 44 24 70 fe ff ff ff 48 8b 05 ec 2a 58 01");
#elif defined(GAMELE3)
INIT_FIND_PATTERN_POSTHOOK(appCreateBitmap, /*40 55 53 56 57 */ "41 54 41 55 41 56 41 57 48 8d ac 24 78 f8 ff ff 48 81 ec 88 08 00 00 48 c7 44 24 60 fe ff ff ff");
#endif

if (auto rc = InterfacePtr->InstallHook(MYHOOK "appCreateBitmap", appCreateBitmap, appCreateBitmap_hook, (void**)&appCreateBitmap_orig);
rc != SPIReturn::Success)
{
writeln(L"Attach - failed to hook appCreateBitmap: %d / %s", rc, SPIReturnToString(rc));
return false;
}
return true;
}

SPI_IMPLEMENT_DETACH
{
return true;
}
// ASI code is all defined in PNGScreenShots.h in the LEASIMods repo
10 changes: 2 additions & 8 deletions LE3PNGScreenshots/LE3PNGScreenshots.vcxproj
View file
Original file line number Diff line number Diff line change
Expand Up @@ -89,17 +89,11 @@
</Link>
</ItemDefinitionGroup>
<ItemGroup>
<ClCompile Include="fpng\fpng.cpp" />
<ClCompile Include="LE3PNGScreenShots.cpp" />
</ItemGroup>
<ItemGroup>
<ClInclude Include="..\Common.h" />
<ClInclude Include="..\Interface.h" />
<ClInclude Include="fpng\fpng.h" />
</ItemGroup>
<ItemGroup>
<ProjectReference Include="..\LE1-SDK\LE1-SDK.vcxproj">
<Project>{e8071203-085e-4b90-9c5c-39d546ed6fe2}</Project>
<ProjectReference Include="..\LE3-SDK\LE3-SDK.vcxproj">
<Project>{8da34ede-7d5c-4da3-8841-78ab05e8c6d2}</Project>
</ProjectReference>
</ItemGroup>
<Import Project="$(VCTargetsPath)\Microsoft.Cpp.targets" />
Expand Down
Loading

0 comments on commit 7f1a8d4

Please sign in to comment.