Logo ROOT  

Reference Guide
 
Loading...
Searching...
No Matches
FitAwmi.C File Reference
Tutorials » Spectrum tutorials

Detailed Description

View in nbviewer Open in SWAN
This macro fits the source spectrum using the AWMI algorithm from the "TSpectrumFit" class ("TSpectrum" class is used to find peaks).

To try this macro, in a ROOT (5 or 6) prompt, do:

root > .x FitAwmi.C

or:

root > .x FitAwmi.C++
root > FitAwmi(); // re-run with another random set of peaks
created -9.7 11.9683 3
created -9.1 39.8942 10
created -8.5 23.9365 6
created -7.9 11.9683 3
created -7.3 7.97885 2
created -6.7 27.926 7
created -6.1 15.9577 4
created -5.5 7.97885 2
created -4.9 31.9154 8
created -4.3 27.926 7
created -3.7 23.9365 6
created -3.1 15.9577 4
created -2.5 11.9683 3
created -1.9 23.9365 6
created -1.3 3.98942 1
created -0.7 27.926 7
created -0.1 23.9365 6
created 0.5 39.8942 10
created 1.1 31.9154 8
created 1.7 15.9577 4
created 2.3 31.9154 8
created 2.9 27.926 7
created 3.5 35.9048 9
created 4.1 3.98942 1
created 4.7 3.98942 1
created 5.3 11.9683 3
created 5.9 27.926 7
created 6.5 11.9683 3
created 7.1 7.97885 2
created 7.7 11.9683 3
created 8.3 19.9471 5
created 8.9 19.9471 5
created 9.5 19.9471 5
the total number of created peaks = 33 with sigma = 0.1
the total number of found peaks = 33 with sigma = 0.100002 (+-3.65148e-05)
fit chi^2 = 3.48892e-06
found -9.1 (+-0.000258641) 39.8939 (+-0.101928) 10.0001 (+-0.000836474)
found 0.5 (+-0.000259438) 39.8942 (+-0.101966) 10.0002 (+-0.000836783)
found 3.5 (+-0.0002723) 35.9045 (+-0.0966862) 9.0001 (+-0.000793454)
found 1.1 (+-0.000290412) 31.9155 (+-0.0912158) 8.00018 (+-0.000748562)
found -4.9 (+-0.000289429) 31.9152 (+-0.0911782) 8.00012 (+-0.000748253)
found 2.3 (+-0.000289987) 31.9153 (+-0.0911985) 8.00014 (+-0.00074842)
found -6.7 (+-0.000309046) 27.9257 (+-0.0852764) 7.00008 (+-0.00069982)
found -4.3 (+-0.00031093) 27.9261 (+-0.0853401) 7.00018 (+-0.000700342)
found -0.699998 (+-0.000309013) 27.9258 (+-0.0852773) 7.00009 (+-0.000699827)
found 2.9 (+-0.000311453) 27.9263 (+-0.0853585) 7.00022 (+-0.000700494)
found 5.9 (+-0.000309112) 27.9257 (+-0.0852782) 7.00008 (+-0.000699835)
found -8.5 (+-0.000335783) 23.9367 (+-0.0790092) 6.00017 (+-0.000648388)
found -3.7 (+-0.000335561) 23.9366 (+-0.0790013) 6.00014 (+-0.000648324)
found -1.9 (+-0.000333253) 23.9363 (+-0.0789357) 6.00005 (+-0.000647785)
found -0.0999989 (+-0.000336844) 23.937 (+-0.0790402) 6.00022 (+-0.000648643)
found 8.3 (+-0.00036722) 19.9471 (+-0.0721088) 5.0001 (+-0.00059176)
found 8.9 (+-0.000367917) 19.9473 (+-0.0721259) 5.00013 (+-0.000591901)
found 9.5 (+-0.000364324) 19.9472 (+-0.0720509) 5.00013 (+-0.000591285)
found -6.1 (+-0.000411347) 15.9579 (+-0.0645128) 4.00012 (+-0.000529424)
found -3.1 (+-0.000411614) 15.9579 (+-0.0645173) 4.00012 (+-0.000529461)
found 1.7 (+-0.000413982) 15.9582 (+-0.0645661) 4.00021 (+-0.000529861)
found -7.9 (+-0.00047576) 11.9685 (+-0.0558814) 3.0001 (+-0.00045859)
found 6.5 (+-0.000476147) 11.9685 (+-0.0558877) 3.00012 (+-0.000458642)
found -9.69999 (+-0.000476724) 11.9686 (+-0.0558939) 3.00013 (+-0.000458693)
found -2.5 (+-0.000477086) 11.9686 (+-0.0559012) 3.00013 (+-0.000458752)
found 5.3 (+-0.000475118) 11.9685 (+-0.0558735) 3.0001 (+-0.000458525)
found 7.7 (+-0.000475327) 11.9684 (+-0.0558745) 3.00009 (+-0.000458533)
found -5.5 (+-0.000587829) 7.97932 (+-0.0456814) 2.00016 (+-0.000374884)
found -7.3 (+-0.000586489) 7.97921 (+-0.045667) 2.00013 (+-0.000374765)
found 7.1 (+-0.000583797) 7.97901 (+-0.0456376) 2.00008 (+-0.000374525)
found -1.3 (+-0.000840441) 3.99002 (+-0.0323536) 1.00017 (+-0.000265509)
found 4.09998 (+-0.000833566) 3.98987 (+-0.0323178) 1.00013 (+-0.000265216)
found 4.7 (+-0.000827162) 3.98956 (+-0.0322796) 1.00005 (+-0.000264902)
#include "TROOT.h"
#include "TMath.h"
#include "TRandom.h"
#include "TH1.h"
#include "TF1.h"
#include "TCanvas.h"
#include "TSpectrum.h"
#include "TSpectrumFit.h"
#include "TPolyMarker.h"
#include "TList.h"
#include <iostream>
TH1F *FitAwmi_Create_Spectrum(void) {
Int_t nbins = 1000;
Double_t xmin = -10., xmax = 10.;
delete gROOT->FindObject("h"); // prevent "memory leak"
TH1F *h = new TH1F("h", "simulated spectrum", nbins, xmin, xmax);
h->SetStats(kFALSE);
TF1 f("f", "TMath::Gaus(x, [0], [1], 1)", xmin, xmax);
// f.SetParNames("mean", "sigma");
gRandom->SetSeed(0); // make it really random
// create well separated peaks with exactly known means and areas
// note: TSpectrumFit assumes that all peaks have the same sigma
Double_t sigma = (xmax - xmin) / Double_t(nbins) * Int_t(gRandom->Uniform(2., 6.));
Int_t npeaks = 0;
while (xmax > (xmin + 6. * sigma)) {
npeaks++;
xmin += 3. * sigma; // "mean"
f.SetParameters(xmin, sigma);
Double_t area = 1. * Int_t(gRandom->Uniform(1., 11.));
h->Add(&f, area, ""); // "" ... or ... "I"
std::cout << "created "
<< xmin << " "
<< (area / sigma / TMath::Sqrt(TMath::TwoPi())) << " "
<< area << std::endl;
xmin += 3. * sigma;
}
std::cout << "the total number of created peaks = " << npeaks
<< " with sigma = " << sigma << std::endl;
return h;
}
void FitAwmi(void) {
TH1F *h = FitAwmi_Create_Spectrum();
TCanvas *cFit = ((TCanvas *)(gROOT->GetListOfCanvases()->FindObject("cFit")));
if (!cFit) cFit = new TCanvas("cFit", "cFit", 10, 10, 1000, 700);
else cFit->Clear();
h->Draw("L");
Int_t i, nfound, bin;
Int_t nbins = h->GetNbinsX();
Double_t *source = new Double_t[nbins];
Double_t *dest = new Double_t[nbins];
for (i = 0; i < nbins; i++) source[i] = h->GetBinContent(i + 1);
TSpectrum *s = new TSpectrum(); // note: default maxpositions = 100
// searching for candidate peaks positions
nfound = s->SearchHighRes(source, dest, nbins, 2., 2., kFALSE, 10000, kFALSE, 0);
// filling in the initial estimates of the input parameters
Bool_t *FixPos = new Bool_t[nfound];
Bool_t *FixAmp = new Bool_t[nfound];
for(i = 0; i < nfound; i++) FixAmp[i] = FixPos[i] = kFALSE;
Double_t *Pos, *Amp = new Double_t[nfound]; // ROOT 6
Pos = s->GetPositionX(); // 0 ... (nbins - 1)
for (i = 0; i < nfound; i++) {
bin = 1 + Int_t(Pos[i] + 0.5); // the "nearest" bin
Amp[i] = h->GetBinContent(bin);
}
TSpectrumFit *pfit = new TSpectrumFit(nfound);
pfit->SetFitParameters(0, (nbins - 1), 1000, 0.1, pfit->kFitOptimChiCounts,
pfit->kFitAlphaHalving, pfit->kFitPower2,
pfit->kFitTaylorOrderFirst);
pfit->SetPeakParameters(2., kFALSE, Pos, FixPos, Amp, FixAmp);
// pfit->SetBackgroundParameters(source[0], kFALSE, 0., kFALSE, 0., kFALSE);
pfit->FitAwmi(source);
Double_t *Positions = pfit->GetPositions();
Double_t *PositionsErrors = pfit->GetPositionsErrors();
Double_t *Amplitudes = pfit->GetAmplitudes();
Double_t *AmplitudesErrors = pfit->GetAmplitudesErrors();
Double_t *Areas = pfit->GetAreas();
Double_t *AreasErrors = pfit->GetAreasErrors();
delete gROOT->FindObject("d"); // prevent "memory leak"
TH1F *d = new TH1F(*h); d->SetNameTitle("d", ""); d->Reset("M");
for (i = 0; i < nbins; i++) d->SetBinContent(i + 1, source[i]);
Double_t x1 = d->GetBinCenter(1), dx = d->GetBinWidth(1);
Double_t sigma, sigmaErr;
pfit->GetSigma(sigma, sigmaErr);
// current TSpectrumFit needs a sqrt(2) correction factor for sigma
sigma /= TMath::Sqrt2(); sigmaErr /= TMath::Sqrt2();
// convert "bin numbers" into "x-axis values"
sigma *= dx; sigmaErr *= dx;
std::cout << "the total number of found peaks = " << nfound
<< " with sigma = " << sigma << " (+-" << sigmaErr << ")"
<< std::endl;
std::cout << "fit chi^2 = " << pfit->GetChi() << std::endl;
for (i = 0; i < nfound; i++) {
bin = 1 + Int_t(Positions[i] + 0.5); // the "nearest" bin
Pos[i] = d->GetBinCenter(bin);
Amp[i] = d->GetBinContent(bin);
// convert "bin numbers" into "x-axis values"
Positions[i] = x1 + Positions[i] * dx;
PositionsErrors[i] *= dx;
Areas[i] *= dx;
AreasErrors[i] *= dx;
std::cout << "found "
<< Positions[i] << " (+-" << PositionsErrors[i] << ") "
<< Amplitudes[i] << " (+-" << AmplitudesErrors[i] << ") "
<< Areas[i] << " (+-" << AreasErrors[i] << ")"
<< std::endl;
}
d->SetLineColor(kRed); d->SetLineWidth(1);
d->Draw("SAME L");
TPolyMarker *pm = ((TPolyMarker*)(h->GetListOfFunctions()->FindObject("TPolyMarker")));
if (pm) {
h->GetListOfFunctions()->Remove(pm);
delete pm;
}
pm = new TPolyMarker(nfound, Pos, Amp);
h->GetListOfFunctions()->Add(pm);
pm->SetMarkerStyle(23);
pm->SetMarkerColor(kRed);
pm->SetMarkerSize(1);
// cleanup
delete pfit;
delete [] Amp;
delete [] FixAmp;
delete [] FixPos;
delete s;
delete [] dest;
delete [] source;
return;
}
d
#define d(i)
Definition RSha256.hxx:102
f
#define f(i)
Definition RSha256.hxx:104
h
#define h(i)
Definition RSha256.hxx:106
Bool_t
bool Bool_t
Definition RtypesCore.h:63
Int_t
int Int_t
Definition RtypesCore.h:45
kFALSE
constexpr Bool_t kFALSE
Definition RtypesCore.h:101
Double_t
double Double_t
Definition RtypesCore.h:59
kRed
@ kRed
Definition Rtypes.h:66
TRangeDynCast
ROOT::Detail::TRangeCast< T, true > TRangeDynCast
TRangeDynCast is an adapter class that allows the typed iteration through a TCollection.
Definition TCollection.h:358
dest
Option_t Option_t TPoint TPoint const char GetTextMagnitude GetFillStyle GetLineColor GetLineWidth GetMarkerStyle GetTextAlign GetTextColor GetTextSize void char Point_t Rectangle_t dest
Definition TGWin32VirtualXProxy.cxx:164
x1
Option_t Option_t TPoint TPoint const char x1
Definition TGWin32VirtualXProxy.cxx:70
xmin
float xmin
Definition THbookFile.cxx:95
xmax
float xmax
Definition THbookFile.cxx:95
gROOT
#define gROOT
Definition TROOT.h:414
gRandom
R__EXTERN TRandom * gRandom
Definition TRandom.h:62
TCanvas
The Canvas class.
Definition TCanvas.h:23
TF1
1-Dim function class
Definition TF1.h:233
TH1F
1-D histogram with a float per channel (see TH1 documentation)
Definition TH1.h:621
TPolyMarker
A PolyMarker is defined by an array on N points in a 2-D space.
Definition TPolyMarker.h:31
TRandom::SetSeed
virtual void SetSeed(ULong_t seed=0)
Set the random generator seed.
Definition TRandom.cxx:615
TRandom::Uniform
virtual Double_t Uniform(Double_t x1=1)
Returns a uniform deviate on the interval (0, x1).
Definition TRandom.cxx:682
TSpectrumFit
Advanced 1-dimensional spectra fitting functions.
Definition TSpectrumFit.h:18
TSpectrum
Advanced Spectra Processing.
Definition TSpectrum.h:18
TSpectrum::SearchHighRes
Int_t SearchHighRes(Double_t *source, Double_t *destVector, Int_t ssize, Double_t sigma, Double_t threshold, bool backgroundRemove, Int_t deconIterations, bool markov, Int_t averWindow)
One-dimensional high-resolution peak search function.
Definition TSpectrum.cxx:2129
TSpectrum::GetPositionX
Double_t * GetPositionX() const
Definition TSpectrum.h:58
sigma
const Double_t sigma
Definition h1analysisProxy.h:11
TMath::Sqrt2
constexpr Double_t Sqrt2()
Definition TMath.h:86
TMath::Sqrt
Double_t Sqrt(Double_t x)
Returns the square root of x.
Definition TMath.h:662
TMath::TwoPi
constexpr Double_t TwoPi()
Definition TMath.h:44
Author

Definition in file FitAwmi.C.