Fixes for Lc->v0+X and DStar (Francesco, Rossella)

[u/mrichter/AliRoot.git] / JETAN / AliAnalysisTaskJetBackgroundSubtract.h

diff --git a/JETAN/AliAnalysisTaskJetBackgroundSubtract.h b/JETAN/AliAnalysisTaskJetBackgroundSubtract.h
index ca507206b2350852d60edcdf3311bc84e84798a4..9f0e0a193e79a0d0db4e39cf0cf022e9ef633566 100644 (file)
--- a/JETAN/AliAnalysisTaskJetBackgroundSubtract.h
+++ b/JETAN/AliAnalysisTaskJetBackgroundSubtract.h
@@ -50,14 +50,15 @@ class AliAnalysisTaskJetBackgroundSubtract : public AliAnalysisTaskSE
     virtual void   AddJetBranch(const char* c){fJBArray->Add(new TObjString(c));}
     virtual void   SetSubtractionMethod(Int_t i){fSubtraction = i;}
     virtual Int_t  GetSubtractionMethod(){return fSubtraction;}
+    virtual void   SetKeepJets(Bool_t b = kTRUE){fKeepJets = b;}
     virtual void   SetBackgroundBranch(char* c){fBackgroundBranch = c;}  
-    virtual void   SetNonStdFile(char* c){fNonStdFile = c;}  
+    virtual void   SetNonStdOutputFile(char* c){fNonStdFile = c;}  
     virtual void   SetToReplace(char* c){fReplaceString1 = c;}  
     const char*    GetToReplace(){return fReplaceString1.Data();}  
     virtual void   SetReplacementMask(char* c){fReplaceString2 = c;}  
-    const char*   GetReplacementMask(){fReplaceString2.Data();}  
+    const char*    GetReplacementMask(){return fReplaceString2.Data();}  
    
-    enum {kNoSubtract = 0,kArea,k4Area};
+    enum {kNoSubtract = 0,kArea,k4Area,kRhoRecalc,kRhoRC};
  
  private:
 
@@ -66,8 +67,12 @@ class AliAnalysisTaskJetBackgroundSubtract : public AliAnalysisTaskSE
     AliAnalysisTaskJetBackgroundSubtract(const AliAnalysisTaskJetBackgroundSubtract&);
     AliAnalysisTaskJetBackgroundSubtract& operator=(const AliAnalysisTaskJetBackgroundSubtract&);
     Bool_t RescaleJetMomentum(AliAODJet *jet,Float_t pT);
+    Bool_t RescaleJet4vector(AliAODJet *jet,TLorentzVector backgroundv);
+    Int_t  MultFromJetRefs(TClonesArray* jets);
+    Double_t RecalcRho(TClonesArray* fbkgclusters,Double_t meanarea);
+    Double_t RhoRC(TClonesArray* fbkgclustersRC);
     void ResetOutJets();
-
+    void PrintAODContents();
 
     AliAODEvent     *fAODOut;                // ! where we take the jets from and they are modified
     AliAODEvent     *fAODIn;                 // ! where we may take the background from, only in case we do not find it in the output
@@ -79,11 +84,23 @@ class AliAnalysisTaskJetBackgroundSubtract : public AliAnalysisTaskSE
     TString         fReplaceString1;     // To construct the new output name  
     TString         fReplaceString2;     // To construct the new output name   
     Int_t           fSubtraction;       // Parameter for subtraction mode 
+    Bool_t          fKeepJets;          // keeps the jets with negative p_t rescaled to 0.1 GeV
     TList *fInJetArrayList; //! transient list to make ease the handling of input jets
     TList *fOutJetArrayList; //! transient list to make ease the reset of output jets
+
+    TH2F*         fh2CentvsRho;   //! centrality vs background density  
+    TH2F*         fh2CentvsSigma;   //! centrality vs background sigma  
+    TH2F*         fh2MultvsRho;   //! centrality vs background density  
+    TH2F*         fh2MultvsSigma;   //! centrality vs background sigma  
+    TH2F*         fh2ShiftEta;   //! extended correction Eta
+    TH2F*         fh2ShiftPhi;   //! extended correction Phi
+    TH2F*         fh2ShiftEtaLeading;   //! extended correction Eta leading jet
+    TH2F*         fh2ShiftPhiLeading;   //! extended correction Phi leading jet
+
+ 
     TList *fHistList; //! the histograms output list
    
-    ClassDef(AliAnalysisTaskJetBackgroundSubtract, 1) 
+    ClassDef(AliAnalysisTaskJetBackgroundSubtract, 6) 
 };
  
 #endif