PSDCutFinder.h

Go to the documentation of this file.

#ifndef _PSD_CUT_FINDER_H_
#define _PSD_CUT_FINDER_H_
 
 
#include "TF1.h"
#include "TTree.h"
#include "TH1.h"
#include "TH2.h"
#include "TH3.h"
#include "TGraph.h"
#include "TCanvas.h"
 
#include <string>
#include <vector>
 
 
struct FitParameters
{
  
  Double_t mu;
  Double_t D;
  Double_t sigma1;
  Double_t sigma2;
  Double_t area1;
  Double_t area2;
 
};
 
 
class PSDCutFinder
{
public:
  PSDCutFinder(TFile* a_rootFile, std::string a_psdHistoName);
  
  PSDCutFinder(TH2F* a_psdHisto);
  
  ~PSDCutFinder();
  
  void setProjections(double a_xmin,
                      double a_xmax,
                      const std::vector<unsigned int> &a_bins);
  
  void setProjections(double a_xmin,
                      double a_xmax,
                      unsigned int a_start,
                      unsigned int a_stop, 
                      unsigned int a_step=1);
 
  void drawPolyCut(std::string a_psdHistoName);
 
  void clear(); 
  
  std::vector< std::pair<Double_t, Double_t> > 
  findPeaks(unsigned int a_hindex);
 
  static Double_t totalFit(Double_t *x, Double_t *par);
 
  std::vector<Double_t> 
  doFit( unsigned int a_hindex, std::vector< std::pair<Double_t, Double_t> > a_peaks );
 
  const std::vector<unsigned int> &getBins() const;
  
  std::vector<TH1D*> m_projY;
 
private:
  FitParameters getCutParameters();
 
private:
  std::string m_psdHistoName;
  TH2F* m_psdHisto;
  std::vector<unsigned int> m_bins;
  
    
  double m_xmin;
  double m_xmax;
  int m_npeaks;
};
 
#endif