+/**************************************************************************
+ * Copyright(c) 1998-2003, ALICE Experiment at CERN, All rights reserved. *
+ * *
+ * Author: The ALICE Off-line Project. *
+ * Contributors are mentioned in the code where appropriate. *
+ * *
+ * Permission to use, copy, modify and distribute this software and its *
+ * documentation strictly for non-commercial purposes is hereby granted *
+ * without fee, provided that the above copyright notice appears in all *
+ * copies and that both the copyright notice and this permission notice *
+ * appear in the supporting documentation. The authors make no claims *
+ * about the suitability of this software for any purpose. It is *
+ * provided "as is" without express or implied warranty. *
+ **************************************************************************/
#include <AliITSVertexerPPZ.h>
-#include <Riostream.h>
#include <TArrayF.h>
-#include <TDirectory.h>
#include <TH1F.h>
-#include <TMath.h>
#include <TObjArray.h>
#include <TTree.h>
#include <TClonesArray.h>
-#include "AliRun.h"
-#include "AliITS.h"
+#include "AliITSDetTypeRec.h"
#include "AliITSgeom.h"
+#include "AliITSLoader.h"
#include "AliITSRecPoint.h"
/////////////////////////////////////////////////////////////////////////
// //
//______________________________________________________________________
-AliITSVertexerPPZ::AliITSVertexerPPZ():AliITSVertexer() {
+AliITSVertexerPPZ::AliITSVertexerPPZ():AliITSVertexer(),
+fFirstL1(0),
+fLastL1(0),
+fFirstL2(0),
+fLastL2(0),
+fDiffPhiMax(0),
+fX0(0),
+fY0(0),
+fZFound(0),
+fZsig(0),
+fWindow(0){
// Default Constructor
SetDiffPhiMax(0);
- fX0 = 0.;
- fY0 = 0.;
SetFirstLayerModules(0);
SetSecondLayerModules(0);
- fZFound = 0;
- fZsig = 0.;
- fITS = 0;
+ //fITS = 0;
+ SetWindow(0);
}
-AliITSVertexerPPZ::AliITSVertexerPPZ(TFile *infile, TFile *outfile, Float_t x0, Float_t y0):AliITSVertexer(infile,outfile) {
+AliITSVertexerPPZ::AliITSVertexerPPZ(TString fn, Float_t x0, Float_t y0):AliITSVertexer(fn),
+fFirstL1(0),
+fLastL1(0),
+fFirstL2(0),
+fLastL2(0),
+fDiffPhiMax(0),
+fX0(x0),
+fY0(y0),
+fZFound(0),
+fZsig(0),
+fWindow(0) {
// Standard constructor
- if(!fInFile)Fatal("AliITSVertexerPPZ","No inputfile has been defined!");
SetDiffPhiMax();
fX0 = x0;
fY0 = y0;
SetFirstLayerModules();
SetSecondLayerModules();
- fZFound = 0;
- fZsig = 0.;
- fITS = 0;
+ //fITS = 0;
+ SetWindow();
}
//______________________________________________________________________
AliITSVertexerPPZ::~AliITSVertexerPPZ() {
// Default Destructor
- fITS = 0;
+ //fITS = 0;
}
//________________________________________________________
void AliITSVertexerPPZ::EvalZ(TH1F *hist,Int_t sepa, Int_t ncoinc, TArrayF *zval) {
+
+ //Evaluation of Z
+ Float_t deltaVal = hist->GetBinWidth(1)*fWindow; // max window in Z for searching
fZFound=0;
fZsig=0;
- Int_t N=0;
- Int_t NbinNotZero=0;
+ Int_t nN=0;
+ Int_t nbinNotZero=0;
Float_t totst = 0.;
Float_t totst2 = 0.;
Float_t curz = 0.;
for(Int_t i=1;i<=sepa;i++){
Float_t cont=hist->GetBinContent(i);
- curz = hist->GetBinLowEdge(i)+0.5*hist->GetBinWidth(i);
+ if(cont!=0)curz = hist->GetBinLowEdge(i)+0.5*hist->GetBinWidth(i);
totst+=cont;
totst2+=cont*cont;
- N++;
- if(cont!=0)NbinNotZero++;
+ nN++;
+ if(cont!=0)nbinNotZero++;
+ }
+ if(nbinNotZero==0){fZFound=-100; fZsig=-100; return;}
+ if(nbinNotZero==1){
+ fZFound = curz;
+ fZsig=0;
+ fCurrentVertex = new AliESDVertex(fZFound,fZsig,nbinNotZero);
+ return;
+ }
+ Float_t errsq = totst2/(nN-1)-totst*totst/nN/(nN-1);
+ if(errsq>=0){
+ totst2=TMath::Sqrt(errsq);
}
- if(NbinNotZero==0){fZFound=-100; fZsig=-100; return;}
- if(NbinNotZero==1){fZFound = curz; fZsig=-100; return;}
- totst2=TMath::Sqrt(totst2/(N-1)-totst*totst/N/(N-1));
- totst /= N;
+ else {
+ Error("EvalZ","Negative variance: %d - %12.7f - %12.7f",nN,totst2,totst);
+ fZFound=-100;
+ fZsig=-100;
+ return;
+ }
+ totst /= nN;
+ Float_t cut = totst+totst2*2.;
+ if(fDebug>1)cout<<"totst, totst2, cut: "<<totst<<", "<<totst2<<", "<<cut<<endl;
Float_t val1=hist->GetBinLowEdge(sepa);
Float_t val2=hist->GetBinLowEdge(1);
+ Float_t valm = 0.;
+ Float_t zmax = 0.;
for(Int_t i=1;i<=sepa;i++){
Float_t cont=hist->GetBinContent(i);
- if(cont>(totst+totst2*2.)){
+ if(cont>valm){
+ valm = cont;
+ zmax = hist->GetBinLowEdge(i)+0.5*hist->GetBinWidth(1);
+ }
+ if(cont>cut){
curz=hist->GetBinLowEdge(i);
if(curz<val1)val1=curz;
if(curz>val2)val2=curz;
}
}
val2+=hist->GetBinWidth(1);
- if(fDebug>0)cout<<"Values for Z finding: "<<val1<<" "<<val2<<endl;
+ if((val2-val1)>deltaVal){
+ val1 = zmax-deltaVal/2.;
+ val2 = zmax+deltaVal/2.;
+ if(fDebug>0)cout<<"val1 and val2 recomputed\n";
+ }
+ if(fDebug>0)cout<<"Values for Z finding: "<<val1<<" "<<val2<<" "<<val2-val1<<endl;
fZFound=0;
fZsig=0;
- N=0;
+ nN=0;
for(Int_t i=0; i<ncoinc; i++){
Float_t z=zval->At(i);
if(z<val1)continue;
if(z>val2)continue;
+
fZFound+=z;
fZsig+=z*z;
- N++;
+ /* weights defined by the curvature
+ Float_t wei = 1./curv->At(i);
+ fZFound+=z*wei;
+ fZsig+=wei;
+ */
+ nN++;
+ }
+ if(nN<1){fZFound=-110; fZsig=-110; return;}
+ if(nN==1){
+ fZsig = 0;
+ fCurrentVertex = new AliESDVertex(fZFound,fZsig,nN);
+ return;
+ }
+ errsq = (fZsig/(nN-1)-fZFound*fZFound/nN/(nN-1))/nN;
+ if(errsq>=0.){
+ fZsig=TMath::Sqrt(errsq);
}
- if(N<=1){fZFound=-110; fZsig=-110; return;}
- fZsig=TMath::Sqrt((fZsig/(N-1)-fZFound*fZFound/N/(N-1))/N);
- fZFound=fZFound/N;
- fCurrentVertex = new AliITSVertex(fZFound,fZsig,N);
+ else {
+ Error("evalZ","Negative variance: %d - %12.7f %12.7f",nN,fZsig,fZFound);
+ fZsig=0;
+ }
+ fZFound=fZFound/nN;
+ /* weights defined by the curvature
+ fZsig=1./fZsig;
+ fZFound*=fZsig;
+ fZsig = TMath::Sqrt(fZsig);
+ */
+ fCurrentVertex = new AliESDVertex(fZFound,fZsig,nN);
}
//______________________________________________________________________
-AliITSVertex* AliITSVertexerPPZ::FindVertexForCurrentEvent(Int_t evnumber){
- // Defines the AliITSVertex for the current event
+AliESDVertex* AliITSVertexerPPZ::FindVertexForCurrentEvent(Int_t evnumber){
+ // Defines the AliESDVertex for the current event
fCurrentVertex = 0;
fZFound = -999;
fZsig = -999;
- if(!gAlice)gAlice = (AliRun*)fInFile->Get("gAlice");
- if(!gAlice){
- Error("FindVertexForCurrentEvent","The AliRun object is not available - nothing done");
- return fCurrentVertex;
- }
- if(!fITS) {
- fITS = (AliITS*)gAlice->GetModule("ITS");
- if(!fITS) {
- Error("FindVertexForCurrentEvent","AliITS object was not found");
- return fCurrentVertex;
- }
- }
- AliITSgeom *geom = fITS->GetITSgeom();
+ AliRunLoader *rl =AliRunLoader::GetRunLoader();
+ AliITSLoader* iTSloader = (AliITSLoader*)rl->GetLoader("ITSLoader");
+ TDirectory * olddir = gDirectory;
+ rl->CdGAFile();
+ AliITSgeom* geom = (AliITSgeom*)gDirectory->Get("AliITSgeom");
+ olddir->cd();
+
+ AliITSDetTypeRec detTypeRec;
+
if(!geom) {
Error("FindVertexForCurrentEvent","ITS geometry is not defined");
return fCurrentVertex;
}
- TTree *TR=0;
- TClonesArray *ITSrec = 0;
+ TTree *tR=0;
+ TClonesArray *itsRec = 0;
Float_t lc[3]; for(Int_t ii=0; ii<3; ii++) lc[ii]=0.;
Float_t gc[3]; for(Int_t ii=0; ii<3; ii++) gc[ii]=0.;
Float_t lc2[3]; for(Int_t ii=0; ii<3; ii++) lc2[ii]=0.;
Float_t gc2[3]; for(Int_t ii=0; ii<3; ii++) gc2[ii]=0.;
- TR = gAlice->TreeR();
- if(!TR){
+ tR = iTSloader->TreeR();
+ if(!tR){
Error("FindVertexForCurrentEvent","TreeR not found");
return fCurrentVertex;
}
- ITSrec = fITS->RecPoints(); // uses slow points or fast points if slow are
+ detTypeRec.SetTreeAddressR(tR);
+ itsRec = detTypeRec.RecPoints();
// missing
- TBranch *branch = TR->GetBranch("ITSRecPoints");
+ // TClonesArray dummy("AliITSRecPoint",10000), *clusters=&dummy;
+ TBranch *branch;
+ branch = tR->GetBranch("ITSRecPoints");
if(!branch){
- branch = TR->GetBranch("ITSRecPointsF");
- // Warning("FindVertexForCurrentEvent","Using Fast points");
+ branch = tR->GetBranch("ITSRecPointsF");
}
+ //}
if(!branch){
Error("FindVertexForCurrentEvent","branch for ITS rec points not found");
return fCurrentVertex;
Int_t firipixe=0;
for(Int_t module= fFirstL1; module<=fLastL1;module++){
branch->GetEvent(module);
- Int_t nrecp1 = ITSrec->GetEntries();
+ Int_t nrecp1 = itsRec->GetEntries();
for(Int_t i=0; i<nrecp1;i++){
- AliITSRecPoint *current = (AliITSRecPoint*)ITSrec->At(i);
- lc[0]=current->GetX();
- lc[2]=current->GetZ();
+ AliITSRecPoint *current = (AliITSRecPoint*)itsRec->At(i);
+ lc[0]=current->GetDetLocalX();
+ lc[2]=current->GetDetLocalZ();
geom->LtoG(module,lc,gc);
zave+=gc[2];
zave2+=gc[2]*gc[2];
firipixe++;
}
- fITS->ResetRecPoints();
+ detTypeRec.ResetRecPoints();
}
if(firipixe>1){
rmszav=TMath::Sqrt(zave2/(firipixe-1)-zave*zave/firipixe/(firipixe-1));
zave=zave/firipixe;
+ if(fDebug>1)cout<<"++++++++++++++++++++++++++++++++++++++++++++++++++++++\n Number of firing pixels: "<<firipixe<<endl;
}
else {
fZFound = -200;
Int_t sepa=(Int_t)((zlim2-zlim1)*10.+1.);
zlim2=zlim1 + sepa/10.;
TH1F *zvdis = new TH1F("z_ev","zv distr",sepa,zlim1,zlim2);
- if(fDebug>0)cout<<"Z limits: "<<zlim1<<" "<<zlim2<<"; Bins= "<<sepa<<endl;
+ if(fDebug>0){
+ cout<<"Z limits: "<<zlim1<<" "<<zlim2<<"; Bins= "<<sepa<<endl;
+ cout<<"Bin width: "<<zvdis->GetBinWidth(1)*10000.<<" micron\n";
+ }
Int_t sizarr=100;
TArrayF *zval = new TArrayF(sizarr);
+ // TArrayF *curv = new TArrayF(sizarr);
Int_t ncoinc=0;
for(Int_t module= fFirstL1; module<=fLastL1;module++){
if(fDebug>0)cout<<"processing module "<<module<<" \r";
branch->GetEvent(module);
- Int_t nrecp1 = ITSrec->GetEntries();
+ Int_t nrecp1 = itsRec->GetEntries();
TObjArray *poiL1 = new TObjArray(nrecp1);
- for(Int_t i=0; i<nrecp1;i++)poiL1->AddAt(ITSrec->At(i),i);
- fITS->ResetRecPoints();
+ for(Int_t i=0; i<nrecp1;i++)poiL1->AddAt(itsRec->At(i),i);
+ detTypeRec.ResetRecPoints();
for(Int_t i=0; i<nrecp1;i++){
AliITSRecPoint *current = (AliITSRecPoint*)poiL1->At(i);
- lc[0]=current->GetX();
- lc[2]=current->GetZ();
+ lc[0]=current->GetDetLocalX();
+ lc[2]=current->GetDetLocalZ();
geom->LtoG(module,lc,gc);
gc[0]-=fX0;
gc[1]-=fY0;
if(fDebug>1)cout<<"module "<<module<<" "<<gc[0]<<" "<<gc[1]<<" "<<gc[2]<<" "<<phi1<<" \n";
for(Int_t modul2=fFirstL2; modul2<=fLastL2; modul2++){
branch->GetEvent(modul2);
- Int_t nrecp2 = ITSrec->GetEntries();
+ Int_t nrecp2 = itsRec->GetEntries();
for(Int_t j=0; j<nrecp2;j++){
- AliITSRecPoint *recp = (AliITSRecPoint*)ITSrec->At(j);
- lc2[0]=recp->GetX();
- lc2[2]=recp->GetZ();
+ AliITSRecPoint *recp = (AliITSRecPoint*)itsRec->At(j);
+ lc2[0]=recp->GetDetLocalX();
+ lc2[2]=recp->GetDetLocalZ();
geom->LtoG(modul2,lc2,gc2);
gc2[0]-=fX0;
gc2[1]-=fY0;
if(diff<fDiffPhiMax ){
zvdis->Fill(zr0);
zval->AddAt(zr0,ncoinc);
+ /* uncomment these lines to use curvature as a weight
+ Float_t cu = Curv(0.,0.,gc[0],gc[1],gc2[0],gc[1]);
+ curv->AddAt(cu,ncoinc);
+ */
ncoinc++;
if(ncoinc==(sizarr-1)){
sizarr+=100;
zval->Set(sizarr);
+ //uncomment next line to use curvature as weight
+ // curv->Set(sizarr);
}
}
}
}
- fITS->ResetRecPoints();
+ detTypeRec.ResetRecPoints();
}
}
delete poiL1;
if(fDebug>0){
cout<<endl<<"Number of coincidences = "<<ncoinc<<endl;
}
+ // EvalZ(zvdis,sepa,ncoinc,zval,curv);
EvalZ(zvdis,sepa,ncoinc,zval);
delete zvdis;
delete zval;
+ // delete curv;
if(fCurrentVertex){
char name[30];
sprintf(name,"Vertex_%d",evnumber);
- fCurrentVertex->SetName(name);
+ // fCurrentVertex->SetName(name);
fCurrentVertex->SetTitle("vertexer: PPZ");
}
+
return fCurrentVertex;
}
//______________________________________________________________________
void AliITSVertexerPPZ::FindVertices(){
// computes the vertices of the events in the range FirstEvent - LastEvent
- if(!fOutFile){
- Error("FindVertices","The output file is not available - nothing done");
- return;
- }
- if(!gAlice)gAlice = (AliRun*)fInFile->Get("gAlice");
- if(!gAlice){
- Error("FindVertices","The AliRun object is not available - nothing done");
- return;
- }
- TDirectory *curdir = gDirectory;
- fInFile->cd();
- for(Int_t i=fFirstEvent;i<fLastEvent;i++){
- gAlice->GetEvent(i);
+ AliRunLoader *rl = AliRunLoader::GetRunLoader();
+ AliITSLoader* iTSloader = (AliITSLoader*) rl->GetLoader("ITSLoader");
+ iTSloader->ReloadRecPoints();
+ for(Int_t i=fFirstEvent;i<=fLastEvent;i++){
+ cout<<"Processing event "<<i<<endl;
+ rl->GetEvent(i);
FindVertexForCurrentEvent(i);
if(fCurrentVertex){
WriteCurrentVertex();
}
}
}
- curdir->cd();
}
//________________________________________________________
cout <<" Second layer first and last modules: "<<fFirstL2<<", ";
cout <<fLastL2<<endl;
cout <<" Max Phi difference: "<<fDiffPhiMax<<endl;
+ cout <<" Window for Z search: "<<fWindow<<endl;
cout <<" Current Z "<<fZFound<<"; Z sig "<<fZsig<<endl;
cout <<" Debug flag: "<<fDebug<<endl;
- if(fInFile)cout <<" The input file is open\n";
- if(!fInFile)cout <<"The input file is not open\n";
- if(fOutFile)cout <<"The output file is open\n";
- if(fOutFile)cout <<"The output file not is open\n";
cout<<"First event to be processed "<<fFirstEvent;
cout<<"\n Last event to be processed "<<fLastEvent<<endl;
}
+
+//________________________________________________________
+Float_t AliITSVertexerPPZ::Curv(Double_t x1,Double_t y1,
+ Double_t x2,Double_t y2,
+ Double_t x3,Double_t y3)
+{
+ //-----------------------------------------------------------------
+ // Initial approximation of the track curvature (Y. Belikov) squared
+ //-----------------------------------------------------------------
+ Double_t d=(x2-x1)*(y3-y2)-(x3-x2)*(y2-y1);
+ Double_t a=0.5*((y3-y2)*(y2*y2-y1*y1+x2*x2-x1*x1)-
+ (y2-y1)*(y3*y3-y2*y2+x3*x3-x2*x2));
+ Double_t b=0.5*((x2-x1)*(y3*y3-y2*y2+x3*x3-x2*x2)-
+ (x3-x2)*(y2*y2-y1*y1+x2*x2-x1*x1));
+
+ Double_t xr=TMath::Abs(d/(d*x1-a)), yr=d/(d*y1-b);
+
+ Float_t val = (Float_t)(( -xr*yr/TMath::Sqrt(xr*xr+yr*yr))
+ *( -xr*yr/TMath::Sqrt(xr*xr+yr*yr)));
+ return val;
+}
git://git.uio.no / u / mrichter / AliRoot.git / blobdiff