Skip to content

Commit

Permalink
feat(natives/five): SET_RUNTIME_TEXTURE_IMAGE command
Browse files Browse the repository at this point in the history 
Another end user request: https://forum.cfx.re/t/5015014
  • Loading branch information
@blattersturm
blattersturm committed Feb 4, 2023
1 parent 0485939 commit cbfe089
Show file tree
Hide file tree
Showing 2 changed files with 186 additions and 45 deletions.
209 changes: 164 additions & 45 deletions code/components/extra-natives-five/src/RuntimeAssetNatives.cpp
View file
Expand Up @@ -75,6 +75,8 @@ class RuntimeTex

bool SetPixelData(const void* data, size_t length);

bool LoadImage(const char* fileName);

void Commit();

inline void SetReferenceData(fwRefContainer<fwRefCountable> reference)
Expand Down Expand Up @@ -358,20 +360,16 @@ RuntimeTex* RuntimeTxd::CreateTextureFromDui(const char* name, const char* duiHa

#pragma comment(lib, "windowscodecs.lib")

ComPtr<IWICImagingFactory> g_imagingFactory;
static ComPtr<IWICImagingFactory> g_imagingFactory;

RuntimeTex* RuntimeTxd::CreateTextureFromImage(const char* name, const char* fileName)
static ComPtr<IWICBitmapSource> ImageToBitmapSource(std::string_view fileName)
{
if (!m_txd)
{
return nullptr;
}

if (!g_imagingFactory)
{
HRESULT hr = CoCreateInstance(CLSID_WICImagingFactory1, nullptr, CLSCTX_INPROC_SERVER, __uuidof(IWICImagingFactory), (void**)g_imagingFactory.GetAddressOf());
}

ComPtr<IWICBitmapSource> source;
ComPtr<IStream> stream;

std::string fileNameString(fileName);
Expand All @@ -382,7 +380,7 @@ RuntimeTex* RuntimeTxd::CreateTextureFromImage(const char* name, const char* fil

if (f == std::string::npos)
{
return nullptr;
return {};
}

fileNameString = fileNameString.substr(f + 7);
Expand All @@ -391,14 +389,16 @@ RuntimeTex* RuntimeTxd::CreateTextureFromImage(const char* name, const char* fil
UrlDecode(fileNameString, decodedURL, false);

decodedURL.erase(std::remove_if(decodedURL.begin(), decodedURL.end(), [](char c)
{
return std::isspace<char>(c, std::locale::classic());
}), decodedURL.end());
{
return std::isspace<char>(c, std::locale::classic());
}),
decodedURL.end());

size_t length = decodedURL.length();
size_t paddingNeeded = 4 - (length % 4);

if ((paddingNeeded == 1 || paddingNeeded == 2) && decodedURL[length - 1] != '=') {
if ((paddingNeeded == 1 || paddingNeeded == 2) && decodedURL[length - 1] != '=')
{
decodedURL.resize(length + paddingNeeded, '=');
}

Expand All @@ -412,12 +412,12 @@ RuntimeTex* RuntimeTxd::CreateTextureFromImage(const char* name, const char* fil

if (!FX_SUCCEEDED(fx::GetCurrentScriptRuntime(&runtime)))
{
return nullptr;
return {};
}

fx::Resource* resource = reinterpret_cast<fx::Resource*>(runtime->GetParentObject());

stream = vfs::CreateComStream(vfs::OpenRead(resource->GetPath() + "/" + fileName));
stream = vfs::CreateComStream(vfs::OpenRead(fmt::sprintf("%s/%s", resource->GetPath(), fileName)));
}

ComPtr<IWICBitmapDecoder> decoder;
Expand All @@ -432,51 +432,169 @@ RuntimeTex* RuntimeTxd::CreateTextureFromImage(const char* name, const char* fil

if (SUCCEEDED(hr))
{
ComPtr<IWICBitmapSource> source;
ComPtr<IWICBitmapSource> convertedSource;
// try to convert to a pixel format we like
frame.As(&source);
}
}

UINT width = 0, height = 0;
return source;
}

frame->GetSize(&width, &height);
RuntimeTex* RuntimeTxd::CreateTextureFromImage(const char* name, const char* fileName)
{
if (!m_txd)
{
return nullptr;
}

// try to convert to a pixel format we like
frame.As(&source);
if (auto source = ImageToBitmapSource(fileName))
{
ComPtr<IWICBitmapSource> convertedSource;

UINT width = 0, height = 0;
source->GetSize(&width, &height);

// try to convert to a pixel format we like
HRESULT hr = WICConvertBitmapSource(GUID_WICPixelFormat32bppBGRA, source.Get(), convertedSource.GetAddressOf());

if (SUCCEEDED(hr))
{
source = convertedSource;
}

// create a pixel data buffer
std::unique_ptr<uint32_t[]> pixelData(new uint32_t[width * height]);

hr = source->CopyPixels(nullptr, width * 4, width * height * 4, reinterpret_cast<BYTE*>(pixelData.get()));

if (SUCCEEDED(hr))
{
rage::grcTextureReference reference;
memset(&reference, 0, sizeof(reference));
reference.width = width;
reference.height = height;
reference.depth = 1;
reference.stride = width * 4;
reference.format = 11; // should correspond to DXGI_FORMAT_B8G8R8A8_UNORM
reference.pixelData = (uint8_t*)pixelData.get();

auto tex = std::make_shared<RuntimeTex>(rage::grcTextureFactory::getInstance()->createImage(&reference, nullptr), pixelData.get(), width * height * 4);
m_txd->Add(name, tex->GetTexture());

m_textures[name] = tex;

scrBindAddSafePointer(tex.get());
return tex.get();
}
}

return nullptr;
}

bool RuntimeTex::LoadImage(const char* fileName)
{
if (!m_texture)
{
return false;
}

auto completion = [this](const ComPtr<IWICBitmapSource>& bitmapSource)
{
auto source = bitmapSource;
ComPtr<IWICBitmapSource> convertedSource;

hr = WICConvertBitmapSource(GUID_WICPixelFormat32bppBGRA, source.Get(), convertedSource.GetAddressOf());
// try to convert to a pixel format we like
HRESULT hr = WICConvertBitmapSource(GUID_WICPixelFormat32bppBGRA, source.Get(), convertedSource.GetAddressOf());

if (SUCCEEDED(hr))
if (SUCCEEDED(hr))
{
source = convertedSource;
}

// set up the pixel data buffer
UINT width = 0, height = 0;
source->GetSize(&width, &height);

size_t length = (size_t(width) * size_t(height) * 4);

if (length == m_backingPixels.size())
{
if (SUCCEEDED(source->CopyPixels(nullptr, width * 4, width * height * 4, reinterpret_cast<BYTE*>(&m_backingPixels[0]))))
{
source = convertedSource;
Commit();

return true;
}
}

// create a pixel data buffer
std::unique_ptr<uint32_t[]> pixelData(new uint32_t[width * height]);
return false;
};

hr = source->CopyPixels(nullptr, width * 4, width * height * 4, reinterpret_cast<BYTE*>(pixelData.get()));
if (auto source = ImageToBitmapSource(fileName))
{
UINT width = 0, height = 0;
source->GetSize(&width, &height);

if (SUCCEEDED(hr))
if (width == m_texture->GetWidth() && height == m_texture->GetHeight())
{
return completion(source);
}
else
{
struct Item
{
rage::grcTextureReference reference;
memset(&reference, 0, sizeof(reference));
reference.width = width;
reference.height = height;
reference.depth = 1;
reference.stride = width * 4;
reference.format = 11; // should correspond to DXGI_FORMAT_B8G8R8A8_UNORM
reference.pixelData = (uint8_t*)pixelData.get();

auto tex = std::make_shared<RuntimeTex>(rage::grcTextureFactory::getInstance()->createImage(&reference, nullptr), pixelData.get(), width * height * 4);
m_txd->Add(name, tex->GetTexture());

m_textures[name] = tex;

scrBindAddSafePointer(tex.get());
return tex.get();
}
ComPtr<IWICBitmapSource> bitmap;
int targetWidth;
int targetHeight;
decltype(completion) cb;

Item(ComPtr<IWICBitmapSource>&& bitmap, int targetWidth, int targetHeight, decltype(completion)&& cb)
: bitmap(std::move(bitmap)), targetWidth(targetWidth), targetHeight(targetHeight), cb(std::move(cb))
{

}

void Work()
{
ComPtr<IWICBitmapScaler> scaler;

if (SUCCEEDED(g_imagingFactory->CreateBitmapScaler(&scaler)))
{
if (SUCCEEDED(scaler->Initialize(
bitmap.Get(),
targetWidth,
targetHeight,
WICBitmapInterpolationModeFant)))
{
ComPtr<IWICBitmapSource> outBitmap;
scaler.As(&outBitmap);

OnNextMainFrame([cb = std::move(cb), bitmap = std::move(outBitmap)]()
{
cb(bitmap);
});
}
}
}

static DWORD WINAPI StaticWork(LPVOID arg)
{
auto self = static_cast<Item*>(arg);
self->Work();
delete self;

return 0;
}
};

auto workItem = new Item(std::move(source), m_texture->GetWidth(), m_texture->GetHeight(), std::move(completion));
QueueUserWorkItem(&Item::StaticWork, workItem, 0);

return true;
}
}

return nullptr;
return false;
}

#define VFS_GET_RAGE_PAGE_FLAGS 0x20001
Expand Down Expand Up @@ -1006,6 +1124,7 @@ static InitFunction initFunction([]()
.AddMethod("GET_RUNTIME_TEXTURE_PITCH", &RuntimeTex::GetPitch)
.AddMethod("SET_RUNTIME_TEXTURE_PIXEL", &RuntimeTex::SetPixel)
.AddMethod("SET_RUNTIME_TEXTURE_ARGB_DATA", &RuntimeTex::SetPixelData)
.AddMethod("SET_RUNTIME_TEXTURE_IMAGE", &RuntimeTex::LoadImage)
.AddMethod("COMMIT_RUNTIME_TEXTURE", &RuntimeTex::Commit);

fx::ScriptEngine::RegisterNativeHandler("REGISTER_ARCHETYPES", [](fx::ScriptContext& context)
Expand Down
22 changes: 22 additions & 0 deletions ext/native-decls/SetRuntimeTextureImage.md
View file
@@ -0,0 +1,22 @@
---
ns: CFX
apiset: client
---
## SET_RUNTIME_TEXTURE_IMAGE

```c
BOOL SET_RUNTIME_TEXTURE_IMAGE(long tex, char* fileName);
```
Replaces the pixel data in a runtime texture with the image data from a file in the current resource, or a data URL.
If the bitmap is a different size compared to the existing texture, it will be resampled.
This command may end up executed asynchronously, and only update the texture data at a later time.
## Parameters
* **tex**: A runtime texture handle.
* **fileName**: The file name of an image to load, or a base64 "data:" URL. This should preferably be a PNG, and has to be specified as a `file` in the resource manifest.
## Return value
TRUE for success, FALSE for failure.

0 comments on commit cbfe089

Please sign in to comment.