CbmRichHitProducer.cxx

Go to the documentation of this file.

 
#include "CbmRichHitProducer.h"
 
#include "CbmDigiManager.h"
#include "CbmEvent.h"
#include "CbmRichDetectorData.h"  // for CbmRichPmtData, CbmRichPixelData
#include "CbmRichDigi.h"
#include "CbmRichDigiMapManager.h"
#include "CbmRichGeoManager.h"
#include "CbmRichHit.h"
#include "CbmRichPoint.h"
#include "FairLogger.h"
#include "TClonesArray.h"
 
 
#include <iostream>
 
using namespace std;
 
 
CbmRichHitProducer::CbmRichHitProducer()
  : FairTask("CbmRichHitProducer")
  , fDigiMan(nullptr)
  , fRichHits(NULL)
  , fCbmEvents(NULL)
  , fEventNum(0)
  , fRotationNeeded(true)
  , fHitError(0.6 / sqrt(12.))
 
{}
 
CbmRichHitProducer::~CbmRichHitProducer() {
  FairRootManager* manager = FairRootManager::Instance();
  manager->Write();
}
 
void CbmRichHitProducer::SetParContainers() {}
 
InitStatus CbmRichHitProducer::Init() {
  FairRootManager* manager = FairRootManager::Instance();
 
  fCbmEvents = dynamic_cast<TClonesArray*>(manager->GetObject("Event"));
  if (fCbmEvents == nullptr) {
    LOG(info) << ": CbmEvent NOT found \n \n \n";
  } else {
    LOG(info) << ": CbmEvent found \n \n \n";
  }
 
  fDigiMan = CbmDigiManager::Instance();
  fDigiMan->Init();
  if (!fDigiMan->IsPresent(ECbmModuleId::kRich)) {
    Fatal("CbmRichHitProducer::Init", "No RichDigi array!");
  }
 
  fRichHits = new TClonesArray("CbmRichHit");
  manager->Register(
    "RichHit", "RICH", fRichHits, IsOutputBranchPersistent("RichHit"));
 
  return kSUCCESS;
}
 
void CbmRichHitProducer::Exec(Option_t* /*option*/) {
  fEventNum++;
  LOG(info) << "CbmRichHitProducer Event (or TimeSlice) " << fEventNum;
 
  fRichHits->Delete();
 
  // if CbmEvent does not exist then process standard event.
  // if CbmEvent exists then proceed all events in time slice.
  Int_t nUnits = (fCbmEvents != nullptr) ? fCbmEvents->GetEntriesFast() : 1;
 
  for (Int_t iUnit = 0; iUnit < nUnits; iUnit++) {
    CbmEvent* event = (fCbmEvents != nullptr)
                        ? static_cast<CbmEvent*>(fCbmEvents->At(iUnit))
                        : nullptr;
    ProcessData(event);
  }
}
 
void CbmRichHitProducer::ProcessData(CbmEvent* event) {
  if (event != NULL) {
    LOG(info) << "CbmRichHitProducer CbmEvent mode. CbmEvent # "
              << event->GetNumber();
    Int_t nofDigis = event->GetNofData(ECbmDataType::kRichDigi);
    LOG(info) << "nofDigis: " << nofDigis;
 
    for (Int_t iDigi = 0; iDigi < nofDigis; iDigi++) {
      Int_t digiIndex = event->GetIndex(ECbmDataType::kRichDigi, iDigi);
      ProcessDigi(event, digiIndex);
    }
 
  } else {
    for (Int_t iDigi = 0; iDigi < fDigiMan->GetNofDigis(ECbmModuleId::kRich);
         iDigi++) {
      ProcessDigi(event, iDigi);
    }
  }
}
 
void CbmRichHitProducer::ProcessDigi(CbmEvent* event, Int_t digiIndex) {
  const CbmRichDigi* digi = fDigiMan->Get<CbmRichDigi>(digiIndex);
  if (digi == nullptr) return;
  if (digi->GetAddress() < 0) return;
  CbmRichPixelData* data =
    CbmRichDigiMapManager::GetInstance().GetPixelDataByAddress(
      digi->GetAddress());
  TVector3 posPoint;
  posPoint.SetXYZ(data->fX, data->fY, data->fZ);
  TVector3 detPoint;
 
  CbmRichGeoManager::GetInstance().RotatePoint(
    &posPoint, &detPoint, !fRotationNeeded);
  AddHit(event, detPoint, digi->GetTime(), digiIndex);
}
 
 
void CbmRichHitProducer::AddHit(CbmEvent* event,
                                TVector3& posHit,
                                Double_t time,
                                Int_t index) {
  Int_t nofHits = fRichHits->GetEntriesFast();
  new ((*fRichHits)[nofHits]) CbmRichHit();
  CbmRichHit* hit = (CbmRichHit*) fRichHits->At(nofHits);
  hit->SetPosition(posHit);
  hit->SetDx(fHitError);
  hit->SetDy(fHitError);
  hit->SetRefId(index);
  hit->SetTime(time);
 
  if (event != NULL) { event->AddData(ECbmDataType::kRichHit, nofHits); }
}
 
void CbmRichHitProducer::Finish() { fRichHits->Clear(); }
 
 
ClassImp(CbmRichHitProducer)