#ifndef DILEPTONTRACKROTATOR_H
#define DILEPTONTRACKROTATOR_H

//#############################################################
//#                                                           #
//#         Class PairAnalysisTrackRotator                    #
//#                                                           #
//#  Authors:                                                 #
//#   Julian    Book,     Uni Ffm / Julian.Book@cern.ch       #
//#                                                           #
//#############################################################

#include <TNamed.h>
#include <TMath.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