-//$Id$
+// @(#) $Id$
// Author: Anders Vestbo <mailto:vestbo@fi.uib.no>
-//*-- Copyright © ASV
-
-#include "AliL3MemHandler.h"
-#include "AliL3Logging.h"
-#include "AliL3HoughTransformer.h"
-#include "AliL3Defs.h"
-#include "AliL3Transform.h"
-#include "AliL3DigitData.h"
-#include "AliL3Histogram.h"
+//*-- Copyright © ALICE HLT Group
+/** \class AliL3HoughTransformer
+<pre>
//_____________________________________________________________
// AliL3HoughTransformer
//
// Hough transformation class
//
+</pre>
+*/
+
+#include "AliL3StandardIncludes.h"
+
+#include "AliL3Logging.h"
+#include "AliL3HoughTransformer.h"
+#include "AliL3MemHandler.h"
+#include "AliL3Transform.h"
+#include "AliL3DigitData.h"
+#include "AliL3HistogramAdaptive.h"
+
+#if __GNUC__ == 3
+using namespace std;
+#endif
ClassImp(AliL3HoughTransformer)
{
//Default constructor
fParamSpace = 0;
+ fDoMC = kFALSE;;
+ fEtaOverlap=kFALSE;
+#ifdef do_mc
+ fTrackID = 0;
+#endif
}
-AliL3HoughTransformer::AliL3HoughTransformer(Int_t slice,Int_t patch,Int_t n_eta_segments) : AliL3HoughBaseTransformer(slice,patch,n_eta_segments)
+AliL3HoughTransformer::AliL3HoughTransformer(Int_t slice,Int_t patch,Int_t netasegments,Bool_t DoEtaOverlap,Bool_t /*DoMC*/) : AliL3HoughBaseTransformer(slice,patch,netasegments)
{
//Normal constructor
fParamSpace = 0;
+ fDoMC = kFALSE;
+ fEtaOverlap = DoEtaOverlap;
+ fDoMC=kFALSE;
+#ifdef do_mc
+ fTrackID = 0;
+#endif
}
AliL3HoughTransformer::~AliL3HoughTransformer()
{
+ // Dtor
DeleteHistograms();
+#ifdef do_mc
+ if(fTrackID)
+ {
+ for(Int_t i=0; i<GetNEtaSegments(); i++)
+ {
+ if(!fTrackID[i]) continue;
+ delete fTrackID[i];
+ }
+ delete [] fTrackID;
+ }
+#endif
}
void AliL3HoughTransformer::DeleteHistograms()
{
+ // Clean up
if(!fParamSpace)
return;
for(Int_t i=0; i<GetNEtaSegments(); i++)
delete fParamSpace[i];
}
delete [] fParamSpace;
+ fParamSpace = 0;
}
-void AliL3HoughTransformer::CreateHistograms(Int_t nxbin,Double_t pt_min,
- Int_t nybin,Double_t phimin,Double_t phimax)
+void AliL3HoughTransformer::CreateHistograms(Float_t ptmin,Float_t ptmax,Float_t ptres,
+ Int_t nybin,Float_t psi)
+{
+ //Create histograms.
+ //_Only_ to be used in case of the adaptive histograms!
+ //phimax is given in radians!!
+
+ if(ptmin > ptmax)
+ {
+ cerr<<"AliL3HoughTransformer::CreateHistograms: Error in ptrange "<<ptmin<<" "<<ptmax<<endl;
+ return;
+ }
+ if(psi < 0)
+ {
+ cerr<<"AliL3HoughTransformer::CreateHistograms: Wrong psi-angle "<<psi<<endl;
+ return;
+ }
+
+ fParamSpace = new AliL3Histogram*[GetNEtaSegments()];
+ Char_t histname[256];
+ Int_t i;
+ for(i=0; i<GetNEtaSegments(); i++)
+ {
+ sprintf(histname,"paramspace_%d",i);
+ fParamSpace[i] = new AliL3HistogramAdaptive(histname,ptmin,ptmax,ptres,nybin,-psi,psi);
+ }
+}
+
+void AliL3HoughTransformer::CreateHistograms(Int_t nxbin,Float_t ptmin,
+ Int_t nybin,Float_t phimin,Float_t phimax)
{
//Create the histograms (parameter space).
//These are 2D histograms, span by kappa (curvature of track) and phi0 (emission angle with x-axis).
//The arguments give the range and binning;
//nxbin = #bins in kappa
//nybin = #bins in phi0
- //pt_min = mimium Pt of track (corresponding to maximum kappa)
- //phi_min = mimimum phi0 (degrees)
- //phi_max = maximum phi0 (degrees)
+ //ptmin = mimium Pt of track (corresponding to maximum kappa)
+ //phimin = mimimum phi0
+ //phimax = maximum phi0
- Double_t bfact = 0.0029980;
- Double_t bfield = 0.2;
- Double_t x = bfact*bfield/pt_min;
- CreateHistograms(nxbin,-1.*x,x,nybin,phimin*ToRad,phimax*ToRad);
+ Double_t x = AliL3Transform::GetBFact()*AliL3Transform::GetBField()/ptmin;
+ //Double_t torad = AliL3Transform::Pi()/180;
+
+ CreateHistograms(nxbin,-1.*x,x,nybin,phimin/**torad*/,phimax/**torad*/);
}
-void AliL3HoughTransformer::CreateHistograms(Int_t nxbin,Double_t xmin,Double_t xmax,
- Int_t nybin,Double_t ymin,Double_t ymax)
+void AliL3HoughTransformer::CreateHistograms(Int_t nxbin,Float_t xmin,Float_t xmax,
+ Int_t nybin,Float_t ymin,Float_t ymax)
{
+ //Create the histograms (parameter space).
+ //nxbin = #bins in X
+ //nybin = #bins in Y
+ //xmin xmax ymin ymax = histogram limits in X and Y
fParamSpace = new AliL3Histogram*[GetNEtaSegments()];
for(Int_t i=0; i<GetNEtaSegments(); i++)
{
sprintf(histname,"paramspace_%d",i);
+ //fParamSpace[i] = new AliL3HistogramAdaptive(histname,0.5,1.5,0.05,nybin,ymin,ymax);
fParamSpace[i] = new AliL3Histogram(histname,"",nxbin,xmin,xmax,nybin,ymin,ymax);
}
+
+#ifdef do_mc
+ if(fDoMC)
+ {
+ AliL3Histogram *hist = fParamSpace[0];
+ Int_t ncells = (hist->GetNbinsX()+2)*(hist->GetNbinsY()+2);
+ cout<<"Transformer: Allocating "<<GetNEtaSegments()*ncells*sizeof(AliL3TrackIndex)<<" bytes to fTrackID"<<endl;
+ fTrackID = new AliL3TrackIndex*[GetNEtaSegments()];
+ for(Int_t i=0; i<GetNEtaSegments(); i++)
+ fTrackID[i] = new AliL3TrackIndex[ncells];
+ }
+#endif
}
void AliL3HoughTransformer::Reset()
for(Int_t i=0; i<GetNEtaSegments(); i++)
fParamSpace[i]->Reset();
+
+#ifdef do_mc
+ if(fDoMC)
+ {
+ AliL3Histogram *hist = fParamSpace[0];
+ Int_t ncells = (hist->GetNbinsX()+2)*(hist->GetNbinsY()+2);
+ for(Int_t i=0; i<GetNEtaSegments(); i++)
+ memset(fTrackID[i],0,ncells*sizeof(AliL3TrackIndex));
+ }
+#endif
}
-
-Int_t AliL3HoughTransformer::GetEtaIndex(Double_t eta)
+Int_t AliL3HoughTransformer::GetEtaIndex(Double_t eta) const
{
//Return the histogram index of the corresponding eta.
return (Int_t)index;
}
+void AliL3HoughTransformer::GetEtaIndexes(Double_t eta,Int_t *indexes) const
+{
+ //Return histogram indexes in case of overlapping etaslices.
+
+ Double_t etaslice = (GetEtaMax() - GetEtaMin())/GetNEtaSegments();
+ Int_t index = (Int_t)((eta-GetEtaMin())/etaslice);
+ if(index%2 == 0)
+ {
+ indexes[0] = index;
+ indexes[1] = index - 1;
+ }
+ else
+ {
+ indexes[0] = index - 1;
+ indexes[1] = index;
+ }
+}
+
+AliL3Histogram *AliL3HoughTransformer::GetHistogram(Int_t etaindex)
+{
+ // Return a pointer to the histogram which contains etaindex eta slice
+ if(!fParamSpace || etaindex >= GetNEtaSegments() || etaindex < 0)
+ return 0;
+ if(!fParamSpace[etaindex])
+ return 0;
+ return fParamSpace[etaindex];
+}
+
+Double_t AliL3HoughTransformer::GetEta(Int_t etaindex,Int_t /*slice*/) const
+{
+ // Return eta calculated in the middle of the eta slice
+ Double_t etaslice = (GetEtaMax()-GetEtaMin())/GetNEtaSegments();
+ Double_t eta=0;
+ if(fEtaOverlap)
+ {
+ Int_t index = etaindex + 1;
+ eta=(Double_t)((index)*etaslice);
+ }
+ else
+ eta=(Double_t)((etaindex+0.5)*etaslice);
+ return eta;
+}
+
void AliL3HoughTransformer::TransformCircle()
{
//Transform the input data with a circle HT.
//The function loops over all the data, and transforms each pixel with the equations:
//
- //kappa = 2/R*sin(phi - phi0)
+ //kappa = 2/r*sin(phi - phi0)
//
- //where R = sqrt(x*x +y*y), and phi = arctan(y/x)
+ //where r = sqrt(x*x +y*y), and phi = arctan(y/x)
//
//Each pixel then transforms into a curve in the (kappa,phi0)-space. In order to find
- //which histogram in which the pixel should be transformed, the eta-value is calcluated
+ //which histogram in which the pixel should be transformed, the eta-value is calculated
//and the proper histogram index is found by GetEtaIndex(eta).
<<"No input data "<<ENDLOG;
return;
}
- if(!fTransform)
- {
- LOG(AliL3Log::kError,"AliL3HoughTransformer::TransformCircle","Transformer")
- <<"No AliL3Transform object"<<ENDLOG;
- return;
- }
//Loop over the padrows:
- for(Int_t i=NRows[GetPatch()][0]; i<=NRows[GetPatch()][1]; i++)
+ for(Int_t i=AliL3Transform::GetFirstRow(GetPatch()); i<=AliL3Transform::GetLastRow(GetPatch()); i++)
{
//Get the data on this padrow:
AliL3DigitData *digPt = tempPt->fDigitData;
if(i != (Int_t)tempPt->fRow)
{
- printf("AliL3HoughTransform::TransformCircle : Mismatching padrow numbering\n");
+ cerr<<"AliL3HoughTransform::TransformCircle : Mismatching padrow numbering "<<i<<" "<<(Int_t)tempPt->fRow<<endl;
continue;
}
UShort_t charge = digPt[j].fCharge;
UChar_t pad = digPt[j].fPad;
UShort_t time = digPt[j].fTime;
- if(charge <= GetThreshold())
+ if((Int_t)charge <= GetLowerThreshold())
continue;
+
+ if((Int_t)charge > GetUpperThreshold())
+ charge = GetUpperThreshold();
+
Int_t sector,row;
Float_t xyz[3];
-
+
//Transform data to local cartesian coordinates:
- fTransform->Slice2Sector(GetSlice(),i,sector,row);
- fTransform->Raw2Local(xyz,sector,row,(Int_t)pad,(Int_t)time);
-
+ AliL3Transform::Slice2Sector(GetSlice(),i,sector,row);
+ AliL3Transform::Raw2Local(xyz,sector,row,(Int_t)pad,(Int_t)time);
+
//Calculate the eta:
- Double_t eta = fTransform->GetEta(xyz);
+ Double_t eta = AliL3Transform::GetEta(xyz);
//Get the corresponding index, which determines which histogram to fill:
- Int_t eta_index = GetEtaIndex(eta);
- if(eta_index < 0 || eta_index >= GetNEtaSegments())
+ Int_t etaindex = GetEtaIndex(eta);
+
+ if(etaindex < 0 || etaindex >= GetNEtaSegments())
continue;
//Get the correct histogrampointer:
- AliL3Histogram *hist = fParamSpace[eta_index];
+ AliL3Histogram *hist = fParamSpace[etaindex];
if(!hist)
{
- printf("AliL3HoughTransformer::TransformCircle : Error getting histogram in index %d\n",eta_index);
+ cerr<<"AliL3HoughTransformer::TransformCircle : Error getting histogram in index "<<etaindex<<endl;
continue;
}
-
+
//Do the transformation:
- Float_t R = sqrt(xyz[0]*xyz[0] + xyz[1]*xyz[1]);
- Float_t phi = fTransform->GetPhi(xyz);
+ Float_t r = sqrt(xyz[0]*xyz[0] + xyz[1]*xyz[1]);
+ Float_t phi = AliL3Transform::GetPhi(xyz);
+
//Fill the histogram along the phirange
for(Int_t b=hist->GetFirstYbin(); b<=hist->GetLastYbin(); b++)
{
Float_t phi0 = hist->GetBinCenterY(b);
- Float_t kappa = 2*sin(phi - phi0)/R;
+ Float_t kappa = 2*sin(phi - phi0)/r;
+ //hist->Fill(kappa,phi0,(int)rint(log((Float_t)charge)));
hist->Fill(kappa,phi0,charge);
+ //hist->Fill(kappa,phi0,1);
+#ifdef do_mc
+ if(fDoMC)
+ {
+ Int_t bin = hist->FindBin(kappa,phi0);
+ for(Int_t t=0; t<3; t++)
+ {
+ Int_t label = digPt[j].fTrackID[t];
+ if(label < 0) break;
+ UInt_t c;
+ for(c=0; c<MaxTrack; c++)
+ if(fTrackID[etaindex][bin].fLabel[c] == label || fTrackID[etaindex][bin].fNHits[c] == 0)
+ break;
+ if(c == MaxTrack-1) cerr<<"AliL3HoughTransformer::TransformCircle : Array reached maximum!! "<<c<<endl;
+ fTrackID[etaindex][bin].fLabel[c] = label;
+ fTrackID[etaindex][bin].fNHits[c]++;
+ }
+ }
+#endif
}
}
}
}
-void AliL3HoughTransformer::TransformCircleC(Int_t row_range)
+struct AliL3Digit {
+ Int_t fRow; // Digit padrow
+ Double_t fR; // Digit radius in local coordinate system
+ Double_t fPhi; // Digit Phi angle in local coordinate system
+ Int_t fCharge; // Digit charge
+ AliL3Digit *fNext; // Next digit
+};
+
+struct AliL3EtaContainer {
+ AliL3Digit *fFirst; //First digit
+ AliL3Digit *fLast; //Last digit
+};
+
+void AliL3HoughTransformer::TransformCircleC(Int_t *rowrange,Int_t every)
{
//Circle transform, using combinations of every 2 points lying
//on different padrows and within the same etaslice.
if(!tempPt)
LOG(AliL3Log::kError,"AliL3HoughTransformer::TransformCircleC","Data")
<<"No input data "<<ENDLOG;
-
- if(!fTransform)
+
+ Int_t minrow = AliL3Transform::GetFirstRow(GetPatch());
+ Int_t maxrow = AliL3Transform::GetLastRow(GetPatch());
+ if(rowrange)
{
- LOG(AliL3Log::kError,"AliL3HoughTransformer::TransformCircleC","Transformer")
- <<"No AliL3Transform object"<<ENDLOG;
- return;
+ minrow = rowrange[0];
+ maxrow = rowrange[1];
+ if(minrow < AliL3Transform::GetFirstRow(GetPatch()) || minrow >= AliL3Transform::GetLastRow(GetPatch()))
+ minrow = AliL3Transform::GetFirstRow(GetPatch());
+ if(maxrow < AliL3Transform::GetFirstRow(GetPatch()) || maxrow >= AliL3Transform::GetLastRow(GetPatch()))
+ maxrow = AliL3Transform::GetLastRow(GetPatch());
+ if(minrow > maxrow || maxrow==minrow)
+ {
+ cerr<<"AliL3HoughTransformer::TransformCircleC : Bad row range "<<minrow<<" "<<maxrow<<endl;
+ return;
+ }
}
-
+ else
+ {
+ minrow = AliL3Transform::GetFirstRow(GetPatch());
+ maxrow = AliL3Transform::GetLastRow(GetPatch());
+ }
+
Int_t counter=0;
- for(Int_t i=NRows[GetPatch()][0]; i<=NRows[GetPatch()][1]; i++)
+ for(Int_t i=AliL3Transform::GetFirstRow(GetPatch()); i<=AliL3Transform::GetLastRow(GetPatch()); i++)
{
counter += tempPt->fNDigit;
AliL3MemHandler::UpdateRowPointer(tempPt);
}
- struct Digit {
- Int_t row;
- Double_t r;
- Double_t phi;
- Int_t eta_index;
- Int_t charge;
- };
-
- Digit *digits = new Digit[counter];
- cout<<"Allocating "<<counter*sizeof(Digit)<<" bytes to digitsarray"<<endl;
+ Int_t bound = (GetNEtaSegments()+1)*(AliL3Transform::GetNRows(GetPatch())+1);
+ AliL3EtaContainer *etaPt = new AliL3EtaContainer[bound];
+ memset(etaPt,0,bound*sizeof(AliL3EtaContainer));
- Int_t total_digits=counter;
- Int_t sector,row,tot_charge,pad,time,charge;
- Double_t r1,r2,phi1,phi2,eta,kappa,phi_0;
+ AliL3Digit *digits = new AliL3Digit[counter];
+ cout<<"Allocating "<<counter*sizeof(AliL3Digit)<<" bytes to digitsarray"<<endl;
+ memset(digits,0,counter*sizeof(AliL3Digit));
+
+ Int_t sector,row,totcharge,pad,time,charge;
+ Double_t r1,r2,phi1,phi2,eta,kappa,phi0;
Float_t xyz[3];
counter=0;
tempPt = GetDataPointer();
- for(Int_t i=NRows[GetPatch()][0]; i<=NRows[GetPatch()][1]; i++)
+ cout<<"Calculating digits in patch "<<GetPatch()<<endl;
+ for(Int_t i=AliL3Transform::GetFirstRow(GetPatch()); i<=AliL3Transform::GetLastRow(GetPatch()); i++)
{
AliL3DigitData *digPt = tempPt->fDigitData;
for(UInt_t di=0; di<tempPt->fNDigit; di++)
charge = digPt[di].fCharge;
pad = digPt[di].fPad;
time = digPt[di].fTime;
- fTransform->Slice2Sector(GetSlice(),i,sector,row);
- fTransform->Raw2Local(xyz,sector,row,(Int_t)pad,(Int_t)time);
- eta = fTransform->GetEta(xyz);
- digits[counter].row = i;
- digits[counter].r = sqrt(xyz[0]*xyz[0] + xyz[1]*xyz[1]);
- digits[counter].phi = atan2(xyz[1],xyz[0]);
- digits[counter].eta_index = GetEtaIndex(eta);
- digits[counter].charge = charge;
+ AliL3Transform::Slice2Sector(GetSlice(),i,sector,row);
+ AliL3Transform::Raw2Local(xyz,sector,row,(Int_t)pad,(Int_t)time);
+ eta = AliL3Transform::GetEta(xyz);
+
+ digits[counter].fRow = i;
+ digits[counter].fR = sqrt(xyz[0]*xyz[0] + xyz[1]*xyz[1]);
+ digits[counter].fPhi = atan2(xyz[1],xyz[0]);
+ digits[counter].fCharge = charge;
+
+ if(!fEtaOverlap)
+ {
+ Int_t etaindex = GetEtaIndex(eta);
+
+ Int_t index = (GetNEtaSegments()+1)*(i-AliL3Transform::GetFirstRow(GetPatch())) + etaindex;
+
+ if(index > 0 && index < bound)
+ {
+ if(etaPt[index].fFirst == 0)
+ etaPt[index].fFirst = &digits[counter];
+ else
+ (etaPt[index].fLast)->fNext = &digits[counter];
+ etaPt[index].fLast = &digits[counter];
+ }
+ }
+ else
+ {
+ Int_t etaindex[2];
+ GetEtaIndexes(eta,etaindex);
+ Int_t index[2];
+ index[0] = (GetNEtaSegments()+1)*(i-AliL3Transform::GetFirstRow(GetPatch())) + etaindex[0];
+ index[1] = (GetNEtaSegments()+1)*(i-AliL3Transform::GetFirstRow(GetPatch())) + etaindex[1];
+ if(index[0] == index[1])
+ {
+ cerr<<"Same etaindexes "<<index[0]<<" "<<index[1]<<endl;
+ exit(5);
+ }
+
+ Int_t ind = index[0];
+ if(ind > 0 && ind < bound)
+ {
+ if(etaPt[ind].fFirst == 0)
+ etaPt[ind].fFirst = &digits[counter];
+ else
+ (etaPt[ind].fLast)->fNext = &digits[counter];
+ etaPt[ind].fLast = &digits[counter];
+ }
+
+ ind = index[1];
+ if(ind > 0 && ind < bound)
+ {
+ if(etaPt[ind].fFirst == 0)
+ etaPt[ind].fFirst = &digits[counter];
+ else
+ (etaPt[ind].fLast)->fNext = &digits[counter];
+ etaPt[ind].fLast = &digits[counter];
+ }
+ }
+
counter++;
}
AliL3MemHandler::UpdateRowPointer(tempPt);
}
- for(Int_t i=0; i<total_digits; i++)
+ cout<<"Doing the combinatorics"<<endl;
+
+ AliL3Digit *dPt1,*dPt2;
+
+ for(Int_t e=0; e<GetNEtaSegments(); e++)
{
- if(digits[i].eta_index < 0 || digits[i].eta_index >= GetNEtaSegments()) continue;
- Int_t ind = digits[i].eta_index;
-
- for(Int_t j=i+1; j<total_digits; j++)
+ for(Int_t i=minrow; i<=maxrow; i+=every)
{
- if(digits[i].row == digits[j].row) continue;
- if(digits[i].eta_index != digits[j].eta_index) continue;
- if(digits[i].row + row_range < digits[j].row) break;
+ Int_t index1 = (GetNEtaSegments()+1)*(i-AliL3Transform::GetFirstRow(GetPatch())) + e;
- //Get the correct histogrampointer:
- AliL3Histogram *hist = fParamSpace[ind];
- if(!hist)
+ for(dPt1 = (AliL3Digit*)etaPt[index1].fFirst; dPt1 != 0; dPt1 = (AliL3Digit*)dPt1->fNext)
{
- printf("AliL3HoughTransformer::TransformCircleC() : No histogram at index %d\n",ind);
- continue;
+ for(Int_t j=i+every; j<=maxrow; j+=every)
+ {
+ Int_t index2 = (GetNEtaSegments()+1)*(j-AliL3Transform::GetFirstRow(GetPatch())) + e;
+
+ for(dPt2 = (AliL3Digit*)etaPt[index2].fFirst; dPt2 != 0; dPt2 = (AliL3Digit*)dPt2->fNext)
+ {
+ if(dPt1->fRow == dPt2->fRow)
+ {
+ cerr<<"same row; indexes "<<index1<<" "<<index2<<endl;
+ exit(5);
+ }
+
+ //Get the correct histogrampointer:
+ AliL3Histogram *hist = fParamSpace[e];
+ if(!hist)
+ {
+ printf("AliL3HoughTransformer::TransformCircleC() : No histogram at index %d\n",i);
+ continue;
+ }
+
+ //Do the transform:
+ r1 = dPt1->fR;
+ phi1 = dPt1->fPhi;
+ r2 = dPt2->fR;
+ phi2 = dPt2->fPhi;
+ phi0 = atan( (r2*sin(phi1)-r1*sin(phi2))/(r2*cos(phi1)-r1*cos(phi2)) );
+ kappa = 2*sin(phi2-phi0)/r2;
+ totcharge = dPt1->fCharge + dPt2->fCharge;
+ hist->Fill(kappa,phi0,totcharge);
+
+ }
+ }
}
-
- r1 = digits[i].r;
- phi1 = digits[i].phi;
- r2 = digits[j].r;
- phi2 = digits[j].phi;
- phi_0 = atan( (r2*sin(phi1)-r1*sin(phi2))/(r2*cos(phi1)-r1*cos(phi2)) );
- kappa = 2*sin(phi2-phi_0)/r2;
- tot_charge = digits[i].charge + digits[j].charge;
- hist->Fill(kappa,phi_0,tot_charge);
}
}
+
+ cout<<"done"<<endl;
+ delete [] etaPt;
delete [] digits;
+
}
-void AliL3HoughTransformer::TransformLine()
+void AliL3HoughTransformer::TransformLine(Int_t *rowrange,Float_t *phirange)
{
//Do a line transform on the data.
-
+
AliL3DigitRowData *tempPt = GetDataPointer();
if(!tempPt)
{
<<"No input data "<<ENDLOG;
return;
}
- if(!fTransform)
+
+ Int_t minrow = AliL3Transform::GetFirstRow(GetPatch());
+ Int_t maxrow = AliL3Transform::GetLastRow(GetPatch());
+ if(rowrange)
{
- LOG(AliL3Log::kError,"AliL3HoughTransformer::TransformLine","Transformer")
- <<"No AliL3Transform object"<<ENDLOG;
- return;
+ minrow = rowrange[0];
+ maxrow = rowrange[1];
+ if(minrow < AliL3Transform::GetFirstRow(GetPatch()) || minrow >= AliL3Transform::GetLastRow(GetPatch()))
+ minrow = AliL3Transform::GetFirstRow(GetPatch());
+ if(maxrow < AliL3Transform::GetFirstRow(GetPatch()) || maxrow >= AliL3Transform::GetLastRow(GetPatch()))
+ maxrow = AliL3Transform::GetLastRow(GetPatch());
+ if(minrow > maxrow || maxrow==minrow)
+ {
+ cerr<<"AliL3HoughTransformer::TransformCircleC : Bad row range "<<minrow<<" "<<maxrow<<endl;
+ return;
+ }
}
-
- for(Int_t i=NRows[GetPatch()][0]; i<=NRows[GetPatch()][1]; i++)
+
+ for(Int_t i=minrow; i<=maxrow; i++)
{
AliL3DigitData *digPt = tempPt->fDigitData;
if(i != (Int_t)tempPt->fRow)
UShort_t charge = digPt[j].fCharge;
UChar_t pad = digPt[j].fPad;
UShort_t time = digPt[j].fTime;
- if(charge < GetThreshold())
+ if(charge < GetLowerThreshold())
continue;
Int_t sector,row;
Float_t xyz[3];
- fTransform->Slice2Sector(GetSlice(),i,sector,row);
- fTransform->Raw2Local(xyz,sector,row,(Int_t)pad,(Int_t)time);
- Float_t eta = fTransform->GetEta(xyz);
- Int_t eta_index = GetEtaIndex(eta);//(Int_t)(eta/etaslice);
- if(eta_index < 0 || eta_index >= GetNEtaSegments())
+ AliL3Transform::Slice2Sector(GetSlice(),i,sector,row);
+ AliL3Transform::Raw2Local(xyz,sector,row,(Int_t)pad,(Int_t)time);
+
+ if(phirange)
+ {
+ Float_t phi = AliL3Transform::GetPhi(xyz);
+ if(phi < phirange[0] || phi > phirange[1])
+ continue;
+ }
+ Float_t eta = AliL3Transform::GetEta(xyz);
+ Int_t etaindex = GetEtaIndex(eta);//(Int_t)(eta/etaslice);
+ if(etaindex < 0 || etaindex >= GetNEtaSegments())
continue;
+ xyz[0] = xyz[0] - AliL3Transform::Row2X(minrow);
+
//Get the correct histogram:
- AliL3Histogram *hist = fParamSpace[eta_index];
+ AliL3Histogram *hist = fParamSpace[etaindex];
if(!hist)
{
- printf("AliL3HoughTransformer::TransformLine : Error getting histogram in index %d\n",eta_index);
+ printf("AliL3HoughTransformer::TransformLine : Error getting histogram in index %d\n",etaindex);
continue;
}
for(Int_t xbin=hist->GetFirstXbin(); xbin<hist->GetLastXbin(); xbin++)
}
}
+
+struct AliL3LDigit {
+ Int_t fRow; // Digit rowpad
+ Int_t fCharge; // Digit charge
+ Float_t fY; // Y position of the digit in the local coor system
+ AliL3LDigit *fNext; // Next digit
+};
+struct AliL3LEtaContainer {
+ AliL3LDigit *fFirst; //First digit
+ AliL3LDigit *fLast; //Last digit
+};
+void AliL3HoughTransformer::TransformLineC(Int_t *rowrange,Float_t *phirange)
+{
+ //Circle transform ??
+ AliL3DigitRowData *tempPt = GetDataPointer();
+ if(!tempPt)
+ LOG(AliL3Log::kError,"AliL3HoughTransformer::TransformCircleC","Data")
+ <<"No input data "<<ENDLOG;
+
+
+ Int_t counter=0;
+ for(Int_t i=AliL3Transform::GetFirstRow(GetPatch()); i<=AliL3Transform::GetLastRow(GetPatch()); i++)
+ {
+ counter += tempPt->fNDigit;
+ AliL3MemHandler::UpdateRowPointer(tempPt);
+ }
+
+ Int_t bound = (GetNEtaSegments()+1)*(AliL3Transform::GetNRows(GetPatch())+1);
+ AliL3LEtaContainer *etaPt = new AliL3LEtaContainer[bound];
+ memset(etaPt,0,bound*sizeof(AliL3LEtaContainer));
+
+ AliL3LDigit *digits = new AliL3LDigit[counter];
+ cout<<"Allocating "<<counter*sizeof(AliL3LDigit)<<" bytes to digitsarray"<<endl;
+ memset(digits,0,counter*sizeof(AliL3LDigit));
+
+ Int_t sector,row;
+ Float_t xyz[3];
+
+ counter=0;
+ tempPt = GetDataPointer();
+
+ cout<<"Calculating digits in patch "<<GetPatch()<<endl;
+ for(Int_t i=AliL3Transform::GetFirstRow(GetPatch()); i<=AliL3Transform::GetLastRow(GetPatch()); i++)
+ {
+ AliL3DigitData *digPt = tempPt->fDigitData;
+ for(UInt_t di=0; di<tempPt->fNDigit; di++)
+ {
+ Int_t charge = digPt[di].fCharge;
+ Int_t pad = digPt[di].fPad;
+ Int_t time = digPt[di].fTime;
+ AliL3Transform::Slice2Sector(GetSlice(),i,sector,row);
+ AliL3Transform::Raw2Local(xyz,sector,row,(Int_t)pad,(Int_t)time);
+ Double_t eta = AliL3Transform::GetEta(xyz);
+
+ Float_t phi = atan2(xyz[1],xyz[0]);
+ if(phi < phirange[0] || phi > phirange[1]) continue;
+
+ digits[counter].fRow = i;
+ digits[counter].fY = xyz[1];
+ digits[counter].fCharge = charge;
+
+ Int_t etaindex = GetEtaIndex(eta);
+ Int_t index = (GetNEtaSegments()+1)*(i-AliL3Transform::GetFirstRow(GetPatch())) + etaindex;
+
+ if(index > 0 && index < bound)
+ {
+ if(etaPt[index].fFirst == 0)
+ etaPt[index].fFirst = &digits[counter];
+ else
+ (etaPt[index].fLast)->fNext = &digits[counter];
+ etaPt[index].fLast = &digits[counter];
+ }
+ counter++;
+ }
+ AliL3MemHandler::UpdateRowPointer(tempPt);
+ }
+
+ cout<<"Doing the combinatorics"<<endl;
+
+ AliL3LDigit *dPt1,*dPt2;
+
+ for(Int_t e=0; e<GetNEtaSegments(); e++)
+ {
+ for(Int_t i=rowrange[0]; i<=rowrange[1]; i++)
+ {
+ Int_t index1 = (GetNEtaSegments()+1)*(i-AliL3Transform::GetFirstRow(GetPatch())) + e;
+
+ for(dPt1 = (AliL3LDigit*)etaPt[index1].fFirst; dPt1 != 0; dPt1 = (AliL3LDigit*)dPt1->fNext)
+ {
+ for(Int_t j=i+1; j<=rowrange[1]; j++)
+ {
+ Int_t index2 = (GetNEtaSegments()+1)*(j-AliL3Transform::GetFirstRow(GetPatch())) + e;
+
+ for(dPt2 = (AliL3LDigit*)etaPt[index2].fFirst; dPt2 != 0; dPt2 = (AliL3LDigit*)dPt2->fNext)
+ {
+ if(dPt1->fRow == dPt2->fRow)
+ {
+ cerr<<"same row; indexes "<<index1<<" "<<index2<<endl;
+ exit(5);
+ }
+
+ //Get the correct histogrampointer:
+ AliL3Histogram *hist = fParamSpace[e];
+ if(!hist)
+ {
+ printf("AliL3HoughTransformer::TransformCircleC() : No histogram at index %d\n",i);
+ continue;
+ }
+
+ //Do the transform:
+ float x1 = AliL3Transform::Row2X(dPt1->fRow) - AliL3Transform::Row2X(rowrange[0]);
+ float x2 = AliL3Transform::Row2X(dPt2->fRow) - AliL3Transform::Row2X(rowrange[0]);
+ float y1 = dPt1->fY;
+ float y2 = dPt2->fY;
+ float theta = atan2(x2-x1,y1-y2);
+ float rho = x1*cos(theta)+y1*sin(theta);
+ hist->Fill(theta,rho,1);//dPt1->charge+dPt2->charge);
+ }
+ }
+ }
+ }
+ }
+
+ cout<<"done"<<endl;
+ delete [] etaPt;
+ delete [] digits;
+}
+
+#ifdef do_mc
+Int_t AliL3HoughTransformer::GetTrackID(Int_t etaindex,Double_t kappa,Double_t psi) const
+{
+ // Returns the MC label for a given peak found in the Hough space
+ if(!fDoMC)
+ {
+ cerr<<"AliL3HoughTransformer::GetTrackID : Flag switched off"<<endl;
+ return -1;
+ }
+
+ if(etaindex < 0 || etaindex > GetNEtaSegments())
+ {
+ cerr<<"AliL3HoughTransformer::GetTrackID : Wrong etaindex "<<etaindex<<endl;
+ return -1;
+ }
+ AliL3Histogram *hist = fParamSpace[etaindex];
+ Int_t bin = hist->FindBin(kappa,psi);
+ Int_t label=-1;
+ Int_t max=0;
+ for(UInt_t i=0; i<MaxTrack; i++)
+ {
+ Int_t nhits=fTrackID[etaindex][bin].fNHits[i];
+ if(nhits == 0) break;
+ if(nhits > max)
+ {
+ max = nhits;
+ label = fTrackID[etaindex][bin].fLabel[i];
+ }
+ }
+ //nhits = max;
+ return label;
+#else
+ Int_t AliL3HoughTransformer::GetTrackID(Int_t /*etaindex*/,Double_t /*kappa*/,Double_t /*psi*/) const
+{
+ // Returns the MC label for a given peak found in the Hough space
+ if(!fDoMC)
+ {
+ cerr<<"AliL3HoughTransformer::GetTrackID : Flag switched off"<<endl;
+ return -1;
+ }
+
+ cout<<"AliL3HoughTransformer::GetTrackID : Compile with do_mc flag!"<<endl;
+ return -1;
+#endif
+}
+
git://git.uio.no / u / mrichter / AliRoot.git / blobdiff