53 lines
1.7 KiB
C++
53 lines
1.7 KiB
C++
/* SPDX-License-Identifier: GPL-2.0-or-later */
|
|
|
|
#include "BLI_string_utf8.h"
|
|
|
|
#include "node_function_util.hh"
|
|
|
|
namespace blender::nodes::node_fn_replace_string_cc {
|
|
|
|
static void fn_node_replace_string_declare(NodeDeclarationBuilder &b)
|
|
{
|
|
b.add_input<decl::String>(N_("String"));
|
|
b.add_input<decl::String>(N_("Find")).description(N_("The string to find in the input string"));
|
|
b.add_input<decl::String>(N_("Replace"))
|
|
.description(N_("The string to replace each match with"));
|
|
b.add_output<decl::String>(N_("String"));
|
|
}
|
|
|
|
static std::string replace_all(const StringRefNull str,
|
|
const StringRefNull from,
|
|
const StringRefNull to)
|
|
{
|
|
if (from.is_empty()) {
|
|
return str;
|
|
}
|
|
char *new_str_ptr = BLI_str_replaceN(str.c_str(), from.c_str(), to.c_str());
|
|
std::string new_str{new_str_ptr};
|
|
MEM_freeN(new_str_ptr);
|
|
return new_str;
|
|
}
|
|
|
|
static void fn_node_replace_string_build_multi_function(NodeMultiFunctionBuilder &builder)
|
|
{
|
|
static fn::CustomMF_SI_SI_SI_SO<std::string, std::string, std::string, std::string> substring_fn{
|
|
"Replace", [](const std::string &str, const std::string &find, const std::string &replace) {
|
|
return replace_all(str, find, replace);
|
|
}};
|
|
builder.set_matching_fn(&substring_fn);
|
|
}
|
|
|
|
} // namespace blender::nodes::node_fn_replace_string_cc
|
|
|
|
void register_node_type_fn_replace_string()
|
|
{
|
|
namespace file_ns = blender::nodes::node_fn_replace_string_cc;
|
|
|
|
static bNodeType ntype;
|
|
|
|
fn_node_type_base(&ntype, FN_NODE_REPLACE_STRING, "Replace String", NODE_CLASS_CONVERTER);
|
|
ntype.declare = file_ns::fn_node_replace_string_declare;
|
|
ntype.build_multi_function = file_ns::fn_node_replace_string_build_multi_function;
|
|
nodeRegisterType(&ntype);
|
|
}
|