Ref.h

Go to the documentation of this file.

#ifndef DataFormats_Common_Ref_h
#define DataFormats_Common_Ref_h

/*----------------------------------------------------------------------
  
Ref: A template for a interproduct reference to a member of a product_.

----------------------------------------------------------------------*/
/*----------------------------------------------------------------------
//  This defines the public interface to the class Ref<C, T, F>.
//  C                         is the collection type.
//  T (default C::value_type) is the type of an element in the collection.
//
//  ProductID productID       is the product ID of the collection.
//  key_type itemKey          is the key of the element in the collection.
//  C::value_type *itemPtr    is a C++ pointer to the element 
//  Ref<C, T, F> const& ref   is another Ref<C, T, F>

//  Constructors
    Ref(); // Default constructor
    Ref(Ref<C, T> const& ref);  // Copy constructor  (default, not explicitly specified)

    Ref(Handle<C> const& handle, key_type itemKey);
    Ref(ProductID pid, key_type itemKey, EDProductGetter const* prodGetter);

//  Destructor
    virtual ~Ref() {}

    // Operators and methods
    Ref<C, T>& operator=(Ref<C, T> const&);     // assignment (default, not explicitly specified)
    T const& operator*() const;         // dereference
    T const* const operator->() const;      // member dereference
    bool operator==(Ref<C, T> const& ref) const; // equality
    bool operator!=(Ref<C, T> const& ref) const; // inequality
    bool operator<(Ref<C, T> const& ref) const; // ordering
    bool isNonnull() const;         // true if an object is referenced
    bool isNull() const;            // equivalent to !isNonnull()
    bool operator!() const;         // equivalent to !isNonnull()
    ----------------------------------------------------------------------*/

#include "DataFormats/Common/interface/CMS_CLASS_VERSION.h"
#include "DataFormats/Common/interface/EDProductfwd.h"
#include "DataFormats/Common/interface/EDProductGetter.h"
#include "DataFormats/Common/interface/Handle.h"
#include "DataFormats/Common/interface/OrphanHandle.h"
#include "DataFormats/Common/interface/RefCore.h"
#include "DataFormats/Common/interface/RefCoreWithIndex.h"
#include "DataFormats/Common/interface/TestHandle.h"
#include "DataFormats/Common/interface/traits.h"
#include "DataFormats/Provenance/interface/ProductID.h"

#include "boost/functional.hpp"

#include <vector>
#include <type_traits>

#include "DataFormats/Common/interface/RefTraits.h"

namespace edm {
  //Use SFINAE to test for embedded type with name key_compare
  template <typename, typename = void>
  struct has_key_compare : std::false_type {};
  template <typename T>
  struct has_key_compare<T, std::void_t<typename T::key_compare>> : std::true_type {};

  template <typename C, typename K>
  bool compare_key(K const& lhs, K const& rhs) {
    if constexpr (has_key_compare<C>::value) {
      using comparison_functor = typename C::key_compare;
      return comparison_functor()(lhs, rhs);
    } else {
      return lhs < rhs;
    }
  }

  template <typename C, typename T, typename F>
  class RefVector;

  template <typename T>
  class RefToBaseVector;

  template <typename C,
            typename T = typename refhelper::ValueTrait<C>::value,
            typename F = typename refhelper::FindTrait<C, T>::value>
  class Ref {
  private:
    typedef refhelper::FindRefVectorUsingAdvance<RefVector<C, T, F>> VF;
    typedef refhelper::FindRefVectorUsingAdvance<RefToBaseVector<T>> VBF;
    friend class RefVectorIterator<C, T, F>;
    friend class RefVector<C, T, F>;
    friend class RefVector<RefVector<C, T, F>, T, VF>;
    friend class RefVector<RefVector<C, T, F>, T, VBF>;

