NicaUnigenEvent.cxx

Go to the documentation of this file.

/*
 * NicaUnigen.cxx
 *
 *  Created on: 23-06-2014
 *      Author: Daniel Wielanek
 *              E-mail: daniel.wielanek@gmail.com
 *              Warsaw University of Technology, Faculty of Physics
 */
 
#include "NicaUnigenEvent.h"
#include "FairRootManager.h"
#include "NicaUnigenEventInterface.h"
 
NicaUnigenEvent::NicaUnigenEvent() : NicaMCEvent("NicaUnigenTrack") {}
 
void NicaUnigenEvent::Update() {
  UEvent* temp   = ((NicaUnigenEventInterface*) fSource)->fEvent;
  fB             = temp->GetB();
  fPhi           = temp->GetPhi();
  fTotalTracksNo = temp->GetNpa();
  fTracks->Clear();
  for (int i = 0; i < fTotalTracksNo; i++) {
    UParticle* particle    = temp->GetParticle(i);
    TParticlePDG* pdg_part = fPDG->GetParticle(particle->GetPdg());
    Double_t charge        = 0;
    if (pdg_part) { charge = pdg_part->Charge() / 3.0; }
    NicaMCTrack* target_track = (NicaMCTrack*) fTracks->ConstructedAt(i);
    if (particle->GetMate() < 0) {
      target_track->SetPrimary(kTRUE);
    } else {
      target_track->SetPrimary(kFALSE);
    }
    target_track->SetID(i);
    target_track->SetCharge(charge);
    target_track->SetPdg(particle->GetPdg());
    target_track->GetLink()->Clear();
    target_track->GetLink()->SetLink(0, i);
    target_track->GetMomentum()->SetPxPyPzE(
      particle->Px(), particle->Py(), particle->Pz(), particle->E());
    target_track->GetFreezoutPosition()->SetXYZT(
      particle->X(), particle->Y(), particle->Z(), particle->T());
    target_track->SetStatus(particle->GetStatus());
    target_track->SetEvent(this);
  }
}
 
void NicaUnigenEvent::Clear(Option_t* opt) {
  NicaMCEvent::Clear(opt);
  if (fSource) fSource->Clear(opt);
}
 
NicaUnigenEvent::NicaUnigenEvent(const NicaUnigenEvent& other)
  : NicaMCEvent(other) {}
 
void NicaUnigenEvent::CreateSource() {
  fSource = new NicaUnigenEventInterface();
}
 
NicaUnigenEvent::~NicaUnigenEvent() {}
 
TString NicaUnigenEvent::GetFormatName() const { return "UnigenFormat"; }
 
Bool_t NicaUnigenEvent::ExistInTree() const {
  FairRootManager* manager = FairRootManager::Instance();
  if (manager->CheckBranch("UEvent.")) { return kTRUE; }
  return kFALSE;
}