Factorization of the different raw fitting algorithms in EMCAL (Per Thomas)
[u/mrichter/AliRoot.git] / EMCAL / AliCaloRawAnalyzer.h
CommitLineData
168c7b3c 1// -*- mode: c++ -*-
57839add 2#ifndef ALICALORAWANALYZER_H
3#define ALICALORAWANALYZER_H
d655d7dd 4/**************************************************************************
5 * This file is property of and copyright by *
e37e3c84 6 * the Relatvistic Heavy Ion Group (RHIG), Yale University, US, 2009 *
d655d7dd 7 * *
8 * Primary Author: Per Thomas Hille <p.t.hille@fys.uio.no> *
9 * *
10 * Contributors are mentioned in the code where appropriate. *
11 * Please report bugs to p.t.hille@fys.uio.no *
12 * *
13 * Permission to use, copy, modify and distribute this software and its *
14 * documentation strictly for non-commercial purposes is hereby granted *
15 * without fee, provided that the above copyright notice appears in all *
16 * copies and that both the copyright notice and this permission notice *
17 * appear in the supporting documentation. The authors make no claims *
18 * about the suitability of this software for any purpose. It is *
19 * provided "as is" without express or implied warranty. *
20 **************************************************************************/
21
22//Base class for extraction
23//of signal amplitude and peak position
57839add 24//From CALO Calorimeter RAW data
d655d7dd 25#include "Rtypes.h"
26#include "TObject.h"
92d9f317 27///#define MAXSAMPLES 1008 //CRAP PTH
e37e3c84 28#include <vector>
168c7b3c 29#include "AliCaloConstants.h"
92d9f317 30using namespace ALTRO;
31using namespace CALO;
168c7b3c 32
d655d7dd 33
57839add 34class AliCaloBunchInfo;
99730246 35#include "AliCaloFitResults.h"
d655d7dd 36
57839add 37class AliCaloRawAnalyzer : public TObject
d655d7dd 38{
92d9f317 39 friend class AliCaloRawAnalyzerFactory;
40
d655d7dd 41 public:
48a2e3eb 42 AliCaloRawAnalyzer(const char *name="AliCaloRawAnalyzer", const char *nameshort="RawAna");
57839add 43 virtual ~AliCaloRawAnalyzer();
99730246 44 virtual AliCaloFitResults Evaluate( const std::vector<AliCaloBunchInfo> &/*bunchvector*/,
45 const UInt_t /*altrocfg1*/, const UInt_t /*altrocfg2*/ ) {return AliCaloFitResults( Ret::kInvalid , Ret::kInvalid);}
168c7b3c 46 //enum fitAlgorithm {kCrude, kPeakFinder, kLMS, kFastFit, kNeuralNet, kNONE};
ce95bae9 47 void PrintBunches( const std::vector<AliCaloBunchInfo> &bunchvector ) const;
57839add 48 void PrintBunch( const AliCaloBunchInfo &bunch ) const ;
168c7b3c 49
92d9f317 50 int PreFitEvaluateSamples( const std::vector<AliCaloBunchInfo> &bunchvector,
e37e3c84 51 const UInt_t altrocfg1, const UInt_t altrocfg2, Int_t & index,
92d9f317 52 Float_t & maxf, short & maxamp, short & maxampindex,
53 Float_t & ped, int & first, int & last, const int acut);
54
d655d7dd 55 void SetTimeConstraint(const int min, const int max );
56 void SetVerbose(bool verbose = true){ fVerbose = verbose; };
57 void SetIsZeroSuppressed(const bool iszs = true) { fIsZerosupressed = iszs; } ;
58 void SetAmpCut(const Float_t cut) { fAmpCut = cut ; } ;
57839add 59 void SetFitArrayCut(const Int_t cut) { fFitArrayCut = cut ; } ;
60 void SetNsampleCut(const Int_t cut) { fNsampleCut = cut ; } ;
2cd0ffda 61 void SetOverflowCut(const Int_t cut) { fOverflowCut = cut ; } ;
f57baa2d 62 void SetNsamplePed(const Int_t i) { fNsamplePed = i ; } ;
63
64 bool GetIsZeroSuppressed() const { return fIsZerosupressed;} ;
65 Float_t GetAmpCut() const { return fAmpCut; } ;
66 Int_t GetFitArrayCut() const { return fFitArrayCut; } ;
67 Int_t GetNsampleCut() const { return fNsampleCut; } ;
2cd0ffda 68 Int_t GetOverflowCut() const { return fOverflowCut; } ;
f57baa2d 69 Int_t GetNsamplePed() const { return fNsamplePed; } ;
57839add 70
71 // access to array info
72 Double_t GetReversed(const int i) const { return fReversed[i]; }
e37e3c84 73 const char * GetAlgoName() const { return fName; };
48a2e3eb 74 const char * GetAlgoAbbr() const { return fNameShort; };
168c7b3c 75 Algo::fitAlgorithm GetAlgo() const { return fAlgo; };
d655d7dd 76
947c8e28 77 Double_t CalculateChi2(const Double_t amp, const Double_t time,
78 const Int_t first, const Int_t last,
79 const Double_t adcErr=1,
80 const Double_t tau=2.35);
81
82 void CalculateMeanAndRMS(const Int_t first, const Int_t last,
83 Double_t & mean, Double_t & rms);
92d9f317 84 void SetL1Phase(const Double_t phase) {fL1Phase = phase;};
85
86protected:
3d841a9f 87public: // PAI
92d9f317 88 short Max( const AliCaloBunchInfo *const bunch, int *const maxindex);
89 UShort_t Max(const UShort_t *data, const int length );
507751ce 90 bool CheckBunchEdgesForMax( const AliCaloBunchInfo *const bunch) const;
92d9f317 91 bool IsInTimeRange( const int maxindex, const int maxtime, const int mintime );
57839add 92 Float_t ReverseAndSubtractPed( const AliCaloBunchInfo *bunch, const UInt_t altrocfg1, const UInt_t altrocfg2, double *outarray ) const;
92d9f317 93 int SelectBunch( const std::vector<AliCaloBunchInfo> &bunchvector, short *const maxampbin, short *const maxamplitude );
94 void SelectSubarray( const Double_t *date, const int length, const short maxindex, int *const first, int *const last, const int cut);
57839add 95 Float_t EvaluatePedestal(const UShort_t * const data, const int length ) const;
92d9f317 96 Double_t fReversed[ALTROMAXSAMPLES]; //Reversed sequence of samples (pedestalsubtracted)
d655d7dd 97 // private:
98 int fMinTimeIndex; //The timebin of the max signal value must be between fMinTimeIndex and fMaxTimeIndex
99 int fMaxTimeIndex; //The timebin of the max signal value must be between fMinTimeIndex and fMaxTimeIndex
100 int fFitArrayCut; //Cut on ADC value (after ped. subtraction) for signals used for fit
101 Float_t fAmpCut; //Max ADC - pedestal must be higher than this befor attemting to extract the amplitude
102 int fNsampleCut; //Minimum number of sample require before attemting to extract signal parameters
2cd0ffda 103 int fOverflowCut; // value when ADC starts to saturate
f57baa2d 104 int fNsamplePed; //Number of samples used for pedestal calculation (first in bunch)
d655d7dd 105 bool fIsZerosupressed; //Wether or not the data is zeros supressed, by default its assumed that the baseline is also subtracted if set to true
106 bool fVerbose; //Print debug information to std out if set to true
e37e3c84 107 char fName[256]; // Name of the algorithm
48a2e3eb 108 char fNameShort[256]; // Abbrevation for the name
168c7b3c 109 // CaloConstants fAlgo;
110 Algo::fitAlgorithm fAlgo;
92d9f317 111 // FILE *fFp;
112 Double_t fL1Phase;
168c7b3c 113
92d9f317 114 Double_t fAmp; // The amplitude in entities of ADC counts
115 Double_t fTof; // The amplitude in entities of ADC counts
116
f57baa2d 117 ClassDef(AliCaloRawAnalyzer, 2)
e37e3c84 118
d655d7dd 119};
120
121#endif