PndMcCloner.h

Go to the documentation of this file.

#pragma once

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

using namespace std;

class TClonesArray;

class PndMcCloner : public FairTask {

 public:
  PndMcCloner();

  ~PndMcCloner();

  virtual InitStatus Init();

  virtual void Exec(Option_t *);

  void SetCleanMc(Bool_t opt = kTRUE) { fCleanMC = opt; }

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

  TClonesArray *fInputArray = nullptr;

  TClonesArray *fPidChargedArray = nullptr;

  TClonesArray *fPidNeutralArray = nullptr;

  TClonesArray *fOutputArray = nullptr;

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

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

  ClassDef(PndMcCloner, 2);
};