/cvmfs/atlas.cern.ch/repo/sw/ASG/AnalysisBase/2.4.31/xAODMissingET/xAODMissingET/versions/MissingETAssociationMap_v1.h

// -*- c++ -*-
#ifndef XAODMISSINGET_VERSIONS_MISSINGETASSOCIATIONMAP_V1_H
#define XAODMISSINGET_VERSIONS_MISSINGETASSOCIATIONMAP_V1_H

#include "AthContainers/DataVector.h"

#include "xAODJet/Jet.h"
#include "xAODMissingET/versions/MissingETAssociation_v1.h"

#include <string>
#include <vector>
#include <map>

#include <iostream>

namespace xAOD
{

  class MissingETAssociationMap_v1 : public DataVector<MissingETAssociation_v1>
  {
  public:

    MissingETAssociationMap_v1(SG::OwnershipPolicy own=SG::OWN_ELEMENTS, SG::IndexTrackingPolicy trackIndices = SG::DEFAULT_TRACK_INDICES);
    MissingETAssociationMap_v1(MissingETAssociationMap_v1::iterator first, MissingETAssociationMap_v1::iterator last,
                 SG::OwnershipPolicy own=SG::OWN_ELEMENTS, SG::IndexTrackingPolicy trackIndices = SG::DEFAULT_TRACK_INDICES);
    virtual ~MissingETAssociationMap_v1();

    const_iterator find(const Jet* pJet)   const;
    iterator find(const Jet* pJet);
    size_t   findIndex(const Jet* pJet) const;
    //
    const_iterator findByJetConst(const IParticle* pConst) const;
    iterator       findByJetConst(const IParticle* pConst);
    const_iterator findByJetConst(ElementLink<IParticleContainer> constLink) const;
    iterator       findByJetConst(ElementLink<IParticleContainer> constLink);
    size_t findIndexByJetConst(const IParticle* pConst) const;
    size_t findIndexByJetConst(ElementLink<IParticleContainer> pConstLink) const;
    const MissingETAssociation_v1* getMiscAssociation() const;
    MissingETAssociation_v1* getMiscAssociation();
    bool setJetConstituents(const Jet* pJet) const;
    bool setJetConstituents(const std::vector<ElementLink<IParticleContainer> >& constLinks, const Jet* pJet) const;
    bool setJetConstituents(const std::vector<ElementLink<IParticleContainer> >& constLinks, size_t jetIndex) const;
    bool identifyOverlaps();
    void clearOverlaps();
    const IParticleContainer* getUniqueSignals(const IParticleContainer* signals,MissingETBase::UsageHandler::Policy p=MissingETBase::UsageHandler::TrackCluster) const;
    const IParticleContainer* getOverlapRemovedSignals(const IParticleContainer* signals,MissingETBase::UsageHandler::Policy p=MissingETBase::UsageHandler::TrackCluster) const;
    /* override base DataVector methods that can remove elements
     * in order to invalidate the cache
     */
    void resize(size_type sz);
    void pop_back();
    void clear();
    void sort();
    template <class COMPARE> void sort(COMPARE comp);
    void clear (SG::OwnershipPolicy ownPolicy);
    void clear (SG::OwnershipPolicy ownPolicy,SG::IndexTrackingPolicy trackIndices);
    iterator erase(iterator position);
    iterator erase(iterator first, iterator last);

    void resetObjSelectionFlags() const;
    void resetAssocCache() const;

  protected:

    void f_setJetConstMap(std::map<ElementLink<IParticleContainer>, size_t> map);
    void f_clearJetConstMap();
    mutable std::map<ElementLink<IParticleContainer>, size_t> m_jetConstLinks;  
    const_iterator f_findConst(const Jet* pJet) const;
    void f_setConstCache(const_iterator fCont) const;

    iterator f_find(const Jet* pJet);
    void f_setCache(iterator fCont);

    void resetCache() const;

  private:

    mutable const Jet* m_lastRefJet;
    mutable size_t m_lastContribIndex;
    mutable ElementLink<IParticleContainer> m_lastConstLink;
    mutable size_t m_miscAssocIndex;

    static size_t m_jetConstLinkReserve;

  }; // class xAOD::MissingETAssociationMap_v1
} // namespace xAOD

#include "xAODCore/BaseInfo.h"
SG_BASE(xAOD::MissingETAssociationMap_v1, DataVector<xAOD::MissingETAssociation_v1>);

#include "xAODMissingET/versions/MissingETAssociationMap_v1.icc"
#endif

---