PndMvdEventMergerTask.cxx

Go to the documentation of this file.

// -------------------------------------------------------------------------
// -----                PndMvdEventMergerTask source file             -----
// -------------------------------------------------------------------------

#include "TClonesArray.h"
#include "TArrayD.h"

#include "FairRootManager.h"
#include "PndMvdEventMergerTask.h"
#include "FairRun.h"
#include "FairRuntimeDb.h"

// -----   Default constructor   -------------------------------------------
PndMvdEventMergerTask::PndMvdEventMergerTask() :
  FairTask("MVD Event Merger")
{
        std::cout << "-E- PndMvdEventMergerTask: The default constructor should not be used!" << std::endl;
        fEventNr = 0;
}

PndMvdEventMergerTask::PndMvdEventMergerTask(TString signalBranch, TString bgFile, TString bgBranch, Int_t events, Int_t mergedEvents, Bool_t signalIsBg) :
        FairTask("MVD Event Merger")
{
        fSignalBranch = signalBranch;
        fBgFile = bgFile;
        fBgBranch = bgBranch;
        fNMergedEvents = mergedEvents;
        fSignalIsBg = signalIsBg;
        fEventNr = 0;
        fNEvents = events;
}
// -------------------------------------------------------------------------


// -----   Destructor   ----------------------------------------------------
PndMvdEventMergerTask::~PndMvdEventMergerTask()
{
        delete (fMerger);
}
// -------------------------------------------------------------------------

// -----   Initialization  of Parameter Containers -------------------------
void PndMvdEventMergerTask::SetParContainers()
{
  // called before Init()
  // Get Base Container
  //FairRun* ana = FairRun::Instance();
  //FairRuntimeDb* rtdb=ana->GetRuntimeDb();
}

InitStatus PndMvdEventMergerTask::ReInit()
{
  SetParContainers();
  return kSUCCESS;
}

// -----   Public method Init   --------------------------------------------
InitStatus PndMvdEventMergerTask::Init()
{
    //FairRun* ana = FairRun::Instance();
    FairRootManager* ioman = FairRootManager::Instance();

  if ( ! ioman )
    {
      std::cout << "-E- PndMvdEventMergerTask::Init: "
     << "RootManager not instantiated!" << std::endl;
      return kFATAL;
    }

  if (fSignalIsBg == false){
          fSignalArray = (TClonesArray*) ioman->GetObject(fSignalBranch);

          if ( ! fSignalArray )
            {
              std::cout << "-W- PndMvdEventMergerTask::Init: "
             << "No MVDPoint array!" << std::endl;
              return kERROR;
          }
  }


  // Create and register output array
  fMergedArray = new TClonesArray("PndTpcCluster");
  ioman->Register("PndTpcClusterMerged", "TPC", fMergedArray, kTRUE);

  fMerger = new PndMvdEventMerger(fBgFile, fBgBranch, fNEvents, fNMergedEvents);


   std::cout << "-I- PndMvdEventMergerTask: Intialisation successfull" << std::endl;

  return kSUCCESS;
}
// -------------------------------------------------------------------------

// -----   Public method Exec   --------------------------------------------
void PndMvdEventMergerTask::Exec(Option_t*)
{

        TClonesArray* bg;
        if (fEventNr < fNMergedEvents)
                bg = fMerger->GetEvent(fEventNr);
        else bg = fMerger->GetEvent(0);

        fMergedArray->Delete();

        if (fSignalIsBg == false){
                fMerger->AddTClonesArray(bg, fSignalArray);
        }
  for (int i = 0; i < bg->GetEntriesFast(); i++){
    new ((*fMergedArray)[i])PndTpcCluster(*(PndTpcCluster*)bg->At(i));
  }

  // Event summary

  if (fVerbose > 0){
    std::cout << "-I- PndMvdEventMergerTask: Merged " << fMergedArray->GetEntriesFast() << " events in Event Nr: " << fEventNr << std::endl;
  }
  fEventNr++;
}

ClassImp(PndMvdEventMergerTask);