RegionCuts.hpp

Go to the documentation of this file.

#ifndef REGION_CUTS_HPP
#define REGION_CUTS_HPP
 
#include <TCutG.h>
#include <TH2F.h>
#include <TString.h>

namespace RegionCutStore {
TString Dir();
TString Path(const char *name, Int_t reac);
void Save(Int_t reac, TCutG *cut_an, TCutG *cut_aa,
          Double_t n_an_assigned = -1.0);
TCutG *Load(const char *name, Int_t reac);
TCutG *LoadDrawn(const char *name, Int_t reac);
Double_t LoadAssigned(const char *name, Int_t reac);
} // namespace RegionCutStore

struct Gauss2D {
  Double_t amp = 0.0; 
  Double_t mx = 0.0;  
  Double_t sx = 0.0;  
  Double_t my = 0.0;  
  Double_t sy = 0.0;  
  Double_t rho = 0.0; 
};

struct RegionFit {
  Gauss2D beam; 
  Gauss2D reac; 
  Double_t x_lo = 0.0, x_hi = 0.0, y_lo = 0.0, y_hi = 0.0; 
  Double_t n_beam = 0.0, n_reac = 0.0;
  Bool_t has_reac = kTRUE;
  Bool_t ok = kFALSE; 
  TString why;        
};
 
namespace RegionCutStore {
void SaveFit(Int_t reac, const RegionFit &fit);

Bool_t LoadFit(Int_t reac, RegionFit &fit);
} // namespace RegionCutStore

namespace RegionCutFinder {
RegionFit FitMixture(TH2F *scatter, Int_t reac, Double_t x_lo, Double_t x_hi,
                     Double_t y_lo, Double_t y_hi);

TCutG *EllipseCut(const char *name, const Gauss2D &g, Double_t nsigma,
                  Int_t npts = 64);

RegionFit FitBeam(TH2F *scatter, Int_t reac, Double_t x_lo, Double_t x_hi,
                  Double_t y_lo, Double_t y_hi);

Double_t AboveRidge(const Gauss2D &beam, Double_t x, Double_t y);

TCutG *RidgeBandCut(const char *name, const Gauss2D &beam, Double_t nsig_lo,
                    Double_t nsig_hi, Double_t x_lo, Double_t x_hi,
                    Double_t y_lo, Double_t y_hi);

Double_t CountInside(TH2F *scatter, TCutG *cut, const RegionFit &fit);

void SaveFigures(TH2F *scatter, Int_t reac, const RegionFit &fit, TCutG *an,
                 TCutG *aa, const TString &subdir);
} // namespace RegionCutFinder
 
#endif