AliMCParticleContainerToyModel.cxx

Go to the documentation of this file.

/*************************************************************************
* Copyright(c) 1998-2016, ALICE Experiment at CERN, All rights reserved. *
*                                                                        *
* Author: The ALICE Off-line Project.                                    *
* Contributors are mentioned in the code where appropriate.              *
*                                                                        *
* Permission to use, copy, modify and distribute this software and its   *
* documentation strictly for non-commercial purposes is hereby granted   *
* without fee, provided that the above copyright notice appears in all   *
* copies and that both the copyright notice and this permission notice   *
* appear in the supporting documentation. The authors make no claims     *
* about the suitability of this software for any purpose. It is          *
* provided "as is" without express or implied warranty.                  *
**************************************************************************/

#include "AliMCParticleContainerToyModel.h"
#include <TMath.h>
#include <TRandom3.h>
ClassImp(AliMCParticleContainerToyModel);

AliMCParticleContainerToyModel::AliMCParticleContainerToyModel() :
  AliMCParticleContainer(),
  fTrackScalePt(1),
  fTrackEtaWindow(0.9),
  fRandomizeEtaPhi(0)
 
 

{
}

AliMCParticleContainerToyModel::AliMCParticleContainerToyModel(const char *name) :
  AliMCParticleContainer(name),
  fTrackScalePt(1),
  fTrackEtaWindow(0.9),
  fRandomizeEtaPhi(0) 
{
}

Bool_t AliMCParticleContainerToyModel::GetMomentumFromParticle(TLorentzVector &mom, const AliAODMCParticle* track, Double_t mass) const
{
  Bool_t r = AliMCParticleContainer::GetMomentumFromParticle(mom, track, mass);
  ScalePtOfLorentzVector(mom);
  SetRandomEtaPhiOfLorentzVector(mom);
  return r;
}

Bool_t AliMCParticleContainerToyModel::GetMomentum(TLorentzVector &mom, Int_t i) const
{
  Bool_t r = AliMCParticleContainer::GetMomentum(mom, i);
  ScalePtOfLorentzVector(mom);
  SetRandomEtaPhiOfLorentzVector(mom);
  return r;
}

Bool_t AliMCParticleContainerToyModel::GetAcceptMomentum(TLorentzVector &mom, Int_t i) const
{
  Bool_t r = AliMCParticleContainer::GetAcceptMomentum(mom, i);
  ScalePtOfLorentzVector(mom);
  SetRandomEtaPhiOfLorentzVector(mom);
  return r;
}

Bool_t AliMCParticleContainerToyModel::GetNextMomentum(TLorentzVector &mom)
{
  Bool_t r = AliMCParticleContainer::GetNextMomentum(mom);
  ScalePtOfLorentzVector(mom);
  SetRandomEtaPhiOfLorentzVector(mom);
  return r;
}

Bool_t AliMCParticleContainerToyModel::GetNextAcceptMomentum(TLorentzVector &mom)
{
  Bool_t r = AliMCParticleContainer::GetNextAcceptMomentum(mom);
  ScalePtOfLorentzVector(mom);
  SetRandomEtaPhiOfLorentzVector(mom);
  return r;
}

void AliMCParticleContainerToyModel::ScalePtOfLorentzVector(TLorentzVector &mom) const
{
  if(fTrackScalePt<1.){
   
    Double_t pTscale = fTrackScalePt*mom.Pt();
    Double_t phiscale   = mom.Phi();
    Double_t thetascale = 2.*TMath::ATan(TMath::Exp(-1.*(mom.Eta())));
    Double_t pXscale    = pTscale * TMath::Cos(phiscale);
    Double_t pYscale    = pTscale * TMath::Sin(phiscale);
    Double_t pZscale    = pTscale/TMath::Tan(thetascale);
    Double_t pscale=TMath::Sqrt(pTscale*pTscale+pZscale*pZscale);
    mom.SetPxPyPzE(pXscale, pYscale, pZscale, pscale);
  }
}
void AliMCParticleContainerToyModel::SetRandomEtaPhiOfLorentzVector(TLorentzVector &mom) const
{
  if(fRandomizeEtaPhi==1){
   
   
      
    Double_t pTscale = mom.Pt();
 
    Double_t etascale = 2.*fTrackEtaWindow * gRandom->Rndm() - fTrackEtaWindow;
    Double_t phiscale = 2.* TMath::Pi() * gRandom->Rndm();

    
    Double_t thetascale = 2.*TMath::ATan(TMath::Exp(-1.*(etascale)));
    Double_t pXscale    = pTscale * TMath::Cos(phiscale);
    Double_t pYscale    = pTscale * TMath::Sin(phiscale);
    Double_t pZscale    = pTscale/TMath::Tan(thetascale);
    Double_t pscale=TMath::Sqrt(pTscale*pTscale+pZscale*pZscale);
    mom.SetPxPyPzE(pXscale, pYscale, pZscale, pscale);
    
  }
}

void AliMCParticleContainerToyModel::ExecOnce()
{
  
    if (gRandom) delete gRandom;
     gRandom = new TRandom3(0);
  }