StFcsPulseAna.h

//Author:David Kapukchyan
/*
@[December 17, 2022]
> SetFitPars can now set a function parameters for a range of in npeaks as opposed to all peaks

@[June 25, 2022]
 > Added Refinements to drawing and copying. Also made some changes to how the tunnel probability is used to merge the peaks.

@[March 3, 2022]
 > Copied StFcsPulseFit so that I can inherit from and use the PeakAna class methods from MyTools
*/

#ifndef STFCSPULSEANA_H
#define STFCSPULSEANA_H

//ROOT Headers
#include "TMath.h"
#include "TLegend.h"
#include "TLegendEntry.h"
#include "TF1.h"
#include "TGraphAsymmErrors.h"
#include "TMultiGraph.h"
#include "TH1.h"

#include "TLine.h"
#include "TPaveText.h"
#include "TPaveStats.h"

#include "StFcsDbMaker/StFcsDbPulse.h"
#include "PeakAna.h"

class StFcsPulseAna : public PeakAna
{
 public:
  StFcsPulseAna();                    
  StFcsPulseAna( std::string name );  
  explicit StFcsPulseAna( TGraph* Sig, std::string name = "StFcsPulseAna"); 
  StFcsPulseAna(const StFcsPulseAna& old, const char* post_name="_copy", TGraph* graph=0); 
  StFcsPulseAna& operator=(const StFcsPulseAna& rhs);   

  virtual ~StFcsPulseAna();  

  virtual void Copy(TObject& obj) const;              
  virtual TObject* Clone(const char* newname) const;  

  virtual Int_t AnalyzeForPeak();  
  virtual Int_t AnalyzeForPeak(Double_t peak, Double_t width){ return PeakAna::AnalyzeForPeak(peak,width); }//<! same as #AnalyzeForPeak() but also sets search parameters for peak
  virtual Int_t AnalyzeForNoisyPeak(){ return PeakAna::AnalyzeForNoisyPeak(); }//<! First calls #ConvertToAna() function and then #AnalyzeForPeak() on the converted data

  static double MaxwellBoltzmannDist(double* x, double* p);
  static void GetMBPars(const double& xpeak, const double& xrise, const double& yh, const double& ped, double& height, double& scale );

  virtual TGraphAsymmErrors* GetData()const{return (TGraphAsymmErrors*)mG_Data;} 

  const char* GetName() const {return mName.c_str();} 
  std::string& Name() {return mName;}                 
  const std::string& Name() const {return mName;}     

  void setDbPulse(StFcsDbPulse* p){mDbPulse = p;}     
  
  void ResetFinder();   
  void ResetBaseline(); 
  void ResetSum();      
  
  Int_t Sum(Int_t Start, Int_t End);                
  Int_t SumWindow();                                
  Double_t GausFit(Int_t Start=0,Int_t End=0);      
  void SignalMBPars(double& height, double& scale); 
  double SumMB();                                   
  Double_t MBFit(Int_t Start=0,Int_t End=0);        
  Double_t PulseFit(Int_t Start=0, Int_t End=0);    

  void SetFitPars(TF1*& func, int start=-1, int end=-1);   

  static void FillAdc(TGraphAsymmErrors* g, unsigned short& counter, int Start, unsigned short* adcdata);
  static int SumDep0(TGraphAsymmErrors* gdata, int Start, int ped=0);     
  static int SumDep0Mod(TGraphAsymmErrors* gdata, int Start, int ped=0);  

  void SetFitSignal(TF1* func){mF1_SignalFit=func;}      
  void SetBaselineFit(TF1* func){mF1_BaselineFit=func;}  

  void SetZS(){SetBaseline(0,0.39);} 

  void AnalyzeForPedestal(); 
  
  TH1F* BaselineHist()const{return mH1_Baseline;} 
  TF1* SignalFit(const char option ='n');//Replace with a function that actually calls "fit" and then returns the function?Arguments can be values and range?A little more complicated since hard to tell when user would want to call a fit or just get the function itself?Maybe no arguments means give me function, arguments means do a refit?
  TF1* SignalFit() const {return mF1_SignalFit;}    
  TF1* BaselineFit()const{return mF1_BaselineFit;}  

  virtual StFcsPulseAna* DrawCopy(Option_t* opt="",const char* name_postfix = "_copy", TGraph* graph=0) const; 

  virtual void Print(Option_t* opt="") const; 

  virtual void MergeByProbability(std::vector<PeakWindow>& merged) const; 

 protected:
  void Init(); 

  StFcsDbPulse* mDbPulse; 

  TH1F* mH1_Baseline;     
  //Main fit functions
  TF1*  mF1_SignalFit;    
  TF1*  mF1_BaselineFit;  

  bool FindBaseline();    

 private:
  
  std::string mName;      

  bool mWindowSum;        
  bool mFitSum;           
    
  //Variables to hold sum values.  This is to avoid tedious recomputation
  Int_t mSumAdc;          
  Double_t mSumAdcFit;    

  ClassDef(StFcsPulseAna,1)

};

#endif