PndMcClonerMulti.h

Go to the documentation of this file.

//****************************************************************************
//*                   This file is part of PandaRoot.                        *
//*                                                                          *
//*            PandaRoot is distributed under the terms of the               *
//*              GNU General Public License (GPL) version 3,                 *
//*                 copied verbatim in the file "LICENSE".                   *
//*                                                                          *
//*  Copyright (C) 2006 - 2024 FAIR GmbH and copyright holders of PandaRoot  *
//*     The copyright holders are listed in the file "COPYRIGHTHOLDERS".     *
//*               The authors are listed in the file "AUTHORS".              *
//****************************************************************************

#pragma once

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

using namespace std;

class TClonesArray;

class PndMcClonerMulti : public FairTask {

 public:
  PndMcClonerMulti();

  ~PndMcClonerMulti();

  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 CloneMCTracks();
  void CloneAndCleanMCTrack();
  void CorrectMotherIndices();
  void CorrectPidIndices();
  void CloneMcTrack(Int_t mcidx = -1);
  void MarkMcTree(Int_t mcindex = -1);

  TClonesArray *fInputArray;

  TClonesArray *fPidChargedArray[5];

  TClonesArray *fPidNeutralArray;

  TClonesArray *fOutputArray;

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

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

  TString fPidHypoStr[5];

  TString fTrackBranchNamePidHypo;

  ClassDef(PndMcClonerMulti, 2);
};