Merge remote-tracking branch 'angavrilov/master'

NEWS

@@ -10,8 +10,6 @@ DFHack future
     - fastdwarf: new mode using debug flags, and some internal consistency fixes.
     - added a small stand-alone utility for applying and removing binary patches.
     - removebadthoughts: add --dry-run option
-  New commands:
-    - fix-armory: activates a plugin that makes armor stands and weapon racks be used again.
   New GUI scripts:
     - gui/guide-path: displays the cached path for minecart Guide orders.
     - gui/workshop-job: displays inputs of a workshop job and allows tweaking them.
@@ -21,6 +19,10 @@ DFHack future
     - properly considers minecarts assigned to routes busy.
     - code for deducing job outputs rewritten in lua for flexibility.
     - logic fix: collecting webs produces silk, and ungathered webs are not thread.
+  New Fix Armory plugin:
+    Together with a couple of binary patches and the gui/assign-rack script,
+    this plugin makes weapon racks, armor stands, chests and cabinets in
+    properly designated barracks be used again for storage of squad equipment.
 
 DFHack v0.34.11-r2
 

Readme.html

@@ -1862,14 +1862,15 @@ armor stands, and in containers.</p>
 <div class="note">
 <p class="first admonition-title">Note</p>
 <p>In order to actually be used, weapon racks have to be patched and
-assigned to a squad. See documentation for <tt class="docutils literal"><span class="pre">gui/assign-rack</span></tt> below.</p>
+manually assigned to a squad. See documentation for <tt class="docutils literal"><span class="pre">gui/assign-rack</span></tt>
+below.</p>
 <p class="last">Also, the default capacity of armor stands is way too low, so check out
 <a class="reference external" href="http://www.bay12games.com/dwarves/mantisbt/view.php?id=1445">http://www.bay12games.com/dwarves/mantisbt/view.php?id=1445</a>
 for a patch addressing that too.</p>
 </div>
 <p>Note that the buildings in the armory are used as follows:</p>
 <ul class="simple">
-<li>Weapon racks when fixed are used to store any assigned weapons.
+<li>Weapon racks (when patched) are used to store any assigned weapons.
 Each rack belongs to a specific squad, and can store up to 5 weapons.</li>
 <li>Armor stands belong to specific squad members and are used for
 armor and shields. By default one stand can store one item of each
@@ -1878,11 +1879,9 @@ which should be sufficient.</li>
 <li>Cabinets are used to store assigned clothing for a specific squad member.
 They are <strong>never</strong> used to store owned clothing.</li>
 <li>Chests (boxes, etc) are used for a flask, backpack or quiver assigned
-to the squad member. Due to a bug, food is dropped out of the backpack
+to the squad member. Due to a probable bug, food is dropped out of the
-when it is stored.</li>
+backpack when it is stored.</li>
 </ul>
-<p>Contrary to the common misconception, all these uses are controlled by the
-<em>Individual Equipment</em> usage flag; the Squad Equipment mode means nothing.</p>
 <div class="warning">
 <p class="first admonition-title">Warning</p>
 <p class="last">Although armor stands, cabinets and chests properly belong only to one
@@ -1890,6 +1889,21 @@ squad member, the owner of the building used to create the barracks will
 randomly use any containers inside the room. Thus, it is recommended to
 always create the armory from a weapon rack.</p>
 </div>
+<p>Contrary to the common misconception, all these uses are controlled by the
+<em>Individual Equipment</em> usage flag. The <em>Squad Equipment</em> flag is actually
+intended for ammo, but the game does even less in that area than for armor
+and weapons. This plugin implements the following rules almost from scratch:</p>
+<ul class="simple">
+<li>Combat ammo is stored in chests inside rooms with Squad Equipment enabled.</li>
+<li>If a chest is assigned to a squad member due to Individual Equipment also
+being set, it is only used for that squad's ammo; otherwise, any squads
+with Squad Equipment on the room will use all of the chests at random.</li>
+<li>Training ammo is stored in chests inside archery ranges designated from
+archery targets, and controlled by the same Train flag as archery training
+itself. This is inspired by some defunct code for weapon racks.</li>
+</ul>
+<p>There are some minor traces in the game code to suggest that the first of
+these rules is intended by Toady; the rest are invented by this plugin.</p>
 </div>
 </div>
 <div class="section" id="mode-switch-and-reclaim">

Readme.rst

