Coverity

[u/mrichter/AliRoot.git] / PWG3 / vertexingHF / AliCFTaskVertexingHF.h

diff --git a/PWG3/vertexingHF/AliCFTaskVertexingHF.h b/PWG3/vertexingHF/AliCFTaskVertexingHF.h
index 28cc22e25bef764b4538476b724a74f9e13c0c7c..2e2eb92dbec918baa00c8d09c8f7eae0ffaf61b0 100644 (file)
--- a/PWG3/vertexingHF/AliCFTaskVertexingHF.h
+++ b/PWG3/vertexingHF/AliCFTaskVertexingHF.h
@@ -58,6 +58,10 @@ public:
                kStepRecoPID         = 9
        };
 
+       enum {
+               kSnail = 0,    // slow configuration, all variables
+               kCheetah = 1   // fast configuration, only a subset of variables
+       };
        
        AliCFTaskVertexingHF();
        AliCFTaskVertexingHF(const Char_t* name, AliRDHFCuts* cuts);
@@ -102,6 +106,22 @@ public:
 
        void SetFakeSelection(Int_t fakeSel = 0) {fFakeSelection=fakeSel;}
        Int_t GetFakeSelection(){return fFakeSelection;}
+
+       void SetRejectCandidateIfNotFromQuark(Bool_t opt){fRejectIfNoQuark=opt;}
+       Bool_t GetRejectCandidateIfNotFromQuark(){return fRejectIfNoQuark;}
+
+       void SetUseMCVertex(Bool_t opt){fUseMCVertex=opt;}
+       Bool_t GetUseMCVertex(){return fUseMCVertex;}
+       
+
+       void SetKeepDsViaPhi(){fDsOption=1;}
+       void SetKeepDsViaK0star(){fDsOption=2;}
+       void SetKeepAllDs(){fDsOption=3;}
+
+       Bool_t ProcessDs(Int_t returnCodeDs) const;
+
+       void SetConfiguration(Int_t configuration) {(configuration == kSnail) ? Printf("Slow configuration chosen, all variables will be used!") : Printf("Fast configuration chosen, all variablesOnly pt, y, phi, ct, fake, z_vtx, centrality and multiplicity will be used!"); fConfiguration = configuration;} 
+       Int_t GetConfiguration() const {return fConfiguration;} 
        
 protected:
        AliCFManager   *fCFManager;   //  pointer to the CF manager
@@ -130,9 +150,12 @@ protected:
        Char_t fSign;                 // flag to decide wheter to keep D0 only (0), D0bar only (1), or both D0 and D0bar (2)
         Bool_t fCentralitySelection;  //flag to switch off the centrality selection
        Int_t  fFakeSelection;  //selection flag for fakes tracks 
+       Bool_t fRejectIfNoQuark;  // flag to remove events not geenrated with PYTHIA
+       Bool_t fUseMCVertex;  // flag to use MC vertex (useful when runnign in pp)
+       Int_t  fDsOption;     // Ds decay option
+       Int_t fConfiguration; // configuration (slow / fast) of the CF --> different variables will be allocated (all / reduced number)
 
-
-       ClassDef(AliCFTaskVertexingHF,6); // class for HF corrections as a function of many variables
+       ClassDef(AliCFTaskVertexingHF,8); // class for HF corrections as a function of many variables
 };
 
 #endif