|
| 1 | +#include "lua_func.hpp" |
| 2 | + |
| 3 | +#include <algorithm> |
| 4 | + |
| 5 | +Obj::Obj(lua_State *L) : L(L) { |
| 6 | + lua_getglobal(L, "obj"); |
| 7 | + obj_ref = luaL_ref(L, LUA_REGISTRYINDEX); |
| 8 | +} |
| 9 | + |
| 10 | +Obj::~Obj() { luaL_unref(L, LUA_REGISTRYINDEX, obj_ref); } |
| 11 | + |
| 12 | +Param |
| 13 | +Obj::get_param() { |
| 14 | + constexpr float e = 1.0e-4f; |
| 15 | + |
| 16 | + float shutter_angle = static_cast<float>(std::clamp(get_number(2, 180), 0.0, 360.0)); |
| 17 | + std::uint32_t render_smp_lim = static_cast<std::uint32_t>(std::max(get_integer(3, 256ll), 1ll)); |
| 18 | + std::uint32_t preview_smp_lim = static_cast<std::uint32_t>(std::max(get_integer(4, 0ll), 0ll)); |
| 19 | + std::uint32_t ext = static_cast<std::uint32_t>(std::clamp(get_integer(5, 2ll), 0ll, 2ll)); |
| 20 | + bool resize = get_boolean(6, true); |
| 21 | + std::uint32_t geo_cache = static_cast<std::uint32_t>(std::clamp(get_integer(7, 0ll), 0ll, 2ll)); |
| 22 | + std::uint32_t geo_ctrl = static_cast<std::uint32_t>(std::clamp(get_integer(8, 0ll), 0ll, 3ll)); |
| 23 | + float mix = static_cast<float>(std::clamp(get_number(9, 0.0) * 0.01, 0.0, 1.0)); |
| 24 | + bool print_info = get_boolean(10, false); |
| 25 | + |
| 26 | + std::uint32_t smp_lim = (!preview_smp_lim || get_saving()) ? render_smp_lim : preview_smp_lim; |
| 27 | + bool is_valid = shutter_angle > e && smp_lim > 1u; |
| 28 | + |
| 29 | + return Param{get_string(1, "motion_blur"), |
| 30 | + shutter_angle, |
| 31 | + smp_lim, |
| 32 | + ext, |
| 33 | + resize, |
| 34 | + geo_cache, |
| 35 | + geo_ctrl, |
| 36 | + mix, |
| 37 | + print_info, |
| 38 | + is_valid}; |
| 39 | +} |
| 40 | + |
| 41 | +Input |
| 42 | +Obj::get_input(std::uint32_t ext) { |
| 43 | + constexpr std::array<const char *, 6> fields{"cx", "cy", "ox", "oy", "rz", "zoom"}; |
| 44 | + constexpr std::array<const char *, 6> tracks{"cx", "cy", "x", "y", "rz", "zoom"}; |
| 45 | + |
| 46 | + Input input{}; |
| 47 | + |
| 48 | + lua_rawgeti(L, LUA_REGISTRYINDEX, obj_ref); |
| 49 | + |
| 50 | + lua_getfield(L, -1, "getpixel"); |
| 51 | + lua_call(L, 0, 2); |
| 52 | + input.res = Vec2<float>(static_cast<float>(lua_tonumber(L, -2)), static_cast<float>(lua_tonumber(L, -1))); |
| 53 | + lua_pop(L, 2); |
| 54 | + |
| 55 | + input.obj_id = static_cast<std::size_t>(get_integer(11, 0ll)); |
| 56 | + lua_getfield(L, -1, "index"); |
| 57 | + lua_getfield(L, -2, "num"); |
| 58 | + input.obj_idx = static_cast<std::size_t>(lua_tointeger(L, -2)); |
| 59 | + input.obj_num = static_cast<std::size_t>(lua_tointeger(L, -1)); |
| 60 | + lua_pop(L, 2); |
| 61 | + |
| 62 | + lua_getfield(L, -1, "frame"); |
| 63 | + lua_getfield(L, -2, "totalframe"); |
| 64 | + lua_getfield(L, -3, "framerate"); |
| 65 | + input.frame = static_cast<std::size_t>(lua_tointeger(L, -3)); |
| 66 | + std::size_t total_frame = static_cast<std::size_t>(lua_tointeger(L, -2)); |
| 67 | + const double fps = lua_tonumber(L, -1); |
| 68 | + lua_pop(L, 3); |
| 69 | + |
| 70 | + input.is_last = {input.obj_idx == input.obj_num - 1, input.frame == total_frame - 1}; |
| 71 | + |
| 72 | + const double dt = 1.0 / fps; |
| 73 | + input.geo_curr.set_state(true, input.frame); |
| 74 | + |
| 75 | + for (std::size_t i = 0; i < 6; ++i) { |
| 76 | + lua_getfield(L, -1, fields[i]); |
| 77 | + lua_getfield(L, -2, "getvalue"); |
| 78 | + lua_pushstring(L, tracks[i]); |
| 79 | + lua_call(L, 1, 1); |
| 80 | + input.geo_curr[i] = static_cast<float>(lua_tonumber(L, -2)); |
| 81 | + input.tf_curr[i] = static_cast<float>(lua_tonumber(L, -1)); |
| 82 | + lua_pop(L, 2); |
| 83 | + } |
| 84 | + |
| 85 | + if (ext && !input.frame) { |
| 86 | + switch (ext) { |
| 87 | + case 1: |
| 88 | + for (std::size_t i = 0; i < 6; ++i) { |
| 89 | + lua_getfield(L, -1, "getvalue"); |
| 90 | + lua_pushstring(L, tracks[i]); |
| 91 | + lua_pushnumber(L, dt); |
| 92 | + lua_call(L, 2, 1); |
| 93 | + const double v1 = lua_tonumber(L, -1); |
| 94 | + input.tf_prev[i] = static_cast<float>(input.tf_curr[i] * 2.0 - v1); |
| 95 | + lua_pop(L, 1); |
| 96 | + } |
| 97 | + break; |
| 98 | + case 2: { |
| 99 | + const double dt2 = dt * 2.0; |
| 100 | + for (std::size_t i = 0; i < 6; ++i) { |
| 101 | + lua_getfield(L, -1, "getvalue"); |
| 102 | + lua_pushstring(L, tracks[i]); |
| 103 | + lua_pushnumber(L, dt); |
| 104 | + lua_call(L, 2, 1); |
| 105 | + lua_getfield(L, -2, "getvalue"); |
| 106 | + lua_pushstring(L, tracks[i]); |
| 107 | + lua_pushnumber(L, dt2); |
| 108 | + lua_call(L, 2, 1); |
| 109 | + const double v1 = lua_tonumber(L, -2); |
| 110 | + const double v2 = lua_tonumber(L, -1); |
| 111 | + input.tf_prev[i] = static_cast<float>(input.tf_curr[i] * 3.0 - v1 * 3.0 + v2); |
| 112 | + lua_pop(L, 2); |
| 113 | + } |
| 114 | + } break; |
| 115 | + default: |
| 116 | + input.tf_prev = input.tf_curr; |
| 117 | + break; |
| 118 | + } |
| 119 | + } else if (input.frame) { |
| 120 | + lua_getfield(L, -1, "time"); |
| 121 | + const double time_prev = lua_tonumber(L, -1) - dt; |
| 122 | + lua_pop(L, 1); |
| 123 | + |
| 124 | + for (std::size_t i = 0; i < 6; ++i) { |
| 125 | + lua_getfield(L, -1, "getvalue"); |
| 126 | + lua_pushstring(L, tracks[i]); |
| 127 | + lua_pushnumber(L, time_prev); |
| 128 | + lua_call(L, 2, 1); |
| 129 | + input.tf_prev[i] = static_cast<float>(lua_tonumber(L, -1)); |
| 130 | + lua_pop(L, 1); |
| 131 | + } |
| 132 | + } |
| 133 | + |
| 134 | + input.pivot = input.tf_curr.get_center() + input.geo_curr.get_center(); |
| 135 | + |
| 136 | + lua_pop(L, 1); |
| 137 | + return input; |
| 138 | +} |
| 139 | + |
| 140 | +bool |
| 141 | +Obj::get_saving() { |
| 142 | + lua_rawgeti(L, LUA_REGISTRYINDEX, obj_ref); |
| 143 | + lua_getfield(L, -1, "getinfo"); |
| 144 | + lua_pushstring(L, "saving"); |
| 145 | + lua_call(L, 1, 1); |
| 146 | + bool saving = lua_toboolean(L, -1); |
| 147 | + lua_pop(L, 2); |
| 148 | + return saving; |
| 149 | +} |
| 150 | + |
| 151 | +void |
| 152 | +Obj::resize(const std::array<Vec2<float>, 2> &margin, const Vec2<float> ¢er) { |
| 153 | + lua_rawgeti(L, LUA_REGISTRYINDEX, obj_ref); |
| 154 | + lua_getfield(L, -1, "effect"); |
| 155 | + lua_pushstring(L, "領域拡張"); |
| 156 | + lua_pushstring(L, "上"); |
| 157 | + lua_pushinteger(L, static_cast<lua_Integer>(margin[0].get_y())); |
| 158 | + lua_pushstring(L, "下"); |
| 159 | + lua_pushinteger(L, static_cast<lua_Integer>(margin[1].get_y())); |
| 160 | + lua_pushstring(L, "左"); |
| 161 | + lua_pushinteger(L, static_cast<lua_Integer>(margin[0].get_x())); |
| 162 | + lua_pushstring(L, "右"); |
| 163 | + lua_pushinteger(L, static_cast<lua_Integer>(margin[1].get_x())); |
| 164 | + lua_call(L, 9, 0); |
| 165 | + |
| 166 | + lua_pushnumber(L, center.get_x()); |
| 167 | + lua_setfield(L, -2, "cx"); |
| 168 | + lua_pushnumber(L, center.get_y()); |
| 169 | + lua_setfield(L, -2, "cy"); |
| 170 | + |
| 171 | + lua_pop(L, 1); |
| 172 | +} |
| 173 | + |
| 174 | +void |
| 175 | +Obj::pixel_shader(const std::string &name, const std::vector<float> &constants) { |
| 176 | + lua_rawgeti(L, LUA_REGISTRYINDEX, obj_ref); |
| 177 | + lua_getfield(L, -1, "pixelshader"); |
| 178 | + lua_pushstring(L, name.c_str()); |
| 179 | + lua_pushstring(L, "object"); |
| 180 | + lua_pushstring(L, "object"); |
| 181 | + |
| 182 | + lua_newtable(L); |
| 183 | + for (int i = 0; i < constants.size(); ++i) { |
| 184 | + lua_pushnumber(L, constants[i]); |
| 185 | + lua_rawseti(L, -2, i + 1); |
| 186 | + } |
| 187 | + |
| 188 | + lua_call(L, 4, 0); |
| 189 | + lua_pop(L, 1); |
| 190 | +} |
| 191 | + |
| 192 | +void |
| 193 | +Obj::print(const std::string &str) { |
| 194 | + lua_getglobal(L, "debug_print"); |
| 195 | + lua_pushstring(L, str.c_str()); |
| 196 | + lua_call(L, 1, 0); |
| 197 | +} |
0 commit comments