@@ -1084,7 +1084,8 @@ armor stands, and in containers.
 .. note::
 
   In order to actually be used, weapon racks have to be patched and
-  assigned to a squad. See documentation for ``gui/assign-rack`` below.
+  manually assigned to a squad. See documentation for ``gui/assign-rack``
+  below.
 
   Also, the default capacity of armor stands is way too low, so check out
   http://www.bay12games.com/dwarves/mantisbt/view.php?id=1445
@@ -1092,7 +1093,7 @@ armor stands, and in containers.
 
 Note that the buildings in the armory are used as follows:
 
-* Weapon racks when fixed are used to store any assigned weapons.
+* Weapon racks (when patched) are used to store any assigned weapons.
   Each rack belongs to a specific squad, and can store up to 5 weapons.
 
 * Armor stands belong to specific squad members and are used for
@@ -1104,11 +1105,8 @@ Note that the buildings in the armory are used as follows:
   They are **never** used to store owned clothing.
 
 * Chests (boxes, etc) are used for a flask, backpack or quiver assigned
-  to the squad member. Due to a bug, food is dropped out of the backpack
+  to the squad member. Due to a probable bug, food is dropped out of the
-  when it is stored.
+  backpack when it is stored.
-
-Contrary to the common misconception, all these uses are controlled by the
-*Individual Equipment* usage flag; the Squad Equipment mode means nothing.
 
 .. warning::
 
@@ -1117,6 +1115,24 @@ Contrary to the common misconception, all these uses are controlled by the
   randomly use any containers inside the room. Thus, it is recommended to
   always create the armory from a weapon rack.
 
+Contrary to the common misconception, all these uses are controlled by the
+*Individual Equipment* usage flag. The *Squad Equipment* flag is actually
+intended for ammo, but the game does even less in that area than for armor
+and weapons. This plugin implements the following rules almost from scratch:
+
+* Combat ammo is stored in chests inside rooms with Squad Equipment enabled.
+
+* If a chest is assigned to a squad member due to Individual Equipment also
+  being set, it is only used for that squad's ammo; otherwise, any squads
+  with Squad Equipment on the room will use all of the chests at random.
+
+* Training ammo is stored in chests inside archery ranges designated from
+  archery targets, and controlled by the same Train flag as archery training
+  itself. This is inspired by some defunct code for weapon racks.
+
+There are some minor traces in the game code to suggest that the first of
+these rules is intended by Toady; the rest are invented by this plugin.
+
 
 Mode switch and reclaim
 =======================

plugins/fix-armory.cpp

@@ -22,6 +22,7 @@
 #include "df/squad.h"
 #include "df/unit.h"
 #include "df/squad_position.h"
+#include "df/squad_ammo_spec.h"
 #include "df/items_other_id.h"
 #include "df/item_weaponst.h"
 #include "df/item_armorst.h"
@@ -33,12 +34,15 @@
 #include "df/item_flaskst.h"
 #include "df/item_backpackst.h"
 #include "df/item_quiverst.h"
+#include "df/item_ammost.h"
 #include "df/building_weaponrackst.h"
 #include "df/building_armorstandst.h"
 #include "df/building_cabinetst.h"
 #include "df/building_boxst.h"
+#include "df/building_squad_use.h"
 #include "df/job.h"
 #include "df/general_ref_building_holderst.h"
+#include "df/general_ref_building_destinationst.h"
 #include "df/barrack_preference_category.h"
 
 #include <stdlib.h>
@@ -86,6 +90,48 @@ DFhackCExport command_result plugin_shutdown (color_ostream &out)
     return CR_OK;
 }
 
