INav4MomLinkContainer.h

Go to the documentation of this file.

 
/*
  Copyright (C) 2002-2020 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef NAVFOURMOM_INAV4MOMLINKCONTAINER_H
#define NAVFOURMOM_INAV4MOMLINKCONTAINER_H 1
 
// STL includes
#include <vector>
 
#include "AthLinks/ElementLink.h"
#include "AthenaKernel/CLASS_DEF.h"
#include "NavFourMom/INav4MomLink.h"
#include "EventKernel/INavigable4Momentum.h"
#include "NavFourMom/INavigable4MomentumCollection.h"
 
 
// This defines the new container
class INav4MomLinkContainer : public std::vector< INav4MomLink >
{
};
 
// This defines the ID which is used by StoreGate
CLASS_DEF( INav4MomLinkContainer , 1194917823 , 1 )
 
 
// This defines how to convert an INav4MomLinkContainer to an INavigable4MomentumCollection
class INav4MomLinkContainerToINavigable4MomentumCollectionConverter
  : public SG::CopyConversion<INav4MomLinkContainer,
                              INavigable4MomentumCollection>
{
public:
  virtual void convert ( const INav4MomLinkContainer& src,
                         INavigable4MomentumCollection& dst ) const
  {
    size_t sz = src.size();
    if (dst.size() != sz) {
      dst.clear (SG::VIEW_ELEMENTS);
      dst.reserve (sz);
      for (size_t i = 0; i < sz; i++) {
        const INavigable4Momentum* p = *(src[i]).cptr();
        // FIXME: Ok, since the target ends up recorded in the event
        // store as const.  But should change the interfaces so that
        // we get a ConstDataVector to fill.
        INavigable4Momentum* p_nc ATLAS_THREAD_SAFE = const_cast<INavigable4Momentum*> (p);
        dst.push_back (p_nc);
      }
    }
  }
};
 
// This tells StoreGate about the conversion
SG_ADD_COPY_CONVERSION (INav4MomLinkContainer, INav4MomLinkContainerToINavigable4MomentumCollectionConverter);
 
 
#endif // NAVFOURMOM_INAV4MOMLINKCONTAINER_H