JaggedVecConversions.cxx

Go to the documentation of this file.

/*
 * Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration.
 */
 
 
#include "AthContainers/tools/JaggedVecConversions.h"
 
 
namespace SG { namespace detail {
 

void JaggedVecProxyBase::resize1 (size_t elt_index, index_type n_new)
{
  int n_old = elt (elt_index).size (elt_index);
  adjust1 (elt_index, n_old, static_cast<int>(n_new) - n_old);
}
 

void JaggedVecProxyBase::adjust1 (size_t elt_index, index_type index, int n_add)
{
  // Return right away if there's nothing to do.
  if (n_add == 0) return;
 
  // Find the IAuxTypeVector for the payload variable.
  IAuxTypeVector* linkedVec;
  if (m_linkedVec.index() == 0) {
    // We already have it.
    linkedVec = std::get<0>(m_linkedVec);
  }
  else {
    // Look it up from the container.
    SG::auxid_t auxid = std::get<1>(m_linkedVec);
    linkedVec = m_container.getStore()->linkedVector (auxid);
 
    // Remember it to possibly use again.
    m_linkedVec = linkedVec;
  }
 
  // Indices from the element that we're modifying.
  Elt_t& e = elt(elt_index);
  size_t beg = e.begin(elt_index);
  size_t end = e.end();
 
  // Backfill trailing zeros up to the current element if needed.
  if (end == 0 && m_elts.back().end() == 0) {
    if (size_t npayload = linkedVec->getDataSpan().size) {
      Elt_t::Shift shift (npayload);
      size_t i = elt_index;
      while (m_elts[i].end() == 0) {
        if (i == 0) break;
        --i;
      }
      if (m_elts[i].end() != 0) {
        std::fill (m_elts.data()+i+1, m_elts.data()+elt_index+1, Elt_t(m_elts[i].end()));
        beg = end = npayload;
      }
    }
  }
 
  // Shift the payload items.
  if (!linkedVec->shift (beg+index, n_add)) {
    m_container.clearCache (linkedVec->auxid());
  }
 
  // Adjust the indices in the jagged vector elements.
  // First the element that we're modifying...
  e = JaggedVecEltBase (end + n_add);
  // .. then the remaining elements
  JaggedVecEltBase::Shift shift (n_add);
  for (auto pos = m_elts.begin() + elt_index+1; pos < m_elts.end(); ++pos)
  {
    if (pos->end() == 0 && (pos-1)->end() == linkedVec->getDataSpan().size) {
      // Stop if we get to trailing zeros to avoid N^2 behavior.
      break;
    }
    shift (*pos);
  }
}
 
 
} } // namespace SG::detail