BaselineWatcher.cpp

Go to the documentation of this file.

#include <iostream>
#include <iomanip>
#include <fstream>
#include <sstream>
#include <vector>
#include <string>
#include <map>

#include "TCanvas.h"
#include "TROOT.h"
#include "TStyle.h"
#include "TH1.h"
#include "TF1.h"
#include "TSpectrum.h"
#include "TObjArray.h"
#include "BaselineWatcher.h"
#include "TBrowser.h"
#include "TMath.h"
#include "TVirtualFitter.h"
#include "TPavesText.h"
#include "TObject.h"
#include "FitSpectra.h"

#include "LoupeOnPad.h"

using namespace std;

//ClassImp(BaselineWatcher);

BaselineWatcher::BaselineWatcher(const char* name, const char *title) : SpectraViewerWatcher(name,title)
{
    fDefinePersoBackground = false;
}

void BaselineWatcher::Exec(Option_t *option)
{
    TString Opt = option;
    Opt.ToLower();

    TString FitType = "CBDS";

    if(Opt.Contains("baseline"))
    {
        TString CrystalName = fCrystalName;
        CrystalName.ToLower();

        if(Opt.Contains("all") || Opt.Contains(CrystalName))
        {
            ShowBaselinesMap();

            if(Opt.Contains("calib"))
            {
                if(Opt.Contains("gaus")) FitType = "Gaus";

                FitAll(FitType);
            }
        }
    }
}

void BaselineWatcher::FitAll(Option_t *FitType)
{
    cout<<"Baseline parameters for crystal "<<fCrystalName<<endl;
    cout<<"Seg"<<"\t\t\t"<<"GausMean"<<"\t\t"<<"FWHM"<<endl;

    TCanvas *CurrentCanvas = gPad->GetCanvas();

    int Pad=0;
    while(CurrentCanvas->cd(Pad) != CurrentCanvas->cd(Pad+1))
    {
        Pad++;
        CurrentCanvas->cd(Pad);
        FitSelected(FitType,true);
    }

    CurrentCanvas->cd();
    CurrentCanvas->Update();
    CurrentCanvas->Modified();

    ofstream FileOut(((TString)"FitResults/" + fCrystalName + "_Baselines.fit"));
    FileOut<<"#Seg"<<"\t"<<"GausMean"<<"\t"<<"FWHM"<<endl;
    for(int i=0 ; i<38 ; i++)FileOut<<i<<"\t"<<TabMean[i]<<"\t\t"<<TabFWHM[i]<<endl;
    FileOut.close();
}

void BaselineWatcher::FitSelected(Option_t *FitType, bool all)
{
    FitSpectra *FS = new FitSpectra(FitType,"baseline");
    FS->GetCurrentHistogram();
    if(fDefinePersoBackground)
    {
        FS->SetPersoBackground(fNumberIterations,fDirection,fFilterOrder,fSmoothing,fsmoothingWindow,fCompton);
    }
    FS->Fit();


    int HistNbr = -1;

    if(FS->GetHistogram()!=0x0)
    {
        if(!all)
        {
            cout<<"Baseline parameters for crystal "<<fCrystalName<<endl;
            cout<<"Seg"<<"\t\t\t"<<"GausMean"<<"\t\t"<<"FWHM"<<endl;
        }

        TString Name = FS->GetHistName();
        TObjArray *array = Name.Tokenize("_");
        TObject *LastToken = array->At(array->GetLast());

        HistNbr = ((TString)LastToken->GetName()).Atoi();

        PadPos HistPosition = fHistPositionList[HistNbr];

        TString SegName = HistPosition.SegName;

        double Mean = FS->GetMean().at(0);
        double FWHM = FS->GetFWHM().at(0);

        TabMean[HistNbr] = Mean;
        TabFWHM[HistNbr] = FWHM;

        if(SegName.Contains("Core"))cout<<"Seg "<<HistNbr<<"("<<SegName<<") :\t"<<Mean<<"\t\t"<<FWHM<<endl;
        else cout<<"Seg "<<HistNbr<<"("<<SegName<<") :  \t"<<Mean<<"\t\t"<<FWHM<<endl;

    }
    else if(gPad->GetListOfPrimitives()->GetSize()>1)
    {
        TString Name = FS->GetHistName();
        TObjArray *array = Name.Tokenize("_");
        TObject *LastToken = array->At(array->GetLast());

        HistNbr = ((TString)LastToken->GetName()).Atoi();

        PadPos HistPosition = fHistPositionList[HistNbr];

        TString SegName = HistPosition.SegName;

        if(SegName.Contains("Core"))cout<<"Seg "<<HistNbr<<"("<<SegName<<") :\t"<<0<<"\t\t"<<0<<"\t Broken Segment"<<endl;
        else cout<<"Seg "<<HistNbr<<"("<<SegName<<") :  \t"<<0<<"\t\t"<<0<<"\t Broken Segment"<<endl;

        TabMean[HistNbr] = 0.;
        TabFWHM[HistNbr] = 0.;
    }

    //delete FS;
}

