PndLmdSigCleanTask Class Reference

#include <PndLmdSigCleanTask.h>

Inheritance diagram for PndLmdSigCleanTask:

Public Member Functions
	PndLmdSigCleanTask ()
	PndLmdSigCleanTask (Double_t pBeam, TString dir)
	Set up beam momuntum value. More...
	~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)
bool	CheckXY (FairTrackParP *fTrk)
void	SetDX (double dx)
void	SetDY (double dy)

Private Member Functions
	ClassDef (PndLmdSigCleanTask, 2)

Private Attributes
TClonesArray *	fTrackParFinal
PndGeoHandling *	fGeoH
int	fEventNr
Double_t	fPbeam
bool	fXYcut
TClonesArray *	fRecBPTracks
TClonesArray *	fTrkArray
TString	fdir
TString	fmethodName
Float_t	axrec
Float_t	ayrec
Float_t	aprec
Float_t	azrec
Float_t	athrec
Float_t	aphrec
TMVA::Reader *	reader
double	fdX
double	fdY
TH1 *	hResponse

Detailed Description

Definition at line 37 of file PndLmdSigCleanTask.h.

Constructor & Destructor Documentation

PndLmdSigCleanTask::PndLmdSigCleanTask

(

)

Default constructor

Definition at line 36 of file PndLmdSigCleanTask.cxx.

    : FairTask("Cleaning Tracks Task for PANDA Lmd"),
      fEventNr(0),
      fdX(0),
      fdY(0) {
  // tprop = new TNtuple();
}

PndLmdSigCleanTask::PndLmdSigCleanTask	(	Double_t	pBeam,
		TString	dir
	)

Set up beam momuntum value.

Definition at line 45 of file PndLmdSigCleanTask.cxx.

References fdir, fPbeam, fXYcut, and hResponse.

    : FairTask("Cleaning Tracks  Task for PANDA Lmd"),
      fEventNr(0),
      fdX(0),
      fdY(0) {
  fdir = dir;
  fPbeam = pBeam;
  cout << "Beam Momentum in this run is " << fPbeam << endl;
  hResponse = new TH1D("hResponse", "", 1e3, -1, 1);
  fXYcut = false;
}

PndLmdSigCleanTask::~PndLmdSigCleanTask

(

)

Destructor

Definition at line 58 of file PndLmdSigCleanTask.cxx.

{}

Member Function Documentation

bool PndLmdSigCleanTask::Check

(

FairTrackParH *

fTrk

)

Definition at line 174 of file PndLmdSigCleanTask.cxx.

References CheckMom(), CheckMVA(), fabs(), fPbeam, and res.

Referenced by Exec().

                         {  // answer TRUE means "it's a signal"
  bool res;
  if (fabs(fPbeam - 1.5) < 1e-1)
    res = CheckMVA(fTrk);  // MVA was trained only on sample with 1.5 GeV/c
                           // simulation ///TODO: extension to other energies?
  else {
    TVector3 MomRecBP = fTrk->GetMomentum();
    double prec = MomRecBP.Mag();
    res = CheckMom(prec);
  }
  return res;
}

bool PndLmdSigCleanTask::CheckMom

(

double

prec

)

Definition at line 188 of file PndLmdSigCleanTask.cxx.

References fPbeam, and res.

Referenced by Check().

                                             {
  bool res;
  if (abs(prec - fPbeam) > 3e-4)
    res = false;
  else
    res = true;
  return res;
}

bool PndLmdSigCleanTask::CheckMVA

(

FairTrackParH *

fTrk

)

Definition at line 197 of file PndLmdSigCleanTask.cxx.

References aphrec, aprec, athrec, axrec, ayrec, azrec, Double_t, fdX, fdY, fmethodName, hResponse, Pi, reader, and res.

