NicaCbmMCEvent.cxx

Go to the documentation of this file.

/*
 * NicaCbmMCEvent.cxx
 *
 *  Created on: 20 wrz 2018
 *      Author: Daniel Wielanek
 *              E-mail: daniel.wielanek@gmail.com
 *              Warsaw University of Technology, Faculty of Physics
 */
#include "NicaCbmMCEvent.h"
#include "CbmMCTrack.h"
#include "FairRootManager.h"
#include "NicaCbmMCEventInterface.h"
#include "NicaCbmMCTrack.h"
 
NicaCbmMCEvent::NicaCbmMCEvent() : NicaMCEvent("NicaCbmMCTrack") {
  fSource = new NicaCbmMCEventInterface();
}
 
NicaCbmMCEvent::~NicaCbmMCEvent() {
  // TODO Auto-generated destructor stub
}
 
void NicaCbmMCEvent::CreateSource() { fSource = new NicaCbmMCEventInterface(); }
 
void NicaCbmMCEvent::Update() {
  FairMCEventHeader* event =
    (FairMCEventHeader*) ((NicaCbmMCEventInterface*) fSource)->fEvent;
  TClonesArray* tracks =
    (TClonesArray*) ((NicaCbmMCEventInterface*) fSource)->fCbmTracks;
  fB = event->GetB();
  fVertex->SetXYZT(event->GetX(), event->GetY(), event->GetZ(), event->GetT());
  fTracks->Clear();
  fTotalTracksNo = tracks->GetEntriesFast();
  fTracks->ExpandCreateFast(fTotalTracksNo);
  for (int i = 0; i < tracks->GetEntriesFast(); i++) {
    CbmMCTrack* track  = (CbmMCTrack*) tracks->UncheckedAt(i);
    Double_t charge    = CalculateCharge(track->GetPdgCode());
    NicaCbmMCTrack* mc = (NicaCbmMCTrack*) fTracks->UncheckedAt(i);
    mc->SetEvent(this);
    mc->Update(track, charge);
    mc->GetLink()->Clear();
    mc->GetLink()->SetLink(0, i);
  }
}
 
NicaCbmMCEvent::NicaCbmMCEvent(const NicaCbmMCEvent& other)
  : NicaMCEvent(other) {}
 
Bool_t NicaCbmMCEvent::ExistInTree() const {
  FairRootManager* manager = FairRootManager::Instance();
  Int_t header             = manager->CheckBranch("MCEventHeader.")
                 + manager->CheckBranch("EventHeader.");
  Int_t tracks = manager->CheckBranch("MCTrack");
  if ((header + tracks) >= 2) { return kTRUE; }
  return kFALSE;
}