CbmMuchHitProducerIdeal.cxx

Go to the documentation of this file.

/* Copyright (C) 2006-2021 GSI Helmholtzzentrum fuer Schwerionenforschung, Darmstadt
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Anna Senger, Andrey Lebedev, Denis Bertini [committer], Florian Uhlig */
 
#include "CbmMuchHitProducerIdeal.h"
 
#include "CbmMuchGeoScheme.h"
#include "CbmMuchPixelHit.h"
#include "CbmMuchPoint.h"
#include "FairRootManager.h"
#include "TClonesArray.h"
#include "TFile.h"
#include "TRandom.h"
#include "TVector3.h"
 
#include <Logger.h>
 
#include <cmath>
 
using std::fabs;
 
CbmMuchHitProducerIdeal::CbmMuchHitProducerIdeal(const char* digiFileName, Int_t Id)
  : FairTask("CbmMuchHitProducerIdeal")
  , fMuchPoints(NULL)
  , fMuchPixelHits(NULL)
  , fSigmaX(0.01)
  , fSigmaY(0.01)
  , fSigmaZ(0.)
  , fId(Id)
  , fDigiFile(digiFileName)
  , fGeoScheme(NULL)
{
}
 
CbmMuchHitProducerIdeal::~CbmMuchHitProducerIdeal() {}
 
InitStatus CbmMuchHitProducerIdeal::Init()
{
  FairRootManager* ioman = FairRootManager::Instance();
  fMuchPoints            = static_cast<TClonesArray*>(ioman->GetObject("MuchPoint"));
  fMuchPixelHits         = new TClonesArray("CbmMuchPixelHit", 100);
  ioman->Register("MuchPixelHit", "MUCH", fMuchPixelHits, IsOutputBranchPersistent("MuchPixelHit"));
 
  // Initialize GeoScheme
  fGeoScheme = CbmMuchGeoScheme::Instance();
  TFile* oldFile     = gFile;
  TDirectory* oldDir = gDirectory;
 
  TFile* file = new TFile(fDigiFile);
  LOG_IF(fatal, !file) << "Could not open file " << fDigiFile;
  TObjArray* stations = file->Get<TObjArray>("stations");
  LOG_IF(fatal, !stations) << "TObjArray stations not found in file " << fDigiFile;
  file->Close();
  file->Delete();
  gFile      = oldFile;
  gDirectory = oldDir;
 
  fGeoScheme->Init(stations, fId);
  return kSUCCESS;
}
 
void CbmMuchHitProducerIdeal::Exec(Option_t*)
{
  static Int_t eventNo = 0;
  fMuchPixelHits->Clear();
  Int_t iHit          = 0;
  Int_t nofMuchPoints = fMuchPoints->GetEntriesFast();
  for (Int_t iPoint = 0; iPoint < nofMuchPoints; iPoint++) {
    const CbmMuchPoint* point = static_cast<const CbmMuchPoint*>(fMuchPoints->At(iPoint));
 
    // Smear position
    Double_t dX = gRandom->Gaus(0, fSigmaX);
    Double_t dY = gRandom->Gaus(0, fSigmaY);
    Double_t dZ = gRandom->Gaus(0, fSigmaZ);
 
    dX = (fabs(dX) < 3 * fSigmaX) ? dX : (dX > 0) ? 3 * fSigmaX : -3 * fSigmaX;
    dY = (fabs(dY) < 3 * fSigmaY) ? dY : (dY > 0) ? 3 * fSigmaY : -3 * fSigmaY;
    dZ = (fabs(dZ) < 3 * fSigmaZ) ? dZ : (dZ > 0) ? 3 * fSigmaZ : -3 * fSigmaZ;
 
    TVector3 hitPos(point->GetXIn() + dX, point->GetYIn() + dY, point->GetZIn() + dZ);
    TVector3 hitPosErr(fSigmaX, fSigmaY, fSigmaZ);
 
    Int_t address = point->GetDetectorID();
    Int_t planeId = fGeoScheme->GetLayerSideNr(address);
 
    new ((*fMuchPixelHits)[iHit++]) CbmMuchPixelHit(address, hitPos, hitPosErr, 0, iPoint, planeId);
  }
  eventNo++;
  LOG(info) << "CbmMuchHitProducerIdeal::Exec: eventNo=" << eventNo << " nofPoints=" << fMuchPoints->GetEntriesFast()
            << " nofHits=" << fMuchPixelHits->GetEntriesFast();
}
 
void CbmMuchHitProducerIdeal::Finish() {}
 
ClassImp(CbmMuchHitProducerIdeal)