139 lines
4.8 KiB
C++
139 lines
4.8 KiB
C++
#include <hex/data_processor/node.hpp>
|
|
|
|
#include <hex/helpers/fmt.hpp>
|
|
|
|
#include <hex/api/localization.hpp>
|
|
#include <hex/providers/provider.hpp>
|
|
|
|
namespace hex::dp {
|
|
|
|
int Node::s_idCounter = 1;
|
|
|
|
Node::Node(std::string unlocalizedTitle, std::vector<Attribute> attributes) : m_id(Node::s_idCounter++), m_unlocalizedTitle(std::move(unlocalizedTitle)), m_attributes(std::move(attributes)) {
|
|
for (auto &attr : this->m_attributes)
|
|
attr.setParentNode(this);
|
|
}
|
|
|
|
const std::vector<u8>& Node::getBufferOnInput(u32 index) {
|
|
auto attribute = this->getConnectedInputAttribute(index);
|
|
|
|
if (attribute == nullptr)
|
|
throwNodeError(hex::format("Nothing connected to input '{0}'", LangEntry(this->m_attributes[index].getUnlocalizedName())));
|
|
|
|
if (attribute->getType() != Attribute::Type::Buffer)
|
|
throwNodeError("Tried to read buffer from non-buffer attribute");
|
|
|
|
markInputProcessed(index);
|
|
attribute->getParentNode()->process();
|
|
|
|
auto &outputData = attribute->getOutputData();
|
|
|
|
if (outputData.empty())
|
|
throwNodeError("No data available at connected attribute");
|
|
|
|
return outputData;
|
|
}
|
|
|
|
const i128& Node::getIntegerOnInput(u32 index) {
|
|
auto attribute = this->getConnectedInputAttribute(index);
|
|
|
|
auto &outputData = [&] -> std::vector<u8>& {
|
|
if (attribute != nullptr) {
|
|
if (attribute->getType() != Attribute::Type::Integer)
|
|
throwNodeError("Tried to read integer from non-integer attribute");
|
|
|
|
markInputProcessed(index);
|
|
attribute->getParentNode()->process();
|
|
|
|
return attribute->getOutputData();
|
|
} else {
|
|
return this->getAttribute(index).getOutputData();
|
|
}
|
|
}();
|
|
|
|
if (outputData.empty())
|
|
throwNodeError("No data available at connected attribute");
|
|
|
|
if (outputData.size() < sizeof(i128))
|
|
throwNodeError("Not enough data provided for integer");
|
|
|
|
return *reinterpret_cast<i128 *>(outputData.data());
|
|
}
|
|
|
|
const long double& Node::getFloatOnInput(u32 index) {
|
|
auto attribute = this->getConnectedInputAttribute(index);
|
|
|
|
auto &outputData = [&] -> std::vector<u8>& {
|
|
if (attribute != nullptr) {
|
|
if (attribute->getType() != Attribute::Type::Float)
|
|
throwNodeError("Tried to read integer from non-float attribute");
|
|
|
|
markInputProcessed(index);
|
|
attribute->getParentNode()->process();
|
|
|
|
return attribute->getOutputData();
|
|
} else {
|
|
return this->getAttribute(index).getOutputData();
|
|
}
|
|
}();
|
|
|
|
if (outputData.empty())
|
|
throwNodeError("No data available at connected attribute");
|
|
|
|
if (outputData.size() < sizeof(long double))
|
|
throwNodeError("Not enough data provided for float");
|
|
|
|
return *reinterpret_cast<long double *>(outputData.data());
|
|
}
|
|
|
|
void Node::setBufferOnOutput(u32 index, std::span<const u8> data) {
|
|
if (index >= this->getAttributes().size())
|
|
throwNodeError("Attribute index out of bounds!");
|
|
|
|
auto &attribute = this->getAttributes()[index];
|
|
|
|
if (attribute.getIOType() != Attribute::IOType::Out)
|
|
throwNodeError("Tried to set output data of an input attribute!");
|
|
|
|
attribute.getOutputData() = { data.begin(), data.end() };
|
|
}
|
|
|
|
void Node::setIntegerOnOutput(u32 index, i128 integer) {
|
|
if (index >= this->getAttributes().size())
|
|
throwNodeError("Attribute index out of bounds!");
|
|
|
|
auto &attribute = this->getAttributes()[index];
|
|
|
|
if (attribute.getIOType() != Attribute::IOType::Out)
|
|
throwNodeError("Tried to set output data of an input attribute!");
|
|
|
|
std::vector<u8> buffer(sizeof(integer), 0);
|
|
std::memcpy(buffer.data(), &integer, sizeof(integer));
|
|
|
|
attribute.getOutputData() = buffer;
|
|
}
|
|
|
|
void Node::setFloatOnOutput(u32 index, long double floatingPoint) {
|
|
if (index >= this->getAttributes().size())
|
|
throwNodeError("Attribute index out of bounds!");
|
|
|
|
auto &attribute = this->getAttributes()[index];
|
|
|
|
if (attribute.getIOType() != Attribute::IOType::Out)
|
|
throwNodeError("Tried to set output data of an input attribute!");
|
|
|
|
std::vector<u8> buffer(sizeof(floatingPoint), 0);
|
|
std::memcpy(buffer.data(), &floatingPoint, sizeof(floatingPoint));
|
|
|
|
attribute.getOutputData() = buffer;
|
|
}
|
|
|
|
void Node::setOverlayData(u64 address, const std::vector<u8> &data) {
|
|
if (this->m_overlay == nullptr)
|
|
throwNodeError("Tried setting overlay data on a node that's not the end of a chain!");
|
|
|
|
this->m_overlay->setAddress(address);
|
|
this->m_overlay->getData() = data;
|
|
}
|
|
|
|
} |