PndHypGeSpectrumAnalyser.h

Go to the documentation of this file.

/*
 * PndHypGeSpectrumAnalyser.h
 *
 * Copyright 2013 Marcell Steinen <steinen@kph.uni-mainz.de>
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
 * MA 02110-1301, USA.
 *
 *
 */

// "IsA()" Error is given by ClassDef line. This line must be used for Go4 and PANDAroot.

#ifndef PNDHYPGESPECTRUMANALYSER_H
#define PNDHYPGESPECTRUMANALYSER_H

#include "TH1D.h"
#include "TSpectrum.h"
#include "TMath.h"
#include "TF1.h"
#include "TGraphErrors.h"
#include "TCanvas.h"
#include "TFile.h"

#include <iostream>
#include <vector>
#include <map>
#include <fstream>

using namespace std;

class PndHypGeSpectrumAnalyser {
 public:
  PndHypGeSpectrumAnalyser(TH1D *hEnergySpec_ext, Int_t nPeaks = 2, Int_t FuncWidthExt = 100);
  PndHypGeSpectrumAnalyser(TH1D *hEnergySpec_ext, vector<double> *Energies_ext, Int_t FuncWidthExt = 100); // constructor with energies from external array
  PndHypGeSpectrumAnalyser(TH1D *hEnergySpec_ext, TString Nuclei, Int_t FuncWidthExt = 100);               // constructor with energies from internal database
  virtual ~PndHypGeSpectrumAnalyser();

  TH1D *GetEnergySpectrum() { return fhEnergySpectrum; }
  void SetEnergySpectrum(TH1D *hEnergySpec_ext);

  Int_t AnalyseSpectrum();
  Int_t AnalyseSimulationSpectrum();
  Int_t FindPeaks();
  Int_t FitPeaks();

  Int_t DrawCalibratedSpectrum();
  Int_t DrawSimulationSpectrum();
  Int_t DoEnergyCalibration();
  Int_t CalculateFWHM();
  Int_t CalculateFWTM();
  Double_t CalculateEfficiency();
  Double_t GetFWHM511();
  Double_t GetFWTM511();
  Double_t GetFWHMCo1();
  Double_t GetFWTMCo1();
  Double_t GetFWHMCo();
  Double_t GetFWTMCo();
  Double_t GetFWHMAl();
  Double_t GetFWTMAl();
  Double_t GetEfficiency();
  Double_t GetEfficiencyError();

  Double_t GetCountsCo(); // NIY

  Int_t CompareNuclei(TString NucleiName); // database comparison

  void SetTxtFileOutputName(TString TxtFilename_ext);
  void SetRootFileOutputName(TString RootFilename_ext);
  void SetOutputPath(TString OutputPath_ext);

  Int_t ExportToTextFile(TString TxtFilename_ext = "test.txt");
  Int_t ExportToRootFile(TString RootFilename_ext = "test.root");

  void SetSearchRange(Double_t RangeMin, Double_t RangeMax);

  void SetGaussianFitting();
  Bool_t IsGaussianFitting();

  void SetFreeSkewedFitting();
  Bool_t IsFreeSkewedFitting();

  void SetSecondGausianFitting();
  Bool_t IsSecondGausianFitting();

  void SetNewFunctionFitting();
  Bool_t IsNewFunctionFitting();
  void SetNoDrawingMode(Bool_t NoDrawingMode_ext = true); // NIY
  Bool_t IsNoDrawingMode();                               // NIY

  void SetInputHistogramResolution(Double_t InputHistogramResolution_ext);
  void SetNumberOfSimEvents(Int_t NumberOfSimEvents_ext, Bool_t OnlyTwoPiSimulated = 1);

 private:
  Double_t FindUpperSigmaFitLimit(Double_t threshold, Float_t StartingPoint);
  void PeakSort();
  Double_t Calibrate(Double_t Channel);

  Double_t CalculateLowerErrorLeft(Double_t threshold, TH1D *histo, Double_t StartingValue);
  Double_t CalculateUpperErrorLeft(Double_t threshold, TH1D *histo, Double_t StartingValue);
  Double_t CalculateLowerErrorRight(Double_t threshold, TH1D *histo, Double_t StartingValue);
  Double_t CalculateUpperErrorRight(Double_t threshold, TH1D *histo, Double_t StartingValue);

  /* add your private declarations */
  TCanvas *fEnergySpecCanvas;
  TH1D *fhEnergySpectrum;

  TCanvas *fCalSpecCanvas;
  TH1D *fhCalibratedSpectrum;

  TCanvas *fSubstractCanvas;
  TH1D *fhSubstractSpectrum;
  TSpectrum *fSpectrum;
  Float_t *PeaksPosX; 
  Float_t *PeaksPosY; 

  Int_t nPeaks;
  Int_t nPeaksFound; // number of Peaks found by fSpectrum->Search("...")

  TF1 *FitFunc[50];
  TF1 *FitFuncWithoutBg[50];
  TF1 *CalibratedFunction[50];
  TF1 *FuncGaus[50];
  TF1 *FuncSmoothedStep[50];
  TF1 *FuncLinear[50];
  TF1 *FuncTail[50];

  Int_t FuncWidth;
  vector<double> PeakFitX;
  vector<double> PeakFitXError;
  vector<double> PeakCounts;

  TGraphErrors *fgCalibration;
  TCanvas *fCalCanvas;
  TF1 *CalFunc;

  Double_t Amplitude[50];
  Double_t FWHMHeight[50];
  Double_t FWHMlow[50];
  Double_t FWHMlowLeftBorder[50];
  Double_t FWHMlowRightBorder[50];
  Double_t FWHMlowLeftBorderEnergy[50];
  Double_t FWHMlowRightBorderEnergy[50];
  Double_t FWHMhigh[50];
  Double_t FWHMhighLeftBorder[50];
  Double_t FWHMhighRightBorder[50];
  Double_t FWHMhighLeftBorderEnergy[50];
  Double_t FWHMhighRightBorderEnergy[50];
  Double_t FWHMlowEnergy[50];
  Double_t FWHMhighEnergy[50];

  Double_t FWTMHeight[50];
  Double_t FWTMlow[50];
  Double_t FWTMhigh[50];
  Double_t FWTMlowEnergy[50];
  Double_t FWTMhighEnergy[50];

  TH1D *fhFitErrorhistogram[50];

  map<double, double> FWHM_ch;
  map<double, double> FWHM_en;
  map<double, double> FWTM_ch;
  map<double, double> FWTM_en;

  vector<double> Energies;     // vector of peak energies
  vector<double> EnergyErrors; // vector of peak energy errors

  TString OutputPath;

  ofstream TxtFile;
  TString TxtFilename;

  TFile *RootFile;
  TString RootFilename;

  Bool_t BreakProgram;

  Int_t fFittingMethod;

  Bool_t NoDrawingMode;

  Double_t fFEPEfficiency;
  Double_t fFEPEfficiencyError;
  Double_t fInputHistogramResolution;
  Int_t fNumberOfSimEvents;
  Int_t fSolidAngleCorrectionFactor;

  ClassDef(PndHypGeSpectrumAnalyser, 1) // This line must be used for Go4 and PANDAroot !!!!!!!!!!!   gives "IsA()" Error if compiled with g++
};

#endif /* PNDHYPGESPECTRUMANALYSER_H */