@ -766,7 +766,7 @@ static df::unit *getAnyUnit(df::viewscreen *top)
{
case df::viewscreen_petst::List:
if (!vector_get(screen->is_vermin, screen->cursor))
return (df::unit*)vector_get(screen->animal, screen->cursor);
return vector_get(screen->animal, screen->cursor).unit;
return NULL;
case df::viewscreen_petst::SelectTrainer:
@ -12,9 +12,15 @@ module DFHack
ref.unit_tg if ref.kind_of?(GeneralRefUnit)
when :viewscreen_unitlistst
v = curview
# TODO fix xml to use enums everywhere
page = DFHack::ViewscreenUnitlistst_TPage.int(v.page)
v.units[page][v.cursor_pos[page]]
v.units[v.page][v.cursor_pos[v.page]]
when :viewscreen_petst
case v.mode
when :List
v.animal[v.cursor].unit if !v.is_vermin[v.cursor]
when :SelectTrainer
v.trainer_unit[v.trainer_cursor]
end
else
case ui.main.mode
when :ViewUnits
@ -356,7 +356,7 @@ DEFINE_SORT_HANDLER(unit_sorters, pet, "/List", animals)
std::vector<df::unit*> units;
for (size_t i = 0; i < animals->animal.size(); i++)
units.push_back(animals->is_vermin[i] ? NULL : (df::unit*)animals->animal[i]);
units.push_back(animals->is_vermin[i] ? NULL : animals->animal[i].unit);
if (compute_order(*pout, L, top, &order, units))