Update of the alignment-data fileting macro including a fix for the access to the...

[u/mrichter/AliRoot.git] / PWG1 / AliESDresolMakerFast.cxx

diff --git a/PWG1/AliESDresolMakerFast.cxx b/PWG1/AliESDresolMakerFast.cxx
index 4666e62cafb1ff9fb3fc840ad61e65c9ca2159dc..8f45120938d58150503f75a490712e2d9efaefde 100644 (file)
--- a/PWG1/AliESDresolMakerFast.cxx
+++ b/PWG1/AliESDresolMakerFast.cxx
@@ -29,7 +29,6 @@
   // Make esd chain
   //
   .x ~/rootlogon.C
-  gSystem->Load("libSTAT.so");
   gSystem->AddIncludePath("-I$ALICE_ROOT/TPC/macros");
   gROOT->LoadMacro("$ALICE_ROOT/TPC/macros/AliXRDPROOFtoolkit.cxx+");  
   AliXRDPROOFtoolkit tool;
@@ -38,13 +37,25 @@
   //
   // Load macros
   //
+  gSystem->Load("libSTAT.so");
   .L $ALICE_ROOT/PWG1/AliESDresolParams.cxx+
   .L $ALICE_ROOT/PWG1/AliESDresolMakerFast.cxx+
-  AliESDresolParams params; 
   TCut cutDCA="Tracks[].fCchi2<100&&abs(Tracks[].fP[4])<8&&abs(Tracks[].fP[3])<1&&sqrt(Tracks[].fC[0])/(0.2+abs(Tracks[].fP[4]))<0.02&&abs(Tracks[].fX)<3&&Tracks[].fITSncls>4&&Tracks.fTPCncls>40"
   //
+  // Create resolution
+  //
   TObjArray * array = AliESDresolMakerFast::MakeParamPrimFast(tree,cutDCA,0.95,2000);
-
+  AliESDresolParams params; 
+  params.SetResolPrimFast(array);
+  params.SetInstance(&params);
+  TFile f("resolParams.root","recreate");
+  param.Write("resolParams");
+  //
+  //
+  TF2 f2sy("f2sy","AliESDresolParams::SGetResolPrimFast(0,x,y)",0,10,0,2);
+  TF1 f1sy("f1sy","AliESDresolParams::SGetResolPrimFast(0,x,0)",0,10);
+  f2sy->Draw("surf2")
+  f1sy->Draw();
 
 */
 
