gistfile1.txt

//! zinc

library ZnPublicLibrary {

    Table pets;
    
    function GetPlayersPet (player p) -> unit {
        group g = CreateGroup();
        GroupEnumUnitsOfPlayer(g, p, Condition( function() -> boolean {
            return pets.has(GetUnitTypeId(GetEnumUnit()));
        }));
        return FirstOfGroup(g);
    }
    
    function onInit() {
        pets = Table.create();
        pets[UNIT_FAWN]              = 1;
	pets[UNIT_ELK_PET]           = 1;
	pets[UNIT_ADULT_ELK]         = 1;
	pets[UNIT_WOLF_PUP]          = 1;
	pets[UNIT_JUNGLE_WOLF]       = 1;
	pets[UNIT_ADULT_JUNGLE_WOLF] = 1;
	pets[UNIT_BEAR_CUB]          = 1;
	pets[UNIT_JUNGLE_BEAR]       = 1;
	pets[UNIT_ADULT_JUNGLE_BEAR] = 1;
    }
}

//! endzinc