MissingETAssociationMap_v1.h

Go to the documentation of this file.

// -*- c++ -*-
 
/*
  Copyright (C) 2002-2021 CERN for the benefit of the ATLAS collaboration
*/
 
#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
{
 
  // Forward declarations
    class MissingETAssociation_v1;
    class MissingETAssociationHelper;
 
  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(const MissingETAssociationMap_v1::iterator& first,
                               const 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(const ElementLink<IParticleContainer>& constLink) const;
    iterator       findByJetConst(const ElementLink<IParticleContainer>& constLink);
    size_t findIndexByJetConst(const IParticle* pConst) const;
    size_t findIndexByJetConst(const ElementLink<IParticleContainer>& pConstLink) const;
    const MissingETAssociation_v1* getMiscAssociation() const;
    MissingETAssociation_v1* getMiscAssociation();

    bool setJetConstituents(const Jet* pJet);
    bool setJetConstituents(const std::vector<ElementLink<IParticleContainer> >& constLinks, const Jet* pJet);
    bool setJetConstituents(const std::vector<ElementLink<IParticleContainer> >& constLinks, size_t jetIndex);
    bool identifyOverlaps();
    void clearOverlaps();
    const IParticleContainer* getUniqueSignals(const IParticleContainer* signals,MissingETBase::UsageHandler::Policy p=MissingETBase::UsageHandler::TrackCluster) const;
    const IParticleContainer* getOverlapRemovedSignals(const MissingETAssociationHelper& helper,const IParticleContainer* signals,MissingETBase::UsageHandler::Policy p=MissingETBase::UsageHandler::TrackCluster) const;
 
  protected:
 
    void f_setJetConstMap(std::map<ElementLink<IParticleContainer>, size_t> map);
    void f_clearJetConstMap();
    std::map<ElementLink<IParticleContainer>, size_t> m_jetConstLinks;  

    const_iterator f_findConst(const Jet* pJet) const;
    iterator f_find(const Jet* pJet);
 
 
  }; // class xAOD::MissingETAssociationMap_v1
} // namespace xAOD
 
#include "xAODCore/BaseInfo.h"
SG_BASE(xAOD::MissingETAssociationMap_v1, DataVector<xAOD::MissingETAssociation_v1>);
 
#endif