+/*
+ * PART 1 - Stop restockpiling of items stored in the armory.
+ *
+ * For everything other than ammo this is quite straightforward,
+ * since the uniform switch code already tries to store items
+ * in barracks containers, and it is thus known what the intention
+ * is. Moreover these containers know which squad and member they
+ * belong to.
+ *
+ * For ammo there is no such code (in fact, ammo is never removed
+ * from a quiver, except when it is dropped itself), so I had to
+ * apply some improvisation. There is one place where BOX containers
+ * with Squad Equipment set are used as an anchor location for a
+ * pathfinding check when assigning ammo, so presumably that's
+ * the correct place. I however wanted to also differentiate
+ * training ammo, so came up with the following rules:
+ *
+ *  1.  Combat ammo and ammo without any allowed use can be stored
+ *      in BOXes marked for Squad Equipment, either directly or via
+ *      containing room. No-allowed-use ammo is assumed to be reserved
+ *      for emergency combat use, or something like that.
+ *  1a. If assigned to a squad position, that box can be used _only_
+ *      for ammo assigned to that specific _squad_. Otherwise, if
+ *      multiple squads can use this room, they will store their
+ *      ammo all mixed up.
+ *  2.  Training ammo can be stored in BOXes within archery ranges
+ *      (designated from archery target) that are enabled for Training.
+ *      Train-only ammo in particular can _only_ be stored in such
+ *      boxes. The inspiration for this comes from some broken code
+ *      for weapon racks in Training rooms.
+ *
+ * As an additional feature (partially needed due to the constraints
+ * of working from an external hack), this plugin also blocks instant
+ * queueing of stockpiling jobs for items blocked on the ground, if
+ * these items are assigned to any squad.
+ *
+ * Since there apparently still are bugs that cause uniform items to be
+ * momentarily dropped on ground, this delay is set not to the minimally
+ * necessary 50 ticks, but to 0.5 - 1.0 in-game days, so as to provide a
+ * grace period during which the items can be instantly picked up again.
+ */
+
 // Check if the item is assigned to any use controlled by the military tab
 static bool is_assigned_item(df::item *item)
 {
@@ -100,6 +146,8 @@ static bool is_assigned_item(df::item *item)
     // Exclude weapons used by miners, wood cutters etc
     switch (type) {
     case item_type::WEAPON:
+        // the game code also checks this for ammo, funnily enough
+        // maybe it's not just for weapons?..
         if (binsearch_index(ui->equipment.work_weapons, item->id) >= 0)
             return false;
         break;
@@ -111,6 +159,76 @@ static bool is_assigned_item(df::item *item)
     return true;
 }
 
+// Check if this ammo item is assigned to this squad with one of the specified uses
+static bool is_squad_ammo(df::item *item, df::squad *squad, bool train, bool combat)
+{
+    for (size_t i = 0; i < squad->ammunition.size(); i++)
+    {
+        auto spec = squad->ammunition[i];
+        bool cs = spec->flags.bits.use_combat;
+        bool ts = spec->flags.bits.use_training;
+
+        // no-use ammo assumed to be combat
+        if (((cs || !ts) && combat) || (ts && train))
+        {
+            if (binsearch_index(spec->assigned, item->id) >= 0)
+                return true;
+        }
+    }
+
+    return false;
+}
+
+// Recursively check room parents to find out if this ammo item is allowed here
+static bool can_store_ammo_rec(df::item *item, df::building *holder, int squad_id)
+{
+    auto squads = holder->getSquads();
+
+    if (squads)
+    {
+        bool target = holder->getType() == building_type::ArcheryTarget;
+
+        for (size_t i = 0; i < squads->size(); i++)
+        {
+            auto use = (*squads)[i];
+
+            // For containers assigned to a squad, only consider that squad
+            if (squad_id >= 0 && use->squad_id != squad_id)
+                continue;
+
+            // Squad Equipment -> combat
+            bool combat = use->mode.bits.squad_eq;
+            // Archery target with Train -> training
+            bool train = target && use->mode.bits.train;
+
+            if (combat || train)
+            {
+                auto squad = df::squad::find(use->squad_id);
+
+                if (squad && is_squad_ammo(item, squad, combat, train))
+                    return true;
+            }
+        }
+    }
+
+    for (size_t i = 0; i < holder->parents.size(); i++)
+        if (can_store_ammo_rec(item, holder->parents[i], squad_id))
+            return true;
+
+    return false;
+}
+
+// Check if the ammo item can be stored in this container
+static bool can_store_ammo(df::item *item, df::building *holder)
+{
+    // Only chests
+    if (holder->getType() != building_type::Box)
+        return false;
+
+    // with appropriate flags set
+    return can_store_ammo_rec(item, holder, holder->getSpecificSquad());
+}
+
 // Check if the item is assigned to the squad member who owns this armory building
 static bool belongs_to_position(df::item *item, df::building *holder)
 {
@@ -154,36 +272,77 @@ static bool is_in_armory(df::item *item)
         return false;
 
     // If indeed in a building, check if it is the right one
-    return belongs_to_position(item, holder);
+    if (item->getType() == item_type::AMMO)
+        return can_store_ammo(item, holder);
+    else
+        return belongs_to_position(item, holder);
 }
 
+/*
+ * Hooks used to affect stockpiling code as it runs, and prevent it
+ * from doing unwanted stuff.
+ *
+ * Toady can simply add these checks directly to the stockpiling code;
+ * we have to abuse some handy item vmethods.
+ */
+
 template<class Item> struct armory_hook : Item {
     typedef Item interpose_base;
 
+    /*
+     * This vmethod is called by the actual stockpiling code before it
+     * tries to queue a job, and is normally used to prevent stockpiling
+     * of uncollected webs.
+     */
     DEFINE_VMETHOD_INTERPOSE(bool, isCollected, ())
     {
-        // Normally this vmethod is used to prevent stockpiling of uncollected webs.
+        // Block stockpiling of items in the armory.
-        // This uses it to also block stockpiling of items in the armory.
         if (is_in_armory(this))
             return false;
 
-        // Also never let items in process of being removed from uniform be stockpiled at once
+        /*
+         * When an item is removed from inventory due to Pickup Equipment
+         * process, the unit code directly invokes the stockpiling code
+         * and thus creates the job even before the item is actually dropped
+         * on the ground. We don't want this at all, especially due to the
+         * grace period idea.
+         *
+         * With access to source, that code can just be changed to simply
+         * drop the item on ground, without running stockpiling code.
+         */
         if (this->flags.bits.in_inventory)
         {
             auto holder = Items::getHolderUnit(this);
 
-            if (holder && ::binsearch_index(holder->military.uniform_drop, this->id) >= 0)
+            // When that call happens, the item is still in inventory
+            if (holder && is_assigned_item(this))
             {
-                if (is_assigned_item(this))
+                // And its ID is is this vector
+                if (::binsearch_index(holder->military.uniform_drop, this->id) >= 0)
                     return false;
             }
         }
 
+        // Call the original vmethod
         return INTERPOSE_NEXT(isCollected)();
     }
 
+    /*
+     * This vmethod is used to actually put the item on the ground.
+     * When it does that, it also adds it to a vector of items to be
+     * instanly restockpiled by a loop in another part of the code.
+     *
+     * We don't want this either, even more than when removing from
+     * uniform, because this can happen in lots of various situations,
+     * including deconstructed containers etc, and we want our own
+     * armory code below to have a chance to look at the item.
+     *
+     * The logical place for this code is in the loop that processes
+     * that vector, but that part is not virtual.
+     */
     DEFINE_VMETHOD_INTERPOSE(bool, moveToGround, (int16_t x, int16_t y, int16_t z))
     {
+        // First, let it do its work
         bool rv = INTERPOSE_NEXT(moveToGround)(x, y, z);
 
         // Prevent instant restockpiling of dropped assigned items.
@@ -196,6 +355,8 @@ template<class Item> struct armory_hook : Item {
             if (erase_from_vector(ovec, &df::item::id, this->id))
             {
                 // and queue it to be checked normally in 0.5-1 in-game days
+                // (this is a grace period in case the uniform is dropped just
+                // for a moment due to a momentary glitch)
                 this->stockpile_countdown = 12 + random_int(12);
                 this->stockpile_delay = 0;
             }
@@ -215,6 +376,7 @@ template<> IMPLEMENT_VMETHOD_INTERPOSE(armory_hook<df::item_shieldst>, isCollect
 template<> IMPLEMENT_VMETHOD_INTERPOSE(armory_hook<df::item_flaskst>, isCollected);
 template<> IMPLEMENT_VMETHOD_INTERPOSE(armory_hook<df::item_backpackst>, isCollected);
 template<> IMPLEMENT_VMETHOD_INTERPOSE(armory_hook<df::item_quiverst>, isCollected);
+template<> IMPLEMENT_VMETHOD_INTERPOSE(armory_hook<df::item_ammost>, isCollected);
 
 template<> IMPLEMENT_VMETHOD_INTERPOSE(armory_hook<df::item_weaponst>, moveToGround);
 template<> IMPLEMENT_VMETHOD_INTERPOSE(armory_hook<df::item_armorst>, moveToGround);
@@ -226,10 +388,25 @@ template<> IMPLEMENT_VMETHOD_INTERPOSE(armory_hook<df::item_shieldst>, moveToGro
 template<> IMPLEMENT_VMETHOD_INTERPOSE(armory_hook<df::item_flaskst>, moveToGround);
 template<> IMPLEMENT_VMETHOD_INTERPOSE(armory_hook<df::item_backpackst>, moveToGround);
 template<> IMPLEMENT_VMETHOD_INTERPOSE(armory_hook<df::item_quiverst>, moveToGround);
+template<> IMPLEMENT_VMETHOD_INTERPOSE(armory_hook<df::item_ammost>, moveToGround);
+
+/*
+ * PART 2 - Actively queue jobs to haul assigned items to the armory.
+ *
+ * The logical place for this of course is in the same code that decides
+ * to put stuff in stockpiles, alongside the checks to prevent moving
+ * stuff away from the armory. We just run it independently every 50
+ * simulation frames.
+ */
 
 // Check if this item is loose and can be moved to armory
 static bool can_store_item(df::item *item)
 {
+    // bad id, or cooldown timer still counting
+    if (!item || item->stockpile_countdown > 0)
+        return false;
+
+    // bad flags?
     if (item->flags.bits.in_job ||
         item->flags.bits.removed ||
         item->flags.bits.in_building ||
@@ -239,6 +416,7 @@ static bool can_store_item(df::item *item)
         item->flags.bits.on_fire)
         return false;
 
+    // in unit inventory?
     auto top = item;
 
     while (top->flags.bits.in_inventory)
@@ -251,6 +429,7 @@ static bool can_store_item(df::item *item)
     if (Items::getGeneralRef(top, general_ref_type::UNIT_HOLDER))
         return false;
 
+    // already in armory?
     if (is_in_armory(item))
         return false;
 
@@ -280,6 +459,8 @@ static bool try_store_item(df::building *target, df::item *item)
 
     job->pos = tpos;
 
+    bool dest = false;
+
     // Choose the job type - correct matching is needed so that
     // later canStoreItem calls would take the job into account.
     switch (target->getType()) {
@@ -295,9 +476,11 @@ static bool try_store_item(df::building *target, df::item *item)
             break;
         case building_type::Cabinet:
             job->job_type = job_type::StoreItemInCabinet;
+            dest = true;
             break;
         default:
             job->job_type = job_type::StoreItemInChest;
+            dest = true;
             break;
     }
 
@@ -314,6 +497,19 @@ static bool try_store_item(df::building *target, df::item *item)
     target->jobs.push_back(job);
     job->references.push_back(href);
 
+    // Two of the jobs need this link to find the job in canStoreItem().
+    // They also don't actually need BUILDING_HOLDER, but it doesn't hurt.
+    if (dest)
+    {
+        auto rdest = df::allocate<df::general_ref_building_destinationst>();
+
+        if (rdest)
+        {
+            rdest->building_id = target->id;
+            job->references.push_back(rdest);
+        }
+    }
+
     // add to job list
     Job::linkIntoWorld(job);
     return true;
@@ -340,14 +536,11 @@ static void try_store_item_set(std::vector<int32_t> &items, df::squad *squad, df
     {
         auto item = df::item::find(items[j]);
 
-        // bad id, or cooldown timer still counting
-        if (!item || item->stockpile_countdown > 0)
-            continue;
-
         // not loose
         if (!can_store_item(item))
             continue;
 
+        // queue jobs to put it in the appropriate container
         if (item->isWeapon())
             try_store_item(squad->rack_combat, item);
         else if (item->isClothing())
@@ -359,6 +552,118 @@ static void try_store_item_set(std::vector<int32_t> &items, df::squad *squad, df
     }
 }
 
+// For storing ammo, use a data structure sorted by free space, to even out the load
+typedef std::map<int, std::set<df::building*> > ammo_box_set;
+
+// Enumerate boxes in the room, adding them to the set
+static void index_boxes(df::building *root, ammo_box_set &group, int squad_id)
+{
+    if (root->getType() == building_type::Box)
+    {
+        int id = root->getSpecificSquad();
+
+        if (id < 0 || id == squad_id)
+        {
+            //color_ostream_proxy out(Core::getInstance().getConsole());
+            //out.print("%08x %d\n", unsigned(root), root->getFreeCapacity(true));
+
+            group[root->getFreeCapacity(true)].insert(root);
+        }
+    }
+
+    for (size_t i = 0; i < root->children.size(); i++)
+        index_boxes(root->children[i], group, squad_id);
+}
+
+// Loop over the set from most empty to least empty
+static bool try_store_ammo(df::item *item, ammo_box_set &group)
+{
+    int volume = item->getVolume();
+
+    for (auto it = group.rbegin(); it != group.rend(); ++it)
+    {
+        if (it->first < volume)
+            break;
+
+        for (auto it2 = it->second.begin(); it2 != it->second.end(); ++it2)
+        {
+            auto bld = *it2;
+
+            if (try_store_item(bld, item))
+            {
+                it->second.erase(bld);
+                group[bld->getFreeCapacity(true)].insert(bld);
+                return true;
+            }
+        }
+    }
+
+    return false;
+}
+
+// Collect chests for ammo storage
+static void index_ammo_boxes(df::squad *squad, ammo_box_set &train_set, ammo_box_set &combat_set)
+{
+    for (size_t j = 0; j < squad->rooms.size(); j++)
+    {
+        auto room = squad->rooms[j];
+        auto bld = df::building::find(room->building_id);
+
+        // Chests in rooms marked for Squad Equipment used for combat ammo
+        if (room->mode.bits.squad_eq)
+            index_boxes(bld, combat_set, squad->id);
+
+        // Chests in archery ranges used for training ammo
+        if (room->mode.bits.train && bld->getType() == building_type::ArcheryTarget)
+            index_boxes(bld, train_set, squad->id);
+    }
+}
+
+// Store ammo into appropriate chests
+static void try_store_ammo(df::squad *squad)
+{
+    bool indexed = false;
+    ammo_box_set train_set, combat_set;
+
+    for (size_t i = 0; i < squad->ammunition.size(); i++)
+    {
+        auto spec = squad->ammunition[i];
+        bool cs = spec->flags.bits.use_combat;
+        bool ts = spec->flags.bits.use_training;
+
+        for (size_t j = 0; j < spec->assigned.size(); j++)
+        {
+            auto item = df::item::find(spec->assigned[j]);
+
+            // not loose
+            if (!can_store_item(item))
+                continue;
+
+            // compute the maps lazily
+            if (!indexed)
+            {
+                indexed = true;
+                index_ammo_boxes(squad, train_set, combat_set);
+            }
+
+            // BUG: if the same container is in both sets,
+            // when a job is queued, the free space in the other
+            // set will not be updated, which could lead to uneven
+            // loading - but not to overflowing the container!
+
+            // As said above, combat goes into Squad Equipment
+            if (cs && try_store_ammo(item, combat_set))
+                continue;
+            // Training goes into Archery Range with Train
+            if (ts && try_store_ammo(item, train_set))
+                continue;
+            // No use goes into combat
+            if (!(ts || cs) && try_store_ammo(item, combat_set))
+                continue;
+        }
+    }
+}
+
 static bool is_enabled = false;
 
 DFhackCExport command_result plugin_onupdate(color_ostream &out, state_change_event event)
@@ -383,6 +688,8 @@ DFhackCExport command_result plugin_onupdate(color_ostream &out, state_change_ev
 
             try_store_item_set(pos->assigned_items, squad, pos);
         }
+
+        try_store_ammo(squad);
     }
 
     return CR_OK;
@@ -408,6 +715,7 @@ static void enable_hooks(color_ostream &out, bool enable)
     enable_hook(out, INTERPOSE_HOOK(armory_hook<df::item_flaskst>, isCollected), enable);
     enable_hook(out, INTERPOSE_HOOK(armory_hook<df::item_backpackst>, isCollected), enable);
     enable_hook(out, INTERPOSE_HOOK(armory_hook<df::item_quiverst>, isCollected), enable);
+    enable_hook(out, INTERPOSE_HOOK(armory_hook<df::item_ammost>, isCollected), enable);
 
     enable_hook(out, INTERPOSE_HOOK(armory_hook<df::item_weaponst>, moveToGround), enable);
     enable_hook(out, INTERPOSE_HOOK(armory_hook<df::item_armorst>, moveToGround), enable);
@@ -419,6 +727,7 @@ static void enable_hooks(color_ostream &out, bool enable)
     enable_hook(out, INTERPOSE_HOOK(armory_hook<df::item_flaskst>, moveToGround), enable);
     enable_hook(out, INTERPOSE_HOOK(armory_hook<df::item_backpackst>, moveToGround), enable);
     enable_hook(out, INTERPOSE_HOOK(armory_hook<df::item_quiverst>, moveToGround), enable);
+    enable_hook(out, INTERPOSE_HOOK(armory_hook<df::item_ammost>, moveToGround), enable);
 }
 
 static void enable_plugin(color_ostream &out)