Make jagged_vector_buffer usable with a vecmem::vector for its size.

Author: krasznaa

core/include/vecmem/containers/data/jagged_vector_buffer.hpp

@@ -86,10 +86,11 @@ class jagged_vector_buffer : public jagged_vector_view<TYPE> {
     /// @param type The type (resizable or not) of the buffer
     ///
     template <typename SIZE_TYPE = std::size_t,
+              typename SIZE_ALLOC = std::allocator<SIZE_TYPE>,
               std::enable_if_t<std::is_integral<SIZE_TYPE>::value &&
                                    std::is_unsigned<SIZE_TYPE>::value,
                                bool> = true>
-    jagged_vector_buffer(const std::vector<SIZE_TYPE>& capacities,
+    jagged_vector_buffer(const std::vector<SIZE_TYPE, SIZE_ALLOC>& capacities,
                          memory_resource& resource,
                          memory_resource* host_access_resource = nullptr,
                          buffer_type type = buffer_type::fixed_size);

core/include/vecmem/containers/impl/jagged_vector_buffer.ipp

@@ -69,13 +69,14 @@ jagged_vector_buffer<TYPE>::jagged_vector_buffer(
                            type) {}
 
 template <typename TYPE>
-template <typename SIZE_TYPE,
+template <typename SIZE_TYPE, typename SIZE_ALLOC,
           std::enable_if_t<std::is_integral<SIZE_TYPE>::value &&
                                std::is_unsigned<SIZE_TYPE>::value,
                            bool> >
 jagged_vector_buffer<TYPE>::jagged_vector_buffer(
-    const std::vector<SIZE_TYPE>& capacities, memory_resource& resource,
-    memory_resource* host_access_resource, buffer_type type)
+    const std::vector<SIZE_TYPE, SIZE_ALLOC>& capacities,
+    memory_resource& resource, memory_resource* host_access_resource,
+    buffer_type type)
     : base_type(capacities.size(), nullptr),
       m_outer_memory(::allocate_jagged_buffer_outer_memory<TYPE>(
           (host_access_resource == nullptr ? 0 : capacities.size()), resource)),

tests/core/test_core_jagged_vector_view.cpp

@@ -1,7 +1,7 @@
 /*
  * VecMem project, part of the ACTS project (R&D line)
  *
- * (c) 2021-2023 CERN for the benefit of the ACTS project
+ * (c) 2021-2025 CERN for the benefit of the ACTS project
  *
  * Mozilla Public License Version 2.0
  */
@@ -217,7 +217,7 @@ TEST_F(core_jagged_vector_view_test, sizeless) {
 
     // Create a resizable buffer for a jagged vector.
     vecmem::data::jagged_vector_buffer<int> output_data(
-        std::vector<std::size_t>(3, 0), m_mem, nullptr,
+        vecmem::vector<std::size_t>(3, 0, &m_mem), m_mem, nullptr,
         vecmem::data::buffer_type::resizable);
     copy.setup(output_data)->wait();