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"

ClassImp(AliTrackContainerToyModel);

AliTrackContainerToyModel::AliTrackContainerToyModel() :
  AliTrackContainer(),
  fTrackScalePt(1)
{
}

AliTrackContainerToyModel::AliTrackContainerToyModel(const char *name) :
  AliTrackContainer(name),
  fTrackScalePt(1)
{
}

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

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

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

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

Bool_t AliTrackContainerToyModel::GetNextAcceptMomentum(TLorentzVector &mom)
{
  Bool_t r = AliTrackContainer::GetNextAcceptMomentum(mom);
  ScalePtOfLorentzVector(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);
  }
}