PndGFRecoTask.h

Go to the documentation of this file.

#pragma once

#include "PndPersistencyTask.h"
#include "PndTrack.h"
#include "PndGFRecoFit.h"
#include "FairLogger.h"

#include "TString.h"
#include "TClonesArray.h"
#include "TMath.h"

#include <vector>
//#include "eventDisplay/include/EventDisplay.h"

class TClonesArray;
class MeasurementFactory;
class TDatabasePDG;

struct PndGFPidHypo {

  PndGFPidHypo(){};
  PndGFPidHypo(int pdgvalue)
  {
    switch (TMath::Abs(pdgvalue)) {
    case 11:
      fPdgNumber = -11;
      fParticleName = "Electron";
      break;
    case 13:
      fPdgNumber = -13;
      fParticleName = "Muon";
      break;
    case 211:
      fPdgNumber = 211;
      fParticleName = "Pion";
      break;
    case 321:
      fPdgNumber = 321;
      fParticleName = "Kaon";
      break;
    case 2212:
      fPdgNumber = 2212;
      fParticleName = "Proton";
      break;
    default: LOG(error) << "Wrong particle ID given: " << pdgvalue; LOG(error) << "Allowed values are: 11, 13, 211, 321, 2212";
    }
  };
  int fPdgNumber = 0;
  TString fParticleName;
  TClonesArray *fArray = nullptr;
};

class PndGFRecoTask : public PndPersistencyTask {
 public:
  // Constructors/Destructors ---------
  PndGFRecoTask(const char *name = "Genfit", Int_t iVerbose = 0);
  ~PndGFRecoTask();

  void AddParticleHypo(int pdgCode);

  void SetMultiPID();

  void UseIdealPID(bool val = true, Int_t defaultPID = 211)
  {
    fIdealPID = val;
    fDefaultPID = defaultPID;
  }

  void SetTrackInBranchName(TString name) { fInputBranchName = name; }

  void SetTrackOutBranchName(TString name) { fOutputBranchName = name; }

  void SetFitter(genfit::AbsKalmanFitter *fitter);

  void SetNBusyCut(int val) { fNBusyCut = val; }

  void UseEventDisplay(bool val = true) { fUseEventDisplay = val; }

  // Operations ----------------------
  virtual InitStatus Init();
  void SetParContainers();
  virtual void Exec(Option_t *opt);
  virtual void FinishTask() {}

 protected:
  int GetIdealPdgCode(PndTrack *track);
  void DeleteOldTracks()
  {
    for (auto &track : fPidHypothesis) {
      track.fArray->Delete();
    }
  }

 private:
  TClonesArray *fPrefitTrack = nullptr;     //< Input from track finder
  TClonesArray *fMCTrack = nullptr;         //< MC tracks needed for ideal PID
  std::vector<PndGFPidHypo> fPidHypothesis; //< Pid Hypothesis to fit with. Can be more than one.
  TString fInputBranchName = "";            //< Name of branch with prefit track data
  TString fOutputBranchName = "";           //< Name of output branch with fitted track data
  bool fIdealPID = false;                   //< Flag to use ideal PID instead of PID hypothesis
  bool fMultiPID = true;                    //< Flag
  int fDefaultPID = 211;                    //< PID assumption if no PID hypothesis is given
  TDatabasePDG *fPdg = nullptr;             //< Needed by GetIdealPdgCode
  int fNBusyCut = 50;                       //< Events with tracks above NBusyCut are ignored
  bool fUseEventDisplay = false;            //< Enables genfit2 event display for PndGFRecoFit
  PndGFRecoFit fFit;

  ClassDef(PndGFRecoTask, 2);
};