mirror of
				https://github.com/godotengine/godot.git
				synced 2025-10-30 21:21:10 +00:00 
			
		
		
		
	
		
			
				
	
	
		
			345 lines
		
	
	
	
		
			7.8 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			345 lines
		
	
	
	
		
			7.8 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
| #include "baked_light_editor_plugin.h"
 | |
| #include "scene/gui/box_container.h"
 | |
| #include "scene/3d/mesh_instance.h"
 | |
| #include "io/marshalls.h"
 | |
| #include "io/resource_saver.h"
 | |
| 
 | |
| 
 | |
| 
 | |
| 
 | |
| 
 | |
| void BakedLightEditor::_end_baking() {
 | |
| 
 | |
| 	baker->clear();
 | |
| 	set_process(false);
 | |
| 	button_bake->set_pressed(false);
 | |
| 	bake_info->set_text("");
 | |
| }
 | |
| 
 | |
| void BakedLightEditor::_node_removed(Node *p_node) {
 | |
| 
 | |
| 	if(p_node==node) {
 | |
| 		_end_baking();
 | |
| 		node=NULL;
 | |
| 
 | |
| 		hide();
 | |
| 	}
 | |
| 
 | |
| }
 | |
| 
 | |
| 
 | |
| 
 | |
| 
 | |
| 
 | |
| void BakedLightEditor::_notification(int p_option) {
 | |
| 
 | |
| 
 | |
| 	if (p_option==NOTIFICATION_ENTER_TREE) {
 | |
| 
 | |
| 		button_bake->set_icon(get_icon("Bake","EditorIcons"));
 | |
| 		button_reset->set_icon(get_icon("Reload","EditorIcons"));
 | |
| 		button_make_lightmaps->set_icon(get_icon("LightMap","EditorIcons"));
 | |
| 	}
 | |
| 
 | |
| 	if (p_option==NOTIFICATION_PROCESS) {
 | |
| 
 | |
| 		if (baker->is_baking() && !baker->is_paused()) {
 | |
| 
 | |
| 			update_timeout-=get_process_delta_time();
 | |
| 			if (update_timeout<0) {
 | |
| 
 | |
| 				if (baker->get_baked_light()!=node->get_baked_light()) {
 | |
| 					_end_baking();
 | |
| 					return;
 | |
| 				}
 | |
| 
 | |
| 				uint64_t t = OS::get_singleton()->get_ticks_msec();
 | |
| 
 | |
| #ifdef DEBUG_CUBES
 | |
| 				double norm =  baker->get_normalization();
 | |
| 				float max_lum=0;
 | |
| 
 | |
| 				{
 | |
| 					DVector<Color>::Write cw=colors.write();
 | |
| 					BakedLightBaker::Octant *octants=baker->octant_pool.ptr();
 | |
| 					BakedLightBaker::Octant *oct = &octants[baker->leaf_list];
 | |
| 					int vert_idx=0;
 | |
| 
 | |
| 					while(oct) {
 | |
| 
 | |
| 
 | |
| 
 | |
| 						Color colors[8];
 | |
| 						for(int i=0;i<8;i++) {
 | |
| 
 | |
| 							colors[i].r=oct->light_accum[i][0]/norm;
 | |
| 							colors[i].g=oct->light_accum[i][1]/norm;
 | |
| 							colors[i].b=oct->light_accum[i][2]/norm;
 | |
| 
 | |
| 							float lum = colors[i].get_v();
 | |
| 							//if (lum<0.05)
 | |
| 							//	color.a=0;
 | |
| 							if (lum>max_lum)
 | |
| 								max_lum=lum;
 | |
| 
 | |
| 						}
 | |
| 						static const int vert2cub[36]={7,3,1,1,5,7,7,6,2,2,3,7,7,5,4,4,6,7,2,6,4,4,0,2,4,5,1,1,0,4,1,3,2,2,0,1};
 | |
| 						for (int i=0;i<36;i++) {
 | |
| 
 | |
| 
 | |
| 							cw[vert_idx++]=colors[vert2cub[i]];
 | |
| 						}
 | |
| 
 | |
| 						if (oct->next_leaf)
 | |
| 							oct=&octants[oct->next_leaf];
 | |
| 						else
 | |
| 							oct=NULL;
 | |
| 
 | |
| 					}
 | |
| 				}
 | |
| 				print_line("MSCOL: "+itos(OS::get_singleton()->get_ticks_msec()-t));
 | |
| 				t = OS::get_singleton()->get_ticks_msec();
 | |
| 
 | |
| 				Array a;
 | |
| 				a.resize(Mesh::ARRAY_MAX);
 | |
| 				a[Mesh::ARRAY_VERTEX]=vertices;
 | |
| 				a[Mesh::ARRAY_COLOR]=colors;
 | |
| 				while(mesh->get_surface_count())
 | |
| 					mesh->surface_remove(0);
 | |
| 				mesh->add_surface(Mesh::PRIMITIVE_TRIANGLES,a);
 | |
| 				mesh->surface_set_material(0,material);
 | |
| #endif
 | |
| 				ERR_FAIL_COND(node->get_baked_light().is_null());
 | |
| 
 | |
| 				baker->update_octree_images(octree_texture,light_texture);
 | |
| 				baker->update_octree_sampler(octree_sampler);
 | |
| 			//	print_line("sampler size: "+itos(octree_sampler.size()*4));
 | |
| 
 | |
| #if 1
 | |
| //debug
 | |
| 				Image img(baker->baked_octree_texture_w,baker->baked_octree_texture_h,0,Image::FORMAT_RGBA,octree_texture);
 | |
| 				Ref<ImageTexture> it = memnew( ImageTexture );
 | |
| 				it->create_from_image(img);
 | |
| 				ResourceSaver::save("baked_octree.png",it);
 | |
| 
 | |
| 
 | |
| #endif
 | |
| 
 | |
| 
 | |
| 				uint64_t rays_snap = baker->get_rays_thrown();
 | |
| 				int rays_sec = (rays_snap-last_rays_time)*1.0-(update_timeout);
 | |
| 				last_rays_time=rays_snap;
 | |
| 
 | |
| 				bake_info->set_text("rays/s: "+itos(rays_sec));
 | |
| 				update_timeout=1;
 | |
| 				print_line("MSUPDATE: "+itos(OS::get_singleton()->get_ticks_msec()-t));
 | |
| 				t=OS::get_singleton()->get_ticks_msec();
 | |
| 				node->get_baked_light()->set_octree(octree_texture);
 | |
| 				node->get_baked_light()->set_light(light_texture);
 | |
| 				node->get_baked_light()->set_sampler_octree(octree_sampler);
 | |
| 				node->get_baked_light()->set_edited(true);
 | |
| 
 | |
| 				print_line("MSSET: "+itos(OS::get_singleton()->get_ticks_msec()-t));
 | |
| 
 | |
| 
 | |
| 
 | |
| 			}
 | |
| 		}
 | |
| 	}
 | |
| }
 | |
