private void AddEntityToRecheckVisualAfterLoadList(SmartEntity entity, Buff buff)
{
BuildingComponent buildingComp = entity.BuildingComp;
TroopComponent troopComp = entity.TroopComp;
BuffComponent buffComp = entity.BuffComp;
if (buildingComp != null)
{
buffComp.RegisterForVisualReAddOnBuilding(buff);
}
else if (troopComp != null)
{
buffComp.RegisterForVisualReAddOnTroop(buff);
}
}
