OpenRCT2/src/openrct2/core/Collections.hpp

102 lines
3.0 KiB
C++

/*****************************************************************************
* Copyright (c) 2014-2018 OpenRCT2 developers
*
* For a complete list of all authors, please refer to contributors.md
* Interested in contributing? Visit https://github.com/OpenRCT2/OpenRCT2
*
* OpenRCT2 is licensed under the GNU General Public License version 3.
*****************************************************************************/
#pragma once
#include "../common.h"
#include "Memory.hpp"
#include "String.hpp"
#include <initializer_list>
namespace Collections
{
template<typename TCollection, typename TItem>
static void AddRange(TCollection& collection, std::initializer_list<TItem> initializerList)
{
collection.insert(collection.end(), initializerList.begin(), initializerList.end());
}
template<typename TCollection, typename TItem, typename TComparer>
static bool Contains(TCollection& collection, TItem needle, TComparer comparer)
{
for (TItem item : collection)
{
if (comparer(item, needle))
{
return true;
}
}
return false;
}
template<typename TCollection, typename TItem, typename TComparer>
static size_t IndexOf(TCollection& collection, TItem needle, TComparer comparer)
{
size_t index = 0;
for (TItem item : collection)
{
if (comparer(item, needle))
{
return index;
}
index++;
}
return SIZE_MAX;
}
template<typename TCollection, typename TPred> static size_t IndexOf(TCollection& collection, TPred predicate)
{
size_t index = 0;
for (auto item : collection)
{
if (predicate(item))
{
return index;
}
index++;
}
return SIZE_MAX;
}
#pragma region String helpers
template<typename TCollection> static bool Contains(TCollection& collection, const char* item, bool ignoreCase = false)
{
return Contains(
collection, item, [ignoreCase](const char* a, const char* b) { return String::Equals(a, b, ignoreCase); });
}
template<typename TCollection> static size_t IndexOf(TCollection& collection, const char* item, bool ignoreCase = false)
{
return IndexOf(
collection, item, [ignoreCase](const char* a, const char* b) { return String::Equals(a, b, ignoreCase); });
}
template<typename TCollection> static typename TCollection::value_type* ToArray(const TCollection& collection)
{
size_t count = collection.size();
if (count == 0)
{
return nullptr;
}
auto* items = Memory::AllocateArray<typename TCollection::value_type>(count);
size_t i = 0;
for (const auto& item : collection)
{
items[i] = item;
i++;
}
return items;
}
#pragma endregion
} // namespace Collections