PndDrcLutFill.h

Go to the documentation of this file.

// -----------------------------------------
// PndDrcLutFill.h
//
// Created on: 08.07.2013
// Author: R.Dzhygadlo at gsi.de
// -----------------------------------------
// Class allows to fill look-up tables for reconstruction in DIRC

#ifndef PNDDRCLUTFILL_H
#define PNDDRCLUTFILL_H

#include "FairTask.h"
#include "TClonesArray.h"
#include "PndMCTrack.h"
#include "PndDrcBarPoint.h"
#include "PndDrcPDPoint.h"
#include "PndDrcEVPoint.h"
#include "PndDrcHit.h"
#include "PndDrcPDHit.h"
#include "PndDrcDigi.h"
#include "PndGeoDrc.h"

#include "TString.h"
#include "TFile.h"
#include "TTree.h"

class PndDrcLutFill : public FairTask {

 public:
  // Default constructor
  PndDrcLutFill();

  // Standard constructors
  PndDrcLutFill(int verbose);
  PndDrcLutFill(int verbose, TString outfilename);

  // Destructor
  virtual ~PndDrcLutFill();

  virtual InitStatus Init();

  // Executed task
  virtual void Exec(Option_t *option);

  // Finish task
  virtual void Finish();

  // Look-up table initialization
  void InitLut();

  void SetOutputFile(TString outfilename = "luttab.root") { fOutputFile = outfilename; }

 private:
  void ProcessPhotonHit();

  PndGeoDrc *fGeo;
  int fDetectorID;
  double fBboxNum, fPipehAngle, fDphi;

  TClonesArray *fMCArray; // DRC MCPoints in the photon detector
  TClonesArray *fBarPointArray;
  TClonesArray *fPDPointArray; // DRC points in the photon detector
  TClonesArray *fEVPointArray;
  TClonesArray *fDigiArray;
  TClonesArray *fPDHitArray; // DRC Hits in the photon detector
  std::array<TClonesArray *, 5> fLut;

  TFile *fFile;
  TTree *fTree;

  PndMCTrack *fMCTrack;
  PndDrcBarPoint *fBarPoint;
  PndDrcPDPoint *fPDPoint;
  PndDrcDigi *fDigi;
  PndDrcPDHit *fPDHit;
  PndDrcEVPoint *fEVPoint;

  // Set the parameters to the default values.
  void SetDefaultParameters();

  // Verbosity level
  int fVerbose;
  int nevents;
  TString fOutputFile;

  ClassDef(PndDrcLutFill, 1)
};

#endif