]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - PWGJE/EMCALJetTasks/AliJetContainer.h
changes confusing histo names
[u/mrichter/AliRoot.git] / PWGJE / EMCALJetTasks / AliJetContainer.h
index 227fb2cf33e1780e5701cf455ec313c6a0440f20..037c6ee416bd0baa3b37db5ceb85c86d7a9485de 100644 (file)
@@ -1,9 +1,7 @@
 #ifndef AliJetContainer_H
 #define AliJetContainer_H
 
-//
-// container with name, TClonesArray and cuts for jets
-//
+// $Id$
 
 class TClonesArray;
 class TString;
@@ -13,6 +11,7 @@ class AliEmcalJet;
 class AliVEvent;
 class AliParticleContainer;
 class AliClusterContainer;
+class AliLocalRhoParameter;
 
 #include "AliRhoParameter.h"
 
@@ -35,37 +34,36 @@ class AliJetContainer : public AliEmcalContainer {
   void SetEMCALGeometry();
   void SetEMCALGeometry(AliEMCALGeometry *p) {fGeom = p;}
   void LoadRho(AliVEvent *event);
+  void LoadLocalRho(AliVEvent *event);
 
   void                        SetJetAcceptanceType(JetAcceptanceType type)         { fJetAcceptanceType          = type ; }
-
   void                        ResetCuts();
-
   void                        SetJetEtaPhiEMCAL() ;
   void                        SetJetEtaPhiTPC()   ;
   void                        SetRunNumber(Int_t r)                                { fRunNumber = r;                      }
-
   void                        SetJetEtaLimits(Float_t min, Float_t max)            { fJetMinEta = min, fJetMaxEta = max ; }
   void                        SetJetPhiLimits(Float_t min, Float_t max)            { fJetMinPhi = min, fJetMaxPhi = max ; }
   void                        SetJetAreaCut(Float_t cut)                           { fJetAreaCut     = cut              ; }
-  void                        SetPercAreaCut(Float_t p)                            { if(fJetRadius==0.) AliWarning("JetRadius not set. Area cut will be 0"); fJetAreaCut = p*TMath::Pi()*fJetRadius*fJetRadius; }
-
+  void                        SetPercAreaCut(Float_t p)                            { if(fJetRadius==0.) AliWarning("JetRadius not set. Area cut will be 0"); 
+                                                                                     fJetAreaCut = p*TMath::Pi()*fJetRadius*fJetRadius; }
+  void                        SetZLeadingCut(Float_t zemc, Float_t zch)            { fZLeadingEmcCut = zemc; fZLeadingChCut = zch ; }
+  void                        SetNEFCut(Float_t min = 0., Float_t max = 1.)        { fNEFMinCut = min; fNEFMaxCut = max;  }
   void                        SetAreaEmcCut(Double_t a = 0.99)                     { fAreaEmcCut     = a                ; }
   void                        SetJetPtCut(Float_t cut)                             { fJetPtCut       = cut              ; }
   void                        SetJetRadius(Float_t r)                              { fJetRadius      = r                ; } 
   virtual void                SetRhoName(const char *n)                            { fRhoName        = n                ; }
+  virtual void                SetLocalRhoName(const char *n)                       { fLocalRhoName   = n                ; }
+  void                        SetFlavourCut(Int_t myflavour)                       { fFlavourSelection = myflavour;}
   void                        SetMaxClusterPt(Float_t b)                           { fMaxClusterPt   = b                ; }
   void                        SetMaxTrackPt(Float_t b)                             { fMaxTrackPt     = b                ; }
   void                        SetPtBiasJetClus(Float_t b)                          { fPtBiasJetClus  = b                ; }
   void                        SetNLeadingJets(Int_t t)                             { fNLeadingJets   = t                ; }
   void                        SetPtBiasJetTrack(Float_t b)                         { fPtBiasJetTrack = b                ; }
-
   void                        SetLeadingHadronType(Int_t t)                        { fLeadingHadronType = t             ; }
   void                        SetJetBitMap(UInt_t m)                               { fJetBitMap      = m                ; }
   void                        SetJetTrigger(UInt_t t=AliVEvent::kEMCEJE)           { fJetTrigger     = t                ; }
-
   void                        ConnectParticleContainer(AliParticleContainer *c)    { fParticleContainer = c             ; }
   void                        ConnectClusterContainer(AliClusterContainer *c)      { fClusterContainer  = c             ; }
-
   AliEmcalJet                *GetLeadingJet(const char* opt="")          ;
   AliEmcalJet                *GetJet(Int_t i)                       const;
   AliEmcalJet                *GetAcceptJet(Int_t i)                 const;
@@ -75,14 +73,21 @@ class AliJetContainer : public AliEmcalContainer {
   AliEmcalJet                *GetNextJet(Int_t i=-1)                     ;
   void                        GetMomentum(TLorentzVector &mom, Int_t i) const;
   Bool_t                      AcceptJet(AliEmcalJet* jet)           const;
-  Bool_t                      AcceptBiasJet(AliEmcalJet* jet)       const; 
+  Bool_t                      AcceptBiasJet(AliEmcalJet* jet)       const;
+  Int_t                       GetFlavourCut()                       const    {return fFlavourSelection;}
   Int_t                       GetNJets()                            const    {return GetNEntries();}
   Double_t                    GetLeadingHadronPt(AliEmcalJet* jet)  const;
   void                        GetLeadingHadronMomentum(TLorentzVector &mom, AliEmcalJet* jet)  const;
+  Double_t                    GetZ(AliEmcalJet *jet, TLorentzVector mom) const;
+  Double_t                    GetZLeadingEmc(AliEmcalJet *jet)      const;
+  Double_t                    GetZLeadingCharged(AliEmcalJet *jet)  const;
   AliRhoParameter            *GetRhoParameter()                              {return fRho;}
-  Double_t                    GetRhoVal()                           const    { if (fRho) return fRho->GetVal(); else return 0;}
+  Double_t                    GetRhoVal()                           const    {if (fRho) return fRho->GetVal(); else return 0;}
   const TString&              GetRhoName()                          const    {return fRhoName;}
+  AliLocalRhoParameter       *GetLocalRhoParameter()                const    {return fLocalRho;}
+  const TString&              GetLocalRhoName()                     const    {return fLocalRhoName;}
   Double_t                    GetJetPtCorr(Int_t i)                 const;
+  Double_t                    GetJetPtCorrLocal(Int_t i)            const;
   Float_t                     GetJetRadius()                        const    {return fJetRadius;}
   Float_t                     GetJetEtaMin()                        const    {return fJetMinEta;}
   Float_t                     GetJetEtaMax()                        const    {return fJetMaxEta;}
@@ -97,6 +102,8 @@ class AliJetContainer : public AliEmcalContainer {
   JetAcceptanceType           fJetAcceptanceType;    //  acceptance type
   Float_t                     fJetRadius;            //  jet radius
   TString                     fRhoName;              //  Name of rho object
+  TString                     fLocalRhoName;         //  Name of local rho object
+  Int_t                       fFlavourSelection;     //  selection on jet flavour
   Float_t                     fPtBiasJetTrack;       //  select jets with a minimum pt track
   Float_t                     fPtBiasJetClus;        //  select jets with a minimum pt cluster
   Float_t                     fJetPtCut;             //  cut on jet pt
@@ -108,6 +115,10 @@ class AliJetContainer : public AliEmcalContainer {
   Float_t                     fJetMaxPhi;            //  maximum phi jet acceptance  
   Float_t                     fMaxClusterPt;         //  maximum cluster constituent pt to accept the jet
   Float_t                     fMaxTrackPt;           //  maximum track constituent pt to accept the jet
+  Float_t                     fZLeadingEmcCut;       //  maximum z,leading neutral
+  Float_t                     fZLeadingChCut;        //  maximum z,leading charged
+  Float_t                     fNEFMinCut;            //  minimum NEF in a jet
+  Float_t                     fNEFMaxCut;            //  maximum NEF in a jet
   Int_t                       fLeadingHadronType;    //  0 = charged, 1 = neutral, 2 = both
   Int_t                       fNLeadingJets;         //  how many jets are to be considered the leading jet(s)
   UInt_t                      fJetBitMap;            //  bit map of accepted jets
@@ -116,6 +127,7 @@ class AliJetContainer : public AliEmcalContainer {
   AliClusterContainer        *fClusterContainer;     //  cluster container (jet constituents)
 
   AliRhoParameter            *fRho;                  //! event rho for these jets
+  AliLocalRhoParameter       *fLocalRho;             //! event local rho for these jets
   AliEMCALGeometry           *fGeom;                 //! emcal geometry
   Int_t                       fRunNumber;            //! run number
 
@@ -123,10 +135,8 @@ class AliJetContainer : public AliEmcalContainer {
   AliJetContainer(const AliJetContainer& obj); // copy constructor
   AliJetContainer& operator=(const AliJetContainer& other); // assignment
 
-  ClassDef(AliJetContainer,3);
+  ClassDef(AliJetContainer,6);
 
 };
 
 #endif
-
-