DataFormats/Common/interface/RefVectorBase.h - master - githubmirror/cmssw

DataFormats/Common/interface/RefVectorBase.h
154 строки5 KB
Matti Kortelainen
Move stored types in io_v1 namespace, reset class versions to 3
22 май 2026, 18:18
22 май 2026, 18:18909f324
#ifndef DataFormats_Common_RefVectorBase_h
#define DataFormats_Common_RefVectorBase_h

/*----------------------------------------------------------------------
  
RefVectorBase: Base class for a vector of interproduct references.


----------------------------------------------------------------------*/

#include "DataFormats/Common/interface/CMS_CLASS_VERSION.h"
#include "DataFormats/Common/interface/EDProductfwd.h"
#include "DataFormats/Common/interface/RefCore.h"
#include <vector>

namespace edm {

  class EDProductGetter;

  class RefVectorMemberPointersHolder {
  public:
    RefVectorMemberPointersHolder() {}
    std::vector<void const*> const& memberPointers() const { return memberPointers_; }
    std::vector<void const*>& memberPointers() { return memberPointers_; }

  private:
    std::vector<void const*> memberPointers_;
  };

  template <typename KEY>
  class RefVectorBase {
  public:
    typedef std::vector<KEY> keys_type;
    typedef KEY key_type;
    typedef typename keys_type::size_type size_type;
    /// Default constructor needed for reading from persistent store. Not for direct use.
    RefVectorBase() : product_(), keys_() {}
    RefVectorBase(RefVectorBase const& rhs)
        : product_(rhs.product_), keys_(rhs.keys_), memberPointersHolder_(rhs.memberPointersHolder_) {}
    RefVectorBase(RefVectorBase&& rhs) noexcept
        : product_(std::move(rhs.product_)),
          keys_(std::move(rhs.keys_)),
          memberPointersHolder_(std::move(rhs.memberPointersHolder_)) {}

    RefVectorBase& operator=(RefVectorBase const& rhs) {
      RefVectorBase temp(rhs);
      this->swap(temp);
      return *this;
    }
    RefVectorBase& operator=(RefVectorBase&& rhs) noexcept {
      product_ = std::move(rhs.product_);
      keys_ = std::move(rhs.keys_);
      memberPointersHolder_ = std::move(rhs.memberPointersHolder_);
      return *this;
    }

    explicit RefVectorBase(ProductID const& productID,
                           void const* prodPtr = nullptr,
                           EDProductGetter const* prodGetter = nullptr)
        : product_(productID, prodPtr, prodGetter, false), keys_() {}

    /// Destructor
    ~RefVectorBase() noexcept {}

    /// Accessor for product ID and product getter
    RefCore const& refCore() const { return product_; }

    void const* cachedMemberPointer(size_type idx) const {
      return memberPointers().empty() ? nullptr : memberPointers()[idx];
    }

    /// Accessor for vector of keys and pointers
    keys_type const& keys() const { return keys_; }

    /// Is vector empty?
    bool empty() const { return keys_.empty(); }

    /// Size of vector
    size_type size() const { return keys_.size(); }

    void pushBack(RefCore const& product, KEY const& key) {
      product_.pushBackRefItem(product);
      if (product.productPtr() != nullptr) {
        if (memberPointers().empty()) {
          memberPointersHolder_.memberPointers().resize(keys_.size(), nullptr);
        }
        memberPointersHolder_.memberPointers().push_back(product.productPtr());
        keys_.push_back(key);
        return;
      } else {
        if (!memberPointers().empty()) {
          memberPointersHolder_.memberPointers().push_back(nullptr);
        }
        keys_.push_back(key);
      }
    }

    /// Capacity of vector
    size_type capacity() const { return keys_.capacity(); }

    /// Reserve space for vector
    void reserve(size_type n) { keys_.reserve(n); }

    /// erase an element from the vector
    typename keys_type::iterator eraseAtIndex(size_type index) {
      memberPointersHolder_.memberPointers().erase(memberPointersHolder_.memberPointers().begin() + index);
      return keys_.erase(keys_.begin() + index);
    }

    /// clear the vector
    void clear() {
      keys_.clear();
      memberPointersHolder_.memberPointers().clear();
      product_ = RefCore();
    }

    /// swap two vectors
    void swap(RefVectorBase<KEY>& other) noexcept {
      product_.swap(other.product_);
      keys_.swap(other.keys_);
      memberPointersHolder_.memberPointers().swap(other.memberPointersHolder_.memberPointers());
    }

    //Needed for ROOT storage
    CMS_CLASS_VERSION(3)

  private:
    std::vector<void const*> const& memberPointers() const { return memberPointersHolder_.memberPointers(); }

    RefCore product_;
    keys_type keys_;
    RefVectorMemberPointersHolder memberPointersHolder_;
  };

  /// Equality operator
  template <typename KEY>
  bool operator==(RefVectorBase<KEY> const& lhs, RefVectorBase<KEY> const& rhs) {
    return lhs.refCore() == rhs.refCore() && lhs.keys() == rhs.keys();
  }

  /// Inequality operator
  template <typename KEY>
  bool operator!=(RefVectorBase<KEY> const& lhs, RefVectorBase<KEY> const& rhs) {
    return !(lhs == rhs);
  }

  /// swap two vectors
  template <typename KEY>
  inline void swap(RefVectorBase<KEY>& a, RefVectorBase<KEY>& b) {
    a.swap(b);
  }

}  // namespace edm
#endif