updates from salvatore

[u/mrichter/AliRoot.git] / PWGGA / EMCALJetTasks / AliFJWrapper.h

diff --git a/PWGGA/EMCALJetTasks/AliFJWrapper.h b/PWGGA/EMCALJetTasks/AliFJWrapper.h
index 2246f49f277160d76b0b981e426d15de6a8bfd36..c9baf2eea00910bb55227711ac5aed1c6032fe3b 100644 (file)
--- a/PWGGA/EMCALJetTasks/AliFJWrapper.h
+++ b/PWGGA/EMCALJetTasks/AliFJWrapper.h
@@ -31,6 +31,7 @@ class AliFJWrapper
   const char*                           GetName()            const { return fName;                       }
   const char*                           GetTitle()           const { return fTitle;                      }
   Double_t                              GetJetArea         (UInt_t idx) const;
+  fastjet::PseudoJet                    GetJetAreaVector   (UInt_t idx) const;
   Double_t                              GetJetSubtractedPt (UInt_t idx) const;
   virtual std::vector<double>           GetSubtractedJetsPts(Double_t median_pt = -1, Bool_t sorted = kFALSE);
 
@@ -254,6 +255,19 @@ Double_t AliFJWrapper::GetJetArea(UInt_t idx) const
   return retval;
 }
 
+//_________________________________________________________________________________________________
+fastjet::PseudoJet AliFJWrapper::GetJetAreaVector(UInt_t idx) const
+{
+  // Get the jet area as vector.
+  fastjet::PseudoJet retval;
+  if ( idx < fInclusiveJets.size() ) {
+    retval = fClustSeq->area_4vector(fInclusiveJets[idx]);
+  } else {
+    AliError(Form("[e] ::GetJetArea wrong index: %d",idx));
+  }
+  return retval;
+}
+
 //_________________________________________________________________________________________________
 std::vector<double> AliFJWrapper::GetSubtractedJetsPts(Double_t median_pt, Bool_t sorted)
 { 
@@ -304,6 +318,7 @@ void AliFJWrapper::GetMedianAndSigma(Double_t &median, Double_t &sigma, Int_t re
 
   if (!fClustSeq) {
     AliError("[e] Run the jfinder first.");
+    return;
   }
 
   Double_t mean_area = 0;