void BaselineWatcher::ShowBaselinesMap(int LibraryNumber, Option_t *option)
{
    TString opt = (TString)option;
    TCanvas *c = 0x0;

    if(opt.Contains("same",TString::kIgnoreCase)) c = (TCanvas*)gPad->GetCanvas();
    else
    {
        TString Name = "Baselines_" + fCrystalName;
        TString Title = "Mapping of the baselines for crystal " + fCrystalName;

        c = CanvasVisu::BuildSegmentedCanvas(Name,Title);
    }
    if(c == 0x0) return;

    c->cd();

    TList *ListOfHists = (TList*)fListOfLibrairies->At(LibraryNumber);

    for(int i=0 ; i<fNSg ; i++)
    {
        PadPos HistPosition = fHistPositionList[i];

        TPad *pad = (TPad*)c->cd(HistPosition.PadNmbr);

        TH1 *h = 0x0;

        TPaveText *p = (TPaveText*)pad->GetListOfPrimitives()->FindObject("InfoPad");

        h = ((TH1*)ListOfHists->At(i));
        h->GetXaxis()->SetLabelSize(0.07);
        h->GetYaxis()->SetLabelSize(0.07);

        int col = 0;
        for(int k=0 ; k<pad->GetListOfPrimitives()->GetSize() ; k++)
        {
            if(pad->GetListOfPrimitives()->At(k)->InheritsFrom("TH1"))col++;
        }

        h->SetLineColor(GetColor(col));
        h->SetFillColor(0);
        h->SetStats(0);
        h->Draw(opt);

        if(p)p->Draw();

        pad->Modified();
        pad->Update();
    }
    for(int i=0 ; i<fNC ; i++)
    {
        PadPos HistPosition = fHistPositionList[fNSg+i];

        TPad *pad = (TPad*)c->cd(HistPosition.PadNmbr);

        TH1 *h = 0x0;

        TPaveText *p = (TPaveText*)pad->GetListOfPrimitives()->FindObject("InfoPad");

        h = ((TH1*)ListOfHists->At(fNSg+i));
        h->GetXaxis()->SetLabelSize(0.07);
        h->GetYaxis()->SetLabelSize(0.07);

        int col = 0;
        for(int k=0 ; k<pad->GetListOfPrimitives()->GetSize() ; k++)
        {
            if(pad->GetListOfPrimitives()->At(k)->InheritsFrom("TH1"))col++;
        }
        h->SetLineColor(GetColor(col));
        h->SetFillColor(0);
        h->SetStats(0);
        h->Draw(opt);

        if(p)p->Draw();

        pad->Modified();
        pad->Update();
    }

    if(!opt.Contains("same"))
    {
        new LoupeOnPad(c);
        SetLoupe(false);
    }
}

void BaselineWatcher::SetPersoBackground(int NumberIterations, int Direction,int FilterOrder,bool Smoothing,int smoothingWindow,bool Compton)
{
    fDefinePersoBackground = true;
    fNumberIterations = NumberIterations;
    fDirection = Direction;
    fFilterOrder = FilterOrder;
    fSmoothing = Smoothing;
    fsmoothingWindow = smoothingWindow;
    fCompton = Compton;
}