// Author: Anders Vestbo <mailto:vestbo@fi.uib.no>
//*-- Copyright © ALICE HLT Group
+#include <strings.h>
#include "AliL3StandardIncludes.h"
#ifndef no_root
#include "AliL3Histogram.h"
#include "AliL3TrackArray.h"
#include "AliL3HoughTrack.h"
+#include "AliL3Transform.h"
+#include "AliL3HoughTransformerRow.h"
-#if GCCVERSION == 3
+#if __GNUC__ == 3
using namespace std;
#endif
+/** \class AliL3HoughMaxFinder
+<pre>
//_____________________________________________________________
// AliL3HoughMaxFinder
//
// Maximum finder
+//
+</pre>
+*/
ClassImp(AliL3HoughMaxFinder)
{
//Default constructor
fThreshold = 0;
+ fCurrentEtaSlice = -1;
fHistoType=0;
fXPeaks=0;
fYPeaks=0;
+ fWeight=0;
fNPeaks=0;
+ fN1PeaksPrevEtaSlice=0;
+ fN2PeaksPrevEtaSlice=0;
+ fSTARTXPeaks=0;
+ fSTARTYPeaks=0;
+ fENDXPeaks=0;
+ fENDYPeaks=0;
+ fSTARTETAPeaks=0;
+ fENDETAPeaks=0;
fNMax=0;
+ fGradX=1;
+ fGradY=1;
#ifndef no_root
fNtuppel = 0;
#endif
}
-
AliL3HoughMaxFinder::AliL3HoughMaxFinder(Char_t *histotype,Int_t nmax,AliL3Histogram *hist)
{
//Constructor
//fTracks = new AliL3TrackArray("AliL3HoughTrack");
if(strcmp(histotype,"KappaPhi")==0) fHistoType='c';
if(strcmp(histotype,"DPsi")==0) fHistoType='l';
+
+ fCurrentEtaSlice = -1;
if(hist)
fCurrentHisto = hist;
+ fGradX=1;
+ fGradY=1;
fNMax=nmax;
fXPeaks = new Float_t[fNMax];
fYPeaks = new Float_t[fNMax];
fWeight = new Int_t[fNMax];
+ fSTARTXPeaks = new Int_t[fNMax];
+ fSTARTYPeaks = new Int_t[fNMax];
+ fENDXPeaks = new Int_t[fNMax];
+ fENDYPeaks = new Int_t[fNMax];
+ fSTARTETAPeaks = new Int_t[fNMax];
+ fENDETAPeaks = new Int_t[fNMax];
#ifndef no_root
fNtuppel = 0;
#endif
fThreshold=0;
}
-
AliL3HoughMaxFinder::~AliL3HoughMaxFinder()
{
//Destructor
delete [] fYPeaks;
if(fWeight)
delete [] fWeight;
+ if(fSTARTXPeaks)
+ delete [] fSTARTXPeaks;
+ if(fSTARTYPeaks)
+ delete [] fSTARTYPeaks;
+ if(fENDXPeaks)
+ delete [] fENDXPeaks;
+ if(fENDYPeaks)
+ delete [] fENDYPeaks;
+ if(fSTARTETAPeaks)
+ delete [] fSTARTETAPeaks;
+ if(fENDETAPeaks)
+ delete [] fENDETAPeaks;
#ifndef no_root
if(fNtuppel)
delete fNtuppel;
void AliL3HoughMaxFinder::Reset()
{
+ // Method to reinit the Peak Finder
for(Int_t i=0; i<fNMax; i++)
{
fXPeaks[i]=0;
fYPeaks[i]=0;
fWeight[i]=0;
+ fSTARTXPeaks[i]=0;
+ fSTARTYPeaks[i]=0;
+ fENDXPeaks[i]=0;
+ fENDYPeaks[i]=0;
+ fSTARTETAPeaks[i]=0;
+ fENDETAPeaks[i]=0;
}
fNPeaks=0;
+ fN1PeaksPrevEtaSlice=0;
+ fN2PeaksPrevEtaSlice=0;
}
void AliL3HoughMaxFinder::CreateNtuppel()
{
+ // Fill a NTuple with the peak parameters
#ifndef no_root
//content#; neighbouring bins of the peak.
fNtuppel = new TNtuple("ntuppel","Peak charateristics","kappa:phi0:weigth:content3:content5:content1:content7");
void AliL3HoughMaxFinder::WriteNtuppel(Char_t *filename)
{
+ // Write the NTuple with the peak parameters
#ifndef no_root
TFile *file = TFile::Open(filename,"RECREATE");
if(!file)
void AliL3HoughMaxFinder::FindAbsMaxima()
{
-
+ // Simple Peak Finder in the Hough space
if(!fCurrentHisto)
{
cerr<<"AliL3HoughMaxFinder::FindAbsMaxima : No histogram"<<endl;
}
AliL3Histogram *hist = fCurrentHisto;
+ if(hist->GetNEntries() == 0)
+ return;
+
Int_t xmin = hist->GetFirstXbin();
Int_t xmax = hist->GetLastXbin();
Int_t ymin = hist->GetFirstYbin();
Int_t ymax = hist->GetLastYbin();
Int_t bin;
- Double_t value,max_value=0;
+ Double_t value,maxvalue=0;
- Int_t max_xbin=0,max_ybin=0;
+ Int_t maxxbin=0,maxybin=0;
for(Int_t xbin=xmin; xbin<=xmax; xbin++)
{
for(Int_t ybin=ymin; ybin<=ymax; ybin++)
{
bin = hist->GetBin(xbin,ybin);
value = hist->GetBinContent(bin);
- if(value>max_value)
+ if(value>maxvalue)
{
- max_value = value;
- max_xbin = xbin;
- max_ybin = ybin;
+ maxvalue = value;
+ maxxbin = xbin;
+ maxybin = ybin;
}
}
}
+ if(maxvalue == 0)
+ return;
+
if(fNPeaks > fNMax)
{
cerr<<"AliL3HoughMaxFinder::FindAbsMaxima : Array out of range : "<<fNPeaks<<endl;
return;
}
-
- Double_t max_x = hist->GetBinCenterX(max_xbin);
- Double_t max_y = hist->GetBinCenterY(max_ybin);
- fXPeaks[fNPeaks] = max_x;
- fYPeaks[fNPeaks] = max_y;
- fWeight[fNPeaks] = (Int_t)max_value;
+ Double_t maxx = hist->GetBinCenterX(maxxbin);
+ Double_t maxy = hist->GetBinCenterY(maxybin);
+ fXPeaks[fNPeaks] = maxx;
+ fYPeaks[fNPeaks] = maxy;
+ fWeight[fNPeaks] = (Int_t)maxvalue;
+
fNPeaks++;
#ifndef no_root
if(fNtuppel)
{
- Int_t bin3 = hist->GetBin(max_xbin-1,max_ybin);
- Int_t bin5 = hist->GetBin(max_xbin+1,max_ybin);
- Int_t bin1 = hist->GetBin(max_xbin,max_ybin-1);
- Int_t bin7 = hist->GetBin(max_xbin,max_ybin+1);
+ Int_t bin3 = hist->GetBin(maxxbin-1,maxybin);
+ Int_t bin5 = hist->GetBin(maxxbin+1,maxybin);
+ Int_t bin1 = hist->GetBin(maxxbin,maxybin-1);
+ Int_t bin7 = hist->GetBin(maxxbin,maxybin+1);
- fNtuppel->Fill(max_x,max_y,max_value,hist->GetBinContent(bin3),hist->GetBinContent(bin5),hist->GetBinContent(bin1),hist->GetBinContent(bin7));
+ fNtuppel->Fill(maxx,maxy,maxvalue,hist->GetBinContent(bin3),hist->GetBinContent(bin5),hist->GetBinContent(bin1),hist->GetBinContent(bin7));
}
#endif
}
void AliL3HoughMaxFinder::FindBigMaxima()
{
-
+ // Another Peak finder
AliL3Histogram *hist = fCurrentHisto;
+
+ if(hist->GetNEntries() == 0)
+ return;
+
Int_t xmin = hist->GetFirstXbin();
Int_t xmax = hist->GetLastXbin();
Int_t ymin = hist->GetFirstYbin();
Int_t ymax = hist->GetLastYbin();
- Int_t bin[25],bin_index;
+ Int_t bin[25],binindex;
Double_t value[25];
for(Int_t xbin=xmin+2; xbin<xmax-3; xbin++)
{
for(Int_t ybin=ymin+2; ybin<ymax-3; ybin++)
{
- bin_index=0;
+ binindex=0;
for(Int_t xb=xbin-2; xb<=xbin+2; xb++)
{
for(Int_t yb=ybin-2; yb<=ybin+2; yb++)
{
- bin[bin_index]=hist->GetBin(xb,yb);
- value[bin_index]=hist->GetBinContent(bin[bin_index]);
- bin_index++;
+ bin[binindex]=hist->GetBin(xb,yb);
+ value[binindex]=hist->GetBinContent(bin[binindex]);
+ binindex++;
}
-
}
if(value[12]==0) continue;
Int_t b=0;
while(1)
{
- if(value[b] > value[12] || b==bin_index) break;
+ if(value[b] > value[12] || b==binindex) break;
b++;
//printf("b %d\n",b);
}
- if(b == bin_index)
+ if(b == binindex)
{
//Found maxima
if(fNPeaks > fNMax)
return;
}
- Double_t max_x = hist->GetBinCenterX(xbin);
- Double_t max_y = hist->GetBinCenterY(ybin);
- fXPeaks[fNPeaks] = max_x;
- fYPeaks[fNPeaks] = max_y;
+ Double_t maxx = hist->GetBinCenterX(xbin);
+ Double_t maxy = hist->GetBinCenterY(ybin);
+ fXPeaks[fNPeaks] = maxx;
+ fYPeaks[fNPeaks] = maxy;
fNPeaks++;
}
}
}
}
-void AliL3HoughMaxFinder::FindMaxima(Double_t grad_x,Double_t grad_y)
+void AliL3HoughMaxFinder::FindMaxima(Int_t threshold)
{
//Locate all the maxima in input histogram.
//Maxima is defined as bins with more entries than the
//immediately neighbouring bins.
+ if(fCurrentHisto->GetNEntries() == 0)
+ return;
+
Int_t xmin = fCurrentHisto->GetFirstXbin();
Int_t xmax = fCurrentHisto->GetLastXbin();
Int_t ymin = fCurrentHisto->GetFirstYbin();
&& value[4]>value[7] && value[4]>value[8])
{
//Found a local maxima
- Float_t max_x = fCurrentHisto->GetBinCenterX(xbin);
- Float_t max_y = fCurrentHisto->GetBinCenterY(ybin);
+ Float_t maxx = fCurrentHisto->GetBinCenterX(xbin);
+ Float_t maxy = fCurrentHisto->GetBinCenterY(ybin);
- if((Int_t)value[4] <= fThreshold) continue;//central bin below threshold
+ if((Int_t)value[4] <= threshold) continue;//central bin below threshold
if(fNPeaks >= fNMax)
{
cout<<"AliL3HoughMaxFinder::FindMaxima : Array out of range "<<fNPeaks<<endl;
}
//Check the gradient:
- //if(value[1]/value[4] > grad_y || value[7]/value[4] > grad_y)
- //continue;
-
- fXPeaks[fNPeaks] = max_x;
- fYPeaks[fNPeaks] = max_y;
+ if(value[3]/value[4] > fGradX && value[5]/value[4] > fGradX)
+ continue;
+
+ if(value[1]/value[4] > fGradY && value[7]/value[4] > fGradY)
+ continue;
+
+ fXPeaks[fNPeaks] = maxx;
+ fYPeaks[fNPeaks] = maxy;
fWeight[fNPeaks] = (Int_t)value[4];
fNPeaks++;
Bool_t bigger = kFALSE;
for(Int_t p=0; p<fNPeaks; p++)
{
- if(fabs(max_x - fXPeaks[p]) < max_kappa && fabs(max_y - fYPeaks[p]) < max_phi0)
+ if(fabs(maxx - fXPeaks[p]) < max_kappa && fabs(maxy - fYPeaks[p]) < max_phi0)
{
bigger = kTRUE;
if(value[4] > fWeight[p]) //this peak is bigger.
{
- fXPeaks[p] = max_x;
- fYPeaks[p] = max_y;
+ fXPeaks[p] = maxx;
+ fYPeaks[p] = maxy;
fWeight[p] = (Int_t)value[4];
}
else
}
if(!bigger) //there were no overlapping peaks.
{
- fXPeaks[fNPeaks] = max_x;
- fYPeaks[fNPeaks] = max_y;
+ fXPeaks[fNPeaks] = maxx;
+ fYPeaks[fNPeaks] = maxy;
fWeight[fNPeaks] = (Int_t)value[4];
fNPeaks++;
}
}
-AliL3TrackArray *AliL3HoughMaxFinder::LookForPeaks(AliL3Histogram *hist,Int_t nbins)
+struct AliL3Window
{
+ Int_t fStart; // Start
+ Int_t fSum; // Sum
+};
+
+void AliL3HoughMaxFinder::FindAdaptedPeaks(Int_t kappawindow,Float_t cutratio)
+{
+ //Peak finder which looks for peaks with a certain shape.
+ //The first step involves a pre-peak finder, which looks for peaks
+ //in windows (size controlled by kappawindow) summing over each psi-bin.
+ //These pre-preaks are then matched between neighbouring kappa-bins to
+ //look for real 2D peaks exhbiting the typical cross-shape in the Hough circle transform.
+ //The maximum bin within this region is marked as the peak itself, and
+ //a few checks is performed to avoid the clear fake peaks (asymmetry check etc.)
+
+
+ AliL3Histogram *hist = fCurrentHisto;
- AliL3TrackArray *tracks = new AliL3TrackArray("AliL3HoughTrack");
- AliL3HoughTrack *track;
+ if(!hist)
+ {
+ cerr<<"AliL3HoughMaxFinder : No histogram!"<<endl;
+ return;
+ }
+
+ if(hist->GetNEntries() == 0)
+ return;
+
Int_t xmin = hist->GetFirstXbin();
Int_t xmax = hist->GetLastXbin();
Int_t ymin = hist->GetFirstYbin();
Int_t ymax = hist->GetLastYbin();
+
+
+ //Start by looking for pre-peaks:
+
+ AliL3Window **localmaxima = new AliL3Window*[hist->GetNbinsY()];
- Int_t weight_loc;
- for(Int_t xbin=xmin+nbins; xbin <= xmax - nbins; xbin++)
+ Short_t *nmaxs = new Short_t[hist->GetNbinsY()];
+ Int_t n,lastsum,sum;
+ Bool_t sumwasrising;
+ for(Int_t ybin=ymin; ybin<=ymax; ybin++)
{
- for(Int_t ybin=ymin+nbins; ybin <= ymax - nbins; ybin++)
+ localmaxima[ybin-ymin] = new AliL3Window[hist->GetNbinsX()];
+ nmaxs[ybin-ymin] = 0;
+ sumwasrising=0;
+ lastsum=0;
+ n=0;
+ for(Int_t xbin=xmin; xbin<=xmax-kappawindow; xbin++)
{
- weight_loc=0;
- for(Int_t xbin_loc = xbin-nbins; xbin_loc <= xbin+nbins; xbin_loc++)
- {
- for(Int_t ybin_loc = ybin-nbins; ybin_loc <= ybin+nbins; ybin_loc++)
- {
- Int_t bin_loc = hist->GetBin(xbin_loc,ybin_loc);
- weight_loc += (Int_t)hist->GetBinContent(bin_loc);
- }
- }
+ sum=0;
+ for(Int_t lbin=xbin; lbin<xbin+kappawindow; lbin++)
+ sum += hist->GetBinContent(hist->GetBin(lbin,ybin));
- if(weight_loc > 0)
+ if(sum < lastsum)
{
- track = (AliL3HoughTrack*)tracks->NextTrack();
- track->SetTrackParameters(hist->GetBinCenterX(xbin),hist->GetBinCenterY(ybin),weight_loc);
+ if(sum > fThreshold)
+ if(sumwasrising)//Previous sum was a local maxima
+ {
+ localmaxima[ybin-ymin][nmaxs[ybin-ymin]].fStart = xbin-1;
+ localmaxima[ybin-ymin][nmaxs[ybin-ymin]].fSum = lastsum;
+ nmaxs[ybin-ymin]++;
+ }
+
+ sumwasrising=0;
}
-
+ else if(sum > 0)
+ sumwasrising=1;
+ lastsum=sum;
}
}
- tracks->QSort();
- AliL3HoughTrack *track1,*track2;
-
- for(Int_t i=1; i<tracks->GetNTracks(); i++)
+ Int_t match=0;
+ Int_t *starts = new Int_t[hist->GetNbinsY()+1];
+ Int_t *maxs = new Int_t[hist->GetNbinsY()+1];
+
+ for(Int_t ybin=ymax; ybin >= ymin+1; ybin--)
{
- track1 = (AliL3HoughTrack*)tracks->GetCheckedTrack(i);
- if(!track1) continue;
- Int_t xbin1 = hist->FindXbin(track1->GetKappa());
- Int_t ybin1 = hist->FindXbin(track1->GetPhi0());
- for(Int_t j=0; j < i; j++)
+ for(Int_t i=0; i<nmaxs[ybin-ymin]; i++)
{
- track2 = (AliL3HoughTrack*)tracks->GetCheckedTrack(j);
- if(!track2) continue;
- Int_t xbin2 = hist->FindXbin(track2->GetKappa());
- Int_t ybin2 = hist->FindYbin(track2->GetPhi0());
- if(abs(xbin1-xbin2) < 10 && abs(ybin1-ybin2) < 10)
+ Int_t lw = localmaxima[ybin-ymin][i].fSum;
+
+ if(lw<0)
+ continue; //already used
+
+ Int_t maxvalue=0,maxybin=0,maxxbin=0,maxwindow=0;
+ for(Int_t k=localmaxima[ybin-ymin][i].fStart; k<localmaxima[ybin-ymin][i].fStart + kappawindow; k++)
+ if(hist->GetBinContent(hist->GetBin(k,ybin)) > maxvalue)
+ {
+ maxvalue = hist->GetBinContent(hist->GetBin(k,ybin));
+ maxybin = ybin;
+ maxxbin = k;
+ }
+
+ //start expanding in the psi-direction:
+
+ Int_t lb = localmaxima[ybin-ymin][i].fStart;
+ //Int_t ystart=ybin;
+ starts[ybin] = localmaxima[ybin-ymin][i].fStart;
+ maxs[ybin] = maxxbin;
+ Int_t yl=ybin-1,nybins=1;
+
+ //cout<<"Starting search at ybin "<<ybin<<" start "<<lb<<" with sum "<<localmaxima[ybin-ymin][i].sum<<endl;
+ while(yl >= ymin)
{
- tracks->Remove(i);
- break;
+ Bool_t found=0;
+ for(Int_t j=0; j<nmaxs[yl-ymin]; j++)
+ {
+ if( localmaxima[yl-ymin][j].fStart - lb < 0) continue;
+ if( localmaxima[yl-ymin][j].fStart < lb + kappawindow + match &&
+ localmaxima[yl-ymin][j].fStart >= lb && localmaxima[yl-ymin][j].fSum > 0)
+ {
+
+ //cout<<"match at ybin "<<yl<<" yvalue "<<hist->GetBinCenterY(yl)<<" start "<<localmaxima[yl-ymin][j].start<<" sum "<<localmaxima[yl-ymin][j].sum<<endl;
+
+ Int_t lmaxvalue=0,lmaxxbin=0;
+ for(Int_t k=localmaxima[yl-ymin][j].fStart; k<localmaxima[yl-ymin][j].fStart + kappawindow; k++)
+ {
+ if(hist->GetBinContent(hist->GetBin(k,yl)) > maxvalue)
+ {
+ maxvalue = hist->GetBinContent(hist->GetBin(k,yl));
+ maxxbin = k;
+ maxybin = yl;
+ maxwindow = j;
+ }
+ if(hist->GetBinContent(hist->GetBin(k,yl)) > lmaxvalue)//local maxima value
+ {
+ lmaxvalue=hist->GetBinContent(hist->GetBin(k,yl));
+ lmaxxbin=k;
+ }
+ }
+ nybins++;
+ starts[yl] = localmaxima[yl-ymin][j].fStart;
+ maxs[yl] = lmaxxbin;
+ localmaxima[yl-ymin][j].fSum=-1; //Mark as used
+ found=1;
+ lb = localmaxima[yl-ymin][j].fStart;
+ break;//Since we found a match in this bin, we dont have to search it anymore, goto next bin.
+ }
+ }
+ if(!found || yl == ymin)//no more local maximas to be matched, so write the final peak and break the expansion:
+ {
+ if(nybins > 4)
+ {
+ //cout<<"ystart "<<ystart<<" and nybins "<<nybins<<endl;
+
+ Bool_t truepeak=kTRUE;
+
+ //cout<<"Maxima found at xbin "<<maxxbin<<" ybin "<<maxybin<<" value "<<maxvalue<<endl;
+ //cout<<"Starting to sum at xbin "<<starts[maxybin-ymin]<<endl;
+
+
+ //Look in a window on both sides to probe the asymmetry
+ Float_t right=0,left=0;
+ for(Int_t w=maxxbin+1; w<=maxxbin+3; w++)
+ {
+ for(Int_t r=maxybin+1; r<=maxybin+3; r++)
+ {
+ right += (Float_t)hist->GetBinContent(hist->GetBin(w,r));
+ }
+ }
+
+ for(Int_t w=maxxbin-1; w>=maxxbin-3; w--)
+ {
+ for(Int_t r=maxybin+1; r<=maxybin+3; r++)
+ {
+ left += (Float_t)hist->GetBinContent(hist->GetBin(w,r));
+ }
+ }
+
+ //cout<<"ratio "<<right/left<<endl;
+
+ Float_t upperratio=1,lowerratio=1;
+ if(left)
+ upperratio = right/left;
+
+ right=left=0;
+ for(Int_t w=maxxbin+1; w<=maxxbin+3; w++)
+ {
+ for(Int_t r=maxybin-1; r>=maxybin-3; r--)
+ {
+ right += (Float_t)hist->GetBinContent(hist->GetBin(w,r));
+ }
+ }
+
+ for(Int_t w=maxxbin-1; w>=maxxbin-3; w--)
+ {
+ for(Int_t r=maxybin-1; r>=maxybin-3; r--)
+ {
+ left += (Float_t)hist->GetBinContent(hist->GetBin(w,r));
+ }
+ }
+
+ //cout<<"ratio "<<left/right<<endl;
+
+ if(right)
+ lowerratio = left/right;
+
+ if(upperratio > cutratio || lowerratio > cutratio)
+ truepeak=kFALSE;
+
+ if(truepeak)
+ {
+
+ fXPeaks[fNPeaks] = hist->GetBinCenterX(maxxbin);
+ fYPeaks[fNPeaks] = hist->GetBinCenterY(maxybin);
+ fWeight[fNPeaks] = maxvalue;
+ fNPeaks++;
+
+ /*
+ //Calculate the peak using weigthed means:
+ Float_t sum=0;
+ fYPeaks[fNPeaks]=0;
+ for(Int_t k=maxybin-1; k<=maxybin+1; k++)
+ {
+ Float_t lsum = 0;
+ for(Int_t l=starts[k]; l<starts[k]+kappawindow; l++)
+ {
+ lsum += (Float_t)hist->GetBinContent(hist->GetBin(l,k));
+ sum += (Float_t)hist->GetBinContent(hist->GetBin(l,k));
+ }
+ fYPeaks[fNPeaks] += lsum*hist->GetBinCenterY(k);
+ }
+ fYPeaks[fNPeaks] /= sum;
+ Int_t ybin1,ybin2;
+ if(fYPeaks[fNPeaks] < hist->GetBinCenterY(hist->FindYbin(fYPeaks[fNPeaks])))
+ {
+ ybin1 = hist->FindYbin(fYPeaks[fNPeaks])-1;
+ ybin2 = ybin1+1;
+ }
+ else
+ {
+ ybin1 = hist->FindYbin(fYPeaks[fNPeaks]);
+ ybin2 = ybin1+1;
+ }
+
+ Float_t kappa1=0,kappa2=0;
+ sum=0;
+ for(Int_t k=starts[ybin1]; k<starts[ybin1] + kappawindow; k++)
+ {
+ kappa1 += hist->GetBinCenterX(k)*hist->GetBinContent(hist->GetBin(k,ybin1));
+ sum += (Float_t)hist->GetBinContent(hist->GetBin(k,ybin1));
+ }
+ kappa1 /= sum;
+ sum=0;
+ for(Int_t k=starts[ybin2]; k<starts[ybin2] + kappawindow; k++)
+ {
+ kappa2 += hist->GetBinCenterX(k)*hist->GetBinContent(hist->GetBin(k,ybin2));
+ sum += (Float_t)hist->GetBinContent(hist->GetBin(k,ybin2));
+ }
+ kappa2 /= sum;
+
+ fXPeaks[fNPeaks] = ( kappa1*( hist->GetBinCenterY(ybin2) - fYPeaks[fNPeaks] ) +
+ kappa2*( fYPeaks[fNPeaks] - hist->GetBinCenterY(ybin1) ) ) /
+ (hist->GetBinCenterY(ybin2) - hist->GetBinCenterY(ybin1));
+
+ fNPeaks++;
+ */
+ }
+ }
+ break;
+ }
+ else
+ yl--;//Search continues...
}
}
-
}
- tracks->Compress();
- return tracks;
+
+ for(Int_t i=0; i<hist->GetNbinsY(); i++)
+ delete localmaxima[i];
+
+ delete [] localmaxima;
+ delete [] nmaxs;
+ delete [] starts;
+ delete [] maxs;
}
-AliL3HoughTrack *AliL3HoughMaxFinder::CalculatePeakInWindow(Int_t *maxbin,Int_t t0,Int_t t1,Double_t t2,Int_t t3)
+struct AliL3PreYPeak
{
- //Try to expand the area around the maxbin +- t0
-
- if(!fCurrentHisto)
+ Int_t fStartPosition; // Start position in X
+ Int_t fEndPosition; // End position in X
+ Int_t fMinValue; // Minimum value inside the prepeak
+ Int_t fMaxValue; // Maximum value inside the prepeak
+ Int_t fPrevValue; // Neighbour values
+ Int_t fLeftValue; // Neighbour values
+ Int_t fRightValue; // Neighbour values
+};
+
+void AliL3HoughMaxFinder::FindAdaptedRowPeaks(Int_t kappawindow,Int_t xsize,Int_t ysize)
+{
+ // Peak finder which is working over the Hough Space provided by the AliL3HoughTransformerRow class
+ AliL3Histogram *hist = fCurrentHisto;
+
+ if(!hist)
{
- printf("AliL3HoughMaxFinder::LocatePeak : No histogram\n");
- return 0;
+ cerr<<"AliL3HoughMaxFinder : No histogram!"<<endl;
+ return;
}
- AliL3Histogram *hist = fCurrentHisto;
+
+ if(hist->GetNEntries() == 0)
+ return;
+
Int_t xmin = hist->GetFirstXbin();
Int_t xmax = hist->GetLastXbin();
Int_t ymin = hist->GetFirstYbin();
Int_t ymax = hist->GetLastYbin();
+ Int_t nxbins = hist->GetNbinsX()+2;
+ Int_t *content = hist->GetContentArray();
- Int_t xlow = maxbin[0]-t0;
- if(xlow < xmin)
- xlow = xmin;
- Int_t xup = maxbin[0]+t0;
- if(xup > xmax)
- xup = xmax;
- Int_t ylow = maxbin[1]-t0;
- if(ylow < ymin)
- ylow = ymin;
- Int_t yup = maxbin[1]+t0;
- if(yup > ymax)
- yup = ymax;
-
- Int_t nbinsx = hist->GetNbinsX()+1;
+ //Start by looking for pre-peaks:
- Int_t *m = new Int_t[nbinsx];
- Int_t *m_low = new Int_t[nbinsx];
- Int_t *m_up = new Int_t[nbinsx];
-
- recompute:
-
- Int_t max_x=0,sum=0,max_xbin=0,bin;
+ AliL3PreYPeak **localmaxima = new AliL3PreYPeak*[hist->GetNbinsY()];
- for(Int_t xbin=xlow; xbin<=xup; xbin++)
+ Short_t *nmaxs = new Short_t[hist->GetNbinsY()];
+ memset(nmaxs,0,hist->GetNbinsY()*sizeof(Short_t));
+ Int_t lastvalue=0,value=0;
+ for(Int_t ybin=fNextRow[ymin]; ybin<=ymax; ybin = fNextRow[ybin+1])
{
- for(Int_t ybin=ylow; ybin <= yup; ybin++)
+ localmaxima[ybin-ymin] = new AliL3PreYPeak[nxbins-2];
+ lastvalue = 0;
+ Bool_t found = 0;
+ for(Int_t xbin=xmin; xbin<=xmax; xbin++)
{
- sum = 0;
- for(Int_t y=ybin; y <= ybin+t1; y++)
+ value = content[xbin + nxbins*ybin]; //value = hist->GetBinContent(xbin + nxbins*ybin); //value = hist->GetBinContent(hist->GetBin(xbin,ybin));
+ if(value > 0)
{
- if(y>yup) break; //reached the upper limit in y.
- //Inside window
- bin = hist->GetBin(xbin,y);
- sum += (Int_t)hist->GetBinContent(bin);
-
+ if((value - lastvalue) > 1)
+ {
+ localmaxima[ybin-ymin][nmaxs[ybin-ymin]].fStartPosition = xbin;
+ localmaxima[ybin-ymin][nmaxs[ybin-ymin]].fEndPosition = xbin;
+ localmaxima[ybin-ymin][nmaxs[ybin-ymin]].fMinValue = value;
+ localmaxima[ybin-ymin][nmaxs[ybin-ymin]].fMaxValue = value;
+ localmaxima[ybin-ymin][nmaxs[ybin-ymin]].fPrevValue = 0;
+ localmaxima[ybin-ymin][nmaxs[ybin-ymin]].fLeftValue = lastvalue;
+ found = 1;
+ }
+ if(abs(value - lastvalue) <= 1)
+ {
+ if(found) {
+ localmaxima[ybin-ymin][nmaxs[ybin-ymin]].fEndPosition = xbin;
+ if(value>localmaxima[ybin-ymin][nmaxs[ybin-ymin]].fMaxValue)
+ localmaxima[ybin-ymin][nmaxs[ybin-ymin]].fMaxValue = value;
+ if(value<localmaxima[ybin-ymin][nmaxs[ybin-ymin]].fMinValue)
+ localmaxima[ybin-ymin][nmaxs[ybin-ymin]].fMinValue = value;
+ }
+ }
+ if((value - lastvalue) < -1)
+ {
+ if(found) {
+ localmaxima[ybin-ymin][nmaxs[ybin-ymin]].fRightValue = value;
+ nmaxs[ybin-ymin]++;
+ found = 0;
+ }
+ }
}
- if(sum > m[xbin]) //Max value locally in this xbin
+ else
{
- m[xbin]=sum;
- m_low[xbin]=ybin;
- m_up[xbin]=ybin + t1;
+ if(found) {
+ localmaxima[ybin-ymin][nmaxs[ybin-ymin]].fRightValue = value;
+ nmaxs[ybin-ymin]++;
+ found = 0;
+ }
}
-
- }
-
- if(m[xbin] > max_x) //Max value globally in x-direction
- {
- max_xbin = xbin;
- max_x = m[xbin];//sum;
+ lastvalue = value;
+
}
+ if(found) {
+ localmaxima[ybin-ymin][nmaxs[ybin-ymin]].fRightValue = 0;
+ nmaxs[ybin-ymin]++;
+ }
}
- //printf("max_xbin %d max_x %d m_low %d m_up %d\n",max_xbin,max_x,m_low[max_xbin],m_up[max_xbin]);
- //printf("ylow %f yup %f\n",hist->GetBinCenterY(m_low[max_xbin]),hist->GetBinCenterY(m_up[max_xbin]));
+
+ AliL3Pre2DPeak maxima[500];
+ Int_t nmaxima = 0;
- //Determine a width in the x-direction
- Int_t x_low=0,x_up=0;
- for(Int_t xbin=max_xbin-1; xbin >= xmin; xbin--)
+ for(Int_t ybin=ymax; ybin >= ymin; ybin--)
{
- if(m[xbin] < max_x*t2)
+ for(Int_t i=0; i<nmaxs[ybin-ymin]; i++)
{
- x_low = xbin+1;
- break;
- }
- }
- for(Int_t xbin = max_xbin+1; xbin <=xmax; xbin++)
- {
- if(m[xbin] < max_x*t2)
- {
- x_up = xbin-1;
- break;
- }
- }
- printf("x_low %d x_up %d\n",x_low,x_up);
+ Int_t localminvalue = localmaxima[ybin-ymin][i].fMinValue;
+ Int_t localmaxvalue = localmaxima[ybin-ymin][i].fMaxValue;
+ Int_t localprevvalue = localmaxima[ybin-ymin][i].fPrevValue;
+ Int_t localnextvalue = 0;
+ Int_t localleftvalue = localmaxima[ybin-ymin][i].fLeftValue;
+ Int_t localrightvalue = localmaxima[ybin-ymin][i].fRightValue;
- Double_t top=0,butt=0,value,x_peak;
- if(x_up - x_low + 1 > t3)
- {
- t1 -= 1;
- printf("\nxrange out if limit x_up %d x_low %d\n\n",x_low,x_up);
- if(t1 > 1)
- goto recompute;
- else
- {
- x_peak = hist->GetBinCenterX(max_xbin);
- goto moveon;
- }
- }
-
- //printf("xlow %f xup %f\n",hist->GetBinCenterX(x_low),hist->GetBinCenterX(x_up));
- //printf("Spread in x %d\n",x_up-x_low +1);
+ if(localminvalue<0)
+ continue; //already used
- //Now, calculate the center of mass in x-direction
- for(Int_t xbin=x_low; xbin <= x_up; xbin++)
- {
- value = hist->GetBinCenterX(xbin);
- top += value*m[xbin];
- butt += m[xbin];
- }
- x_peak = top/butt;
-
- moveon:
-
- //Find the peak in y direction:
- Int_t x_l = hist->FindXbin(x_peak);
- if(hist->GetBinCenterX(x_l) > x_peak)
- x_l--;
+ //start expanding in the psi-direction:
- Int_t x_u = x_l + 1;
-
- if(hist->GetBinCenterX(x_l) > x_peak || hist->GetBinCenterX(x_u) <= x_peak)
- printf("\nAliL3HoughMaxFinder::FindPeak : Wrong xrange %f %f %f\n\n",hist->GetBinCenterX(x_l),x_peak,hist->GetBinCenterX(x_u));
-
- //printf("\nxlow %f xup %f\n",hist->GetBinCenterX(x_l),hist->GetBinCenterX(x_u));
+ Int_t localxstart = localmaxima[ybin-ymin][i].fStartPosition;
+ Int_t localxend = localmaxima[ybin-ymin][i].fEndPosition;
+ Int_t tempxstart = localmaxima[ybin-ymin][i].fStartPosition;
+ Int_t tempxend = localmaxima[ybin-ymin][i].fEndPosition;
- value=top=butt=0;
-
- //printf("ylow %f yup %f\n",hist->GetBinCenterY(m_low[x_l]),hist->GetBinCenterY(m_up[x_l]));
- //printf("ylow %f yup %f\n",hist->GetBinCenterY(m_low[x_u]),hist->GetBinCenterY(m_up[x_u]));
-
- for(Int_t ybin=m_low[x_l]; ybin <= m_up[x_l]; ybin++)
- {
- value = hist->GetBinCenterY(ybin);
- bin = hist->GetBin(x_l,ybin);
- top += value*hist->GetBinContent(bin);
- butt += hist->GetBinContent(bin);
- }
- Double_t y_peak_low = top/butt;
-
- //printf("y_peak_low %f\n",y_peak_low);
+ Int_t localy=ybin-1,nybins=1;
- value=top=butt=0;
- for(Int_t ybin=m_low[x_u]; ybin <= m_up[x_u]; ybin++)
- {
- value = hist->GetBinCenterY(ybin);
- bin = hist->GetBin(x_u,ybin);
- top += value*hist->GetBinContent(bin);
- butt += hist->GetBinContent(bin);
+ while(localy >= ymin)
+ {
+ Bool_t found=0;
+ for(Int_t j=0; j<nmaxs[localy-ymin]; j++)
+ {
+ if( (localmaxima[localy-ymin][j].fStartPosition <= (tempxend + kappawindow)) && (localmaxima[localy-ymin][j].fEndPosition >= (tempxstart - kappawindow)))
+ {
+ if((localmaxima[localy-ymin][j].fMinValue <= localmaxvalue) && (localmaxima[localy-ymin][j].fMaxValue >= localminvalue))
+ {
+ if(localmaxima[localy-ymin][j].fEndPosition > localxend)
+ localxend = localmaxima[localy-ymin][j].fEndPosition;
+ if(localmaxima[localy-ymin][j].fStartPosition < localxstart)
+ localxstart = localmaxima[localy-ymin][j].fStartPosition;
+ tempxstart = localmaxima[localy-ymin][j].fStartPosition;
+ tempxend = localmaxima[localy-ymin][j].fEndPosition;
+ if(localmaxima[localy-ymin][j].fMinValue < localminvalue)
+ localminvalue = localmaxima[localy-ymin][j].fMinValue;
+ if(localmaxima[localy-ymin][j].fMaxValue > localmaxvalue)
+ localmaxvalue = localmaxima[localy-ymin][j].fMaxValue;
+ if(localmaxima[localy-ymin][j].fRightValue > localrightvalue)
+ localrightvalue = localmaxima[localy-ymin][j].fRightValue;
+ if(localmaxima[localy-ymin][j].fLeftValue > localleftvalue)
+ localleftvalue = localmaxima[localy-ymin][j].fLeftValue;
+ localmaxima[localy-ymin][j].fMinValue = -1;
+ found = 1;
+ nybins++;
+ break;
+ }
+ else
+ {
+ if(localmaxvalue > localmaxima[localy-ymin][j].fPrevValue)
+ localmaxima[localy-ymin][j].fPrevValue = localmaxvalue;
+ if(localmaxima[localy-ymin][j].fMaxValue > localnextvalue)
+ localnextvalue = localmaxima[localy-ymin][j].fMaxValue;
+ }
+ }
+ }
+ if(!found || localy == ymin)//no more local maximas to be matched, so write the final peak and break the expansion:
+ {
+ if((nybins > ysize) && ((localxend-localxstart+1) > xsize) && (localmaxvalue > localprevvalue) && (localmaxvalue > localnextvalue) && (localmaxvalue > localleftvalue) && (localmaxvalue > localrightvalue))
+ // if((nybins > ysize) && ((localxend-localxstart+1) > xsize))
+ {
+ maxima[nmaxima].fX = ((Float_t)localxstart+(Float_t)localxend)/2.0;
+ maxima[nmaxima].fY = ((Float_t)ybin+(Float_t)(localy+1))/2.0;
+ maxima[nmaxima].fSizeX = localxend-localxstart+1;
+ maxima[nmaxima].fSizeY = nybins;
+ maxima[nmaxima].fWeight = (localminvalue+localmaxvalue)/2;
+ maxima[nmaxima].fStartX = localxstart;
+ maxima[nmaxima].fEndX = localxend;
+ maxima[nmaxima].fStartY = localy +1;
+ maxima[nmaxima].fEndY = ybin;
+#ifdef do_mc
+ // cout<<"Peak found at: "<<((Float_t)localxstart+(Float_t)localxend)/2.0<<" "<<((Float_t)ybin+(Float_t)(localy+1))/2.0<<" "<<localminvalue<<" "<<localmaxvalue<<" "<<" with weight "<<(localminvalue+localmaxvalue)/2<<" and size "<<localxend-localxstart+1<<" by "<<nybins<<endl;
+#endif
+ nmaxima++;
+ }
+ break;
+ }
+ else
+ localy--;//Search continues...
+ }
+ }
}
- Double_t y_peak_up = top/butt;
-
- //printf("y_peak_up %f\n",y_peak_up);
-
- Double_t x_value_up = hist->GetBinCenterX(x_u);
- Double_t x_value_low = hist->GetBinCenterX(x_l);
-
- Double_t y_peak = (y_peak_low*(x_value_up - x_peak) + y_peak_up*(x_peak - x_value_low))/(x_value_up - x_value_low);
-
- //Find the weight:
- bin = hist->FindBin(x_peak,y_peak);
- Int_t weight = (Int_t)hist->GetBinContent(bin);
-
- AliL3HoughTrack *track = new AliL3HoughTrack();
- track->SetTrackParameters(x_peak,y_peak,weight);
+ //remove fake tracks
- //Reset area around peak
- for(Int_t xbin=x_low; xbin<=x_up; xbin++)
+ for(Int_t i = 0; i < (nmaxima - 1); i++)
{
- for(Int_t ybin=m_low[xbin]; ybin<=m_up[xbin]; ybin++)
+ // if(maxima[i].fWeight < 0) continue;
+ for(Int_t j = i + 1; j < nmaxima; j++)
{
- bin = hist->GetBin(xbin,ybin);
- hist->SetBinContent(bin,0);
+ // if(maxima[j].fWeight < 0) continue;
+ MergeRowPeaks(&maxima[i],&maxima[j],5.0);
}
}
-
- delete [] m;
- delete [] m_low;
- delete [] m_up;
-
- return track;
-
-
-}
-
-AliL3HoughTrack *AliL3HoughMaxFinder::FindPeakLine(Double_t rho,Double_t theta)
-{
- //Peak finder based on a second line transformation on kappa-phi space,
- //to use as a baseline.
- if(!fCurrentHisto)
- {
- printf("AliL3HoughTransformer::FindPeakLine : No input histogram\n");
- return 0;
- }
-
- //get the line parameters:
- Double_t a = -1./tan(theta);
- Double_t b = rho/sin(theta);
-
- printf("rho %f theta %f\n",rho,theta);
- //now, start looking along the line.
- Int_t xmin = fCurrentHisto->GetFirstXbin();
- Int_t xmax = fCurrentHisto->GetLastXbin();
-
- Int_t max_weight=0;
- Double_t max_bin[2];
- for(Int_t xbin=xmin; xbin<=xmax; xbin++)
- {
- Double_t x = fCurrentHisto->GetBinCenterX(xbin);
- Double_t y = a*x + b;
- Int_t bin = fCurrentHisto->FindBin(x,y);
- //printf("x %f y %f weight %d\n",x,y,fCurrentHisto->GetBinContent(bin));
- if(fCurrentHisto->GetBinContent(bin) > max_weight)
- {
- max_weight = (Int_t)fCurrentHisto->GetBinContent(bin);
- max_bin[0] = x;
- max_bin[1] = y;
+ //merge tracks in neighbour eta slices
+ Int_t currentnpeaks = fNPeaks;
+ for(Int_t i = 0; i < nmaxima; i++) {
+ if(maxima[i].fWeight < 0) continue;
+ Bool_t merged = kFALSE;
+ for(Int_t j = fN1PeaksPrevEtaSlice; j < fN2PeaksPrevEtaSlice; j++) {
+ // if(fWeight[j] < 0) continue;
+ // Merge only peaks with limited size in eta
+ if((fENDETAPeaks[j]-fSTARTETAPeaks[j]) >= 2) continue;
+ if((maxima[i].fStartX <= fENDXPeaks[j]+1) && (maxima[i].fEndX >= fSTARTXPeaks[j]-1) && (maxima[i].fStartY <= fENDYPeaks[j]+1) && (maxima[i].fEndY >= fSTARTYPeaks[j]-1)){
+ //merge
+ merged = kTRUE;
+ if(fWeight[j] > 0) {
+ fXPeaks[fNPeaks] = (hist->GetPreciseBinCenterX(maxima[i].fX)+(fENDETAPeaks[j]-fSTARTETAPeaks[j]+1)*fXPeaks[j])/(fENDETAPeaks[j]-fSTARTETAPeaks[j]+2);
+ fYPeaks[fNPeaks] = (hist->GetPreciseBinCenterY(maxima[i].fY)+(fENDETAPeaks[j]-fSTARTETAPeaks[j]+1)*fYPeaks[j])/(fENDETAPeaks[j]-fSTARTETAPeaks[j]+2);
+ fSTARTXPeaks[fNPeaks] = maxima[i].fStartX;
+ fSTARTYPeaks[fNPeaks] = maxima[i].fStartY;
+ fENDXPeaks[fNPeaks] = maxima[i].fEndX;
+ fENDYPeaks[fNPeaks] = maxima[i].fEndY;
+
+ fWeight[fNPeaks] = abs((Int_t)maxima[i].fWeight) + abs(fWeight[j]);
+ fSTARTETAPeaks[fNPeaks] = fSTARTETAPeaks[j];
+ fENDETAPeaks[fNPeaks] = fCurrentEtaSlice;
+ fNPeaks++;
}
+ fWeight[j] = -abs(fWeight[j]);
+ }
}
-
- AliL3HoughTrack *track = new AliL3HoughTrack();
- track->SetTrackParameters(max_bin[0],max_bin[1],max_weight);
- return track;
+ fXPeaks[fNPeaks] = hist->GetPreciseBinCenterX(maxima[i].fX);
+ fYPeaks[fNPeaks] = hist->GetPreciseBinCenterY(maxima[i].fY);
+ if(!merged)
+ fWeight[fNPeaks] = abs((Int_t)maxima[i].fWeight);
+ else
+ fWeight[fNPeaks] = -abs((Int_t)maxima[i].fWeight);
+ fSTARTXPeaks[fNPeaks] = maxima[i].fStartX;
+ fSTARTYPeaks[fNPeaks] = maxima[i].fStartY;
+ fENDXPeaks[fNPeaks] = maxima[i].fEndX;
+ fENDYPeaks[fNPeaks] = maxima[i].fEndY;
+ fSTARTETAPeaks[fNPeaks] = fCurrentEtaSlice;
+ fENDETAPeaks[fNPeaks] = fCurrentEtaSlice;
+ fNPeaks++;
+ }
+
+ fN1PeaksPrevEtaSlice = currentnpeaks;
+ fN2PeaksPrevEtaSlice = fNPeaks;
+
+ for(Int_t ybin=fNextRow[ymin]; ybin<=ymax; ybin = fNextRow[ybin+1])
+ delete [] localmaxima[ybin-ymin];
+
+ delete [] localmaxima;
+ delete [] nmaxs;
}
-void AliL3HoughMaxFinder::FindPeak1(Int_t y_window,Int_t x_bin_sides)
+void AliL3HoughMaxFinder::FindPeak1(Int_t ywindow,Int_t xbinsides)
{
//Testing mutliple peakfinding.
//The algorithm searches the histogram for prepreaks by looking in windows
- //for each bin on the xaxis. The size of these windows is controlled by y_window.
+ //for each bin on the xaxis. The size of these windows is controlled by ywindow.
//Then the prepreaks are sorted according to their weight (sum inside window),
//and the peak positions are calculated by taking the weighted mean in both
- //x and y direction. The size of the peak in x-direction is controlled by x_bin_sides.
+ //x and y direction. The size of the peak in x-direction is controlled by xbinsides.
if(!fCurrentHisto)
{
printf("AliL3HoughMaxFinder::FindPeak1 : No input histogram\n");
return;
}
-
- //Int_t y_window=2;
- //Int_t x_bin_sides=1;
+ if(fCurrentHisto->GetNEntries()==0)
+ return;
+
+ //Int_t ywindow=2;
+ //Int_t xbinsides=1;
//Float_t max_kappa = 0.001;
//Float_t max_phi0 = 0.08;
- Int_t max_sum=0;
+ Int_t maxsum=0;
Int_t xmin = fCurrentHisto->GetFirstXbin();
Int_t xmax = fCurrentHisto->GetLastXbin();
Int_t ymax = fCurrentHisto->GetLastYbin();
Int_t nbinsx = fCurrentHisto->GetNbinsX()+1;
- AxisWindow **windowPt = new AxisWindow*[nbinsx];
- AxisWindow **anotherPt = new AxisWindow*[nbinsx];
+ AliL3AxisWindow **windowPt = new AliL3AxisWindow*[nbinsx];
+ AliL3AxisWindow **anotherPt = new AliL3AxisWindow*[nbinsx];
for(Int_t i=0; i<nbinsx; i++)
{
- windowPt[i] = new AxisWindow;
- bzero((void*)windowPt[i],sizeof(AxisWindow));
+ windowPt[i] = new AliL3AxisWindow;
+#if defined(__DECCXX)
+ bzero((char *)windowPt[i],sizeof(AliL3AxisWindow));
+#else
+ bzero((void*)windowPt[i],sizeof(AliL3AxisWindow));
+#endif
anotherPt[i] = windowPt[i];
}
for(Int_t xbin=xmin; xbin<=xmax; xbin++)
{
- max_sum = 0;
- for(Int_t ybin=ymin; ybin<=ymax-y_window; ybin++)
+ maxsum = 0;
+ for(Int_t ybin=ymin; ybin<=ymax-ywindow; ybin++)
{
- Int_t sum_in_window=0;
- for(Int_t b=ybin; b<ybin+y_window; b++)
+ Int_t suminwindow=0;
+ for(Int_t b=ybin; b<ybin+ywindow; b++)
{
//inside window
Int_t bin = fCurrentHisto->GetBin(xbin,b);
- sum_in_window += (Int_t)fCurrentHisto->GetBinContent(bin);
+ suminwindow += (Int_t)fCurrentHisto->GetBinContent(bin);
}
- if(sum_in_window > max_sum)
+ if(suminwindow > maxsum)
{
- max_sum = sum_in_window;
- windowPt[xbin]->ymin = ybin;
- windowPt[xbin]->ymax = ybin + y_window;
- windowPt[xbin]->weight = sum_in_window;
- windowPt[xbin]->xbin = xbin;
+ maxsum = suminwindow;
+ windowPt[xbin]->fYmin = ybin;
+ windowPt[xbin]->fYmax = ybin + ywindow;
+ windowPt[xbin]->fWeight = suminwindow;
+ windowPt[xbin]->fXbin = xbin;
}
}
}
for(Int_t i=0; i<nbinsx; i++)
{
top=butt=0;
- Int_t xbin = windowPt[i]->xbin;
+ Int_t xbin = windowPt[i]->fXbin;
if(xbin<xmin || xbin > xmax-1) continue;
//Check if this is really a local maxima
- if(anotherPt[xbin-1]->weight > anotherPt[xbin]->weight ||
- anotherPt[xbin+1]->weight > anotherPt[xbin]->weight)
+ if(anotherPt[xbin-1]->fWeight > anotherPt[xbin]->fWeight ||
+ anotherPt[xbin+1]->fWeight > anotherPt[xbin]->fWeight)
continue;
- for(Int_t j=windowPt[i]->ymin; j<windowPt[i]->ymax; j++)
+ for(Int_t j=windowPt[i]->fYmin; j<windowPt[i]->fYmax; j++)
{
//Calculate the mean in y direction:
- Int_t bin = fCurrentHisto->GetBin(windowPt[i]->xbin,j);
+ Int_t bin = fCurrentHisto->GetBin(windowPt[i]->fXbin,j);
top += (fCurrentHisto->GetBinCenterY(j))*(fCurrentHisto->GetBinContent(bin));
butt += fCurrentHisto->GetBinContent(bin);
}
- //if(butt < fThreshold)
- // continue;
+ if(butt < fThreshold)
+ continue;
- fXPeaks[fNPeaks] = fCurrentHisto->GetBinCenterX(windowPt[i]->xbin);
+ fXPeaks[fNPeaks] = fCurrentHisto->GetBinCenterX(windowPt[i]->fXbin);
fYPeaks[fNPeaks] = top/butt;
fWeight[fNPeaks] = (Int_t)butt;
//cout<<"mean in y "<<ypeaks[n]<<" on x "<<windowPt[i]->xbin<<" content "<<butt<<endl;
for(Int_t i=0; i<fNPeaks; i++)
{
Int_t xbin = fCurrentHisto->FindXbin(fXPeaks[i]);
- if(xbin - x_bin_sides < xmin || xbin + x_bin_sides > xmax) continue;
+ if(xbin - xbinsides < xmin || xbin + xbinsides > xmax) continue;
top=butt=0;
ytop=0,ybutt=0;
w=0;
- prev = xbin - x_bin_sides+1;
- for(Int_t j=xbin-x_bin_sides; j<=xbin+x_bin_sides; j++)
+ prev = xbin - xbinsides+1;
+ for(Int_t j=xbin-xbinsides; j<=xbin+xbinsides; j++)
{
/*
//Check if the windows are overlapping:
prev = j;
*/
- top += fCurrentHisto->GetBinCenterX(j)*anotherPt[j]->weight;
- butt += anotherPt[j]->weight;
+ top += fCurrentHisto->GetBinCenterX(j)*anotherPt[j]->fWeight;
+ butt += anotherPt[j]->fWeight;
- for(Int_t k=anotherPt[j]->ymin; k<anotherPt[j]->ymax; k++)
+ for(Int_t k=anotherPt[j]->fYmin; k<anotherPt[j]->fYmax; k++)
{
Int_t bin = fCurrentHisto->GetBin(j,k);
ytop += (fCurrentHisto->GetBinCenterY(k))*(fCurrentHisto->GetBinContent(bin));
delete windowPt[i];
delete [] windowPt;
delete [] anotherPt;
-
}
-void AliL3HoughMaxFinder::SortPeaks(struct AxisWindow **a,Int_t first,Int_t last)
+void AliL3HoughMaxFinder::SortPeaks(struct AliL3AxisWindow **a,Int_t first,Int_t last)
{
//General sorting routine
//Sort according to PeakCompare()
- static struct AxisWindow *tmp;
+ static struct AliL3AxisWindow *tmp;
static int i; // "static" to save stack space
int j;
}
-Int_t AliL3HoughMaxFinder::PeakCompare(struct AxisWindow *a,struct AxisWindow *b)
+Int_t AliL3HoughMaxFinder::PeakCompare(struct AliL3AxisWindow *a,struct AliL3AxisWindow *b) const
{
- if(a->weight < b->weight) return 1;
- if(a->weight > b->weight) return -1;
+ // Peak comparison based on peaks weight
+ if(a->fWeight < b->fWeight) return 1;
+ if(a->fWeight > b->fWeight) return -1;
return 0;
-
}
void AliL3HoughMaxFinder::FindPeak(Int_t t1,Double_t t2,Int_t t3)
return;
}
AliL3Histogram *hist = fCurrentHisto;
+ if(hist->GetNEntries()==0)
+ return;
Int_t xmin = hist->GetFirstXbin();
Int_t xmax = hist->GetLastXbin();
Int_t nbinsx = hist->GetNbinsX()+1;
Int_t *m = new Int_t[nbinsx];
- Int_t *m_low = new Int_t[nbinsx];
- Int_t *m_up = new Int_t[nbinsx];
+ Int_t *mlow = new Int_t[nbinsx];
+ Int_t *mup = new Int_t[nbinsx];
recompute: //this is a goto.
for(Int_t i=0; i<nbinsx; i++)
{
m[i]=0;
- m_low[i]=0;
- m_up[i]=0;
+ mlow[i]=0;
+ mup[i]=0;
}
- Int_t max_x=0,sum=0,max_xbin=0,bin=0;
+ Int_t maxx=0,sum=0,maxxbin=0,bin=0;
for(Int_t xbin=xmin; xbin<=xmax; xbin++)
{
if(sum > m[xbin]) //Max value locally in this xbin
{
m[xbin]=sum;
- m_low[xbin]=ybin;
- m_up[xbin]=ybin + t1;
+ mlow[xbin]=ybin;
+ mup[xbin]=ybin + t1;
}
}
- if(m[xbin] > max_x) //Max value globally in x-direction
+ if(m[xbin] > maxx) //Max value globally in x-direction
{
- max_xbin = xbin;
- max_x = m[xbin];//sum;
+ maxxbin = xbin;
+ maxx = m[xbin];//sum;
}
}
- //printf("max_xbin %d max_x %d m_low %d m_up %d\n",max_xbin,max_x,m_low[max_xbin],m_up[max_xbin]);
- //printf("ylow %f yup %f\n",hist->GetBinCenterY(m_low[max_xbin]),hist->GetBinCenterY(m_up[max_xbin]));
+ //printf("maxxbin %d maxx %d mlow %d mup %d\n",maxxbin,maxx,mlow[maxxbin],mup[maxxbin]);
+ //printf("ylow %f yup %f\n",hist->GetBinCenterY(mlow[maxxbin]),hist->GetBinCenterY(mup[maxxbin]));
//Determine a width in the x-direction
- Int_t x_low=0,x_up=0;
+ Int_t xlow=0,xup=0;
- for(Int_t xbin=max_xbin-1; xbin >= xmin; xbin--)
+ for(Int_t xbin=maxxbin-1; xbin >= xmin; xbin--)
{
- if(m[xbin] < max_x*t2)
+ if(m[xbin] < maxx*t2)
{
- x_low = xbin+1;
+ xlow = xbin+1;
break;
}
}
- for(Int_t xbin = max_xbin+1; xbin <=xmax; xbin++)
+ for(Int_t xbin = maxxbin+1; xbin <=xmax; xbin++)
{
- if(m[xbin] < max_x*t2)
+ if(m[xbin] < maxx*t2)
{
- x_up = xbin-1;
+ xup = xbin-1;
break;
}
}
- Double_t top=0,butt=0,value,x_peak;
- if(x_up - x_low + 1 > t3)
+ Double_t top=0,butt=0,value,xpeak;
+ if(xup - xlow + 1 > t3)
{
t1 -= 1;
- printf("\nxrange out if limit x_up %d x_low %d t1 %d\n\n",x_low,x_up,t1);
+ printf("\nxrange out if limit xup %d xlow %d t1 %d\n\n",xlow,xup,t1);
if(t1 > 1)
goto recompute;
else
{
- x_peak = hist->GetBinCenterX(max_xbin);
+ xpeak = hist->GetBinCenterX(maxxbin);
goto moveon;
}
}
- //printf("xlow %f xup %f\n",hist->GetBinCenterX(x_low),hist->GetBinCenterX(x_up));
- //printf("Spread in x %d\n",x_up-x_low +1);
+ //printf("xlow %f xup %f\n",hist->GetBinCenterX(xlow),hist->GetBinCenterX(xup));
+ //printf("Spread in x %d\n",xup-xlow +1);
//Now, calculate the center of mass in x-direction
- for(Int_t xbin=x_low; xbin <= x_up; xbin++)
+ for(Int_t xbin=xlow; xbin <= xup; xbin++)
{
value = hist->GetBinCenterX(xbin);
top += value*m[xbin];
butt += m[xbin];
}
- x_peak = top/butt;
+ xpeak = top/butt;
moveon:
//Find the peak in y direction:
- Int_t x_l = hist->FindXbin(x_peak);
- if(hist->GetBinCenterX(x_l) > x_peak)
- x_l--;
+ Int_t xl = hist->FindXbin(xpeak);
+ if(hist->GetBinCenterX(xl) > xpeak)
+ xl--;
- Int_t x_u = x_l + 1;
+ Int_t xu = xl + 1;
- if(hist->GetBinCenterX(x_l) > x_peak || hist->GetBinCenterX(x_u) <= x_peak)
- printf("\nAliL3HoughMaxFinder::FindPeak : Wrong xrange %f %f %f\n\n",hist->GetBinCenterX(x_l),x_peak,hist->GetBinCenterX(x_u));
+ if(hist->GetBinCenterX(xl) > xpeak || hist->GetBinCenterX(xu) <= xpeak)
+ printf("\nAliL3HoughMaxFinder::FindPeak : Wrong xrange %f %f %f\n\n",hist->GetBinCenterX(xl),xpeak,hist->GetBinCenterX(xu));
- //printf("\nxlow %f xup %f\n",hist->GetBinCenterX(x_l),hist->GetBinCenterX(x_u));
+ //printf("\nxlow %f xup %f\n",hist->GetBinCenterX(xl),hist->GetBinCenterX(xu));
value=top=butt=0;
- //printf("ylow %f yup %f\n",hist->GetBinCenterY(m_low[x_l]),hist->GetBinCenterY(m_up[x_l]));
- //printf("ylow %f yup %f\n",hist->GetBinCenterY(m_low[x_u]),hist->GetBinCenterY(m_up[x_u]));
+ //printf("ylow %f yup %f\n",hist->GetBinCenterY(mlow[xl]),hist->GetBinCenterY(mup[xl]));
+ //printf("ylow %f yup %f\n",hist->GetBinCenterY(mlow[xu]),hist->GetBinCenterY(mup[xu]));
- for(Int_t ybin=m_low[x_l]; ybin <= m_up[x_l]; ybin++)
+ for(Int_t ybin=mlow[xl]; ybin <= mup[xl]; ybin++)
{
value = hist->GetBinCenterY(ybin);
- bin = hist->GetBin(x_l,ybin);
+ bin = hist->GetBin(xl,ybin);
top += value*hist->GetBinContent(bin);
butt += hist->GetBinContent(bin);
}
- Double_t y_peak_low = top/butt;
+ Double_t ypeaklow = top/butt;
- //printf("y_peak_low %f\n",y_peak_low);
+ //printf("ypeaklow %f\n",ypeaklow);
value=top=butt=0;
- for(Int_t ybin=m_low[x_u]; ybin <= m_up[x_u]; ybin++)
+ for(Int_t ybin=mlow[xu]; ybin <= mup[xu]; ybin++)
{
value = hist->GetBinCenterY(ybin);
- bin = hist->GetBin(x_u,ybin);
+ bin = hist->GetBin(xu,ybin);
top += value*hist->GetBinContent(bin);
butt += hist->GetBinContent(bin);
}
- Double_t y_peak_up = top/butt;
+ Double_t ypeakup = top/butt;
- //printf("y_peak_up %f\n",y_peak_up);
+ //printf("ypeakup %f\n",ypeakup);
- Double_t x_value_up = hist->GetBinCenterX(x_u);
- Double_t x_value_low = hist->GetBinCenterX(x_l);
+ Double_t xvalueup = hist->GetBinCenterX(xu);
+ Double_t xvaluelow = hist->GetBinCenterX(xl);
- Double_t y_peak = (y_peak_low*(x_value_up - x_peak) + y_peak_up*(x_peak - x_value_low))/(x_value_up - x_value_low);
+ Double_t ypeak = (ypeaklow*(xvalueup - xpeak) + ypeakup*(xpeak - xvaluelow))/(xvalueup - xvaluelow);
//Find the weight:
- //bin = hist->FindBin(x_peak,y_peak);
+ //bin = hist->FindBin(xpeak,ypeak);
//Int_t weight = (Int_t)hist->GetBinContent(bin);
//AliL3HoughTrack *track = new AliL3HoughTrack();
- //track->SetTrackParameters(x_peak,y_peak,weight);
- fXPeaks[fNPeaks]=x_peak;
- fYPeaks[fNPeaks]=y_peak;
+ //track->SetTrackParameters(xpeak,ypeak,weight);
+ fXPeaks[fNPeaks]=xpeak;
+ fYPeaks[fNPeaks]=ypeak;
fWeight[fNPeaks]=(Int_t)hist->GetBinContent(bin);
fNPeaks++;
delete [] m;
- delete [] m_low;
- delete [] m_up;
+ delete [] mlow;
+ delete [] mup;
//return track;
+}
-
+Float_t AliL3HoughMaxFinder::GetXPeakSize(Int_t i) const
+{
+ // Get X size of a peak
+ if(i<0 || i>fNMax)
+ {
+ STDCERR<<"AliL3HoughMaxFinder::GetXPeakSize : Invalid index "<<i<<STDENDL;
+ return 0;
+ }
+ Float_t binwidth = fCurrentHisto->GetBinWidthX();
+ return binwidth*(fENDXPeaks[i]-fSTARTXPeaks[i]+1);
}
+Float_t AliL3HoughMaxFinder::GetYPeakSize(Int_t i) const
+{
+ // Get Y size of a peak
+ if(i<0 || i>fNMax)
+ {
+ STDCERR<<"AliL3HoughMaxFinder::GetYPeak : Invalid index "<<i<<STDENDL;
+ return 0;
+ }
+ Float_t binwidth = fCurrentHisto->GetBinWidthY();
+ return binwidth*(fENDYPeaks[i]-fSTARTYPeaks[i]+1);
+}
+
+Bool_t AliL3HoughMaxFinder::MergeRowPeaks(AliL3Pre2DPeak *maxima1, AliL3Pre2DPeak *maxima2, Float_t distance)
+{
+ // Check the distance between tracks corresponding to given Hough space peaks and if the
+ // distance is smaller than some threshold value marks the smaller peak as fake
+ AliL3Histogram *hist = fCurrentHisto;
+ Int_t nxbins = hist->GetNbinsX()+2;
+
+ Int_t xtrack1=0,xtrack2=0,ytrack1=0,ytrack2=0;
+ Int_t deltax = 9999;
+ for(Int_t ix1 = maxima1->fStartX; ix1 <= maxima1->fEndX; ix1++) {
+ for(Int_t ix2 = maxima2->fStartX; ix2 <= maxima2->fEndX; ix2++) {
+ if(abs(ix1 - ix2) < deltax) {
+ deltax = abs(ix1 - ix2);
+ xtrack1 = ix1;
+ xtrack2 = ix2;
+ }
+ }
+ }
+ Int_t deltay = 9999;
+ for(Int_t iy1 = maxima1->fStartY; iy1 <= maxima1->fEndY; iy1++) {
+ for(Int_t iy2 = maxima2->fStartY; iy2 <= maxima2->fEndY; iy2++) {
+ if(abs(iy1 - iy2) < deltay) {
+ deltay = abs(iy1 - iy2);
+ ytrack1 = iy1;
+ ytrack2 = iy2;
+ }
+ }
+ }
+ Int_t firstrow1 = fTrackFirstRow[xtrack1 + nxbins*ytrack1];
+ Int_t lastrow1 = fTrackLastRow[xtrack1 + nxbins*ytrack1];
+ Int_t firstrow2 = fTrackFirstRow[xtrack1 + nxbins*ytrack1];
+ Int_t lastrow2 = fTrackLastRow[xtrack1 + nxbins*ytrack1];
+ Int_t firstrow,lastrow;
+ if(firstrow1 < firstrow2)
+ firstrow = firstrow2;
+ else
+ firstrow = firstrow1;
+
+ if(lastrow1 > lastrow2)
+ lastrow = lastrow2;
+ else
+ lastrow = lastrow1;
+
+ AliL3HoughTrack track1;
+ Float_t x1 = hist->GetPreciseBinCenterX(xtrack1);
+ Float_t y1 = hist->GetPreciseBinCenterY(ytrack1);
+ Float_t psi1 = atan((x1-y1)/(AliL3HoughTransformerRow::GetBeta1()-AliL3HoughTransformerRow::GetBeta2()));
+ Float_t kappa1 = 2.0*(x1*cos(psi1)-AliL3HoughTransformerRow::GetBeta1()*sin(psi1));
+ track1.SetTrackParameters(kappa1,psi1,1);
+ Float_t firsthit1[3];
+ if(!track1.GetCrossingPoint(firstrow,firsthit1)) return kFALSE;
+ Float_t lasthit1[3];
+ if(!track1.GetCrossingPoint(lastrow,lasthit1)) return kFALSE;
+
+ AliL3HoughTrack track2;
+ Float_t x2 = hist->GetPreciseBinCenterX(xtrack2);
+ Float_t y2 = hist->GetPreciseBinCenterY(ytrack2);
+ Float_t psi2 = atan((x2-y2)/(AliL3HoughTransformerRow::GetBeta1()-AliL3HoughTransformerRow::GetBeta2()));
+ Float_t kappa2 = 2.0*(x2*cos(psi2)-AliL3HoughTransformerRow::GetBeta1()*sin(psi2));
+ track2.SetTrackParameters(kappa2,psi2,1);
+ Float_t firsthit2[3];
+ if(!track2.GetCrossingPoint(firstrow,firsthit2)) return kFALSE;
+ Float_t lasthit2[3];
+ if(!track2.GetCrossingPoint(lastrow,lasthit2)) return kFALSE;
+
+ Float_t padpitchlow = AliL3Transform::GetPadPitchWidth(AliL3Transform::GetPatch(firstrow));
+ Float_t padpitchup = AliL3Transform::GetPadPitchWidth(AliL3Transform::GetPatch(lastrow));
+ // check the distance between tracks at the edges
+ // cout<<"Check "<<firsthit1[1]<<" "<<firsthit2[1]<<" "<<padpitchlow<<" "<<lasthit1[1]<<" "<<lasthit2[1]<<" "<<padpitchup<<" "<<xtrack1<<" "<<ytrack1<<" "<<xtrack2<<" "<<ytrack2<<endl;
+ if((fabs(firsthit1[1]-firsthit2[1])/padpitchlow + fabs(lasthit1[1]-lasthit2[1])/padpitchup) < distance) {
+ if(maxima1->fSizeX*maxima1->fSizeY > maxima2->fSizeX*maxima2->fSizeY)
+ maxima2->fWeight = -fabs(maxima2->fWeight);
+ if(maxima1->fSizeX*maxima1->fSizeY < maxima2->fSizeX*maxima2->fSizeY)
+ maxima1->fWeight = -fabs(maxima1->fWeight);
+ if(maxima1->fSizeX*maxima1->fSizeY == maxima2->fSizeX*maxima2->fSizeY) {
+ if(maxima1->fStartX > maxima2->fStartX)
+ maxima1->fStartX = maxima2->fStartX;
+ if(maxima1->fStartY > maxima2->fStartY)
+ maxima1->fStartY = maxima2->fStartY;
+ if(maxima1->fEndX < maxima2->fEndX)
+ maxima1->fEndX = maxima2->fEndX;
+ if(maxima1->fEndY < maxima2->fEndY)
+ maxima1->fEndY = maxima2->fEndY;
+ maxima1->fX = ((Float_t)maxima1->fStartX + (Float_t)maxima1->fEndX)/2.0;
+ maxima1->fY = ((Float_t)maxima1->fStartY + (Float_t)maxima1->fEndY)/2.0;
+ maxima1->fSizeX = (maxima1->fEndX - maxima1->fStartX + 1);
+ maxima1->fSizeY = (maxima1->fEndY - maxima1->fStartY + 1);
+ maxima2->fWeight = -fabs(maxima2->fWeight);
+ }
+ return kTRUE;
+ }
+ return kFALSE;
+}
git://git.uio.no / u / mrichter / AliRoot.git / blobdiff