X-Git-Url: http://git.uio.no/git/?a=blobdiff_plain;f=TRD%2FAliTRDclusterizer.cxx;h=205cf96049eecd9720314bcd8f56beab43c75c1a;hb=2efdb82126d743111d102c1ffd636ae9e4935cbf;hp=584dac756a122cefab99b704aecb3f97315173ee;hpb=25ca55ce1319759ab4aef50a95d664a47f1cd597;p=u%2Fmrichter%2FAliRoot.git diff --git a/TRD/AliTRDclusterizer.cxx b/TRD/AliTRDclusterizer.cxx index 584dac756a1..205cf96049e 100644 --- a/TRD/AliTRDclusterizer.cxx +++ b/TRD/AliTRDclusterizer.cxx @@ -1,3 +1,4 @@ + /************************************************************************** * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. * * * @@ -17,25 +18,41 @@ /////////////////////////////////////////////////////////////////////////////// // // -// TRD cluster finder base class // +// TRD cluster finder // // // /////////////////////////////////////////////////////////////////////////////// -#include +#include #include +#include #include #include #include "AliRunLoader.h" #include "AliLoader.h" +#include "AliRawReader.h" #include "AliLog.h" +#include "AliAlignObj.h" #include "AliTRDclusterizer.h" #include "AliTRDcluster.h" -#include "AliTRDrecPoint.h" +#include "AliTRDReconstructor.h" #include "AliTRDgeometry.h" +#include "AliTRDdataArrayF.h" +#include "AliTRDdataArrayI.h" +#include "AliTRDdataArrayS.h" +#include "AliTRDdigitsManager.h" +#include "AliTRDrawData.h" #include "AliTRDcalibDB.h" +#include "AliTRDrecoParam.h" #include "AliTRDCommonParam.h" +#include "AliTRDtransform.h" +#include "AliTRDSignalIndex.h" +#include "AliTRDrawStreamBase.h" +#include "AliTRDfeeParam.h" + +#include "Cal/AliTRDCalROC.h" +#include "Cal/AliTRDCalDet.h" ClassImp(AliTRDclusterizer) @@ -45,24 +62,48 @@ AliTRDclusterizer::AliTRDclusterizer() ,fRunLoader(NULL) ,fClusterTree(NULL) ,fRecPoints(NULL) + ,fDigitsManager(NULL) + ,fAddLabels(kTRUE) + ,fRawVersion(2) + ,fIndexesOut(NULL) + ,fIndexesMaxima(NULL) + ,fTransform(new AliTRDtransform(0)) + ,fLUTbin(0) + ,fLUT(NULL) { // // AliTRDclusterizer default constructor // + fRawVersion = AliTRDfeeParam::Instance()->GetRAWversion(); + } //_____________________________________________________________________________ -AliTRDclusterizer::AliTRDclusterizer(const Text_t* name, const Text_t* title) +AliTRDclusterizer::AliTRDclusterizer(const Text_t *name, const Text_t *title) :TNamed(name,title) ,fRunLoader(NULL) ,fClusterTree(NULL) ,fRecPoints(NULL) + ,fDigitsManager(new AliTRDdigitsManager()) + ,fAddLabels(kTRUE) + ,fRawVersion(2) + ,fIndexesOut(NULL) + ,fIndexesMaxima(NULL) + ,fTransform(new AliTRDtransform(0)) + ,fLUTbin(0) + ,fLUT(NULL) { // // AliTRDclusterizer constructor // + fDigitsManager->CreateArrays(); + + fRawVersion = AliTRDfeeParam::Instance()->GetRAWversion(); + + FillLUT(); + } //_____________________________________________________________________________ @@ -71,11 +112,21 @@ AliTRDclusterizer::AliTRDclusterizer(const AliTRDclusterizer &c) ,fRunLoader(NULL) ,fClusterTree(NULL) ,fRecPoints(NULL) + ,fDigitsManager(NULL) + ,fAddLabels(kTRUE) + ,fRawVersion(2) + ,fIndexesOut(NULL) + ,fIndexesMaxima(NULL) + ,fTransform(NULL) + ,fLUTbin(0) + ,fLUT(0) { // // AliTRDclusterizer copy constructor // + FillLUT(); + } //_____________________________________________________________________________ @@ -85,10 +136,41 @@ AliTRDclusterizer::~AliTRDclusterizer() // AliTRDclusterizer destructor // - if (fRecPoints) { - fRecPoints->Delete(); - delete fRecPoints; - } + if (fRecPoints) + { + fRecPoints->Delete(); + delete fRecPoints; + } + + if (fDigitsManager) + { + delete fDigitsManager; + fDigitsManager = NULL; + } + + if (fIndexesOut) + { + delete fIndexesOut; + fIndexesOut = NULL; + } + + if (fIndexesMaxima) + { + delete fIndexesMaxima; + fIndexesMaxima = NULL; + } + + if (fTransform) + { + delete fTransform; + fTransform = NULL; + } + + if (fLUT) + { + delete [] fLUT; + fLUT = NULL; + } } @@ -99,9 +181,10 @@ AliTRDclusterizer &AliTRDclusterizer::operator=(const AliTRDclusterizer &c) // Assignment operator // - if (this != &c) { - ((AliTRDclusterizer &) c).Copy(*this); - } + if (this != &c) + { + ((AliTRDclusterizer &) c).Copy(*this); + } return *this; } @@ -113,8 +196,16 @@ void AliTRDclusterizer::Copy(TObject &c) const // Copy function // - ((AliTRDclusterizer &) c).fClusterTree = NULL; - ((AliTRDclusterizer &) c).fRecPoints = NULL; + ((AliTRDclusterizer &) c).fClusterTree = NULL; + ((AliTRDclusterizer &) c).fRecPoints = NULL; + ((AliTRDclusterizer &) c).fDigitsManager = NULL; + ((AliTRDclusterizer &) c).fAddLabels = fAddLabels; + ((AliTRDclusterizer &) c).fRawVersion = fRawVersion; + ((AliTRDclusterizer &) c).fIndexesOut = NULL; + ((AliTRDclusterizer &) c).fIndexesMaxima = NULL; + ((AliTRDclusterizer &) c).fTransform = NULL; + ((AliTRDclusterizer &) c).fLUTbin = 0; + ((AliTRDclusterizer &) c).fLUT = NULL; } @@ -225,7 +316,9 @@ Bool_t AliTRDclusterizer::WriteClusters(Int_t det) detRecPoints->AddLast(c); } else { - AliError("Attempt to write a cluster with unexpected detector index\n"); + AliError(Form("Attempt to write a cluster with unexpected detector index: got=%d expected=%d\n" + ,c->GetDetector() + ,det)); } } @@ -233,7 +326,7 @@ Bool_t AliTRDclusterizer::WriteClusters(Int_t det) fClusterTree->Fill(); delete detRecPoints; - + return kTRUE; } @@ -268,83 +361,860 @@ Bool_t AliTRDclusterizer::WriteClusters(Int_t det) } +//_____________________________________________________________________________ +void AliTRDclusterizer::ResetHelperIndexes(AliTRDSignalIndex *indexesIn) +{ + // + // Reset the helper indexes + // + + if (fIndexesOut) + { + // carefull here - we assume that only row number may change - most probable + if (indexesIn->GetNrow() <= fIndexesOut->GetNrow()) + fIndexesOut->ResetContent(); + else + fIndexesOut->ResetContentConditional(indexesIn->GetNrow() + , indexesIn->GetNcol() + , indexesIn->GetNtime()); + } + else + { + fIndexesOut = new AliTRDSignalIndex(indexesIn->GetNrow() + , indexesIn->GetNcol() + , indexesIn->GetNtime()); + } + + if (fIndexesMaxima) + { + // carefull here - we assume that only row number may change - most probable + if (indexesIn->GetNrow() <= fIndexesMaxima->GetNrow()) + { + fIndexesMaxima->ResetContent(); + } + else + { + fIndexesMaxima->ResetContentConditional(indexesIn->GetNrow() + , indexesIn->GetNcol() + , indexesIn->GetNtime()); + } + } + else + { + fIndexesMaxima = new AliTRDSignalIndex(indexesIn->GetNrow() + , indexesIn->GetNcol() + , indexesIn->GetNtime()); + } + +} //_____________________________________________________________________________ -AliTRDcluster* AliTRDclusterizer::AddCluster(Double_t *pos, Int_t timebin - , Int_t det, Double_t amp - , Int_t *tracks, Double_t *sig - , Int_t iType, Float_t center) +Bool_t AliTRDclusterizer::ReadDigits() { // - // Add a cluster for the TRD + // Reads the digits arrays from the input aliroot file // - AliTRDcluster *c = new AliTRDcluster(); - - c->SetDetector(det); - c->SetQ(amp); - c->SetX(pos[2]); - c->SetY(pos[0]); - c->SetZ(pos[1]); - c->SetSigmaY2(sig[0]); - c->SetSigmaZ2(sig[1]); - c->SetLocalTimeBin(timebin); - c->SetCenter(center); + if (!fRunLoader) { + AliError("No run loader available"); + return kFALSE; + } - if (tracks) { - c->AddTrackIndex(tracks); + AliLoader* loader = fRunLoader->GetLoader("TRDLoader"); + if (!loader->TreeD()) { + loader->LoadDigits(); } - switch (iType) { - case 0: - c->Set2pad(); - break; - case 1: - c->Set3pad(); - break; - case 2: - c->Set4pad(); - break; - case 3: - c->Set5pad(); - break; - case 4: - c->SetLarge(); - break; - }; - - RecPoints()->Add(c); - return c; + // Read in the digit arrays + return (fDigitsManager->ReadDigits(loader->TreeD())); + +} + +//_____________________________________________________________________________ +Bool_t AliTRDclusterizer::ReadDigits(TTree *digitsTree) +{ + // + // Reads the digits arrays from the input tree + // + + // Read in the digit arrays + return (fDigitsManager->ReadDigits(digitsTree)); + +} + +//_____________________________________________________________________________ +Bool_t AliTRDclusterizer::ReadDigits(AliRawReader *rawReader) +{ + // + // Reads the digits arrays from the ddl file + // + + AliTRDrawData raw; + fDigitsManager = raw.Raw2Digits(rawReader); + + return kTRUE; + +} + +//_____________________________________________________________________________ +Bool_t AliTRDclusterizer::MakeClusters() +{ + // + // Creates clusters from digits + // + + // Propagate info from the digits manager + if (fAddLabels == kTRUE) + { + fAddLabels = fDigitsManager->UsesDictionaries(); + } + + Bool_t fReturn = kTRUE; + for (Int_t i = 0; i < AliTRDgeometry::kNdet; i++) + { + + AliTRDdataArrayS *digitsIn = (AliTRDdataArrayS *) fDigitsManager->GetDigits(i); + // This is to take care of switched off super modules + if (!digitsIn->HasData()) + { + continue; + } + digitsIn->Expand(); + AliTRDSignalIndex* indexes = fDigitsManager->GetIndexes(i); + if (indexes->IsAllocated() == kFALSE) + { + fDigitsManager->BuildIndexes(i); + } + + Bool_t fR = kFALSE; + if (indexes->HasEntry()) + { + if (fAddLabels) + { + for (Int_t iDict = 0; iDict < AliTRDdigitsManager::kNDict; iDict++) + { + AliTRDdataArrayI *tracksIn = 0; + tracksIn = (AliTRDdataArrayI *) fDigitsManager->GetDictionary(i,iDict); + tracksIn->Expand(); + } + } + fR = MakeClusters(i); + fReturn = fR && fReturn; + } + + if (fR == kFALSE) + { + WriteClusters(i); + ResetRecPoints(); + } + + // No compress just remove + fDigitsManager->RemoveDigits(i); + fDigitsManager->RemoveDictionaries(i); + fDigitsManager->ClearIndexes(i); + + } + + return fReturn; + +} + +//_____________________________________________________________________________ +Bool_t AliTRDclusterizer::Raw2Clusters(AliRawReader *rawReader) +{ + // + // Creates clusters from raw data + // + + return Raw2ClustersChamber(rawReader); + +} + +//_____________________________________________________________________________ +Bool_t AliTRDclusterizer::Raw2ClustersChamber(AliRawReader *rawReader) +{ + // + // Creates clusters from raw data + // + + // Create the digits manager + if (!fDigitsManager) + { + fDigitsManager = new AliTRDdigitsManager(); + fDigitsManager->CreateArrays(); + } + + fDigitsManager->SetUseDictionaries(fAddLabels); + + AliTRDrawStreamBase *pinput = AliTRDrawStreamBase::GetRawStream(rawReader); + AliTRDrawStreamBase &input = *pinput; + + AliInfo(Form("Stream version: %s", input.IsA()->GetName())); + + Int_t det = 0; + while ((det = input.NextChamber(fDigitsManager)) >= 0) + { + Bool_t iclusterBranch = kFALSE; + if (fDigitsManager->GetIndexes(det)->HasEntry()) + { + iclusterBranch = MakeClusters(det); + } + if (iclusterBranch == kFALSE) + { + WriteClusters(det); + ResetRecPoints(); + } + fDigitsManager->RemoveDigits(det); + fDigitsManager->RemoveDictionaries(det); + fDigitsManager->ClearIndexes(det); + } + + delete fDigitsManager; + fDigitsManager = NULL; + + delete pinput; + pinput = NULL; + return kTRUE; + +} + +//_____________________________________________________________________________ +Bool_t AliTRDclusterizer::MakeClusters(Int_t det) +{ + // + // Generates the cluster. + // + + // Get the digits + // digits should be expanded beforehand! + // digitsIn->Expand(); + AliTRDdataArrayS *digitsIn = (AliTRDdataArrayS *) fDigitsManager->GetDigits(det); + + // This is to take care of switched off super modules + if (!digitsIn->HasData()) + { + return kFALSE; + } + + AliTRDSignalIndex *indexesIn = fDigitsManager->GetIndexes(det); + if (indexesIn->IsAllocated() == kFALSE) + { + AliError("Indexes do not exist!"); + return kFALSE; + } + + AliTRDcalibDB *calibration = AliTRDcalibDB::Instance(); + if (!calibration) + { + AliFatal("No AliTRDcalibDB instance available\n"); + return kFALSE; + } + + // ADC thresholds + // There is no ADC threshold anymore, and simParam should not be used in clusterizer. KO + Float_t adcThreshold = 0; + + if (!AliTRDReconstructor::RecoParam()) + { + AliError("RecoParam does not exist\n"); + return kFALSE; + } + + // Threshold value for the maximum + Float_t maxThresh = AliTRDReconstructor::RecoParam()->GetClusMaxThresh(); + // Threshold value for the digit signal + Float_t sigThresh = AliTRDReconstructor::RecoParam()->GetClusSigThresh(); + + // Iteration limit for unfolding procedure + const Float_t kEpsilon = 0.01; + const Int_t kNclus = 3; + const Int_t kNsig = 5; + + Int_t iUnfold = 0; + Double_t ratioLeft = 1.0; + Double_t ratioRight = 1.0; + + Double_t padSignal[kNsig]; + Double_t clusterSignal[kNclus]; + + Int_t icham = indexesIn->GetChamber(); + Int_t iplan = indexesIn->GetPlane(); + Int_t isect = indexesIn->GetSM(); + + // Start clustering in the chamber + + Int_t idet = AliTRDgeometry::GetDetector(iplan,icham,isect); + if (idet != det) + { + AliError("Strange Detector number Missmatch!"); + return kFALSE; + } + + // TRD space point transformation + fTransform->SetDetector(det); + + Int_t ilayer = AliGeomManager::kTRD1 + iplan; + Int_t imodule = icham + AliTRDgeometry::Ncham() * isect; + UShort_t volid = AliGeomManager::LayerToVolUID(ilayer,imodule); + + Int_t nColMax = digitsIn->GetNcol(); + Int_t nTimeTotal = digitsIn->GetNtime(); + + // Detector wise calibration object for the gain factors + const AliTRDCalDet *calGainFactorDet = calibration->GetGainFactorDet(); + // Calibration object with pad wise values for the gain factors + AliTRDCalROC *calGainFactorROC = calibration->GetGainFactorROC(idet); + // Calibration value for chamber wise gain factor + Float_t calGainFactorDetValue = calGainFactorDet->GetValue(idet); + + Int_t nClusters = 0; + + AliTRDdataArrayF *digitsOut = new AliTRDdataArrayF(digitsIn->GetNrow() + ,digitsIn->GetNcol() + ,digitsIn->GetNtime()); + + ResetHelperIndexes(indexesIn); + + // Apply the gain and the tail cancelation via digital filter + TailCancelation(digitsIn + ,digitsOut + ,indexesIn + ,fIndexesOut + ,nTimeTotal + ,adcThreshold + ,calGainFactorROC + ,calGainFactorDetValue); + + Int_t row = 0; + Int_t col = 0; + Int_t time = 0; + Int_t iPad = 0; + + fIndexesOut->ResetCounters(); + while (fIndexesOut->NextRCTbinIndex(row, col, time)) + { + + Float_t signalM = TMath::Abs(digitsOut->GetDataUnchecked(row,col,time)); + + // Look for the maximum + if (signalM >= maxThresh) + { + + if (col + 1 >= nColMax || col-1 < 0) + continue; + + Float_t signalL = TMath::Abs(digitsOut->GetDataUnchecked(row,col+1,time)); + Float_t signalR = TMath::Abs(digitsOut->GetDataUnchecked(row,col-1,time)); + + if ((TMath::Abs(signalL) <= signalM) && + (TMath::Abs(signalR) < signalM)) + { + if ((TMath::Abs(signalL) >= sigThresh) || + (TMath::Abs(signalR) >= sigThresh)) + { + // Maximum found, mark the position by a negative signal + digitsOut->SetDataUnchecked(row,col,time,-signalM); + fIndexesMaxima->AddIndexTBin(row,col,time); + } + } + + } + + } + + // 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 + fIndexesMaxima->ResetCounters(); + while (fIndexesMaxima->NextRCTbinIndex(row, col, time)) + { + + // 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)); + } + + // 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++; + + // 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; + 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)); + } + // 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; + iUnfold = 1; + } + + // The position of the cluster in COL direction relative to the center pad (pad units) + Double_t clusterPosCol = 0.0; + if (AliTRDReconstructor::RecoParam()->LUTOn()) + { + // Calculate the position of the cluster by using the + // lookup table method + clusterPosCol = 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)); + } + clusterPosCol = GetCOG(padSignal); + } + + // 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))); + } + + // Transform the local cluster coordinates into calibrated + // space point positions defined in the local tracking system. + // Here the calibration for T0, Vdrift and ExB is applied as well. + Double_t clusterXYZ[6]; + clusterXYZ[0] = clusterPosCol; + clusterXYZ[1] = clusterSignal[0]; + clusterXYZ[2] = clusterSignal[1]; + clusterXYZ[3] = clusterSignal[2]; + clusterXYZ[4] = 0.0; + clusterXYZ[5] = 0.0; + Int_t clusterRCT[3]; + clusterRCT[0] = row; + clusterRCT[1] = col; + clusterRCT[2] = 0; + + Bool_t out = kTRUE; + if (fTransform->Transform(clusterXYZ, clusterRCT, ((UInt_t) time), out, 0)) { + + // Add the cluster to the output array + // The track indices will be stored later + Float_t clusterPos[3]; + clusterPos[0] = clusterXYZ[0]; + clusterPos[1] = clusterXYZ[1]; + clusterPos[2] = clusterXYZ[2]; + Float_t clusterSig[2]; + clusterSig[0] = clusterXYZ[4]; + clusterSig[1] = clusterXYZ[5]; + Double_t clusterCharge = clusterXYZ[3]; + Char_t clusterTimeBin = ((Char_t) clusterRCT[2]); + AliTRDcluster *cluster = new AliTRDcluster(idet + ,clusterCharge + ,clusterPos + ,clusterSig + ,0x0 + ,((Char_t) nPadCount) + ,signals + ,((UChar_t) col) + ,((UChar_t) row) + ,((UChar_t) time) + ,clusterTimeBin + ,clusterPosCol + ,volid); + cluster->SetInChamber(!out); + + // 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); + + RecPoints()->Add(cluster); + + // 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: Transform ok ? + + } // if: Maximum found ? + + } + + delete digitsOut; + + if (fAddLabels) + { + AddLabels(idet, firstClusterROC, nClusterROC); + } + + // Write the cluster and reset the array + WriteClusters(idet); + ResetRecPoints(); + + return kTRUE; } //_____________________________________________________________________________ -Double_t AliTRDclusterizer::CalcXposFromTimebin(Float_t timebin, Int_t idet - , Int_t col, Int_t row) +Bool_t AliTRDclusterizer::AddLabels(Int_t idet, Int_t firstClusterROC, Int_t nClusterROC) { // - // Calculates the local x position in the detector from the timebin, - // depends on the drift velocity and t0 + // Add the track indices to the found clusters // - AliTRDcalibDB *calibration = AliTRDcalibDB::Instance(); + const Int_t kNclus = 3; + const Int_t kNdict = AliTRDdigitsManager::kNDict; + const Int_t kNtrack = kNdict * kNclus; + + Int_t iClusterROC = 0; + + Int_t row = 0; + Int_t col = 0; + Int_t time = 0; + Int_t iPad = 0; + + // 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 + AliTRDdataArrayI *tracksIn = 0; + for (Int_t iDict = 0; iDict < kNdict; iDict++) { + + // tracksIn should be expanded beforehand! + tracksIn = (AliTRDdataArrayI *) fDigitsManager->GetDictionary(idet,iDict); + + // 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; + } + + } + + } + + // 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]); + + } + + delete [] idxTracks; + + return kTRUE; + +} + +//_____________________________________________________________________________ +Double_t AliTRDclusterizer::GetCOG(Double_t signal[5]) const +{ + // + // Get COG position + // Used for clusters with more than 3 pads - where LUT not applicable + // + + 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; + +} + +//_____________________________________________________________________________ +Double_t AliTRDclusterizer::Unfold(Double_t eps, Int_t plane, Double_t *padSignal) +{ + // + // Method to unfold neighbouring maxima. + // The charge ratio on the overlapping pad is calculated + // until there is no more change within the range given by eps. + // The resulting ratio is then returned to the calling method. + // + + AliTRDcalibDB *calibration = AliTRDcalibDB::Instance(); if (!calibration) { - AliError("Cannot find calibration object"); - return -1; + AliError("No AliTRDcalibDB instance available\n"); + return kFALSE; } - AliTRDCommonParam *parCom = AliTRDCommonParam::Instance(); - if (!parCom) { - AliError("Could not get common parameters\n"); - return kFALSE; + + Int_t irc = 0; + Int_t itStep = 0; // Count iteration steps + + Double_t ratio = 0.5; // Start value for ratio + Double_t prevRatio = 0.0; // Store previous ratio + + 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 the iteration + while ((TMath::Abs(prevRatio - ratio) > eps) && (itStep < 10)) { + + itStep++; + prevRatio = ratio; + + // 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]); + + // 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]; + + // Apply pad response to parameters + irc = calibration->PadResponse(ampLeft ,maxLeft ,plane,newLeftSignal ); + irc = calibration->PadResponse(ampRight,maxRight,plane,newRightSignal); + + // Calculate new overlapping ratio + ratio = TMath::Min((Double_t) 1.0 + ,newLeftSignal[2] / (newLeftSignal[2] + newRightSignal[0])); + } - Float_t vdrift = calibration->GetVdrift(idet,col,row); - Float_t t0 = calibration->GetT0(idet,col,row); - Float_t samplingFrequency = parCom->GetSamplingFrequency(); + return ratio; - timebin -= t0; +} - return timebin / samplingFrequency * vdrift; +//_____________________________________________________________________________ +void AliTRDclusterizer::TailCancelation(AliTRDdataArrayS *digitsIn + , AliTRDdataArrayF *digitsOut + , AliTRDSignalIndex *indexesIn + , AliTRDSignalIndex *indexesOut + , Int_t nTimeTotal + , Float_t adcThreshold + , AliTRDCalROC *calGainFactorROC + , Float_t calGainFactorDetValue) +{ + // + // Applies the tail cancelation and gain factors: + // Transform digitsIn to digitsOut + // + + Int_t iRow = 0; + Int_t iCol = 0; + Int_t iTime = 0; + + Double_t *inADC = new Double_t[nTimeTotal]; // ADC data before tail cancellation + Double_t *outADC = new Double_t[nTimeTotal]; // ADC data after tail cancellation + indexesIn->ResetCounters(); + while (indexesIn->NextRCIndex(iRow, iCol)) + { + Float_t calGainFactorROCValue = calGainFactorROC->GetValue(iCol,iRow); + Double_t gain = calGainFactorDetValue + * calGainFactorROCValue; + + for (iTime = 0; iTime < nTimeTotal; iTime++) + { + // Apply gain gain factor + inADC[iTime] = digitsIn->GetDataUnchecked(iRow,iCol,iTime); + inADC[iTime] /= gain; + outADC[iTime] = inADC[iTime]; + } + + // Apply the tail cancelation via the digital filter + if (AliTRDReconstructor::RecoParam()->TCOn()) + { + DeConvExp(inADC,outADC,nTimeTotal,AliTRDReconstructor::RecoParam()->GetTCnexp()); + } + + indexesIn->ResetTbinCounter(); + while (indexesIn->NextTbinIndex(iTime)) + { + // Store the amplitude of the digit if above threshold + if (outADC[iTime] > adcThreshold) + { + digitsOut->SetDataUnchecked(iRow,iCol,iTime,outADC[iTime]); + indexesOut->AddIndexTBin(iRow,iCol,iTime); + } + } // while itime + + } // while irow icol + + delete [] inADC; + delete [] outADC; + + return; + +} + +//_____________________________________________________________________________ +void AliTRDclusterizer::DeConvExp(Double_t *source, Double_t *target + , Int_t n, Int_t nexp) +{ + // + // Tail cancellation by deconvolution for PASA v4 TRF + // + + 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; + + rates[0] = TMath::Exp(-dt/(r1)); + rates[1] = TMath::Exp(-dt/(r2)); + + Int_t i = 0; + Int_t k = 0; + + Double_t reminder[2]; + Double_t correction = 0.0; + Double_t result = 0.0; + + // Attention: computation order is important + 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]; + } + + } } @@ -362,7 +1232,7 @@ void AliTRDclusterizer::ResetRecPoints() } //_____________________________________________________________________________ -TObjArray* AliTRDclusterizer::RecPoints() +TObjArray *AliTRDclusterizer::RecPoints() { // // Returns the list of rec points @@ -375,3 +1245,201 @@ TObjArray* AliTRDclusterizer::RecPoints() return fRecPoints; } + +//_____________________________________________________________________________ +void AliTRDclusterizer::FillLUT() +{ + // + // Create the LUT + // + + const Int_t kNlut = 128; + + fLUTbin = AliTRDgeometry::kNplan * kNlut; + + // The lookup table from Bogdan + Float_t lut[AliTRDgeometry::kNplan][kNlut] = { + { + 0.0070, 0.0150, 0.0224, 0.0298, 0.0374, 0.0454, 0.0533, 0.0611, + 0.0684, 0.0755, 0.0827, 0.0900, 0.0975, 0.1049, 0.1120, 0.1187, + 0.1253, 0.1318, 0.1385, 0.1453, 0.1519, 0.1584, 0.1646, 0.1704, + 0.1762, 0.1821, 0.1879, 0.1938, 0.1996, 0.2053, 0.2108, 0.2160, + 0.2210, 0.2260, 0.2310, 0.2361, 0.2411, 0.2461, 0.2509, 0.2557, + 0.2602, 0.2646, 0.2689, 0.2732, 0.2774, 0.2816, 0.2859, 0.2901, + 0.2942, 0.2983, 0.3022, 0.3061, 0.3099, 0.3136, 0.3172, 0.3207, + 0.3242, 0.3278, 0.3312, 0.3347, 0.3382, 0.3416, 0.3450, 0.3483, + 0.3515, 0.3547, 0.3579, 0.3609, 0.3639, 0.3669, 0.3698, 0.3727, + 0.3756, 0.3785, 0.3813, 0.3842, 0.3870, 0.3898, 0.3926, 0.3952, + 0.3979, 0.4005, 0.4032, 0.4057, 0.4082, 0.4108, 0.4132, 0.4157, + 0.4181, 0.4205, 0.4228, 0.4252, 0.4275, 0.4299, 0.4322, 0.4345, + 0.4367, 0.4390, 0.4412, 0.4434, 0.4456, 0.4478, 0.4499, 0.4520, + 0.4541, 0.4562, 0.4583, 0.4603, 0.4623, 0.4643, 0.4663, 0.4683, + 0.4702, 0.4722, 0.4741, 0.4758, 0.4774, 0.4790, 0.4805, 0.4824, + 0.4844, 0.4863, 0.4883, 0.4902, 0.4921, 0.4940, 0.4959, 0.4978 + }, + { + 0.0072, 0.0156, 0.0235, 0.0313, 0.0394, 0.0478, 0.0561, 0.0642, + 0.0718, 0.0792, 0.0868, 0.0947, 0.1025, 0.1101, 0.1172, 0.1241, + 0.1309, 0.1378, 0.1449, 0.1518, 0.1586, 0.1650, 0.1710, 0.1770, + 0.1830, 0.1891, 0.1952, 0.2011, 0.2070, 0.2125, 0.2177, 0.2229, + 0.2280, 0.2332, 0.2383, 0.2435, 0.2484, 0.2533, 0.2581, 0.2627, + 0.2670, 0.2714, 0.2757, 0.2799, 0.2842, 0.2884, 0.2927, 0.2968, + 0.3008, 0.3048, 0.3086, 0.3123, 0.3159, 0.3195, 0.3231, 0.3266, + 0.3301, 0.3335, 0.3370, 0.3404, 0.3438, 0.3471, 0.3504, 0.3536, + 0.3567, 0.3598, 0.3628, 0.3657, 0.3686, 0.3715, 0.3744, 0.3772, + 0.3800, 0.3828, 0.3856, 0.3884, 0.3911, 0.3938, 0.3965, 0.3991, + 0.4016, 0.4042, 0.4067, 0.4092, 0.4116, 0.4140, 0.4164, 0.4187, + 0.4211, 0.4234, 0.4257, 0.4280, 0.4302, 0.4325, 0.4347, 0.4369, + 0.4391, 0.4413, 0.4434, 0.4456, 0.4477, 0.4497, 0.4518, 0.4538, + 0.4558, 0.4578, 0.4598, 0.4618, 0.4637, 0.4656, 0.4675, 0.4694, + 0.4713, 0.4732, 0.4750, 0.4766, 0.4781, 0.4797, 0.4813, 0.4832, + 0.4851, 0.4870, 0.4888, 0.4906, 0.4925, 0.4942, 0.4960, 0.4978 + }, + { + 0.0075, 0.0163, 0.0246, 0.0328, 0.0415, 0.0504, 0.0592, 0.0674, + 0.0753, 0.0832, 0.0914, 0.0996, 0.1077, 0.1154, 0.1225, 0.1296, + 0.1369, 0.1442, 0.1515, 0.1585, 0.1652, 0.1714, 0.1776, 0.1839, + 0.1902, 0.1965, 0.2025, 0.2085, 0.2141, 0.2194, 0.2247, 0.2299, + 0.2352, 0.2405, 0.2457, 0.2507, 0.2557, 0.2604, 0.2649, 0.2693, + 0.2737, 0.2780, 0.2823, 0.2867, 0.2909, 0.2951, 0.2992, 0.3033, + 0.3072, 0.3110, 0.3146, 0.3182, 0.3218, 0.3253, 0.3288, 0.3323, + 0.3357, 0.3392, 0.3426, 0.3459, 0.3492, 0.3524, 0.3555, 0.3586, + 0.3616, 0.3645, 0.3674, 0.3703, 0.3731, 0.3759, 0.3787, 0.3815, + 0.3843, 0.3870, 0.3897, 0.3925, 0.3950, 0.3976, 0.4002, 0.4027, + 0.4052, 0.4076, 0.4101, 0.4124, 0.4148, 0.4171, 0.4194, 0.4217, + 0.4239, 0.4262, 0.4284, 0.4306, 0.4328, 0.4350, 0.4371, 0.4393, + 0.4414, 0.4435, 0.4455, 0.4476, 0.4496, 0.4516, 0.4536, 0.4555, + 0.4575, 0.4594, 0.4613, 0.4632, 0.4650, 0.4669, 0.4687, 0.4705, + 0.4723, 0.4741, 0.4758, 0.4773, 0.4789, 0.4804, 0.4821, 0.4839, + 0.4857, 0.4875, 0.4893, 0.4910, 0.4928, 0.4945, 0.4961, 0.4978 + }, + { + 0.0078, 0.0171, 0.0258, 0.0345, 0.0438, 0.0532, 0.0624, 0.0708, + 0.0791, 0.0875, 0.0962, 0.1048, 0.1130, 0.1206, 0.1281, 0.1356, + 0.1432, 0.1508, 0.1582, 0.1651, 0.1716, 0.1780, 0.1845, 0.1910, + 0.1975, 0.2038, 0.2099, 0.2155, 0.2210, 0.2263, 0.2317, 0.2371, + 0.2425, 0.2477, 0.2528, 0.2578, 0.2626, 0.2671, 0.2715, 0.2759, + 0.2803, 0.2846, 0.2890, 0.2933, 0.2975, 0.3016, 0.3056, 0.3095, + 0.3132, 0.3168, 0.3204, 0.3239, 0.3274, 0.3309, 0.3344, 0.3378, + 0.3412, 0.3446, 0.3479, 0.3511, 0.3543, 0.3574, 0.3603, 0.3633, + 0.3662, 0.3690, 0.3718, 0.3747, 0.3774, 0.3802, 0.3829, 0.3857, + 0.3883, 0.3910, 0.3936, 0.3962, 0.3987, 0.4012, 0.4037, 0.4061, + 0.4085, 0.4109, 0.4132, 0.4155, 0.4177, 0.4200, 0.4222, 0.4244, + 0.4266, 0.4288, 0.4309, 0.4331, 0.4352, 0.4373, 0.4394, 0.4414, + 0.4435, 0.4455, 0.4475, 0.4494, 0.4514, 0.4533, 0.4552, 0.4571, + 0.4590, 0.4608, 0.4626, 0.4645, 0.4662, 0.4680, 0.4698, 0.4715, + 0.4733, 0.4750, 0.4766, 0.4781, 0.4796, 0.4812, 0.4829, 0.4846, + 0.4863, 0.4880, 0.4897, 0.4914, 0.4930, 0.4946, 0.4963, 0.4979 + }, + { + 0.0081, 0.0178, 0.0270, 0.0364, 0.0463, 0.0562, 0.0656, 0.0744, + 0.0831, 0.0921, 0.1013, 0.1102, 0.1183, 0.1261, 0.1339, 0.1419, + 0.1499, 0.1576, 0.1648, 0.1715, 0.1782, 0.1849, 0.1917, 0.1984, + 0.2048, 0.2110, 0.2167, 0.2223, 0.2278, 0.2333, 0.2389, 0.2444, + 0.2497, 0.2548, 0.2598, 0.2645, 0.2691, 0.2735, 0.2780, 0.2824, + 0.2868, 0.2912, 0.2955, 0.2997, 0.3038, 0.3078, 0.3116, 0.3152, + 0.3188, 0.3224, 0.3259, 0.3294, 0.3329, 0.3364, 0.3398, 0.3432, + 0.3465, 0.3497, 0.3529, 0.3561, 0.3591, 0.3620, 0.3649, 0.3677, + 0.3705, 0.3733, 0.3761, 0.3788, 0.3816, 0.3843, 0.3869, 0.3896, + 0.3922, 0.3948, 0.3973, 0.3998, 0.4022, 0.4047, 0.4070, 0.4094, + 0.4117, 0.4139, 0.4162, 0.4184, 0.4206, 0.4227, 0.4249, 0.4270, + 0.4291, 0.4313, 0.4334, 0.4354, 0.4375, 0.4395, 0.4415, 0.4435, + 0.4455, 0.4474, 0.4493, 0.4512, 0.4531, 0.4550, 0.4568, 0.4586, + 0.4604, 0.4622, 0.4639, 0.4657, 0.4674, 0.4691, 0.4708, 0.4725, + 0.4742, 0.4758, 0.4773, 0.4788, 0.4803, 0.4819, 0.4836, 0.4852, + 0.4869, 0.4885, 0.4901, 0.4917, 0.4933, 0.4948, 0.4964, 0.4979 + }, + { + 0.0085, 0.0189, 0.0288, 0.0389, 0.0497, 0.0603, 0.0699, 0.0792, + 0.0887, 0.0985, 0.1082, 0.1170, 0.1253, 0.1336, 0.1421, 0.1505, + 0.1587, 0.1662, 0.1733, 0.1803, 0.1874, 0.1945, 0.2014, 0.2081, + 0.2143, 0.2201, 0.2259, 0.2316, 0.2374, 0.2431, 0.2487, 0.2541, + 0.2593, 0.2642, 0.2689, 0.2735, 0.2781, 0.2826, 0.2872, 0.2917, + 0.2961, 0.3003, 0.3045, 0.3086, 0.3125, 0.3162, 0.3198, 0.3235, + 0.3270, 0.3306, 0.3342, 0.3377, 0.3411, 0.3446, 0.3479, 0.3511, + 0.3543, 0.3575, 0.3605, 0.3634, 0.3663, 0.3691, 0.3720, 0.3748, + 0.3775, 0.3803, 0.3830, 0.3857, 0.3884, 0.3911, 0.3937, 0.3962, + 0.3987, 0.4012, 0.4036, 0.4060, 0.4084, 0.4107, 0.4129, 0.4152, + 0.4174, 0.4196, 0.4218, 0.4239, 0.4261, 0.4282, 0.4303, 0.4324, + 0.4344, 0.4365, 0.4385, 0.4405, 0.4425, 0.4445, 0.4464, 0.4483, + 0.4502, 0.4521, 0.4539, 0.4558, 0.4576, 0.4593, 0.4611, 0.4629, + 0.4646, 0.4663, 0.4680, 0.4697, 0.4714, 0.4730, 0.4747, 0.4759, + 0.4769, 0.4780, 0.4790, 0.4800, 0.4811, 0.4827, 0.4843, 0.4859, + 0.4874, 0.4889, 0.4905, 0.4920, 0.4935, 0.4950, 0.4965, 0.4979 + } + }; + + if (fLUT) { + delete [] fLUT; + } + fLUT = new Double_t[fLUTbin]; + + for (Int_t iplan = 0; iplan < AliTRDgeometry::kNplan; iplan++) { + for (Int_t ilut = 0; ilut < kNlut; ilut++ ) { + fLUT[iplan*kNlut+ilut] = lut[iplan][ilut]; + } + } + +} + +//_____________________________________________________________________________ +Double_t AliTRDclusterizer::LUTposition(Int_t iplane, Double_t ampL + , Double_t ampC, Double_t ampR) const +{ + // + // Calculates the cluster position using the lookup table. + // Method provided by Bogdan Vulpescu. + // + + const Int_t kNlut = 128; + + Double_t pos; + Double_t x = 0.0; + Double_t xmin; + Double_t xmax; + Double_t xwid; + + Int_t side = 0; + Int_t ix; + + Double_t xMin[AliTRDgeometry::kNplan] = { 0.006492, 0.006377, 0.006258 + , 0.006144, 0.006030, 0.005980 }; + Double_t xMax[AliTRDgeometry::kNplan] = { 0.960351, 0.965870, 0.970445 + , 0.974352, 0.977667, 0.996101 }; + + if (ampL > ampR) { + x = (ampL - ampR) / ampC; + side = -1; + } + else if (ampL < ampR) { + x = (ampR - ampL) / ampC; + side = +1; + } + + if (ampL != ampR) { + + xmin = xMin[iplane] + 0.000005; + xmax = xMax[iplane] - 0.000005; + xwid = (xmax - xmin) / 127.0; + + if (x < xmin) { + pos = 0.0000; + } + else if (x > xmax) { + pos = side * 0.5000; + } + else { + ix = (Int_t) ((x - xmin) / xwid); + pos = side * fLUT[iplane*kNlut+ix]; + } + + } + else { + + pos = 0.0; + + } + + return pos; + +}