PndMasterPidTask.cxx

Go to the documentation of this file.

// -------------------------------------------------------------------------
// -----                    PndMasterPidTask source file               -----
// -----                   Created 01/02/16  by S. Spataro             -----
// -----                       Wrapper for  pid tasks                  -----
// -------------------------------------------------------------------------


#include "PndMasterPidTask.h"
#include "PndMasterTask.h"

#include "PndPidCorrelator.h"
#include "PndPidBremCorrector.h"
#include "PndMcCloner.h"
#include "PndPidIdealAssociatorTask.h"
#include "PndPidMvdAssociatorTask.h"
#include "PndPidMdtHCAssociatorTask.h"
#include "PndPidDrcAssociatorTask.h"
#include "PndPidDiscAssociatorTask.h"
#include "PndPidSttAssociatorTask.h"
#include "PndPidEmcBayesAssociatorTask.h"
#include "PndPidSciTAssociatorTask.h"
#include "PndPidFtofAssociatorTask.h"
#include "PndPidRichAssociatorTask.h"

// -----   Default constructor   -------------------------------------------
PndMasterPidTask::PndMasterPidTask(TString options) :
  PndMasterTask("Master Pid Task"), fOptions(options)
{
  pid = {-1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1};

  // -----   Correlation   ---------------------------------
  PndPidCorrelator* corr = NULL;
  this->Add(corr = new PndPidCorrelator()); // 1
  pid.kPndPidCorrelator = GetListOfTasks()->GetSize()-1;
  TString barrelbranchname="SttMvdGemGenTrack";
  if (fOptions.Contains("nogem")||fOptions.Contains("gem0")) {
    barrelbranchname="SttMvdGenTrack";
  }
  if (fOptions.Contains("filtered")) barrelbranchname+="_filtered";
  if (fOptions.Contains("fakeonline")) barrelbranchname+="_fakeonline";
  corr->SetBarrelTrackBranch(barrelbranchname);

  TString fwdbranchname="FtsIdealGenTrack";
  if (fOptions.Contains("fakeonline")) fwdbranchname+="_fakeonline";
  corr->SetForwardTrackBranch(fwdbranchname);

  if (fOptions.Contains("piddebug")) corr->SetDebugMode(kTRUE);
  if (fOptions.Contains("pidfast")) corr->SetFast(kTRUE);
  if (fOptions.Contains("pidnoswim")) corr->SetBackPropagate(kFALSE);

  // -----   Bremsstrahlung Correction ----------------------
  this->Add(new PndPidBremCorrector()); // 2
  pid.kPndPidBremCorrector = GetListOfTasks()->GetSize()-1;

  // -----   MC Cloner   ------------------------------------
  PndMcCloner *clone = NULL;
  this->Add(clone = new PndMcCloner()); // 3
  pid.kPndMcCloner = GetListOfTasks()->GetSize()-1;
  // Option to clean the MCTrack TClonesArray from particles which were not interacting with sensitive detectors
  clone->SetCleanMc();

  // -----   Classifiers   ----------------------------------
  this->Add(new PndPidIdealAssociatorTask()); // 4
  pid.kPndPidIdealAssociatorTask = GetListOfTasks()->GetSize()-1;

  this->Add(new PndPidMvdAssociatorTask()); // 5
  pid.kPndPidMvdAssociatorTask = GetListOfTasks()->GetSize()-1;

  this->Add(new PndPidMdtHCAssociatorTask()); // 6
  pid.kPndPidMdtHCAssociatorTask = GetListOfTasks()->GetSize()-1;

  this->Add(new PndPidDrcAssociatorTask()); // 7
  pid.kPndPidDrcAssociatorTask = GetListOfTasks()->GetSize()-1;

  if ( !fOptions.Contains("day1") && !fOptions.Contains("phase1") )
  {
    this->Add(new PndPidDiscAssociatorTask()); // 8
    pid.kPndPidDiscAssociatorTask = GetListOfTasks()->GetSize()-1;
  }

  this->Add(new PndPidSttAssociatorTask()); // 9
  pid.kPndPidSttAssociatorTask = GetListOfTasks()->GetSize()-1;

  this->Add(new PndPidEmcBayesAssociatorTask()); // 10
  pid.kPndPidEmcBayesAssociatorTask = GetListOfTasks()->GetSize()-1;

  this->Add(new PndPidSciTAssociatorTask()); // 11
  pid.kPndPidSciTAssociatorTask = GetListOfTasks()->GetSize()-1;

  this->Add(new PndPidFtofAssociatorTask()); // 12
  pid.kPndPidFtofAssociatorTask = GetListOfTasks()->GetSize()-1;

  if ( !fOptions.Contains("day1") && !fOptions.Contains("phase1") )
  {
    this->Add(new PndPidRichAssociatorTask()); // 13
    pid.kPndPidRichAssociatorTask = GetListOfTasks()->GetSize()-1;
  }

  SetVerbose(0);

}
// -------------------------------------------------------------------------

void PndMasterPidTask::SetPersistency(Bool_t pers)
{
  if (!pers) LOG(INFO) << "It makes no sense to have pid persistency switched OFF!" << FairLogger::endl;
  if (!pers) LOG(INFO) << "Or, if you prefer... this functionality has not been implemented yet" << FairLogger::endl;

  return;
}

// -----   Destructor   ----------------------------------------------------
PndMasterPidTask::~PndMasterPidTask()
{
}
// -------------------------------------------------------------------------

ClassImp(PndMasterPidTask);