| 
 | |
| 
 | |
| void BakedLightEditor::_menu_option(int p_option) {
 | |
| 
 | |
| 
 | |
| 	switch(p_option) {
 | |
| 
 | |
| 
 | |
| 		case MENU_OPTION_BAKE: {
 | |
| 
 | |
| 			ERR_FAIL_COND(!node);
 | |
| 			ERR_FAIL_COND(node->get_baked_light().is_null());
 | |
| 			baker->bake(node->get_baked_light(),node);
 | |
| 			node->get_baked_light()->set_mode(BakedLight::MODE_OCTREE);
 | |
| 			update_timeout=0;
 | |
| 			set_process(true);
 | |
| 
 | |
| 
 | |
| 		} break;
 | |
| 		case MENU_OPTION_CLEAR: {
 | |
| 
 | |
| 
 | |
| 
 | |
| 		} break;
 | |
| 
 | |
| 	}
 | |
| }
 | |
| 
 | |
| void BakedLightEditor::_bake_pressed() {
 | |
| 
 | |
| 	ERR_FAIL_COND(!node);
 | |
| 	if (node->get_baked_light().is_null()) {
 | |
| 		err_dialog->set_text(TTR("BakedLightInstance does not contain a BakedLight resource."));
 | |
| 		err_dialog->popup_centered_minsize();
 | |
| 		button_bake->set_pressed(false);
 | |
| 		return;
 | |
| 	}
 | |
| 
 | |
| 	if (baker->is_baking()) {
 | |
| 
 | |
| 		baker->set_pause(!button_bake->is_pressed());
 | |
| 		if (baker->is_paused()) {
 | |
| 
 | |
| 			set_process(false);
 | |
| 			bake_info->set_text("");
 | |
| 			button_reset->show();
 | |
| 			button_make_lightmaps->show();
 | |
| 
 | |
| 		} else {
 | |
| 
 | |
| 			update_timeout=0;
 | |
| 			set_process(true);
 | |
| 			button_make_lightmaps->hide();
 | |
| 			button_reset->hide();
 | |
| 		}
 | |
| 	} else {
 | |
| 		baker->bake(node->get_baked_light(),node);
 | |
| 		node->get_baked_light()->set_mode(BakedLight::MODE_OCTREE);
 | |
| 		update_timeout=0;
 | |
| 
 | |
| 		last_rays_time=0;
 | |
| 
 | |
| 		set_process(true);
 | |
| 	}
 | |
| 
 | |
| }
 | |
| 
 | |
| void BakedLightEditor::_clear_pressed(){
 | |
| 
 | |
| 	baker->clear();
 | |
| 	button_bake->set_pressed(false);
 | |
| 	bake_info->set_text("");
 | |
| 
 | |
| }
 | |
| 
 | |
| void BakedLightEditor::edit(BakedLightInstance *p_baked_light) {
 | |
| 
 | |
| 	if (p_baked_light==NULL || node==p_baked_light) {
 | |
| 		return;
 | |
| 	}
 | |
| 	if (node && node!=p_baked_light)
 | |
| 		_end_baking();
 | |
| 
 | |
| 
 | |
| 	node=p_baked_light;
 | |
| 	//_end_baking();
 | |
| 
 | |
| }
 | |
| 
 | |
