AliMCParticleContainerToyModel.h

Go to the documentation of this file.

#ifndef ALIMCPARTICLECONTAINERTOYMODEL_H
#define ALIMCPARTICLECONTAINERTOYMODEL_H

/**************************************************************************
* 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.                  *
**************************************************************************/

class AliAODEvent;

#include "AliAODEvent.h"
#include "AliMCParticleContainer.h"


class AliMCParticleContainerToyModel : public AliMCParticleContainer {
public:
  AliMCParticleContainerToyModel();
  AliMCParticleContainerToyModel(const char *name);

  void SetTrackScalePt(Double_t t) { fTrackScalePt  = t; }
  void SetRandomizeEtaPhi(Double_t t) { fRandomizeEtaPhi  = t; }

  virtual Bool_t GetMomentumFromParticle(TLorentzVector &mom, const AliAODMCParticle* track, Double_t mass) const;
  virtual Bool_t GetMomentum(TLorentzVector &mom, Int_t i) const;
  virtual Bool_t GetAcceptMomentum(TLorentzVector &mom, Int_t i) const;
  virtual Bool_t GetNextMomentum(TLorentzVector &mom);
  virtual Bool_t GetNextAcceptMomentum(TLorentzVector &mom);
  
  void ScalePtOfLorentzVector(TLorentzVector &mom) const;
   void SetRandomEtaPhiOfLorentzVector(TLorentzVector &mom) const;
protected:
  void                   ExecOnce(); 
  Double_t               fTrackScalePt;           //scaling of the track pT by given fraction (0....1)
  Double_t               fTrackEtaWindow;         //eta acceptance 
  Double_t               fRandomizeEtaPhi;        //break correlations by setting random eta phi to the tracks
 

 private:
  ClassDef(AliMCParticleContainerToyModel, 1);
};
#endif