godot/tools/editor/plugins/multimesh_editor_plugin.cpp

/*************************************************************************/
/*  multimesh_editor_plugin.cpp                                          */
/*************************************************************************/
/*                       This file is part of:                           */
/*                           GODOT ENGINE                                */
/*                    http://www.godotengine.org                         */
/*************************************************************************/
/* Copyright (c) 2007-2017 Juan Linietsky, Ariel Manzur.                 */
/*                                                                       */
/* Permission is hereby granted, free of charge, to any person obtaining */
/* a copy of this software and associated documentation files (the       */
/* "Software"), to deal in the Software without restriction, including   */
/* without limitation the rights to use, copy, modify, merge, publish,   */
/* distribute, sublicense, and/or sell copies of the Software, and to    */
/* permit persons to whom the Software is furnished to do so, subject to */
/* the following conditions:                                             */
/*                                                                       */
/* The above copyright notice and this permission notice shall be        */
/* included in all copies or substantial portions of the Software.       */
/*                                                                       */
/* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,       */
/* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF    */
/* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.*/
/* IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY  */
/* CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,  */
/* TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE     */
/* SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.                */
/*************************************************************************/
#include "multimesh_editor_plugin.h"

#include "scene/gui/box_container.h"
#include "scene/3d/mesh_instance.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 = ms_node->cast_to<MeshInstance>();

		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 = ss_node->cast_to<MeshInstance>();

	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;
	}

	//make all faces local

	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();
#if 0
	node->get_multimesh()->set_instance_count(populate_amount->get_val());
	node->populate_parent(populate_rotate_random->get_val(),populate_tilt_random->get_val(),populate_scale_random->get_val(),populate_scale->get_val());


	ERR_EXPLAIN("Parent is not of type VisualInstance.");
	ERR_FAIL_COND(!get_parent() || !get_parent()->is_type("VisualInstance"));

	ERR_EXPLAIN("Multimesh not present.");
	ERR_FAIL_COND(multimesh.is_null());

	VisualInstance *vi = get_parent()->cast_to<VisualInstance>();
	ERR_EXPLAIN("Parent is not of type VisualInstance, can't be populated.");
	ERR_FAIL_COND(!vi);

#endif
	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,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);

		// ok FINALLY get face
		Face3 face = r[index];
		//now compute some position inside the face...

		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.orthonormalize();


		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);
	//populate_dialog->set_child_rect(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(p_object->cast_to<MultiMeshInstance>());
}

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()
{
}