CbmL1TrdTrackFinderSts.h

Go to the documentation of this file.

// -----------------------------------------------------------------------
// -----                    CbmL1TrdTrackFinderSts                   -----
// -----                 Created 6/12/05  by D. Kresan               -----
// -----------------------------------------------------------------------
 
#ifndef CBM_L1_TRD_TRACK_FINDER_STS
#define CBM_L1_TRD_TRACK_FINDER_STS
 
#include "CbmTrdTrackFinder.h"
 
#include "CbmTrdTrack.h"
 
#include <map>
#include <vector>
 
class TClonesArray;
class CbmTrdHit;
class CbmKFTrack;
class TH1F;
class TH2F;
 
 
class CbmL1TrdTrackFinderSts : public CbmTrdTrackFinder {
private:
  CbmL1TrdTrackFinderSts(const CbmL1TrdTrackFinderSts&);
  CbmL1TrdTrackFinderSts operator=(const CbmL1TrdTrackFinderSts&);
 
private:
  Int_t fEvents;                         // Number of events processed
  Int_t fVerbose;                        // Verbosity level
  TClonesArray* fArrayTrdPoint;          // Array of TRD points
  TClonesArray* fArrayTrdHit;            // Array of TRD hits
  TClonesArray* fArrayStsTrack;          // Array of STS tracks
  TClonesArray* fArrayStsTrackM;         // Array of STS tracks
  TClonesArray* fArrayKFTrdHit;          // Array of KF TRD hits
  std::vector<CbmTrdTrack*> fvTrdTrack;  // Working array of TRD tracks
  TClonesArray* fArrayTrdTrack;          // Output Array of TRD tracks
  Int_t fPid;                            // PID assumption
  Int_t fNoTrdStations;                  // Number of TRD stations
  Int_t fNoTrdPerStation;                // Number of TRD layers per station
  Int_t fNoTrdHits[12];                  // Number of TRD hits per station
  Int_t fTrdHitIndex[12][10000];         // Indices of TRD hits per station
  std::map<Int_t, Bool_t> fmapHitUsed;   // Map from hit index to boolean flag
  TH1F* fh_chi2hit;                      // Control histogramm
  TH2F* fh_chi2hit_plane;                // Control histogramm
  TH2F* fh_resx_plane_true;              // Control histogramm
  TH2F* fh_resy_plane_true;              // Control histogramm
  TH2F* fh_resx_plane_fake;              // Control histogramm
  TH2F* fh_resy_plane_fake;              // Control histogramm
  TH2F* fh_resx_mom_true;                // Control histogramm
  TH2F* fh_resy_mom_true;                // Control histogramm
  TH2F* fh_pullx_plane_true;             // Control histogramm
  TH2F* fh_pully_plane_true;             // Control histogramm
  TH2F* fh_pullx_plane_fake;             // Control histogramm
  TH2F* fh_pully_plane_fake;             // Control histogramm
  std::map<Int_t, Bool_t> fLostTracks;
 
  void CreateHistogramms();
  void DataBranches();
  void TrdLayout();
  void SortTrdHits();
  void Process();
  void
  Sts2Trd(Double_t pmin, Double_t pmax, Double_t chi2min, Double_t chi2max);
  void ProcessAllStations();
  void MoveOut();
  void ProcessStation(CbmTrdTrack* pTrack, const Int_t& station);
  void UpdateTrack(Int_t station, CbmTrdTrack* track);
  virtual void Clear(const Option_t* a = 0);
  void RemoveFakes();
  Bool_t Overlap(CbmKFTrack& track, CbmTrdHit* pHit);
  Double_t GetChi2Hit(CbmKFTrack& track, CbmTrdHit* pHit);
 
public:
  CbmL1TrdTrackFinderSts();
  CbmL1TrdTrackFinderSts(Int_t verbose);
  virtual ~CbmL1TrdTrackFinderSts();
 
  void Init();
  Int_t DoFind(TClonesArray* hitArray, TClonesArray* trackArray);
  void WriteHistogramms();
 
  inline void SetVerbose(Int_t verbose) { fVerbose = verbose; }
 
  static Bool_t CompareChi2(const CbmTrdTrack* a, const CbmTrdTrack* b) {
    return (a->GetChiSq() / static_cast<Double_t>(a->GetNDF())
            < b->GetChiSq() / static_cast<Double_t>(b->GetNDF()));
  };
 
  ClassDef(CbmL1TrdTrackFinderSts, 1);
};
 
 
#endif