ERBeamDetPID.h

/********************************************************************************
 *              Copyright (C) Joint Institute for Nuclear Research              *
 *                                                                              *
 *              This software is distributed under the terms of the             *
 *         GNU Lesser General Public Licence version 3 (LGPL) version 3,        *
 *                  copied verbatim in the file "LICENSE"                       *
 ********************************************************************************/

#ifndef ERBeamDetPID_H
#define ERBeamDetPID_H

#include "TClonesArray.h"
#include "TParticlePDG.h"

#include "FairIon.h"

#include "ERTask.h"
#include "ERBeamDetTrack.h"
#include "ERBeamDetParticle.h"
#include "ERBeamDetSetup.h"

class ERBeamDetPID : public ERTask {
public:
  ERBeamDetPID();
  ERBeamDetPID(Int_t verbose);
  ~ERBeamDetPID() = default;
  virtual InitStatus Init();
  virtual void Exec(Option_t* opt);
  virtual void Reset();
  void SetPID(Int_t pdg) {fPID = pdg;}
  void SetBoxPID(Double_t tof1, Double_t tof2, Double_t dE1, Double_t dE2);
  void SetOffsetToF(Double_t offsetToF) {fOffsetToF = offsetToF;}
  void SetIonMass(Double_t mass) {fIonMass = mass;}
  void SetProbabilityThreshold(Double_t probabilityThreshold) {
    fProbabilityThreshold = probabilityThreshold; 
  }
  void SetIonMassNumber(Int_t a);
  void SetIonMass(Float_t mass){fIonMass = mass;}
protected:
  std::pair<float, float> CalcEkinAndTimeOnTarget(ERBeamDetTrack& track, int pid, float mom,
                                                  float mass, float time_on_tof5);
  //Paramaters
  ERBeamDetSetup* fBeamDetSetup = nullptr; 
  //Input arrays
  TClonesArray* fBeamDetToFDigi1 = nullptr; 
  TClonesArray* fBeamDetToFDigi2 = nullptr; 
  TClonesArray* fBeamDetTrack = nullptr; 
  Int_t fPID = -1; 
  Double_t fToF1 = 0., fToF2 = 0.; 
  Double_t fdE1 = 0., fdE2 = 0.; 
  Double_t fOffsetToF = 0.; 
  Double_t fIonMass = 0.; 
  Double_t fProbabilityThreshold = 0.; 
  TParticlePDG* fIon = nullptr;
  //Output arrays
  TClonesArray* fProjectile = nullptr; 
private:
  ERBeamDetParticle* AddParticle(Int_t pid, TLorentzVector tofState, 
                                 TLorentzVector targetState, float time_on_target,
                                 float probability);
  ClassDef(ERBeamDetPID,1)
};
#endif