]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - EMCAL/AliCaloRawAnalyzerPeakFinder.cxx
Update (Rossella)
[u/mrichter/AliRoot.git] / EMCAL / AliCaloRawAnalyzerPeakFinder.cxx
index f839260957e58f5c14776fecdf90af3d16af2bd4..beb6b3a7cc94907b90538a52d1124f25f705a84d 100644 (file)
@@ -1,11 +1,11 @@
 /**************************************************************************
- * This file is property of and copyright by the Experimental Nuclear     *
- * Physics Group, Dep. of Physics                                         *
- * University of Oslo, Norway, 2007                                       *
+ * This file is property of and copyright by                              *
+ * the Relativistic Heavy Ion Group (RHIG), Yale University, US, 2009     *
+ *                                                                        *
+ * Primary Author: Per Thomas Hille  <perthomas.hille@yale.edu>           *
  *                                                                        *
- * Author: Per Thomas Hille <perthi@fys.uio.no> for the ALICE HLT Project.*
  * Contributors are mentioned in the code where appropriate.              *
- * Please report bugs to perthi@fys.uio.no                                *
+ * Please report bugs to   perthomas.hille@yale.edu                       *
  *                                                                        *
  * Permission to use, copy, modify and distribute this software and its   *
  * documentation strictly for non-commercial purposes is hereby granted   *
@@ -16,6 +16,7 @@
  * provided "as is" without express or implied warranty.                  *
  **************************************************************************/
 
+
 // The Peak-Finder algorithm
 // The amplitude is extracted  as a
 // weighted sum of the samples using the 
 #include "AliCaloRawAnalyzerPeakFinder.h"
 #include "AliCaloBunchInfo.h"
 #include "AliCaloFitResults.h"
-#include <iostream>
-#include "unistd.h"
 #include "TMath.h"
 #include "AliLog.h"
+#include "AliCDBEntry.h"
+#include "AliCDBManager.h"
+#include "TFile.h"
+#include "AliCaloPeakFinderVectors.h"
 
 using namespace std;
 
-AliCaloRawAnalyzerPeakFinder::AliCaloRawAnalyzerPeakFinder() :AliCaloRawAnalyzer(), 
-                                                               fTof(0), 
-                                                               fAmp(0)
+
+ClassImp( AliCaloRawAnalyzerPeakFinder )
+
+
+AliCaloRawAnalyzerPeakFinder::AliCaloRawAnalyzerPeakFinder() :AliCaloRawAnalyzer("Peak-Finder", "PF"),  
+                                                             fAmp(0),
+                                                             fPeakFinderVectors(0),
+                                                             fRunOnAlien(false)
 {
-  //comment
+  //Comment
+  InitOCDB(fRunOnAlien);
+  fPeakFinderVectors = new AliCaloPeakFinderVectors() ;
+  ResetVectors();
+  LoadVectorsOCDB();
+}
 
-  fNsampleCut = 5;
 
+void 
+AliCaloRawAnalyzerPeakFinder::InitOCDB(bool alien) const
+{
+  // Setting the default OCDB pathe depending on wether we work locally or on the GRID.
+  AliCDBManager::Instance()->SetDefaultStorage(  alien == true ? "alien://$ALICE_ROOT/OCDB" : "local://$ALICE_ROOT/OCDB");
+  AliCDBManager::Instance()->SetRun(100);
+}
+
+
+void  
+AliCaloRawAnalyzerPeakFinder::ResetVectors()
+{
+  //As name implies
   for(int i=0; i < MAXSTART; i++)
     {
       for(int j=0; j < SAMPLERANGE; j++ )
        {
-         fPFAmpVectors[i][j] = new double[100];
-         fPFTofVectors[i][j] = new double[100];
-         
          for(int k=0; k < 100; k++ )
            {
              fPFAmpVectors[i][j][k] = 0; 
              fPFTofVectors[i][j][k] = 0;
+             fPFAmpVectorsCoarse[i][j][k] = 0;
+             fPFTofVectorsCoarse[i][j][k] = 0; 
            }
        }
     }
-  LoadVectors();
 }
 
 
@@ -70,11 +93,35 @@ AliCaloRawAnalyzerPeakFinder::~AliCaloRawAnalyzerPeakFinder()
        {
          delete[] fPFAmpVectors[i][j];
          delete[] fPFTofVectors[i][j];
+         delete[] fPFAmpVectorsCoarse[i][j];
+         delete[] fPFTofVectorsCoarse[i][j];
        }
     }
 }
 
 
