Function setMaterialMaskFilter

plugins/buildingplan/buildingplan.cpp:810–867

}

static int setMaterialMaskFilter(lua_State *L) {
    color_ostream *out = Lua::GetOutput(L);
    if (!out)
        out = &Core::getInstance().getConsole();
    df::building_type type = (df::building_type)luaL_checkint(L, 1);
    int16_t subtype = luaL_checkint(L, 2);
    int32_t custom = luaL_checkint(L, 3);
    int index = luaL_checkint(L, 4);
    DEBUG(control,*out).print(
            "entering setMaterialMaskFilter building_type={} subtype={} custom={} index={}\n",
            ENUM_AS_STR(type), subtype, custom, index);
    BuildingTypeKey key(type, subtype, custom);
    auto &filters = get_item_filters(*out, key).getItemFilters();
    if (index < 0 || filters.size() <= (size_t)index)
        return 0;
    uint32_t mask = 0;
    vector<string> cats;
    Lua::GetVector(L, cats, 5);
    for (auto &cat : cats) {
        if (cat == "stone")
            mask |= stone_cat.whole;
        else if (cat == "wood")
            mask |= wood_cat.whole;
        else if (cat == "metal")
            mask |= metal_cat.whole;
        else if (cat == "glass")
            mask |= glass_cat.whole;
        else if (cat == "gem")
            mask |= gem_cat.whole;
        else if (cat == "clay")
            mask |= clay_cat.whole;
        else if (cat == "cloth")
            mask |= cloth_cat.whole;
        else if (cat == "silk")
            mask |= silk_cat.whole;
        else if (cat == "yarn")
            mask |= yarn_cat.whole;
    }
    DEBUG(control,*out).print(
            "setting material mask filter for building_type={} subtype={} custom={} index={} to {:x}\n",
            ENUM_AS_STR(type), subtype, custom, index, mask);
    ItemFilter filter = filters[index];
    filter.setMaterialMask(mask);
    set<MaterialInfo> new_mats;
    if (mask) {
        // remove materials from the list that don't match the mask
        const auto &mats = filter.getMaterials();
        const df::dfhack_material_category mat_mask(mask);
        for (auto & mat : mats) {
            if (mat.matches(mat_mask))
                new_mats.emplace(mat);
        }
    }
    filter.setMaterials(new_mats);
    get_item_filters(*out, key).setItemFilter(*out, filter, index);
    Lua::CallLuaModuleFunction(*out, "plugins.buildingplan", "signal_reset");
    return 0;
}