  public:
    typedef C product_type;
    typedef T value_type;
    typedef T const element_type;  //used for generic programming
    typedef F finder_type;
    typedef typename boost::binary_traits<F>::second_argument_type argument_type;
    typedef typename std::remove_cv<typename std::remove_reference<argument_type>::type>::type key_type;

    static key_type invalidKey() { return key_traits<key_type>::value; }

    Ref() : product_(), index_(key_traits<key_type>::value) {}

    Ref(Handle<C> const& handle, key_type itemKey, bool setNow = true);

    Ref(OrphanHandle<C> const& handle, key_type itemKey, bool setNow = true);

    //  An exception will be thrown if an attempt is made to persistify
    //  any object containing this Ref.  Also, in the future work will
    //  be done to throw an exception if an attempt is made to put any object
    //  containing this Ref into an event(or run or lumi).
    Ref(C const* product, key_type itemKey, bool setNow = true);

    //  An exception will be thrown if an attempt is made to persistify
    //  any object containing this Ref.  Also, in the future work will
    Ref(TestHandle<C> const& handle, key_type itemKey, bool setNow = true);

    Ref(ProductID const& productID, key_type itemKey, EDProductGetter const* prodGetter)
        : product_(productID, nullptr, mustBeNonZero(prodGetter, "Ref", productID), false), index_(itemKey) {}

    //  It is an error (not diagnosable at compile- or run-time) to call
    //  this constructor with a pointer to a T unless the pointed-to T
    //  object is already in a collection of type C stored in the
    //  Event. The given ProductID must be the id of the collection in
    //  the Event.

    Ref(ProductID const& iProductID, T const* item, key_type itemKey, C const* /* iProduct */)
        : product_(iProductID, item, 0, false), index_(itemKey) {}

    Ref(ProductID const& iProductID, T const* item, key_type itemKey)
        : product_(iProductID, item, nullptr, false), index_(itemKey) {}

    Ref(ProductID const& iProductID, T const* item, key_type itemKey, bool transient)
        : product_(iProductID, item, nullptr, transient), index_(itemKey) {}


    explicit Ref(ProductID const& iId) : product_(iId, nullptr, nullptr, false), index_(key_traits<key_type>::value) {}

    Ref(RefProd<C> const& refProd, key_type itemKey);

    ~Ref() {}

    T const& operator*() const;

    T const* operator->() const;

    T const* get() const { return isNull() ? nullptr : this->operator->(); }

    bool isNull() const { return !isNonnull(); }

    bool isNonnull() const { return index_ != edm::key_traits<key_type>::value; }

    bool operator!() const { return isNull(); }

    ProductID id() const { return product_.id(); }

    EDProductGetter const* productGetter() const { return product_.productGetter(); }

    key_type key() const { return index_; }

    // This one just for backward compatibility.  Will be removed soon.
    key_type index() const { return index_; }

    bool hasProductCache() const { return product_.productPtr() != nullptr; }

    bool isAvailable() const;

    bool isTransient() const { return product_.isTransient(); }

    RefCore const& refCore() const { return product_; }

    //Used by ROOT storage
    CMS_CLASS_VERSION(10)
    //  private:
    // Constructor from member of RefVector
    Ref(RefCore const& iRefCore, key_type const& iKey) : product_(iRefCore), index_(iKey) {}

  private:
    // Compile time check that the argument is a C* or C const*
    // or derived from it.
    void checkTypeAtCompileTime(C const*) {}

    RefCore product_;
    key_type index_;
  };

  //***************************
  //Specialization for a vector
  //***************************
#define REF_FOR_VECTOR_ARGS                                              \
  std::vector<E>, typename refhelper::ValueTrait<std::vector<E>>::value, \
      typename refhelper::FindTrait<std::vector<E>, typename refhelper::ValueTrait<std::vector<E>>::value>::value

  template <typename E>
  class Ref<REF_FOR_VECTOR_ARGS> {
  private:
    typedef typename refhelper::ValueTrait<std::vector<E>>::value T;
    typedef
        typename refhelper::FindTrait<std::vector<E>, typename refhelper::ValueTrait<std::vector<E>>::value>::value F;

