BPhysAddMuonBasedInvMass.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2017 CERN for the benefit of the ATLAS collaboration
*/
 
//
#ifndef DERIVATIONFRAMEWORK_BPhysAddMuonBasedInvMass_H
#define DERIVATIONFRAMEWORK_BPhysAddMuonBasedInvMass_H
 
#include <string>
#include <vector>
#include <set>
#include <map>
 
#include "AthenaBaseComps/AthAlgTool.h"
#include "DerivationFrameworkInterfaces/IAugmentationTool.h"
#include "GaudiKernel/ToolHandle.h"
#include "xAODBPhys/BPhysHelper.h"
#include "EventPrimitives/EventPrimitives.h"
#include "ITrackToVertex/ITrackToVertex.h"
 
namespace DerivationFramework {
  //
  // typedefs -- to abbreviate long lines
  //
  typedef std::vector<const xAOD::TrackParticle*> TrackBag;
  typedef std::vector<const xAOD::Muon*>          MuonBag;
  struct BasedInvCache;
  class BPhysAddMuonBasedInvMass : public extends<AthAlgTool, IAugmentationTool> {
 
  public:
      BPhysAddMuonBasedInvMass(const std::string& t, const std::string& n,
                   const IInterface* p);
      
      virtual StatusCode initialize();
      virtual StatusCode finalize();
      virtual StatusCode addBranches() const;
 
  protected:
      std::pair<double, double> getMuCalcMass(xAOD::BPhysHelper& vtx,
                          const std::vector<double>&
                          trkMasses,
                          int nMuRequested, BasedInvCache &cache) const;
      std::pair<TrackBag, int> getTracksWithMuons(xAOD::BPhysHelper& vtx, BasedInvCache&) const;
      std::pair<double,double>
    getInvariantMassWithError(TrackBag trksIn,
                  const std::vector<double> &massHypotheses,
                  const Amg::Vector3D& pos, BasedInvCache&) const;
      //         any primary vertex.
      double getMinChi2ToAnyPV(xAOD::BPhysHelper& vtx,
                   const xAOD::VertexContainer* pvContainer,
                   const std::vector<int>& pvtypes,
                   const int minNTracksInPV,
                   const int mode,
                   const xAOD::BPhysHelper::pv_type&
                   pvAssocType,
             BasedInvCache &cache) const;
      double getTrackPVChi2(const xAOD::TrackParticle& track,
                const Amg::Vector3D& pos, BasedInvCache &cache) const;
      AmgSymMatrix(3) getMomentumCov(const xAOD::TrackParticle* track) const;
      AmgSymMatrix(3) getMomentumCov(const Trk::Perigee* perigee) const;
      AmgSymMatrix(3) getMomentumCov(const AmgVector(5)& pars,
                     const AmgSymMatrix(5)& cov) const;
      MuonBag findAllMuonsInDecay(xAOD::BPhysHelper& vtx) const;
      TrackBag getIdTracksForMuons(MuonBag& muons) const;
      const xAOD::TrackParticle* adjustTrackParticle(const xAOD::Muon* muon, BasedInvCache &cache)
    const;
      void initPvAssocTypeVec();
  private:      
      std::string                      m_branchPrefix;
      std::string                      m_vertexContainerName;
      std::vector<double>              m_trkMasses;
      ToolHandle<Reco::ITrackToVertex> m_trackToVertexTool;
      bool                             m_adjustToMuonKinematics;
      int                              m_addMinChi2ToAnyPVMode;
      std::string                      m_pvContainerName;
      int                              m_minNTracksInPV;
      std::vector<int>                 m_pvTypesToConsider;
      int                              m_doVertexType;
 
      std::vector<xAOD::BPhysHelper::pv_type> m_pvAssocTypes;
 
  }; // class
} // namespace
 
#endif // DERIVATIONFRAMEWORK_BPhysAddMuonBasedInvMass_H