3 /**************************************************************************
4 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
7 * for The ALICE HLT Project. *
9 * Permission to use, copy, modify and distribute this software and its *
10 * documentation strictly for non-commercial purposes is hereby granted *
11 * without fee, provided that the above copyright notice appears in all *
12 * copies and that both the copyright notice and this permission notice *
13 * appear in the supporting documentation. The authors make no claims *
14 * about the suitability of this software for any purpose. It is *
15 * provided "as is" without express or implied warranty. *
16 **************************************************************************/
18 // @file AliHLTTRDCalibrationComponent.cxx
21 // @brief A TRDCalibration processing component for the HLT.
33 #include "TProfile2D.h"
35 #include "AliHLTReadoutList.h"
37 #include "AliHLTTRDCalibrationComponent.h"
38 #include "AliHLTTRDDefinitions.h"
39 #include "AliHLTTRDUtils.h"
41 #include "AliCDBManager.h"
42 #include "AliCDBStorage.h"
43 #include "AliRawReaderMemory.h"
45 #include "AliTRDCalPad.h"
46 #include "AliTRDCalDet.h"
48 #include "AliTRDCalibraFillHisto.h"
49 #include "AliTRDtrackV1.h"
51 #include "AliTRDCalibraFit.h"
52 #include "AliTRDCalibraMode.h"
53 #include "AliTRDCalibraVector.h"
54 #include "AliTRDCalibraVdriftLinearFit.h"
55 #include "AliTRDReconstructor.h"
56 #include "AliTRDrecoParam.h"
62 ClassImp(AliHLTTRDCalibrationComponent);
64 AliHLTTRDCalibrationComponent::AliHLTTRDCalibrationComponent()
65 : AliHLTCalibrationProcessor(),
66 fTRDCalibraFillHisto(NULL),
72 fSavedTimeBins(kFALSE),
77 fTakeAllEvents(kFALSE)
79 // Default constructor
82 AliHLTTRDCalibrationComponent::~AliHLTTRDCalibrationComponent()
87 const char* AliHLTTRDCalibrationComponent::GetComponentID()
89 // Return the component ID const char *
90 return "TRDCalibration"; // The ID of this component
93 void AliHLTTRDCalibrationComponent::GetInputDataTypes( vector<AliHLTComponentDataType>& list)
95 // Get the list of input data
96 list.clear(); // We do not have any requirements for our input data type(s).
97 list.push_back(AliHLTTRDDefinitions::fgkTracksDataType);
100 AliHLTComponentDataType AliHLTTRDCalibrationComponent::GetOutputDataType()
102 // Get the output data type
103 return kAliHLTMultipleDataType;
104 // return AliHLTTRDDefinitions::fgkCalibrationDataType;
108 int AliHLTTRDCalibrationComponent::GetOutputDataTypes(AliHLTComponentDataTypeList& tgtList)
110 // Get the output data type
112 tgtList.push_back(AliHLTTRDDefinitions::fgkCalibrationDataType);
113 tgtList.push_back(AliHLTTRDDefinitions::fgkEORCalibrationDataType);
114 return tgtList.size();
117 void AliHLTTRDCalibrationComponent::GetOutputDataSize( unsigned long& constBase, double& inputMultiplier )
119 // Get the output data size
120 constBase = fOutputSize;
124 AliHLTComponent* AliHLTTRDCalibrationComponent::Spawn()
126 // Spawn function, return new instance of this class
127 return new AliHLTTRDCalibrationComponent;
130 Int_t AliHLTTRDCalibrationComponent::ScanArgument( int argc, const char** argv )
132 // perform initialization. We check whether our relative output size is specified in the arguments.
136 fTrgStrings = new TObjArray();
140 HLTDebug("argv[%d] == %s", i, argv[i] );
141 if ( !strcmp( argv[i], "output_size" ) )
145 HLTError("Missing output_size parameter");
148 HLTDebug("argv[%d+1] == %s", i, argv[i+1] );
149 fOutputSize = strtoul( argv[i+1], &cpErr, 0 );
152 HLTError("Cannot convert output_size parameter '%s'", argv[i+1] );
155 HLTInfo("Output size set to %lu %%", fOutputSize );
159 if ( !strcmp( argv[i], "-minClusters" ) )
163 HLTError("Missing minClusters parameter");
166 HLTDebug("argv[%d+1] == %s", i, argv[i+1] );
167 fMinClusters = strtoul( argv[i+1], &cpErr, 0 );
171 if ( !strcmp( argv[i], "-minTracklets" ) )
175 HLTError("Missing minTracklets parameter");
178 HLTDebug("argv[%d+1] == %s", i, argv[i+1] );
179 fMinTracklets = strtoul( argv[i+1], &cpErr, 0 );
183 if ( !strcmp( argv[i], "-TrgStr" ) )
187 HLTError("Missing parameter for mbTriggerString");
190 HLTDebug("argv[%d+1] == %s", i, argv[i+1] );
191 fTrgStrings->Add(new TObjString(argv[i+1]));
196 if ( !strcmp( argv[i], "-acceptTrgStr" ) )
202 if ( !strcmp( argv[i], "-rejectTrgStr" ) )
208 if ( !strcmp( argv[i], "-takeAllEvents" ) )
210 fTakeAllEvents = kTRUE;
217 HLTError("Unknown option '%s'", argv[i] );
224 Int_t AliHLTTRDCalibrationComponent::InitCalibration()
228 fTrgStrings = new TObjArray();
230 if(!AliCDBManager::Instance()->IsDefaultStorageSet()){
231 HLTError("DefaultStorage is not set in CDBManager");
234 if(AliCDBManager::Instance()->GetRun()<0){
235 HLTError("Run Number is not set in CDBManager");
238 HLTInfo("CDB default storage: %s; RunNo: %i", (AliCDBManager::Instance()->GetDefaultStorage()->GetBaseFolder()).Data(), AliCDBManager::Instance()->GetRun());
240 if(fTrgStrings->GetEntriesFast()>0 && !fAccRejTrg){
241 HLTError("Trigger string(s) given, but acceptTrgStr or rejectTrgStr not selected");
245 fTRDCalibraFillHisto = AliTRDCalibraFillHisto::Instance();
246 fTRDCalibraFillHisto->SetIsHLT(kTRUE);
247 fTRDCalibraFillHisto->SetHisto2d(); // choose to use histograms
248 fTRDCalibraFillHisto->SetCH2dOn(); // choose to calibrate the gain
249 fTRDCalibraFillHisto->SetPH2dOn(); // choose to calibrate the drift velocity
250 fTRDCalibraFillHisto->SetPRF2dOn(); // choose to look at the PRF
251 fTRDCalibraFillHisto->SetIsHLT(); // per detector
252 //fTRDCalibraFillHisto->SetDebugLevel(1);// debug
253 fTRDCalibraFillHisto->SetFillWithZero(kTRUE);
254 fTRDCalibraFillHisto->SetLinearFitterOn(kTRUE);
255 fTRDCalibraFillHisto->SetNumberBinCharge(100);
257 fTracksArray = new TClonesArray("AliTRDtrackV1");
258 fOutArray = new TObjArray(4);
259 fAfterRunArray=new TObjArray(5);
260 fDisplayArray=new TObjArray(4);
262 HLTDebug("run SetupCTPData");
268 Int_t AliHLTTRDCalibrationComponent::DeinitCalibration()
271 // Deinitialization of the component
273 HLTDebug("DeinitCalibration");
274 delete fTracksArray; fTracksArray=0;
275 //fTRDCalibraFillHisto->Destroy();
276 //fOutArray->Delete();
277 delete fOutArray; fOutArray=0;
278 fAfterRunArray->Delete();
279 delete fAfterRunArray; fAfterRunArray=0;
280 fDisplayArray->Delete();
281 delete fDisplayArray; fDisplayArray=0;
282 fTrgStrings->Delete();
283 delete fTrgStrings; fTrgStrings=0;
287 Int_t AliHLTTRDCalibrationComponent::ProcessCalibration(const AliHLTComponent_EventData& /*evtData*/,
288 const AliHLTComponent_BlockData* /*blocks*/,
289 AliHLTComponent_TriggerData& /*trigData*/,
290 AliHLTUInt8_t* /*outputPtr*/,
291 AliHLTUInt32_t& /*size*/,
292 vector<AliHLTComponent_BlockData>& /*outputBlocks*/)
296 TClonesArray* TCAarray[18] = {0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0};
297 Int_t usedEntries = 0;
298 Int_t blockOrObject = 0;
299 Int_t nTimeBins = -1;
301 for (const AliHLTComponentBlockData* pBlock=GetFirstInputBlock(AliHLTTRDDefinitions::fgkTracksDataType); pBlock; pBlock=GetNextInputBlock())
303 TCAarray[0] = fTracksArray;
304 AliHLTTRDUtils::ReadTracks(TCAarray[0], pBlock->fPtr, pBlock->fSize, &nTimeBins);
309 for(const TObject *iter = GetFirstInputObject(AliHLTTRDDefinitions::fgkHiLvlTracksDataType); iter; iter = GetNextInputObject())
312 HLTError("You may not mix high level and low level!");
316 TCAarray[usedEntries] = dynamic_cast<TClonesArray*>(const_cast<TObject*>(iter));
317 if(!TCAarray[usedEntries])continue;
318 TObjString* strg = dynamic_cast<TObjString*>(const_cast<TObject*>(GetNextInputObject()));
321 nTimeBins = strg->String().Atoi();
331 HLTFatal("Number of timebins is negative!");
334 HLTDebug("Saving number of time bins which was read from input block. Value is: %d", nTimeBins);
335 fTRDCalibraFillHisto->Init2Dhistos(nTimeBins); // initialise the histos
336 fTRDCalibraFillHisto->SetNumberClusters(fMinClusters); // At least fMinClusters clusters
337 fTRDCalibraFillHisto->SetNumberClustersf(nTimeBins); // Not more than %d clusters
338 fSavedTimeBins=kTRUE;
341 Bool_t TriggerPassed = fTakeAllEvents;
345 TriggerPassed=kFALSE;
346 for(int i = 0; i < fTrgStrings->GetEntriesFast(); i++){
347 const TObjString *const obString=(TObjString*)fTrgStrings->At(i);
348 const TString tString=obString->GetString();
349 if(CheckCTPTrigger(tString.Data())>0){TriggerPassed=kTRUE; break;}
354 for(int i = 0; i < fTrgStrings->GetEntriesFast(); i++){
355 const TObjString *const obString=(TObjString*)fTrgStrings->At(i);
356 const TString tString=obString->GetString();
357 if(CheckCTPTrigger(tString.Data())>0){TriggerPassed=kFALSE; break;}
362 fTRDCalibraFillHisto->SetCH2dOn(TriggerPassed);
363 fTRDCalibraFillHisto->SetPH2dOn(TriggerPassed);
364 for(int i=0; i<usedEntries; i++){
365 const TClonesArray* inArr = TCAarray[i];
366 Int_t nbEntries = inArr->GetEntries();
367 HLTDebug(" %i TRDtracks in tracksArray", nbEntries);
368 AliTRDtrackV1* trdTrack = 0x0;
369 for (Int_t ii = 0; ii < nbEntries; ii++){
370 HLTDebug("%i/%i: ", ii+1, nbEntries);
371 trdTrack = (AliTRDtrackV1*)inArr->At(ii);
372 if(trdTrack->GetNumberOfTracklets()<fMinTracklets)continue;
373 fTRDCalibraFillHisto->UpdateHistogramsV1(trdTrack);
374 // for(int i3=0; i3<7; i3++)
375 // if(trdTrack->GetTracklet(i3))trdTrack->GetTracklet(i3)->Bootstrap(fReconstructor);
379 if(!fOutArray->At(0))FormOutput(0);
380 if(!fDisplayArray->At(0))FormOutput(1);
381 PushBack(fDisplayArray, AliHLTTRDDefinitions::fgkCalibrationDataType);
384 TCAarray[0]->Delete();
392 * Form output array of histrograms
394 //============================================================================
395 void AliHLTTRDCalibrationComponent::FormOutput(Int_t param)
398 TH2I *hCH2d = fTRDCalibraFillHisto->GetCH2d();
399 if(!param)fOutArray->Add(hCH2d);
400 else fDisplayArray->Add(hCH2d);
402 // drift velocity histo
403 TProfile2D *hPH2d = fTRDCalibraFillHisto->GetPH2d();
404 if(!param)fOutArray->Add(hPH2d);
405 else fDisplayArray->Add(hPH2d);
408 TProfile2D *hPRF2d = fTRDCalibraFillHisto->GetPRF2d();
409 if(!param)fOutArray->Add(hPRF2d);
410 else fDisplayArray->Add(hPRF2d);
414 AliTRDCalibraVdriftLinearFit *hVdriftLinearFitOne=(AliTRDCalibraVdriftLinearFit *)fTRDCalibraFillHisto->GetVdriftLinearFit();
415 fOutArray->Add(hVdriftLinearFitOne);
418 TH2S *hVdriftLinearFitOne = (TH2S *)(((AliTRDCalibraVdriftLinearFit *)fTRDCalibraFillHisto->GetVdriftLinearFit())->GetLinearFitterHisto(10,kTRUE));
419 fDisplayArray->Add(hVdriftLinearFitOne);
422 HLTDebug("GetCH2d = 0x%x; NEntries = %i; size = %i", hCH2d, hCH2d->GetEntries(), sizeof(*hCH2d));
424 HLTDebug("GetPH2d = 0x%x; NEntries = %i; size = %i", hPH2d, hPH2d->GetEntries(), sizeof(*hPH2d));
426 HLTDebug("GetPRF2d = 0x%x; NEntries = %i; size = %i", hPRF2d, hPRF2d->GetEntries(), sizeof(*hPRF2d));
428 //HLTDebug("GetVdriftLinearFit = 0x%x; size = %i", hVdriftLinearFitOne, sizeof(hVdriftLinearFitOne));
430 HLTDebug("output Array: pointer = 0x%x; NEntries = %i; size = %i", fOutArray, fOutArray->GetEntries(), sizeof(fOutArray));
434 Int_t AliHLTTRDCalibrationComponent::ShipDataToFXS(const AliHLTComponentEventData& /*evtData*/, AliHLTComponentTriggerData& /*trigData*/)
436 //fTRDCalibraFillHisto->DestroyDebugStreamer();
438 AliHLTReadoutList rdList(AliHLTReadoutList::kTRD);
442 fOutArray->Remove(fOutArray->FindObject("AliTRDCalibraVdriftLinearFit"));
443 //fOutArray->Remove(fOutArray->FindObject("PRF2d"));
444 //fOutArray->Remove(fOutArray->FindObject("PH2d"));
445 //fOutArray->Remove(fOutArray->FindObject("CH2d"));
447 if(!(fOutArray->FindObject("CH2d"))) {
448 TH2I * ch2d = new TH2I("CH2d","Nz0Nrphi0",100,0.0,300.0,540,0,540);
449 fOutArray->Add(ch2d);
452 if(!(fOutArray->FindObject("PH2d"))) {
453 TProfile2D * ph2d = new TProfile2D("PH2d","Nz0Nrphi0",30,-0.05,2.95,540,0,540);
454 fOutArray->Add(ph2d);
457 if(!(fOutArray->FindObject("PRF2d"))) {
458 TProfile2D * prf2d = new TProfile2D("PRF2d","Nz0Nrphi0Ngp3",60,-9.0,9.0,540,0,540);
459 fOutArray->Add(prf2d);
462 HLTDebug("Size of the fOutArray is %d\n",fOutArray->GetEntriesFast());
464 PushToFXS((TObject*)fOutArray, "TRD", "GAINDRIFTPRF", &rdList );
465 //PushToFXS((TObject*)fOutArray->FindObject("CH2d"), "TRD", "GAINDRIFTPRF", rdList.Buffer() );
470 Int_t AliHLTTRDCalibrationComponent::EORCalibration()
472 //Also Fill histograms for the online display
473 TH2I *hCH2d=(TH2I*)fOutArray->FindObject("CH2d");
474 TProfile2D *hPH2d=(TProfile2D*)fOutArray->FindObject("PH2d");
475 TProfile2D *hPRF2d= (TProfile2D*)fOutArray->FindObject("PRF2d");
476 AliTRDCalibraVdriftLinearFit* hVdriftLinearFit = (AliTRDCalibraVdriftLinearFit*)fOutArray->FindObject("AliTRDCalibraVdriftLinearFit");
479 if(!hCH2d || !hPH2d || !hPRF2d || !hVdriftLinearFit) return 0;
482 AliTRDCalibraFit *calibra = AliTRDCalibraFit::Instance();
485 calibra->SetMinEntries(100);
486 calibra->AnalyseCH(hCH2d);
487 Int_t nbtg = 6*4*18*((Int_t) ((AliTRDCalibraMode *)calibra->GetCalibraMode())->GetDetChamb0(0))
488 + 6* 18*((Int_t) ((AliTRDCalibraMode *)calibra->GetCalibraMode())->GetDetChamb2(0));
489 Int_t nbfit = calibra->GetNumberFit();
490 Int_t nbE = calibra->GetNumberEnt();
491 TH1F *coefgain = 0x0;
494 // (nbfit >= 0.2*nbE)) {
495 // create the cal objects
496 //calibra->PutMeanValueOtherVectorFit(1,kTRUE);
497 TObjArray object = calibra->GetVectorFit();
498 AliTRDCalDet *objgaindet = calibra->CreateDetObjectGain(&object,kFALSE);
499 coefgain = objgaindet->MakeHisto1DAsFunctionOfDet();
501 calibra->ResetVectorFit();
503 // vdrift second method
504 calibra->SetMinEntries(100); // If there is less than 100
505 hVdriftLinearFit->FillPEArray();
506 calibra->AnalyseLinearFitters(hVdriftLinearFit);
508 nbfit = calibra->GetNumberFit();
509 nbE = calibra->GetNumberEnt();
510 TH1F *coefdriftsecond = 0x0;
513 // (nbfit >= 0.1*nbE)) {
514 // create the cal objects
515 //calibra->PutMeanValueOtherVectorFit(1,kTRUE);
516 object = calibra->GetVectorFit();
517 AliTRDCalDet *objdriftvelocitydetsecond = calibra->CreateDetObjectVdrift(&object,kTRUE);
518 objdriftvelocitydetsecond->SetTitle("secondmethodvdrift");
519 coefdriftsecond = objdriftvelocitydetsecond->MakeHisto1DAsFunctionOfDet();
521 calibra->ResetVectorFit();
523 // vdrift first method
524 calibra->SetMinEntries(100*20); // If there is less than 20000
525 calibra->AnalysePH(hPH2d);
526 nbtg = 6*4*18*((Int_t) ((AliTRDCalibraMode *)calibra->GetCalibraMode())->GetDetChamb0(1))
527 + 6* 18*((Int_t) ((AliTRDCalibraMode *)calibra->GetCalibraMode())->GetDetChamb2(1));
528 nbfit = calibra->GetNumberFit();
529 nbE = calibra->GetNumberEnt();
530 TH1F *coefdrift = 0x0;
534 // (nbfit >= 0.2*nbE)) {
535 // create the cal objects
536 //calibra->PutMeanValueOtherVectorFit(1,kTRUE);
537 //calibra->PutMeanValueOtherVectorFit2(1,kTRUE);
538 object = calibra->GetVectorFit();
539 AliTRDCalDet *objdriftvelocitydet = calibra->CreateDetObjectVdrift(&object,kTRUE);
540 coefdrift = objdriftvelocitydet->MakeHisto1DAsFunctionOfDet();
541 object = calibra->GetVectorFit2();
542 AliTRDCalDet *objtime0det = calibra->CreateDetObjectT0(&object,kTRUE);
543 coeft0 = objtime0det->MakeHisto1DAsFunctionOfDet();
545 calibra->ResetVectorFit();
549 calibra->SetMinEntries(200);
550 calibra->AnalysePRFMarianFit(hPRF2d);
551 nbtg = 6*4*18*((Int_t) ((AliTRDCalibraMode *)calibra->GetCalibraMode())->GetDetChamb0(2))
552 + 6* 18*((Int_t) ((AliTRDCalibraMode *)calibra->GetCalibraMode())->GetDetChamb2(2));
553 nbfit = calibra->GetNumberFit();
554 nbE = calibra->GetNumberEnt();
558 // (nbfit >= 0.95*nbE)) {
559 // create cal pad objects
560 object = calibra->GetVectorFit();
561 TObject *objPRFpad = calibra->CreatePadObjectPRF(&object);
562 coefprf = ((AliTRDCalPad *) objPRFpad)->MakeHisto1D();
564 calibra->ResetVectorFit();
567 coefgain->SetName("coefgain");
568 coefprf->SetName("coefprf");
569 coefdrift->SetName("coefdrift");
570 coefdriftsecond->SetName("coefdriftsecond");
571 coeft0->SetName("coeft0");
572 if(coefgain) fAfterRunArray->Add(coefgain);
573 if(coefprf) fAfterRunArray->Add(coefprf);
574 if(coefdrift) fAfterRunArray->Add(coefdrift);
575 if(coefdriftsecond) fAfterRunArray->Add(coefdriftsecond);
576 if(coeft0) fAfterRunArray->Add(coeft0);
579 if(coefgain||coefprf||coefdrift||coeft0||coefdriftsecond) {
580 PushBack(fAfterRunArray, AliHLTTRDDefinitions::fgkEORCalibrationDataType);
583 // TString fileName="/tmp/CalibHistoDump_run";
584 // fileName+=AliCDBManager::Instance()->GetRun();
585 // fileName+=".root";
586 // HLTInfo("Dumping Histogram file to %s",fileName.Data());
587 // TFile* file = TFile::Open(fileName, "RECREATE");
588 // fAfterRunArray->Write();
589 // fOutArray->Write();
591 // HLTInfo("Histogram file dumped");