    typedef refhelper::FindRefVectorUsingAdvance<RefVector<std::vector<E>, T, F>> VF;
    typedef refhelper::FindRefVectorUsingAdvance<RefToBaseVector<T>> VBF;
    friend class RefVectorIterator<std::vector<E>, T, F>;
    friend class RefVector<std::vector<E>, T, F>;
    friend class RefVector<RefVector<std::vector<E>, T, F>, T, VF>;
    friend class RefVector<RefVector<std::vector<E>, T, F>, T, VBF>;

  public:
    typedef std::vector<E> product_type;
    typedef typename refhelper::ValueTrait<std::vector<E>>::value value_type;
    typedef value_type const element_type;  //used for generic programming
    typedef typename refhelper::FindTrait<std::vector<E>, typename refhelper::ValueTrait<std::vector<E>>::value>::value
        finder_type;
    typedef typename boost::binary_traits<F>::second_argument_type argument_type;
    typedef unsigned int key_type;

    static key_type invalidKey() { return key_traits<key_type>::value; }

    Ref() : product_() {}

    Ref(Handle<product_type> const& handle, key_type itemKey, bool setNow = true);

    Ref(OrphanHandle<product_type> const& handle, key_type itemKey, bool setNow = true);

    //  An exception will be thrown if an attempt is made to persistify
    //  any object containing this Ref.  Also, in the future work will
    //  be done to throw an exception if an attempt is made to put any object
    //  containing this Ref into an event(or run or lumi).
    Ref(product_type const* product, key_type itemKey, bool setNow = true);

    //  An exception will be thrown if an attempt is made to persistify
    //  any object containing this Ref.  Also, in the future work will
    Ref(TestHandle<product_type> const& handle, key_type itemKey, bool setNow = true);

    Ref(ProductID const& productID, key_type itemKey, EDProductGetter const* prodGetter)
        : product_(productID, nullptr, mustBeNonZero(prodGetter, "Ref", productID), false, itemKey) {}

    //  It is an error (not diagnosable at compile- or run-time) to call
    //  this constructor with a pointer to a T unless the pointed-to T
    //  object is already in a collection of type C stored in the
    //  Event. The given ProductID must be the id of the collection in
    //  the Event.

    Ref(ProductID const& iProductID, T const* item, key_type itemKey, product_type const* /* iProduct */)
        : product_(iProductID, item, nullptr, false, itemKey) {}

    Ref(ProductID const& iProductID, T const* item, key_type itemKey)
        : product_(iProductID, item, nullptr, false, itemKey) {}

    Ref(ProductID const& iProductID, T const* item, key_type itemKey, bool transient)
        : product_(iProductID, item, nullptr, transient, itemKey) {}


    explicit Ref(ProductID const& iId) : product_(iId, nullptr, nullptr, false, key_traits<key_type>::value) {}

    Ref(RefProd<product_type> const& refProd, key_type itemKey);

    ~Ref() {}

    T const& operator*() const;

    T const* operator->() const;

    T const* get() const { return isNull() ? nullptr : this->operator->(); }

    bool isNull() const { return !isNonnull(); }

    bool isNonnull() const { return key() != edm::key_traits<key_type>::value; }

    bool operator!() const { return isNull(); }

    ProductID id() const { return product_.id(); }

    EDProductGetter const* productGetter() const { return product_.productGetter(); }

    key_type key() const { return product_.index(); }

    // This one just for backward compatibility.  Will be removed soon.
    key_type index() const { return product_.index(); }

    bool hasProductCache() const { return product_.productPtr() != nullptr; }

    bool isAvailable() const;

    bool isTransient() const { return product_.isTransient(); }

    RefCore const& refCore() const { return product_.toRefCore(); }

    //Used by ROOT storage
    CMS_CLASS_VERSION(11)
    //  private:
    // Constructor from member of RefVector
    Ref(RefCore const& iRefCore, key_type const& iKey) : product_(iRefCore, iKey) {}

