/data/refman/pasoursint/CMSSW_5_3_1/src/TrackingTools/TrackAssociator/interface/TrackDetectorAssociator.h

Go to the documentation of this file.

#ifndef TrackAssociator_TrackDetectorAssociator_h
#define TrackAssociator_TrackDetectorAssociator_h 1

// -*- C++ -*-
//
// Package:    TrackAssociator
// Class:      TrackDetectorAssociator
// 
/*

 Description: main class of tools to associate a track to calorimeter and muon detectors

*/
//
// Original Author:  Dmytro Kovalskyi
//         Created:  Fri Apr 21 10:59:41 PDT 2006
// $Id: TrackDetectorAssociator.h,v 1.21 2011/04/07 09:02:44 innocent Exp $
//
//

#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "DataFormats/Common/interface/Handle.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/Framework/interface/ESWatcher.h"
#include "DataFormats/Common/interface/OrphanHandle.h"

#include "DataFormats/TrackReco/interface/TrackBase.h"
#include "Geometry/CaloGeometry/interface/CaloGeometry.h"
#include "Geometry/Records/interface/CaloGeometryRecord.h"
#include "TrackingTools/GeomPropagators/interface/Propagator.h"
#include "MagneticField/Records/interface/IdealMagneticFieldRecord.h"

#include "TrackingTools/TrackAssociator/interface/DetIdAssociator.h"
#include "TrackingTools/TrackAssociator/interface/TrackDetMatchInfo.h"
#include "TrackingTools/TrackAssociator/interface/CachedTrajectory.h"

#include "DataFormats/CaloTowers/interface/CaloTower.h"
#include "DataFormats/EcalRecHit/interface/EcalRecHit.h"
#include "DataFormats/TrackReco/interface/Track.h"
#include "DataFormats/TrackingRecHit/interface/RecSegment.h"

#include "SimDataFormats/Track/interface/SimTrack.h"
#include "SimDataFormats/Vertex/interface/SimVertex.h"

#include "TrackingTools/TrackAssociator/interface/TrackAssociatorParameters.h"

#include "FWCore/Utilities/interface/Visibility.h"


class TrackDetectorAssociator {
 public:
   TrackDetectorAssociator();
   ~TrackDetectorAssociator();
   
   typedef TrackAssociatorParameters AssociatorParameters;
   enum Direction { Any, InsideOut, OutsideIn };
   
   TrackDetMatchInfo            associate( const edm::Event&,
                                           const edm::EventSetup&,
                                           const FreeTrajectoryState&,
                                           const AssociatorParameters& );
   TrackDetMatchInfo            associate( const edm::Event& iEvent,
                                           const edm::EventSetup& iSetup,
                                           const AssociatorParameters& parameters,
                                           const FreeTrajectoryState* innerState,
                                           const FreeTrajectoryState* outerState=0);
   TrackDetMatchInfo            associate( const edm::Event&,
                                           const edm::EventSetup&,
                                           const reco::Track&,
                                           const AssociatorParameters&,
                                           Direction direction = Any );
   TrackDetMatchInfo            associate( const edm::Event&,
                                           const edm::EventSetup&,
                                           const SimTrack&,
                                           const SimVertex&,
                                           const AssociatorParameters& );
   TrackDetMatchInfo            associate( const edm::Event&,
                                           const edm::EventSetup&,
                                           const GlobalVector&,
                                           const GlobalPoint&,
                                           const int,
                                           const AssociatorParameters& );
   
   void setPropagator( const Propagator* );
   
   void useDefaultPropagator();
   
   static FreeTrajectoryState getFreeTrajectoryState( const edm::EventSetup&, 
                                                      const reco::Track& );
   static FreeTrajectoryState getFreeTrajectoryState( const edm::EventSetup&, 
                                                      const SimTrack&, 
                                                      const SimVertex& );
   static FreeTrajectoryState getFreeTrajectoryState( const edm::EventSetup&,
                                                      const GlobalVector&,
                                                      const GlobalPoint&,
                                                      const int);
        
   static bool                crossedIP(const reco::Track& track);

 private:
   DetIdAssociator::MapRange getMapRange( const std::pair<float,float>& delta,
                                          const float dR ) dso_internal;

   void fillEcal(       const edm::Event&,
                        TrackDetMatchInfo&, 
                        const AssociatorParameters&) dso_internal;
   
   void fillCaloTowers( const edm::Event&,
                        TrackDetMatchInfo&,
                        const AssociatorParameters&) dso_internal;
   
   void fillHcal(       const edm::Event&,
                        TrackDetMatchInfo&,
                        const AssociatorParameters&) dso_internal;
   
   void fillHO(         const edm::Event&,
                        TrackDetMatchInfo&,
                        const AssociatorParameters&) dso_internal;
  
   void fillPreshower(  const edm::Event& iEvent,
                        TrackDetMatchInfo& info,
                        const AssociatorParameters&) dso_internal;
   
   void fillMuon(       const edm::Event&,
                        TrackDetMatchInfo&,
                        const AssociatorParameters&) dso_internal;
   
   void fillCaloTruth(  const edm::Event&,
                        TrackDetMatchInfo&,
                        const AssociatorParameters&) dso_internal;
   
   bool addTAMuonSegmentMatch(TAMuonChamberMatch&,
                            const RecSegment*,
                            const AssociatorParameters&) dso_internal;
   
   void getTAMuonChamberMatches(std::vector<TAMuonChamberMatch>& matches,
                              const AssociatorParameters& parameters) dso_internal;
  
   void           init( const edm::EventSetup&) dso_internal;
   
   math::XYZPoint getPoint( const GlobalPoint& point)  dso_internal
     {
        return math::XYZPoint(point.x(),point.y(),point.z());
     }
   
   math::XYZPoint getPoint( const LocalPoint& point)  dso_internal
     {
        return math::XYZPoint(point.x(),point.y(),point.z());
     }
   
   math::XYZVector getVector( const GlobalVector& vec)  dso_internal
     {
        return math::XYZVector(vec.x(),vec.y(),vec.z());
     }
   
   math::XYZVector getVector( const LocalVector& vec)  dso_internal
     {
        return math::XYZVector(vec.x(),vec.y(),vec.z());
     }
   
   const Propagator* ivProp_;
   Propagator* defProp_;
   CachedTrajectory cachedTrajectory_;
   bool useDefaultPropagator_;
   
   edm::ESHandle<DetIdAssociator> ecalDetIdAssociator_;
   edm::ESHandle<DetIdAssociator> hcalDetIdAssociator_;
   edm::ESHandle<DetIdAssociator>   hoDetIdAssociator_;
   edm::ESHandle<DetIdAssociator> caloDetIdAssociator_;
   edm::ESHandle<DetIdAssociator> muonDetIdAssociator_;
   edm::ESHandle<DetIdAssociator> preshowerDetIdAssociator_;
   
   edm::ESHandle<CaloGeometry> theCaloGeometry_;
   edm::ESHandle<GlobalTrackingGeometry> theTrackingGeometry_;
   
   edm::ESWatcher<IdealMagneticFieldRecord>     theMagneticFeildWatcher_;
};
#endif