Method addToSquad

library/modules/Military.cpp:426–476

}

bool Military::addToSquad(int32_t unit_id, int32_t squad_id, int32_t squad_pos)
{
    df::unit* unit = df::unit::find(unit_id);
    if (unit == nullptr || unit->military.squad_id != -1) return false;

    df::historical_figure* hf = df::historical_figure::find(unit->hist_figure_id);
    if (hf == nullptr)
        return false;

    df::squad* squad = df::squad::find(squad_id);
    if (squad == nullptr) return false;

    if (squad_pos == -1)
    {
        for (int p = 0; p < 10; p++)
        {
            auto pp = vector_get(squad->positions, p);
            if (pp == nullptr || pp->occupant == -1)
            {
                squad_pos = p;
                break;
            }
        }
    }
    if (squad_pos == -1) return false;

    // this function cannot (currently) change the squad commander
    if (squad_pos == 0) return false;

    df::squad_position* pos = vector_get(squad->positions, squad_pos);
    if (pos == nullptr)
        pos = squad->positions[squad_pos] = df::allocate<df::squad_position>();

    pos->occupant = hf->id;
    // does anything else need to be set here?

    unit->military.squad_id = squad->id;
    unit->military.squad_position = squad_pos;

    add_soldier_entity_link(hf, squad, squad_pos);

    #define F(flag) df::equipment_update::mask_##flag
    auto constexpr update_flags = F(weapon) | F(armor) | F(shoes) | F(shield) | F(helm) | F(gloves)
                                | F(ammo) | F(pants) | F(backpack) | F(quiver) | F(flask);
    #undef F

    squad->ammo.update.whole |= update_flags;
    df::global::plotinfo->equipment.update.whole |= update_flags;

    return true;
}

bool Military::removeFromSquad(int32_t unit_id)
{