Fixes to the raw-data reco. 1. Possibility to select the max adc search window. 2...
[u/mrichter/AliRoot.git] / VZERO / AliVZEROReconstructor.cxx
index 2353668..8f809b5 100644 (file)
@@ -34,6 +34,8 @@
 #include "AliESDfriend.h"
 #include "AliESDVZEROfriend.h"
 #include "AliVZEROdigit.h"
+#include "AliVZEROCalibData.h"
+#include "AliRunInfo.h"
 
 ClassImp(AliVZEROReconstructor)
 
@@ -87,7 +89,7 @@ void AliVZEROReconstructor::Init()
 //______________________________________________________________________
 void AliVZEROReconstructor::ConvertDigits(AliRawReader* rawReader, TTree* digitsTree) const
 {
-// converts RAW to digits - pedestal is subtracted 
+// converts RAW to digits 
 
   if (!digitsTree) {
     AliError("No digits tree!");
@@ -101,28 +103,61 @@ void AliVZEROReconstructor::ConvertDigits(AliRawReader* rawReader, TTree* digits
 
   rawReader->Reset();
   AliVZERORawStream rawStream(rawReader);
-  if (rawStream.Next()) {  
-     Float_t ADC_max[64], adc[64]; 
-     Bool_t BBFlag[64], BGFlag[64], integrator[64]; 
+  if (rawStream.Next()) { 
+     Float_t adc[64]; 
      Float_t time[64], width[64];  
+     Bool_t BBFlag[64], BGFlag[64], integrator[64];
      for(Int_t i=0; i<64; i++) {
-         // Search for the maximum charge in the train of 21 LHC clocks 
-         // regardless of the integrator which has been operated:
-         ADC_max[i] = 0.0;
-        Int_t imax = 0;
-         for(Int_t iClock=0; iClock<21; iClock++){
-             if(rawStream.GetPedestal(i,iClock) > ADC_max[i])  
-               {ADC_max[i]= rawStream.GetPedestal(i,iClock);
-                imax      = iClock;}
-         }
-        // Convert i (FEE channel numbering) to j (aliroot channel numbering)
-        Int_t j       =  rawStream.GetOfflineChannel(i);
-        adc[j]        =  ADC_max[i];
-        time[j]       =  rawStream.GetTime(i);
-        width[j]      =  rawStream.GetWidth(i);
-        BBFlag[j]     =  rawStream.GetBBFlag(i,imax);
-        BGFlag[j]     =  rawStream.GetBGFlag(i,imax); 
-        integrator[j] =  rawStream.GetIntegratorFlag(i,imax); 
+       Int_t j   =  rawStream.GetOfflineChannel(i);
+       adc[j]    = 0.0;
+       time[j]   = 0.0;
+       width[j]  = 0.0;
+       BBFlag[j] = kFALSE;
+       BGFlag[j] = kFALSE;
+       integrator[j] = kFALSE;
+       // Search for the maximum charge in the train of 21 LHC clocks 
+       // regardless of the integrator which has been operated:
+       Float_t maxadc = 0;
+       Int_t imax = -1;
+       Float_t adcPedSub[21];
+       for(Int_t iClock=0; iClock<21; iClock++){
+        Bool_t iIntegrator = rawStream.GetIntegratorFlag(i,iClock);
+        Int_t k = j+64*iIntegrator;
+        adcPedSub[iClock] = rawStream.GetPedestal(i,iClock) - fCalibData->GetPedestal(k);
+        if(adcPedSub[iClock] <= GetRecoParam()->GetNSigmaPed()*fCalibData->GetSigma(k)) {
+          adcPedSub[iClock] = 0;
+          continue;
+        }
+        if(iClock < GetRecoParam()->GetStartClock() || iClock > GetRecoParam()->GetEndClock()) continue;
+        if(adcPedSub[iClock] > maxadc) {
+          maxadc = adcPedSub[iClock];
+          imax   = iClock;
+        }
+       }
+
+       AliDebug(2,Form("Channel %d (online), %d (offline)",i,j)); 
+        if (imax != -1) {
+          Int_t start = imax - GetRecoParam()->GetNPreClocks();
+          if (start < 0) start = 0;
+          Int_t end = imax + GetRecoParam()->GetNPostClocks();
+          if (end > 20) end = 20;
+          for(Int_t iClock = start; iClock <= end; iClock++) {
+            if (iClock == imax)
+              adc[j] += rawStream.GetPedestal(i,iClock);
+            else 
+              adc[j] += adcPedSub[iClock];
+
+            AliDebug(2,Form("clock = %d adc = %f",iClock,rawStream.GetPedestal(i,iClock))); 
+          }
+          // Convert i (FEE channel numbering) to j (aliroot channel numbering)
+
+          Int_t board   = j / 8;
+          time[j]       =  rawStream.GetTime(i)/ (25./256.) * fCalibData->GetTimeResolution(board);
+          width[j]      =  rawStream.GetWidth(i) / 0.4 * fCalibData->GetWidthResolution(board);
+          BBFlag[j]     =  rawStream.GetBBFlag(i,imax);
+          BGFlag[j]     =  rawStream.GetBGFlag(i,imax); 
+          integrator[j] =  rawStream.GetIntegratorFlag(i,imax); 
+        }
 
         // Filling the esd friend object
         fESDVZEROfriend->SetBBScalers(j,rawStream.GetBBScalers(i));
@@ -139,8 +174,8 @@ void AliVZEROReconstructor::ConvertDigits(AliRawReader* rawReader, TTree* digits
             fESDVZEROfriend->SetBBFlag(j,iEv,rawStream.GetBBFlag(i,iEv));
             fESDVZEROfriend->SetBGFlag(j,iEv,rawStream.GetBGFlag(i,iEv));
         }
-        fESDVZEROfriend->SetTime(j,rawStream.GetTime(i));
-        fESDVZEROfriend->SetWidth(j,rawStream.GetWidth(i));
+        fESDVZEROfriend->SetTime(j,time[j]);
+        fESDVZEROfriend->SetWidth(j,width[j]);
      }  
 
      // Filling the esd friend object
@@ -156,20 +191,25 @@ void AliVZEROReconstructor::ConvertDigits(AliRawReader* rawReader, TTree* digits
 
      // Channels(aliroot numbering) will be ordered in the tree
      for(Int_t iChannel = 0; iChannel < 64; iChannel++) {
-         new ((*digitsArray)[digitsArray->GetEntriesFast()])
+         if(fCalibData->IsChannelDead(iChannel)){
+           adc[iChannel]  = (Float_t) kInvalidADC; 
+           time[iChannel] = (Float_t) kInvalidTime;     
+         }
+        if (adc[iChannel] > 0)
+          new ((*digitsArray)[digitsArray->GetEntriesFast()])
              AliVZEROdigit(iChannel, adc[iChannel], time[iChannel],
                           width[iChannel], BBFlag[iChannel], BGFlag[iChannel],integrator[iChannel]);
-     }
+        
+     }          
+     digitsTree->Fill();
   }
-
-  digitsTree->Fill();
 }
 
 //______________________________________________________________________
 void AliVZEROReconstructor::FillESD(TTree* digitsTree, TTree* /*clustersTree*/,
                                    AliESDEvent* esd) const
 {
-// fills multiplicities to the ESD
+// fills multiplicities to the ESD - pedestal is now subtracted
     
   if (!digitsTree) {
       AliError("No digits tree!");
@@ -207,22 +247,19 @@ void AliVZEROReconstructor::FillESD(TTree* digitsTree, TTree* /*clustersTree*/,
         AliVZEROdigit* digit = (AliVZEROdigit*)digitsArray->At(d);      
         Int_t  pmNumber      = digit->PMNumber(); 
         // Pedestal retrieval and suppression: 
-        Float_t  pedestal      = fCalibData->GetPedestal(d);
+       Bool_t   integrator  = digit->Integrator();
+        Float_t  pedestal    = fCalibData->GetPedestal(d + 64*integrator);
         adc[pmNumber]   =  digit->ADC() - pedestal; 
         time[pmNumber]  =  digit->Time();
        width[pmNumber] =  digit->Width();
        BBFlag[pmNumber]=  digit->BBFlag();
        BGFlag[pmNumber]=  digit->BGFlag();
-        // printf("PM = %d,  MIP per ADC channel = %f \n",pmNumber, fCalibData->GetMIPperADC(pmNumber));
-               //AliInfo(Form("PM = %d,  ADC = %f TDC %f",pmNumber, digit->ADC(),digit->Time()));
-        // cut of ADC at 1MIP/2 
-       if(fCollisionMode >0) { 
-          Float_t MIP = 2.0;  
-           if (adc[pmNumber] > (MIP/2.) ) mult[pmNumber] += adc[pmNumber]*(1.0/MIP) ;
-          }
-           else{    
-           if (adc[pmNumber] > (1.0/fCalibData->GetMIPperADC(pmNumber) /2.) ) 
-              mult[pmNumber] += adc[pmNumber]*fCalibData->GetMIPperADC(pmNumber);
+
+       AliDebug(2,Form("PM = %d ADC = %f TDC %f",pmNumber, digit->ADC(),digit->Time()));
+
+       Int_t k = pmNumber+64*integrator;
+       if(adc[pmNumber] > (fCalibData->GetPedestal(k) + GetRecoParam()->GetNSigmaPed()*fCalibData->GetSigma(k))) {
+           mult[pmNumber] += adc[pmNumber]*fCalibData->GetMIPperADC(pmNumber);
         }          
     } // end of loop over digits
   } // end of loop over events in digits tree
@@ -292,51 +329,28 @@ AliCDBStorage* AliVZEROReconstructor::SetStorage(const char *uri)
 //____________________________________________________________________________
 void AliVZEROReconstructor::GetCollisionMode()
 {
-// Retrieves the collision mode from GRP data
+  // Retrieval of collision mode 
 
-// Initialization of the GRP entry 
-// To be replace by a call to AliRunInfo object
-  
-  AliCDBEntry*  entry = AliCDBManager::Instance()->Get("GRP/GRP/Data");
-  AliGRPObject* grpData = 0x0;
-   
-  if(entry){
-    TMap* m = dynamic_cast<TMap*>(entry->GetObject());  // old GRP entry
-    if(m){
-       grpData = new AliGRPObject();
-       grpData->ReadValuesFromMap(m);
-    }
-    else{
-       grpData = dynamic_cast<AliGRPObject*>(entry->GetObject());  // new GRP entry
-    }
-  }
-
-  if(!grpData) { AliError("No GRP entry found in OCDB!");
-                  return; }
-
-// Retrieval of simulated collision mode 
-
-  TString beamType = grpData->GetBeamType();
+  TString beamType = GetRunInfo()->GetBeamType();
   if(beamType==AliGRPObject::GetInvalidString()){
-     AliError("GRP/GRP/Data entry:  missing value for the beam type !");
-     AliError("\t VZERO cannot retrieve beam type\n");
+     AliError("VZERO cannot retrieve beam type");
      return;
   }
 
-   if( (beamType.CompareTo("P-P") ==0)  || (beamType.CompareTo("p-p") ==0) ){
-       fCollisionMode=0;
+  if( (beamType.CompareTo("P-P") ==0)  || (beamType.CompareTo("p-p") ==0) ){
+    fCollisionMode=0;
+  }
+  else if( (beamType.CompareTo("Pb-Pb") ==0)  || (beamType.CompareTo("A-A") ==0) ){
+    fCollisionMode=1;
   }
-   else if( (beamType.CompareTo("Pb-Pb") ==0)  || (beamType.CompareTo("A-A") ==0) ){
-       fCollisionMode=1;
-   }
     
-  fBeamEnergy = grpData->GetBeamEnergy();
+  fBeamEnergy = GetRunInfo()->GetBeamEnergy();
   if(fBeamEnergy==AliGRPObject::GetInvalidFloat()) {
-     AliError("GRP/GRP/Data entry:  missing value for the beam energy ! Using 0");
+     AliError("Missing value for the beam energy ! Using 0");
      fBeamEnergy = 0.;
   }
   
-//    printf("\n ++++++ Beam type and collision mode retrieved as %s %d @ %1.3f GeV ++++++\n\n",beamType.Data(), fCollisionMode, fBeamEnergy);
+  AliDebug(1,Form("\n ++++++ Beam type and collision mode retrieved as %s %d @ %1.3f GeV ++++++\n\n",beamType.Data(), fCollisionMode, fBeamEnergy));
 
 }