X-Git-Url: http://git.uio.no/git/?a=blobdiff_plain;ds=sidebyside;f=TRD%2FAliTRDclusterizerV1.cxx;h=5f73b867b638191dfed90a2b1c38d0071b66c975;hb=d49e463beff82523b876df0e69879d7b316c72dd;hp=520c521eb19ea48bda138dad26cc7f8e9a89aac9;hpb=94de381804dbc3c6cc154b06fc58284dc8266705;p=u%2Fmrichter%2FAliRoot.git diff --git a/TRD/AliTRDclusterizerV1.cxx b/TRD/AliTRDclusterizerV1.cxx index 520c521eb19..5f73b867b63 100644 --- a/TRD/AliTRDclusterizerV1.cxx +++ b/TRD/AliTRDclusterizerV1.cxx @@ -1,3 +1,4 @@ + /************************************************************************** * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. * * * @@ -13,83 +14,73 @@ * provided "as is" without express or implied warranty. * **************************************************************************/ -/* -$Log$ -Revision 1.7 2000/06/27 13:08:50 cblume -Changed to Copy(TObject &A) to appease the HP-compiler - -Revision 1.6 2000/06/09 11:10:07 cblume -Compiler warnings and coding conventions, next round - -Revision 1.5 2000/06/08 18:32:58 cblume -Make code compliant to coding conventions - -Revision 1.4 2000/06/07 16:27:01 cblume -Try to remove compiler warnings on Sun and HP - -Revision 1.3 2000/05/08 16:17:27 cblume -Merge TRD-develop - -Revision 1.1.4.1 2000/05/08 15:09:01 cblume -Introduce AliTRDdigitsManager - -Revision 1.1 2000/02/28 18:58:54 cblume -Add new TRD classes - -*/ +/* $Id$ */ /////////////////////////////////////////////////////////////////////////////// // // -// TRD cluster finder for the slow simulator. +// TRD cluster finder // // // /////////////////////////////////////////////////////////////////////////////// #include #include +#include +#include + +#include "AliRun.h" +#include "AliRunLoader.h" +#include "AliLoader.h" +#include "AliRawReader.h" +#include "AliLog.h" #include "AliTRDclusterizerV1.h" -#include "AliTRDmatrix.h" #include "AliTRDgeometry.h" -#include "AliTRDdigitizer.h" -#include "AliTRDrecPoint.h" #include "AliTRDdataArrayF.h" +#include "AliTRDdataArrayI.h" +#include "AliTRDdigitsManager.h" +#include "AliTRDpadPlane.h" +#include "AliTRDrawData.h" +#include "AliTRDcalibDB.h" +#include "AliTRDSimParam.h" +#include "AliTRDRecParam.h" +#include "AliTRDCommonParam.h" +#include "AliTRDcluster.h" ClassImp(AliTRDclusterizerV1) //_____________________________________________________________________________ -AliTRDclusterizerV1::AliTRDclusterizerV1():AliTRDclusterizer() +AliTRDclusterizerV1::AliTRDclusterizerV1() + :AliTRDclusterizer() + ,fDigitsManager(NULL) { // // AliTRDclusterizerV1 default constructor // - fDigitsManager = NULL; - } //_____________________________________________________________________________ -AliTRDclusterizerV1::AliTRDclusterizerV1(const Text_t* name, const Text_t* title) - :AliTRDclusterizer(name,title) +AliTRDclusterizerV1::AliTRDclusterizerV1(const Text_t *name, const Text_t *title) + :AliTRDclusterizer(name,title) + ,fDigitsManager(new AliTRDdigitsManager()) { // - // AliTRDclusterizerV1 default constructor + // AliTRDclusterizerV1 constructor // - fDigitsManager = new AliTRDdigitsManager(); - - Init(); + fDigitsManager->CreateArrays(); } //_____________________________________________________________________________ AliTRDclusterizerV1::AliTRDclusterizerV1(const AliTRDclusterizerV1 &c) + :AliTRDclusterizer(c) + ,fDigitsManager(NULL) { // // AliTRDclusterizerV1 copy constructor // - ((AliTRDclusterizerV1 &) c).Copy(*this); - } //_____________________________________________________________________________ @@ -101,6 +92,7 @@ AliTRDclusterizerV1::~AliTRDclusterizerV1() if (fDigitsManager) { delete fDigitsManager; + fDigitsManager = NULL; } } @@ -118,301 +110,494 @@ AliTRDclusterizerV1 &AliTRDclusterizerV1::operator=(const AliTRDclusterizerV1 &c } //_____________________________________________________________________________ -void AliTRDclusterizerV1::Copy(TObject &c) +void AliTRDclusterizerV1::Copy(TObject &c) const { // // Copy function // - ((AliTRDclusterizerV1 &) c).fClusMaxThresh = fClusMaxThresh; - ((AliTRDclusterizerV1 &) c).fClusSigThresh = fClusSigThresh; - ((AliTRDclusterizerV1 &) c).fClusMethod = fClusMethod; - ((AliTRDclusterizerV1 &) c).fDigitsManager = NULL; + ((AliTRDclusterizerV1 &) c).fDigitsManager = 0; AliTRDclusterizer::Copy(c); } //_____________________________________________________________________________ -void AliTRDclusterizerV1::Init() +Bool_t AliTRDclusterizerV1::ReadDigits() { // - // Initializes the cluster finder + // Reads the digits arrays from the input aliroot file // - // The default parameter for the clustering - fClusMaxThresh = 5.0; - fClusSigThresh = 2.0; - fClusMethod = 1; + if (!fRunLoader) { + AliError("No run loader available"); + return kFALSE; + } + + AliLoader* loader = fRunLoader->GetLoader("TRDLoader"); + if (!loader->TreeD()) { + loader->LoadDigits(); + } + + // Read in the digit arrays + return (fDigitsManager->ReadDigits(loader->TreeD())); } //_____________________________________________________________________________ -Bool_t AliTRDclusterizerV1::ReadDigits() +Bool_t AliTRDclusterizerV1::ReadDigits(AliRawReader *rawReader) { // - // Reads the digits arrays from the input aliroot file + // Reads the digits arrays from the ddl file // - if (!fInputFile) { - printf("AliTRDclusterizerV1::ReadDigits -- "); - printf("No input file open\n"); - return kFALSE; - } + AliTRDrawData raw; + fDigitsManager = raw.Raw2Digits(rawReader); - // Read in the digit arrays - return (fDigitsManager->ReadDigits()); + return kTRUE; } //_____________________________________________________________________________ -Bool_t AliTRDclusterizerV1::MakeCluster() +Bool_t AliTRDclusterizerV1::MakeClusters() { // // Generates the cluster. // - Int_t row, col, time; - - // Get the pointer to the detector class and check for version 1 - AliTRD *trd = (AliTRD*) gAlice->GetDetector("TRD"); - if (trd->IsVersion() != 1) { - printf("AliTRDclusterizerV1::MakeCluster -- "); - printf("TRD must be version 1 (slow simulator).\n"); - return kFALSE; - } + Int_t row = 0; + Int_t col = 0; + Int_t time = 0; + Int_t icham = 0; + Int_t iplan = 0; + Int_t isect = 0; + Int_t iPad = 0; + + AliTRDdataArrayI *digitsIn; + AliTRDdataArrayI *tracksIn; // Get the geometry - AliTRDgeometry *geo = trd->GetGeometry(); - - printf("AliTRDclusterizerV1::MakeCluster -- "); - printf("Start creating clusters.\n"); + AliTRDgeometry *geo = AliTRDgeometry::GetGeometry(fRunLoader); - AliTRDdataArrayI *digits; + AliTRDcalibDB *calibration = AliTRDcalibDB::Instance(); + if (!calibration) { + AliError("No AliTRDcalibDB instance available\n"); + return kFALSE; + } + + AliTRDSimParam *simParam = AliTRDSimParam::Instance(); + if (!simParam) { + AliError("No AliTRDSimParam instance available\n"); + return kFALSE; + } + + AliTRDRecParam *recParam = AliTRDRecParam::Instance(); + if (!recParam) { + AliError("No AliTRDRecParam instance available\n"); + return kFALSE; + } + + AliTRDCommonParam *commonParam = AliTRDCommonParam::Instance(); + if (!commonParam) { + AliError("Could not get common parameters\n"); + return kFALSE; + } - // Parameters - Float_t maxThresh = fClusMaxThresh; // threshold value for maximum - Float_t signalThresh = fClusSigThresh; // threshold value for digit signal - Int_t clusteringMethod = fClusMethod; // clustering method option (for testing) + // ADC threshols + Float_t ADCthreshold = simParam->GetADCthreshold(); + // Threshold value for the maximum + Float_t maxThresh = recParam->GetClusMaxThresh(); + // Threshold value for the digit signal + Float_t sigThresh = recParam->GetClusSigThresh(); // Iteration limit for unfolding procedure const Float_t kEpsilon = 0.01; - const Int_t kNclus = 3; const Int_t kNsig = 5; + const Int_t kNdict = AliTRDdigitsManager::kNDict; + const Int_t kNtrack = kNdict * kNclus; - Int_t chamBeg = 0; - Int_t chamEnd = kNcham; - if (trd->GetSensChamber() >= 0) { - chamBeg = trd->GetSensChamber(); - chamEnd = chamBeg + 1; - } - Int_t planBeg = 0; - Int_t planEnd = kNplan; - if (trd->GetSensPlane() >= 0) { - planBeg = trd->GetSensPlane(); - planEnd = planBeg + 1; - } - Int_t sectBeg = 0; - Int_t sectEnd = kNsect; - - // *** Start clustering *** in every chamber - for (Int_t icham = chamBeg; icham < chamEnd; icham++) { - for (Int_t iplan = planBeg; iplan < planEnd; iplan++) { - for (Int_t isect = sectBeg; isect < sectEnd; isect++) { - - if (trd->GetSensSector() >= 0) { - Int_t sens1 = trd->GetSensSector(); - Int_t sens2 = sens1 + trd->GetSensSectorRange(); - sens2 -= ((Int_t) (sens2 / kNsect)) * kNsect; - if (sens1 < sens2) { - if ((isect < sens1) || (isect >= sens2)) continue; - } - else { - if ((isect < sens1) && (isect >= sens2)) continue; - } - } + Int_t iType = 0; + Int_t iUnfold = 0; + Double_t ratioLeft = 1.0; + Double_t ratioRight = 1.0; - Int_t idet = geo->GetDetector(iplan,icham,isect); + Int_t iClusterROC = 0; - Int_t nClusters = 0; - printf("AliTRDclusterizerV1::MakeCluster -- "); - printf("Analyzing chamber %d, plane %d, sector %d.\n" - ,icham,iplan,isect); - - Int_t nRowMax = geo->GetRowMax(iplan,icham,isect); - Int_t nColMax = geo->GetColMax(iplan); - Int_t nTimeMax = geo->GetTimeMax(); - - // Create a detector matrix to keep maxima - AliTRDmatrix *digitMatrix = new AliTRDmatrix(nRowMax,nColMax,nTimeMax - ,isect,icham,iplan); - // Create a matrix to contain maximum flags - AliTRDmatrix *maximaMatrix = new AliTRDmatrix(nRowMax,nColMax,nTimeMax - ,isect,icham,iplan); - - // Read in the digits - digits = fDigitsManager->GetDigits(idet); - - // Loop through the detector pixel - for (time = 0; time < nTimeMax; time++) { - for ( col = 0; col < nColMax; col++) { - for ( row = 0; row < nRowMax; row++) { - - Int_t signal = digits->GetData(row,col,time); - Int_t index = digits->GetIndex(row,col,time); - - // Fill the detector matrix - if (signal > signalThresh) { - // Store the signal amplitude - digitMatrix->SetSignal(row,col,time,signal); - // Store the digits number - digitMatrix->AddTrack(row,col,time,index); - } + Double_t padSignal[kNsig]; + Double_t clusterSignal[kNclus]; + Double_t clusterPads[kNclus]; - } - } - } + Int_t chamBeg = 0; + Int_t chamEnd = AliTRDgeometry::Ncham(); + Int_t planBeg = 0; + Int_t planEnd = AliTRDgeometry::Nplan(); + Int_t sectBeg = 0; + Int_t sectEnd = AliTRDgeometry::Nsect(); + Int_t nTimeTotal = calibration->GetNumberOfTimeBins(); - // Loop chamber and find maxima in digitMatrix - for ( row = 0; row < nRowMax; row++) { - for ( col = 1; col < nColMax; col++) { - for (time = 0; time < nTimeMax; time++) { - - if (digitMatrix->GetSignal(row,col,time) - < digitMatrix->GetSignal(row,col - 1,time)) { - // really maximum? - if (col > 1) { - if (digitMatrix->GetSignal(row,col - 2,time) - < digitMatrix->GetSignal(row,col - 1,time)) { - // yes, so set maximum flag - maximaMatrix->SetSignal(row,col - 1,time,1); - } - else maximaMatrix->SetSignal(row,col - 1,time,0); - } - } - - } // time - } // col - } // row - - // now check maxima and calculate cluster position - for ( row = 0; row < nRowMax; row++) { - for ( col = 1; col < nColMax; col++) { - for (time = 0; time < nTimeMax; time++) { - - if ((maximaMatrix->GetSignal(row,col,time) > 0) - && (digitMatrix->GetSignal(row,col,time) > maxThresh)) { - - // Ratio resulting from unfolding - Float_t ratio = 0; - // Signals on max and neighbouring pads - Float_t padSignal[kNsig] = {0}; - // Signals from cluster - Float_t clusterSignal[kNclus] = {0}; - // Cluster pad info - Float_t clusterPads[kNclus] = {0}; - // Cluster digit info - Int_t clusterDigit[kNclus] = {0}; - - Int_t iPad; - for (iPad = 0; iPad < kNclus; iPad++) { - clusterSignal[iPad] = digitMatrix->GetSignal(row,col-1+iPad,time); - clusterDigit[iPad] = digitMatrix->GetTrack(row,col-1+iPad,time,0); - } + Int_t dummy[9] = { 0, 0, 0, 0, 0, 0, 0, 0, 0 }; - // neighbouring maximum on right side? - if (col < nColMax - 2) { - if (maximaMatrix->GetSignal(row,col + 2,time) > 0) { + AliDebug(1,Form("Number of Time Bins = %d.\n",nTimeTotal)); - for (iPad = 0; iPad < 5; iPad++) { - padSignal[iPad] = digitMatrix->GetSignal(row,col-1+iPad,time); - } + // Start clustering in every chamber + for (icham = chamBeg; icham < chamEnd; icham++) { + for (iplan = planBeg; iplan < planEnd; iplan++) { + for (isect = sectBeg; isect < sectEnd; isect++) { - // unfold: - ratio = Unfold(kEpsilon, padSignal); + Int_t idet = geo->GetDetector(iplan,icham,isect); - // set signal on overlapping pad to ratio - clusterSignal[2] *= ratio; + // Get the digits + digitsIn = fDigitsManager->GetDigits(idet); + // This is to take care of switched off super modules + if (digitsIn->GetNtime() == 0) { + continue; + } + digitsIn->Expand(); + AliTRDdataArrayI *tracksTmp = fDigitsManager->GetDictionary(idet,0); + tracksTmp->Expand(); + + AliDebug(1,Form("Analyzing chamber %d, plane %d, sector %d.\n" + ,icham,iplan,isect)); + + Int_t nRowMax = commonParam->GetRowMax(iplan,icham,isect); + Int_t nColMax = commonParam->GetColMax(iplan); + + AliTRDpadPlane *padPlane = commonParam->GetPadPlane(iplan,icham); + + Int_t nClusters = 0; + Int_t nClusters2pad = 0; + Int_t nClusters3pad = 0; + Int_t nClusters4pad = 0; + Int_t nClusters5pad = 0; + Int_t nClustersLarge = 0; + + // Apply the gain and the tail cancelation via digital filter + AliTRDdataArrayF *digitsOut = new AliTRDdataArrayF(digitsIn->GetNrow() + ,digitsIn->GetNcol() + ,digitsIn->GetNtime()); + Transform(digitsIn,digitsOut,idet,nRowMax,nColMax,nTimeTotal,ADCthreshold); + + // Input digits are not needed any more + digitsIn->Compress(1,0); + + // Loop through the chamber and find the maxima + for ( row = 0; row < nRowMax; row++) { + for ( col = 2; col < nColMax; col++) { + for (time = 0; time < nTimeTotal; time++) { + + Float_t signalL = TMath::Abs(digitsOut->GetDataUnchecked(row,col ,time)); + Float_t signalM = TMath::Abs(digitsOut->GetDataUnchecked(row,col-1,time)); + Float_t signalR = TMath::Abs(digitsOut->GetDataUnchecked(row,col-2,time)); + + // Look for the maximum + if (signalM >= maxThresh) { + if ((TMath::Abs(signalL) <= signalM) && + (TMath::Abs(signalR) <= signalM)) { + if ((TMath::Abs(signalL) + TMath::Abs(signalR)) > sigThresh) { + // Maximum found, mark the position by a negative signal + digitsOut->SetDataUnchecked(row,col-1,time,-signalM); + } + } + } + + } + } + } + tracksTmp->Compress(1,0); + + // The index to the first cluster of a given ROC + Int_t firstClusterROC = -1; + // The number of cluster in a given ROC + Int_t nClusterROC = 0; + + // Now check the maxima and calculate the cluster position + for ( row = 0; row < nRowMax ; row++) { + for (time = 0; time < nTimeTotal; time++) { + for ( col = 1; col < nColMax-1; col++) { + + // Maximum found ? + if (digitsOut->GetDataUnchecked(row,col,time) < 0.0) { - } + for (iPad = 0; iPad < kNclus; iPad++) { + Int_t iPadCol = col - 1 + iPad; + clusterSignal[iPad] = + TMath::Abs(digitsOut->GetDataUnchecked(row,iPadCol,time)); } - - // Calculate the position of the cluster - switch (clusteringMethod) { - case 1: - // method 1: simply center of mass - clusterPads[0] = row + 0.5; - clusterPads[1] = col - 0.5 + (clusterSignal[2] - clusterSignal[0]) / - (clusterSignal[0] + clusterSignal[1] + clusterSignal[2]); - clusterPads[2] = time + 0.5; - - nClusters++; - break; + + // Count the number of pads in the cluster + Int_t nPadCount = 0; + Int_t ii; + // Look to the left + ii = 0; + while (TMath::Abs(digitsOut->GetDataUnchecked(row,col-ii ,time)) >= sigThresh) { + nPadCount++; + ii++; + if (col-ii < 0) break; + } + // Look to the right + ii = 0; + while (TMath::Abs(digitsOut->GetDataUnchecked(row,col+ii+1,time)) >= sigThresh) { + nPadCount++; + ii++; + if (col+ii+1 >= nColMax) break; + } + nClusters++; + switch (nPadCount) { case 2: - // method 2: integral gauss fit on 3 pads - TH1F *hPadCharges = new TH1F("hPadCharges", "Charges on center 3 pads" - , 5, -1.5, 3.5); - for (Int_t iCol = -1; iCol <= 3; iCol++) { - if (clusterSignal[iCol] < 1) clusterSignal[iCol] = 1; - hPadCharges->Fill(iCol, clusterSignal[iCol]); + iType = 0; + nClusters2pad++; + break; + case 3: + iType = 1; + nClusters3pad++; + break; + case 4: + iType = 2; + nClusters4pad++; + break; + case 5: + iType = 3; + nClusters5pad++; + break; + default: + iType = 4; + nClustersLarge++; + break; + }; + + // Look for 5 pad cluster with minimum in the middle + Bool_t fivePadCluster = kFALSE; + if (col < (nColMax - 3)) { + if (digitsOut->GetDataUnchecked(row,col+2,time) < 0) { + fivePadCluster = kTRUE; + } + if ((fivePadCluster) && (col < (nColMax - 5))) { + if (digitsOut->GetDataUnchecked(row,col+4,time) >= sigThresh) { + fivePadCluster = kFALSE; + } + } + if ((fivePadCluster) && (col > 1)) { + if (digitsOut->GetDataUnchecked(row,col-2,time) >= sigThresh) { + fivePadCluster = kFALSE; + } + } + } + + // 5 pad cluster + // Modify the signal of the overlapping pad for the left part + // of the cluster which remains from a previous unfolding + if (iUnfold) { + clusterSignal[0] *= ratioLeft; + iType = 5; + iUnfold = 0; + } + + // Unfold the 5 pad cluster + if (fivePadCluster) { + for (iPad = 0; iPad < kNsig; iPad++) { + padSignal[iPad] = TMath::Abs(digitsOut->GetDataUnchecked(row + ,col-1+iPad + ,time)); } - hPadCharges->Fit("gaus", "IQ", "SAME", -0.5, 2.5); - TF1 *fPadChargeFit = hPadCharges->GetFunction("gaus"); - Double_t colMean = fPadChargeFit->GetParameter(1); + // Unfold the two maxima and set the signal on + // the overlapping pad to the ratio + ratioRight = Unfold(kEpsilon,iplan,padSignal); + ratioLeft = 1.0 - ratioRight; + clusterSignal[2] *= ratioRight; + iType = 5; + iUnfold = 1; + } - clusterPads[0] = row + 0.5; - clusterPads[1] = col - 1.5 + colMean; - clusterPads[2] = time + 0.5; + Double_t clusterCharge = clusterSignal[0] + + clusterSignal[1] + + clusterSignal[2]; + + // The position of the cluster + clusterPads[0] = row + 0.5; + // Take the shift of the additional time bins into account + clusterPads[2] = time + 0.5; + + if (recParam->LUTOn()) { + // Calculate the position of the cluster by using the + // lookup table method + clusterPads[1] = recParam->LUTposition(iplan,clusterSignal[0] + ,clusterSignal[1] + ,clusterSignal[2]); + } + else { + // Calculate the position of the cluster by using the + // center of gravity method + for (Int_t i = 0; i < kNsig; i++) { + padSignal[i] = 0.0; + } + padSignal[2] = TMath::Abs(digitsOut->GetDataUnchecked(row,col ,time)); // Central pad + padSignal[1] = TMath::Abs(digitsOut->GetDataUnchecked(row,col-1,time)); // Left pad + padSignal[3] = TMath::Abs(digitsOut->GetDataUnchecked(row,col+1,time)); // Right pad + if ((col > 2) && + (TMath::Abs(digitsOut->GetDataUnchecked(row,col-2,time)) < padSignal[1])) { + padSignal[0] = TMath::Abs(digitsOut->GetDataUnchecked(row,col-2,time)); + } + if ((col < nColMax - 3) && + (TMath::Abs(digitsOut->GetDataUnchecked(row,col+2,time)) < padSignal[3])) { + padSignal[4] = TMath::Abs(digitsOut->GetDataUnchecked(row,col+2,time)); + } + clusterPads[1] = GetCOG(padSignal); + } + + Double_t q0 = clusterSignal[0]; + Double_t q1 = clusterSignal[1]; + Double_t q2 = clusterSignal[2]; + Double_t clusterSigmaY2 = (q1 * (q0 + q2) + 4.0 * q0 * q2) + / (clusterCharge*clusterCharge); + + // + // Calculate the position and the error + // + + // Correct for t0 + Int_t clusterTimeBin = TMath::Nint(time - calibration->GetT0(idet,col,row)); + Double_t colSize = padPlane->GetColSize(col); + Double_t rowSize = padPlane->GetRowSize(row); + + Double_t clusterPos[3]; + clusterPos[0] = padPlane->GetColPos(col) - (clusterPads[1] + 0.5) * colSize; + clusterPos[1] = padPlane->GetRowPos(row) - 0.5 * rowSize; + clusterPos[2] = CalcXposFromTimebin(clusterPads[2],idet,col,row); + Double_t clusterSig[2]; + clusterSig[0] = (clusterSigmaY2 + 1.0/12.0) * colSize*colSize; + clusterSig[1] = rowSize * rowSize / 12.0; + + // Add the cluster to the output array + // The track indices will be stored later + AliTRDcluster *cluster = AddCluster(clusterPos + ,clusterTimeBin + ,idet + ,clusterCharge + ,dummy + ,clusterSig + ,iType + ,clusterPads[1]); + + // Store the amplitudes of the pads in the cluster for later analysis + Short_t signals[7] = { 0, 0, 0, 0, 0, 0, 0 }; + for (Int_t jPad = col-3; jPad <= col+3; jPad++) { + if ((jPad < 0) || + (jPad >= nColMax-1)) { + continue; + } + signals[jPad-col+3] = TMath::Nint(TMath::Abs(digitsOut->GetDataUnchecked(row,jPad,time))); + } + cluster->SetSignals(signals); + + // Temporarily store the row, column and time bin of the center pad + // Used to later on assign the track indices + cluster->SetLabel( row,0); + cluster->SetLabel( col,1); + cluster->SetLabel(time,2); + + // Store the index of the first cluster in the current ROC + if (firstClusterROC < 0) { + firstClusterROC = RecPoints()->GetEntriesFast() - 1; + } + // Count the number of cluster in the current ROC + nClusterROC++; + + } // if: Maximum found ? + + } // loop: pad columns + } // loop: time bins + } // loop: pad rows + + delete digitsOut; + + // + // Add the track indices to the found clusters + // + + // Temporary array to collect the track indices + Int_t *idxTracks = new Int_t[kNtrack*nClusterROC]; + + // Loop through the dictionary arrays one-by-one + // to keep memory consumption low + for (Int_t iDict = 0; iDict < kNdict; iDict++) { + + tracksIn = fDigitsManager->GetDictionary(idet,iDict); + tracksIn->Expand(); + + // Loop though the clusters found in this ROC + for (iClusterROC = 0; iClusterROC < nClusterROC; iClusterROC++) { + + AliTRDcluster *cluster = (AliTRDcluster *) + RecPoints()->UncheckedAt(firstClusterROC+iClusterROC); + row = cluster->GetLabel(0); + col = cluster->GetLabel(1); + time = cluster->GetLabel(2); + + for (iPad = 0; iPad < kNclus; iPad++) { + Int_t iPadCol = col - 1 + iPad; + Int_t index = tracksIn->GetDataUnchecked(row,iPadCol,time) - 1; + idxTracks[3*iPad+iDict + iClusterROC*kNtrack] = index; + } - delete hPadCharges; + } - nClusters++; - break; - } + // Compress the arrays + tracksIn->Compress(1,0); - Float_t clusterCharge = clusterSignal[0] - + clusterSignal[1] - + clusterSignal[2]; + } - // Add the cluster to the output array - trd->AddRecPoint(clusterPads,clusterDigit,idet,clusterCharge); + // Copy the track indices into the cluster + // Loop though the clusters found in this ROC + for (iClusterROC = 0; iClusterROC < nClusterROC; iClusterROC++) { + + AliTRDcluster *cluster = (AliTRDcluster *) + RecPoints()->UncheckedAt(firstClusterROC+iClusterROC); + cluster->SetLabel(-9999,0); + cluster->SetLabel(-9999,1); + cluster->SetLabel(-9999,2); + + cluster->AddTrackIndex(&idxTracks[iClusterROC*kNtrack]); - } - } // time - } // col - } // row + } - printf("AliTRDclusterizerV1::MakeCluster -- "); - printf("Number of clusters found: %d\n",nClusters); + delete [] idxTracks; - delete digitMatrix; - delete maximaMatrix; + // Write the cluster and reset the array + WriteClusters(idet); + ResetRecPoints(); - } // isect - } // iplan - } // icham + } // loop: Sectors + } // loop: Planes + } // loop: Chambers - printf("AliTRDclusterizerV1::MakeCluster -- "); - printf("Total number of points found: %d\n" - ,trd->RecPoints()->GetEntries()); + return kTRUE; - // Get the pointer to the cluster branch - TTree *clusterTree = gAlice->TreeR(); +} - // Fill the cluster-branch - printf("AliTRDclusterizerV1::MakeCluster -- "); - printf("Fill the cluster tree.\n"); - clusterTree->Fill(); - printf("AliTRDclusterizerV1::MakeCluster -- "); - printf("Done.\n"); +//_____________________________________________________________________________ +Double_t AliTRDclusterizerV1::GetCOG(Double_t signal[5]) +{ + // + // Get COG position + // Used for clusters with more than 3 pads - where LUT not applicable + // - return kTRUE; + Double_t sum = signal[0] + + signal[1] + + signal[2] + + signal[3] + + signal[4]; + + Double_t res = (0.0 * (-signal[0] + signal[4]) + + (-signal[1] + signal[3])) / sum; + + return res; } //_____________________________________________________________________________ -Float_t AliTRDclusterizerV1::Unfold(Float_t eps, Float_t* padSignal) +Double_t AliTRDclusterizerV1::Unfold(Double_t eps, Int_t plane, Double_t *padSignal) { // // Method to unfold neighbouring maxima. @@ -421,41 +606,47 @@ Float_t AliTRDclusterizerV1::Unfold(Float_t eps, Float_t* padSignal) // The resulting ratio is then returned to the calling method. // - Int_t itStep = 0; // count iteration steps + AliTRDcalibDB *calibration = AliTRDcalibDB::Instance(); + if (!calibration) { + AliError("No AliTRDcalibDB instance available\n"); + return kFALSE; + } + + Int_t irc = 0; + Int_t itStep = 0; // Count iteration steps - Float_t ratio = 0.5; // start value for ratio - Float_t prevRatio = 0; // store previous ratio + Double_t ratio = 0.5; // Start value for ratio + Double_t prevRatio = 0.0; // Store previous ratio - Float_t newLeftSignal[3] = {0}; // array to store left cluster signal - Float_t newRightSignal[3] = {0}; // array to store right cluster signal + Double_t newLeftSignal[3] = { 0.0, 0.0, 0.0 }; // Array to store left cluster signal + Double_t newRightSignal[3] = { 0.0, 0.0, 0.0 }; // Array to store right cluster signal + Double_t newSignal[3] = { 0.0, 0.0, 0.0 }; - // start iteration: + // Start the iteration while ((TMath::Abs(prevRatio - ratio) > eps) && (itStep < 10)) { itStep++; prevRatio = ratio; - // cluster position according to charge ratio - Float_t maxLeft = (ratio*padSignal[2] - padSignal[0]) / - (padSignal[0] + padSignal[1] + ratio*padSignal[2]); - Float_t maxRight = (padSignal[4] - (1-ratio)*padSignal[2]) / - ((1-ratio)*padSignal[2] + padSignal[3] + padSignal[4]); - - // set cluster charge ratio - Float_t ampLeft = padSignal[1]; - Float_t ampRight = padSignal[3]; + // Cluster position according to charge ratio + Double_t maxLeft = (ratio*padSignal[2] - padSignal[0]) + / (padSignal[0] + padSignal[1] + ratio*padSignal[2]); + Double_t maxRight = (padSignal[4] - (1-ratio)*padSignal[2]) + / ((1.0 - ratio)*padSignal[2] + padSignal[3] + padSignal[4]); - // apply pad response to parameters - newLeftSignal[0] = ampLeft*PadResponse(-1 - maxLeft); - newLeftSignal[1] = ampLeft*PadResponse( 0 - maxLeft); - newLeftSignal[2] = ampLeft*PadResponse( 1 - maxLeft); + // Set cluster charge ratio + irc = calibration->PadResponse(1.0,maxLeft ,plane,newSignal); + Double_t ampLeft = padSignal[1] / newSignal[1]; + irc = calibration->PadResponse(1.0,maxRight,plane,newSignal); + Double_t ampRight = padSignal[3] / newSignal[1]; - newRightSignal[0] = ampRight*PadResponse(-1 - maxRight); - newRightSignal[1] = ampRight*PadResponse( 0 - maxRight); - newRightSignal[2] = ampRight*PadResponse( 1 - maxRight); + // Apply pad response to parameters + irc = calibration->PadResponse(ampLeft ,maxLeft ,plane,newLeftSignal ); + irc = calibration->PadResponse(ampRight,maxRight,plane,newRightSignal); - // calculate new overlapping ratio - ratio = newLeftSignal[2]/(newLeftSignal[2] + newRightSignal[0]); + // Calculate new overlapping ratio + ratio = TMath::Min((Double_t)1.0,newLeftSignal[2] / + (newLeftSignal[2] + newRightSignal[0])); } @@ -464,22 +655,143 @@ Float_t AliTRDclusterizerV1::Unfold(Float_t eps, Float_t* padSignal) } //_____________________________________________________________________________ -Float_t AliTRDclusterizerV1::PadResponse(Float_t x) +void AliTRDclusterizerV1::Transform(AliTRDdataArrayI *digitsIn + , AliTRDdataArrayF *digitsOut + , Int_t idet, Int_t nRowMax + , Int_t nColMax, Int_t nTimeTotal + , Float_t ADCthreshold) +{ + // + // Apply gain factor + // Apply tail cancelation: Transform digitsIn to digitsOut + // + + Int_t iRow = 0; + Int_t iCol = 0; + Int_t iTime = 0; + + AliTRDRecParam *recParam = AliTRDRecParam::Instance(); + if (!recParam) { + AliError("No AliTRDRecParam instance available\n"); + return; + } + AliTRDcalibDB *calibration = AliTRDcalibDB::Instance(); + if (!calibration) { + AliError("No AliTRDcalibDB instance available\n"); + return; + } + + Double_t *inADC = new Double_t[nTimeTotal]; // ADC data before tail cancellation + Double_t *outADC = new Double_t[nTimeTotal]; // ADC data after tail cancellation + + AliDebug(1,Form("Tail cancellation (nExp = %d) for detector %d.\n" + ,recParam->GetTCnexp(),idet)); + + for (iRow = 0; iRow < nRowMax; iRow++ ) { + for (iCol = 0; iCol < nColMax; iCol++ ) { + + for (iTime = 0; iTime < nTimeTotal; iTime++) { + + // + // Add gain + // + Double_t gain = calibration->GetGainFactor(idet,iCol,iRow); + if (gain == 0.0) { + AliError("Not a valid gain\n"); + } + inADC[iTime] = digitsIn->GetDataUnchecked(iRow,iCol,iTime); + inADC[iTime] /= gain; + outADC[iTime] = inADC[iTime]; + + } + + // Apply the tail cancelation via the digital filter + if (recParam->TCOn()) { + DeConvExp(inADC,outADC,nTimeTotal,recParam->GetTCnexp()); + } + + for (iTime = 0; iTime < nTimeTotal; iTime++) { + + // Store the amplitude of the digit if above threshold + if (outADC[iTime] > ADCthreshold) { + AliDebug(2,Form(" iRow = %d, iCol = %d, iTime = %d, adc = %f\n" + ,iRow,iCol,iTime,outADC[iTime])); + digitsOut->SetDataUnchecked(iRow,iCol,iTime,outADC[iTime]); + } + + } + + } + } + + delete [] inADC; + delete [] outADC; + + return; + +} + +//_____________________________________________________________________________ +void AliTRDclusterizerV1::DeConvExp(Double_t *source, Double_t *target + , Int_t n, Int_t nexp) { // - // The pad response for the chevron pads. - // We use a simple Gaussian approximation which should be good - // enough for our purpose. + // Tail cancellation by deconvolution for PASA v4 TRF // - // The parameters for the response function - const Float_t kA = 0.8872; - const Float_t kB = -0.00573; - const Float_t kC = 0.454; - const Float_t kC2 = kC*kC; + Double_t rates[2]; + Double_t coefficients[2]; + + // Initialization (coefficient = alpha, rates = lambda) + Double_t R1 = 1.0; + Double_t R2 = 1.0; + Double_t C1 = 0.5; + Double_t C2 = 0.5; + + if (nexp == 1) { // 1 Exponentials + R1 = 1.156; + R2 = 0.130; + C1 = 0.066; + C2 = 0.000; + } + if (nexp == 2) { // 2 Exponentials + R1 = 1.156; + R2 = 0.130; + C1 = 0.114; + C2 = 0.624; + } + + coefficients[0] = C1; + coefficients[1] = C2; + + Double_t Dt = 0.1; - Float_t pr = kA * (kB + TMath::Exp(-x*x / (2. * kC2))); + rates[0] = TMath::Exp(-Dt/(R1)); + rates[1] = TMath::Exp(-Dt/(R2)); + + Int_t i = 0; + Int_t k = 0; - return (pr); + Double_t reminder[2]; + Double_t correction; + Double_t result; + + // Attention: computation order is important + correction = 0.0; + for (k = 0; k < nexp; k++) { + reminder[k] = 0.0; + } + for (i = 0; i < n; i++) { + result = (source[i] - correction); // No rescaling + target[i] = result; + + for (k = 0; k < nexp; k++) { + reminder[k] = rates[k] * (reminder[k] + coefficients[k] * result); + } + correction = 0.0; + for (k = 0; k < nexp; k++) { + correction += reminder[k]; + } + } }