X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=nageru%2Fscene.cpp;h=2612de639961e8336d3eb4a6a8d8f8fb1f5f842a;hb=e4ebe67b6523038484df30d1d5fafc6055398acc;hp=6c3a0393b8e83d47c7ca6fd202c668f69bf4bf38;hpb=e9b906258a07b55ad62738552130df51dd872482;p=nageru diff --git a/nageru/scene.cpp b/nageru/scene.cpp index 6c3a039..2612de6 100644 --- a/nageru/scene.cpp +++ b/nageru/scene.cpp @@ -42,7 +42,9 @@ size_t Scene::compute_chain_number(bool is_main_chain) const { assert(chains.size() > 0); assert(chains.size() % 2 == 0); - size_t chain_number = compute_chain_number_for_block(blocks.size() - 1); + bitset<256> disabled = find_disabled_blocks(size_t(-1)); + + size_t chain_number = compute_chain_number_for_block(blocks.size() - 1, disabled); assert(chain_number < chains.size() / 2); if (is_main_chain) { chain_number += chains.size() / 2; @@ -50,20 +52,74 @@ size_t Scene::compute_chain_number(bool is_main_chain) const return chain_number; } -size_t Scene::compute_chain_number_for_block(size_t block_idx) const +size_t Scene::compute_chain_number_for_block(size_t block_idx, const bitset<256> &disabled) const { Block *block = blocks[block_idx]; size_t chain_number; + + size_t currently_chosen_alternative; + if (disabled.test(block_idx)) { + // It doesn't matter, so pick 0 as the canonical choice + // (this is the only one that is actually instantiated). + currently_chosen_alternative = 0; + } else { + currently_chosen_alternative = block->currently_chosen_alternative; + } + assert(currently_chosen_alternative < block->alternatives.size()); + if (block_idx == 0) { assert(block->cardinality_base == 1); - chain_number = block->currently_chosen_alternative; + chain_number = currently_chosen_alternative; } else { - chain_number = compute_chain_number_for_block(block_idx - 1) + block->cardinality_base * block->currently_chosen_alternative; + chain_number = compute_chain_number_for_block(block_idx - 1, disabled) + block->cardinality_base * currently_chosen_alternative; } - assert(block->currently_chosen_alternative < int(block->alternatives.size())); return chain_number; } +bitset<256> Scene::find_disabled_blocks(size_t chain_idx) const +{ + assert(blocks.size() < 256); + + bitset<256> ret; + find_disabled_blocks(chain_idx, blocks.size() - 1, /*currently_disabled=*/false, &ret); + return ret; +} + +void Scene::find_disabled_blocks(size_t chain_idx, size_t block_idx, bool currently_disabled, bitset<256> *disabled) const +{ + if (currently_disabled) { + disabled->set(block_idx); + } + Block *block = blocks[block_idx]; + EffectType chosen_type = block->alternatives[block->chosen_alternative(chain_idx)]->effect_type; + for (size_t input_idx = 0; input_idx < block->inputs.size(); ++input_idx) { + if (chosen_type == IDENTITY_EFFECT && input_idx > 0) { + // Multi-input effect that has been replaced by + // IdentityEffect, so every effect but the first are + // disabled and will not participate in the chain. + find_disabled_blocks(chain_idx, block->inputs[input_idx], /*currently_disabled=*/true, disabled); + } else { + // Just keep on recursing down. + find_disabled_blocks(chain_idx, block->inputs[input_idx], currently_disabled, disabled); + } + } +} + +bool Scene::is_noncanonical_chain(size_t chain_idx) const +{ + bitset<256> disabled = find_disabled_blocks(chain_idx); + if (disabled.none()) { + return false; + } + assert(blocks.size() < 256); + for (size_t block_idx = 0; block_idx < blocks.size(); ++block_idx) { + if (disabled.test(block_idx) && blocks[block_idx]->chosen_alternative(chain_idx) != 0) { + return true; + } + } + return false; +} + int Scene::add_input(lua_State* L) { assert(lua_gettop(L) == 1 || lua_gettop(L) == 2); @@ -109,6 +165,37 @@ int Scene::add_input(lua_State* L) return wrap_lua_existing_object_nonowned(L, "Block", block); } +void Scene::find_inputs_for_block(lua_State *L, Scene *scene, Block *block) +{ + if (lua_gettop(L) == 2) { + // Implicitly the last added effect. + assert(!scene->blocks.empty()); + block->inputs.push_back(scene->blocks.size() - 1); + return; + } + + for (int idx = 3; idx <= lua_gettop(L); ++idx) { + Block *input_block = nullptr; + if (luaL_testudata(L, idx, "Block")) { + input_block = *(Block **)luaL_checkudata(L, idx, "Block"); + } else { + EffectBlueprint *blueprint = *(EffectBlueprint **)luaL_checkudata(L, idx, "EffectBlueprint"); + + // Search through all the blocks to figure out which one contains this effect. + for (Block *block : scene->blocks) { + if (find(block->alternatives.begin(), block->alternatives.end(), blueprint) != block->alternatives.end()) { + input_block = block; + break; + } + } + if (input_block == nullptr) { + luaL_error(L, "Input effect in parameter #%d has not been added to this scene", idx - 1); + } + } + block->inputs.push_back(input_block->idx); + } +} + int Scene::add_effect(lua_State* L) { assert(lua_gettop(L) >= 2); @@ -130,33 +217,7 @@ int Scene::add_effect(lua_State* L) block->alternatives.push_back(blueprint); } - // Find the inputs. - if (lua_gettop(L) == 2) { - assert(!scene->blocks.empty()); - block->inputs.push_back(scene->blocks.size() - 1); - } else { - for (int idx = 3; idx <= lua_gettop(L); ++idx) { - Block *input_block = nullptr; - if (luaL_testudata(L, idx, "Block")) { - input_block = *(Block **)luaL_checkudata(L, idx, "Block"); - } else { - EffectBlueprint *blueprint = *(EffectBlueprint **)luaL_checkudata(L, idx, "EffectBlueprint"); - - // Search through all the blocks to figure out which one contains this effect. - for (Block *block : scene->blocks) { - if (find(block->alternatives.begin(), block->alternatives.end(), blueprint) != block->alternatives.end()) { - input_block = block; - break; - } - } - if (input_block == nullptr) { - luaL_error(L, "Input effect in parameter #%d has not been added to this scene", idx - 1); - } - } - block->inputs.push_back(input_block->idx); - } - } - + find_inputs_for_block(L, scene, block); scene->blocks.push_back(block); return wrap_lua_existing_object_nonowned(L, "Block", block); @@ -167,7 +228,6 @@ int Scene::add_optional_effect(lua_State* L) assert(lua_gettop(L) >= 2); Scene *scene = (Scene *)luaL_checkudata(L, 1, "Scene"); - // NOTE: We only support effects with a single parent, since that's what IdentityEffect does. Block *block = new Block; block->idx = scene->blocks.size(); @@ -177,7 +237,7 @@ int Scene::add_optional_effect(lua_State* L) // An IdentityEffect will be the alternative for when the effect is disabled. block->alternatives.push_back(new EffectBlueprint(IDENTITY_EFFECT)); - block->inputs.push_back(scene->blocks.size() - 1); + find_inputs_for_block(L, scene, block); scene->blocks.push_back(block); return wrap_lua_existing_object_nonowned(L, "Block", block); @@ -185,13 +245,21 @@ int Scene::add_optional_effect(lua_State* L) Effect *Scene::instantiate_effects(const Block *block, size_t chain_idx, Scene::Instantiation *instantiation) { + // Find the chosen alternative for this block in this instance. + EffectType chosen_type = block->alternatives[block->chosen_alternative(chain_idx)]->effect_type; + vector inputs; for (size_t input_idx : block->inputs) { inputs.push_back(instantiate_effects(blocks[input_idx], chain_idx, instantiation)); - } - // Find the chosen alternative for this block in this instance. - EffectType chosen_type = block->alternatives[block->chosen_alternative(chain_idx)]->effect_type; + // As a special case, we allow IdentityEffect to take only one input + // even if the other alternative (or alternatives) is multi-input. + // Thus, even if there are more than one inputs, instantiate only + // the first one. + if (chosen_type == IDENTITY_EFFECT) { + break; + } + } Effect *effect; switch (chosen_type) { @@ -252,7 +320,13 @@ int Scene::finalize(lua_State* L) } const size_t cardinality = base; - const size_t total_cardinality = cardinality * (only_one_mode ? 1 : 2); + size_t real_cardinality = 0; + for (size_t chain_idx = 0; chain_idx < cardinality; ++chain_idx) { + if (!scene->is_noncanonical_chain(chain_idx)) { + ++real_cardinality; + } + } + const size_t total_cardinality = real_cardinality * (only_one_mode ? 1 : 2); if (total_cardinality > 200) { print_warning(L, "The given Scene will instantiate %zu different versions. This will take a lot of time and RAM to compile; see if you could limit some options by e.g. locking the input type in some cases (by giving a fixed input to add_input()).\n", total_cardinality); @@ -261,7 +335,8 @@ int Scene::finalize(lua_State* L) Block *output_block = scene->blocks.back(); for (bool is_main_chain : { false, true }) { for (size_t chain_idx = 0; chain_idx < cardinality; ++chain_idx) { - if (only_one_mode && is_main_chain != chosen_mode) { + if ((only_one_mode && is_main_chain != chosen_mode) || + scene->is_noncanonical_chain(chain_idx)) { scene->chains.emplace_back(); continue; } @@ -531,6 +606,20 @@ int Block_enable(lua_State *L) return 0; } +int Block_enable_if(lua_State *L) +{ + assert(lua_gettop(L) == 2); + Block *block = *(Block **)luaL_checkudata(L, 1, "Block"); + + if (block->alternatives.size() != 2 || + block->alternatives[1]->effect_type != IDENTITY_EFFECT) { + luaL_error(L, "enable_if() called on something that wasn't added with add_optional_effect()"); + } + bool enabled = checkbool(L, 2); + block->currently_chosen_alternative = enabled ? 0 : 1; + return 0; +} + int Block_disable(lua_State *L) { assert(lua_gettop(L) == 1);