Module:BaseDataModel: Difference between revisions
From Against the Storm Official Wiki
m (cleaning up lua documentation to prep for better inheritance) |
m (Prepping for better module inheritance and robustness. no functional changes yet) |
||
Line 8: | Line 8: | ||
---I. Provides a standard set of getters for building data and permits overrides from derived classes to specialize the building data, to allow for exceptions, etc. | ---I. Provides a standard set of getters for building data and permits overrides from derived classes to specialize the building data, to allow for exceptions, etc. | ||
--- | --- | ||
---II. Provides a standard set of getters for recipe queries. | ---II. Provides a standard set of getters for recipe queries. All queries return an array of pairs of building IDs and recipes, pulled straight from JSON. Data from those pairs needs to be retrieved with the third interface. | ||
--- | --- | ||
---III. Provides a standard set of getters for data within individual recipes, to allow derived classes to specialize based on how their data is structured. Recipe data should not be accessed directly because of these differences. | ---III. Provides a standard set of getters for data within individual recipes, to allow derived classes to specialize based on how their data is structured. Recipe data should not be accessed directly because of these differences. | ||
--- | |||
--- | --- | ||
---@field getID function building interface method | ---@field getID function building interface method | ||
Line 25: | Line 26: | ||
---@field getStorage function building interface method | ---@field getStorage function building interface method | ||
---@field isServiceBuilding function building interface method | ---@field isServiceBuilding function building interface method | ||
--- | |||
---@field _ --- --- --- | |||
--- | |||
---@field getIDsAndRecipesWhereProductID function query interface method | |||
---@field getIDsAndRecipesWhereBuildingID function query interface method | |||
---@field getIDsAndRecipesWhereIngredientID function query interface method | |||
---@field getIDsAndRecipesWhereProductIDAndBuildingID function query interface method | |||
---@field getIDsAndRecipesWhereProductIDAndIngredientID function query interface method | |||
---@field getIDsAndRecipesWhereIngredientIDAndBuildingID function query interface method | |||
--- | --- | ||
---@field _ --- --- --- | ---@field _ --- --- --- | ||
Line 58: | Line 68: | ||
---@protected | ---@protected | ||
---Protected data schema defining Json field names for accessing data. | ---Protected data schema defining Json field names for accessing data. | ||
BaseDataModel.schema = { | BaseDataModel.schema = { | ||
ID = "id", | ID = "id", | ||
Line 80: | Line 89: | ||
WORKPLACES = "workplaces", | WORKPLACES = "workplaces", | ||
RECIPES = "recipes", | |||
RECIPE = { | RECIPE = { | ||
GRADE = "grade", | GRADE = "grade", | ||
Line 102: | Line 112: | ||
---Secondary category, to be defined by derived classes. | ---Secondary category, to be defined by derived classes. | ||
BaseDataModel.schema.CATEGORY2 = "category2" | BaseDataModel.schema.CATEGORY2 = "category2" | ||
BaseDataModel.schema.IS_SERVICE = "isService" | |||
-- This section is deprecated. | -- This section is deprecated. These should be handled by derived classes, not here. | ||
local INDEX_RECIPE_GRADE_ALT = "gradeId" -- alt for service buildings | local INDEX_RECIPE_GRADE_ALT = "gradeId" -- alt for service buildings | ||
local INDEX_RECIPE_INGREDIENT_OPTION_AMOUNT = "amount" | local INDEX_RECIPE_INGREDIENT_OPTION_AMOUNT = "amount" | ||
local INDEX_RECIPE_INGREDIENT_OPTION_ID = "name" | local INDEX_RECIPE_INGREDIENT_OPTION_ID = "name" | ||
local INDEX_RECIPE_ING_SERVICE_GOODS = "goods" -- alt to ingredients for service buildings | local INDEX_RECIPE_ING_SERVICE_GOODS = "goods" -- alt to ingredients for service buildings | ||
local INDEX_RECIPE_PRO_DEPOSITS = "seekedDeposits" -- alt to product for huts and camps | local INDEX_RECIPE_PRO_DEPOSITS = "seekedDeposits" -- alt to product for huts and camps | ||
local INDEX_RECIPE_PRO_SERVICE = "servedNeed" -- alt to product for service buildings | local INDEX_RECIPE_PRO_SERVICE = "servedNeed" -- alt to product for service buildings | ||
local INDEX_RECIPE_PLANTING_TIME = "plantingTime" | local INDEX_RECIPE_PLANTING_TIME = "plantingTime" | ||
local INDEX_RECIPE_HARVESTING_TIME = "harvestingTime" | local INDEX_RECIPE_HARVESTING_TIME = "harvestingTime" | ||
Line 129: | Line 133: | ||
---Enum converts grade strings (keys) to numbers (values). | ---Enum converts grade strings (keys) to numbers (values). | ||
local LOOKUP_CONVERT_GRADE_TO_NUMBER = { | local LOOKUP_CONVERT_GRADE_TO_NUMBER = { | ||
["Grade0"] = 0, | ["Grade0"] = 0, | ||
Line 269: | Line 272: | ||
---@protected | ---@protected | ||
---Initializes the specified record. | |||
---Initializes the specified record | |||
--- | --- | ||
---**Likely overridden in derived classes.** | ---**Likely overridden in derived classes.** | ||
Line 276: | Line 278: | ||
---@param id buildingIDString | ---@param id buildingIDString | ||
function BaseDataModel:initializeRecord(id) | function BaseDataModel:initializeRecord(id) | ||
-- | |||
local building = findID(self, id) | |||
-- Initialize the extra values not in the data. | |||
building[self.schema.CATEGORY2] = "Building" | |||
building[self.schema.IS_SERVICE] = false | |||
end | end | ||
---@public | ---@public | ||
---Gets the ID associated with the specified name in this instance's data. | ---Gets the ID associated with the specified name in this instance's data. | ||
--- | --- | ||
Line 297: | Line 302: | ||
---@public | ---@public | ||
---Gets the category for the specified ID from this instance's data. | ---Gets the category for the specified ID from this instance's data. | ||
--- | --- | ||
Line 312: | Line 316: | ||
---@public | ---@public | ||
---Gets the secondary category for the specified ID from this instance's data. | ---Gets the secondary category for the specified ID from this instance's data. | ||
--- | --- | ||
Line 329: | Line 332: | ||
---@public | ---@public | ||
---Gets the city score for the specified ID from this instance's data. | ---Gets the city score for the specified ID from this instance's data. | ||
--- | --- | ||
Line 344: | Line 346: | ||
---@public | ---@public | ||
---Gets the construction goods for the specified ID from this instance's data. | ---Gets the construction goods for the specified ID from this instance's data. | ||
---Returns the array of construction goods, straight from the table, in subtables of IDs and amounts. | ---Returns the array of construction goods, straight from the table, in subtables of IDs and amounts. | ||
Line 362: | Line 363: | ||
---@public | ---@public | ||
---Gets the time required for construction for the specified ID from this instance's data. | ---Gets the time required for construction for the specified ID from this instance's data. | ||
--- | --- | ||
Line 377: | Line 377: | ||
---@public | ---@public | ||
---Gets the description for the specified ID from this instance's data. | ---Gets the description for the specified ID from this instance's data. | ||
--- | --- | ||
Line 392: | Line 391: | ||
---@public | ---@public | ||
---Gets the icon filename for the specified ID from this instance's data. | ---Gets the icon filename for the specified ID from this instance's data. | ||
--- | --- | ||
Line 409: | Line 407: | ||
---@public | ---@public | ||
---Checks whether the specified ID is movable from this instance's data. | ---Checks whether the specified ID is movable from this instance's data. | ||
--- | --- | ||
Line 424: | Line 421: | ||
---@public | ---@public | ||
---Gets the name for the specified ID from this instance's data. | ---Gets the name for the specified ID from this instance's data. | ||
--- | --- | ||
Line 439: | Line 435: | ||
---@public | ---@public | ||
---Gets the number of workplaces for the specified ID from this instance's data. | ---Gets the number of workplaces for the specified ID from this instance's data. | ||
--- | --- | ||
Line 454: | Line 449: | ||
---@public | ---@public | ||
---Gets the size for ths specified ID from this instance's data, expressed as an X-by-Y string. | ---Gets the size for ths specified ID from this instance's data, expressed as an X-by-Y string. | ||
--- | --- | ||
Line 469: | Line 463: | ||
---@public | ---@public | ||
---Gets the storage capacity for the specified ID from this instance's data. | ---Gets the storage capacity for the specified ID from this instance's data. | ||
--- | --- | ||
Line 490: | Line 483: | ||
---@public | ---@public | ||
---Checks whether the specified ID is a service-type building from this instance's data. | ---Checks whether the specified ID is a service-type building from this instance's data. | ||
--- | --- | ||
Line 513: | Line 505: | ||
--region Public building recipe query interface | --region Public building recipe query interface | ||
- | ---@public | ||
---Gets all this instance's buildings' recipes that have the specified product. Returns an array of pairs of building IDs and recipe data. | |||
-- | |||
- | |||
-- | |||
--- | --- | ||
--- | ---**Likely overridden by derived classes.** | ||
--- | --- | ||
---Benchmarking: ~0.0001 | ---Benchmarking: ~0.0001 seconds | ||
--- | --- | ||
---@param productID string the product | ---@param productID string the product | ||
---@return table array of pairs of | ---@return table array of pairs of building IDs and recipe data, or {} if none found | ||
function BaseDataModel:getIDsAndRecipesWhereProductID(productID) | function BaseDataModel:getIDsAndRecipesWhereProductID(productID) | ||
Line 558: | Line 522: | ||
local ret = {} | local ret = {} | ||
for id, building in pairs(self.dataTable) do | for id, building in pairs(self.dataTable) do | ||
for _, recipe in ipairs(building[ | for _, recipe in ipairs(building[self.schema.RECIPES]) do | ||
-- If there's a product subtable with an amount. | -- If there's a product subtable with an amount. | ||
if recipe[ | if recipe[self.schema.RECIPE.PRODUCTS] then | ||
if recipe[ | if recipe[self.schema.RECIPE.PRODUCTS][self.schema.RECIPE.PRODUCT.ID] == productID then | ||
-- Add as an buildingID-recipe pair | |||
table.insert(ret, { | table.insert(ret, { | ||
["buildingID"] = id, | ["buildingID"] = id, | ||
Line 572: | Line 537: | ||
elseif recipe[INDEX_RECIPE_PRO_DEPOSITS] then | elseif recipe[INDEX_RECIPE_PRO_DEPOSITS] then | ||
if recipe[INDEX_RECIPE_PRO_DEPOSITS] == productID then | if recipe[INDEX_RECIPE_PRO_DEPOSITS] == productID then | ||
-- Add as an buildingID-recipe pair | |||
table.insert(ret, { | table.insert(ret, { | ||
["buildingID"] = id, | ["buildingID"] = id, | ||
Line 581: | Line 547: | ||
elseif recipe[INDEX_RECIPE_PRO_SERVICE] then | elseif recipe[INDEX_RECIPE_PRO_SERVICE] then | ||
if recipe[INDEX_RECIPE_PRO_SERVICE] == productID then | if recipe[INDEX_RECIPE_PRO_SERVICE] == productID then | ||
-- Add as an buildingID-recipe pair | |||
table.insert(ret, { | table.insert(ret, { | ||
["buildingID"] = id, | ["buildingID"] = id, | ||
Line 592: | Line 559: | ||
end | end | ||
--- | ---@public | ||
--- | ---Gets all this instance's building's recipes from the specified building ID. Returns an array of pairs of building IDs and recipe data. | ||
--- | |||
---**Likely overridden by derived classes.** | |||
--- | --- | ||
---Benchmarking: ~0.0000 seconds | ---Benchmarking: ~0.0000 seconds | ||
--- | --- | ||
---@param buildingID buildingIDString | ---@param buildingID buildingIDString | ||
---@return table array of pairs of | ---@return table array of pairs of building IDs and recipe data, or {} if none found | ||
function BaseDataModel:getIDsAndRecipesWhereBuildingID(buildingID) | function BaseDataModel:getIDsAndRecipesWhereBuildingID(buildingID) | ||
Line 607: | Line 576: | ||
local ret = {} | local ret = {} | ||
for _, recipe in ipairs(building[ | for _, recipe in ipairs(building[self.schema.RECIPES]) do | ||
-- Add as an buildingID-recipe pair | |||
table.insert(ret, { | table.insert(ret, { | ||
["buildingID"] = buildingID, | ["buildingID"] = buildingID, | ||
Line 616: | Line 586: | ||
end | end | ||
--- | ---@public | ||
--- | ---Gets all this instance's buildings' recipes that use the specified ingredients. Returns an array of pairs of building IDs and recipe data. | ||
--- | |||
---**Likely overridden by derived classes.** | |||
--- | --- | ||
---Benchmarking: ~0.0003 seconds | ---Benchmarking: ~0.0003 seconds | ||
--- | --- | ||
---@param ingredientID string the ingredient | ---@param ingredientID string the ingredient | ||
---@return table array of pairs of | ---@return table array of pairs of building IDs and recipe data, or {} if none found | ||
function BaseDataModel:getIDsAndRecipesWhereIngredientID(ingredientID) | function BaseDataModel:getIDsAndRecipesWhereIngredientID(ingredientID) | ||
Line 631: | Line 603: | ||
local ret = {} | local ret = {} | ||
for id, building in pairs(self.dataTable) do | for id, building in pairs(self.dataTable) do | ||
for _, recipe in ipairs(building[ | for _, recipe in ipairs(building[self.schema.RECIPES]) do | ||
local ingredientsList = {} | local ingredientsList = {} | ||
-- Some recipes don't have ingredients; skip them obviously. | -- Some recipes don't have ingredients; skip them obviously. | ||
if recipe[ | if recipe[self.schema.RECIPE.INGREDIENTS] then | ||
ingredientsList = recipe[ | ingredientsList = recipe[self.schema.RECIPE.INGREDIENTS] | ||
elseif recipe[INDEX_RECIPE_ING_SERVICE_GOODS] then | elseif recipe[INDEX_RECIPE_ING_SERVICE_GOODS] then | ||
ingredientsList = { recipe[INDEX_RECIPE_ING_SERVICE_GOODS] } | ingredientsList = { recipe[INDEX_RECIPE_ING_SERVICE_GOODS] } | ||
Line 643: | Line 615: | ||
for _, ingredientSlot in ipairs(ingredientsList) do | for _, ingredientSlot in ipairs(ingredientsList) do | ||
for _, option in ipairs(ingredientSlot) do | for _, option in ipairs(ingredientSlot) do | ||
if option[ | if option[self.schema.RECIPE.INGREDIENT.OPTION.ID] == ingredientID then | ||
-- Add as an buildingID-recipe pair | |||
table.insert(ret, { | table.insert(ret, { | ||
["buildingID"] = id, | ["buildingID"] = id, | ||
Line 658: | Line 631: | ||
end | end | ||
--- | ---@public | ||
---Gets all this instance's specified building's recipes that produce the specified product. Returns an array of pairs of building IDs and recipe data. | |||
--- | |||
---**Likely overridden by derived classes.** | |||
--- | |||
---(Benchmarking: ~0.0000 seconds with largest data file) | |||
--- | |||
---@param productID string the product | |||
---@param buildingID buildingIDString | |||
---@return table array of pairs of building IDs and recipe data, or {} if none found | |||
function BaseDataModel:getIDsAndRecipesWhereProductIDAndBuildingID(productID, buildingID) | |||
local building = findID(self, buildingID) | |||
if not building then | |||
return {} | |||
end | |||
local ret = {} | |||
for _, recipe in ipairs(building[self.schema.RECIPES]) do | |||
-- If there's a product subtable with an amount. | |||
if recipe[self.schema.RECIPE.PRODUCTS] then | |||
if recipe[self.schema.RECIPE.PRODUCTS][self.schema.RECIPE.PRODUCT.ID] == productID then | |||
-- Add as an buildingID-recipe pair | |||
table.insert(ret, { | |||
["buildingID"] = buildingID, | |||
["recipe"] = recipe, | |||
}) | |||
-- Only one ever per building, so return now | |||
return ret | |||
end | |||
-- There could also be a seeked deposit. | |||
elseif recipe[INDEX_RECIPE_PRO_DEPOSITS] then | |||
if recipe[INDEX_RECIPE_PRO_DEPOSITS] == productID then | |||
-- Add as an buildingID-recipe pair | |||
table.insert(ret, { | |||
["buildingID"] = buildingID, | |||
["recipe"] = recipe, | |||
}) | |||
-- Only one ever per building, so return now | |||
return ret | |||
end | |||
-- Or a service provided. | |||
elseif recipe[INDEX_RECIPE_PRO_SERVICE] then | |||
if recipe[INDEX_RECIPE_PRO_SERVICE] == productID then | |||
-- Add as an buildingID-recipe pair | |||
table.insert(ret, { | |||
["buildingID"] = buildingID, | |||
["recipe"] = recipe, | |||
}) | |||
-- Only one ever per building, so return now | |||
return ret | |||
end | |||
end | |||
end | |||
return {} | |||
end | |||
---@public | |||
---Gets all this instance's buildings' recipes that result in the specified product and use the specified ingredient. Returns an array of pairs of building IDs and recipe data. | |||
--- | |||
---**Likely overridden by derived classes.** | |||
--- | |||
---@param productID string the product ID | ---@param productID string the product ID | ||
---@param ingredientID string the ingredient ID | ---@param ingredientID string the ingredient ID | ||
---@return table array of pairs of | ---@return table array of pairs of building IDs and recipe data, or {} if none found | ||
function BaseDataModel:getIDsAndRecipesWhereProductIDAndIngredientID(productID, ingredientID) | function BaseDataModel:getIDsAndRecipesWhereProductIDAndIngredientID(productID, ingredientID) | ||
Line 670: | Line 707: | ||
local ret = {} | local ret = {} | ||
for id, building in pairs(self.dataTable) do | for id, building in pairs(self.dataTable) do | ||
for _, recipe in ipairs(building[ | for _, recipe in ipairs(building[self.schema.RECIPES]) do | ||
-- If there's a product subtable with an amount. | -- If there's a product subtable with an amount. | ||
if recipe[ | if recipe[self.schema.RECIPE.PRODUCTS] then | ||
if recipe[ | if recipe[self.schema.RECIPE.PRODUCTS][self.schema.RECIPE.PRODUCT.ID] == productID then | ||
-- Products have ingredients lists. | -- Products have ingredients lists. | ||
if recipe[ | if recipe[self.schema.RECIPE.INGREDIENTS] then | ||
for _, ingredientSlot in ipairs(recipe[ | for _, ingredientSlot in ipairs(recipe[self.schema.RECIPE.INGREDIENTS]) do | ||
for _, option in ipairs(ingredientSlot) do | for _, option in ipairs(ingredientSlot) do | ||
if option[ | if option[self.schema.RECIPE.INGREDIENT.OPTION.ID] == ingredientID then | ||
-- Add as an buildingID-recipe pair | |||
table.insert(ret, { | table.insert(ret, { | ||
["buildingID"] = id, | ["buildingID"] = id, | ||
Line 703: | Line 741: | ||
for _, option in ipairs(ingredientSlot) do | for _, option in ipairs(ingredientSlot) do | ||
if option[INDEX_RECIPE_INGREDIENT_OPTION_ID] == ingredientID then | if option[INDEX_RECIPE_INGREDIENT_OPTION_ID] == ingredientID then | ||
-- Add as an buildingID-recipe pair | |||
table.insert(ret, { | table.insert(ret, { | ||
["buildingID"] = id, | ["buildingID"] = id, | ||
Line 718: | Line 757: | ||
end | end | ||
--- | ---@public | ||
--- | ---Gets all this instance's specified building's recipes that use the specified ingredient. Returns an array of pairs of building IDs and recipe data. | ||
--- | --- | ||
--- | ---**Likely overridden by derived classes.** | ||
--- | --- | ||
---(Benchmarking: ~0.0000 seconds with largest data file) | ---(Benchmarking: ~0.0000 seconds with largest data file) | ||
Line 780: | Line 766: | ||
---@param ingredientID string | ---@param ingredientID string | ||
---@param buildingID buildingIDString | ---@param buildingID buildingIDString | ||
---@return table array of pairs of | ---@return table array of pairs of building IDs and recipe data, or {} if none found | ||
function BaseDataModel:getIDsAndRecipesWhereIngredientIDAndBuildingID(ingredientID, buildingID) | function BaseDataModel:getIDsAndRecipesWhereIngredientIDAndBuildingID(ingredientID, buildingID) | ||
Line 789: | Line 775: | ||
local ret = {} | local ret = {} | ||
for _, recipe in ipairs(building[ | for _, recipe in ipairs(building[self.schema.RECIPES]) do | ||
local ingredientsList = {} | local ingredientsList = {} | ||
-- Some recipes don't have any ingredients; skip those obviously. | -- Some recipes don't have any ingredients; skip those obviously. | ||
if recipe[ | if recipe[self.schema.RECIPE.INGREDIENTS] then | ||
ingredientsList = recipe[ | ingredientsList = recipe[self.schema.RECIPE.INGREDIENTS] | ||
elseif recipe[INDEX_RECIPE_ING_SERVICE_GOODS] then | elseif recipe[INDEX_RECIPE_ING_SERVICE_GOODS] then | ||
ingredientsList = recipe[INDEX_RECIPE_ING_SERVICE_GOODS] | ingredientsList = recipe[INDEX_RECIPE_ING_SERVICE_GOODS] | ||
Line 801: | Line 787: | ||
for _, ingredientSlot in ipairs(ingredientsList) do | for _, ingredientSlot in ipairs(ingredientsList) do | ||
for _, option in ipairs(ingredientSlot) do | for _, option in ipairs(ingredientSlot) do | ||
if option[ | if option[self.schema.RECIPE.INGREDIENT.OPTION.ID] == ingredientID then | ||
-- Add as an buildingID-recipe pair | |||
table.insert(ret, { | table.insert(ret, { | ||
["buildingID"] = buildingID, | ["buildingID"] = buildingID, | ||
Line 821: | Line 808: | ||
---@public | ---@public | ||
---Gets the building ID from the provided recipe data. | ---Gets the building ID from the provided recipe data. | ||
Line 827: | Line 813: | ||
---@return buildingIDString | ---@return buildingIDString | ||
function BaseDataModel:getRecipeBuildingID(recipeData) | function BaseDataModel:getRecipeBuildingID(recipeData) | ||
return recipeData.buildingID | return recipeData.buildingID | ||
end | end | ||
---@public | ---@public | ||
---Gets the efficiency grade from the provided recipe. | ---Gets the efficiency grade from the provided recipe. | ||
Line 838: | Line 824: | ||
function BaseDataModel:getRecipeGrade(recipeData) | function BaseDataModel:getRecipeGrade(recipeData) | ||
if recipeData.recipe[ | if recipeData.recipe[self.schema.RECIPE.GRADE] then | ||
return LOOKUP_CONVERT_GRADE_TO_NUMBER[recipeData.recipe[ | return LOOKUP_CONVERT_GRADE_TO_NUMBER[recipeData.recipe[self.schema.RECIPE.GRADE]] | ||
elseif recipeData.recipe[INDEX_RECIPE_GRADE_ALT] then | elseif recipeData.recipe[INDEX_RECIPE_GRADE_ALT] then | ||
Line 851: | Line 837: | ||
---@public | ---@public | ||
---Gets the production time from the provided recipe. | ---Gets the production time from the provided recipe. | ||
--- | --- | ||
Line 860: | Line 845: | ||
function BaseDataModel:getRecipeTime(recipeData) | function BaseDataModel:getRecipeTime(recipeData) | ||
if recipeData.recipe[ | if recipeData.recipe[self.schema.RECIPE.PRODUCTION_TIME] then | ||
return recipeData.recipe[ | return recipeData.recipe[self.schema.RECIPE.PRODUCTION_TIME] | ||
elseif recipeData.recipe[INDEX_RECIPE_PLANTING_TIME] and recipeData.recipe[INDEX_RECIPE_HARVESTING_TIME] then | elseif recipeData.recipe[INDEX_RECIPE_PLANTING_TIME] and recipeData.recipe[INDEX_RECIPE_HARVESTING_TIME] then | ||
Line 877: | Line 862: | ||
---@public | ---@public | ||
---Checks whether the provided recipe results in a service. | |||
---Checks whether the provided recipe results in a | |||
--- | --- | ||
---@param recipeData recipePairTable | ---@param recipeData recipePairTable | ||
Line 884: | Line 868: | ||
function BaseDataModel:isRecipeProvidingService(recipeData) | function BaseDataModel:isRecipeProvidingService(recipeData) | ||
-- This is in the base data model a property of the whole building, not a specific recipe. | |||
return | local building = findID(self, recipeData.buildingID) | ||
if building then | |||
return building[self.schema.IS_SERVICE] | |||
else | else | ||
return | return nil | ||
end | end | ||
end | end | ||
---@public | ---@public | ||
---Gets the product ID from the provided recipe. | ---Gets the product ID from the provided recipe. | ||
--- | --- | ||
Line 901: | Line 887: | ||
function BaseDataModel:getRecipeProductID(recipeData) | function BaseDataModel:getRecipeProductID(recipeData) | ||
if recipeData.recipe[ | if recipeData.recipe[self.schema.RECIPE.PRODUCTS] then | ||
return recipeData.recipe[ | return recipeData.recipe[self.schema.RECIPE.PRODUCTS][self.schema.RECIPE.PRODUCT.ID] | ||
elseif recipeData.recipe[INDEX_RECIPE_PRO_DEPOSITS] then | elseif recipeData.recipe[INDEX_RECIPE_PRO_DEPOSITS] then | ||
Line 913: | Line 899: | ||
---@public | ---@public | ||
---Gets the amount of product produced by the provided recipe. | ---Gets the amount of product produced by the provided recipe. | ||
Line 920: | Line 905: | ||
function BaseDataModel:getRecipeProductAmount(recipeData) | function BaseDataModel:getRecipeProductAmount(recipeData) | ||
if recipeData.recipe[ | if recipeData.recipe[self.schema.RECIPE.PRODUCTS] then | ||
return recipeData.recipe[ | return recipeData.recipe[self.schema.RECIPE.PRODUCTS][self.schema.RECIPE.PRODUCT.AMOUNT] | ||
else | else | ||
Line 930: | Line 915: | ||
---@public | ---@public | ||
---Gets the number of slots for ingredients for the provided recipe. | ---Gets the number of slots for ingredients for the provided recipe. | ||
--- | --- | ||
Line 939: | Line 923: | ||
function BaseDataModel:getRecipeNumIngredientSlots(recipeData) | function BaseDataModel:getRecipeNumIngredientSlots(recipeData) | ||
if recipeData.recipe[ | if recipeData.recipe[self.schema.RECIPE.INGREDIENTS] then | ||
return #recipeData.recipe[ | return #recipeData.recipe[self.schema.RECIPE.INGREDIENTS] | ||
elseif recipeData.recipe[INDEX_RECIPE_ING_SERVICE_GOODS] then | elseif recipeData.recipe[INDEX_RECIPE_ING_SERVICE_GOODS] then | ||
Line 952: | Line 936: | ||
---@public | ---@public | ||
---Gets the number of options for the specified ingredient slot in the provided recipe. | ---Gets the number of options for the specified ingredient slot in the provided recipe. | ||
--- | --- | ||
Line 964: | Line 947: | ||
function BaseDataModel:getRecipeIngredientNumOptions(recipeData, i) | function BaseDataModel:getRecipeIngredientNumOptions(recipeData, i) | ||
if recipeData.recipe[ | if recipeData.recipe[self.schema.RECIPE.INGREDIENTS] then | ||
return #recipeData.recipe[ | return #recipeData.recipe[self.schema.RECIPE.INGREDIENTS][i] | ||
elseif recipeData.recipe[INDEX_RECIPE_ING_SERVICE_GOODS] then | elseif recipeData.recipe[INDEX_RECIPE_ING_SERVICE_GOODS] then | ||
Line 978: | Line 961: | ||
---@public | ---@public | ||
---Gets the good ID for the specified option for the specified ingredient slot in the provided recipe. | ---Gets the good ID for the specified option for the specified ingredient slot in the provided recipe. | ||
--- | --- | ||
Line 991: | Line 973: | ||
function BaseDataModel:getRecipeIngredientOptionIDAt(recipeData, i, j) | function BaseDataModel:getRecipeIngredientOptionIDAt(recipeData, i, j) | ||
if recipeData.recipe[ | if recipeData.recipe[self.schema.RECIPE.INGREDIENTS] then | ||
local optionsList = recipeData.recipe[ | local optionsList = recipeData.recipe[self.schema.RECIPE.INGREDIENTS][i] | ||
return optionsList[j][ | return optionsList[j][self.schema.RECIPE.INGREDIENT.OPTION.ID] | ||
elseif recipeData.recipe[INDEX_RECIPE_ING_SERVICE_GOODS] then | elseif recipeData.recipe[INDEX_RECIPE_ING_SERVICE_GOODS] then | ||
Line 1,001: | Line 983: | ||
---@public | ---@public | ||
---Gets the good ID for the specified option for the specified ingredient slot in the provided recipe. | ---Gets the good ID for the specified option for the specified ingredient slot in the provided recipe. | ||
--- | --- | ||
Line 1,014: | Line 995: | ||
function BaseDataModel:getRecipeIngredientOptionAmountAt(recipeData, i, j) | function BaseDataModel:getRecipeIngredientOptionAmountAt(recipeData, i, j) | ||
if recipeData.recipe[ | if recipeData.recipe[self.schema.RECIPE.INGREDIENTS] then | ||
local optionsList = recipeData.recipe[ | local optionsList = recipeData.recipe[self.schema.RECIPE.INGREDIENTS][i] | ||
return optionsList[j][ | return optionsList[j][self.schema.RECIPE.INGREDIENT.OPTION.AMOUNT] | ||
elseif recipeData.recipe[INDEX_RECIPE_ING_SERVICE_GOODS] then | elseif recipeData.recipe[INDEX_RECIPE_ING_SERVICE_GOODS] then |
Revision as of 04:54, 6 November 2024
Documentation for this module may be created at Module:BaseDataModel/doc
---@class BaseDataModel --- ---@alias buildingIDString string the unique identifier for a building ---@alias buildingRecordTable table a full record for one building ---@alias recipePairTable table a record for one recipe-building pair ---@alias constructionGoodsTable table an array of construction goods' IDs and amounts --- ---I. Provides a standard set of getters for building data and permits overrides from derived classes to specialize the building data, to allow for exceptions, etc. --- ---II. Provides a standard set of getters for recipe queries. All queries return an array of pairs of building IDs and recipes, pulled straight from JSON. Data from those pairs needs to be retrieved with the third interface. --- ---III. Provides a standard set of getters for data within individual recipes, to allow derived classes to specialize based on how their data is structured. Recipe data should not be accessed directly because of these differences. --- --- ---@field getID function building interface method ---@field getCategory function building interface method ---@field getCityScore function building interface method ---@field getConstructionCosts function building interface method ---@field getConstructionTime function building interface method ---@field getDescription function building interface method ---@field getIcon function building interface method ---@field isMovable function building interface method ---@field getName function building interface method ---@field getNumberOfWorkplaces function building interface method ---@field getSize function building interface method ---@field getStorage function building interface method ---@field isServiceBuilding function building interface method --- ---@field _ --- --- --- --- ---@field getIDsAndRecipesWhereProductID function query interface method ---@field getIDsAndRecipesWhereBuildingID function query interface method ---@field getIDsAndRecipesWhereIngredientID function query interface method ---@field getIDsAndRecipesWhereProductIDAndBuildingID function query interface method ---@field getIDsAndRecipesWhereProductIDAndIngredientID function query interface method ---@field getIDsAndRecipesWhereIngredientIDAndBuildingID function query interface method --- ---@field _ --- --- --- --- ---@field getRecipeBuildingID function recipe interface method ---@field getRecipeGrade function recipe interface method ---@field getRecipeTime function recipe interface method ---@field isRecipeProvidingService function recipe interface method ---@field getRecipeProductID function recipe interface method ---@field getRecipeProductAmount function recipe interface method ---@field getRecipeNumIngredientSlots function recipe interface method ---@field getRecipeIngredientNumOptions function recipe interface method ---@field getRecipeIngredientOptionIDAt function recipe interface method ---@field getRecipeIngredientOptionAmountAt function recipe interface method --- ---@field _ --- --- --- --- ---@field dataTable table Primary data table of _buildingRecordTables_, keyed by _buildingIDString_ ---@field mapNamesToIDs table Lookup table, returns a _buildingIDString_ from a building's display name ---@field schema table data schema, with field names and subtables of field names local BaseDataModel = {} --region Dependencies local JsonUtils = require("Module:JsonUtils") --endregion --region Protected data schema ---@protected ---Protected data schema defining Json field names for accessing data. BaseDataModel.schema = { ID = "id", NAME = "displayName", CATEGORY1 = "category", CITY_SCORE = "cityScore", CONSTRUCTION_TIME = "constructionTime", CONSTRUCTION_GOODS = "requiredGoods", CONSTRUCTION = { ID = "name", AMOUNT = "amount", }, DESCRIPTION = "description", IS_ESSENTIAL = "initiallyEssential", IS_MOVABLE = "movable", SIZE_X = "sizeX", SIZE_Y = "sizeY", STORAGE_CAP = "storage", WORKPLACES = "workplaces", RECIPES = "recipes", RECIPE = { GRADE = "grade", PRODUCTION_TIME = "productionTime", PRODUCTS = "product", PRODUCT = { ID = "name", AMOUNT = "amount", }, INGREDIENTS = "ingredients", INGREDIENT = { OPTION = { ID = "name", AMOUNT = "amount", }, }, }, } ---Secondary category, to be defined by derived classes. BaseDataModel.schema.CATEGORY2 = "category2" BaseDataModel.schema.IS_SERVICE = "isService" -- This section is deprecated. These should be handled by derived classes, not here. local INDEX_RECIPE_GRADE_ALT = "gradeId" -- alt for service buildings local INDEX_RECIPE_INGREDIENT_OPTION_AMOUNT = "amount" local INDEX_RECIPE_INGREDIENT_OPTION_ID = "name" local INDEX_RECIPE_ING_SERVICE_GOODS = "goods" -- alt to ingredients for service buildings local INDEX_RECIPE_PRO_DEPOSITS = "seekedDeposits" -- alt to product for huts and camps local INDEX_RECIPE_PRO_SERVICE = "servedNeed" -- alt to product for service buildings local INDEX_RECIPE_PLANTING_TIME = "plantingTime" local INDEX_RECIPE_HARVESTING_TIME = "harvestingTime" local INDEX_RECIPE_GATHERING_TIME = "gatheringTime" local INDEX_STORAGE_TANK = "baseTankCapacity" --endregion --region Private constants ---Enum converts grade strings (keys) to numbers (values). local LOOKUP_CONVERT_GRADE_TO_NUMBER = { ["Grade0"] = 0, ["Grade1"] = 1, ["Grade2"] = 2, ["Grade3"] = 3, } --endregion --region Localization string constants local ERROR_MESSAGE_INSTANCE_NOT_INITIALIZED = "This instance's data table was never initialized. Please check how and where this method was called to diagnose the problem" --endregion --region Private member variables --defined as @fields above in class definition --endregion --region Private methods ---Loads the JSON data into the two structured member variables for data access. --- ---(Benchmarking 2024-11-04: ~0.018 seconds for biggest data file) --- ---@param instance BaseDataModel ---@param dataFile string filename of Json data to load into this instance ---@return table, table instance's data structures --when uncommented for debugging local function load(instance, dataFile) -- Utility module loads the data from JSON. local rawTable = JsonUtils.convertJSONToLuaTable(dataFile) -- Lightweight (no deep copying) restructuring for faster look-ups. instance.dataTable = {} instance.mapNamesToIDs = {} for _, record in ipairs(rawTable) do local id = record[instance.schema.ID] local name = record[instance.schema.NAME] instance.dataTable[id] = record instance.mapNamesToIDs[name] = id -- If derived classes have anything to do per-record, let them here. instance:initializeRecord(id) end --return instance.dataTable, instance.mapNamesToIDs --uncomment when debugging end ---Gets the record with that ID from the instance's data table. --- ---(Benchmarking 2024-11-04: ~0 seconds for biggest data file) --- ---@param instance BaseDataModel ---@param id buildingIDString ---@return buildingRecordTable or nil if not found local function findID(instance, id) if not instance.dataTable then error(ERROR_MESSAGE_INSTANCE_NOT_INITIALIZED) end return instance.dataTable[id] or nil end ---Gets the record with that display name from the instance's data table. If the data model has multiple buildings with the same name, this will return only one of them. --- ---(Benchmarking 2024-11-04: ~0.0000 seconds for biggest data file) --- ---@param instance BaseDataModel ---@param name string the display name ---@return buildingRecordTable or nil if not found local function findName(instance, name) if not instance.dataTable or not instance.mapNamesToIDs then error(ERROR_MESSAGE_INSTANCE_NOT_INITIALIZED) end local id = instance.mapNamesToIDs[name] if not id then return nil end return instance.dataTable[id] or nil end ---Deeply copies the provided table and returns the copy. Does not modify the original. --- ---@param original table to copy ---@return table the copy local function deepCopy(original) local copy if type(original) == "table" then copy = {} for key, value in pairs(original) do copy[key] = deepCopy(value) end else copy = original end return copy end --endregion --region Public constructor ---@public ---@constructor ---Instantiates a new BaseDataModel by loading the specified data file into the instance. --- ---(Benchmarking: ~0.0170 seconds with largest data file.) --- ---@param dataFile string filename of Json data to load into this instance ---@return BaseDataModel a new instance storing the specified data function BaseDataModel.new(dataFile) local instance = {} setmetatable(instance, { __index = BaseDataModel }) instance.INDEX = deepCopy(BaseDataModel.schema) load(instance, dataFile) return instance end --endregion --region Public building interface ---@protected ---Initializes the specified record. --- ---**Likely overridden in derived classes.** --- ---@param id buildingIDString function BaseDataModel:initializeRecord(id) local building = findID(self, id) -- Initialize the extra values not in the data. building[self.schema.CATEGORY2] = "Building" building[self.schema.IS_SERVICE] = false end ---@public ---Gets the ID associated with the specified name in this instance's data. --- ---@param name string the display name ---@return buildingIDString or nil if not found function BaseDataModel:getID(name) local building = findName(self, name) if building then return building[self.schema.ID] else return nil end end ---@public ---Gets the category for the specified ID from this instance's data. --- ---@param id buildingIDString ---@return string category, or nil if not found function BaseDataModel:getCategory(id) local building = findID(self, id) if building then return building[self.schema.CATEGORY1] else return nil end end ---@public ---Gets the secondary category for the specified ID from this instance's data. --- ---@param id buildingIDString ---@return string secondary category, or nil if not found function BaseDataModel:getSecondCategory(id) local building = findID(self, id) if building then return building[self.schema.CATEGORY2] else return nil end end ---@public ---Gets the city score for the specified ID from this instance's data. --- ---@param id buildingIDString ---@return number city score, or nil if not found function BaseDataModel:getCityScore(id) local building = findID(self, id) if building then return building[self.schema.CITY_SCORE] else return nil end end ---@public ---Gets the construction goods for the specified ID from this instance's data. ---Returns the array of construction goods, straight from the table, in subtables of IDs and amounts. --- ---See the data schema for details. --- ---@param id buildingIDString ---@return constructionGoodsTable an array of construction goods' IDs and amounts, or nil if not found function BaseDataModel:getConstructionCosts(id) local building = findID(self, id) if building then return building[self.schema.CONSTRUCTION] else return nil end end ---@public ---Gets the time required for construction for the specified ID from this instance's data. --- ---@param id buildingIDString ---@return number the number of seconds, or nil if not found function BaseDataModel:getConstructionTime(id) local building = findID(self, id) if building then return building[self.schema.CONSTRUCTION_TIME] else return nil end end ---@public ---Gets the description for the specified ID from this instance's data. --- ---@param id buildingIDString ---@return string the description including sprite markup, or nil if not found function BaseDataModel:getDescription(id) local building = findID(self, id) if building then return building[self.schema.DESCRIPTION] else return nil end end ---@public ---Gets the icon filename for the specified ID from this instance's data. --- ---**Likely overridden by derived classes.** --- ---@param id buildingIDString ---@return string the icon filename, or nil if not found function BaseDataModel:getIcon(id) local building = findID(self, id) if building then return id .. "_icon.png" else return nil end end ---@public ---Checks whether the specified ID is movable from this instance's data. --- ---@param id buildingIDString ---@return boolean _true_ if the building can be moved, _false_ if not (or nil if not found) function BaseDataModel:isMovable(id) local building = findID(self, id) if building then return building[self.schema.IS_MOVABLE] else return nil end end ---@public ---Gets the name for the specified ID from this instance's data. --- ---@param id buildingIDString ---@return string the name, or nil if not found function BaseDataModel:getName(id) local building = findID(self, id) if building then return building[self.schema.NAME] else return nil end end ---@public ---Gets the number of workplaces for the specified ID from this instance's data. --- ---@param id buildingIDString ---@return number how many worker slots, or nil if not found function BaseDataModel:getNumberOfWorkplaces(id) local building = findID(self, id) if building then return building[self.schema.WORKPLACES] else return nil end end ---@public ---Gets the size for ths specified ID from this instance's data, expressed as an X-by-Y string. --- ---@param id buildingIDString ---@return string as x-by-y, or nil if not found function BaseDataModel:getSize(id) local building = findID(self, id) if building then return building[self.schema.SIZE_X] .. " × " .. building[self.schema.SIZE_Y] else return nil end end ---@public ---Gets the storage capacity for the specified ID from this instance's data. --- ---**Likely overridden by derived classes.** --- ---@param id buildingIDString ---@return number the storage capacity, or nil if not found function BaseDataModel:getStorage(id) local building = findID(self, id) if building then if building[self.schema.STORAGE_CAP] then return building[self.schema.STORAGE_CAP] elseif building[INDEX_STORAGE_TANK] then return building[INDEX_STORAGE_TANK] end else return nil end end ---@public ---Checks whether the specified ID is a service-type building from this instance's data. --- ---**Likely overridden by derived classes.** --- ---@param id buildingIDString ---@return boolean _true_ if a service building, _false_ if not, or nil if not found function BaseDataModel:isServiceBuilding(id) -- The retrieval isn't used, but it performs necessary error checking. local building = findID(self, id) if building then return false else return nil end end --endregion --region Public building recipe query interface ---@public ---Gets all this instance's buildings' recipes that have the specified product. Returns an array of pairs of building IDs and recipe data. --- ---**Likely overridden by derived classes.** --- ---Benchmarking: ~0.0001 seconds --- ---@param productID string the product ---@return table array of pairs of building IDs and recipe data, or {} if none found function BaseDataModel:getIDsAndRecipesWhereProductID(productID) if not self.dataTable then error(ERROR_MESSAGE_INSTANCE_NOT_INITIALIZED) end local ret = {} for id, building in pairs(self.dataTable) do for _, recipe in ipairs(building[self.schema.RECIPES]) do -- If there's a product subtable with an amount. if recipe[self.schema.RECIPE.PRODUCTS] then if recipe[self.schema.RECIPE.PRODUCTS][self.schema.RECIPE.PRODUCT.ID] == productID then -- Add as an buildingID-recipe pair table.insert(ret, { ["buildingID"] = id, ["recipe"] = recipe, }) end -- There may also be seeked deposits. elseif recipe[INDEX_RECIPE_PRO_DEPOSITS] then if recipe[INDEX_RECIPE_PRO_DEPOSITS] == productID then -- Add as an buildingID-recipe pair table.insert(ret, { ["buildingID"] = id, ["recipe"] = recipe, }) end -- Or a service provided elseif recipe[INDEX_RECIPE_PRO_SERVICE] then if recipe[INDEX_RECIPE_PRO_SERVICE] == productID then -- Add as an buildingID-recipe pair table.insert(ret, { ["buildingID"] = id, ["recipe"] = recipe, }) end end end end return ret end ---@public ---Gets all this instance's building's recipes from the specified building ID. Returns an array of pairs of building IDs and recipe data. --- ---**Likely overridden by derived classes.** --- ---Benchmarking: ~0.0000 seconds --- ---@param buildingID buildingIDString ---@return table array of pairs of building IDs and recipe data, or {} if none found function BaseDataModel:getIDsAndRecipesWhereBuildingID(buildingID) local building = findID(self, buildingID) if not building then return {} end local ret = {} for _, recipe in ipairs(building[self.schema.RECIPES]) do -- Add as an buildingID-recipe pair table.insert(ret, { ["buildingID"] = buildingID, ["recipe"] = recipe, }) end return ret end ---@public ---Gets all this instance's buildings' recipes that use the specified ingredients. Returns an array of pairs of building IDs and recipe data. --- ---**Likely overridden by derived classes.** --- ---Benchmarking: ~0.0003 seconds --- ---@param ingredientID string the ingredient ---@return table array of pairs of building IDs and recipe data, or {} if none found function BaseDataModel:getIDsAndRecipesWhereIngredientID(ingredientID) if not self.dataTable then error(ERROR_MESSAGE_INSTANCE_NOT_INITIALIZED) end local ret = {} for id, building in pairs(self.dataTable) do for _, recipe in ipairs(building[self.schema.RECIPES]) do local ingredientsList = {} -- Some recipes don't have ingredients; skip them obviously. if recipe[self.schema.RECIPE.INGREDIENTS] then ingredientsList = recipe[self.schema.RECIPE.INGREDIENTS] elseif recipe[INDEX_RECIPE_ING_SERVICE_GOODS] then ingredientsList = { recipe[INDEX_RECIPE_ING_SERVICE_GOODS] } end for _, ingredientSlot in ipairs(ingredientsList) do for _, option in ipairs(ingredientSlot) do if option[self.schema.RECIPE.INGREDIENT.OPTION.ID] == ingredientID then -- Add as an buildingID-recipe pair table.insert(ret, { ["buildingID"] = id, ["recipe"] = recipe, }) break end end end end end return ret end ---@public ---Gets all this instance's specified building's recipes that produce the specified product. Returns an array of pairs of building IDs and recipe data. --- ---**Likely overridden by derived classes.** --- ---(Benchmarking: ~0.0000 seconds with largest data file) --- ---@param productID string the product ---@param buildingID buildingIDString ---@return table array of pairs of building IDs and recipe data, or {} if none found function BaseDataModel:getIDsAndRecipesWhereProductIDAndBuildingID(productID, buildingID) local building = findID(self, buildingID) if not building then return {} end local ret = {} for _, recipe in ipairs(building[self.schema.RECIPES]) do -- If there's a product subtable with an amount. if recipe[self.schema.RECIPE.PRODUCTS] then if recipe[self.schema.RECIPE.PRODUCTS][self.schema.RECIPE.PRODUCT.ID] == productID then -- Add as an buildingID-recipe pair table.insert(ret, { ["buildingID"] = buildingID, ["recipe"] = recipe, }) -- Only one ever per building, so return now return ret end -- There could also be a seeked deposit. elseif recipe[INDEX_RECIPE_PRO_DEPOSITS] then if recipe[INDEX_RECIPE_PRO_DEPOSITS] == productID then -- Add as an buildingID-recipe pair table.insert(ret, { ["buildingID"] = buildingID, ["recipe"] = recipe, }) -- Only one ever per building, so return now return ret end -- Or a service provided. elseif recipe[INDEX_RECIPE_PRO_SERVICE] then if recipe[INDEX_RECIPE_PRO_SERVICE] == productID then -- Add as an buildingID-recipe pair table.insert(ret, { ["buildingID"] = buildingID, ["recipe"] = recipe, }) -- Only one ever per building, so return now return ret end end end return {} end ---@public ---Gets all this instance's buildings' recipes that result in the specified product and use the specified ingredient. Returns an array of pairs of building IDs and recipe data. --- ---**Likely overridden by derived classes.** --- ---@param productID string the product ID ---@param ingredientID string the ingredient ID ---@return table array of pairs of building IDs and recipe data, or {} if none found function BaseDataModel:getIDsAndRecipesWhereProductIDAndIngredientID(productID, ingredientID) if not self.dataTable then error(ERROR_MESSAGE_INSTANCE_NOT_INITIALIZED) end local ret = {} for id, building in pairs(self.dataTable) do for _, recipe in ipairs(building[self.schema.RECIPES]) do -- If there's a product subtable with an amount. if recipe[self.schema.RECIPE.PRODUCTS] then if recipe[self.schema.RECIPE.PRODUCTS][self.schema.RECIPE.PRODUCT.ID] == productID then -- Products have ingredients lists. if recipe[self.schema.RECIPE.INGREDIENTS] then for _, ingredientSlot in ipairs(recipe[self.schema.RECIPE.INGREDIENTS]) do for _, option in ipairs(ingredientSlot) do if option[self.schema.RECIPE.INGREDIENT.OPTION.ID] == ingredientID then -- Add as an buildingID-recipe pair table.insert(ret, { ["buildingID"] = id, ["recipe"] = recipe, }) break end end end end end -- There may also be seeked deposits; these never have ingredients, so skip. elseif recipe[INDEX_RECIPE_PRO_DEPOSITS] then -- skip, but I want this to have the same structure as other methods. -- Or a service provided elseif recipe[INDEX_RECIPE_PRO_SERVICE] then if recipe[INDEX_RECIPE_PRO_SERVICE] == productID then -- Services have service goods. if recipe[INDEX_RECIPE_ING_SERVICE_GOODS] then for _, ingredientSlot in ipairs(recipe[INDEX_RECIPE_ING_SERVICE_GOODS]) do for _, option in ipairs(ingredientSlot) do if option[INDEX_RECIPE_INGREDIENT_OPTION_ID] == ingredientID then -- Add as an buildingID-recipe pair table.insert(ret, { ["buildingID"] = id, ["recipe"] = recipe, }) end end end end end end end end return ret end ---@public ---Gets all this instance's specified building's recipes that use the specified ingredient. Returns an array of pairs of building IDs and recipe data. --- ---**Likely overridden by derived classes.** --- ---(Benchmarking: ~0.0000 seconds with largest data file) --- ---@param ingredientID string ---@param buildingID buildingIDString ---@return table array of pairs of building IDs and recipe data, or {} if none found function BaseDataModel:getIDsAndRecipesWhereIngredientIDAndBuildingID(ingredientID, buildingID) local building = findID(self, buildingID) if not building then return {} end local ret = {} for _, recipe in ipairs(building[self.schema.RECIPES]) do local ingredientsList = {} -- Some recipes don't have any ingredients; skip those obviously. if recipe[self.schema.RECIPE.INGREDIENTS] then ingredientsList = recipe[self.schema.RECIPE.INGREDIENTS] elseif recipe[INDEX_RECIPE_ING_SERVICE_GOODS] then ingredientsList = recipe[INDEX_RECIPE_ING_SERVICE_GOODS] end for _, ingredientSlot in ipairs(ingredientsList) do for _, option in ipairs(ingredientSlot) do if option[self.schema.RECIPE.INGREDIENT.OPTION.ID] == ingredientID then -- Add as an buildingID-recipe pair table.insert(ret, { ["buildingID"] = buildingID, ["recipe"] = recipe, }) break end end end end return ret end --endregion --region Public recipe data retrieval interface ---@public ---Gets the building ID from the provided recipe data. ---@param recipeData recipePairTable ---@return buildingIDString function BaseDataModel:getRecipeBuildingID(recipeData) return recipeData.buildingID end ---@public ---Gets the efficiency grade from the provided recipe. ---@param recipeData recipePairTable ---@return number efficiency grade function BaseDataModel:getRecipeGrade(recipeData) if recipeData.recipe[self.schema.RECIPE.GRADE] then return LOOKUP_CONVERT_GRADE_TO_NUMBER[recipeData.recipe[self.schema.RECIPE.GRADE]] elseif recipeData.recipe[INDEX_RECIPE_GRADE_ALT] then return LOOKUP_CONVERT_GRADE_TO_NUMBER[recipeData.recipe[INDEX_RECIPE_GRADE_ALT]] else -- Default of grade zero. return 0 end end ---@public ---Gets the production time from the provided recipe. --- ---**Likely overridden by derived classes.** --- ---@param recipeData recipePairTable ---@return number seconds function BaseDataModel:getRecipeTime(recipeData) if recipeData.recipe[self.schema.RECIPE.PRODUCTION_TIME] then return recipeData.recipe[self.schema.RECIPE.PRODUCTION_TIME] elseif recipeData.recipe[INDEX_RECIPE_PLANTING_TIME] and recipeData.recipe[INDEX_RECIPE_HARVESTING_TIME] then return recipeData.recipe[INDEX_RECIPE_PLANTING_TIME] + recipeData.recipe[INDEX_RECIPE_HARVESTING_TIME] elseif recipeData.recipe[INDEX_RECIPE_GATHERING_TIME] then return recipeData.recipe[INDEX_RECIPE_GATHERING_TIME] else -- Services don't have a time. return 0 end end ---@public ---Checks whether the provided recipe results in a service. --- ---@param recipeData recipePairTable ---@return boolean _true_ if service, _false_ if product function BaseDataModel:isRecipeProvidingService(recipeData) -- This is in the base data model a property of the whole building, not a specific recipe. local building = findID(self, recipeData.buildingID) if building then return building[self.schema.IS_SERVICE] else return nil end end ---@public ---Gets the product ID from the provided recipe. --- ---**Likely overridden by derived classes.** --- ---@param recipeData recipePairTable ---@return string product ID function BaseDataModel:getRecipeProductID(recipeData) if recipeData.recipe[self.schema.RECIPE.PRODUCTS] then return recipeData.recipe[self.schema.RECIPE.PRODUCTS][self.schema.RECIPE.PRODUCT.ID] elseif recipeData.recipe[INDEX_RECIPE_PRO_DEPOSITS] then return recipeData.recipe[INDEX_RECIPE_PRO_DEPOSITS] elseif recipeData.recipe[INDEX_RECIPE_PRO_SERVICE] then return recipeData.recipe[INDEX_RECIPE_PRO_SERVICE] end end ---@public ---Gets the amount of product produced by the provided recipe. ---@param recipeData recipePairTable ---@return number stack size function BaseDataModel:getRecipeProductAmount(recipeData) if recipeData.recipe[self.schema.RECIPE.PRODUCTS] then return recipeData.recipe[self.schema.RECIPE.PRODUCTS][self.schema.RECIPE.PRODUCT.AMOUNT] else -- Default, for deposits, from which gatherers always bring home 1 and for services, which always provide 1 return 1 end end ---@public ---Gets the number of slots for ingredients for the provided recipe. --- ---If the recipe has no ingredients (like for farms and camps), this safely returns zero. --- ---@param recipeData recipePairTable ---@return number of ingredients slots of options function BaseDataModel:getRecipeNumIngredientSlots(recipeData) if recipeData.recipe[self.schema.RECIPE.INGREDIENTS] then return #recipeData.recipe[self.schema.RECIPE.INGREDIENTS] elseif recipeData.recipe[INDEX_RECIPE_ING_SERVICE_GOODS] then return #recipeData.recipe[INDEX_RECIPE_ING_SERVICE_GOODS] else --Camps, for example, don't have ingredients. return 0 end end ---@public ---Gets the number of options for the specified ingredient slot in the provided recipe. --- ---This should never be called if the ingredients table doesn't exist. --- ---**Likely overridden by derived classes.** --- ---@param recipeData recipePairTable ---@param i number index of which ingredient slot ---@return number options for that ingredient slot function BaseDataModel:getRecipeIngredientNumOptions(recipeData, i) if recipeData.recipe[self.schema.RECIPE.INGREDIENTS] then return #recipeData.recipe[self.schema.RECIPE.INGREDIENTS][i] elseif recipeData.recipe[INDEX_RECIPE_ING_SERVICE_GOODS] then -- Should always be 1, but let's not assume return #recipeData.recipe[INDEX_RECIPE_ING_SERVICE_GOODS] else -- Any other cases have no options at all, for example camps. return 0 end end ---@public ---Gets the good ID for the specified option for the specified ingredient slot in the provided recipe. --- ---This should never be called if the ingredients table doesn't exist. --- ---**Likely overridden by derived classes.** --- ---@param recipeData recipePairTable ---@param i number index of which ingredient slot ---@param j table index of which option at that slot ---@return string good ID function BaseDataModel:getRecipeIngredientOptionIDAt(recipeData, i, j) if recipeData.recipe[self.schema.RECIPE.INGREDIENTS] then local optionsList = recipeData.recipe[self.schema.RECIPE.INGREDIENTS][i] return optionsList[j][self.schema.RECIPE.INGREDIENT.OPTION.ID] elseif recipeData.recipe[INDEX_RECIPE_ING_SERVICE_GOODS] then return recipeData.recipe[INDEX_RECIPE_ING_SERVICE_GOODS][i][INDEX_RECIPE_INGREDIENT_OPTION_ID] end end ---@public ---Gets the good ID for the specified option for the specified ingredient slot in the provided recipe. --- ---This should never be called if the ingredients table doesn't exist. --- ---**Likely overridden by derived classes.** --- ---@param recipeData recipePairTable ---@param i number index of which ingredient slot ---@param j table index of which option at that slot ---@return number stack size function BaseDataModel:getRecipeIngredientOptionAmountAt(recipeData, i, j) if recipeData.recipe[self.schema.RECIPE.INGREDIENTS] then local optionsList = recipeData.recipe[self.schema.RECIPE.INGREDIENTS][i] return optionsList[j][self.schema.RECIPE.INGREDIENT.OPTION.AMOUNT] elseif recipeData.recipe[INDEX_RECIPE_ING_SERVICE_GOODS] then return recipeData.recipe[INDEX_RECIPE_ING_SERVICE_GOODS][i][INDEX_RECIPE_INGREDIENT_OPTION_AMOUNT] else return 0 end end --endregion return BaseDataModel