PWG2/SPECTRA -> PWGLF/SPECTRA migration

[u/mrichter/AliRoot.git] / PWGLF / SPECTRA / MultEvShape / AliAnalysisCentralCutESD.h

diff --git a/PWGLF/SPECTRA/MultEvShape/AliAnalysisCentralCutESD.h b/PWGLF/SPECTRA/MultEvShape/AliAnalysisCentralCutESD.h
new file mode 100644 (file)
index 0000000..5c49739
--- /dev/null
+++ b/PWGLF/SPECTRA/MultEvShape/AliAnalysisCentralCutESD.h
@@ -0,0 +1,75 @@
+#ifndef ALIANALYSISCENTRALCUTESD_H
+#define ALIANALYSISCENTRALCUTESD_H
+
+
+/*
+ * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved.
+ * See cxx source for full Copyright notice
+ * $Id$
+ */
+
+// ---------------------------------------------------
+//  ESD particle level cuts for azimuthal isotropic
+//  expansion in highly central collisions analysis
+//  author: Cristian Andrei
+//           acristian@niham.nipne.ro
+// ---------------------------------------------------
+
+
+#include <TString.h>
+#include <TPDGCode.h>
+
+#include "AliAnalysisCuts.h"
+
+
+class TObject;
+class TList;
+class TF1;
+
+class AliESDtrack;
+
+
+class AliAnalysisCentralCutESD: public AliAnalysisCuts {
+public:
+    AliAnalysisCentralCutESD(const char *name="AliAnalysisCentralCutESD", const char *title="ESD_cuts");
+    virtual ~AliAnalysisCentralCutESD();
+
+    Bool_t  IsSelected(TObject* obj);
+    Bool_t  IsSelected(TList* /*list*/) {return kTRUE;}
+
+    void SetPartType(PDG_t type) {fReqPID = kTRUE; fPartType = type;}
+    void SetPIDtype(TString type) {fPIDtype = type;}
+    void SetPriorFunctions(Bool_t pfunc) {fPriorsFunc = pfunc;}
+    
+    void SetReqIsCharged() {fReqCharge = kTRUE;} 
+
+private:
+  AliAnalysisCentralCutESD(const AliAnalysisCentralCutESD &ref);
+  AliAnalysisCentralCutESD& operator=(const AliAnalysisCentralCutESD &ref);
+
+    Bool_t fReqPID;  //kTRUE -> run the PID
+
+    Bool_t fReqCharge; //kTRUE -> only charged particles selected
+
+    PDG_t fPartType; //can be kPi*, kK* or kProton
+
+    TString fPIDtype; //PID method -> can be Custom or Bayesian
+    Bool_t fPriorsFunc; // if kTRUE -> use function priors 
+
+    Double_t fPartPriors[10]; //prior probabilities
+    TF1  *fElectronFunction; //momentum dependence of the prior probs
+    TF1  *fMuonFunction; //momentum dependence of the prior probs
+    TF1  *fPionFunction; //momentum dependence of the prior probs
+    TF1  *fKaonFunction; //momentum dependence of the prior probs
+    TF1  *fProtonFunction; //momentum dependence of the prior probs
+
+    Bool_t IsA(AliESDtrack *track, PDG_t type);
+
+    Bool_t IsCharged  (AliESDtrack* const track) const;
+
+    Double_t GetPriors(Int_t i, Double_t P);
+
+    ClassDef(AliAnalysisCentralCutESD, 1);
+};
+
+#endif