PairAnalysisTrackRotator.h

Go to the documentation of this file.

#ifndef DILEPTONTRACKROTATOR_H
#define DILEPTONTRACKROTATOR_H
 
//#############################################################
//#                                                           #
//#         Class PairAnalysisTrackRotator                    #
//#                                                           #
//#  Authors:                                                 #
//#   Julian    Book,     Uni Ffm / Julian.Book@cern.ch       #
//#                                                           #
//#############################################################
 
#include <TMath.h>
#include <TNamed.h>
#include <TRandom3.h>
 
class PairAnalysisTrackRotator : public TNamed {
public:
  enum ERotationType { kRotatePositive, kRotateNegative, kRotateBothRandom };
 
  PairAnalysisTrackRotator();
  PairAnalysisTrackRotator(const char* name, const char* title);
 
  virtual ~PairAnalysisTrackRotator();
 
  //Setters
  void SetIterations(UInt_t niter) { fIterations = niter; }
  void SetRotationType(ERotationType type) { fRotationType = type; }
  void SetStartAnglePhi(Double_t phi) { fStartAnglePhi = phi; }
  void SetConeAnglePhi(Double_t phi) { fConeAnglePhi = phi; }
 
  //Getters
  Int_t GetIterations() const { return fIterations; }
  ERotationType GetRotationType() const { return fRotationType; }
  Double_t GetStartAnglePhi() const { return fStartAnglePhi; }
  Double_t GetConeAnglePhi() const { return fConeAnglePhi; }
 
  Double_t GetAngle() const {
    return fStartAnglePhi + (2 * gRandom->Rndm() - 1) * fConeAnglePhi;
  }
  Double_t GetCharge() const { return TMath::Nint(gRandom->Rndm()); }
 
private:
  UInt_t fIterations          = 1;                  // number of iterations
  ERotationType fRotationType = kRotateBothRandom;  // which track to rotate
  Double_t fStartAnglePhi     = TMath::Pi();  // starting angle for rotation
  Double_t fConeAnglePhi =
    TMath::Pi();  // opening angle in phi for multiple rotation
 
  PairAnalysisTrackRotator(const PairAnalysisTrackRotator& c);
  PairAnalysisTrackRotator& operator=(const PairAnalysisTrackRotator& c);
 
  ClassDef(PairAnalysisTrackRotator, 1)  // Track rotation configuration
};
 
 
#endif