MuonPRDSelectionTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2021 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef MUON_MUONPRDSELECTIONTOOL_H
#define MUON_MUONPRDSELECTIONTOOL_H
 
#include "MuonRecToolInterfaces/IMuonPRDSelectionTool.h"
#include "AthenaBaseComps/AthAlgTool.h"
#include "GaudiKernel/ServiceHandle.h"
#include "GaudiKernel/ToolHandle.h"
 
#include "MuonLayerEvent/MuonSystemExtension.h"
#include "MuonLayerEvent/MuonLayerPrepRawData.h"
#include "MuonLayerEvent/MuonLayerROTs.h"
#include "MuonIdHelpers/IMuonIdHelperSvc.h"
#include "MuonRIO_OnTrack/MuonClusterOnTrack.h"
#include "MuonRecToolInterfaces/IMdtDriftCircleOnTrackCreator.h"
#include "MuonRecToolInterfaces/IMuonClusterOnTrackCreator.h"
#include "MuonPrepRawData/MdtPrepDataCollection.h"
 
#include <vector>
#include <string>
 
namespace Muon {
 
  class MuonPRDSelectionTool : virtual public Muon::IMuonPRDSelectionTool,  public AthAlgTool {
  public:
 
    MuonPRDSelectionTool(const std::string& type, const std::string& name, const IInterface* parent);
    virtual ~MuonPRDSelectionTool()=default;
    StatusCode initialize();
 
    bool calibrateAndSelect( const MuonSystemExtension::Intersection& intersection, const MuonLayerPrepRawData& layerPrepRawData, MuonLayerROTs& layerROTs ) const;
 
    bool calibrateAndSelectMdt( const MuonSystemExtension::Intersection& intersection, const MdtPrepDataCollection& prds, std::vector<const MdtDriftCircleOnTrack*>& rots ) const {
      for( MdtPrepDataCollection::const_iterator it = prds.begin(); it != prds.end();++it ){
        const MdtDriftCircleOnTrack* rot = calibrateAndSelect(intersection,**it);
        if( rot ) rots.push_back(rot);
      }
      return true;
    }
 
    template< class COL >
    bool calibrateAndSelectCluster( const MuonSystemExtension::Intersection& intersection, const COL& prds, std::vector<const MuonClusterOnTrack*>& rots ) const {
      for( typename COL::const_iterator it = prds.begin(); it != prds.end();++it ){
        const MuonClusterOnTrack* rot = calibrateAndSelect(intersection,**it);
        if( rot ) rots.push_back(rot);
      }
      return true;
    }
    
    const MdtDriftCircleOnTrack* calibrateAndSelect( const MuonSystemExtension::Intersection& intersection, const MdtPrepData& mdt ) const;
 
    const MuonClusterOnTrack* calibrateAndSelect( const Trk::TrackParameters& pars, const MuonCluster& clus ) const;
    const MuonClusterOnTrack* calibrateAndSelect( const MuonSystemExtension::Intersection& intersection, const MuonCluster& clus ) const;
 
 
  private:
    Amg::Vector3D intersectMDT( const MdtPrepData& mdt, const Amg::Vector3D& position, const Amg::Vector3D& direction, bool usePlane ) const;
 
 
    bool calibrateAndSelectMdt( const MuonSystemExtension::Intersection& intersection, const MdtPrepDataCollection& prds, MuonLayerROTs& layerROTs ) const {
      std::vector<const MdtDriftCircleOnTrack*> rots;
      calibrateAndSelectMdt(intersection,prds,rots);
      if( !rots.empty() ) layerROTs.addMdts(rots);
      return true;
    }
 
    template< class COL >
    bool calibrateAndSelectCluster( const MuonSystemExtension::Intersection& intersection, const COL& prds, MuonLayerROTs& layerROTs ) const {
      std::vector<const MuonClusterOnTrack*> rots;
      calibrateAndSelectCluster(intersection,prds,rots);
      if( !rots.empty() ) layerROTs.addClusters(rots,m_idHelperSvc->technologyIndex(rots.front()->identify()));
      return true;
    }
    
    ServiceHandle<Muon::IMuonIdHelperSvc> m_idHelperSvc {this, "MuonIdHelperSvc", "Muon::MuonIdHelperSvc/MuonIdHelperSvc"};
    ToolHandle<IMdtDriftCircleOnTrackCreator> m_mdtCreator{this,"MdtDriftCircleOnTrackCreator","Muon::MdtDriftCircleOnTrackCreator/MdtDriftCircleOnTrackCreator"};
    ToolHandle<IMuonClusterOnTrackCreator> m_clusterCreator{this,"MuonClusterOnTrackCreator","Muon::MuonClusterOnTrackCreator/MuonClusterOnTrackCreator"};
 
    double m_distanceToTubeCut;
    double m_secondCoordinateCut;
 
  };
}
 
 
 
#endif