AliEmcalPythiaInfo.h

Go to the documentation of this file.

#ifndef ALIPYTHIAINFO_H
#define ALIPYTHIAINFO_H

/* Copyright(c) 1998-2015, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */

#include <TNamed.h>
#include "AliTLorentzVector.h"

class AliEmcalPythiaInfo : public TNamed{

 public:
  AliEmcalPythiaInfo();
  AliEmcalPythiaInfo(const char* name);

  void SetPartonFlag6(Int_t flag6)                                        { fPartonFlag6   = flag6                ; }
  void SetParton6(Float_t pt, Float_t eta, Float_t phi, Float_t mass=0)   { fParton6.SetPtEtaPhiM(pt,eta,phi,mass); }
  void SetPartonFlag7(Int_t flag7)                                        { fPartonFlag7   = flag7                ; }
  void SetParton7(Float_t pt, Float_t eta, Float_t phi, Float_t mass=0)   { fParton7.SetPtEtaPhiM(pt,eta,phi,mass); }
  void SetPythiaEventWeight(Float_t ptWeight)                             { fPythiaEventWeight = ptWeight         ; }
  
  Int_t                    GetPartonFlag6()       const { return fPartonFlag6         ; }
  Float_t                  GetPartonPt6()         const { return fParton6.Pt()        ; }
  Float_t                  GetPartonEta6()        const { return fParton6.Eta()       ; }
  Float_t                  GetPartonPhi6()        const { return fParton6.Phi_0_2pi() ; }

  Int_t                    GetPartonFlag7()       const { return fPartonFlag7         ; }
  Float_t                  GetPartonPt7()         const { return fParton7.Pt()        ; }
  Float_t                  GetPartonEta7()        const { return fParton7.Eta()       ; }
  Float_t                  GetPartonPhi7()        const { return fParton7.Phi_0_2pi() ; }

  const AliTLorentzVector& GetParton6Momentum()   const { return fParton6             ; }
  const AliTLorentzVector& GetParton7Momentum()   const { return fParton7             ; }

  Float_t                  GetPythiaEventWeight() const { return fPythiaEventWeight   ; }

 private: 
  Int_t                fPartonFlag6       ; 
  AliTLorentzVector    fParton6           ; 
  Int_t                fPartonFlag7       ; 
  AliTLorentzVector    fParton7           ; 
  Float_t              fPythiaEventWeight ; 
    
  AliEmcalPythiaInfo(const AliEmcalPythiaInfo&);
  AliEmcalPythiaInfo& operator=(const AliEmcalPythiaInfo&);
  
  ClassDef(AliEmcalPythiaInfo, 1);
};
#endif