PndLmdSigCleanTask.h

Go to the documentation of this file.

// -------------------------------------------------------------------------
// -----                   PndLmdSigCleanTask header file            -----
// -----                  Created 26/02/14  by A.Karavdina            -----
// -------------------------------------------------------------------------

#ifndef PndLmdSigCleanTask_H
#define PndLmdSigCleanTask_H

// framework includes
#include "FairTask.h"
#include "TH1.h"
//#include "GFTrackCand.h"
//#include "FairGeanePro.h"
#include "FairTrackParH.h"
#include "FairTrackParP.h"
#include "TMatrixD.h"
#include "TNtuple.h"
#include "TVector3.h"
//#include "PndSdsGeoHandling.h"
#include <map>
#include <vector>
#include "PndGeoHandling.h"
// TMVA includes
#include "TMVA/MethodCuts.h"
#include "TMVA/Reader.h"
#include "TMVA/Tools.h"
//#include "PndMultiField.h"
class TClonesArray;
class PndSdsCluster;
class PndLmdSigCleanTask : public FairTask {
 public:
  PndLmdSigCleanTask();

  PndLmdSigCleanTask(Double_t pBeam, TString dir);

  ~PndLmdSigCleanTask();

  virtual void SetParContainers();
  virtual InitStatus Init();

  virtual void Exec(Option_t* opt);

  virtual void Finish();

  bool CheckMVA(FairTrackParH* fTrk);
  bool CheckMom(double prec);
  bool Check(FairTrackParH* fTrk);
  void SetXY(bool setXY) { fXYcut = setXY; };

  bool CheckXY(FairTrackParP* fTrk);
  void SetDX(double dx) { fdX = dx; }
  void SetDY(double dy) { fdY = dy; }

 private:
  TClonesArray* fTrackParFinal;

  PndGeoHandling* fGeoH;
  int fEventNr;
  Double_t fPbeam;
  bool fXYcut;

  // Input Data ----------
  TClonesArray* fRecBPTracks;
  TClonesArray* fTrkArray;

  // MVA Data -------
  TString fdir;
  TString fmethodName;
  Float_t axrec, ayrec, aprec;
  Float_t azrec, athrec, aphrec;
  TMVA::Reader* reader;

  //----------------------------------------
  double fdX;  // beam shift in X
  double fdY;  // beam shift in Y

  TH1* hResponse;

  ClassDef(PndLmdSigCleanTask, 2);
};

#endif