333 lines
10 KiB
C++
333 lines
10 KiB
C++
/*************************************************************************/
|
|
/* baked_light_editor_plugin.cpp */
|
|
/*************************************************************************/
|
|
/* This file is part of: */
|
|
/* GODOT ENGINE */
|
|
/* https://godotengine.org */
|
|
/*************************************************************************/
|
|
/* Copyright (c) 2007-2018 Juan Linietsky, Ariel Manzur. */
|
|
/* Copyright (c) 2014-2018 Godot Engine contributors (cf. AUTHORS.md) */
|
|
/* */
|
|
/* 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 "baked_light_editor_plugin.h"
|
|
#include "io/marshalls.h"
|
|
#include "io/resource_saver.h"
|
|
#include "scene/3d/mesh_instance.h"
|
|
#include "scene/gui/box_container.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);
|
|
const String conf_warning = node->get_configuration_warning();
|
|
if (!conf_warning.empty()) {
|
|
err_dialog->set_text(conf_warning);
|
|
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;
|
|
button_bake->set_pressed(false);
|
|
|
|
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() {
|
|
}
|