mirror of
https://github.com/TrinityCore/TrinityCore.git
synced 2026-01-18 08:28:32 +01:00
bc96df1aae
Core/ScriptedAI: Extend SummonList::DoAction to take a predicate function as parameter and allow specifying a maximum number of units to be selected.
66 lines
2.2 KiB
C++
66 lines
2.2 KiB
C++
/*
|
|
* Copyright (C) 2008-2012 TrinityCore <http://www.trinitycore.org/>
|
|
*
|
|
* This program is free software; you can redistribute it and/or modify it
|
|
* under the terms of the GNU General Public License as published by the
|
|
* Free Software Foundation; either version 2 of the License, or (at your
|
|
* option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful, but WITHOUT
|
|
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
|
|
* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
|
|
* more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License along
|
|
* with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#ifndef TRINITY_CONTAINERS_H
|
|
#define TRINITY_CONTAINERS_H
|
|
|
|
namespace Trinity
|
|
{
|
|
namespace Containers
|
|
{
|
|
template<class T>
|
|
void RandomResizeList(std::list<T> &list, uint32 size)
|
|
{
|
|
size_t list_size = list.size();
|
|
|
|
while (list_size > size)
|
|
{
|
|
typename std::list<T>::iterator itr = list.begin();
|
|
std::advance(itr, urand(0, list_size - 1));
|
|
list.erase(itr);
|
|
--list_size;
|
|
}
|
|
}
|
|
|
|
template<class T, class Predicate>
|
|
void RandomResizeList(std::list<T> &list, Predicate& predicate, uint32 size)
|
|
{
|
|
//! First use predicate filter
|
|
std::list<T> listCopy;
|
|
for (typename std::list<T>::iterator itr = list.begin(); itr != list.end(); ++itr)
|
|
if (predicate(*itr))
|
|
listCopy.push_back(*itr);
|
|
|
|
if (size)
|
|
RandomResizeList(listCopy, size);
|
|
|
|
list = listCopy;
|
|
}
|
|
|
|
/* Select a random element from a container. Note: make sure you explicitly empty check the container */
|
|
template <class C> typename C::value_type const& SelectRandomContainerElement(C const& container)
|
|
{
|
|
typename C::const_iterator it = container.begin();
|
|
std::advance(it, urand(0, container.size() - 1));
|
|
return *it;
|
|
}
|
|
};
|
|
//! namespace Containers
|
|
};
|
|
//! namespace Trinity
|
|
|
|
#endif //! #ifdef TRINITY_CONTAINERS_H
|