Optimize minetest.get_(all)_craft_recipe(s)

Signed off by: ShadowNinja, kwolekr
This commit is contained in:
gregorycu 2015-01-27 00:46:55 +11:00 committed by kwolekr
parent 1e4fb80d46
commit 03e0dd33a8
3 changed files with 107 additions and 180 deletions

@ -27,6 +27,7 @@ with this program; if not, write to the Free Software Foundation, Inc.,
#include "gamedef.h" #include "gamedef.h"
#include "inventory.h" #include "inventory.h"
#include "util/serialize.h" #include "util/serialize.h"
#include "util/numeric.h"
#include "strfnd.h" #include "strfnd.h"
#include "exceptions.h" #include "exceptions.h"
@ -931,85 +932,30 @@ public:
} }
return false; return false;
} }
virtual bool getCraftRecipe(CraftInput &input, CraftOutput &output,
IGameDef *gamedef) const
{
CraftOutput tmpout;
tmpout.item = "";
tmpout.time = 0;
// If output item is empty, abort.
if(output.item.empty())
return false;
// Walk crafting definitions from back to front, so that later
// definitions can override earlier ones.
for(std::vector<CraftDefinition*>::const_reverse_iterator
i = m_craft_definitions.rbegin();
i != m_craft_definitions.rend(); i++)
{
CraftDefinition *def = *i;
/*infostream<<"Checking "<<input.dump()<<std::endl
<<" against "<<def->dump()<<std::endl;*/
try {
tmpout = def->getOutput(input, gamedef);
if((tmpout.item.substr(0,output.item.length()) == output.item) &&
((tmpout.item[output.item.length()] == 0) ||
(tmpout.item[output.item.length()] == ' ')))
{
// Get output, then decrement input (if requested)
input = def->getInput(output, gamedef);
return true;
}
}
catch(SerializationError &e)
{
errorstream<<"getCraftResult: ERROR: "
<<"Serialization error in recipe "
<<def->dump()<<std::endl;
// then go on with the next craft definition
}
}
return false;
}
virtual std::vector<CraftDefinition*> getCraftRecipes(CraftOutput &output, virtual std::vector<CraftDefinition*> getCraftRecipes(CraftOutput &output,
IGameDef *gamedef) const IGameDef *gamedef, unsigned limit=0) const
{ {
std::vector<CraftDefinition*> recipes_list; std::vector<CraftDefinition*> recipes;
CraftInput input;
CraftOutput tmpout; std::map<std::string, std::vector<CraftDefinition*> >::const_iterator
tmpout.item = ""; vec_iter = m_output_craft_definitions.find(output.item);
tmpout.time = 0;
if (vec_iter == m_output_craft_definitions.end())
return recipes;
const std::vector<CraftDefinition*> &vec = vec_iter->second;
recipes.reserve(limit ? MYMIN(limit, vec.size()) : vec.size());
for (std::vector<CraftDefinition*>::const_reverse_iterator for (std::vector<CraftDefinition*>::const_reverse_iterator
i = m_craft_definitions.rbegin(); it = vec.rbegin(); it != vec.rend(); ++it) {
i != m_craft_definitions.rend(); i++) if (limit && recipes.size() >= limit)
{ break;
CraftDefinition *def = *i; recipes.push_back(*it);
}
/*infostream<<"Checking "<<input.dump()<<std::endl return recipes;
<<" against "<<def->dump()<<std::endl;*/
try {
tmpout = def->getOutput(input, gamedef);
if(tmpout.item.substr(0,output.item.length()) == output.item)
{
// Get output, then decrement input (if requested)
input = def->getInput(output, gamedef);
recipes_list.push_back(*i);
}
}
catch(SerializationError &e)
{
errorstream<<"getCraftResult: ERROR: "
<<"Serialization error in recipe "
<<def->dump()<<std::endl;
// then go on with the next craft definition
}
}
return recipes_list;
} }
virtual std::string dump() const virtual std::string dump() const
{ {
@ -1023,11 +969,16 @@ public:
} }
return os.str(); return os.str();
} }
virtual void registerCraft(CraftDefinition *def) virtual void registerCraft(CraftDefinition *def, IGameDef *gamedef)
{ {
verbosestream<<"registerCraft: registering craft definition: " verbosestream<<"registerCraft: registering craft definition: "
<<def->dump()<<std::endl; <<def->dump()<<std::endl;
m_craft_definitions.push_back(def); m_craft_definitions.push_back(def);
CraftInput input;
std::string output_name = craftGetItemName(
def->getOutput(input, gamedef).item, gamedef);
m_output_craft_definitions[output_name].push_back(def);
} }
virtual void clear() virtual void clear()
{ {
@ -1037,6 +988,7 @@ public:
delete *i; delete *i;
} }
m_craft_definitions.clear(); m_craft_definitions.clear();
m_output_craft_definitions.clear();
} }
virtual void serialize(std::ostream &os) const virtual void serialize(std::ostream &os) const
{ {
@ -1053,7 +1005,7 @@ public:
os<<serializeString(tmp_os.str()); os<<serializeString(tmp_os.str());
} }
} }
virtual void deSerialize(std::istream &is) virtual void deSerialize(std::istream &is, IGameDef *gamedef)
{ {
// Clear everything // Clear everything
clear(); clear();
@ -1067,11 +1019,12 @@ public:
std::istringstream tmp_is(deSerializeString(is), std::ios::binary); std::istringstream tmp_is(deSerializeString(is), std::ios::binary);
CraftDefinition *def = CraftDefinition::deSerialize(tmp_is); CraftDefinition *def = CraftDefinition::deSerialize(tmp_is);
// Register // Register
registerCraft(def); registerCraft(def, gamedef);
} }
} }
private: private:
std::vector<CraftDefinition*> m_craft_definitions; std::vector<CraftDefinition*> m_craft_definitions;
std::map<std::string, std::vector<CraftDefinition*> > m_output_craft_definitions;
}; };
IWritableCraftDefManager* createCraftDefManager() IWritableCraftDefManager* createCraftDefManager()