+
+Double_t  
+AliCaloRawAnalyzerPeakFinder::ScanCoarse(const Double_t *const array, const int length ) const
+{
+  // Fisrt (coarce) estimate of Amplitude using the Peak-Finder.
+  // The output of the first iteration is sued to select vectors 
+  // for the second iteration.
+
+  Double_t tmpTof = 0;
+  Double_t tmpAmp= 0;
+
+  for(int i=0; i < length; i++)
+    {
+      tmpTof += fPFTofVectorsCoarse[0][length][i]*array[i]; 
+      tmpAmp += fPFAmpVectorsCoarse[0][length][i]*array[i]; 
+    }
+  
+  tmpTof = tmpTof / tmpAmp ;
+  return tmpTof;
+}
+
+
 AliCaloFitResults 
 AliCaloRawAnalyzerPeakFinder::Evaluate( const vector<AliCaloBunchInfo> &bunchvector, const UInt_t altrocfg1,  const UInt_t altrocfg2 )
 {
@@ -86,87 +133,193 @@ AliCaloRawAnalyzerPeakFinder::Evaluate( const vector<AliCaloBunchInfo> &bunchvec
   short maxamp; //Maximum amplitude
   fAmp = 0;
   int index = SelectBunch( bunchvector,  &maxampindex,  &maxamp );
+
   if( index >= 0)
     {
       Float_t ped = ReverseAndSubtractPed( &(bunchvector.at(index))  ,  altrocfg1, altrocfg2, fReversed  );
       Float_t maxf = TMath::MaxElement(   bunchvector.at(index).GetLength(),  fReversed );
-      
-      if(  maxf < fAmpCut  ||  ( maxamp - ped) > 900  )         // (maxamp - ped) > 900 = Close to saturation (use low gain then)
+      short maxrev = maxampindex  -  bunchvector.at(index).GetStartBin();
+      short timebinOffset = maxampindex - (bunchvector.at( index ).GetLength()-1); 
+            
+      if(  maxf <= fAmpCut  ||  ( maxamp - ped) > 900  )        // (maxamp - ped) > 900 = Close to saturation (use low gain then)
        {
-         //      cout << __FILE__ << __LINE__ <<":, maxamp = " << maxamp << ", ped = "<< ped  << ",. maxf = "<< maxf << ", maxampindex = "<< maxampindex  << endl;
-         return  AliCaloFitResults( maxamp, ped,  -1, maxf,   maxampindex, -1, -1 );
+         return  AliCaloFitResults( maxamp, ped, AliCaloFitResults::kCrude, maxf, timebinOffset);
        }
       
       int first;
       int last;
       
-      if ( maxf > fAmpCut )
-       {
-         SelectSubarray( fReversed,  bunchvector.at(index).GetLength(),  maxampindex -  bunchvector.at(index).GetStartBin(), &first, &last);
+      if ( maxf >= fAmpCut )
+       {         
+         SelectSubarray( fReversed,  bunchvector.at(index).GetLength(), maxrev, &first, &last);
          int nsamples =  last - first;
          if( ( nsamples  )  >= fNsampleCut )
            {
              int startbin = bunchvector.at(index).GetStartBin();  
-             int n = last -first;  
+             int n = last - first;  
              int pfindex = n - fNsampleCut; 
              pfindex = pfindex > SAMPLERANGE ? SAMPLERANGE : pfindex;
 
-             for(int i=0; i < SAMPLERANGE; i++ )
+             int dt =  maxampindex - startbin -2; 
+             int tmpindex = 0;
+
+             Float_t tmptof = ScanCoarse( &fReversed[dt] , n );
+             
+             if( tmptof < -1 )
                {
-                 int dt =  maxampindex - startbin -2;
-                 double tmp =  fReversed[ dt  +i];  
-                 fAmp += fPFAmpVectors[0][pfindex][i]*tmp; 
+                 tmpindex = 0;
                }
+             else
+               if( tmptof  > -1 && tmptof < 100 )
+                 {
+                   tmpindex = 1;
+                 }
+               else
+                 {
+                   tmpindex = 2;
+                 }
+
+             double tof = 0;
              
-             return AliCaloFitResults( maxamp, ped , -1, fAmp, -1, -1, -1 ); 
+             for(int k=0; k < SAMPLERANGE; k++   )
+               {
+                 tof +=  fPFTofVectors[0][pfindex][k]*fReversed[ dt  +k + tmpindex -1 ];   
+               }
+           
+             for( int i=0; i < SAMPLERANGE; i++ )
+               {
+                 {
+                   fAmp += fPFAmpVectors[0][pfindex][i]*fReversed[ dt  +i  +tmpindex -1 ];
+                 }
+               }
+             if( TMath::Abs(  (maxf - fAmp  )/maxf )  >   0.1 )
+               {
+                 fAmp = maxf;
+               }
+             
+             tof = timebinOffset - 0.01*tof/fAmp; // clock ticks
+             
+             return AliCaloFitResults( maxamp, ped , AliCaloFitResults::kFitPar, fAmp, tof, 
+                                       timebinOffset, AliCaloFitResults::kDummy, AliCaloFitResults::kDummy,
+                                       AliCaloFitResults::kDummy, AliCaloFitSubarray(index, maxrev, first, last) );  
            }
          else
            {
-             return AliCaloFitResults( maxamp, ped , -1, maxf, -1, -1, -1 ); 
+             return AliCaloFitResults( maxamp, ped , AliCaloFitResults::kCrude, maxf, timebinOffset); 
+           }
+       }
+      else 
+       {
+         return AliCaloFitResults( maxamp , ped, AliCaloFitResults::kCrude, maxf, timebinOffset);
+       }
+    }
+  return  AliCaloFitResults(AliCaloFitResults::kInvalid, AliCaloFitResults::kInvalid);
+}
+
+
+void   
+AliCaloRawAnalyzerPeakFinder::CopyVectors(const AliCaloPeakFinderVectors *const pfv )
+{
+  // As name implies
+
+  if ( pfv != 0)
+    {
+      for(int i = 0;  i < MAXSTART ; i++)
+       {
+         for( int j=0; j < SAMPLERANGE; j++)  
+           {
+             pfv->GetVector( i, j, fPFAmpVectors[i][j] ,  fPFTofVectors[i][j],    
+                             fPFAmpVectorsCoarse[i][j] , fPFTofVectorsCoarse[i][j]  ); 
+             fPeakFinderVectors->SetVector( i, j, fPFAmpVectors[i][j], fPFTofVectors[i][j],    
+                                            fPFAmpVectorsCoarse[i][j], fPFTofVectorsCoarse[i][j] );   
            }
        }
     }
-  //  cout << __FILE__ << __LINE__ <<  "WARNING, returning amp = -1 "  <<  endl;
+  else
+    {
+      AliFatal( "pfv = ZERO !!!!!!!");
+    } 
+}
+
 
-  return  AliCaloFitResults(-1, -1);
+
+void   
+AliCaloRawAnalyzerPeakFinder::LoadVectorsOCDB()
+{
+  //Loading of Peak-Finder  vectors from the 
+  //Offline Condition Database  (OCDB)
+  AliCDBEntry* entry = AliCDBManager::Instance()->Get("EMCAL/Calib/PeakFinder/");
+  
+  if( entry != 0 )
+    {
+      AliCaloPeakFinderVectors  *pfv = (AliCaloPeakFinderVectors *)entry->GetObject(); 
+      CopyVectors( pfv );
+     }
 }
 
 
 void 
-AliCaloRawAnalyzerPeakFinder::LoadVectors()
+AliCaloRawAnalyzerPeakFinder::LoadVectorsASCII()
 {
-  //Read in the Peak finder vecors from file
+  //Read in the Peak finder vecors from ASCI files
   for(int i = 0;  i < MAXSTART ; i++)
     {
       for( int j=0; j < SAMPLERANGE; j++)
        {
+         char filenameCoarse[256];
          char filename[256];
          int n = j+fNsampleCut;
-         double start = (double)i+0.5;
-         sprintf(filename, "%s/EMCAL/vectors-emcal/start%.1fN%dtau0.235fs10dt1.0.txt", getenv("ALICE_ROOT"), start, n);
-         FILE *fp = fopen(filename, "r");
+         double start = (double)i+0;
+         
+         sprintf(filename,        "%s/EMCAL/vectors-emcal/start%.1fN%dtau0.235fs10dt1.0.txt", getenv("ALICE_ROOT"), start, n);
+         sprintf(filenameCoarse,  "%s/EMCAL/vectors-emcal/start%.1fN%dtau0.235fs10dt3.0.txt", getenv("ALICE_ROOT"), start, n);
          
-         if(fp == 0)
+         FILE *fp  =  fopen(filename, "r");
+         FILE *fpc =  fopen(filenameCoarse, "r");
+
+         if( fp == 0 )
            {
              AliFatal( Form( "could not open file: %s", filename ) );
            }
+       
+         if(fpc == 0)
+           {
+             AliFatal( Form( "could not open file: %s", filenameCoarse ) );
+           }
          else
            {
              for(int m = 0; m < n ; m++ )
                {
-                 fscanf(fp, "%lf\t", &fPFAmpVectors[i][j][m] );
+                 fscanf(fp, "%lf\t", &fPFAmpVectors[i][j][m] );
+                 fscanf(fpc, "%lf\t", &fPFAmpVectorsCoarse[i][j][m] );
                }
-             fscanf(fp, "\n");
-             
+             fscanf(fp,   "\n" );
+             fscanf(fpc,  "\n" );
              for(int m = 0; m < n ; m++ )
                {
-                 fscanf(fp, "%lf\t", &fPFTofVectors[i][j][m] );
+                 fscanf(fp, "%lf\t",   &fPFTofVectors[i][j][m]  );
+                 fscanf(fpc, "%lf\t",  &fPFTofVectorsCoarse[i][j][m]  );  
                }
              
+             fPeakFinderVectors->SetVector( i, j, fPFAmpVectors[i][j], fPFTofVectors[i][j],    
+                                            fPFAmpVectorsCoarse[i][j], fPFTofVectorsCoarse[i][j] );   
+                                                    
              fclose (fp);
+             fclose (fpc);
            }
        }
     }
 }
+
+
+void   
+AliCaloRawAnalyzerPeakFinder::WriteRootFile() const
+{
+  // Utility function to write Peak-Finder vectors to an root file
+  // The output is used to create an OCDB entry.
+  fPeakFinderVectors->PrintVectors();
+  TFile *f = new TFile("peakfindervectors2.root",  "recreate" );
+  fPeakFinderVectors->Write();
+  f->Close();
+  delete f;
+}