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/ConstPtrCache.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 "boost/call_traits.hpp"
#include "boost/type_traits.hpp"
#include "boost/mpl/has_xxx.hpp"
#include "boost/utility/enable_if.hpp"

BOOST_MPL_HAS_XXX_TRAIT_DEF(key_compare)

  template <typename C, typename K>
  typename boost::enable_if<has_key_compare<C>, bool>::type
  compare_key(K const& lhs, K const& rhs) {
    typedef typename C::key_compare comparison_functor;
    return comparison_functor()(lhs, rhs);
  }

  template <typename C, typename K>
  typename boost::disable_if<has_key_compare<C>, bool>::type
  compare_key(K const& lhs, K const& rhs) {
    return lhs < rhs;
  }

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

namespace edm {
  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<RefVector<C, T, F>, T, VF>, T, VF>;
    friend class RefVector<RefVector<C, T, F>, T, VBF>;
    friend class RefVector<RefVector<RefVector<C, T, F>, T, VBF>, 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 boost::remove_cv<typename boost::remove_reference<argument_type>::type>::type key_type;   

    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);

    //  Note. refvector[index] returns a Ref where index is the index into
    //  the RefVector. This index argument is the index into the COLLECTION,
    //  not the index into the RefVector.
    Ref(RefVector<C, T, F> const& refvector, 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, 0, 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 iItemKey, C const* iProduct) :
      product_(iProductID, iProduct, 0, false), index_(iItemKey)
    { }


    explicit Ref(ProductID const& iId) :
      product_(iId, 0, 0, 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() ? 0 : 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();}

    // Accessor must get the product if necessary
    C const* product() const;

    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() != 0;}

    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*) {}

    mutable 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<RefVector<std::vector<E>, T, F>, T, VF>, T, VF>;
    friend class RefVector<RefVector<std::vector<E>, T, F>, T, VBF>;
    friend class RefVector<RefVector<RefVector<std::vector<E>, T, F>, T, VBF>, 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;   
    
    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);
    
    //  Note. refvector[index] returns a Ref where index is the index into
    //  the RefVector. This index argument is the index into the COLLECTION,
    //  not the index into the RefVector.
    Ref(RefVector<product_type, T, F> const& refvector, 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, 0, 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 iItemKey, product_type const* iProduct) :
    product_(iProductID, iProduct, 0, false,iItemKey)
    { }
    
    
    explicit Ref(ProductID const& iId) :
    product_(iId, 0, 0, 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() ? 0 : 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();}
    
    // Accessor must get the product if necessary
    product_type const* product() const;
    
    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() != 0;}
    
    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*) {}
    
    mutable RefCoreWithIndex product_;
    
  };

}

