TrackConstituents.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2019 CERN for the benefit of the ATLAS collaboration
*/
 
/***************************************************************************
                           TrackConstituent.h  -  Description
                             -------------------
    begin   : January, 2005
    authors : Kyle Cranmer, Andreas Wildauer
    email   : cranmer@cern.ch, andreas.wildauer@cern.ch
 
    changes : 
 
***************************************************************************/
 
#ifndef ANALYSIS_TRACKCONSTITUENT_H
#define ANALYSIS_TRACKCONSTITUENT_H
 
#include<vector>
#include<string>
 
#include "Navigation/Navigable.h"
#include "Particle/TrackParticleContainer.h"
#include "ParticleEvent/IConstituent.h"
//class INavigable4MomentumCollection;
//class INavigable4Momentum;
 
namespace Analysis
{
 
  class TrackConstituents : public IConstituent,
                            public Navigable<Rec::TrackParticleContainer, double>
    {
      typedef Navigable<Rec::TrackParticleContainer,double>::external_index_type index_type;
      typedef Navigable<Rec::TrackParticleContainer,double>::constituent_iter  iter_type;
      //      typedef object_iter  iterator;
      
    public:
      TrackConstituents() ;      
      TrackConstituents(NameType& name) ;      
      ~TrackConstituents() = default;
 
      typedef Navigable<Rec::TrackParticleContainer,double>::object_iter  iterator;
 
 
      virtual NameType name() const;
 
      virtual IConstituent* clone() const; 
 
      //iterator begin(); available via Navigable with (*iterator) giving TrackParticle*
      //iterator end(); available via Navigable

      std::vector<const Rec::TrackParticle*>* tracks() const;
      
      unsigned int nTracks() const { return size() ;    }
 
      double getTrackWeight(const Rec::TrackParticle* ) const;
      
      double getTrackWeight(const Rec::TrackParticleContainer*, index_type& ) const;
      
      // Set Methods
 
      virtual void setName(NameType& name);

      void set_track(const Rec::TrackParticleContainer*,const Rec::TrackParticle*, 
             double weight=0.);
      void set_track(const Rec::TrackParticleContainer*, index_type&, double weight=0.);
 
      // LACK OF KNOWLEDGE HACK HERE
      // use also INavigable4MomentumCollection and INavigable4Momentum to add constituents
      // would be nice to have a addConstituents method in the IConstituent interface (dont know if that is possible)
//       void set_track(const INavigable4MomentumCollection*,const INavigable4Momentum*, 
//           double weight=0.);
 
      //compromise for above... use concrete container, but common name for templated methods
      void set_constituent(const Rec::TrackParticleContainer*,const Rec::TrackParticle*, 
             double weight=0.);
      void set_constituent(const Rec::TrackParticleContainer*, index_type&, double weight=0.);
      
 
 
    private:
      std::string m_name;      
 
    }
  ;
  
}
 
 
#endif