Referenced by Check().

                                                     {
  // cout<<"Yes, TMVA game!"<<endl;
  bool res;
  TVector3 PosRecBP = fTrk->GetPosition();
  axrec = float(PosRecBP.X() - fdX);
  ayrec = float(PosRecBP.Y() - fdY);
  azrec = float(PosRecBP.Z());
  TVector3 MomRecBP = fTrk->GetMomentum();
  aprec = float(MomRecBP.Mag());
  Double_t lyambda = fTrk->GetLambda();
  athrec = float(TMath::Pi() / 2. - lyambda);
  aphrec = float(fTrk->GetPhi());
  double mva_response = reader->EvaluateMVA(fmethodName);
  hResponse->Fill(mva_response);
  // if(mva_response>-0.0599) res=true; //BDT

  // if(mva_response<-0.0411){//new training for non-point-like beam
  // [28/04/2014]
  // if(mva_response<0.0947){//new training for non-point-like beam [28/04/2014]
  if (mva_response <
      -0.5) {  // new training for non-point-like beam [28/09/2014]
    //   if(mva_response<0){
    // if(mva_response<-0.1452){
    // if(mva_response<-0.156){
    //  if(mva_response<-0.01){
    res = false;  // BDT //TEST
    //    cout<<"BDT="<<mva_response<<" for PCA:("<<axrec<<", "<<ayrec<<",
    //    "<<azrec<<") athrec = "<<1e3*athrec<<" aprec = "<<aprec<<endl;
  } else
    res = true;
  return res;
}

bool PndLmdSigCleanTask::CheckXY

(

FairTrackParP *

fTrk

)

Definition at line 230 of file PndLmdSigCleanTask.cxx.

References diffX, diffY, fdX, fdY, fVerbose, and res.

Referenced by Exec().

                                                           {
  TVector3 MomRecLMD(fFittedTrkP->GetPx(), fFittedTrkP->GetPy(),
                     fFittedTrkP->GetPz());
  MomRecLMD *= 1. / MomRecLMD.Mag();
  bool dirOKx = true;
  bool dirOKy = true;
  if (fVerbose > 0) cout << "!XThFilt!" << endl;
  double Xref = -19.1 + 1.12 * 1e3 * MomRecLMD.Theta() + fdX;
  double diffX = abs(fFittedTrkP->GetX() - Xref);
  if (fVerbose > 0)
    cout << "fFittedTrkP.GetX() = " << fFittedTrkP->GetX() << " Xref = " << Xref
         << " diffX = " << diffX << endl;
  if (diffX > 3.0) dirOKx = false;
  if (fVerbose > 0) cout << "!YPhFilt!" << endl;
  double Yref = -0.00651 + 0.045 * 1e3 * MomRecLMD.Phi() + fdY;
  double diffY = abs(fFittedTrkP->GetY() - Yref);
  if (fVerbose > 0)
    cout << "fFittedTrkP.GetY() = " << fFittedTrkP->GetY() << " Yref = " << Yref
         << " diffY = " << diffY << endl;
  if (diffY > 4.0) dirOKy = false;

  bool res = false;
  if (dirOKx && dirOKy) res = true;
  return res;
}

PndLmdSigCleanTask::ClassDef ( PndLmdSigCleanTask  , 2    )

private

void PndLmdSigCleanTask::Exec ( Option_t *  opt )

virtual

Virtual method Exec

Definition at line 132 of file PndLmdSigCleanTask.cxx.

References Check(), CheckXY(), fEventNr, fRecBPTracks, fTrackParFinal, fTrkArray, fVerbose, fXYcut, and PndTrack::GetParamFirst().

                                       {
  std::map<int, std::vector<int> > mcHitMap;  // Track ->  MCHits
  fTrackParFinal->Delete();

  if (fVerbose > 2) {
    cout << " ---- Info: " << fEventNr << endl;
  }
  fEventNr++;

  // go through all tracks
  const int nGeaneTrks = fRecBPTracks->GetEntriesFast();
  Int_t counterSigTrk = 0;
  for (Int_t iN = 0; iN < nGeaneTrks;
       iN++) {  // loop over all reconstructed trks
    FairTrackParH* fRes = (FairTrackParH*)fRecBPTracks->At(iN);
    // 1. Trk params before BP
    bool isCleanCand;
    if (fXYcut == kTRUE) {
      PndTrack* trkpnd = (PndTrack*)(fTrkArray->At(iN));
      FairTrackParP fFittedTrkP = trkpnd->GetParamFirst();
      isCleanCand = CheckXY(&fFittedTrkP);
    } else {
      isCleanCand = kTRUE;
    }
    // 2. Trk params after BP
    bool isClean = Check(fRes);
    if (isClean == kTRUE && isCleanCand == kTRUE) {
      new ((*fTrackParFinal)[counterSigTrk])
          FairTrackParH(*(fRes));  // save Track
      counterSigTrk++;
      if (fVerbose > 2) cout << "***** isClean TRUE *****" << endl;
    } else {
      new ((*fTrackParFinal)[counterSigTrk]) FairTrackParH();  // save NULL
      counterSigTrk++;
      if (fVerbose > 2) cout << "***** isClean FALSE *****" << endl;
    }
  }  // [END] loop over all reconstructed trks
  if (fVerbose > 2) cout << "PndLmdSigCleanTask::Exec END!" << endl;
}

