NicaCbmGlobalEventInterface.cxx

Go to the documentation of this file.

/*
 * NicaCbmEventInterface.cxx
 *
 *  Created on: 3 gru 2018
 *      Author: Daniel Wielanek
 *              E-mail: daniel.wielanek@gmail.com
 *              Warsaw University of Technology, Faculty of Physics
 */
 
#include "NicaCbmGlobalEventInterface.h"
 
#include <FairRootManager.h>
#include <RtypesCore.h>
#include <TObjArray.h>
#include <stddef.h>
 
#include "CbmGlobalTrack.h"
#include "CbmStsTrack.h"
#include "CbmTofTrack.h"
 
NicaCbmGlobalEventInterface::NicaCbmGlobalEventInterface()
  : fVertex(NULL), fTracks(NULL) {
  fVertex = new CbmVertex();
  fTracks = new NicaTrackClones("CbmGlobalTrack", "GlobalTrack", "");
}
 
void NicaCbmGlobalEventInterface::ConnectToTree() {
  FairRootManager* manager = FairRootManager::Instance();
  if (CanDeleteEvent()) {
    delete fVertex;
    fTracks->DeleteClones();
  }
  fVertex = (CbmVertex*) manager->GetObject("PrimaryVertex.");
  fTracks->GetFromTree();
}
 
void NicaCbmGlobalEventInterface::Register(Bool_t write) {
  FairRootManager* manager = FairRootManager::Instance();
  manager->Register("PrimaryVertex.", "PrimaryVertex.", fVertex, write);
  fTracks->Register(write);
}
 
void NicaCbmGlobalEventInterface::CopyData(NicaEventInterface* s) {
  NicaCbmGlobalEventInterface* ev = (NicaCbmGlobalEventInterface*) s;
  *fVertex                        = *ev->fVertex;
  fTracks->CopyFrom<CbmGlobalTrack>(ev->fTracks->GetArray());
}
 
void NicaCbmGlobalEventInterface::Compress(Int_t* map, Int_t map_size) {
  fTracks->Compress(map, map_size);
}
 
void NicaCbmGlobalEventInterface::CopyAndCompress(NicaEventInterface* s,
                                                  Int_t* map,
                                                  Int_t map_size) {
  NicaCbmGlobalEventInterface* ev = (NicaCbmGlobalEventInterface*) s;
  *fVertex                        = *ev->fVertex;
  fTracks->CopyCompress<CbmGlobalTrack>(ev->fTracks->GetArray(), map, map_size);
}
 
void NicaCbmGlobalEventInterface::FillTrackInterface(NicaTrackInterface* track,
                                                     Int_t index) {
  NicaCbmGlobalTrackInterface* cbm_track = (NicaCbmGlobalTrackInterface*) track;
  cbm_track->SetRawTrack(fTracks->UncheckedAt(index));
}
 
Int_t NicaCbmGlobalEventInterface::GetTotalTrackNo() const {
  return fTracks->GetEntriesFast();
}
 
TObject* NicaCbmGlobalEventInterface::GetRawTrackPointer(Int_t index) const {
  return fTracks->UncheckedAt(index);
}
 
TLorentzVector NicaCbmGlobalEventInterface::GetVertexError() const {
  return TLorentzVector(0, 0, 0, 0);
}
 
TLorentzVector NicaCbmGlobalEventInterface::GetVertex() const {
  return TLorentzVector(fVertex->GetX(), fVertex->GetY(), fVertex->GetZ(), 0);
}
 
NicaCbmGlobalEventInterface::~NicaCbmGlobalEventInterface() {
  if (CanDeleteEvent()) {
    if (fVertex) delete fVertex;
    if (fTracks) delete fTracks;
  }
}