CSCComparatorDigiFitter.h

Go to the documentation of this file.

#ifndef L1Trigger_CSCTriggerPrimitives_CSCComparatorDigiFitter_h
#define L1Trigger_CSCTriggerPrimitives_CSCComparatorDigiFitter_h

/*
 * class CSCComparatorDigiFitter
 *
 * Fits a straight line to the comparator digis beloning to a stub
 * 
 * This is a helper class, to be used in the first prototype 
 * implementation of the displaced muon trigger. In due time, 
 * the fitting procedure will be integrated in the
 * CSCCathodeLCTProcessor.
 */

#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "FWCore/Utilities/interface/InputTag.h"

#include "Geometry/CSCGeometry/interface/CSCGeometry.h"
#include "Geometry/CSCGeometry/interface/CSCLayerGeometry.h"

#include "DataFormats/MuonDetId/interface/CSCDetId.h"
#include "DataFormats/GeometryVector/interface/GlobalPoint.h"
#include "DataFormats/CSCDigi/interface/CSCComparatorDigiCollection.h"
#include "DataFormats/CSCDigi/interface/CSCCorrelatedLCTDigiCollection.h"

#include <vector>

typedef std::vector<CSCComparatorDigi> CSCComparatorDigiContainer;
typedef std::vector<std::pair<CSCDetId, CSCComparatorDigiContainer> > CSCComparatorDigiContainerIds;

class CSCComparatorDigiFitter
{
 public:

  CSCComparatorDigiFitter() {}
  ~CSCComparatorDigiFitter() {}

  /* CSC trigger geometry */
  void setGeometry(const CSCGeometry* csc_g) {cscGeometry_= csc_g;}

  /* option to discretize the fitted stub phi */
  void setStripBits(int bits) {nStripBits_ = bits;}

  /* use key layer radius */
  void useKeyRadius(bool useKeyRadius) {useKeyRadius_ = useKeyRadius;}
  
  /* fit a straight line to the digis */
  void fit(const CSCDetId& ch_id, const CSCCorrelatedLCTDigi&,
       const CSCComparatorDigiCollection&,
       std::vector<float>& fit_phi_layers,
       std::vector<float>& fit_z_layers, float& keyRadius);
  
 private:
  
  /* collect the comparator digis that match the pattern */
  void matchingComparatorDigisLCT(const CSCDetId& ch_id, const CSCCorrelatedLCTDigi&, const CSCComparatorDigiCollection&);
  
  /* collect the coordinates of comparators */
  void getComparatorDigiCoordinates(const CSCDetId& ch_id, const CSCCorrelatedLCTDigi& stub);
  
  /* is this comparator in the LCT pattern? */
  bool comparatorInLCTPattern(int keyStrip, int pattern, int layer, int halfStrip) const;

  // calculate slope and intercept of fit
  void calculateSlopeIntercept(float& alpha, float& beta);
  
  /* width of the CSC half strips in this detId  */
  float cscHalfStripWidth(const CSCDetId& id) const;

  const CSCGeometry* cscGeometry_;
  int nStripBits_;

  CSCComparatorDigiContainerIds compDigisIds_;
  std::vector<float> phis_;
  std::vector<float> zs_;
  std::vector<float> ephis_;
  std::vector<float> ezs_;
  float radius_;
  bool useKeyRadius_;

  // number of strips and chamber width for each chamber type
  // ME1a ME1b ME12 ME13 ME21 ME22 ME31 ME32 ME41 ME42
  const std::vector<int> strips_ = {48,64,80,64, 80,80,80,80,80,80};
  const std::vector<float> degrees_ = {10.,10.,10.,10.,20.,10.,20.,10.,20.,10.};
};

#endif