committing on behalf of Silvia... new macros for HFE analysis on the train (Markus
[u/mrichter/AliRoot.git] / PWG3 / hfe / macros / ConfigHFEtrd.C
1 AliAnalysisTaskHFE *ConfigHFEtrd(Bool_t useMC){
2   //
3   // HFE standard task configuration
4   //
5
6   AliHFEcuts *hfecuts = new AliHFEcuts("hfeCutsTRD","HFE cuts including TRD PID");
7   hfecuts->CreateStandardCuts();
8   hfecuts->SetMinNClustersTPC(110);
9   hfecuts->SetCutITSpixel(AliHFEextraCuts::kFirst);
10   hfecuts->SetCheckITSLayerStatus(kFALSE);
11   hfecuts->SetSigmaToVertex(10);
12   hfecuts->SetQAOn();
13   hfecuts->SetMinNTrackletsTRD(6);
14
15   AliAnalysisTaskHFE *task = new AliAnalysisTaskHFE("HFEanalysisTRD");
16   printf("task %p\n", task);
17   task->SetHFECuts(hfecuts);
18
19   // Define Variables
20   AliHFEvarManager *vm = task->GetVarManager();
21   vm->AddVariable("pt");
22   vm->AddVariable("eta");
23   vm->AddVariable("phi");
24   vm->AddVariable("charge");
25   vm->AddVariable("source");
26
27   // Define PID
28   AliHFEpid *pid = task->GetPID();
29   pid->AddDetector("TOF", 0);
30   pid->AddDetector("TRD", 1);
31   pid->AddDetector("TPC", 2);
32   AliHFEpidTRD *trdpid = pid->GetDetPID(AliHFEpid::kTRDpid);
33   trdpid->SetPIDMethod(AliHFEpidTRD::kLQ);
34   trdpid->SetElectronEfficiency(0.71);
35   // New threshold parameters for LHC10d
36   const Double_t kNparams = 4;
37   Double_t par70[kNparams] = {2.29791e-02, 1.17777e-02, 4.29038e-02, 2.11955e+00};
38   Double_t par75[kNparams] = {-1.89881e+00, -2.31561e-01, 1.98553e+00, -1.04081e-01};
39   Double_t par80[kNparams] = {-1.85572e+00, -3.88746e-01, 2.01796e+00, -1.61435e-01};
40   Double_t par85[kNparams] = {2.44547e-01, -8.26956e-02, 4.49075e-02, -7.84496e-01};
41   Double_t par90[kNparams] = {1.64799e-01, 2.07216e-01, 4.43009e-01, 1.19677e+00};
42   Double_t par95[kNparams] = {8.43719e-01, 4.95028e-02, 1.11009e-01, 4.58459e+00};  
43   trdpid->SetThresholdParameters(0.71, par70);
44   trdpid->SetThresholdParameters(0.76, par75);
45   trdpid->SetThresholdParameters(0.81, par80);
46   trdpid->SetThresholdParameters(0.86, par85);
47   trdpid->SetThresholdParameters(0.91, par90);
48   trdpid->SetThresholdParameters(0.96, par95);
49   
50   // QA
51   printf("task %p\n", task);
52   task->SetQAOn(AliAnalysisTaskHFE::kPIDqa);
53   task->SetQAOn(AliAnalysisTaskHFE::kMCqa);    
54   task->SwitchOnPlugin(AliAnalysisTaskHFE::kIsElecBackGround);
55   task->SwitchOnPlugin(AliAnalysisTaskHFE::kSecVtx);
56   task->SwitchOnPlugin(AliAnalysisTaskHFE::kDEstep);
57
58   printf("*************************************\n");
59   printf("Configuring standard Task:\n");
60   task->PrintStatus();
61   pid->PrintStatus();
62   printf("*************************************\n");
63   return task;
64 }