| void BakedLightEditor::_bake_lightmaps() {
 | |
| 
 | |
| 	Error err = baker->transfer_to_lightmaps();
 | |
| 	if (err) {
 | |
| 
 | |
| 		err_dialog->set_text("Error baking to lightmaps!\nMake sure that a bake has just\n happened and that lightmaps are\n configured. ");
 | |
| 		err_dialog->popup_centered_minsize();
 | |
| 		return;
 | |
| 	}
 | |
| 
 | |
| 	node->get_baked_light()->set_mode(BakedLight::MODE_LIGHTMAPS);
 | |
| 
 | |
| 
 | |
| }
 | |
| 
 | |
| void BakedLightEditor::_bind_methods() {
 | |
| 
 | |
| 	ObjectTypeDB::bind_method("_menu_option",&BakedLightEditor::_menu_option);
 | |
| 	ObjectTypeDB::bind_method("_bake_pressed",&BakedLightEditor::_bake_pressed);
 | |
| 	ObjectTypeDB::bind_method("_clear_pressed",&BakedLightEditor::_clear_pressed);
 | |
| 	ObjectTypeDB::bind_method("_bake_lightmaps",&BakedLightEditor::_bake_lightmaps);
 | |
| }
 | |
| 
 | |
| BakedLightEditor::BakedLightEditor() {
 | |
| 
 | |
| 
 | |
| 	bake_hbox = memnew( HBoxContainer );
 | |
| 	button_bake = memnew( ToolButton );
 | |
| 	button_bake->set_text(TTR("Bake!"));
 | |
| 	button_bake->set_toggle_mode(true);
 | |
| 	button_reset = memnew( Button );
 | |
| 	button_make_lightmaps = memnew( Button );
 | |
| 	button_bake->set_tooltip("Start/Unpause the baking process.\nThis bakes lighting into the lightmap octree.");
 | |
| 	button_make_lightmaps ->set_tooltip("Convert the lightmap octree to lightmap textures\n(must have set up UV/Lightmaps properly before!).");
 | |
| 
 | |
| 
 | |
| 	bake_info = memnew( Label );
 | |
| 	bake_hbox->add_child( button_bake );
 | |
| 	bake_hbox->add_child( button_reset );
 | |
| 	bake_hbox->add_child( bake_info );
 | |
| 
 | |
| 	err_dialog = memnew( AcceptDialog );
 | |
| 	add_child(err_dialog);
 | |
| 	node=NULL;
 | |
| 	baker = memnew( BakedLightBaker );
 | |
| 
 | |
| 	bake_hbox->add_child(button_make_lightmaps);
 | |
| 	button_make_lightmaps->hide();
 | |
| 
 | |
| 	button_bake->connect("pressed",this,"_bake_pressed");
 | |
| 	button_reset->connect("pressed",this,"_clear_pressed");
 | |
| 	button_make_lightmaps->connect("pressed",this,"_bake_lightmaps");
 | |
| 	button_reset->hide();
 | |
| 	button_reset->set_tooltip(TTR("Reset the lightmap octree baking process (start over)."));
 | |
| 
 | |
| 
 | |
| 	update_timeout=0;
 | |
| 
 | |
| 
 | |
| 
 | |
| }
 | |
| 
 | |
| BakedLightEditor::~BakedLightEditor() {
 | |
| 
 | |
| 	memdelete(baker);
 | |
| }
 | |
| 
 | |
| void BakedLightEditorPlugin::edit(Object *p_object) {
 | |
| 
 | |
| 	baked_light_editor->edit(p_object->cast_to<BakedLightInstance>());
 | |
| }
 | |
| 
 | |
| bool BakedLightEditorPlugin::handles(Object *p_object) const {
 | |
| 
 | |
| 	return p_object->is_type("BakedLightInstance");
 | |
| }
 | |
| 
 | |
| void BakedLightEditorPlugin::make_visible(bool p_visible) {
 | |
| 
 | |
| 	if (p_visible) {
 | |
| 		baked_light_editor->show();
 | |
| 		baked_light_editor->bake_hbox->show();
 | |
| 	} else {
 | |
| 
 | |
| 		baked_light_editor->hide();
 | |
| 		baked_light_editor->bake_hbox->hide();
 | |
| 		baked_light_editor->edit(NULL);
 | |
| 	}
 | |
| 
 | |
| }
 | |
| 
 | |
| BakedLightEditorPlugin::BakedLightEditorPlugin(EditorNode *p_node) {
 | |
| 
 | |
| 	editor=p_node;
 | |
| 	baked_light_editor = memnew( BakedLightEditor );
 | |
| 	editor->get_viewport()->add_child(baked_light_editor);
 | |
| 	add_control_to_container(CONTAINER_SPATIAL_EDITOR_MENU,baked_light_editor->bake_hbox);
 | |
| 	baked_light_editor->hide();
 | |
| 	baked_light_editor->bake_hbox->hide();
 | |
| }
 | |
| 
 | |
| 
 | |
| BakedLightEditorPlugin::~BakedLightEditorPlugin()
 | |
| {
 | |
| }
 | |
| 
 | |
| 
 | 