void PndLmdSigCleanTask::Finish ( )

virtual

Definition at line 172 of file PndLmdSigCleanTask.cxx.

{}

InitStatus PndLmdSigCleanTask::Init ( )

virtual

Definition at line 61 of file PndLmdSigCleanTask.cxx.

References aphrec, aprec, athrec, axrec, ayrec, azrec, fabs(), fdir, fGeoH, fmethodName, fPbeam, fRecBPTracks, fTrackParFinal, fTrkArray, PndGeoHandling::Instance(), reader, and TString.

Referenced by runLumiPixel5cCombFilt().

                                    {
  // Get RootManager
  FairRootManager* ioman = FairRootManager::Instance();
  if (!ioman) {
    std::cout << "-E- PndLmdSigCleanTask::Init: "
              << "RootManager not instantiated!" << std::endl;
    return kFATAL;
  }

  fTrkArray = (TClonesArray*)ioman->GetObject("LMDPndTrackFilt");
  if (!fTrkArray) {
    std::cout << "-W- PndLmdTrkQTask::Init: "
              << "No Track"
              << " array!" << std::endl;
    return kERROR;
  }

  // Get rec.tracks after back propagation
  fRecBPTracks = (TClonesArray*)ioman->GetObject("GeaneTrackFinal");
  if (!fRecBPTracks) {
    std::cout << "-W- PndLmdSigCleanTask::Init: "
              << "No Track after back-propagation"
              << " array!" << std::endl;
    return kERROR;
  }

  fTrackParFinal = new TClonesArray("FairTrackParH");
  ioman->Register("LMDCleanTrack", "PndLmd", fTrackParFinal, kTRUE);

  fGeoH = PndGeoHandling::Instance();
  // FairRun* fRun = FairRun::Instance(); //[R.K. 01/2017] unused variable?
  // FairRuntimeDb* rtdb = fRun->GetRuntimeDb(); //[R.K. 01/2017] unused
  // variable

  // TMVA -----------------------------------------------------
  // This loads the library
  if (fabs(fPbeam - 1.5) < 1e-1) {
    TMVA::Tools::Instance();

    //   TMVA::Reader *
    reader = new TMVA::Reader("!Color:!Silent");

    // Create a set of variables and declare them to the reader
    // - the variable names MUST corresponds in name and type to those given in
    // the weight file(s) used

    reader->AddVariable("axrec", &axrec);
    reader->AddVariable("ayrec", &ayrec);
    reader->AddVariable("azrec", &azrec);
    reader->AddVariable("aprec", &aprec);
    reader->AddVariable("athrec", &athrec);
    reader->AddVariable("aphrec", &aphrec);

    //  TString dir    = "weights/";
    TString prefix = "TMVAClassification";
    fmethodName = "BDT method";
    TString weightfile =
        fdir + prefix + TString("_BDT") + TString(".weights.xml");
    reader->BookMVA(fmethodName, weightfile);
  }
  //------------------------------------------------------
  return kSUCCESS;
}

void PndLmdSigCleanTask::SetDX ( double  dx )

inline

Definition at line 63 of file PndLmdSigCleanTask.h.

References dx, and fdX.

Referenced by runLumiPixel5bCleanSig(), and runLumiPixel5cCombFilt().

{ fdX = dx; }

void PndLmdSigCleanTask::SetDY ( double  dy )

inline

Definition at line 64 of file PndLmdSigCleanTask.h.

References dy, and fdY.

