OTFMcEvent.cxx

/*
 * HalOnTheFlyMcEvent.cxx
 *
 *  Created on: 28 maj 2022
 *      Author: Daniel Wielanek
 *      E-mail: daniel.wielanek@gmail.com
 *      Warsaw University of Technology, Faculty of Physics
 */
 
#include "OTFMcEvent.h"
#include "OTFData.h"
#include "OTFMcEventInterface.h"
 
#include "DataManager.h"
 
#include <RtypesCore.h>
#include <TClonesArray.h>
#include <TObjArray.h>
#include <TObject.h>
#include <iostream>
 
namespace HalOTF {
  McEvent::McEvent() : Hal::McEvent("HalOTF::McTrack") {}
 
  Hal::EventInterface* McEvent::CreateInterface() const { return new McEventInterface(); }
 
  void McEvent::Update(Hal::EventInterface* interface) {
    Clear();
    McEventInterface* s = (McEventInterface*) interface;
    fTotalTracksNo      = s->GetMcEvent()->GetNTracks();
    fTracks->ExpandCreateFast(fTotalTracksNo);
    OTF::McEvent* ev = s->GetMcEvent();
    fB               = ev->GetB();
    fPhi             = ev->GetPhi();
    for (int i = 0; i < fTotalTracksNo; i++) {
      McTrack* track   = (McTrack*) fTracks->UncheckedAt(i);
      OTF::McTrack* mc = (OTF::McTrack*) ev->GetTrack(i);
      track->ResetTrack(i, this);
      track->SetMomentum(mc->GetMomentum().Px(), mc->GetMomentum().Py(), mc->GetMomentum().Pz(), mc->GetMomentum().E());
      track->SetFreezoutPosition(mc->GetFreezout().X(), mc->GetFreezout().Y(), mc->GetFreezout().Z(), mc->GetFreezout().T());
      track->SetMotherIndex(mc->GetMotherId());
      track->SetPdg(mc->GetPdgCode());
    }
  }
 
  Bool_t McEvent::ExistInTree() const {
    Hal::DataManager* manager = Hal::DataManager::Instance();
    if (manager->CheckBranch("OTF::McEvent.")) { return kTRUE; }
    return kFALSE;
  }
 
  McEvent::~McEvent() {}
}  // namespace HalOTF