LxTBTrdTask.h

Go to the documentation of this file.

/*
 * To change this license header, choose License Headers in Project Properties.
 * To change this template file, choose Tools | Templates
 * and open the template in the editor.
 */
 
/* 
 * File:   LxTBTrdTask.h
 * Author: tablyaz
 *
 * Created on October 4, 2016, 6:43 PM
 */
 
#ifndef LXTBTRDTASK_H
#define LXTBTRDTASK_H
 
#include "CbmMCDataArray.h"
#include "FairTask.h"
#include "LxTBBinned.h"
 
#define CUR_TIMEBIN_LENGTH 100
#define CUR_NOF_STATIONS 4
#define CUR_LAST_STATION CUR_NOF_STATIONS - 1
 
class LxTBTrdFinder : public FairTask {
public:
#ifdef LXTB_QA
  struct TrackDataHolder {
    Int_t pointInds[CUR_NOF_STATIONS];
    bool isSignal;
    Int_t pdg;
    Double_t z;
 
    TrackDataHolder() : isSignal(false), pdg(-1), z(0) {
      for (int i = 0; i < CUR_NOF_STATIONS; ++i)
        pointInds[i] = -1;
    }
  };
 
  struct PointDataHolder {
    Double_t x;
    Double_t y;
    Double_t z;
    Double_t t;
    Int_t eventId;
    Int_t trackId;
    Int_t pointId;
    Int_t layerNumber;
  };
#endif  //LXTB_QA
 
public:
  LxTBTrdFinder();
  LxTBTrdFinder(const LxTBTrdFinder&) = delete;
  LxTBTrdFinder& operator=(const LxTBTrdFinder&) = delete;
 
  InitStatus Init();         // Overridden from FairTask
  void Exec(Option_t* opt);  // Overridden from FairTask
  void Finish();             // Overridden from FairTask
 
  void SetTrigDistance(Double_t v) { fTrigDistance = v; }
 
#ifdef LXTB_QA
  void SetNEvents(Int_t v) { fNEvents = v; }
#endif  //LXTB_QA
 
private:
  void HandleGeometry();
  LxTbBinnedFinder* fFinder;
  Double_t fTrigDistance;
  std::list<LxTbBinnedFinder::Chain*> recoTracks;
  unsigned int nof_timebins;
  unsigned int last_timebin;
  TClonesArray* fTrdHits;
  TClonesArray* fTrdClusters;
  TClonesArray* fTrdDigiMatches;
  TClonesArray* fTrdTracks;
  TClonesArray* fGlobalTracks;
#ifdef LXTB_QA
  CbmMCDataArray* fTrdMCPoints;
  std::vector<std::vector<TrackDataHolder>> fMCTracks;
  std::vector<std::vector<PointDataHolder>> fTrdPoints;
  Int_t fNEvents;
#endif  //LXTB_QA
 
  ClassDef(LxTBTrdFinder, 1)
};
 
#endif /* LXTBTRDTASK_H */