Referenced by runLumiPixel5bCleanSig(), and runLumiPixel5cCombFilt().

{ fdY = dy; }

void PndLmdSigCleanTask::SetParContainers ( )

virtual

Virtual method Init

FairRun* ana = FairRun::Instance();

Definition at line 125 of file PndLmdSigCleanTask.cxx.

                                          {
  // Get Base Container
  // FairRuntimeDb* rtdb=ana->GetRuntimeDb();
}

void PndLmdSigCleanTask::SetXY ( bool  setXY )

inline

Definition at line 60 of file PndLmdSigCleanTask.h.

References fXYcut.

Referenced by runLumiPixel5cCombFilt().

{ fXYcut = setXY; };

Member Data Documentation

Float_t PndLmdSigCleanTask::aphrec

private

Definition at line 82 of file PndLmdSigCleanTask.h.

Referenced by CheckMVA(), and Init().

Float_t PndLmdSigCleanTask::aprec

private

Definition at line 81 of file PndLmdSigCleanTask.h.

Referenced by CheckMVA(), and Init().

Float_t PndLmdSigCleanTask::athrec

private

Definition at line 82 of file PndLmdSigCleanTask.h.

Referenced by CheckMVA(), and Init().

Float_t PndLmdSigCleanTask::axrec

private

Definition at line 81 of file PndLmdSigCleanTask.h.

Referenced by CheckMVA(), and Init().

Float_t PndLmdSigCleanTask::ayrec

private

Definition at line 81 of file PndLmdSigCleanTask.h.

Referenced by CheckMVA(), and Init().

Float_t PndLmdSigCleanTask::azrec

private

Definition at line 82 of file PndLmdSigCleanTask.h.

Referenced by CheckMVA(), and Init().

TString PndLmdSigCleanTask::fdir

private

Definition at line 79 of file PndLmdSigCleanTask.h.

Referenced by Init(), and PndLmdSigCleanTask().

double PndLmdSigCleanTask::fdX

private

Definition at line 86 of file PndLmdSigCleanTask.h.

Referenced by CheckMVA(), CheckXY(), and SetDX().

double PndLmdSigCleanTask::fdY

private

Definition at line 87 of file PndLmdSigCleanTask.h.

Referenced by CheckMVA(), CheckXY(), and SetDY().

int PndLmdSigCleanTask::fEventNr

private

Definition at line 70 of file PndLmdSigCleanTask.h.

Referenced by Exec().

PndGeoHandling* PndLmdSigCleanTask::fGeoH

private

Definition at line 69 of file PndLmdSigCleanTask.h.

Referenced by Init().

TString PndLmdSigCleanTask::fmethodName

private

Definition at line 80 of file PndLmdSigCleanTask.h.

Referenced by CheckMVA(), and Init().

Double_t PndLmdSigCleanTask::fPbeam

private

Definition at line 71 of file PndLmdSigCleanTask.h.

Referenced by Check(), CheckMom(), Init(), and PndLmdSigCleanTask().

TClonesArray* PndLmdSigCleanTask::fRecBPTracks

private

Definition at line 75 of file PndLmdSigCleanTask.h.

Referenced by Exec(), and Init().

TClonesArray* PndLmdSigCleanTask::fTrackParFinal

private

Definition at line 67 of file PndLmdSigCleanTask.h.

Referenced by Exec(), and Init().

TClonesArray* PndLmdSigCleanTask::fTrkArray

private

Definition at line 76 of file PndLmdSigCleanTask.h.

Referenced by Exec(), and Init().

bool PndLmdSigCleanTask::fXYcut

private

Definition at line 72 of file PndLmdSigCleanTask.h.

Referenced by Exec(), PndLmdSigCleanTask(), and SetXY().

TH1* PndLmdSigCleanTask::hResponse

private

Definition at line 89 of file PndLmdSigCleanTask.h.

Referenced by CheckMVA(), and PndLmdSigCleanTask().

TMVA::Reader* PndLmdSigCleanTask::reader

private

Definition at line 83 of file PndLmdSigCleanTask.h.

Referenced by CheckMVA(), and Init().

---

The documentation for this class was generated from the following files:

• PndLmdSigCleanTask.h
• PndLmdSigCleanTask.cxx