  private:
    // Compile time check that the argument is a C* or C const*
    // or derived from it.
    void checkTypeAtCompileTime(product_type const*) {}

    RefCoreWithIndex product_;
  };
}  // namespace edm

#include "DataFormats/Common/interface/RefProd.h"
#include "DataFormats/Common/interface/RefCoreGet.h"
#include "DataFormats/Common/interface/RefItemGet.h"

namespace edm {

  template <typename C, typename T, typename F>
  inline Ref<C, T, F>::Ref(Handle<C> const& handle, key_type itemKey, bool)
      : product_(handle.id(), nullptr, nullptr, false), index_(itemKey) {
    if (itemKey == key_traits<key_type>::value)
      return;
    refitem::findRefItem<C, T, F, key_type>(product_, handle.product(), itemKey);
  }

  template <typename E>
  inline Ref<REF_FOR_VECTOR_ARGS>::Ref(Handle<std::vector<E>> const& handle, key_type itemKey, bool)
      : product_(handle.id(), nullptr, nullptr, false, itemKey) {
    if (itemKey == key_traits<key_type>::value)
      return;
    refitem::findRefItem<product_type, value_type, finder_type, key_type>(
        product_.toRefCore(), handle.product(), itemKey);
  }

  template <typename C, typename T, typename F>
  inline Ref<C, T, F>::Ref(OrphanHandle<C> const& handle, key_type itemKey, bool)
      : product_(handle.id(), nullptr, nullptr, false), index_(itemKey) {
    if (itemKey == key_traits<key_type>::value)
      return;
    refitem::findRefItem<C, T, F, key_type>(product_, handle.product(), itemKey);
  }

  template <typename E>
  inline Ref<REF_FOR_VECTOR_ARGS>::Ref(OrphanHandle<std::vector<E>> const& handle, key_type itemKey, bool)
      : product_(handle.id(), nullptr, nullptr, false, itemKey) {
    if (itemKey == key_traits<key_type>::value)
      return;
    refitem::findRefItem<product_type, value_type, finder_type, key_type>(
        product_.toRefCore(), handle.product(), itemKey);
  }

  //  An exception will be thrown if an attempt is made to persistify
  //  any object containing this Ref.  Also, in the future work will
  //  be done to throw an exception if an attempt is made to put any object
  //  containing this Ref into an event(or run or lumi).
  //  Note:  It is legal for the referenced object to be put into the event
  //  and persistified.  It is this Ref itself that cannot be persistified.
  template <typename C, typename T, typename F>
  inline Ref<C, T, F>::Ref(C const* iProduct, key_type itemKey, bool)
      : product_(ProductID(), nullptr, nullptr, true),
        index_(iProduct != nullptr ? itemKey : key_traits<key_type>::value) {
    if (iProduct != nullptr) {
      refitem::findRefItem<C, T, F, key_type>(product_, iProduct, itemKey);
    }
  }

  template <typename E>
  inline Ref<REF_FOR_VECTOR_ARGS>::Ref(std::vector<E> const* iProduct, key_type itemKey, bool)
      : product_(ProductID(), nullptr, nullptr, true, iProduct != nullptr ? itemKey : key_traits<key_type>::value) {
    if (iProduct != nullptr) {
      refitem::findRefItem<product_type, value_type, finder_type, key_type>(product_.toRefCore(), iProduct, itemKey);
    }
  }

  //  An exception will be thrown if an attempt is made to persistify any object containing this Ref.
  template <typename C, typename T, typename F>
  inline Ref<C, T, F>::Ref(TestHandle<C> const& handle, key_type itemKey, bool)
      : product_(handle.id(), nullptr, nullptr, true), index_(itemKey) {
    if (itemKey == key_traits<key_type>::value)
      return;
    refitem::findRefItem<C, T, F, key_type>(product_, handle.product(), itemKey);
  }

