/*************************************************************************/ /* baked_light.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 "baked_light.h" #include "servers/visual_server.h" void BakedLight::set_mode(Mode p_mode) { mode = p_mode; VS::get_singleton()->baked_light_set_mode(baked_light, (VS::BakedLightMode(p_mode))); } BakedLight::Mode BakedLight::get_mode() const { return mode; } void BakedLight::set_octree(const DVector &p_octree) { VS::get_singleton()->baked_light_set_octree(baked_light, p_octree); } DVector BakedLight::get_octree() const { return VS::get_singleton()->baked_light_get_octree(baked_light); } void BakedLight::set_light(const DVector &p_light) { VS::get_singleton()->baked_light_set_light(baked_light, p_light); } DVector BakedLight::get_light() const { return VS::get_singleton()->baked_light_get_light(baked_light); } void BakedLight::set_sampler_octree(const DVector &p_sampler_octree) { VS::get_singleton()->baked_light_set_sampler_octree(baked_light, p_sampler_octree); } DVector BakedLight::get_sampler_octree() const { return VS::get_singleton()->baked_light_get_sampler_octree(baked_light); } void BakedLight::add_lightmap(const Ref &p_texture, Size2 p_gen_size) { LightMap lm; lm.texture = p_texture; lm.gen_size = p_gen_size; lightmaps.push_back(lm); _update_lightmaps(); _change_notify(); } void BakedLight::set_lightmap_gen_size(int p_idx, const Size2 &p_size) { ERR_FAIL_INDEX(p_idx, lightmaps.size()); lightmaps[p_idx].gen_size = p_size; _update_lightmaps(); } Size2 BakedLight::get_lightmap_gen_size(int p_idx) const { ERR_FAIL_INDEX_V(p_idx, lightmaps.size(), Size2()); return lightmaps[p_idx].gen_size; } void BakedLight::set_lightmap_texture(int p_idx, const Ref &p_texture) { ERR_FAIL_INDEX(p_idx, lightmaps.size()); lightmaps[p_idx].texture = p_texture; _update_lightmaps(); } Ref BakedLight::get_lightmap_texture(int p_idx) const { ERR_FAIL_INDEX_V(p_idx, lightmaps.size(), Ref()); return lightmaps[p_idx].texture; } void BakedLight::erase_lightmap(int p_idx) { ERR_FAIL_INDEX(p_idx, lightmaps.size()); lightmaps.remove(p_idx); _update_lightmaps(); _change_notify(); } int BakedLight::get_lightmaps_count() const { return lightmaps.size(); } void BakedLight::clear_lightmaps() { lightmaps.clear(); _update_lightmaps(); _change_notify(); } void BakedLight::_update_lightmaps() { VS::get_singleton()->baked_light_clear_lightmaps(baked_light); for (int i = 0; i < lightmaps.size(); i++) { RID tid; if (lightmaps[i].texture.is_valid()) tid = lightmaps[i].texture->get_rid(); VS::get_singleton()->baked_light_add_lightmap(baked_light, tid, i); } } RID BakedLight::get_rid() const { return baked_light; } Array BakedLight::_get_lightmap_data() const { Array ret; ret.resize(lightmaps.size() * 2); int idx = 0; for (int i = 0; i < lightmaps.size(); i++) { ret[idx++] = Size2(lightmaps[i].gen_size); ret[idx++] = lightmaps[i].texture; } return ret; } void BakedLight::_set_lightmap_data(Array p_array) { lightmaps.clear(); for (int i = 0; i < p_array.size(); i += 2) { Size2 size = p_array[i]; Ref tex = p_array[i + 1]; // ERR_CONTINUE(tex.is_null()); LightMap lm; lm.gen_size = size; lm.texture = tex; lightmaps.push_back(lm); } _update_lightmaps(); } void BakedLight::set_cell_subdivision(int p_subdiv) { cell_subdiv = p_subdiv; } int BakedLight::get_cell_subdivision() const { return cell_subdiv; } void BakedLight::set_initial_lattice_subdiv(int p_size) { lattice_subdiv = p_size; } int BakedLight::get_initial_lattice_subdiv() const { return lattice_subdiv; } void BakedLight::set_plot_size(float p_size) { plot_size = p_size; } float BakedLight::get_plot_size() const { return plot_size; } void BakedLight::set_bounces(int p_size) { bounces = p_size; } int BakedLight::get_bounces() const { return bounces; } void BakedLight::set_cell_extra_margin(float p_margin) { cell_extra_margin = p_margin; } float BakedLight::get_cell_extra_margin() const { return cell_extra_margin; } void BakedLight::set_edge_damp(float p_margin) { edge_damp = p_margin; } float BakedLight::get_edge_damp() const { return edge_damp; } void BakedLight::set_normal_damp(float p_margin) { normal_damp = p_margin; } float BakedLight::get_normal_damp() const { return normal_damp; } void BakedLight::set_tint(float p_margin) { tint = p_margin; } float BakedLight::get_tint() const { return tint; } void BakedLight::set_saturation(float p_margin) { saturation = p_margin; } float BakedLight::get_saturation() const { return saturation; } void BakedLight::set_ao_radius(float p_ao_radius) { ao_radius = p_ao_radius; } float BakedLight::get_ao_radius() const { return ao_radius; } void BakedLight::set_ao_strength(float p_ao_strength) { ao_strength = p_ao_strength; } float BakedLight::get_ao_strength() const { return ao_strength; } void BakedLight::set_realtime_color_enabled(const bool p_realtime_color_enabled) { VS::get_singleton()->baked_light_set_realtime_color_enabled(baked_light, p_realtime_color_enabled); } bool BakedLight::get_realtime_color_enabled() const { return VS::get_singleton()->baked_light_get_realtime_color_enabled(baked_light); } void BakedLight::set_realtime_color(const Color &p_realtime_color) { VS::get_singleton()->baked_light_set_realtime_color(baked_light, p_realtime_color); } Color BakedLight::get_realtime_color() const { return VS::get_singleton()->baked_light_get_realtime_color(baked_light); } void BakedLight::set_realtime_energy(const float p_realtime_energy) { VS::get_singleton()->baked_light_set_realtime_energy(baked_light, p_realtime_energy); } float BakedLight::get_realtime_energy() const { return VS::get_singleton()->baked_light_get_realtime_energy(baked_light); } void BakedLight::set_energy_multiplier(float p_multiplier) { energy_multiply = p_multiplier; } float BakedLight::get_energy_multiplier() const { return energy_multiply; } void BakedLight::set_gamma_adjust(float p_adjust) { gamma_adjust = p_adjust; } float BakedLight::get_gamma_adjust() const { return gamma_adjust; } void BakedLight::set_bake_flag(BakeFlags p_flags, bool p_enable) { flags[p_flags] = p_enable; } bool BakedLight::get_bake_flag(BakeFlags p_flags) const { return flags[p_flags]; } void BakedLight::set_format(Format p_format) { format = p_format; VS::get_singleton()->baked_light_set_lightmap_multiplier(baked_light, format == FORMAT_HDR8 ? 8.0 : 1.0); } BakedLight::Format BakedLight::get_format() const { return format; } void BakedLight::set_transfer_lightmaps_only_to_uv2(bool p_enable) { transfer_only_uv2 = p_enable; } bool BakedLight::get_transfer_lightmaps_only_to_uv2() const { return transfer_only_uv2; } bool BakedLight::_set(const StringName &p_name, const Variant &p_value) { String n = p_name; if (!n.begins_with("lightmap")) return false; int idx = n.get_slicec('/', 1).to_int(); ERR_FAIL_COND_V(idx < 0, false); ERR_FAIL_COND_V(idx > lightmaps.size(), false); String what = n.get_slicec('/', 2); Ref tex; Size2 gens; if (what == "texture") tex = p_value; else if (what == "gen_size") gens = p_value; if (idx == lightmaps.size()) { if (tex.is_valid() || gens != Size2()) add_lightmap(tex, gens); } else { if (tex.is_valid()) set_lightmap_texture(idx, tex); else if (gens != Size2()) set_lightmap_gen_size(idx, gens); } return true; } bool BakedLight::_get(const StringName &p_name, Variant &r_ret) const { String n = p_name; if (!n.begins_with("lightmap")) return false; int idx = n.get_slicec('/', 1).to_int(); ERR_FAIL_COND_V(idx < 0, false); ERR_FAIL_COND_V(idx > lightmaps.size(), false); String what = n.get_slicec('/', 2); if (what == "texture") { if (idx == lightmaps.size()) r_ret = Ref(); else r_ret = lightmaps[idx].texture; } else if (what == "gen_size") { if (idx == lightmaps.size()) r_ret = Size2(); else r_ret = Size2(lightmaps[idx].gen_size); } else return false; return true; } void BakedLight::_get_property_list(List *p_list) const { for (int i = 0; i <= lightmaps.size(); i++) { p_list->push_back(PropertyInfo(Variant::VECTOR2, "lightmaps/" + itos(i) + "/gen_size", PROPERTY_HINT_NONE, "", PROPERTY_USAGE_EDITOR)); p_list->push_back(PropertyInfo(Variant::OBJECT, "lightmaps/" + itos(i) + "/texture", PROPERTY_HINT_RESOURCE_TYPE, "Texture", PROPERTY_USAGE_EDITOR)); } } void BakedLight::_bind_methods() { ObjectTypeDB::bind_method(_MD("set_mode", "mode"), &BakedLight::set_mode); ObjectTypeDB::bind_method(_MD("get_mode"), &BakedLight::get_mode); ObjectTypeDB::bind_method(_MD("set_octree", "octree"), &BakedLight::set_octree); ObjectTypeDB::bind_method(_MD("get_octree"), &BakedLight::get_octree); ObjectTypeDB::bind_method(_MD("set_light", "light"), &BakedLight::set_light); ObjectTypeDB::bind_method(_MD("get_light"), &BakedLight::get_light); ObjectTypeDB::bind_method(_MD("set_sampler_octree", "sampler_octree"), &BakedLight::set_sampler_octree); ObjectTypeDB::bind_method(_MD("get_sampler_octree"), &BakedLight::get_sampler_octree); ObjectTypeDB::bind_method(_MD("add_lightmap", "texture:Texture", "gen_size"), &BakedLight::add_lightmap); ObjectTypeDB::bind_method(_MD("erase_lightmap", "id"), &BakedLight::erase_lightmap); ObjectTypeDB::bind_method(_MD("clear_lightmaps"), &BakedLight::clear_lightmaps); ObjectTypeDB::bind_method(_MD("_set_lightmap_data", "lightmap_data"), &BakedLight::_set_lightmap_data); ObjectTypeDB::bind_method(_MD("_get_lightmap_data"), &BakedLight::_get_lightmap_data); ObjectTypeDB::bind_method(_MD("set_cell_subdivision", "cell_subdivision"), &BakedLight::set_cell_subdivision); ObjectTypeDB::bind_method(_MD("get_cell_subdivision"), &BakedLight::get_cell_subdivision); ObjectTypeDB::bind_method(_MD("set_initial_lattice_subdiv", "cell_subdivision"), &BakedLight::set_initial_lattice_subdiv); ObjectTypeDB::bind_method(_MD("get_initial_lattice_subdiv", "cell_subdivision"), &BakedLight::get_initial_lattice_subdiv); ObjectTypeDB::bind_method(_MD("set_plot_size", "plot_size"), &BakedLight::set_plot_size); ObjectTypeDB::bind_method(_MD("get_plot_size"), &BakedLight::get_plot_size); ObjectTypeDB::bind_method(_MD("set_bounces", "bounces"), &BakedLight::set_bounces); ObjectTypeDB::bind_method(_MD("get_bounces"), &BakedLight::get_bounces); ObjectTypeDB::bind_method(_MD("set_cell_extra_margin", "cell_extra_margin"), &BakedLight::set_cell_extra_margin); ObjectTypeDB::bind_method(_MD("get_cell_extra_margin"), &BakedLight::get_cell_extra_margin); ObjectTypeDB::bind_method(_MD("set_edge_damp", "edge_damp"), &BakedLight::set_edge_damp); ObjectTypeDB::bind_method(_MD("get_edge_damp"), &BakedLight::get_edge_damp); ObjectTypeDB::bind_method(_MD("set_normal_damp", "normal_damp"), &BakedLight::set_normal_damp); ObjectTypeDB::bind_method(_MD("get_normal_damp"), &BakedLight::get_normal_damp); ObjectTypeDB::bind_method(_MD("set_tint", "tint"), &BakedLight::set_tint); ObjectTypeDB::bind_method(_MD("get_tint"), &BakedLight::get_tint); ObjectTypeDB::bind_method(_MD("set_saturation", "saturation"), &BakedLight::set_saturation); ObjectTypeDB::bind_method(_MD("get_saturation"), &BakedLight::get_saturation); ObjectTypeDB::bind_method(_MD("set_ao_radius", "ao_radius"), &BakedLight::set_ao_radius); ObjectTypeDB::bind_method(_MD("get_ao_radius"), &BakedLight::get_ao_radius); ObjectTypeDB::bind_method(_MD("set_ao_strength", "ao_strength"), &BakedLight::set_ao_strength); ObjectTypeDB::bind_method(_MD("get_ao_strength"), &BakedLight::get_ao_strength); ObjectTypeDB::bind_method(_MD("set_realtime_color_enabled", "enabled"), &BakedLight::set_realtime_color_enabled); ObjectTypeDB::bind_method(_MD("get_realtime_color_enabled"), &BakedLight::get_realtime_color_enabled); ObjectTypeDB::bind_method(_MD("set_realtime_color", "tint"), &BakedLight::set_realtime_color); ObjectTypeDB::bind_method(_MD("get_realtime_color"), &BakedLight::get_realtime_color); ObjectTypeDB::bind_method(_MD("set_realtime_energy", "energy"), &BakedLight::set_realtime_energy); ObjectTypeDB::bind_method(_MD("get_realtime_energy"), &BakedLight::get_realtime_energy); ObjectTypeDB::bind_method(_MD("set_format", "format"), &BakedLight::set_format); ObjectTypeDB::bind_method(_MD("get_format"), &BakedLight::get_format); ObjectTypeDB::bind_method(_MD("set_transfer_lightmaps_only_to_uv2", "enable"), &BakedLight::set_transfer_lightmaps_only_to_uv2); ObjectTypeDB::bind_method(_MD("get_transfer_lightmaps_only_to_uv2"), &BakedLight::get_transfer_lightmaps_only_to_uv2); ObjectTypeDB::bind_method(_MD("set_energy_multiplier", "energy_multiplier"), &BakedLight::set_energy_multiplier); ObjectTypeDB::bind_method(_MD("get_energy_multiplier"), &BakedLight::get_energy_multiplier); ObjectTypeDB::bind_method(_MD("set_gamma_adjust", "gamma_adjust"), &BakedLight::set_gamma_adjust); ObjectTypeDB::bind_method(_MD("get_gamma_adjust"), &BakedLight::get_gamma_adjust); ObjectTypeDB::bind_method(_MD("set_bake_flag", "flag", "enabled"), &BakedLight::set_bake_flag); ObjectTypeDB::bind_method(_MD("get_bake_flag", "flag"), &BakedLight::get_bake_flag); ADD_PROPERTY(PropertyInfo(Variant::INT, "mode/mode", PROPERTY_HINT_ENUM, "Octree,Lightmaps"), _SCS("set_mode"), _SCS("get_mode")); ADD_PROPERTY(PropertyInfo(Variant::INT, "baking/format", PROPERTY_HINT_ENUM, "RGB,HDR8,HDR16"), _SCS("set_format"), _SCS("get_format")); ADD_PROPERTY(PropertyInfo(Variant::INT, "baking/cell_subdiv", PROPERTY_HINT_RANGE, "4,14,1"), _SCS("set_cell_subdivision"), _SCS("get_cell_subdivision")); ADD_PROPERTY(PropertyInfo(Variant::INT, "baking/lattice_subdiv", PROPERTY_HINT_RANGE, "1,5,1"), _SCS("set_initial_lattice_subdiv"), _SCS("get_initial_lattice_subdiv")); ADD_PROPERTY(PropertyInfo(Variant::INT, "baking/light_bounces", PROPERTY_HINT_RANGE, "0,3,1"), _SCS("set_bounces"), _SCS("get_bounces")); ADD_PROPERTY(PropertyInfo(Variant::REAL, "baking/plot_size", PROPERTY_HINT_RANGE, "1.0,16.0,0.01"), _SCS("set_plot_size"), _SCS("get_plot_size")); ADD_PROPERTY(PropertyInfo(Variant::REAL, "baking/energy_mult", PROPERTY_HINT_RANGE, "0.01,4096.0,0.01"), _SCS("set_energy_multiplier"), _SCS("get_energy_multiplier")); ADD_PROPERTY(PropertyInfo(Variant::REAL, "baking/gamma_adjust", PROPERTY_HINT_EXP_EASING), _SCS("set_gamma_adjust"), _SCS("get_gamma_adjust")); ADD_PROPERTY(PropertyInfo(Variant::REAL, "baking/saturation", PROPERTY_HINT_RANGE, "0,8,0.01"), _SCS("set_saturation"), _SCS("get_saturation")); ADD_PROPERTYI(PropertyInfo(Variant::BOOL, "baking_flags/diffuse"), _SCS("set_bake_flag"), _SCS("get_bake_flag"), BAKE_DIFFUSE); ADD_PROPERTYI(PropertyInfo(Variant::BOOL, "baking_flags/specular"), _SCS("set_bake_flag"), _SCS("get_bake_flag"), BAKE_SPECULAR); ADD_PROPERTYI(PropertyInfo(Variant::BOOL, "baking_flags/translucent"), _SCS("set_bake_flag"), _SCS("get_bake_flag"), BAKE_TRANSLUCENT); ADD_PROPERTYI(PropertyInfo(Variant::BOOL, "baking_flags/conserve_energy"), _SCS("set_bake_flag"), _SCS("get_bake_flag"), BAKE_CONSERVE_ENERGY); ADD_PROPERTYI(PropertyInfo(Variant::BOOL, "baking_flags/linear_color"), _SCS("set_bake_flag"), _SCS("get_bake_flag"), BAKE_LINEAR_COLOR); ADD_PROPERTY(PropertyInfo(Variant::BOOL, "lightmap/use_only_uv2"), _SCS("set_transfer_lightmaps_only_to_uv2"), _SCS("get_transfer_lightmaps_only_to_uv2")); ADD_PROPERTY(PropertyInfo(Variant::RAW_ARRAY, "octree", PROPERTY_HINT_NONE, "", PROPERTY_USAGE_NOEDITOR), _SCS("set_octree"), _SCS("get_octree")); ADD_PROPERTY(PropertyInfo(Variant::RAW_ARRAY, "light", PROPERTY_HINT_NONE, "", PROPERTY_USAGE_NOEDITOR), _SCS("set_light"), _SCS("get_light")); ADD_PROPERTY(PropertyInfo(Variant::INT_ARRAY, "sampler_octree", PROPERTY_HINT_NONE, "", PROPERTY_USAGE_NOEDITOR), _SCS("set_sampler_octree"), _SCS("get_sampler_octree")); ADD_PROPERTY(PropertyInfo(Variant::ARRAY, "lightmaps", PROPERTY_HINT_NONE, "", PROPERTY_USAGE_NOEDITOR), _SCS("_set_lightmap_data"), _SCS("_get_lightmap_data")); ADD_PROPERTY(PropertyInfo(Variant::REAL, "advanced/cell_margin", PROPERTY_HINT_RANGE, "0.01,0.8,0.01"), _SCS("set_cell_extra_margin"), _SCS("get_cell_extra_margin")); ADD_PROPERTY(PropertyInfo(Variant::REAL, "advanced/edge_damp", PROPERTY_HINT_RANGE, "0.0,8.0,0.1"), _SCS("set_edge_damp"), _SCS("get_edge_damp")); ADD_PROPERTY(PropertyInfo(Variant::REAL, "advanced/normal_damp", PROPERTY_HINT_RANGE, "0.0,1.0,0.01"), _SCS("set_normal_damp"), _SCS("get_normal_damp")); ADD_PROPERTY(PropertyInfo(Variant::REAL, "advanced/light_tint", PROPERTY_HINT_RANGE, "0.0,1.0,0.01"), _SCS("set_tint"), _SCS("get_tint")); ADD_PROPERTY(PropertyInfo(Variant::REAL, "advanced/ao_radius", PROPERTY_HINT_RANGE, "0.0,16.0,0.01"), _SCS("set_ao_radius"), _SCS("get_ao_radius")); ADD_PROPERTY(PropertyInfo(Variant::REAL, "advanced/ao_strength", PROPERTY_HINT_RANGE, "0.0,1.0,0.01"), _SCS("set_ao_strength"), _SCS("get_ao_strength")); ADD_PROPERTY(PropertyInfo(Variant::BOOL, "realtime/enabled"), _SCS("set_realtime_color_enabled"), _SCS("get_realtime_color_enabled")); ADD_PROPERTY(PropertyInfo(Variant::COLOR, "realtime/color", PROPERTY_HINT_COLOR_NO_ALPHA), _SCS("set_realtime_color"), _SCS("get_realtime_color")); ADD_PROPERTY(PropertyInfo(Variant::REAL, "realtime/energy", PROPERTY_HINT_RANGE, "0.01,4096.0,0.01"), _SCS("set_realtime_energy"), _SCS("get_realtime_energy")); BIND_CONSTANT(MODE_OCTREE); BIND_CONSTANT(MODE_LIGHTMAPS); BIND_CONSTANT(BAKE_DIFFUSE); BIND_CONSTANT(BAKE_SPECULAR); BIND_CONSTANT(BAKE_TRANSLUCENT); BIND_CONSTANT(BAKE_CONSERVE_ENERGY); BIND_CONSTANT(BAKE_MAX); } BakedLight::BakedLight() { cell_subdiv = 8; lattice_subdiv = 4; plot_size = 2.5; bounces = 1; energy_multiply = 2.0; gamma_adjust = 0.7; cell_extra_margin = 0.05; edge_damp = 0.0; normal_damp = 0.0; saturation = 1; tint = 0.0; ao_radius = 2.5; ao_strength = 0.7; format = FORMAT_RGB; transfer_only_uv2 = false; flags[BAKE_DIFFUSE] = true; flags[BAKE_SPECULAR] = false; flags[BAKE_TRANSLUCENT] = true; flags[BAKE_CONSERVE_ENERGY] = false; flags[BAKE_LINEAR_COLOR] = false; mode = MODE_OCTREE; baked_light = VS::get_singleton()->baked_light_create(); } BakedLight::~BakedLight() { VS::get_singleton()->free(baked_light); }