PndMcCloner2.h

Go to the documentation of this file.

#ifndef PndMcCloner2_H
#define PndMcCloner2_H 1


#include "FairTask.h"
#include "PndMCTrack.h"
#include <map>

using namespace std;

class TClonesArray;

class PndMcCloner2 : public FairTask
{

 public:

  PndMcCloner2();


  ~PndMcCloner2();


  virtual InitStatus Init();


  virtual void Exec(Option_t*);

  void SetCleanMc(Bool_t opt = kTRUE) { fCleanMC = opt; };
  void SetOutputBranch(TString branch)    { fTrackBranchNamePidHypo = branch; };

 protected:

  void FindUsedMCIndices();
  void CloneMCTrack();
  void CloneAndCleanMCTrack();
  void CorrectMotherIndices();
  void CorrectPidIndices();

  TClonesArray* fInputArray;

  TClonesArray* fPidChargedArray[5];

  TClonesArray* fPidNeutralArray[5];


  TClonesArray* fOutputArray;

  map<Int_t, Int_t> mapMCIndex; // Map <old mc index, new mc index>

  Bool_t fCleanMC; // Flag to clean the MCTrack from unused indices

  TString fPidHypoStr[5];
  Bool_t fHypoFlagCharged[5];                           
  Bool_t fHypoFlagNeutral[5];                           

  TString fTrackBranchNamePidHypo;

  ClassDef(PndMcCloner2,2);

};

#endif