743d93a565fbbc9b48ebf1dbcda870964b0afa61
[u/mrichter/AliRoot.git] / PWG3 / vertexingHF / macros / AddTaskSignificance.C
1 AliAnalysisTaskSESignificance *AddTaskSignificance(TString filename="cuts4SignifMaximDplus.root",Int_t decCh=0,Bool_t readMC=kFALSE,Int_t flagOPartAntiPart=0,Int_t nofsteps=8)
2 {
3   //
4   // Test macro for the AliAnalysisTaskSE for D meson candidates
5   // Invariant mass histogram and
6   // association with MC truth (using MC info in AOD)
7   //  R. Bala, bala@to.infn.it
8   // C. Bianchin, cbianchi@pd.infn.it
9   // Get the pointer to the existing analysis manager via the static access method.
10   //============================================================================
11   AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
12   if (!mgr) {
13     ::Error("AddTaskSignificanceN", "No analysis manager to connect to.");
14     return NULL;
15   }
16
17   TFile* filecuts=new TFile(filename.Data());
18   if(!filecuts->IsOpen()){
19     cout<<"Input file not found: exit"<<endl;
20     return;
21   }
22   
23   AliRDHFCuts *analysiscuts=0x0;
24   TString suffix="";
25   TString suffix2="";
26   if(flagOPartAntiPart==1) suffix2="P"; //P=particle, A=antiparticle
27   if(flagOPartAntiPart==-1) suffix2="A";
28
29   TString cutsobjname="loosercuts";
30   //Analysis cuts
31   switch (decCh){
32   case 0:
33     analysiscuts = (AliRDHFCutsDplustoKpipi*)filecuts->Get(cutsobjname);
34     suffix=Form("Dplus%s",suffix2.Data());
35     break;
36   case 1:
37     analysiscuts = (AliRDHFCutsD0toKpi*)filecuts->Get(cutsobjname);
38     suffix=Form("D0%s",suffix2.Data());
39     break;
40   case 2:
41     analysiscuts = (AliRDHFCutsDstartoKpipi*)filecuts->Get(cutsobjname);
42     suffix=Form("Dstar%s",suffix2.Data());
43     break;
44   case 3:
45     analysiscuts = (AliRDHFCutsDstoKKpi*)filecuts->Get(cutsobjname);
46     suffix=Form("Ds%s",suffix2.Data());
47     break;
48   case 4:
49     analysiscuts = (AliRDHFCutsD0toKpipipi*)filecuts->Get(cutsobjname);
50     suffix=Form("D04%s",suffix2.Data());
51     break;
52   case 5:
53     analysiscuts = (AliRDHFCutsLctopKpi*)filecuts->Get(cutsobjname);
54     suffix=Form("Lc%s",suffix2.Data());
55     break;
56   }
57   //ptbins
58
59   if(!analysiscuts){
60     cout<<"Specific AliRDHFCuts not found"<<endl;
61     return;
62   }
63   
64   const Int_t nptbins=analysiscuts->GetNPtBins();
65   Float_t* ptbins=analysiscuts->GetPtBinLimits();
66
67   // Analysis task    
68   const Int_t npars=analysiscuts->GetNVarsForOpt();//numbers of var for opt
69   Bool_t* varsforopt=analysiscuts->GetVarsForOpt();
70   cout<<"pt bins= "<<nptbins<<"  varsforopt= "<<npars<<"  nvars= "<<analysiscuts->GetNVars()<<endl;
71   Int_t* nofcells=new Int_t[npars];//={4,4,4,4};
72   Float_t** looses;
73   looses=new Float_t*[npars];
74   Float_t** tights;
75   tights=new Float_t*[npars];
76   TString axisTitle[npars];
77   TString parname="";
78
79   for (Int_t ivop=0;ivop<npars;ivop++){
80     looses[ivop]=new Float_t[nptbins];
81     tights[ivop]=new Float_t[nptbins];
82     nofcells[ivop]=nofsteps;
83   }
84
85   Int_t count=0;
86   
87   for (Int_t ip=0;ip<nptbins;ip++){
88     for(Int_t iv=0;iv<analysiscuts->GetNVars();iv++){
89       if(varsforopt[iv]){
90         looses[count][ip]=analysiscuts->GetCutValue(iv,ip);
91         axisTitle[count]=(analysiscuts->GetVarNames())[iv];
92         parname=Form("par%dptbin%d",count,ip);
93         TParameter<float>* par=(TParameter<float>*)filecuts->Get(parname.Data());
94         tights[count][ip]=par->GetVal();
95         count++;
96       }
97     }
98     count=0;
99   }
100   
101   //creation TList of AliMultiDimVector
102   TList *listMDV=new TList();
103   listMDV->SetOwner();
104   listMDV->SetName("listMDV");
105
106   for (Int_t ip=0;ip<nptbins;ip++){
107     Float_t *loosescut=new Float_t[npars];
108     Float_t *tightscut=new Float_t[npars];
109     for(Int_t i=0;i<npars;i++){
110       loosescut[i]=looses[i][ip];
111       tightscut[i]=tights[i][ip];
112     }
113     Float_t ptbincut[2]={ptbins[ip],ptbins[ip+1]};
114     TString mdvname=Form("multiDimVectorPtBin%d",ip);
115     AliMultiDimVector *mv=new AliMultiDimVector(mdvname.Data(),"MultiDimVector",1,ptbincut,npars,nofcells,loosescut,tightscut,axisTitle);
116     listMDV->Add(mv);
117   }
118   
119   AliAnalysisTaskSESignificance *sigTask = new AliAnalysisTaskSESignificance("SignificanceAnalysis",listMDV,analysiscuts,decCh,AliRDHFCuts::kAll);//AliRDHFCuts::kCandidate
120   sigTask->SetReadMC(readMC);
121   //sigTask->SetDoLikeSign(kTRUE);
122   //sigTask->SetBFeedDown(AliAnalysisTaskSESignificance::kBoth);
123   sigTask->SetDebugLevel(3);
124   sigTask->SetFillWithPartAntiPartBoth(flagOPartAntiPart);
125   mgr->AddTask(sigTask);
126
127   TString contname=Form("cinputSig%s",suffix.Data());
128   // Create containers for input/output
129   AliAnalysisDataContainer *cinputSig = mgr->CreateContainer(contname.Data(),TChain::Class(),AliAnalysisManager::kInputContainer);
130   TString outputfile = AliAnalysisManager::GetCommonFileName();
131   TString outputhistos = outputfile += ":PWG3_D2H_Significance"; 
132   contname=Form("coutputSig%s",suffix.Data());
133   AliAnalysisDataContainer *coutputSig = mgr->CreateContainer(contname.Data(),TList::Class(),AliAnalysisManager::kOutputContainer,outputfile.Data());
134   contname=Form("coutputmv%s",suffix.Data());
135   AliAnalysisDataContainer *coutputmv = mgr->CreateContainer(contname.Data(),TList::Class(),AliAnalysisManager::kOutputContainer,outputfile.Data());
136
137
138   mgr->ConnectInput(sigTask,0,mgr->GetCommonInputContainer());
139   
140   mgr->ConnectOutput(sigTask,1,coutputSig);
141   
142   mgr->ConnectOutput(sigTask,2,coutputmv);
143  
144   return sigTask;
145 }
146