AliTrackContainerToyModel.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 "AliTrackContainerToyModel.h"
#include <TMath.h>
#include <TRandom3.h>
ClassImp(AliTrackContainerToyModel);

AliTrackContainerToyModel::AliTrackContainerToyModel() :
  AliTrackContainer(),
  fTrackScalePt(1),
  fTrackEtaWindow(0.9),
  fRandomizeEtaPhi(0)  
{
}

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

Bool_t AliTrackContainerToyModel::GetMomentumFromTrack(TLorentzVector &mom, const AliVTrack* track, Double_t mass) const
{
  Bool_t r = AliTrackContainer::GetMomentumFromTrack(mom, track, mass);
  ScalePtOfLorentzVector(mom);
  SetRandomEtaPhiOfLorentzVector(mom);
  return r;
}

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

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

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

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

void AliTrackContainerToyModel::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 AliTrackContainerToyModel::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 AliTrackContainerToyModel::ExecOnce()
{
      if (gRandom) delete gRandom;
     gRandom = new TRandom3(0);
  }