@@ -85,9 +96,13 @@ TObjArray * AliESDresolMakerFast::MakeParamPrimFast(TTree * tree, TCut & cutDCA,
   // entries  - total number of entries (tracks) used
 
   /* Example
-    
-     TCut cutDCA="Tracks[].fCchi2<100&&abs(Tracks[].fP[4])<8&&abs(Tracks[].fP[3])<1&&sqrt(Tracks[].fC[0])/(0.2+abs(Tracks[].fP[4]))<0.02&&abs(Tracks[].fX)<3&&Tracks[].fITSncls>4&&Tracks.fTPCncls>40"
-     fraction =0.95;
+     // its adjusted cuts
+
+     TCut cut1pt= "sqrt(Tracks[].fC[14])/(1+abs(Tracks[].fP[4]))^2.5<0.01";   
+     TCut cutsy = "sqrt(Tracks[].fC[0])/(0.2+abs(Tracks[].fP[4]))<0.02";
+     
+     TCut cutDCA="Tracks[].fCchi2<100&&abs(Tracks[].fP[4])<8&&abs(Tracks[].fP[3])<1&&abs(Tracks[].fX)<3&&Tracks[].fITSncls>4&&Tracks.fTPCncls>40"+cut1pt+cutsy;
+     fraction =0.90;
      entries  = 1000;
   */
 
@@ -120,7 +135,7 @@ TObjArray * AliESDresolMakerFast::MakeParamPrimFast(TTree * tree, TCut & cutDCA,
   //
   // theta param
   //
-  TString * dcathParam=  TStatToolkit::FitPlane(tree, "sqrt(Tracks[].fC[9])/(0.1+abs(Tracks[].fP[4]))", "(abs(Tracks[].fP[4]))++(abs(Tracks[].fP[4]))^2++(abs(Tracks[].fP[3]))++(abs(Tracks[].fP[3]))^2++(abs(Tracks[].fP[4]))*(abs(Tracks[].fP[3]))",  cutDCA, chi2,npoints,fitParam,covMatrix,fraction,0,entries);  
+  TString * dcathParam=  TStatToolkit::FitPlane(tree, "sqrt(Tracks[].fC[9])/((0.1+abs(Tracks[].fP[4])*(1+abs(Tracks[].fParamP.fP[3]^2))))", "(abs(Tracks[].fP[4]))++(abs(Tracks[].fP[4]))^2++(abs(Tracks[].fP[3]))++(abs(Tracks[].fP[3]))^2++(abs(Tracks[].fP[4]))*(abs(Tracks[].fP[3]))",  cutDCA, chi2,npoints,fitParam,covMatrix,fraction,0,entries);  
   printf("Theta resol\t%s\n",dcathParam->Data());   
   tree->SetAlias("dcathParam",dcathParam->Data());
   array->AddAt(new TVectorD(fitParam),3);
@@ -134,12 +149,35 @@ TObjArray * AliESDresolMakerFast::MakeParamPrimFast(TTree * tree, TCut & cutDCA,
   return array;
 }
 
+/*
+void scalePt(){
+  //
+  TString * dca1ptParamS4=  TStatToolkit::FitPlane(tree, "(Tracks[].fC[14])", "abs(Tracks[].fP[4])^4",  cutDCA, chi2,npoints,fitParam,covMatrix,fraction,0,entries);  
+  printf("1pt resol\t%s\n",dca1ptParamS4->Data());  
+  tree->SetAlias("dca1ptParam4",dca1ptParamS4->Data());
+  //
+  TString * dca1ptParamS34=  TStatToolkit::FitPlane(tree, "(Tracks[].fC[14])", "abs(Tracks[].fP[4])^4++abs(Tracks[].fP[4])^3",  cutDCA, chi2,npoints,fitParam,covMatrix,fraction,0,entries);  
+  printf("1pt resol\t%s\n",dca1ptParamS34->Data());  
+  tree->SetAlias("dca1ptParam34",dca1ptParamS34->Data());
+  //
+  TString * dca1ptParamS234=  TStatToolkit::FitPlane(tree, "(Tracks[].fC[14])", "abs(Tracks[].fP[4])^4++abs(Tracks[].fP[4])^3++abs(Tracks[].fP[4])^2",  cutDCA, chi2,npoints,fitParam,covMatrix,fraction,0,entries);  
+  printf("1pt resol\t%s\n",dca1ptParamS234->Data());  
+  tree->SetAlias("dca1ptParam234",dca1ptParamS234->Data());
+}
+
+*/
+
+
+
+
 
 
 TObjArray * AliESDresolMakerFast::MakeParamRFast(TTree * tree,  TCut &cutV0, Float_t fraction, Int_t entries){
   //
   // 
   //
+  TObjArray *array = new TObjArray;
+ 
   Double_t chi2=0;
   Int_t    npoints=0;
   TVectorD fitParam;
@@ -147,15 +185,15 @@ TObjArray * AliESDresolMakerFast::MakeParamRFast(TTree * tree,  TCut &cutV0, Flo
   //
   //
   /*
+    //TCut cutGood="abs(V0s[].GetEffMass(0,0))<0.05||abs(V0s[].GetEffMass(2,2)-0.5)<0.05"
     TCut  cutV0 = "abs((V0s[].fParamP.fC[0]))<3&&abs(V0s[].fParamP.fP[3])<1&&abs(V0s[].fParamP.fP[4])<8";//
   */
-  TObjArray *array = new TObjArray;
   //
   //
   //
-  TString * v0sigmaY= TStatToolkit::FitPlane(tree,"sqrt(sqrt((V0s[].fParamP.fC[0])))","abs(V0s[].fParamP.fP[4])++V0s[].fParamP.fX++abs(V0s[].fParamP.fP[4])^2++V0s[].fParamP.fX^2++abs(V0s[].fParamP.fP[4])*V0s[].fParamP.fX++abs(V0s[].fParamP.fP[4])*V0s[].fParamP.fX^2", cutV0, chi2,npoints,fitParam,covMatrix,fraction,0,entries);
+  TString * v0sigmaY= TStatToolkit::FitPlane(tree,"sqrt(sqrt((V0s[].fParamP.fC[0]))/(0.2+abs(V0s[].fParamP.fP[4])))","abs(V0s[].fParamP.fP[4])++V0s[].fParamP.fX++abs(V0s[].fParamP.fP[4])^2++V0s[].fParamP.fX^2++abs(V0s[].fParamP.fP[4])*V0s[].fParamP.fX++abs(V0s[].fParamP.fP[4])*V0s[].fParamP.fX^2", cutV0, chi2,npoints,fitParam,covMatrix,fraction,0,entries);
   tree->SetAlias("v0sigmaY",v0sigmaY->Data());
-  array->AddLast(new TVectorD(fitParam));
+  array->AddAt(new TVectorD(fitParam),0);
   //
   //
   //
@@ -165,21 +203,21 @@ TObjArray * AliESDresolMakerFast::MakeParamRFast(TTree * tree,  TCut &cutV0, Flo
   //
   //
   //
-  TString * v0sigmaPhi= TStatToolkit::FitPlane(tree,"sqrt(sqrt((V0s[].fParamP.fC[5])))","abs(V0s[].fParamP.fP[4])++V0s[].fParamP.fX++abs(V0s[].fParamP.fP[4])^2++V0s[].fParamP.fX^2++abs(V0s[].fParamP.fP[4])*V0s[].fParamP.fX++abs(V0s[].fParamP.fP[4])*V0s[].fParamP.fX^2", cutV0, chi2,npoints,fitParam,covMatrix,fraction,0,entries);
+  TString * v0sigmaPhi= TStatToolkit::FitPlane(tree,"sqrt(sqrt((V0s[].fParamP.fC[5]))/(0.1+abs(V0s[].fParamP.fP[4])))","abs(V0s[].fParamP.fP[4])++V0s[].fParamP.fX++abs(V0s[].fParamP.fP[4])^2++abs(V0s[].fParamP.fP[4])*V0s[].fParamP.fX", cutV0, chi2,npoints,fitParam,covMatrix,fraction,0,entries);
   tree->SetAlias("v0sigmaPhi",v0sigmaPhi->Data());
-  array->AddLast(new TVectorD(fitParam));
+  array->AddAt(new TVectorD(fitParam),2);
   //
   //
   //
-  TString * v0sigmaTh= TStatToolkit::FitPlane(tree,"sqrt(sqrt((V0s[].fParamP.fC[9])))","abs(V0s[].fParamP.fP[4])++V0s[].fParamP.fX++abs(V0s[].fParamP.fP[4])^2++V0s[].fParamP.fX^2++abs(V0s[].fParamP.fP[4])*V0s[].fParamP.fX++abs(V0s[].fParamP.fP[4])*V0s[].fParamP.fX^2", cutV0, chi2,npoints,fitParam,covMatrix,fraction,0,entries);
+  TString * v0sigmaTh= TStatToolkit::FitPlane(tree,"sqrt(sqrt((V0s[].fParamP.fC[9]))/((0.1+abs(V0s[].fParamP.fP[4])*(1+abs(V0s[].fParamP.fP[3])^2))))","abs(V0s[].fParamP.fP[4])++V0s[].fParamP.fX++abs(V0s[].fParamP.fP[4])^2++abs(V0s[].fParamP.fP[4])*V0s[].fParamP.fX", cutV0, chi2,npoints,fitParam,covMatrix,fraction,0,entries);
   tree->SetAlias("v0sigmaTh",v0sigmaTh->Data());
-  array->AddLast(new TVectorD(fitParam));
+  array->AddAt(new TVectorD(fitParam),3);
   //
   //
   //
   TString * v0sigma1pt= TStatToolkit::FitPlane(tree,"sqrt(sqrt((V0s[].fParamP.fC[14])))","abs(V0s[].fParamP.fP[4])++V0s[].fParamP.fX++abs(V0s[].fParamP.fP[4])^2++V0s[].fParamP.fX^2++abs(V0s[].fParamP.fP[4])*V0s[].fParamP.fX++abs(V0s[].fParamP.fP[4])*V0s[].fParamP.fX^2", cutV0, chi2,npoints,fitParam,covMatrix,fraction,0,entries);
   tree->SetAlias("v0sigma1pt",v0sigma1pt->Data());
-  array->AddLast(new TVectorD(fitParam));
+  array->AddAt(new TVectorD(fitParam),4);
   return array;
 
 }