AliEMCalTriggerWeightHandler.h

Go to the documentation of this file.

#ifndef ALIEMCALTRIGGERWEIGHTHANDLER_H
#define ALIEMCALTRIGGERWEIGHTHANDLER_H
/* Copyright(c) 1998-2014, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */

#include <TObject.h>

class TF1;
class TObjArray;

class AliGenPythiaEventHeader;
class AliMCEvent;

namespace EMCalTriggerPtAnalysis {

class AliEMCalTriggerPtHardWeight : public TObject {
public:
  AliEMCalTriggerPtHardWeight():  TObject(), fPtMin(-1), fPtMax(-1), fWeight(-1) {}
  AliEMCalTriggerPtHardWeight(double ptmin, double ptmax, double weight) :  TObject(), fPtMin(ptmin), fPtMax(ptmax), fWeight(weight) {}
  virtual ~AliEMCalTriggerPtHardWeight() {}

  void SetPtRange(Double_t ptmin, Double_t ptmax) { fPtMin = ptmin; fPtMax = ptmax; }
  void SetWeight(double weight) { fWeight = weight; }

  bool IsSelected(Double_t pthard) const { return pthard >= fPtMin && pthard < fPtMax; }
  Double_t GetWeight() const { return fWeight; }
  Double_t GetPtMin() const { return fPtMin; }
  Double_t GetPtMax() const { return fPtMax; }
  void GetPtLimits(Double_t &ptmin, Double_t &ptmax) const { ptmin = fPtMin; ptmax = fPtMax; }

private:
  double                  fPtMin;             
  double                  fPtMax;             
  double                  fWeight;            

  ClassDef(AliEMCalTriggerPtHardWeight, 1)
};

class AliEMCalTriggerWeightHandler : public TObject {
public:
  AliEMCalTriggerWeightHandler();
  virtual ~AliEMCalTriggerWeightHandler();

  void SetUseCrossSection(bool useCrossSection) { fUseCrossSection = useCrossSection; }

  void SetWeightModel(const TF1 *model) { fWeightModel = model; }

  void SetWeightForBin(double ptmin, double ptmax, double weight);

  double GetEventWeight(const AliMCEvent *const event) const;
  double GetEventWeight(const AliGenPythiaEventHeader * const header) const;

protected:
  const AliEMCalTriggerPtHardWeight *FindWeight(Double_t pthard) const;

private:
  const TF1         *fWeightModel;          
  TObjArray         *fBinWeights;           
  bool               fUseCrossSection;      

  ClassDef(AliEMCalTriggerWeightHandler, 1);
};

} /* namespace EMCalTriggerPtAnalysis */

#endif /* ALIEMCALTRIGGERWEIGHTHANDLER_H */