]> git.uio.no Git - u/mrichter/AliRoot.git/commitdiff
Update (Zaida, Rossella)
authordainese <dainese@f7af4fe6-9843-0410-8265-dc069ae4e863>
Thu, 24 Feb 2011 22:45:07 +0000 (22:45 +0000)
committerdainese <dainese@f7af4fe6-9843-0410-8265-dc069ae4e863>
Thu, 24 Feb 2011 22:45:07 +0000 (22:45 +0000)
PWG3/vertexingHF/macros/makeTFile4CutsLctopKpi.C

index 59fc24a8e0547c83cee09920543021891208ec9c..67f2c5d02ff1b42211109a0c2586fae73b1dfba3 100644 (file)
@@ -62,6 +62,18 @@ void makeInputAliAnalysisTaskSELctopKpi(){
     prodcutsval[iv]=new Float_t[nptbins];
   }
 
+  //  inv. mass [GeV]
+  // pTK [GeV/c]
+  // pTPi [GeV/c]
+  // d0K [cm]   lower limit!
+  // d0Pi [cm]  lower limit!
+  // dist12 (cm)
+  // sigmavert (cm)
+  // dist prim-sec (cm)
+  // pM=Max{pT1,pT2,pT3} (GeV/c)
+  // cosThetaPoint
+  // Sum d0^2 (cm^2)
+  // dca cut (cm)
   for(Int_t ipt=0;ipt<nptbins;ipt++){
     prodcutsval[0][ipt]=0.18;
     prodcutsval[1][ipt]=0.4;
@@ -71,7 +83,7 @@ void makeInputAliAnalysisTaskSELctopKpi(){
     prodcutsval[5][ipt]=0.01;
     prodcutsval[6][ipt]=0.06;
     prodcutsval[7][ipt]=0.005;
-    prodcutsval[8][ipt]=0.;
+    prodcutsval[8][ipt]=0.7;
     prodcutsval[9][ipt]=0.;
     prodcutsval[10][ipt]=0.;
     prodcutsval[11][ipt]=0.05;
@@ -166,10 +178,6 @@ void makeInputAliAnalysisTaskSELctopKpi(){
   if(pidflag) cout<<"PID is used"<<endl;
   else cout<<"PID is not used"<<endl;
 
-  RDHFLctopKpiProd->SetMinPtCandidate(-1.);
-  RDHFLctopKpiAn->SetMinPtCandidate(-1.);
-  RDHFLctopKpiProd->SetMaxPtCandidate(10000.);
-  RDHFLctopKpiAn->SetMaxPtCandidate(10000.);
   cout<<"This is the object I'm going to save:"<<endl;
   RDHFLctopKpiProd->PrintAll();
   RDHFLctopKpiAn->PrintAll();
@@ -186,3 +194,211 @@ void makeInputAliAnalysisTaskSELctopKpi(){
   delete RDHFLctopKpiAn;
 
 }
+
+
+//macro to make a .root file (for significance maximization) which contains an AliRDHFCutsLctopKpi with loose set of cuts  and TParameter with the tighest value of these cuts
+
+void makeInputAliAnalysisTaskSESignificanceMaximization(){
+
+  AliRDHFCutsLctopKpi* RDHFLctopKpi=new AliRDHFCutsLctopKpi();
+  RDHFLctopKpi->SetName("loosercuts");
+  RDHFLctopKpi->SetTitle("Cuts for significance maximization");
+
+  AliESDtrackCuts* esdTrackCuts=new AliESDtrackCuts();
+  esdTrackCuts->SetRequireSigmaToVertex(kFALSE);
+  //default
+  esdTrackCuts->SetRequireTPCRefit(kTRUE);
+  esdTrackCuts->SetMinNClustersTPC(70);
+  esdTrackCuts->SetRequireITSRefit(kTRUE);
+  esdTrackCuts->SetMinNClustersITS(4);
+
+  esdTrackCuts->SetClusterRequirementITS(AliESDtrackCuts::kSPD,AliESDtrackCuts::kAny);
+  esdTrackCuts->SetMinDCAToVertexXY(0.);
+  esdTrackCuts->SetEtaRange(-0.8,0.8);
+  esdTrackCuts->SetPtRange(0.3,1.e10);
+  
+  RDHFLctopKpi->AddTrackCuts(esdTrackCuts);
+
+  const Int_t nvars=12;
+
+  const Int_t nptbins=4; //change this when adding pt bins!
+  Float_t ptbins[nptbins+1];
+  ptbins[0]=0.;
+  ptbins[1]=2.;
+  ptbins[2]=3.;
+  ptbins[3]=4.;
+  ptbins[4]=9999.;
+
+  RDHFLctopKpi->SetPtBins(nptbins+1,ptbins);
+
+  Float_t** rdcutsvalmine;
+  rdcutsvalmine=new Float_t*[nvars];
+  for(Int_t iv=0;iv<nvars;iv++){
+    rdcutsvalmine[iv]=new Float_t[nptbins];
+  }
+
+  //setting my cut values
+  //  inv. mass [GeV]
+  // pTK [GeV/c]
+  // pTPi [GeV/c]
+  // d0K [cm]   lower limit!
+  // d0Pi [cm]  lower limit!
+  // dist12 (cm)
+  // sigmavert (cm)
+  // dist prim-sec (cm)
+  // pM=Max{pT1,pT2,pT3} (GeV/c)
+  // cosThetaPoint
+  // Sum d0^2 (cm^2)
+  // dca cut (cm)
+  Float_t cutsMatrixLctopKpiStand[nptbins][nvars]=
+  {{0.18,0.4,0.5,0.,0.,0.01,0.06,0.005,0.7,0.,0.,0.05},
+   {0.18,0.4,0.5,0.,0.,0.01,0.06,0.005,0.7,0.,0.,0.05},
+   {0.18,0.4,0.5,0.,0.,0.01,0.06,0.005,0.7,0.,0.,0.05},
+   {0.18,0.4,0.5,0.,0.,0.01,0.06,0.005,0.7,0.,0.,0.05}};
+
+  //CREATE TRANSPOSE MATRIX...REVERSE INDICES as required by AliRDHFCuts
+  Float_t **cutsMatrixTransposeStand=new Float_t*[nvars];
+  for(Int_t iv=0;iv<nvars;iv++)cutsMatrixTransposeStand[iv]=new Float_t[nptbins];
+  for (Int_t ibin=0;ibin<nptbins;ibin++){
+    for (Int_t ivar = 0; ivar<nvars; ivar++){
+      cutsMatrixTransposeStand[ivar][ibin]=cutsMatrixLctopKpiStand[ibin][ivar];
+    }
+  }
+  RDHFLctopKpi->SetCuts(nvars,nptbins,cutsMatrixTransposeStand);
+
+
+  Int_t nvarsforopt=RDHFLctopKpi->GetNVarsForOpt();
+  Int_t dim=4; //set this!!
+  Bool_t *boolforopt;
+  boolforopt=new Bool_t[nvars];
+  if(dim>nvarsforopt){
+    cout<<"Number of variables for optimization has probably changed, check and edit accordingly"<<endl;
+    return;
+  } else {
+    if(dim==nvarsforopt){
+      boolforopt=RDHFLctopKpi->GetVarsForOpt();
+    }else{
+      TString *names;
+      names=new TString[nvars];
+      TString answer="";
+      Int_t checktrue=0;
+      names=RDHFLctopKpi->GetVarNames();
+      for(Int_t i=0;i<nvars;i++){
+       cout<<names[i]<<" for opt? (y/n)"<<endl;
+       cin>>answer;
+       if(answer=="y") {
+         boolforopt[i]=kTRUE;
+         checktrue++;
+       }
+       else boolforopt[i]=kFALSE;
+      }
+      if (checktrue!=dim) {
+       cout<<"Error! You set "<<checktrue<<" kTRUE instead of "<<dim<<endl;
+       return;
+      }
+      RDHFLctopKpi->SetVarsForOpt(dim,boolforopt);
+    }
+  }
+
+
+  Float_t tighterval[dim][nptbins];
+  // 5 dist12 (cm)        <---
+  // 7 dist prim-sec (cm) <---
+  // 9 cosThetaPoint      <---
+  // 10 Sum d0^2 (cm^2)   <---
+  // 11 dca cut (cm)      
+  // {0.18,0.4,0.5,0.,0.,0.01 <--- ,0.06,0.005 <--- ,0.7,0. <--- ,0. <--- ,0.05 }
+
+  //number of steps for each variable is set in the AddTask arguments (default=8)
+  //set this!!
+  for(Int_t ipt=0;ipt<nptbins;ipt++){
+    tighterval[0][ipt]=0.03;
+    tighterval[1][ipt]=0.02;
+    tighterval[2][ipt]=1.;
+    tighterval[3][ipt]=0.01;
+  }
+
+
+
+  TString name=""; 
+  Int_t arrdim=dim*nptbins;
+  cout<<"Will save "<<arrdim<<" TParameter<float>"<<endl;
+  TClonesArray max("TParameter<float>",arrdim);
+  for(Int_t ival=0;ival<dim;ival++){
+    for(Int_t jpt=0;jpt<nptbins;jpt++){
+      name=Form("par%dptbin%d",ival,jpt);
+      cout<<"Setting "<<name.Data()<<" to "<<tighterval[ival][jpt]<<endl;
+      new(max[jpt*dim+ival])TParameter<float>(name.Data(),tighterval[ival][jpt]);
+    }
+  }
+
+  Bool_t flagPID=kFALSE;
+  RDHFLctopKpi->SetUsePID(flagPID);
+
+  RDHFLctopKpi->PrintAll();
+  printf("Use PID? %s\n",flagPID ? "yes" : "no");
+
+    //pid settings
+    //1. kaon: default one
+  AliAODPidHF* pidObjK=new AliAODPidHF();
+  Double_t sigmasK[5]={3.,1.,1.,3.,2.};
+  pidObjK->SetSigma(sigmasK);
+  pidObjK->SetAsym(kTRUE);
+  pidObjK->SetMatch(1);
+  pidObjK->SetTPC(kTRUE);
+  pidObjK->SetTOF(kTRUE);
+  pidObjK->SetITS(kTRUE);
+  Double_t plimK[2]={0.5,0.8};
+  pidObjK->SetPLimit(plimK,2);
+  
+  RDHFLctopKpi->SetPidHF(pidObjK);
+
+    //2. pion 
+  AliAODPidHF* pidObjpi=new AliAODPidHF();
+  pidObjpi->SetTPC(kTRUE);
+  Double_t sigmaspi[5]={3.,0.,0.,0.,0.};
+  pidObjpi->SetSigma(sigmaspi);
+
+  RDHFLctopKpi->SetPidpion(pidObjpi);
+
+  // 3. proton
+  AliAODPidHF* pidObjp=new AliAODPidHF();
+  Double_t sigmasp[5]={3.,1.,1.,3.,2.};
+  pidObjp->SetSigma(sigmasp);
+  pidObjp->SetAsym(kTRUE);
+  pidObjp->SetMatch(1);
+  pidObjp->SetTPC(kTRUE);
+  pidObjp->SetTOF(kTRUE);
+  pidObjp->SetITS(kTRUE);
+  Double_t plimp[2]={1.,2.};
+  pidObjp->SetPLimit(plimp,2);
+
+  RDHFLctopKpi->SetPidprot(pidObjp);
+
+  //activate pileup rejection (for pp)
+  //RDHFLctopKpi->SetOptPileup(AliRDHFCuts::kRejectPileupEvent);
+
+  //Do not recalculate the vertex
+  RDHFLctopKpi->SetRemoveDaughtersFromPrim(kFALSE); //activate for pp
+
+  TString cent="";
+  //centrality selection (Pb-Pb)
+  Float_t minc=20,maxc=80;
+  RDHFLctopKpi->SetMinCentrality(minc);
+  RDHFLctopKpi->SetMaxCentrality(maxc);
+  cent=Form("%.0f%.0f",minc,maxc);
+  //  RDHFLctopKpi->SetUseCentrality(AliRDHFCuts::kCentV0M); //kCentOff,kCentV0M,kCentTRK,kCentTKL,kCentCL1,kCentInvalid
+  RDHFLctopKpi->SetUseCentrality(AliRDHFCuts::kCentOff); //kCentOff,kCentV0M,kCentTRK,kCentTKL,kCentCL1,kCentInvalid
+
+  //temporary
+  //  RDHFLctopKpi->SetFixRefs();
+
+  TFile* fout=new TFile(Form("cuts4SignifMaxim%s%s%sRecVtx%sPileupRej.root", RDHFLctopKpi->GetUseCentrality()==0 ? "pp" : "PbPb",cent.Data(),RDHFLctopKpi->GetIsPrimaryWithoutDaughters() ? "" : "No",RDHFLctopKpi->GetOptPileUp() ? "" : "No"),"recreate");   //set this!! 
+
+  fout->cd();
+  RDHFLctopKpi->Write();
+  max.Write();
+  fout->Close();
+}
+