StandAloneMuonFilter.h

Go to the documentation of this file.

#ifndef RecoMuon_StandAloneTrackFinder_StandAloneMuonFilter_H
#define RecoMuon_StandAloneTrackFinder_StandAloneMuonFilter_H

#include "TrackingTools/TrajectoryState/interface/TrajectoryStateOnSurface.h"
#include "TrackingTools/TrajectoryState/interface/FreeTrajectoryState.h"
#include "TrackingTools/DetLayers/interface/NavigationDirection.h"
#include "DataFormats/TrajectorySeed/interface/PropagationDirection.h"

#include "RecoMuon/TrackingTools/interface/MuonBestMeasurementFinder.h"
#include "FWCore/Framework/interface/ConsumesCollector.h"

class Propagator;
class DetLayer;
class MuonTrajectoryUpdator;
class Trajectory;
class MuonDetLayerMeasurements;
class MeasurementEstimator;
class MuonServiceProxy;

namespace edm {class ParameterSet; class EventSetup; class Event;}

class StandAloneMuonFilter {

 public:
  StandAloneMuonFilter(const edm::ParameterSet& par, const MuonServiceProxy* service,edm::ConsumesCollector& iC);

  virtual ~StandAloneMuonFilter();

  // Operations
  
  void refit(const TrajectoryStateOnSurface& initialState, const DetLayer*, Trajectory& trajectory);

  FreeTrajectoryState lastUpdatedFTS() const {return *theLastUpdatedTSOS.freeTrajectoryState();}

  FreeTrajectoryState lastButOneUpdatedFTS() const {return *theLastButOneUpdatedTSOS.freeTrajectoryState();}
  
  TrajectoryStateOnSurface lastCompatibleTSOS() const {return theLastCompatibleTSOS;}

  TrajectoryStateOnSurface lastUpdatedTSOS() const {return theLastUpdatedTSOS;}

  TrajectoryStateOnSurface lastButOneUpdatedTSOS() const {return theLastButOneUpdatedTSOS;}

  void reset();

  virtual void setEvent(const edm::Event& event);

  int getTotalChamberUsed() const {return totalChambers;}
  int getDTChamberUsed() const {return dtChambers;}
  int getCSCChamberUsed() const {return cscChambers;}
  int getRPCChamberUsed() const {return rpcChambers;}
  int getGEMChamberUsed() const {return gemChambers;}

  int getTotalCompatibleChambers() const {return totalCompatibleChambers;}
  int getDTCompatibleChambers() const {return dtCompatibleChambers;}
  int getCSCCompatibleChambers() const {return cscCompatibleChambers;}
  int getRPCCompatibleChambers() const {return rpcCompatibleChambers;}
  int getGEMCompatibleChambers() const {return gemCompatibleChambers;}

  inline bool goodState() const {return totalChambers >= 2 && 
                   ((dtChambers + cscChambers + gemChambers) >0 ||
                    onlyRPC());}
  
  inline bool isCompatibilitySatisfied() const {return totalCompatibleChambers >= 2 && 
                          ((dtCompatibleChambers + cscCompatibleChambers + gemCompatibleChambers) >0 ||
                           onlyRPC());}
  
  std::vector<const DetLayer*> layers() const {return theDetLayers;}

  const DetLayer* lastDetLayer() const {return theDetLayers.back();}

  PropagationDirection propagationDirection() const;

  NavigationDirection fitDirection() const {return theFitDirection;}
  
  bool onlyRPC() const {return theRPCLoneliness;}

  MeasurementEstimator *estimator() const {return theEstimator;}
  
  const Propagator *propagator() const;
  
  MuonTrajectoryUpdator *updator() const {return theMuonUpdator;}

  void createDefaultTrajectory(const Trajectory &, Trajectory &);


protected:

private:

  void setLastCompatibleTSOS(TrajectoryStateOnSurface tsos) { theLastCompatibleTSOS = tsos;}
  
  void setLastUpdatedTSOS(TrajectoryStateOnSurface tsos) { theLastUpdatedTSOS = tsos;}
  
  void setLastButOneUpdatedTSOS(TrajectoryStateOnSurface tsos) { theLastButOneUpdatedTSOS = tsos;}
  
  void incrementChamberCounters(const DetLayer *layer);
  void incrementCompatibleChamberCounters(const DetLayer *layer);
 
  std::vector<const DetLayer*> compatibleLayers(const DetLayer *initialLayer,
                        const FreeTrajectoryState& fts,
                        PropagationDirection propDir);

  bool update(const DetLayer * layer, 
              const TrajectoryMeasurement * meas,
              Trajectory & trajectory);

  std::vector<TrajectoryMeasurement>
  findBestMeasurements(const DetLayer * layer, const TrajectoryStateOnSurface & tsos);
  
  TrajectoryStateOnSurface theLastCompatibleTSOS;
  TrajectoryStateOnSurface theLastUpdatedTSOS;
  TrajectoryStateOnSurface theLastButOneUpdatedTSOS;

  MuonDetLayerMeasurements *theMeasurementExtractor;
  
  MeasurementEstimator *theEstimator;

  MuonTrajectoryUpdator *theMuonUpdator;
  std::string theMuonUpdatorName;
  
  MuonBestMeasurementFinder *theBestMeasurementFinder;
  MuonBestMeasurementFinder *bestMeasurementFinder() const {return theBestMeasurementFinder;}

  double theMaxChi2;
  double theNSigma;

  NavigationDirection theFitDirection;

  std::vector<const DetLayer*> theDetLayers;

  std::string thePropagatorName;

  std::string theNavigationType;

  bool theRPCLoneliness;

  int totalChambers;
  int dtChambers;
  int cscChambers;
  int rpcChambers;
  int gemChambers;

  int totalCompatibleChambers;
  int dtCompatibleChambers;
  int cscCompatibleChambers;
  int rpcCompatibleChambers;
  int gemCompatibleChambers;
 
  const MuonServiceProxy *theService;
  bool theOverlappingChambersFlag;
};
#endif