resource category and description

Piglit · Jul 13, 2024 · bb1dc7a · bb1dc7a
1 parent ee60def
commit bb1dc7a
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/src/spaceObjects/shipTemplateBasedObject.cpp b/src/spaceObjects/shipTemplateBasedObject.cpp
@@ -183,6 +183,12 @@ REGISTER_SCRIPT_SUBCLASS_NO_CREATE(ShipTemplateBasedObject, SpaceObject)
     /// Returns true, if the operation was sucessfull; false otherwise.
     /// Example: stbo:tryTransferResource("trade item", 1, otherShip)
     REGISTER_SCRIPT_CLASS_FUNCTION(ShipTemplateBasedObject, tryTransferResource);
+    /// Sets a category for a resource for this ship
+    /// Example: stbo:setResourceCategory("empty cargo space", "Cargo")
+    REGISTER_SCRIPT_CLASS_FUNCTION(ShipTemplateBasedObject, setResourceCategory);
+    /// Sets a description for a resource for this ship
+    /// Example: stbo:setResourceCategory("empty cargo space", "Can be filled with cargo")
+    REGISTER_SCRIPT_CLASS_FUNCTION(ShipTemplateBasedObject, setResourceDescription);
     /// [DEPRECATED]
     /// Use ShipTemplateBasedObject:getShieldLevel() with an index value.
     REGISTER_SCRIPT_CLASS_FUNCTION(ShipTemplateBasedObject, getFrontShield);

diff --git a/src/spaceObjects/shipTemplateBasedObject.h b/src/spaceObjects/shipTemplateBasedObject.h
@@ -40,6 +40,8 @@ class ShipTemplateBasedObject : public SpaceObject, public Updatable
     EPlayerShipType player_ship_type;
     ERestockMissileBehaviour restocks_missiles_docked;
 	std::map<string, int> resources;
+	std::map<string, string> resource_categories;
+	std::map<string, string> resource_descriptions;
 
     ScriptSimpleCallback on_destruction;
     ScriptSimpleCallback on_taking_damage;
@@ -134,6 +136,8 @@ class ShipTemplateBasedObject : public SpaceObject, public Updatable
 	bool tryTransformResource(string resource_name_from, int amount_from, string resource_name_to, int amount_to);
 	void transferResource(string resource_name, int amount, P<ShipTemplateBasedObject> other);
 	bool tryTransferResource(string resource_name, int amount, P<ShipTemplateBasedObject> other);
+    void setResourceCategory(string resource_name, string resource_category) { resource_categories[resource_name] = resource_category; }
+    void setResourceDescription(string resource_name, string resource_description) { resource_descriptions[resource_name] = resource_description; }
 };
 
 #endif//SHIP_TEMPLATE_BASED_OBJECT_H