@ -356,10 +356,8 @@ public:
// The main crafting function // The main crafting function
virtual bool getCraftResult(CraftInput &input, CraftOutput &output, virtual bool getCraftResult(CraftInput &input, CraftOutput &output,
bool decrementInput, IGameDef *gamedef) const=0; bool decrementInput, IGameDef *gamedef) const=0;
virtual bool getCraftRecipe(CraftInput &input, CraftOutput &output,
IGameDef *gamedef) const=0;
virtual std::vector<CraftDefinition*> getCraftRecipes(CraftOutput &output, virtual std::vector<CraftDefinition*> getCraftRecipes(CraftOutput &output,
IGameDef *gamedef) const=0; IGameDef *gamedef, unsigned limit=0) const=0;
// Print crafting recipes for debugging // Print crafting recipes for debugging
virtual std::string dump() const=0; virtual std::string dump() const=0;
@ -376,22 +374,20 @@ public:
// The main crafting function // The main crafting function
virtual bool getCraftResult(CraftInput &input, CraftOutput &output, virtual bool getCraftResult(CraftInput &input, CraftOutput &output,
bool decrementInput, IGameDef *gamedef) const=0; bool decrementInput, IGameDef *gamedef) const=0;
virtual bool getCraftRecipe(CraftInput &input, CraftOutput &output,
IGameDef *gamedef) const=0;
virtual std::vector<CraftDefinition*> getCraftRecipes(CraftOutput &output, virtual std::vector<CraftDefinition*> getCraftRecipes(CraftOutput &output,
IGameDef *gamedef) const=0; IGameDef *gamedef, unsigned limit=0) const=0;
// Print crafting recipes for debugging // Print crafting recipes for debugging
virtual std::string dump() const=0; virtual std::string dump() const=0;
// Add a crafting definition. // Add a crafting definition.
// After calling this, the pointer belongs to the manager. // After calling this, the pointer belongs to the manager.
virtual void registerCraft(CraftDefinition *def)=0; virtual void registerCraft(CraftDefinition *def, IGameDef *gamedef) = 0;
// Delete all crafting definitions // Delete all crafting definitions
virtual void clear()=0; virtual void clear()=0;
virtual void serialize(std::ostream &os) const=0; virtual void serialize(std::ostream &os) const=0;
virtual void deSerialize(std::istream &is)=0; virtual void deSerialize(std::istream &is, IGameDef *gamedef) = 0;
}; };
IWritableCraftDefManager* createCraftDefManager(); IWritableCraftDefManager* createCraftDefManager();

