X-Git-Url: http://git.uio.no/git/?a=blobdiff_plain;f=HLT%2FTPCLib%2FAliHLTTPCClusterFinder.cxx;h=d582f80682239d42f93aededa1c92d254dd78066;hb=345e2385f06b5561fbdb0b061d729cefcfe240bd;hp=e171e52cea8f8bd05defe5e055636d81acd14e18;hpb=025ba4dc4c01b4418717a1cb7abd8692658e18bb;p=u%2Fmrichter%2FAliRoot.git diff --git a/HLT/TPCLib/AliHLTTPCClusterFinder.cxx b/HLT/TPCLib/AliHLTTPCClusterFinder.cxx index e171e52cea8..d582f806822 100644 --- a/HLT/TPCLib/AliHLTTPCClusterFinder.cxx +++ b/HLT/TPCLib/AliHLTTPCClusterFinder.cxx @@ -1,5 +1,4 @@ -// @(#) $Id$ -// Original: AliHLTClustFinderNew.cxx,v 1.29 2005/06/14 10:55:21 cvetan Exp +// $Id$ //************************************************************************** //* This file is property of and copyright by the ALICE HLT Project * @@ -19,21 +18,23 @@ //* provided "as is" without express or implied warranty. * //************************************************************************** -/** @file AliHLTTPCClusterFinder.cxx - @author Kenneth Aamodt, Kalliopi Kanaki - @date - @brief Cluster Finder for the TPC -*/ +// @file AliHLTTPCClusterFinder.cxx +// @author Kenneth Aamodt, Kalliopi Kanaki +// @date +// @brief Cluster Finder for the TPC +// @note #include "AliHLTTPCDigitReader.h" -#include "AliHLTTPCRootTypes.h" #include "AliHLTTPCLogging.h" #include "AliHLTTPCClusterFinder.h" #include "AliHLTTPCSpacePointData.h" -#include "AliHLTTPCMemHandler.h" #include "AliHLTTPCPad.h" #include #include +#include +#include "AliTPCcalibDB.h" +#include "AliTPCTransform.h" +#include "AliTPCParam.h" #if __GNUC__ >= 3 using namespace std; @@ -53,7 +54,7 @@ AliHLTTPCClusterFinder::AliHLTTPCClusterFinder() fDeconvPad(kFALSE), fStdout(kFALSE), fCalcerr(kTRUE), - fRawSP(kFALSE), + fFillRawClusters(kFALSE), fFirstRow(0), fLastRow(0), fCurrentRow(0), @@ -71,6 +72,7 @@ AliHLTTPCClusterFinder::AliHLTTPCClusterFinder() fClusters(), fClustersMCInfo(), fMCDigits(), + fRawClusters(), fNumberOfPadsInRow(NULL), fNumberOfRows(0), fRowOfFirstCandidate(0), @@ -81,9 +83,33 @@ AliHLTTPCClusterFinder::AliHLTTPCClusterFinder() fChargeOfCandidatesFalling(kFALSE), f32BitFormat(kFALSE), fDoMC(kFALSE), - fClusterMCVector() + fClusterMCVector(), + fOfflineTransform(NULL), + fOfflineTPCParam( NULL ), + fOfflineTPCRecoParam(*AliTPCRecoParam::GetHLTParam()), + fTimeMeanDiff(2), + fReleaseMemory(0) { //constructor + + //uptate the transform class + + fOfflineTransform = AliTPCcalibDB::Instance()->GetTransform(); + if(!fOfflineTransform){ + HLTError("AliHLTTPCClusterFinder()::UpdateCAlibDB:: Offline transform not in AliTPCcalibDB."); + } + else{ + fOfflineTransform->SetCurrentRecoParam(&fOfflineTPCRecoParam); + } + + fOfflineTPCParam = AliTPCcalibDB::Instance()->GetParameters(); + if( !fOfflineTPCParam ){ + HLTError("AliHLTTPCClusterFinder()::UpdateCAlibDB:: Offline TPC parameters not in AliTPCcalibDB."); + } else { + fOfflineTPCParam->Update(); + fOfflineTPCParam->ReadGeoMatrices(); + } + } AliHLTTPCClusterFinder::~AliHLTTPCClusterFinder(){ @@ -109,6 +135,12 @@ void AliHLTTPCClusterFinder::InitSlice(Int_t slice,Int_t patch,Int_t nmaxpoints) fCurrentPatch = patch; fFirstRow=AliHLTTPCTransform::GetFirstRow(patch); fLastRow=AliHLTTPCTransform::GetLastRow(patch); + + fClusters.clear(); + fClustersMCInfo.clear(); + fMCDigits.clear(); + fClusterMCVector.clear(); + fRawClusters.clear(); } void AliHLTTPCClusterFinder::InitializePadArray(){ @@ -129,6 +161,8 @@ void AliHLTTPCClusterFinder::InitializePadArray(){ memset( fNumberOfPadsInRow, 0, sizeof(Int_t)*(fNumberOfRows)); + fRowPadVector.clear(); + for(UInt_t i=0;iGetBunchSize(); + indexInBunchData = fDigitReader->GetBunchSize()-1; const UShort_t *bunchData= fDigitReader->GetSignalsShort(); + do{ AliHLTTPCClusters candidate; //for(Int_t i=indexInBunchData;iGetBunchSize();i++){ for(Int_t i=indexInBunchData;i>=0;i--){ + // Checks if one need to deconvolute the signals if(bunchData[i]>prevSignal && signalFalling==kTRUE){ if(iGetBunchSize()-1){ // means there are more than one signal left in the bunch @@ -337,7 +377,6 @@ void AliHLTTPCClusterFinder::ReadDataUnsortedDeconvoluteTime(void* ptr,unsigned if(bunchData[i]>candidate.fQMax){ candidate.fQMax=bunchData[i]; } - prevSignal=bunchData[i]; time++; indexInBunchData--; @@ -349,8 +388,7 @@ void AliHLTTPCClusterFinder::ReadDataUnsortedDeconvoluteTime(void* ptr,unsigned candidate.fLastMergedPad=pad; candidate.fRowNumber=row+fDigitReader->GetRowOffset(); } - fRowPadVector[row][pad]->AddClusterCandidate(candidate); - fRowPadVector[row][pad]->AddCandidateDigits(fMCDigits); + fRowPadVector[row][pad]->AddClusterCandidate(candidate); if(indexInBunchDataGetBunchSize()-1){ moreDataInBunch=kFALSE; } @@ -360,6 +398,9 @@ void AliHLTTPCClusterFinder::ReadDataUnsortedDeconvoluteTime(void* ptr,unsigned const UInt_t *bunchData= fDigitReader->GetSignals(); do{ AliHLTTPCClusters candidate; + const AliHLTTPCDigitData* digits = fDigitReader->GetBunchDigits(); + if(fDoMC) fMCDigits.clear(); + for(Int_t i=indexInBunchData;iGetBunchSize();i++){ // Checks if one need to deconvolute the signals if(bunchData[i]>prevSignal && signalFalling==kTRUE){ @@ -386,6 +427,7 @@ void AliHLTTPCClusterFinder::ReadDataUnsortedDeconvoluteTime(void* ptr,unsigned if(bunchData[i]>candidate.fQMax){ candidate.fQMax=bunchData[i]; } + if( fDoMC ) fMCDigits.push_back(digits[i]); prevSignal=bunchData[i]; time++; @@ -399,6 +441,10 @@ void AliHLTTPCClusterFinder::ReadDataUnsortedDeconvoluteTime(void* ptr,unsigned candidate.fRowNumber=row+fDigitReader->GetRowOffset(); } fRowPadVector[row][pad]->AddClusterCandidate(candidate); + if(fDoMC){ + fRowPadVector[row][pad]->AddCandidateDigits(fMCDigits); + fMCDigits.clear(); + } if(indexInBunchDataGetBunchSize()-1){ moreDataInBunch=kFALSE; } @@ -415,8 +461,13 @@ Bool_t AliHLTTPCClusterFinder::ComparePads(AliHLTTPCPad *nextPad,AliHLTTPCCluste //Checking if we have a match on the next pad for(UInt_t candidateNumber=0;candidateNumberfClusterCandidates.size();candidateNumber++){ + if(nextPad->fUsedClusterCandidates[candidateNumber] == 1){ + continue; + } AliHLTTPCClusters *candidate =&nextPad->fClusterCandidates[candidateNumber]; - if(cluster->fMean-candidate->fMean==1 || candidate->fMean-cluster->fMean==1 || cluster->fMean-candidate->fMean==0){ + // if(cluster->fMean-candidate->fMean==1 || candidate->fMean-cluster->fMean==1 || cluster->fMean-candidate->fMean==0){ + + if( abs((Int_t)(cluster->fMean - candidate->fMean)) <= fTimeMeanDiff ){ if(fDeconvPad){ if(candidate->fTotalChargesizeInByte) { + HLTError("not enough space to write raw clusters"); + return 0; + } + memcpy(rawClusters, &fRawClusters[0], fRawClusters.size()*sizeof(AliHLTTPCRawCluster)); + return fRawClusters.size(); +} + void AliHLTTPCClusterFinder::FindClusters(){ // see header file for function documentation @@ -535,10 +594,10 @@ void AliHLTTPCClusterFinder::FindClusters(){ fClusters.push_back(*tmpCandidate); if(fDoMC){ //sort the vector (large->small) according to weight and remove elements above 2 (keep 0 1 and 2) - //sort(fClusterMCVector,fClusterMCVector.size(), MCWeight::CompareWeights ); - ClusterMCInfo tmpClusterMCInfo; + sort(fClusterMCVector.begin(),fClusterMCVector.end(), CompareWeights ); + AliHLTTPCClusterMCLabel tmpClusterMCInfo; - MCWeight zeroMC; + AliHLTTPCClusterMCWeight zeroMC; zeroMC.fMCID=-1; zeroMC.fWeight=0; @@ -596,9 +655,40 @@ void AliHLTTPCClusterFinder::FindClusters(){ WriteClusters(fClusters.size(),clusterlist); delete [] clusterlist; fClusters.clear(); + if( fReleaseMemory ) DeInitializePadArray();// call this when the -releaseMemory flag is set } +Bool_t AliHLTTPCClusterFinder::UpdateCalibDB(){ + + //update the db + AliTPCcalibDB::Instance()->Update(); + + Bool_t ret = 1; + + //uptate the transform class + + fOfflineTransform = AliTPCcalibDB::Instance()->GetTransform(); + if(!fOfflineTransform){ + HLTError("AliHLTTPCClusterFinder()::UpdateCAlibDB:: Offline transform not in AliTPCcalibDB."); + ret = 0; + } + else{ + fOfflineTransform->SetCurrentRecoParam(&fOfflineTPCRecoParam); + } + + fOfflineTPCParam = AliTPCcalibDB::Instance()->GetParameters(); + if( !fOfflineTPCParam ){ + HLTError("AliHLTTPCClusterFinder()::UpdateCAlibDB:: Offline TPC parameters not in AliTPCcalibDB."); + ret = 0; + } else { + fOfflineTPCParam->Update(); + fOfflineTPCParam->ReadGeoMatrices(); + } + + return ret; +} + //---------------------------------- Under this line the old sorted clusterfinder functions can be found -------------------------------- @@ -617,16 +707,17 @@ void AliHLTTPCClusterFinder::PrintClusters(){ } } -void AliHLTTPCClusterFinder::FillMCClusterVector(vector digitData){ - - for(UInt_t d=0;d=0) + (digitData.at(d).fTrackID[1]>=0) + (digitData.at(d).fTrackID[2]>=0); +void AliHLTTPCClusterFinder::FillMCClusterVector(vector *digitData){ + // see header file for class documentation + if( !digitData ) return; + for(UInt_t d=0;dsize();d++){ + Int_t nIDsInDigit = (digitData->at(d).fTrackID[0]>=0) + (digitData->at(d).fTrackID[1]>=0) + (digitData->at(d).fTrackID[2]>=0); for(Int_t id=0; id<3; id++){ - if(digitData.at(d).fTrackID[id]>=0){ + if(digitData->at(d).fTrackID[id]>=0){ Bool_t matchFound = kFALSE; - MCWeight mc; - mc.fMCID = digitData.at(d).fTrackID[id]; - mc.fWeight = ((Float_t)digitData.at(d).fCharge)/nIDsInDigit; + AliHLTTPCClusterMCWeight mc; + mc.fMCID = digitData->at(d).fTrackID[id]; + mc.fWeight = ((Float_t)digitData->at(d).fCharge)/nIDsInDigit; for(UInt_t i=0;i=0 && j+fClustersMCInfo.size()-nclusters < fClustersMCInfo.size()){ + fClustersMCInfo.erase(fClustersMCInfo.begin()+j+fClustersMCInfo.size()-nclusters); // remove the mc info for this cluster since it is not taken into account + } } continue; //discard single pad clusters } if(list[j].fTotalCharge < fThreshold){ if(fDoMC){ - fClustersMCInfo.erase(fClustersMCInfo.begin()+j); // remove the mc ifo for this cluster since it is not taken into account + if(j+(Int_t)fClustersMCInfo.size()-nclusters >=0 && j+fClustersMCInfo.size()-nclusters < fClustersMCInfo.size()){ + fClustersMCInfo.erase(fClustersMCInfo.begin()+j+fClustersMCInfo.size()-nclusters); // remove the mc info for this cluster since it is not taken into account + } } continue; //noise cluster } @@ -1042,20 +1141,25 @@ void AliHLTTPCClusterFinder::WriteClusters(Int_t nclusters,AliClusterData *list) UInt_t q2=list[j].fTotalCharge*list[j].fTotalCharge; // Float_t sy2=list[j].fPad2 * list[j].fTotalCharge - list[j].fPad * list[j].fPad; Float_t sy2=(Float_t)list[j].fPad2 * list[j].fTotalCharge - (Float_t)list[j].fPad * list[j].fPad; + if(q2 == 0) { + LOG(AliHLTTPCLog::kError,"AliHLTTPCClusterFinder::WriteClusters","Cluster width") + <<"zero charge "<< list[j].fTotalCharge <<" on row "<(unsigned)counter) { + fRawClusters[counter].SetPadRow(fCurrentRow); + fRawClusters[counter].SetPad(fpad); + fRawClusters[counter].SetTime(ftime); + } + { AliHLTTPCTransform::Slice2Sector(fCurrentSlice,fCurrentRow,thissector,thisrow); - AliHLTTPCTransform::Raw2Local(xyz,thissector,thisrow,fpad,ftime); - - if(xyz[0]==0) LOG(AliHLTTPCLog::kError,"AliHLTTPCClustFinder","Cluster Finder") - <= fMaxNClusters) - { - LOG(AliHLTTPCLog::kError,"AliHLTTPCClustFinder::WriteClusters","Cluster Finder") - <= fMaxNClusters) + { + LOG(AliHLTTPCLog::kError,"AliHLTTPCClustFinder::WriteClusters","Cluster Finder") + <Transform(x,iSector,0,1); + double y[3] = {x[0], x[1], x[2] }; + + if( fOfflineTPCParam && thissectorGetNSector() ){ + TGeoHMatrix *alignment = fOfflineTPCParam->GetClusterMatrix( thissector ); + if ( alignment ) alignment->LocalToMaster( x, y); + } + + fSpacePointData[counter].fX = y[0]; + fSpacePointData[counter].fY = y[1]; + fSpacePointData[counter].fZ = y[2]; } - fSpacePointData[counter].fZ = xyz[2]; - } else { - fSpacePointData[counter].fX = fCurrentRow; - fSpacePointData[counter].fY = fpad; - fSpacePointData[counter].fZ = ftime; - } + } fSpacePointData[counter].fCharge = list[j].fTotalCharge; fSpacePointData[counter].fPadRow = fCurrentRow; @@ -1114,13 +1237,19 @@ void AliHLTTPCClusterFinder::WriteClusters(Int_t nclusters,AliClusterData *list) fSpacePointData[counter].fQMax = list[j].fQMax; - fSpacePointData[counter].fUsed = kFALSE; // only used / set in AliHLTTPCDisplay - fSpacePointData[counter].fTrackN = -1; // only used / set in AliHLTTPCDisplay + fSpacePointData[counter].SetUsed(kFALSE); // only used / set in AliHLTTPCDisplay + fSpacePointData[counter].SetTrackNumber(-1); // only used / set in AliHLTTPCDisplay Int_t patch=fCurrentPatch; if(patch==-1) patch=0; //never store negative patch number - fSpacePointData[counter].fID = counter - +((fCurrentSlice&0x7f)<<25)+((patch&0x7)<<22);//Uli + fSpacePointData[counter].SetID( fCurrentSlice, patch, counter ); + + if (fFillRawClusters && fRawClusters.size()>(unsigned)counter) { + fRawClusters[counter].SetSigmaY2(fSpacePointData[counter].fSigmaY2); + fRawClusters[counter].SetSigmaZ2(fSpacePointData[counter].fSigmaZ2); + fRawClusters[counter].SetCharge(fSpacePointData[counter].fCharge); + fRawClusters[counter].SetQMax(fSpacePointData[counter].fQMax); + } #ifdef do_mc Int_t trackID[3]; @@ -1140,7 +1269,7 @@ void AliHLTTPCClusterFinder::WriteClusters(Int_t nclusters,AliClusterData *list) // STILL TO FIX ---------------------------------------------------------------------------- #ifdef do_mc -void AliHLTTPCClusterFinder::GetTrackID(Int_t pad,Int_t time,Int_t *trackID){ +void AliHLTTPCClusterFinder::GetTrackID(Int_t pad,Int_t time,Int_t *trackID) const { // see header file for class documentation //get mc id @@ -1178,6 +1307,10 @@ void AliHLTTPCClusterFinder::WriteClusters(Int_t nclusters,AliHLTTPCClusters *li //write cluster to output pointer Int_t thisrow,thissector; UInt_t counter = fNClusters; + + if (fFillRawClusters) { + fRawClusters.resize(nclusters); + } for(int j=0; j(unsigned)counter) { + fRawClusters[counter].SetPadRow(fCurrentRow); + fRawClusters[counter].SetPad(fpad); + fRawClusters[counter].SetTime(ftime); + } + { AliHLTTPCTransform::Slice2Sector(fCurrentSlice,fCurrentRow,thissector,thisrow); AliHLTTPCTransform::Raw2Local(xyz,thissector,thisrow,fpad,ftime); @@ -1253,10 +1391,6 @@ void AliHLTTPCClusterFinder::WriteClusters(Int_t nclusters,AliHLTTPCClusters *li } fSpacePointData[counter].fZ = xyz[2]; - } else { - fSpacePointData[counter].fX = fCurrentRow; - fSpacePointData[counter].fY = fpad; - fSpacePointData[counter].fZ = ftime; } fSpacePointData[counter].fCharge = list[j].fTotalCharge; @@ -1266,13 +1400,19 @@ void AliHLTTPCClusterFinder::WriteClusters(Int_t nclusters,AliHLTTPCClusters *li fSpacePointData[counter].fQMax = list[j].fQMax; - fSpacePointData[counter].fUsed = kFALSE; // only used / set in AliHLTTPCDisplay - fSpacePointData[counter].fTrackN = -1; // only used / set in AliHLTTPCDisplay + fSpacePointData[counter].SetUsed(kFALSE); // only used / set in AliHLTTPCDisplay + fSpacePointData[counter].SetTrackNumber(-1); // only used / set in AliHLTTPCDisplay Int_t patch=fCurrentPatch; if(patch==-1) patch=0; //never store negative patch number - fSpacePointData[counter].fID = counter - +((fCurrentSlice&0x7f)<<25)+((patch&0x7)<<22);//Uli + fSpacePointData[counter].SetID( fCurrentSlice, patch, counter ); + + if (fFillRawClusters && fRawClusters.size()>(unsigned)counter) { + fRawClusters[counter].SetSigmaY2(fSpacePointData[counter].fSigmaY2); + fRawClusters[counter].SetSigmaZ2(fSpacePointData[counter].fSigmaZ2); + fRawClusters[counter].SetCharge(fSpacePointData[counter].fCharge); + fRawClusters[counter].SetQMax(fSpacePointData[counter].fQMax); + } #ifdef do_mc Int_t trackID[3];