diff --git a/src/server/game/Grids/Dynamic/TypeContainer.h b/src/server/game/Grids/Dynamic/TypeContainer.h index 0a44b461929..3d49d222ff2 100644 --- a/src/server/game/Grids/Dynamic/TypeContainer.h +++ b/src/server/game/Grids/Dynamic/TypeContainer.h @@ -132,6 +132,14 @@ public: return Trinity::Find(_elements, handle, (SPECIFIC_TYPE*)NULL); } + template + std::size_t Size() const + { + std::size_t size = 0; + Trinity::Size(_elements, &size, (SPECIFIC_TYPE*)nullptr); + return size; + } + ContainerUnorderedMap& GetElements() { return _elements; } ContainerUnorderedMap const& GetElements() const { return _elements; } diff --git a/src/server/game/Grids/Dynamic/TypeContainerFunctions.h b/src/server/game/Grids/Dynamic/TypeContainerFunctions.h index 32ae871df73..94d9a7e1594 100644 --- a/src/server/game/Grids/Dynamic/TypeContainerFunctions.h +++ b/src/server/game/Grids/Dynamic/TypeContainerFunctions.h @@ -125,6 +125,33 @@ namespace Trinity return ret ? ret : Remove(elements._TailElements, handle, (SPECIFIC_TYPE*)nullptr); } + // Count helpers + template + bool Size(ContainerUnorderedMap const& elements, std::size_t* size, SPECIFIC_TYPE* /*obj*/) + { + *size = elements._element.size(); + return true; + } + + template + bool Size(ContainerUnorderedMap const& /*elements*/, std::size_t* /*size*/, SPECIFIC_TYPE* /*obj*/) + { + return false; + } + + template + bool Size(ContainerUnorderedMap const& /*elements*/, std::size_t* /*size*/, SPECIFIC_TYPE* /*obj*/) + { + return false; + } + + template + bool Size(ContainerUnorderedMap, KEY_TYPE> const& elements, std::size_t* size, SPECIFIC_TYPE* /*obj*/) + { + bool ret = Size(elements._elements, size, (SPECIFIC_TYPE*)nullptr); + return ret ? ret : Size(elements._TailElements, size, (SPECIFIC_TYPE*)nullptr); + } + /* ContainerMapList Helpers */ // count functions template