1 /**************************************************************************
2 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
4 * Author: The ALICE Off-line Project. *
5 * Contributors are mentioned in the code where appropriate. *
7 * Permission to use, copy, modify and distribute this software and its *
8 * documentation strictly for non-commercial purposes is hereby granted *
9 * without fee, provided that the above copyright notice appears in all *
10 * copies and that both the copyright notice and this permission notice *
11 * appear in the supporting documentation. The authors make no claims *
12 * about the suitability of this software for any purpose. It is *
13 * provided "as is" without express or implied warranty. *
14 **************************************************************************/
18 ///////////////////////////////////////////////////////////////////////////////
20 // TRD cluster finder //
22 ///////////////////////////////////////////////////////////////////////////////
24 #include <TClonesArray.h>
25 #include <TObjArray.h>
27 #include "AliRunLoader.h"
28 #include "AliLoader.h"
29 #include "AliTreeLoader.h"
30 #include "AliAlignObj.h"
32 #include "AliTRDclusterizer.h"
33 #include "AliTRDcluster.h"
34 #include "AliTRDReconstructor.h"
35 #include "AliTRDgeometry.h"
36 #include "AliTRDarrayDictionary.h"
37 #include "AliTRDarrayADC.h"
38 #include "AliTRDdigitsManager.h"
39 #include "AliTRDdigitsParam.h"
40 #include "AliTRDrawData.h"
41 #include "AliTRDcalibDB.h"
42 #include "AliTRDtransform.h"
43 #include "AliTRDSignalIndex.h"
44 #include "AliTRDrawStream.h"
45 #include "AliTRDfeeParam.h"
46 #include "AliTRDtrackletWord.h"
47 #include "AliTRDtrackletMCM.h"
48 #include "AliTRDtrackGTU.h"
49 #include "AliESDTrdTrack.h"
51 #include "TTreeStream.h"
53 #include "Cal/AliTRDCalROC.h"
54 #include "Cal/AliTRDCalDet.h"
55 #include "Cal/AliTRDCalSingleChamberStatus.h"
56 #include "Cal/AliTRDCalOnlineGainTableROC.h"
58 ClassImp(AliTRDclusterizer)
60 //_____________________________________________________________________________
61 AliTRDclusterizer::AliTRDclusterizer(const AliTRDReconstructor *const rec)
70 ,fDigitsManager(new AliTRDdigitsManager())
71 ,fTrackletContainer(NULL)
73 ,fTransform(new AliTRDtransform(0))
81 ,fMinLeftRightCutSigma(0)
87 ,fCalGainFactorROC(NULL)
88 ,fCalGainFactorDetValue(0)
91 ,fCalPadStatusROC(NULL)
101 // AliTRDclusterizer default constructor
104 SetBit(kLabels, kTRUE);
105 SetBit(knewDM, kFALSE);
107 fRawVersion = AliTRDfeeParam::Instance()->GetRAWversion();
109 // Initialize debug stream
111 if(fReconstructor->GetRecoParam()->GetStreamLevel(AliTRDrecoParam::kClusterizer) > 1){
112 TDirectory *savedir = gDirectory;
113 //fgGetDebugStream = new TTreeSRedirector("TRD.ClusterizerDebug.root");
120 //_____________________________________________________________________________
121 AliTRDclusterizer::AliTRDclusterizer(const Text_t *name
122 , const Text_t *title
123 , const AliTRDReconstructor *const rec)
132 ,fDigitsManager(new AliTRDdigitsManager())
133 ,fTrackletContainer(NULL)
135 ,fTransform(new AliTRDtransform(0))
143 ,fMinLeftRightCutSigma(0)
149 ,fCalGainFactorROC(NULL)
150 ,fCalGainFactorDetValue(0)
152 ,fCalNoiseDetValue(0)
153 ,fCalPadStatusROC(NULL)
154 ,fCalOnlGainROC(NULL)
163 // AliTRDclusterizer constructor
166 SetBit(kLabels, kTRUE);
167 SetBit(knewDM, kFALSE);
169 fDigitsManager->CreateArrays();
171 fRawVersion = AliTRDfeeParam::Instance()->GetRAWversion();
177 //_____________________________________________________________________________
178 AliTRDclusterizer::AliTRDclusterizer(const AliTRDclusterizer &c)
180 ,fReconstructor(c.fReconstructor)
187 ,fDigitsManager(NULL)
188 ,fTrackletContainer(NULL)
198 ,fMinLeftRightCutSigma(0)
204 ,fCalGainFactorROC(NULL)
205 ,fCalGainFactorDetValue(0)
207 ,fCalNoiseDetValue(0)
208 ,fCalPadStatusROC(NULL)
209 ,fCalOnlGainROC(NULL)
218 // AliTRDclusterizer copy constructor
221 SetBit(kLabels, kTRUE);
222 SetBit(knewDM, kFALSE);
228 //_____________________________________________________________________________
229 AliTRDclusterizer::~AliTRDclusterizer()
232 // AliTRDclusterizer destructor
235 if (fRecPoints/* && IsClustersOwner()*/){
236 fRecPoints->Delete();
241 fTracklets->Delete();
250 if (fDigitsManager) {
251 delete fDigitsManager;
252 fDigitsManager = NULL;
272 //_____________________________________________________________________________
273 AliTRDclusterizer &AliTRDclusterizer::operator=(const AliTRDclusterizer &c)
276 // Assignment operator
281 ((AliTRDclusterizer &) c).Copy(*this);
288 //_____________________________________________________________________________
289 void AliTRDclusterizer::Copy(TObject &c) const
295 ((AliTRDclusterizer &) c).fClusterTree = NULL;
296 ((AliTRDclusterizer &) c).fRecPoints = NULL;
297 ((AliTRDclusterizer &) c).fTrackletTree = NULL;
298 ((AliTRDclusterizer &) c).fDigitsManager = NULL;
299 ((AliTRDclusterizer &) c).fRawVersion = fRawVersion;
300 ((AliTRDclusterizer &) c).fTransform = NULL;
301 ((AliTRDclusterizer &) c).fDigits = NULL;
302 ((AliTRDclusterizer &) c).fDigitsRaw = NULL;
303 ((AliTRDclusterizer &) c).fIndexes = NULL;
304 ((AliTRDclusterizer &) c).fMaxThresh = 0;
305 ((AliTRDclusterizer &) c).fMaxThreshTest = 0;
306 ((AliTRDclusterizer &) c).fSigThresh = 0;
307 ((AliTRDclusterizer &) c).fMinMaxCutSigma= 0;
308 ((AliTRDclusterizer &) c).fMinLeftRightCutSigma = 0;
309 ((AliTRDclusterizer &) c).fLayer = 0;
310 ((AliTRDclusterizer &) c).fDet = 0;
311 ((AliTRDclusterizer &) c).fVolid = 0;
312 ((AliTRDclusterizer &) c).fColMax = 0;
313 ((AliTRDclusterizer &) c).fTimeTotal = 0;
314 ((AliTRDclusterizer &) c).fCalGainFactorROC = NULL;
315 ((AliTRDclusterizer &) c).fCalGainFactorDetValue = 0;
316 ((AliTRDclusterizer &) c).fCalNoiseROC = NULL;
317 ((AliTRDclusterizer &) c).fCalNoiseDetValue = 0;
318 ((AliTRDclusterizer &) c).fCalPadStatusROC = NULL;
319 ((AliTRDclusterizer &) c).fClusterROC = 0;
320 ((AliTRDclusterizer &) c).firstClusterROC= 0;
321 ((AliTRDclusterizer &) c).fNoOfClusters = 0;
322 ((AliTRDclusterizer &) c).fBaseline = 0;
323 ((AliTRDclusterizer &) c).fRawStream = NULL;
327 //_____________________________________________________________________________
328 Bool_t AliTRDclusterizer::Open(const Char_t *name, Int_t nEvent)
331 // Opens the AliROOT file. Output and input are in the same file
334 TString evfoldname = AliConfig::GetDefaultEventFolderName();
335 fRunLoader = AliRunLoader::GetRunLoader(evfoldname);
338 fRunLoader = AliRunLoader::Open(name);
342 AliError(Form("Can not open session for file %s.",name));
353 //_____________________________________________________________________________
354 Bool_t AliTRDclusterizer::OpenOutput()
357 // Open the output file
360 if (!fReconstructor->IsWritingClusters()) return kTRUE;
362 TObjArray *ioArray = 0x0;
364 AliLoader* loader = fRunLoader->GetLoader("TRDLoader");
365 loader->MakeTree("R");
367 fClusterTree = loader->TreeR();
368 fClusterTree->Branch("TRDcluster", "TObjArray", &ioArray, 32000, 0);
374 //_____________________________________________________________________________
375 Bool_t AliTRDclusterizer::OpenOutput(TTree *const clusterTree)
378 // Connect the output tree
382 if (fReconstructor->IsWritingClusters()){
383 TObjArray *ioArray = 0x0;
384 fClusterTree = clusterTree;
385 fClusterTree->Branch("TRDcluster", "TObjArray", &ioArray, 32000, 0);
390 //_____________________________________________________________________________
391 Bool_t AliTRDclusterizer::OpenInput(Int_t nEvent)
394 // Opens a ROOT-file with TRD-hits and reads in the digits-tree
397 // Import the Trees for the event nEvent in the file
398 fRunLoader->GetEvent(nEvent);
404 //_____________________________________________________________________________
405 Bool_t AliTRDclusterizer::WriteClusters(Int_t det)
408 // Fills TRDcluster branch in the tree with the clusters
409 // found in detector = det. For det=-1 writes the tree.
413 (det >= AliTRDgeometry::Ndet())) {
414 AliError(Form("Unexpected detector index %d.\n",det));
418 TObjArray *ioArray = new TObjArray(400);
419 TBranch *branch = fClusterTree->GetBranch("TRDcluster");
421 fClusterTree->Branch("TRDcluster","TObjArray",&ioArray,32000,0);
422 } else branch->SetAddress(&ioArray);
424 Int_t nRecPoints = RecPoints()->GetEntriesFast();
426 for (Int_t i = 0; i < nRecPoints; i++) {
427 AliTRDcluster *c = (AliTRDcluster *) RecPoints()->UncheckedAt(i);
428 if(det != c->GetDetector()) continue;
431 fClusterTree->Fill();
434 Int_t detOld = -1, nw(0);
435 for (Int_t i = 0; i < nRecPoints; i++) {
436 AliTRDcluster *c = (AliTRDcluster *) RecPoints()->UncheckedAt(i);
437 if(c->GetDetector() != detOld){
438 nw += ioArray->GetEntriesFast();
439 fClusterTree->Fill();
441 detOld = c->GetDetector();
445 if(ioArray->GetEntriesFast()){
446 nw += ioArray->GetEntriesFast();
447 fClusterTree->Fill();
450 AliDebug(2, Form("Clusters FOUND[%d] WRITTEN[%d] STATUS[%s]", nRecPoints, nw, nw==nRecPoints?"OK":"FAILED"));
457 //_____________________________________________________________________________
458 Bool_t AliTRDclusterizer::ReadDigits()
461 // Reads the digits arrays from the input aliroot file
465 AliError("No run loader available");
469 AliLoader* loader = fRunLoader->GetLoader("TRDLoader");
470 if (!loader->TreeD()) {
471 loader->LoadDigits();
474 // Read in the digit arrays
475 return (fDigitsManager->ReadDigits(loader->TreeD()));
479 //_____________________________________________________________________________
480 Bool_t AliTRDclusterizer::ReadDigits(TTree *digitsTree)
483 // Reads the digits arrays from the input tree
486 // Read in the digit arrays
487 return (fDigitsManager->ReadDigits(digitsTree));
491 //_____________________________________________________________________________
492 Bool_t AliTRDclusterizer::ReadDigits(AliRawReader *rawReader)
495 // Reads the digits arrays from the ddl file
499 fDigitsManager = raw.Raw2Digits(rawReader);
505 Bool_t AliTRDclusterizer::ReadTracklets()
508 // Reads simulated tracklets from the input aliroot file
511 AliRunLoader *runLoader = AliRunLoader::Instance();
513 AliError("No run loader available");
517 AliLoader* loader = runLoader->GetLoader("TRDLoader");
519 AliDataLoader *trackletLoader = loader->GetDataLoader("tracklets");
520 if (!trackletLoader) {
524 // simulated tracklets
525 trackletLoader->Load();
526 TTree *trackletTree = trackletLoader->Tree();
529 TBranch *trklbranch = trackletTree->GetBranch("mcmtrklbranch");
530 TClonesArray *trklArray = TrackletsArray("AliTRDtrackletMCM");
531 if (trklbranch && trklArray) {
532 AliTRDtrackletMCM *trkl = 0x0;
533 trklbranch->SetAddress(&trkl);
534 for (Int_t iTracklet = 0; iTracklet < trklbranch->GetEntries(); iTracklet++) {
535 trklbranch->GetEntry(iTracklet);
536 new ((*trklArray)[trklArray->GetEntries()]) AliTRDtrackletMCM(*trkl);
544 Bool_t AliTRDclusterizer::ReadTracks()
547 // Reads simulated GTU tracks from the input aliroot file
550 AliRunLoader *runLoader = AliRunLoader::Instance();
553 AliError("No run loader available");
557 AliLoader* loader = runLoader->GetLoader("TRDLoader");
562 AliDataLoader *trackLoader = loader->GetDataLoader("gtutracks");
569 TTree *trackTree = trackLoader->Tree();
574 TClonesArray *trackArray = TracksArray();
575 AliTRDtrackGTU *trk = 0x0;
576 trackTree->SetBranchAddress("TRDtrackGTU", &trk);
577 for (Int_t iTrack = 0; iTrack < trackTree->GetEntries(); iTrack++) {
578 trackTree->GetEntry(iTrack);
579 new ((*trackArray)[trackArray->GetEntries()]) AliESDTrdTrack(*(trk->CreateTrdTrack()));
585 //_____________________________________________________________________________
586 Bool_t AliTRDclusterizer::MakeClusters()
589 // Creates clusters from digits
592 // Propagate info from the digits manager
593 if (TestBit(kLabels)){
594 SetBit(kLabels, fDigitsManager->UsesDictionaries());
597 Bool_t fReturn = kTRUE;
598 for (Int_t i = 0; i < AliTRDgeometry::kNdet; i++){
600 AliTRDarrayADC *digitsIn = (AliTRDarrayADC*) fDigitsManager->GetDigits(i); //mod
601 // This is to take care of switched off super modules
602 if (!digitsIn->HasData()) continue;
604 digitsIn->DeleteNegatives(); // Restore digits array to >=0 values
605 AliTRDSignalIndex* indexes = fDigitsManager->GetIndexes(i);
606 if (indexes->IsAllocated() == kFALSE){
607 fDigitsManager->BuildIndexes(i);
611 if (indexes->HasEntry()){
612 if (TestBit(kLabels)){
614 for (Int_t iDict = 0; iDict < AliTRDdigitsManager::kNDict; iDict++){
615 AliTRDarrayDictionary *tracksIn(NULL); //mod
616 tracksIn = (AliTRDarrayDictionary *) fDigitsManager->GetDictionary(i,iDict); //mod
617 // This is to take care of data reconstruction
618 if (!tracksIn->GetDim()) continue;
619 tracksIn->Expand(); nDict++;
622 AliDebug(1, "MC labels not available. Switch them off.");
623 SetUseLabels(kFALSE);
626 fR = MakeClusters(i);
627 fReturn = fR && fReturn;
631 // if(IsWritingClusters()) WriteClusters(i);
635 // Clear arrays of this chamber, to prepare for next event
636 fDigitsManager->ClearArrays(i);
639 if(fReconstructor->IsWritingClusters()) WriteClusters(-1);
641 AliInfo(Form("Number of found clusters : %d", RecPoints()->GetEntriesFast()));
647 //_____________________________________________________________________________
648 Bool_t AliTRDclusterizer::Raw2Clusters(AliRawReader *rawReader)
651 // Creates clusters from raw data
654 return Raw2ClustersChamber(rawReader);
658 //_____________________________________________________________________________
659 Bool_t AliTRDclusterizer::Raw2ClustersChamber(AliRawReader *rawReader)
662 // Creates clusters from raw data
665 // Create the digits manager
666 if (!fDigitsManager){
667 SetBit(knewDM, kTRUE);
668 fDigitsManager = new AliTRDdigitsManager(kTRUE);
669 fDigitsManager->CreateArrays();
672 fDigitsManager->SetUseDictionaries(TestBit(kLabels));
674 // ----- preparing tracklet output -----
675 if (fReconstructor->IsWritingTracklets()) {
676 AliDataLoader *trklLoader = AliRunLoader::Instance()->GetLoader("TRDLoader")->GetDataLoader("tracklets");
678 //AliInfo("Could not get the tracklets data loader, adding it now!");
679 trklLoader = new AliDataLoader("TRD.Tracklets.root","tracklets", "tracklets");
680 AliRunLoader::Instance()->GetLoader("TRDLoader")->AddDataLoader(trklLoader);
682 AliTreeLoader *trklTreeLoader = dynamic_cast<AliTreeLoader*> (trklLoader->GetBaseLoader("tracklets-raw"));
683 if (!trklTreeLoader) {
684 trklTreeLoader = new AliTreeLoader("tracklets-raw", trklLoader);
685 trklLoader->AddBaseLoader(trklTreeLoader);
687 if (!trklTreeLoader->Tree())
688 trklTreeLoader->MakeTree();
692 fRawStream = new AliTRDrawStream(rawReader);
694 fRawStream->SetReader(rawReader);
696 //if(fReconstructor->IsHLT()){
697 fRawStream->DisableErrorStorage();
700 // register tracklet array for output
701 fRawStream->SetTrackletArray(TrackletsArray("AliTRDtrackletMCM"));
702 fRawStream->SetTrackArray(TracksArray());
705 while ((det = fRawStream->NextChamber(fDigitsManager)) < AliTRDgeometry::kNdet){
706 if (fDigitsManager->GetIndexes(det)->HasEntry()){
708 fDigitsManager->ClearArrays(det);
712 if (fReconstructor->IsWritingTracklets()) {
713 if (AliDataLoader *trklLoader = AliRunLoader::Instance()->GetLoader("TRDLoader")->GetDataLoader("tracklets")) {
715 if (AliTreeLoader *trklTreeLoader = (AliTreeLoader*) trklLoader->GetBaseLoader("tracklets-raw"))
716 trklTreeLoader->WriteData("OVERWRITE");
717 trklLoader->UnloadAll();
722 for (Int_t iSector = 0; iSector < AliTRDgeometry::kNsector; iSector++) {
723 fTrgFlags[iSector] = fRawStream->GetTriggerFlags(iSector);
726 if(fReconstructor->IsWritingClusters()) WriteClusters(-1);
728 if(!TestBit(knewDM)){
729 delete fDigitsManager;
730 fDigitsManager = NULL;
735 AliInfo(Form("Number of found clusters : %d", fNoOfClusters));
740 //_____________________________________________________________________________
741 UChar_t AliTRDclusterizer::GetStatus(Short_t &signal)
744 // Check if a pad is masked
749 if(signal>0 && TESTBIT(signal, 10)){
751 for(int ibit=0; ibit<4; ibit++){
752 if(TESTBIT(signal, 11+ibit)){
753 SETBIT(status, ibit);
754 CLRBIT(signal, 11+ibit);
761 //_____________________________________________________________________________
762 void AliTRDclusterizer::SetPadStatus(const UChar_t status, UChar_t &out) const {
764 // Set the pad status into out
765 // First three bits are needed for the position encoding
770 //_____________________________________________________________________________
771 UChar_t AliTRDclusterizer::GetPadStatus(UChar_t encoding) const {
773 // return the staus encoding of the corrupted pad
775 return static_cast<UChar_t>(encoding >> 3);
778 //_____________________________________________________________________________
779 Int_t AliTRDclusterizer::GetCorruption(UChar_t encoding) const {
781 // Return the position of the corruption
786 //_____________________________________________________________________________
787 Bool_t AliTRDclusterizer::MakeClusters(Int_t det)
790 // Generates the cluster.
794 fDigits = (AliTRDarrayADC *) fDigitsManager->GetDigits(det); //mod
795 fBaseline = fDigitsManager->GetDigitsParam()->GetADCbaseline(det);
797 // This is to take care of switched off super modules
798 if (!fDigits->HasData()) return kFALSE;
800 fIndexes = fDigitsManager->GetIndexes(det);
801 if (fIndexes->IsAllocated() == kFALSE) {
802 AliError("Indexes do not exist!");
806 AliTRDcalibDB* const calibration = AliTRDcalibDB::Instance();
808 AliFatal("No AliTRDcalibDB instance available\n");
812 if (!fReconstructor){
813 AliError("Reconstructor not set\n");
817 const AliTRDrecoParam *const recoParam = fReconstructor->GetRecoParam();
819 fMaxThresh = recoParam->GetClusMaxThresh();
820 fMaxThreshTest = (recoParam->GetClusMaxThresh()/2+fBaseline);
821 fSigThresh = recoParam->GetClusSigThresh();
822 fMinMaxCutSigma = recoParam->GetMinMaxCutSigma();
823 fMinLeftRightCutSigma = recoParam->GetMinLeftRightCutSigma();
824 const Int_t iEveryNTB = recoParam->GetRecEveryNTB();
826 Int_t istack = fIndexes->GetStack();
827 fLayer = fIndexes->GetLayer();
828 Int_t isector = fIndexes->GetSM();
830 // Start clustering in the chamber
832 fDet = AliTRDgeometry::GetDetector(fLayer,istack,isector);
834 AliError(Form("Detector number missmatch! Request[%03d] RAW[%03d]", det, fDet));
838 AliDebug(2, Form("Det[%d] @ Sec[%d] Stk[%d] Ly[%d]", fDet, isector, istack, fLayer));
840 // TRD space point transformation
841 fTransform->SetDetector(det);
843 Int_t iGeoLayer = AliGeomManager::kTRD1 + fLayer;
844 Int_t iGeoModule = istack + AliTRDgeometry::Nstack() * isector;
845 fVolid = AliGeomManager::LayerToVolUID(iGeoLayer,iGeoModule);
847 fColMax = fDigits->GetNcol();
848 fTimeTotal = fDigitsManager->GetDigitsParam()->GetNTimeBins(det);
850 // Check consistency between Geometry and raw data
851 AliTRDpadPlane *pp(fTransform->GetPadPlane());
852 Int_t ncols(pp->GetNcols()), nrows(pp->GetNrows());
853 if(ncols != fColMax) AliDebug(1, Form("N cols missmatch in Digits for Det[%3d] :: Geom[%3d] RAW[%3d]", fDet, ncols, fColMax));
854 if(nrows != fDigits->GetNrow()) AliDebug(1, Form("N rows missmatch in Digits for Det[%3d] :: Geom[%3d] RAW[%3d]", fDet, nrows, fDigits->GetNrow()));
855 if(ncols != fIndexes->GetNcol()) AliDebug(1, Form("N cols missmatch in Digits for Det[%3d] :: Geom[%3d] RAW[%3d]", fDet, ncols, fIndexes->GetNcol()));
856 if(nrows != fIndexes->GetNrow()) AliDebug(1, Form("N rows missmatch in Digits for Det[%3d] :: Geom[%3d] RAW[%3d]", fDet, nrows, fIndexes->GetNrow()));
858 // Check consistency between OCDB and raw data
859 Int_t nTimeOCDB = calibration->GetNumberOfTimeBinsDCS();
860 if(fReconstructor->IsHLT()){
861 if((nTimeOCDB > -1) && (fTimeTotal != nTimeOCDB)){
862 AliWarning(Form("Number of timebins does not match OCDB value (RAW[%d] OCDB[%d]), using raw value"
863 ,fTimeTotal,nTimeOCDB));
867 AliDebug(1, "Undefined number of timebins in OCDB, using value from raw data.");
869 AliError(Form("Number of timebins in raw data is negative, skipping chamber[%3d]!", fDet));
872 }else if(nTimeOCDB == -2){
873 AliError("Mixed number of timebins in OCDB, no reconstruction of TRD data!");
875 }else if(fTimeTotal != nTimeOCDB){
876 AliError(Form("Number of timebins in raw data does not match OCDB value (RAW[%d] OCDB[%d]), skipping chamber[%3d]!"
877 ,fTimeTotal,nTimeOCDB, fDet));
881 AliDebug(1, Form("Using %2d number of timebins for Det[%03d].", fTimeTotal, fDet));
883 // Detector wise calibration object for the gain factors
884 const AliTRDCalDet *calGainFactorDet = calibration->GetGainFactorDet();
885 // Calibration object with pad wise values for the gain factors
886 fCalGainFactorROC = calibration->GetGainFactorROC(fDet);
887 // Calibration value for chamber wise gain factor
888 fCalGainFactorDetValue = calGainFactorDet->GetValue(fDet);
890 // Detector wise calibration object for the noise
891 const AliTRDCalDet *calNoiseDet = calibration->GetNoiseDet();
892 // Calibration object with pad wise values for the noise
893 fCalNoiseROC = calibration->GetNoiseROC(fDet);
894 // Calibration value for chamber wise noise
895 fCalNoiseDetValue = calNoiseDet->GetValue(fDet);
897 // Calibration object with the pad status
898 fCalPadStatusROC = calibration->GetPadStatusROC(fDet);
899 // Calibration object of the online gain
900 fCalOnlGainROC = 0x0;
901 if (calibration->HasOnlineFilterGain()) {
902 fCalOnlGainROC = calibration->GetOnlineGainTableROC(fDet);
905 firstClusterROC = -1;
908 SetBit(kLUT, recoParam->UseLUT());
909 SetBit(kGAUS, recoParam->UseGAUS());
911 // Apply the gain and the tail cancelation via digital filter
912 // Use the configuration from the DCS to find out whether online
913 // tail cancellation was applied
914 if(!calibration->HasOnlineTailCancellation()){
915 // save a copy of raw data
916 if(TestBit(kRawSignal)){
918 fDigitsRaw->~AliTRDarrayADC();
919 new(fDigitsRaw) AliTRDarrayADC(*fDigits);
920 } else fDigitsRaw = new AliTRDarrayADC(*fDigits);
922 TailCancelation(recoParam);
925 MaxStruct curr, last;
926 Int_t nMaximas = 0, nCorrupted = 0;
928 // Here the clusterfining is happening
930 for(curr.time = 0; curr.time < fTimeTotal; curr.time+=iEveryNTB){
931 while(fIndexes->NextRCIndex(curr.row, curr.col)){
932 if(fDigits->GetData(curr.row, curr.col, curr.time) > fMaxThreshTest && IsMaximum(curr, curr.padStatus, &curr.signals[0])){
934 if(curr.col==last.col+2 && curr.row==last.row && curr.time==last.time) FivePadCluster(last, curr);
937 last=curr; curr.fivePad=kFALSE;
941 if(last.row>-1) CreateCluster(last);
943 if(recoParam->GetStreamLevel(AliTRDrecoParam::kClusterizer) > 2 && fReconstructor->IsDebugStreaming()){
944 TTreeSRedirector* fDebugStream = fReconstructor->GetDebugStream(AliTRDrecoParam::kClusterizer);
945 (*fDebugStream) << "MakeClusters"
946 << "Detector=" << det
947 << "NMaxima=" << nMaximas
948 << "NClusters=" << fClusterROC
949 << "NCorrupted=" << nCorrupted
952 if (TestBit(kLabels)) AddLabels();
958 //_____________________________________________________________________________
959 Bool_t AliTRDclusterizer::IsMaximum(const MaxStruct &Max, UChar_t &padStatus, Float_t *const Signals)
962 // Returns true if this row,col,time combination is a maximum.
963 // Gives back the padStatus and the signals of the center pad and the two neighbouring pads.
966 Float_t gain = fCalGainFactorDetValue * fCalGainFactorROC->GetValue(Max.col,Max.row);
967 Float_t onlcf = fCalOnlGainROC ? fCalOnlGainROC->GetGainCorrectionFactor(Max.row,Max.col) : 1;
968 Signals[1] = (fDigits->GetData(Max.row, Max.col, Max.time) - fBaseline) /(onlcf * gain) + 0.5f;
969 if(Signals[1] <= fMaxThresh) return kFALSE;
971 if(Max.col < 1 || Max.col + 1 >= fColMax) return kFALSE;
973 Float_t noiseMiddleThresh = fMinMaxCutSigma*fCalNoiseDetValue*fCalNoiseROC->GetValue(Max.col, Max.row);
974 if (Signals[1] <= noiseMiddleThresh) return kFALSE;
977 fCalPadStatusROC->GetStatus(Max.col-1, Max.row)
978 ,fCalPadStatusROC->GetStatus(Max.col, Max.row)
979 ,fCalPadStatusROC->GetStatus(Max.col+1, Max.row)
983 if((signal = fDigits->GetData(Max.row, Max.col-1, Max.time))){
984 gain = fCalGainFactorDetValue * fCalGainFactorROC->GetValue(Max.col-1,Max.row);
985 onlcf = fCalOnlGainROC ? fCalOnlGainROC->GetGainCorrectionFactor(Max.row,Max.col-1) : 1;
986 Signals[0] = (signal - fBaseline) /( onlcf * gain) + 0.5f;
987 } else Signals[0] = 0.;
988 if((signal = fDigits->GetData(Max.row, Max.col+1, Max.time))){
989 gain = fCalGainFactorDetValue * fCalGainFactorROC->GetValue(Max.col+1,Max.row);
990 onlcf = fCalOnlGainROC ? fCalOnlGainROC->GetGainCorrectionFactor(Max.row,Max.col+1) : 1;
991 Signals[2] = (signal - fBaseline) /( onlcf * gain) + 0.5f;
992 } else Signals[2] = 0.;
994 if(!(status[0] | status[1] | status[2])) {//all pads are good
995 if ((Signals[2] <= Signals[1]) && (Signals[0] < Signals[1])) {
996 if ((Signals[2] > fSigThresh) || (Signals[0] > fSigThresh)) {
997 if(Signals[0]<0) Signals[0]=0.;
998 if(Signals[2]<0) Signals[2]=0.;
999 Float_t noiseSumThresh = fMinLeftRightCutSigma * fCalNoiseDetValue
1000 * fCalNoiseROC->GetValue(Max.col, Max.row);
1001 if ((Signals[2]+Signals[0]+Signals[1]) <= noiseSumThresh) return kFALSE;
1006 } else { // at least one of the pads is bad, and reject candidates with more than 1 problematic pad
1007 if(Signals[0]<0)Signals[0]=0;
1008 if(Signals[2]<0)Signals[2]=0;
1009 if (status[2] && (!(status[0] || status[1])) && Signals[1] > Signals[0] && Signals[0] > fSigThresh) {
1011 SetPadStatus(status[2], padStatus);
1014 else if (status[0] && (!(status[1] || status[2])) && Signals[1] >= Signals[2] && Signals[2] > fSigThresh) {
1016 SetPadStatus(status[0], padStatus);
1019 else if (status[1] && (!(status[0] || status[2])) && ((Signals[2] > fSigThresh) || (Signals[0] > fSigThresh))) {
1020 Signals[1] = fMaxThresh;
1021 SetPadStatus(status[1], padStatus);
1028 //_____________________________________________________________________________
1029 Bool_t AliTRDclusterizer::FivePadCluster(MaxStruct &ThisMax, MaxStruct &NeighbourMax)
1032 // Look for 5 pad cluster with minimum in the middle
1033 // Gives back the ratio
1036 if (ThisMax.col >= fColMax - 3) return kFALSE;
1039 if (ThisMax.col < fColMax - 5){
1040 gain = fCalGainFactorDetValue * fCalGainFactorROC->GetValue(ThisMax.col+4,ThisMax.row);
1041 onlcf = fCalOnlGainROC ? fCalOnlGainROC->GetGainCorrectionFactor(ThisMax.row,ThisMax.col+4) : 1;
1042 if (fDigits->GetData(ThisMax.row, ThisMax.col+4, ThisMax.time) - fBaseline >= fSigThresh * gain * onlcf)
1045 if (ThisMax.col > 1) {
1046 gain = fCalGainFactorDetValue * fCalGainFactorROC->GetValue(ThisMax.col-2,ThisMax.row);
1047 onlcf = fCalOnlGainROC ? fCalOnlGainROC->GetGainCorrectionFactor(ThisMax.row,ThisMax.col-2) : 1;
1048 if (fDigits->GetData(ThisMax.row, ThisMax.col-2, ThisMax.time) - fBaseline >= fSigThresh * gain * onlcf)
1052 const Float_t kEpsilon = 0.01;
1053 Double_t padSignal[5] = {ThisMax.signals[0], ThisMax.signals[1], ThisMax.signals[2],
1054 NeighbourMax.signals[1], NeighbourMax.signals[2]};
1056 // Unfold the two maxima and set the signal on
1057 // the overlapping pad to the ratio
1058 Float_t ratio = Unfold(kEpsilon,fLayer,padSignal);
1059 ThisMax.signals[2] = ThisMax.signals[2]*ratio + 0.5f;
1060 NeighbourMax.signals[0] = NeighbourMax.signals[0]*(1-ratio) + 0.5f;
1061 ThisMax.fivePad=kTRUE;
1062 NeighbourMax.fivePad=kTRUE;
1067 //_____________________________________________________________________________
1068 void AliTRDclusterizer::CreateCluster(const MaxStruct &Max)
1071 // Creates a cluster at the given position and saves it in fRecPoints
1074 Int_t nPadCount = 1;
1075 Short_t signals[7] = { 0, 0, (Short_t)Max.signals[0], (Short_t)Max.signals[1], (Short_t)Max.signals[2], 0, 0 };
1076 if(!fReconstructor->IsHLT()) CalcAdditionalInfo(Max, signals, nPadCount);
1078 AliTRDcluster cluster(fDet, ((UChar_t) Max.col), ((UChar_t) Max.row), ((UChar_t) Max.time), signals, fVolid);
1079 cluster.SetNPads(nPadCount);
1080 cluster.SetQ(Max.signals[0]+Max.signals[1]+Max.signals[2]);
1081 if(TestBit(kLUT)) cluster.SetRPhiMethod(AliTRDcluster::kLUT);
1082 else if(TestBit(kGAUS)) cluster.SetRPhiMethod(AliTRDcluster::kGAUS);
1083 else cluster.SetRPhiMethod(AliTRDcluster::kCOG);
1085 cluster.SetFivePad(Max.fivePad);
1086 // set pads status for the cluster
1087 UChar_t maskPosition = GetCorruption(Max.padStatus);
1089 cluster.SetPadMaskedPosition(maskPosition);
1090 cluster.SetPadMaskedStatus(GetPadStatus(Max.padStatus));
1092 cluster.SetXcorr(fReconstructor->UseClusterRadialCorrection());
1094 // Transform the local cluster coordinates into calibrated
1095 // space point positions defined in the local tracking system.
1096 // Here the calibration for T0, Vdrift and ExB is applied as well.
1097 if(!TestBit(kSkipTrafo)) if(!fTransform->Transform(&cluster)) return;
1098 // Store raw signals in cluster. This MUST be called after position reconstruction !
1099 // Xianguo Lu and Alex Bercuci 19.03.2012
1100 if(TestBit(kRawSignal) && fDigitsRaw){
1101 Float_t tmp(0.), kMaxShortVal(32767.); // protect against data overflow due to wrong gain calibration
1102 Short_t rawSignal[7] = {0};
1103 for(Int_t ipad(Max.col-3), iRawId(0); ipad<=Max.col+3; ipad++, iRawId++){
1104 if(ipad<0 || ipad>=fColMax) continue;
1105 if(!fCalOnlGainROC){
1106 rawSignal[iRawId] = fDigitsRaw->GetData(Max.row, ipad, Max.time);
1109 // Deconvolute online gain calibration when available
1110 // Alex Bercuci 27.04.2012
1111 tmp = (fDigitsRaw->GetData(Max.row, ipad, Max.time) - fBaseline)/fCalOnlGainROC->GetGainCorrectionFactor(Max.row, ipad) + 0.5f;
1112 rawSignal[iRawId] = (Short_t)TMath::Min(tmp, kMaxShortVal);
1114 cluster.SetSignals(rawSignal, kTRUE);
1116 // Temporarily store the Max.Row, column and time bin of the center pad
1117 // Used to later on assign the track indices
1118 cluster.SetLabel(Max.row, 0);
1119 cluster.SetLabel(Max.col, 1);
1120 cluster.SetLabel(Max.time,2);
1122 //needed for HLT reconstruction
1123 AddClusterToArray(&cluster);
1125 // Store the index of the first cluster in the current ROC
1126 if (firstClusterROC < 0) firstClusterROC = fNoOfClusters;
1132 //_____________________________________________________________________________
1133 void AliTRDclusterizer::CalcAdditionalInfo(const MaxStruct &Max, Short_t *const signals, Int_t &nPadCount)
1135 // Calculate number of pads/cluster and
1136 // ADC signals at position 0, 1, 5 and 6
1138 Float_t tmp(0.), kMaxShortVal(32767.); // protect against data overflow due to wrong gain calibration
1139 Float_t gain(1.); Float_t onlcf(1.); Short_t signal(0);
1140 // Store the amplitudes of the pads in the cluster for later analysis
1141 // and check whether one of these pads is masked in the database
1142 signals[3]=Max.signals[1];
1143 Int_t ipad(1), jpad(0);
1144 // Look to the right
1145 while((jpad = Max.col-ipad)){
1146 if(!(signal = fDigits->GetData(Max.row, jpad, Max.time))) break; // empty digit !
1147 gain = fCalGainFactorDetValue * fCalGainFactorROC->GetValue(jpad, Max.row);
1148 onlcf = fCalOnlGainROC ? fCalOnlGainROC->GetGainCorrectionFactor(Max.row,jpad) : 1;
1149 tmp = (signal - fBaseline) / (onlcf * gain) + 0.5f;
1150 signal = (Short_t)TMath::Min(tmp, kMaxShortVal);
1151 if(signal<fSigThresh) break; // signal under threshold
1153 if(ipad<=3) signals[3 - ipad] = signal;
1158 while((jpad = Max.col+ipad)<fColMax){
1159 if(!(signal = fDigits->GetData(Max.row, jpad, Max.time))) break; // empty digit !
1160 gain = fCalGainFactorDetValue * fCalGainFactorROC->GetValue(jpad, Max.row);
1161 onlcf = fCalOnlGainROC ? fCalOnlGainROC->GetGainCorrectionFactor(Max.row,jpad) : 1;
1162 tmp = (signal - fBaseline) / (onlcf * gain) + 0.5f;
1163 signal = (Short_t)TMath::Min(tmp, kMaxShortVal);
1164 if(signal<fSigThresh) break; // signal under threshold
1166 if(ipad<=3) signals[3 + ipad] = signal;
1170 AliDebug(4, Form("Signals[%3d %3d %3d %3d %3d %3d %3d] Npads[%d]."
1171 , signals[0], signals[1], signals[2], signals[3], signals[4], signals[5], signals[6], nPadCount));
1174 //_____________________________________________________________________________
1175 void AliTRDclusterizer::AddClusterToArray(AliTRDcluster* cluster)
1178 // Add a cluster to the array
1181 Int_t n = RecPoints()->GetEntriesFast();
1182 if(n!=fNoOfClusters)AliError(Form("fNoOfClusters != RecPoints()->GetEntriesFast %i != %i \n", fNoOfClusters, n));
1183 new((*RecPoints())[n]) AliTRDcluster(*cluster);
1186 //_____________________________________________________________________________
1187 Bool_t AliTRDclusterizer::AddLabels()
1190 // Add the track indices to the found clusters
1193 const Int_t kNclus = 3;
1194 const Int_t kNdict = AliTRDdigitsManager::kNDict;
1195 const Int_t kNtrack = kNdict * kNclus;
1197 Int_t iClusterROC = 0;
1204 // Temporary array to collect the track indices
1205 Int_t *idxTracks = new Int_t[kNtrack*fClusterROC];
1207 // Loop through the dictionary arrays one-by-one
1208 // to keep memory consumption low
1209 AliTRDarrayDictionary *tracksIn(NULL); //mod
1210 for (Int_t iDict = 0; iDict < kNdict; iDict++) {
1212 // tracksIn should be expanded beforehand!
1213 tracksIn = (AliTRDarrayDictionary *) fDigitsManager->GetDictionary(fDet,iDict);
1215 // Loop though the clusters found in this ROC
1216 for (iClusterROC = 0; iClusterROC < fClusterROC; iClusterROC++) {
1218 AliTRDcluster *cluster = (AliTRDcluster *)
1219 RecPoints()->UncheckedAt(firstClusterROC+iClusterROC);
1220 row = cluster->GetLabel(0);
1221 col = cluster->GetLabel(1);
1222 time = cluster->GetLabel(2);
1224 for (iPad = 0; iPad < kNclus; iPad++) {
1225 Int_t iPadCol = col - 1 + iPad;
1226 Int_t index = tracksIn->GetData(row,iPadCol,time); //Modification of -1 in Track
1227 idxTracks[3*iPad+iDict + iClusterROC*kNtrack] = index;
1234 // Copy the track indices into the cluster
1235 // Loop though the clusters found in this ROC
1236 for (iClusterROC = 0; iClusterROC < fClusterROC; iClusterROC++) {
1238 AliTRDcluster *cluster = (AliTRDcluster *)
1239 RecPoints()->UncheckedAt(firstClusterROC+iClusterROC);
1240 cluster->SetLabel(-9999,0);
1241 cluster->SetLabel(-9999,1);
1242 cluster->SetLabel(-9999,2);
1244 cluster->AddTrackIndex(&idxTracks[iClusterROC*kNtrack]);
1248 delete [] idxTracks;
1254 //_____________________________________________________________________________
1255 Float_t AliTRDclusterizer::Unfold(Double_t eps, Int_t layer, const Double_t *const padSignal) const
1258 // Method to unfold neighbouring maxima.
1259 // The charge ratio on the overlapping pad is calculated
1260 // until there is no more change within the range given by eps.
1261 // The resulting ratio is then returned to the calling method.
1264 AliTRDcalibDB *calibration = AliTRDcalibDB::Instance();
1266 AliError("No AliTRDcalibDB instance available\n");
1271 Int_t itStep = 0; // Count iteration steps
1273 Double_t ratio = 0.5; // Start value for ratio
1274 Double_t prevRatio = 0.0; // Store previous ratio
1276 Double_t newLeftSignal[3] = { 0.0, 0.0, 0.0 }; // Array to store left cluster signal
1277 Double_t newRightSignal[3] = { 0.0, 0.0, 0.0 }; // Array to store right cluster signal
1278 Double_t newSignal[3] = { 0.0, 0.0, 0.0 };
1280 // Start the iteration
1281 while ((TMath::Abs(prevRatio - ratio) > eps) && (itStep < 10)) {
1286 // Cluster position according to charge ratio
1287 Double_t maxLeft = (ratio*padSignal[2] - padSignal[0])
1288 / (padSignal[0] + padSignal[1] + ratio * padSignal[2]);
1289 Double_t maxRight = (padSignal[4] - (1-ratio)*padSignal[2])
1290 / ((1.0 - ratio)*padSignal[2] + padSignal[3] + padSignal[4]);
1292 // Set cluster charge ratio
1293 irc = calibration->PadResponse(1.0, maxLeft, layer, newSignal);
1294 Double_t ampLeft = padSignal[1] / newSignal[1];
1295 irc = calibration->PadResponse(1.0, maxRight, layer, newSignal);
1296 Double_t ampRight = padSignal[3] / newSignal[1];
1298 // Apply pad response to parameters
1299 irc = calibration->PadResponse(ampLeft ,maxLeft ,layer,newLeftSignal );
1300 irc = calibration->PadResponse(ampRight,maxRight,layer,newRightSignal);
1302 // Calculate new overlapping ratio
1303 ratio = TMath::Min((Double_t) 1.0
1304 ,newLeftSignal[2] / (newLeftSignal[2] + newRightSignal[0]));
1312 //_____________________________________________________________________________
1313 void AliTRDclusterizer::TailCancelation(const AliTRDrecoParam* const recoParam)
1316 // Applies the tail cancelation
1319 Int_t nexp = recoParam->GetTCnexp();
1326 TTreeSRedirector *fDebugStream = fReconstructor->GetDebugStream(AliTRDrecoParam::kClusterizer);
1327 Bool_t debugStreaming = recoParam->GetStreamLevel(AliTRDrecoParam::kClusterizer) > 7 && fReconstructor->IsDebugStreaming();
1328 while(fIndexes->NextRCIndex(iRow, iCol))
1330 // if corrupted then don't make the tail cancallation
1331 if (fCalPadStatusROC->GetStatus(iCol, iRow)) continue;
1334 for (iTime = 0; iTime < fTimeTotal; iTime++)
1335 (*fDebugStream) << "TailCancellation"
1341 // Apply the tail cancelation via the digital filter
1342 //DeConvExp(fDigits->GetDataAddress(iRow,iCol),fTimeTotal,nexp);
1343 ApplyTCTM(fDigits->GetDataAddress(iRow,iCol),fTimeTotal,nexp);
1344 } // while irow icol
1351 //_____________________________________________________________________________
1352 void AliTRDclusterizer::ApplyTCTM(Short_t *const arr, const Int_t nTime, const Int_t nexp)
1355 // Steer tail cancellation
1362 DeConvExp(arr,nTime,nexp);
1368 DeConvExp(arr,nTime,1);
1377 //_____________________________________________________________________________
1378 void AliTRDclusterizer::ConvExp(Short_t *const arr, const Int_t nTime)
1384 // Initialization (coefficient = alpha, rates = lambda)
1385 Float_t slope = 1.0;
1386 Float_t coeff = 0.5;
1390 fReconstructor->GetRecoParam()->GetTCParams(par);
1396 rate = TMath::Exp(-dt/(slope));
1398 Float_t reminder = .0;
1399 Float_t correction = 0.0;
1400 Float_t result = 0.0;
1402 for (int i = nTime-1; i >= 0; i--) {
1404 result = arr[i] + correction - fBaseline; // No rescaling
1405 arr[i] = (Short_t)(result + fBaseline + 0.5f);
1409 correction += reminder = rate * (reminder + coeff * result);
1414 //_____________________________________________________________________________
1415 void AliTRDclusterizer::DeConvExp(Short_t *const arr, const Int_t nTime, const Int_t nexp)
1418 // Tail cancellation by deconvolution for PASA v4 TRF
1422 Float_t coefficients[2];
1424 // Initialization (coefficient = alpha, rates = lambda)
1430 if (nexp == 1) { // 1 Exponentials
1436 if (nexp == 2) { // 2 Exponentials
1438 fReconstructor->GetRecoParam()->GetTCParams(par);
1439 r1 = par[0];//1.156;
1440 r2 = par[1];//0.130;
1441 c1 = par[2];//0.114;
1442 c2 = par[3];//0.624;
1445 coefficients[0] = c1;
1446 coefficients[1] = c2;
1450 rates[0] = TMath::Exp(-dt/(r1));
1451 rates[1] = (nexp == 1) ? .0 : TMath::Exp(-dt/(r2));
1453 Float_t reminder[2] = { .0, .0 };
1454 Float_t correction = 0.0;
1455 Float_t result = 0.0;
1457 for (int i = 0; i < nTime; i++) {
1459 result = arr[i] - correction - fBaseline; // No rescaling
1460 arr[i] = (Short_t)(result + fBaseline + 0.5f);
1463 for (int k = 0; k < 2; k++) {
1464 correction += reminder[k] = rates[k] * (reminder[k] + coefficients[k] * result);
1471 //_____________________________________________________________________________
1472 void AliTRDclusterizer::ResetRecPoints()
1475 // Resets the list of rec points
1479 fRecPoints->Clear();
1481 // delete fRecPoints;
1485 //_____________________________________________________________________________
1486 TClonesArray *AliTRDclusterizer::RecPoints()
1489 // Returns the list of rec points
1493 if(!(fRecPoints = AliTRDReconstructor::GetClusters())){
1494 // determine number of clusters which has to be allocated
1495 Float_t nclusters = fReconstructor->GetRecoParam()->GetNClusters();
1497 fRecPoints = new TClonesArray("AliTRDcluster", Int_t(nclusters));
1499 //SetClustersOwner(kTRUE);
1500 AliTRDReconstructor::SetClusters(0x0);
1506 //_____________________________________________________________________________
1507 TClonesArray *AliTRDclusterizer::TrackletsArray(const TString &trkltype)
1510 // Returns the array of on-line tracklets
1513 if (trkltype.Length() != 0) {
1515 fTracklets = new TClonesArray(trkltype, 200);
1517 else if (TClass::GetClass(trkltype.Data()) != fTracklets->GetClass()){
1518 fTracklets->Delete();
1520 fTracklets = new TClonesArray(trkltype, 200);
1526 //_____________________________________________________________________________
1527 TClonesArray* AliTRDclusterizer::TracksArray()
1529 // return array of GTU tracks (create TClonesArray if necessary)
1532 fTracks = new TClonesArray("AliESDTrdTrack",100);