NicaCbmMatchedEventInterface.cxx

Go to the documentation of this file.

/*
 * NicaCbmMatchedEventInterface.cxx
 *
 *  Created on: 12 lip 2019
 *      Author: Daniel Wielanek
 *              E-mail: daniel.wielanek@gmail.com
 *              Warsaw University of Technology, Faculty of Physics
 */
#include "NicaCbmMatchedEventInterface.h"
#include "NicaCbmEvent.h"
#include "NicaCbmMatchedTrackInterface.h"
 
NicaCbmMatchedEventInterface::NicaCbmMatchedEventInterface(NicaCbmSetup mode)
  : NicaCbmEventInterface(mode), fMuchMatches(NULL), fRichMatches(NULL) {
  fStsMatches = new NicaTrackClones("CbmTrackMatchNew", "StsTrackMatch", "STS");
  fTofMatches = new NicaTrackClones("CbmTrackMatchNew", "TofHitMatch", "TOF");
  fTrdMatches = new NicaTrackClones("CbmTrackMatchNew", "TrdTrackMatch", "TRD");
  switch (fMode) {
    case kSis100Electron: {
      fRichMatches =
        new NicaTrackClones("CbmTrackMatchNew", "RichRingMatch", "RICH");
    } break;
    case kSis100Hadron: break;
    case kSis100Muon: {
      fMuchMatches =
        new NicaTrackClones("CbmTrackMatchNew", "MuchTrackMatch", "MUCH");
    } break;
  }
}
 
void NicaCbmMatchedEventInterface::ConnectToTree() {
  NicaCbmEventInterface::ConnectToTree();
  if (CanDeleteEvent()) {
    fStsMatches->DeleteClones();
    fTofMatches->DeleteClones();
    fTrdMatches->DeleteClones();
    switch (fMode) {
      case kSis100Electron: {
        fRichMatches->DeleteClones();
      } break;
      case kSis100Hadron: break;
      case kSis100Muon: {
        fMuchMatches->DeleteClones();
      } break;
    }
  }
  fStsMatches->GetFromTree();
  fTofMatches->GetFromTree();
  fTrdMatches->GetFromTree();
  if (fRichMatches) fRichMatches->GetFromTree();
  if (fMuchMatches) fMuchMatches->GetFromTree();
}
 
void NicaCbmMatchedEventInterface::Register(Bool_t write) {
  NicaCbmEventInterface::Register(write);
  fStsMatches->Register(write);
  fTofMatches->Register(write);
  fTrdMatches->Register(write);
  if (fMuchMatches) fMuchMatches->Register(write);
  if (fRichMatches) fRichMatches->Register(write);
}
 
void NicaCbmMatchedEventInterface::CopyData(NicaEventInterface* s) {
  NicaCbmEventInterface::CopyData(s);
  NicaCbmMatchedEventInterface* interface = (NicaCbmMatchedEventInterface*) s;
  fStsMatches->CopyFrom<CbmTrackMatchNew>(interface->GetStsMatches());
  fTofMatches->CopyFrom<CbmTrackMatchNew>(interface->GetTofMatches());
  fTrdMatches->CopyFrom<CbmTrackMatchNew>(interface->GetTrdMatches());
  if (fMuchMatches)
    fMuchMatches->CopyFrom<CbmTrackMatchNew>(interface->GetMuchMatches());
  if (fRichMatches)
    fRichMatches->CopyFrom<CbmTrackMatchNew>(interface->GetRichMatches());
}
 
void NicaCbmMatchedEventInterface::Compress(Int_t* map, Int_t map_size) {
  NicaCbmEventInterface::Compress(map, map_size);
  //TODO make compression
}
 
void NicaCbmMatchedEventInterface::CopyAndCompress(NicaEventInterface* s,
                                                   Int_t* map,
                                                   Int_t map_size) {
  NicaCbmEventInterface::CopyAndCompress(s, map, map_size);
  NicaCbmMatchedEventInterface* interface = (NicaCbmMatchedEventInterface*) s;
  fStsMatches->CopyFrom<CbmTrackMatchNew>(interface->GetStsMatches());
  fTofMatches->CopyFrom<CbmTrackMatchNew>(interface->GetTofMatches());
  fTrdMatches->CopyFrom<CbmTrackMatchNew>(interface->GetTrdMatches());
  if (fMuchMatches)
    fMuchMatches->CopyFrom<CbmTrackMatchNew>(interface->GetMuchMatches());
  if (fRichMatches)
    fRichMatches->CopyFrom<CbmTrackMatchNew>(interface->GetRichMatches());
}
 
void NicaCbmMatchedEventInterface::FillTrackInterface(NicaTrackInterface* track,
                                                      Int_t index) {
  NicaCbmEventInterface::FillTrackInterface(track, index);
  NicaCbmMatchedTrackInterface* Track = (NicaCbmMatchedTrackInterface*) track;
  CbmGlobalTrack* glob = (CbmGlobalTrack*) fTracks->UncheckedAt(index);
  Track->SetLinkSts(
    (CbmTrackMatchNew*) fStsMatches->UncheckedAt(glob->GetStsTrackIndex()));
  Track->SetLinkTof(
    (CbmTrackMatchNew*) fTofMatches->UncheckedAt(glob->GetTofHitIndex()));
  Track->SetLinkTrd(
    (CbmTrackMatchNew*) fTrdMatches->UncheckedAt(glob->GetTrdTrackIndex()));
  if (fMuchMatches) {
    Track->SetLinkMuch(
      (CbmTrackMatchNew*) fMuchMatches->UncheckedAt(glob->GetMuchTrackIndex()));
  }
  if (fRichMatches) {
    Track->SetLinkRich(
      (CbmTrackMatchNew*) fRichMatches->UncheckedAt(glob->GetRichRingIndex()));
  }
}
 
NicaCbmMatchedEventInterface::~NicaCbmMatchedEventInterface() {
  if (CanDeleteEvent()) {
    if (fStsMatches) delete fStsMatches;
    if (fTofMatches) delete fTofMatches;
    if (fTrdMatches) delete fTrdMatches;
    if (fRichMatches) delete fRichMatches;
    if (fMuchMatches) delete fMuchMatches;
  }
}