Fix Coverity reports

[u/mrichter/AliRoot.git] / PWG3 / dielectron / AliDielectron.h

diff --git a/PWG3/dielectron/AliDielectron.h b/PWG3/dielectron/AliDielectron.h
index 5be7eb492d3c043486e9cb6a90b79d577bd9304b..a33d3dc49c826c936bed4fc9e2b8c2978097d48d 100644 (file)
--- a/PWG3/dielectron/AliDielectron.h
+++ b/PWG3/dielectron/AliDielectron.h
@@ -23,13 +23,13 @@
 #include <AliKFParticle.h>
 
 #include "AliDielectronHistos.h"
-#include "AliDielectronPair.h"
 
 class AliVEvent;
 class THashList;
 class AliDielectronCF;
 class AliDielectronDebugTree;
 class AliDielectronTrackRotator;
+class AliDielectronPair;
 
 //________________________________________________________________
 class AliDielectron : public TNamed {
@@ -76,7 +76,10 @@ public:
   const THashList * GetHistogramList() const { return fHistos?fHistos->GetHistogramList():0x0; }
 
   Bool_t HasCandidates() const { return GetPairArray(1)?GetPairArray(1)->GetEntriesFast()>0:0; }
-
+  Bool_t HasCandidatesLikeSign() const {
+    return (GetPairArray(0)&&GetPairArray(2)) ? (GetPairArray(0)->GetEntriesFast()>0 || GetPairArray(2)->GetEntriesFast()>0) : 0;
+  }
+  
   void SetCFManagerPair(AliDielectronCF * const cf) { fCfManagerPair=cf; }
   AliDielectronCF* GetCFManagerPair() const { return fCfManagerPair; }
 
@@ -84,6 +87,9 @@ public:
 
   void SetTrackRotator(AliDielectronTrackRotator * const rot) { fTrackRotator=rot; }
   AliDielectronTrackRotator* GetTrackRotator() const { return fTrackRotator; }
+
+  void SetHasMC(Bool_t hasMC) { fHasMC = hasMC; }
+  Bool_t GetHasMC() const     { return fHasMC;  }
   
   void SetDebugTree(AliDielectronDebugTree * const tree) { fDebugTree=tree; }
   
@@ -125,7 +131,7 @@ private:
   AliDielectronDebugTree *fDebugTree;  // Debug tree output
 
   Bool_t fPreFilterUnlikeOnly;  //Apply PreFilter either in +- or to ++/--/+- individually
-
+  Bool_t fHasMC;                //If we run with MC, at the moment only needed in AOD
   
   void FillTrackArrays(AliVEvent * const ev, Int_t eventNr=0);
   void PairPreFilter(Int_t arr1, Int_t arr2, TObjArray &arrTracks1, TObjArray &arrTracks2);
@@ -145,7 +151,7 @@ private:
   void ProcessMC();
   
   void  FillHistograms(const AliVEvent *ev);
-  void  FillHistogramsPair(AliDielectronPair *pair);
+  void  FillHistogramsPair(AliDielectronPair *pair,Bool_t fromPreFilter=kFALSE);
   void  FillHistogramsTracks(TObjArray **tracks);
 
   void  FillDebugTree();