@ -173,7 +173,7 @@ int ModApiCraft::l_register_craft(lua_State *L)
CraftDefinition *def = new CraftDefinitionShaped( CraftDefinition *def = new CraftDefinitionShaped(
output, width, recipe, replacements); output, width, recipe, replacements);
craftdef->registerCraft(def); craftdef->registerCraft(def, getServer(L));
} }
/* /*
CraftDefinitionShapeless CraftDefinitionShapeless
@ -205,7 +205,7 @@ int ModApiCraft::l_register_craft(lua_State *L)
CraftDefinition *def = new CraftDefinitionShapeless( CraftDefinition *def = new CraftDefinitionShapeless(
output, recipe, replacements); output, recipe, replacements);
craftdef->registerCraft(def); craftdef->registerCraft(def, getServer(L));
} }
/* /*
CraftDefinitionToolRepair CraftDefinitionToolRepair
@ -216,7 +216,7 @@ int ModApiCraft::l_register_craft(lua_State *L)
CraftDefinition *def = new CraftDefinitionToolRepair( CraftDefinition *def = new CraftDefinitionToolRepair(
additional_wear); additional_wear);
craftdef->registerCraft(def); craftdef->registerCraft(def, getServer(L));
} }
/* /*
CraftDefinitionCooking CraftDefinitionCooking
@ -246,7 +246,7 @@ int ModApiCraft::l_register_craft(lua_State *L)
CraftDefinition *def = new CraftDefinitionCooking( CraftDefinition *def = new CraftDefinitionCooking(
output, recipe, cooktime, replacements); output, recipe, cooktime, replacements);
craftdef->registerCraft(def); craftdef->registerCraft(def, getServer(L));
} }
/* /*
CraftDefinitionFuel CraftDefinitionFuel
@ -270,7 +270,7 @@ int ModApiCraft::l_register_craft(lua_State *L)
CraftDefinition *def = new CraftDefinitionFuel( CraftDefinition *def = new CraftDefinitionFuel(
recipe, burntime, replacements); recipe, burntime, replacements);
craftdef->registerCraft(def); craftdef->registerCraft(def, getServer(L));
} }
else else
{ {
@ -326,88 +326,14 @@ int ModApiCraft::l_get_craft_result(lua_State *L)
return 2; return 2;
} }
// get_craft_recipe(result item)
int ModApiCraft::l_get_craft_recipe(lua_State *L) void push_craft_recipe(lua_State *L, IGameDef *gdef,
const CraftDefinition *recipe,
const CraftOutput &tmpout)
{ {
NO_MAP_LOCK_REQUIRED; CraftInput input = recipe->getInput(tmpout, gdef);
CraftOutput output = recipe->getOutput(input, gdef);
int k = 1;
int input_i = 1;
std::string o_item = luaL_checkstring(L,input_i);
IGameDef *gdef = getServer(L);
ICraftDefManager *cdef = gdef->cdef();
CraftInput input;
CraftOutput output(o_item,0);
bool got = cdef->getCraftRecipe(input, output, gdef);
lua_newtable(L); // output table
if(got){
lua_newtable(L);
for(std::vector<ItemStack>::const_iterator
i = input.items.begin();
i != input.items.end(); i++, k++)
{
if (i->empty())
{
continue;
}
lua_pushinteger(L,k);
lua_pushstring(L,i->name.c_str());
lua_settable(L, -3);
}
lua_setfield(L, -2, "items");
setintfield(L, -1, "width", input.width);
switch (input.method) {
case CRAFT_METHOD_NORMAL:
lua_pushstring(L,"normal");
break;
case CRAFT_METHOD_COOKING:
lua_pushstring(L,"cooking");
break;
case CRAFT_METHOD_FUEL:
lua_pushstring(L,"fuel");
break;
default:
lua_pushstring(L,"unknown");
}
lua_setfield(L, -2, "type");
} else {
lua_pushnil(L);
lua_setfield(L, -2, "items");
setintfield(L, -1, "width", 0);
}
return 1;
}
// get_all_craft_recipes(result item)
int ModApiCraft::l_get_all_craft_recipes(lua_State *L)
{
NO_MAP_LOCK_REQUIRED;
std::string o_item = luaL_checkstring(L,1);
IGameDef *gdef = getServer(L);
ICraftDefManager *cdef = gdef->cdef();
CraftInput input;
CraftOutput output(o_item,0);
std::vector<CraftDefinition*> recipes_list;
recipes_list = cdef->getCraftRecipes(output, gdef);
if (recipes_list.empty()) {
lua_pushnil(L);
return 1;
}
lua_createtable(L, recipes_list.size(), 0);
std::vector<CraftDefinition*>::const_iterator iter = recipes_list.begin();
for (u16 i = 0; iter != recipes_list.end(); iter++) {
CraftOutput tmpout;
tmpout.item = "";
tmpout.time = 0;
tmpout = (*iter)->getOutput(input, gdef);
std::string query = tmpout.item;
char *fmtpos, *fmt = &query[0];
if (strtok_r(fmt, " ", &fmtpos) == output.item) {
input = (*iter)->getInput(output, gdef);
lua_newtable(L);
lua_newtable(L); // items lua_newtable(L); // items
std::vector<ItemStack>::const_iterator iter = input.items.begin(); std::vector<ItemStack>::const_iterator iter = input.items.begin();
for (u16 j = 1; iter != input.items.end(); iter++, j++) { for (u16 j = 1; iter != input.items.end(); iter++, j++) {
@ -432,11 +358,63 @@ int ModApiCraft::l_get_all_craft_recipes(lua_State *L)
lua_pushstring(L, "unknown"); lua_pushstring(L, "unknown");
} }
lua_setfield(L, -2, "type"); lua_setfield(L, -2, "type");
lua_pushstring(L, &tmpout.item[0]); lua_pushstring(L, tmpout.item.c_str());
lua_setfield(L, -2, "output"); lua_setfield(L, -2, "output");
}
void push_craft_recipes(lua_State *L, IGameDef *gdef,
const std::vector<CraftDefinition*> &recipes,
const CraftOutput &output)
{
lua_createtable(L, recipes.size(), 0);
if (recipes.empty()) {
lua_pushnil(L);
return;
}
std::vector<CraftDefinition*>::const_iterator it = recipes.begin();
for (unsigned i = 0; it != recipes.end(); ++it) {
lua_newtable(L);
push_craft_recipe(L, gdef, *it, output);
lua_rawseti(L, -2, ++i); lua_rawseti(L, -2, ++i);
} }
} }
// get_craft_recipe(result item)
int ModApiCraft::l_get_craft_recipe(lua_State *L)
{
NO_MAP_LOCK_REQUIRED;
std::string item = luaL_checkstring(L, 1);
Server *server = getServer(L);
CraftOutput output(item, 0);
std::vector<CraftDefinition*> recipes = server->cdef()
->getCraftRecipes(output, server, 1);
if (recipes.empty()) {
lua_pushnil(L);
lua_setfield(L, -2, "items");
setintfield(L, -1, "width", 0);
return 1;
}
push_craft_recipe(L, server, recipes[0], output);
return 1;
}
// get_all_craft_recipes(result item)
int ModApiCraft::l_get_all_craft_recipes(lua_State *L)
{
NO_MAP_LOCK_REQUIRED;
std::string item = luaL_checkstring(L, 1);
Server *server = getServer(L);
CraftOutput output(item, 0);
std::vector<CraftDefinition*> recipes = server->cdef()
->getCraftRecipes(output, server);
push_craft_recipes(L, server, recipes, output);
return 1; return 1;
} }