O2MCApplicationEvalMat.cxx

Go to the documentation of this file.

// Copyright 2019-2020 CERN and copyright holders of ALICE O2.
// See https://alice-o2.web.cern.ch/copyright for details of the copyright holders.
// All rights not expressly granted are reserved.
//
// This software is distributed under the terms of the GNU General Public
// License v3 (GPL Version 3), copied verbatim in the file "COPYING".
//
// In applying this license CERN does not waive the privileges and immunities
// granted to it by virtue of its status as an Intergovernmental Organization
// or submit itself to any jurisdiction.
 
#include <Steer/O2MCApplicationEvalMat.h>
#include <SimConfig/MatMapParams.h>
#include <Generators/GeneratorGeantinos.h>
#include <TObjArray.h>
#include <TLorentzVector.h>
#include <FairPrimaryGenerator.h>
 
namespace o2
{
namespace steer
{
void O2MCApplicationEvalMat::FinishPrimary()
{
  mMaterialBudgetMap->FinishPrimary(mC1[mMode], mPhi);
  FairMCApplication::FinishPrimary();
}
 
void O2MCApplicationEvalMat::Stepping()
{
  // dispatch to MaterialBudget Map
  if (fMC->IsNewTrack()) {
    TLorentzVector p;
    fMC->TrackMomentum(p);
    Float_t x, y, z;
    fMC->TrackPosition(x, y, z);
 
    mPhi = p.Phi() * 180. / TMath::Pi();
    if (mPhi < 0.) {
      mPhi += 360.;
    }
 
    mC1[0] = p.Theta() * 180. / TMath::Pi();
    mC1[1] = p.Eta();
    mC1[2] = z;
  }
 
  mMaterialBudgetMap->Stepping();
}
 
void O2MCApplicationEvalMat::BeginEvent()
{
  if (!mMaterialBudgetMap) {
    auto& p = o2::conf::MatMapParams::Instance();
    Float_t c1min, c1max;
    Int_t n1;
    if (p.ntheta != 0) {
      // theta-phi binning
      mMode = 0;
      n1 = p.ntheta;
      c1min = p.thetamin;
      c1max = p.thetamax;
    } else if (p.neta != 0) {
      // eta-phi binning
      mMode = 1;
      n1 = p.neta;
      c1min = p.etamin;
      c1max = p.etamax;
    } else if (p.nzv != 0) {
      // z-phi binning
      mMode = 2;
      n1 = p.nzv;
      c1min = p.zvmin;
      c1max = p.zvmax;
    }
    printf("MaterialBudgetMap: %5d %13.3f %13.3f %5d %13.3f %13.3f %13.3f %13.3f\n", n1, c1min, c1max, p.nphi, p.phimin, p.phimax, p.rmax, p.zmax);
    mMaterialBudgetMap = new MaterialBudgetMap("Map", mMode,
                                               n1, c1min, c1max, p.nphi, p.phimin, p.phimax, p.rmin, p.rmax, p.zmax);
 
    auto gen = GetGenerator();
    gen->GetListOfGenerators()->Clear();
    gen->AddGenerator(new o2::eventgen::GeneratorGeantinos(mMode, n1, c1min, c1max, p.nphi, p.phimin, p.phimax, p.rmin, p.rmax, p.zmax));
  }
  mMaterialBudgetMap->BeginEvent();
}
 
void O2MCApplicationEvalMat::FinishEvent()
{
  mMaterialBudgetMap->FinishEvent();
  O2MCApplicationBase::FinishEvent();
}
 
} // namespace steer
} // namespace o2