  template <typename E>
  inline Ref<REF_FOR_VECTOR_ARGS>::Ref(TestHandle<std::vector<E>> const& handle, key_type itemKey, bool)
      : product_(handle.id(), nullptr, nullptr, true, itemKey) {
    if (itemKey == key_traits<key_type>::value)
      return;
    refitem::findRefItem<product_type, value_type, finder_type, key_type>(
        product_.toRefCore(), handle.product(), itemKey);
  }

  template <typename C, typename T, typename F>
  inline Ref<C, T, F>::Ref(RefProd<C> const& refProd, key_type itemKey)
      : product_(refProd.id(), nullptr, refProd.refCore().productGetter(), refProd.refCore().isTransient()),
        index_(itemKey) {
    if (refProd.refCore().productPtr() != nullptr && itemKey != key_traits<key_type>::value) {
      refitem::findRefItem<C, T, F, key_type>(
          product_, static_cast<product_type const*>(refProd.refCore().productPtr()), itemKey);
    }
  }

  template <typename E>
  inline Ref<REF_FOR_VECTOR_ARGS>::Ref(RefProd<std::vector<E>> const& refProd, key_type itemKey)
      : product_(refProd.id(), nullptr, refProd.refCore().productGetter(), refProd.refCore().isTransient(), itemKey) {
    if (refProd.refCore().productPtr() != nullptr && itemKey != key_traits<key_type>::value) {
      refitem::findRefItem<product_type, value_type, finder_type, key_type>(
          product_.toRefCore(), static_cast<product_type const*>(refProd.refCore().productPtr()), itemKey);
    }
  }

  template <typename C, typename T, typename F>
  inline bool Ref<C, T, F>::isAvailable() const {
    if (product_.isAvailable()) {
      return true;
    }
    return isThinnedAvailable<C>(product_, index_);
  }

  template <typename E>
  inline bool Ref<REF_FOR_VECTOR_ARGS>::isAvailable() const {
    if (product_.isAvailable()) {
      return true;
    }
    return isThinnedAvailable<std::vector<E>>(product_.toRefCore(), key());
  }

  template <typename C, typename T, typename F>
  inline T const& Ref<C, T, F>::operator*() const {
    return *getRefPtr<C, T, F>(product_, index_);
  }
  template <typename E>
  inline typename refhelper::ValueTrait<std::vector<E>>::value const& Ref<REF_FOR_VECTOR_ARGS>::operator*() const {
    return *getRefPtr<REF_FOR_VECTOR_ARGS>(product_.toRefCore(), key());
  }

  template <typename C, typename T, typename F>
  inline T const* Ref<C, T, F>::operator->() const {
    return getRefPtr<C, T, F>(product_, index_);
  }
  template <typename E>
  inline typename refhelper::ValueTrait<std::vector<E>>::value const* Ref<REF_FOR_VECTOR_ARGS>::operator->() const {
    return getRefPtr<REF_FOR_VECTOR_ARGS>(product_.toRefCore(), key());
  }

  template <typename C, typename T, typename F>
  inline bool operator==(Ref<C, T, F> const& lhs, Ref<C, T, F> const& rhs) {
    return lhs.key() == rhs.key() && lhs.refCore() == rhs.refCore();
  }

  template <typename C, typename T, typename F>
  inline bool operator!=(Ref<C, T, F> const& lhs, Ref<C, T, F> const& rhs) {
    return !(lhs == rhs);
  }

  template <typename C, typename T, typename F>
  inline bool operator<(Ref<C, T, F> const& lhs, Ref<C, T, F> const& rhs) {
    return (lhs.refCore() == rhs.refCore() ? compare_key<C>(lhs.key(), rhs.key()) : lhs.refCore() < rhs.refCore());
  }

}  // namespace edm

//Handle specialization here
#include "DataFormats/Common/interface/HolderToVectorTrait_Ref_specialization.h"
#endif