From df04acc1b9ef9a5c50e8d460fc3da8f300440a70 Mon Sep 17 00:00:00 2001
From: WerWolv <werwolv98@gmail.com>
Date: Mon, 29 Apr 2024 21:20:22 +0200
Subject: [PATCH] impr: Allow specifying buffer size in integer to buffer cast
 node

---
 .../data_processor_nodes/other_nodes.cpp      | 23 +++++++++++++++----
 1 file changed, 19 insertions(+), 4 deletions(-)

diff --git a/plugins/builtin/source/content/data_processor_nodes/other_nodes.cpp b/plugins/builtin/source/content/data_processor_nodes/other_nodes.cpp
index 63d2799d5..c00401a29 100644
--- a/plugins/builtin/source/content/data_processor_nodes/other_nodes.cpp
+++ b/plugins/builtin/source/content/data_processor_nodes/other_nodes.cpp
@@ -83,15 +83,30 @@ namespace hex::plugin::builtin {
 
     class NodeCastIntegerToBuffer : public dp::Node {
     public:
-        NodeCastIntegerToBuffer() : Node("hex.builtin.nodes.casting.int_to_buffer.header", { dp::Attribute(dp::Attribute::IOType::In, dp::Attribute::Type::Integer, "hex.builtin.nodes.common.input"), dp::Attribute(dp::Attribute::IOType::Out, dp::Attribute::Type::Buffer, "hex.builtin.nodes.common.output") }) { }
+        NodeCastIntegerToBuffer() : Node("hex.builtin.nodes.casting.int_to_buffer.header", { dp::Attribute(dp::Attribute::IOType::In, dp::Attribute::Type::Integer, "hex.builtin.nodes.common.input"), dp::Attribute(dp::Attribute::IOType::In, dp::Attribute::Type::Integer, "hex.builtin.nodes.buffer.size"), dp::Attribute(dp::Attribute::IOType::Out, dp::Attribute::Type::Buffer, "hex.builtin.nodes.common.output") }) { }
 
         void process() override {
             const auto &input = this->getIntegerOnInput(0);
+            auto size = this->getIntegerOnInput(1);
 
-            std::vector<u8> output(sizeof(input), 0x00);
-            std::memcpy(output.data(), &input, sizeof(input));
+            if (size == 0) {
+                for (u32 i = 0; i < sizeof(input); i++) {
+                    if ((input >> (i * 8)) == 0) {
+                        size = i;
+                        break;
+                    }
+                }
 
-            this->setBufferOnOutput(1, output);
+                if (size == 0)
+                    size = 1;
+            } else if (size > sizeof(input)) {
+                throwNodeError("Integers cannot hold more than 16 bytes");
+            }
+
+            std::vector<u8> output(size, 0x00);
+            std::memcpy(output.data(), &input, size);
+
+            this->setBufferOnOutput(2, output);
         }
     };