123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421 |
- #include "multimesh_editor_plugin.h"
- #include "scene/3d/mesh_instance.h"
- #include "scene/gui/box_container.h"
- #include "spatial_editor_plugin.h"
- void MultiMeshEditor::_node_removed(Node *p_node) {
- if (p_node == node) {
- node = NULL;
- hide();
- }
- }
- void MultiMeshEditor::_populate() {
- if (!node)
- return;
- Ref<Mesh> mesh;
- if (mesh_source->get_text() == "") {
- Ref<MultiMesh> multimesh;
- multimesh = node->get_multimesh();
- if (multimesh.is_null()) {
- err_dialog->set_text(TTR("No mesh source specified (and no MultiMesh set in node)."));
- err_dialog->popup_centered_minsize();
- return;
- }
- if (multimesh->get_mesh().is_null()) {
- err_dialog->set_text(TTR("No mesh source specified (and MultiMesh contains no Mesh)."));
- err_dialog->popup_centered_minsize();
- return;
- }
- mesh = multimesh->get_mesh();
- } else {
- Node *ms_node = node->get_node(mesh_source->get_text());
- if (!ms_node) {
- err_dialog->set_text(TTR("Mesh source is invalid (invalid path)."));
- err_dialog->popup_centered_minsize();
- return;
- }
- MeshInstance *ms_instance = Object::cast_to<MeshInstance>(ms_node);
- if (!ms_instance) {
- err_dialog->set_text(TTR("Mesh source is invalid (not a MeshInstance)."));
- err_dialog->popup_centered_minsize();
- return;
- }
- mesh = ms_instance->get_mesh();
- if (mesh.is_null()) {
- err_dialog->set_text(TTR("Mesh source is invalid (contains no Mesh resource)."));
- err_dialog->popup_centered_minsize();
- return;
- }
- }
- if (surface_source->get_text() == "") {
- err_dialog->set_text(TTR("No surface source specified."));
- err_dialog->popup_centered_minsize();
- return;
- }
- Node *ss_node = node->get_node(surface_source->get_text());
- if (!ss_node) {
- err_dialog->set_text(TTR("Surface source is invalid (invalid path)."));
- err_dialog->popup_centered_minsize();
- return;
- }
- GeometryInstance *ss_instance = Object::cast_to<MeshInstance>(ss_node);
- if (!ss_instance) {
- err_dialog->set_text(TTR("Surface source is invalid (no geometry)."));
- err_dialog->popup_centered_minsize();
- return;
- }
- Transform geom_xform = node->get_global_transform().affine_inverse() * ss_instance->get_global_transform();
- PoolVector<Face3> geometry = ss_instance->get_faces(VisualInstance::FACES_SOLID);
- if (geometry.size() == 0) {
- err_dialog->set_text(TTR("Surface source is invalid (no faces)."));
- err_dialog->popup_centered_minsize();
- return;
- }
-
- int gc = geometry.size();
- PoolVector<Face3>::Write w = geometry.write();
- for (int i = 0; i < gc; i++) {
- for (int j = 0; j < 3; j++) {
- w[i].vertex[j] = geom_xform.xform(w[i].vertex[j]);
- }
- }
- w = PoolVector<Face3>::Write();
- PoolVector<Face3> faces = geometry;
- ERR_EXPLAIN(TTR("Parent has no solid faces to populate."));
- int facecount = faces.size();
- ERR_FAIL_COND(!facecount);
- PoolVector<Face3>::Read r = faces.read();
- float area_accum = 0;
- Map<float, int> triangle_area_map;
- for (int i = 0; i < facecount; i++) {
- float area = r[i].get_area();
- if (area < CMP_EPSILON)
- continue;
- triangle_area_map[area_accum] = i;
- area_accum += area;
- }
- ERR_EXPLAIN(TTR("Couldn't map area."));
- ERR_FAIL_COND(triangle_area_map.size() == 0);
- ERR_EXPLAIN(TTR("Couldn't map area."));
- ERR_FAIL_COND(area_accum == 0);
- Ref<MultiMesh> multimesh = memnew(MultiMesh);
- multimesh->set_mesh(mesh);
- int instance_count = populate_amount->get_value();
- multimesh->set_transform_format(MultiMesh::TRANSFORM_3D);
- multimesh->set_color_format(MultiMesh::COLOR_NONE);
- multimesh->set_instance_count(instance_count);
- float _tilt_random = populate_tilt_random->get_value();
- float _rotate_random = populate_rotate_random->get_value();
- float _scale_random = populate_scale_random->get_value();
- float _scale = populate_scale->get_value();
- int axis = populate_axis->get_selected();
- Transform axis_xform;
- if (axis == Vector3::AXIS_Z) {
- axis_xform.rotate(Vector3(1, 0, 0), -Math_PI * 0.5);
- }
- if (axis == Vector3::AXIS_X) {
- axis_xform.rotate(Vector3(0, 0, 1), -Math_PI * 0.5);
- }
- for (int i = 0; i < instance_count; i++) {
- float areapos = Math::random(0.0f, area_accum);
- Map<float, int>::Element *E = triangle_area_map.find_closest(areapos);
- ERR_FAIL_COND(!E)
- int index = E->get();
- ERR_FAIL_INDEX(index, facecount);
-
- Face3 face = r[index];
-
- Vector3 pos = face.get_random_point_inside();
- Vector3 normal = face.get_plane().normal;
- Vector3 op_axis = (face.vertex[0] - face.vertex[1]).normalized();
- Transform xform;
- xform.set_look_at(pos, pos + op_axis, normal);
- xform = xform * axis_xform;
- Basis post_xform;
- post_xform.rotate(xform.basis.get_axis(1), -Math::random(-_rotate_random, _rotate_random) * Math_PI);
- post_xform.rotate(xform.basis.get_axis(2), -Math::random(-_tilt_random, _tilt_random) * Math_PI);
- post_xform.rotate(xform.basis.get_axis(0), -Math::random(-_tilt_random, _tilt_random) * Math_PI);
- xform.basis = post_xform * xform.basis;
-
- xform.basis.scale(Vector3(1, 1, 1) * (_scale + Math::random(-_scale_random, _scale_random)));
- multimesh->set_instance_transform(i, xform);
- }
- node->set_multimesh(multimesh);
- }
- void MultiMeshEditor::_browsed(const NodePath &p_path) {
- NodePath path = node->get_path_to(get_node(p_path));
- if (browsing_source)
- mesh_source->set_text(path);
- else
- surface_source->set_text(path);
- }
- void MultiMeshEditor::_menu_option(int p_option) {
- switch (p_option) {
- case MENU_OPTION_POPULATE: {
- if (_last_pp_node != node) {
- surface_source->set_text("..");
- mesh_source->set_text("..");
- populate_axis->select(1);
- populate_rotate_random->set_value(0);
- populate_tilt_random->set_value(0);
- populate_scale_random->set_value(0);
- populate_scale->set_value(1);
- populate_amount->set_value(128);
- _last_pp_node = node;
- }
- populate_dialog->popup_centered(Size2(250, 380));
- } break;
- }
- }
- void MultiMeshEditor::edit(MultiMeshInstance *p_multimesh) {
- node = p_multimesh;
- }
- void MultiMeshEditor::_browse(bool p_source) {
- browsing_source = p_source;
- std->get_scene_tree()->set_marked(node, false);
- std->popup_centered_ratio();
- if (p_source)
- std->set_title(TTR("Select a Source Mesh:"));
- else
- std->set_title(TTR("Select a Target Surface:"));
- }
- void MultiMeshEditor::_bind_methods() {
- ClassDB::bind_method("_menu_option", &MultiMeshEditor::_menu_option);
- ClassDB::bind_method("_populate", &MultiMeshEditor::_populate);
- ClassDB::bind_method("_browsed", &MultiMeshEditor::_browsed);
- ClassDB::bind_method("_browse", &MultiMeshEditor::_browse);
- }
- MultiMeshEditor::MultiMeshEditor() {
- options = memnew(MenuButton);
- SpatialEditor::get_singleton()->add_control_to_menu_panel(options);
- options->set_text("MultiMesh");
- options->set_icon(EditorNode::get_singleton()->get_gui_base()->get_icon("MultiMeshInstance", "EditorIcons"));
- options->get_popup()->add_item(TTR("Populate Surface"));
- options->get_popup()->connect("id_pressed", this, "_menu_option");
- populate_dialog = memnew(ConfirmationDialog);
- populate_dialog->set_title(TTR("Populate MultiMesh"));
- add_child(populate_dialog);
- VBoxContainer *vbc = memnew(VBoxContainer);
- populate_dialog->add_child(vbc);
-
- HBoxContainer *hbc = memnew(HBoxContainer);
- surface_source = memnew(LineEdit);
- hbc->add_child(surface_source);
- surface_source->set_h_size_flags(SIZE_EXPAND_FILL);
- Button *b = memnew(Button);
- hbc->add_child(b);
- b->set_text("..");
- b->connect("pressed", this, "_browse", make_binds(false));
- vbc->add_margin_child(TTR("Target Surface:"), hbc);
- hbc = memnew(HBoxContainer);
- mesh_source = memnew(LineEdit);
- hbc->add_child(mesh_source);
- mesh_source->set_h_size_flags(SIZE_EXPAND_FILL);
- b = memnew(Button);
- hbc->add_child(b);
- b->set_text("..");
- vbc->add_margin_child(TTR("Source Mesh:"), hbc);
- b->connect("pressed", this, "_browse", make_binds(true));
- populate_axis = memnew(OptionButton);
- populate_axis->add_item(TTR("X-Axis"));
- populate_axis->add_item(TTR("Y-Axis"));
- populate_axis->add_item(TTR("Z-Axis"));
- populate_axis->select(2);
- vbc->add_margin_child(TTR("Mesh Up Axis:"), populate_axis);
- populate_rotate_random = memnew(HSlider);
- populate_rotate_random->set_max(1);
- populate_rotate_random->set_step(0.01);
- vbc->add_margin_child(TTR("Random Rotation:"), populate_rotate_random);
- populate_tilt_random = memnew(HSlider);
- populate_tilt_random->set_max(1);
- populate_tilt_random->set_step(0.01);
- vbc->add_margin_child(TTR("Random Tilt:"), populate_tilt_random);
- populate_scale_random = memnew(SpinBox);
- populate_scale_random->set_min(0);
- populate_scale_random->set_max(1);
- populate_scale_random->set_value(0);
- populate_scale_random->set_step(0.01);
- vbc->add_margin_child(TTR("Random Scale:"), populate_scale_random);
- populate_scale = memnew(SpinBox);
- populate_scale->set_min(0.001);
- populate_scale->set_max(4096);
- populate_scale->set_value(1);
- populate_scale->set_step(0.01);
- vbc->add_margin_child(TTR("Scale:"), populate_scale);
- populate_amount = memnew(SpinBox);
- populate_amount->set_anchor(MARGIN_RIGHT, ANCHOR_END);
- populate_amount->set_begin(Point2(20, 232));
- populate_amount->set_end(Point2(-5, 237));
- populate_amount->set_min(1);
- populate_amount->set_max(65536);
- populate_amount->set_value(128);
- vbc->add_margin_child(TTR("Amount:"), populate_amount);
- populate_dialog->get_ok()->set_text(TTR("Populate"));
- populate_dialog->get_ok()->connect("pressed", this, "_populate");
- std = memnew(SceneTreeDialog);
- populate_dialog->add_child(std);
- std->connect("selected", this, "_browsed");
- _last_pp_node = NULL;
- err_dialog = memnew(AcceptDialog);
- add_child(err_dialog);
- }
- void MultiMeshEditorPlugin::edit(Object *p_object) {
- multimesh_editor->edit(Object::cast_to<MultiMeshInstance>(p_object));
- }
- bool MultiMeshEditorPlugin::handles(Object *p_object) const {
- return p_object->is_class("MultiMeshInstance");
- }
- void MultiMeshEditorPlugin::make_visible(bool p_visible) {
- if (p_visible) {
- multimesh_editor->options->show();
- } else {
- multimesh_editor->options->hide();
- multimesh_editor->edit(NULL);
- }
- }
- MultiMeshEditorPlugin::MultiMeshEditorPlugin(EditorNode *p_node) {
- editor = p_node;
- multimesh_editor = memnew(MultiMeshEditor);
- editor->get_viewport()->add_child(multimesh_editor);
- multimesh_editor->options->hide();
- }
- MultiMeshEditorPlugin::~MultiMeshEditorPlugin() {
- }
|