BeamFit2D.hpp

Go to the documentation of this file.

#ifndef BEAM_FIT_2D_HPP
#define BEAM_FIT_2D_HPP
 
#include <Rtypes.h>
#include <TH1F.h>
#include <TH2F.h>
#include <TString.h>
#include <algorithm>
#include <cmath>
#include <utility>
#include <vector>

struct BeamFit2D {
  Double_t amp = 0;                  
  Double_t mu_x = 0, mu_y = 0;       
  Double_t sigma_x = 0, sigma_y = 0; 
  Double_t rho = 0;                  
  Bool_t ok = kFALSE;                
};

struct Moments2D {
  Double_t mu_x = 0, mu_y = 0;       
  Double_t sigma_x = 0, sigma_y = 0; 
  Double_t rho = 0;                  
  Double_t weight = 0;               
};

class BeamFitUtils {
public:
  static Bool_t InEllipseXY(const BeamFit2D &b, Double_t x, Double_t y,
                            Double_t n);

  static void DrawEllipse(const BeamFit2D &b, Double_t n, Color_t color);

  static Moments2D ComputeMoments(TH2F *h, Int_t lo_bx, Int_t hi_bx,
                                  Int_t lo_by, Int_t hi_by, Double_t thresh,
                                  Double_t bw_x, Double_t bw_y);

  static Moments2D
  ClippedMoments(const std::vector<std::pair<Float_t, Float_t>> &pts,
                 const Moments2D &seed, Double_t clip = 3.0);

  static Moments2D SeedSpotMoments(TH2F *h, Int_t seed_half_bins);

  struct SpotFit {
    BeamFit2D fit;            
    Double_t chi2_ndf = -1.0; 
    Bool_t fit_used = kFALSE; 
  };

  static SpotFit
  FitSpotFromPoints(TH2F *h, std::vector<std::pair<Float_t, Float_t>> &pts,
                    const Moments2D &seed);

  static BeamFit2D FitSpot(TH2F *h, const Moments2D &seed,
                           Double_t window_nsigma = 2.0,
                           Double_t *chi2_ndf = nullptr);

  static Bool_t FitPeak(TH1F *h, Double_t seed_mu, Double_t seed_sigma,
                        Double_t window_nsigma, Bool_t fix_mu, Double_t &mu,
                        Double_t &sigma, Double_t *chi2_ndf = nullptr);
};
 
#endif