MuonSeedCleaner.h

Go to the documentation of this file.

#ifndef RecoMuon_MuonSeedCleaner_H
#define RecoMuon_MuonSeedCleaner_H

#include <FWCore/ParameterSet/interface/ParameterSet.h>
#include <RecoMuon/MeasurementDet/interface/MuonDetLayerMeasurements.h>
#include <DataFormats/TrajectorySeed/interface/TrajectorySeedCollection.h>

//muon service
#include <RecoMuon/TrackingTools/interface/MuonServiceProxy.h>

#include<vector>

class DetLayer; 
class MuonDetLayerGeometry;
class MagneticField;

typedef std::vector<TrajectorySeed> SeedContainer;

class MuonSeedCleaner
{

 public:

  typedef MuonTransientTrackingRecHit::MuonRecHitContainer SegmentContainer;
  typedef std::deque<bool> BoolContainer;

  explicit MuonSeedCleaner(const edm::ParameterSet&);
  
  ~MuonSeedCleaner();
  
  // Operations

  //void setGeometry( const MuonDetLayerGeometry* lgeom ) {muonLayers = lgeom;}

  //void setBField( const MagneticField* theField ) {BField = theField;}

  std::vector<TrajectorySeed> seedCleaner(const edm::EventSetup& eventSetup, std::vector<TrajectorySeed>& seeds );   

  int NRecHitsFromSegment( const TrackingRecHit& rhit );
  int NRecHitsFromSegment( MuonTransientTrackingRecHit *rhit );
  //int NRecHitsFromSegment( const MuonTransientTrackingRecHit& rhit );

  std::vector<int> badSeedLayer;


 private:


  std::vector<SeedContainer> GroupSeeds( std::vector<TrajectorySeed>& seeds );

  TrajectorySeed Chi2LengthSelection(std::vector<TrajectorySeed>& seeds );
  TrajectorySeed LeanHighMomentum( std::vector<TrajectorySeed>& seeds ); 
  TrajectorySeed BiggerCone( std::vector<TrajectorySeed>& seeds ) ;
  TrajectorySeed MoreRecHits( std::vector<TrajectorySeed>& seeds );

  bool MomentumFilter(std::vector<TrajectorySeed>& seeds );
  SeedContainer SeedCandidates( std::vector<TrajectorySeed>& seeds, bool good );
  unsigned int OverlapSegments( const TrajectorySeed& seed1, const TrajectorySeed& seed2 );

  double NChi2OfSegment( const TrackingRecHit& rhit );

  double SeedChi2( const TrajectorySeed& seed );
  int    SeedLength( const TrajectorySeed& seed ); 

  GlobalPoint SeedPosition( const TrajectorySeed& seed );
  GlobalVector SeedMomentum( const TrajectorySeed& seed );

  // obsoleted functions
  SeedContainer LengthFilter(std::vector<TrajectorySeed>& seeds );


  // This Producer private debug flag
  bool debug;

  // Number of Segments from a shower
  int NShowerSeg;
  SegmentContainer ShoweringSegments;   
  std::vector<int> ShoweringLayers; 

  // Cache geometry for current event
  //const MuonDetLayerGeometry* muonLayers;

  // Cache Magnetic Field for current event
  const MagneticField* BField;
 
  // muon service
  MuonServiceProxy* theService;

  // Minimum separation when we can distinguish between 2 muon seeds
  // (to suppress combinatorics)

};
#endif