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