pandemonium_engine/modules/material_maker/nodes/pattern/truchet.cpp

94 lines
2.3 KiB
C++

#include "truchet.h"
#include "../../algos/mm_algos.h"
#include "../../editor/mm_graph_node.h"
#include "../mm_material.h"
Ref<MMNodeUniversalProperty> MMTruchet::get_image() {
return image;
}
void MMTruchet::set_image(const Ref<MMNodeUniversalProperty> &val) {
image = val;
}
int MMTruchet::get_shape() const {
return shape;
}
void MMTruchet::set_shape(const int val) {
shape = val;
set_dirty(true);
}
float MMTruchet::get_size() const {
return size;
}
void MMTruchet::set_size(const float val) {
size = val;
set_dirty(true);
}
void MMTruchet::_init_properties() {
if (!image.is_valid()) {
image.instance();
image->set_default_type(MMNodeUniversalProperty::DEFAULT_TYPE_IMAGE);
}
image->set_output_slot_type(MMNodeUniversalProperty::SLOT_TYPE_IMAGE);
register_output_property(image);
}
void MMTruchet::_register_methods(MMGraphNode *mm_graph_node) {
mm_graph_node->add_slot_texture_universal(image);
Array arr;
arr.push_back("Line");
arr.push_back("Circle");
mm_graph_node->add_slot_enum("get_shape", "set_shape", "Shape", arr);
mm_graph_node->add_slot_float("get_size", "set_size", "Size", 1);
}
void MMTruchet::_render(const Ref<MMMaterial> &material) {
Ref<Image> img = render_image(material);
image->set_value(img);
}
Color MMTruchet::_get_value_for(const Vector2 &uv, const int pseed) {
if (shape == 0) {
return MMAlgos::truchet1c(uv, size, pseed);
}
else if (shape == 1) {
return MMAlgos::truchet2c(uv, size, pseed);
}
return Color();
}
MMTruchet::MMTruchet() {
shape = 0;
size = 4;
}
MMTruchet::~MMTruchet() {
}
void MMTruchet::_bind_methods() {
ClassDB::bind_method(D_METHOD("get_image"), &MMTruchet::get_image);
ClassDB::bind_method(D_METHOD("set_image", "value"), &MMTruchet::set_image);
ADD_PROPERTY(PropertyInfo(Variant::OBJECT, "image", PROPERTY_HINT_RESOURCE_TYPE, "MMNodeUniversalProperty"), "set_image", "get_image");
ClassDB::bind_method(D_METHOD("get_shape"), &MMTruchet::get_shape);
ClassDB::bind_method(D_METHOD("set_shape", "value"), &MMTruchet::set_shape);
ADD_PROPERTY(PropertyInfo(Variant::INT, "shape"), "set_shape", "get_shape");
ClassDB::bind_method(D_METHOD("get_size"), &MMTruchet::get_size);
ClassDB::bind_method(D_METHOD("set_size", "value"), &MMTruchet::set_size);
ADD_PROPERTY(PropertyInfo(Variant::REAL, "size"), "set_size", "get_size");
}