#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 setNow) :
      product_(handle.id(), handle.product(), 0, false), index_(itemKey){
    checkTypeAtCompileTime(handle.product());
    assert(key() == itemKey);
        
    if (setNow) {getPtr_<C, T, F>(product_, index_);}
  }

  template <typename E>
  inline
  Ref<REF_FOR_VECTOR_ARGS>::Ref(Handle<std::vector<E> > const& handle, key_type itemKey, bool setNow) :
  product_(handle.id(), handle.product(), 0, false,itemKey){
    checkTypeAtCompileTime(handle.product());
    assert(key() == itemKey);
    
    if (setNow) {getPtr_<REF_FOR_VECTOR_ARGS>(product_.toRefCore(), index());}
  }

  template <typename C, typename T, typename F>
  inline
  Ref<C, T, F>::Ref(OrphanHandle<C> const& handle, key_type itemKey, bool setNow) :
  product_(handle.id(), handle.product(), 0, false), index_(itemKey) {
    checkTypeAtCompileTime(handle.product());
    assert(key() == itemKey);
        
    if (setNow) {getPtr_<C, T, F>(product_, index_);}
  }

  template <typename E>
  inline
  Ref<REF_FOR_VECTOR_ARGS>::Ref(OrphanHandle<std::vector<E> > const& handle, key_type itemKey, bool setNow) :
  product_(handle.id(), handle.product(), 0, false,itemKey) {
    checkTypeAtCompileTime(handle.product());
    assert(key() == itemKey);
    
    if (setNow) {getPtr_<REF_FOR_VECTOR_ARGS>(product_.toRefCore(), key());}
  }
  
  template <typename C, typename T, typename F>
  inline
  Ref<C, T, F>::Ref(RefVector<C, T, F> const& refvector, key_type itemKey, bool setNow) :
      product_(refvector.id(), refvector.product(), 0, refvector.isTransient()), index_(itemKey) {
    checkTypeAtCompileTime(refvector.product());
    assert(key() == itemKey);
        
    if (setNow) {getPtr_<C, T, F>(product_, index_);}
  }

  template <typename E>
  inline
  Ref<REF_FOR_VECTOR_ARGS>::Ref(RefVector<REF_FOR_VECTOR_ARGS> const& refvector, key_type itemKey, bool setNow) :
  product_(refvector.id(), refvector.product(), 0, refvector.isTransient(),itemKey) {
    checkTypeAtCompileTime(refvector.product());
    assert(key() == itemKey);
    
    if (setNow) {getPtr_<REF_FOR_VECTOR_ARGS>(product_.toRefCore(), key());}
  }
  
  //  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 iItemKey, bool iSetNow) :
    product_(ProductID(), iProduct, 0, true), index_(iProduct != 0 ? iItemKey : key_traits<key_type>::value) {
    checkTypeAtCompileTime(iProduct);
    assert(key() == (iProduct != 0 ? iItemKey : key_traits<key_type>::value));
        
    if (iSetNow && iProduct != 0) {getPtr_<C, T, F>(product_, index_);}
  }

  template <typename E>
  inline
  Ref<REF_FOR_VECTOR_ARGS>::Ref(std::vector<E> const* iProduct, key_type iItemKey, bool iSetNow) :
  product_(ProductID(), iProduct, 0, true,(iProduct != 0 ? iItemKey : key_traits<key_type>::value)) {
    checkTypeAtCompileTime(iProduct);
    assert(key() == (iProduct != 0 ? iItemKey : key_traits<key_type>::value));
    
    if (iSetNow && iProduct != 0) {getPtr_<REF_FOR_VECTOR_ARGS>(product_.toRefCore(), key());}
  }

  //  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 setNow) :
      product_(handle.id(), handle.product(), 0, true), index_(itemKey) {
    checkTypeAtCompileTime(handle.product());
    assert(key() == itemKey);
        
    if (setNow) {getPtr_<C, T, F>(product_, index_);}
  }

  template <typename E>
  inline
  Ref<REF_FOR_VECTOR_ARGS>::Ref(TestHandle<std::vector<E> > const& handle, key_type itemKey, bool setNow) :
  product_(handle.id(), handle.product(), 0, true,itemKey) {
    checkTypeAtCompileTime(handle.product());
    assert(key() == itemKey);
    
    if (setNow) {getPtr_<REF_FOR_VECTOR_ARGS>(product_.toRefCore(), key());}
  }

  template <typename C, typename T, typename F>
  inline
  Ref<C, T, F>::Ref(RefProd<C> const& refProd, key_type itemKey) :
      product_(refProd.id(), refProd.refCore().productPtr(), refProd.refCore().productGetter(), refProd.refCore().isTransient()), index_(itemKey) {
    assert(index() == itemKey);
  }

  template <typename E>
  inline
  Ref<REF_FOR_VECTOR_ARGS>::Ref(RefProd<std::vector<E> > const& refProd, key_type itemKey) :
  product_(refProd.id(), refProd.refCore().productPtr(), refProd.refCore().productGetter(), refProd.refCore().isTransient(),itemKey) {
    assert(index() == itemKey);
  }
  
  
  // Accessor must get the product if necessary
  template <typename C, typename T, typename F>
  inline
  C const*
  Ref<C, T, F>::product() const {
      return isNull() ? 0 : edm::template getProduct<C>(product_);
  }
  template <typename E>
  inline
  std::vector<E> const*
  Ref<REF_FOR_VECTOR_ARGS>::product() const {
    return isNull() ? 0 : edm::template getProduct<std::vector<E> >(product_.toRefCore());
  }

  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 *getPtr<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 *getPtr<REF_FOR_VECTOR_ARGS>(product_.toRefCore(), key());
  }

  template <typename C, typename T, typename F>
  inline
  T const*
  Ref<C, T, F>::operator->() const {
    return getPtr<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 getPtr<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());
  }

}

#include "DataFormats/Common/interface/HolderToVectorTrait.h"
#include "DataFormats/Common/interface/Holder.h"
#include "DataFormats/Common/interface/VectorHolder.h"
#include "DataFormats/Common/interface/RefVector.h"

namespace edm {
  namespace reftobase {

    template <typename T, typename REF>
    struct RefHolderToVector {
      static  std::auto_ptr<BaseVectorHolder<T> > makeVectorHolder() {
    typedef RefVector<typename REF::product_type,
                      typename REF::value_type, 
                          typename REF::finder_type> REFV;
    return std::auto_ptr<BaseVectorHolder<T> >(new VectorHolder<T, REFV>);
      }
      static  std::auto_ptr<RefVectorHolderBase> makeVectorBaseHolder() {
    typedef RefVector<typename REF::product_type,
                      typename REF::value_type, 
                          typename REF::finder_type> REFV;
    return std::auto_ptr<RefVectorHolderBase>(new RefVectorHolder<REFV>);
      }
    };

    template<typename T1, typename C, typename T, typename F>
    struct HolderToVectorTrait<T1, Ref<C, T, F> > {
      typedef RefHolderToVector<T1, Ref<C, T, F> > type;
    };

    template <typename REF>
    struct RefRefHolderToRefVector {
      static std::auto_ptr<RefVectorHolderBase> makeVectorHolder() {
    typedef RefVector<typename REF::product_type,
                      typename REF::value_type, 
                            typename REF::finder_type> REFV;
    return std::auto_ptr<RefVectorHolderBase>(new RefVectorHolder<REFV>);
      }
    };

    template<typename C, typename T, typename F>
    struct RefHolderToRefVectorTrait<Ref<C, T, F> > {
      typedef RefRefHolderToRefVector<Ref<C, T, F> > type;
    };

  }
}
  
#endif