source: JEM-EUSO/esaf_lal/branches/camille/packages/analysis/average_time_distri.C @ 176

int main()
{
    void average_time_distri();
    average_time_distri();
    return 0;
}

//
//  test.C
//
//
//  Created by moretto on 13/06/13.
//
//


#include <TROOT.h>
#include <TRint.h>
#include <TFile.h>
#include <TGFileDialog.h>
#include <TCanvas.h>
#include <TStyle.h>
#include <TTree.h>
#include <TChain.h>
#include <TH1.h>
#include <TH2.h>
#include <TProfile.h>
#include <TF1.h>
#include <TRandom.h>
#include <TRandom1.h>
#include <TMath.h>
#include <TLegend.h>
#include <TGraph.h>
#include <TMarker.h>
#include <TArrow.h>
#include <TGraphErrors.h>
#include <TGraphAsymmErrors.h>
#include <TLatex.h>
#include <TLegend.h>
#include <TLegendEntry.h>
#include <TPaveText.h>
#include <TMath.h>



#include "Riostream.h"
#include "TCanvas.h"
#include "TFile.h"
#include "TTree.h"
#include "TControlBar.h"
#include "TSystem.h"
#include "TGFileDialog.h"
#include "TPaveText.h"
#include "TString.h"
#include "TView.h"
#include "TApplication.h"



#include "EEvent.hh"
#include "EDetPhoton.hh"
#include "EDetector.hh"


#include <iostream>
#include <cmath>
#include <cstring>
#include <list>
#include <utility>
#include <numeric>
#include <functional>
#include <algorithm>
#include <vector>

const char* inputfile1="/Users/moretto/Documents/thesis/ESAF/Code/esaf_lal/esaf/camille_work/KIT_phase1_study/30_ns_TimeResolution__100_event/simu.root";
const char* inputfile2="/Users/moretto/Documents/thesis/ESAF/Code/esaf_lal/esaf/camille_work/KIT_phase1_study/10_ns_TimeResolution__100_event/simu.root";
const char* inputfile3="/Users/moretto/Documents/thesis/ESAF/Code/esaf_lal/esaf/camille_work/KIT_phase1_study/5_ns_TimeResolution__100_event/simu.root";
const char* inputfile4="/Users/moretto/Documents/thesis/ESAF/Code/esaf_lal/esaf/camille_work/KIT_phase1_study/0_ns_TimeResolution__100_event/simu.root";
const char* inputfile5="/Users/moretto/Documents/thesis/ESAF/Code/esaf_lal/esaf/camille_work/KIT_phase1_study/1_ns_TimeResolution__100_event/simu.root";
const char* inputfile6="/Users/moretto/Documents/thesis/ESAF/Code/esaf_lal/esaf/camille_work/KIT_phase1_study/20_ns_TimeResolution__100_event/simu.root";
const char* inputfile7="/Users/moretto/Documents/thesis/ESAF/Code/esaf_lal/esaf/camille_work/KIT_phase1_study/bis_10_ns__100_event/simu.root";
const char* inputfile8="/Users/moretto/Documents/thesis/ESAF/Code/esaf_lal/esaf/camille_work/KIT_phase1_study/60_ns_TimeResolution__100_event/simu.root";


TFile *f;
EEvent*          ev = 0;
EDetector*       gDetector = NULL;
TTree* tree = 0;

int nb_entries = 0;
int nb_photons = 0;
int macrocell_status = 0;
int photon_type = 0;

void init(const char*, int index);
Bool_t loadevent(int i=0);
void process(int num);

//______________________________________________________________________________
void average_time_distri()
{
    init(inputfile8, 1);
    process(1);    
}

//______________________________________________________________________________
void init(const char* fname, int index)
{
    f = new TFile(fname);
    
    char ttree_name[50];
    sprintf(ttree_name, "tree_%d", index);
    
    tree = new TTree (ttree_name, "output tree from esaf simulation");
    tree = (TTree*)gDirectory->Get("etree");
    if ( tree )
    {
        nb_entries = tree->GetEntries();
        //Printf("# tree found with %d events stored.", nb_entries);
    }
    
}

//______________________________________________________________________________
Bool_t loadevent(int i)
{
    if ( !tree )
    {
        Printf("# ERROR: init(): unable to retrieve etree from TFile. Exiting");
        return kFALSE;
    }
    
    // associate the tree and the event
    if ( !ev )
    {
        ev = new EEvent();
        ev->SetBranches(tree);
        
        //Printf("# tree and event associated");
    }
    
    tree->GetEntry(i);
    
    gDetector = ev->GetDetector();
    
    //Printf("# Event %d loaded", i);
    
    return kTRUE;
}

//______________________________________________________________________________
void process(int num)
{
    TH1F *histo[200];
    
    //Get the histo distribution of GTU time arrival of photon for each event
    
    char canvas_histo_name[50];
    sprintf(canvas_histo_name, "c_histo_%d", num);
    
    TCanvas *c1 = new TCanvas(canvas_histo_name, "GTU Distribution");
    c1->Divide(4,4);
    
    for (int i=0; i<nb_entries; i++)
    {
        loadevent(i);
        nb_photons = gDetector->GetNumPhotons();
        //Printf("# Number of photon for event %d => %d", i, nb_photons);
        
        char histo_name[50];
        sprintf(histo_name, "histo_%d_%d", num, i);
        histo[i] = new TH1F(histo_name, "histo for one event", 128, 0, 128);
        histo[i]->Reset();
        
        
        for (int j=0; j<nb_photons; j++)
        {
            macrocell_status = gDetector->GetPhoton(j)->GetMacroCell();
            photon_type = gDetector->GetPhoton(j)->GetType();
            
            if (macrocell_status == 1 & photon_type != 2) //on ne sélectionne que les photons étant détectés par l'électronique & autre que les Cerenkov
            {
                Float_t gtu = gDetector->GetPhoton(j)->GetGtu();
                histo[i]->Fill(gtu);
            }
            
        }
        
        c1->cd(i+1);
        histo[i]->GetXaxis()->SetTitle("GTU");
        histo[i]->Draw();
        
    }
    
    //Loop on the histo array to produce a TProfile
    
    char hprof_name[50];
    sprintf(hprof_name, "hprof_%d", num);
    TProfile *hprof;
    
    hprof = new TProfile(hprof_name, "Average number of photons per GTU", 128, 0, 128);
    hprof->Reset();
    
    for (int k=0; k<nb_entries; k++)
    {
        for (int l=0; l<128; l++)
        {
            hprof->Fill(l, histo[k]->GetBinContent(l));
        }
    }
    
    char canvas_hprof_name[50];
    sprintf(canvas_hprof_name, "c_hprof_%d", num);
    
    TCanvas *c2 = new TCanvas(canvas_hprof_name, "TProfile distribution of the GTU arrival time of photons");
    c2->cd();
    hprof->Draw();
    
    //Save the TProfile one a .root file
    
    TFile file("/Users/moretto/Documents/thesis/ESAF/Code/esaf_lal/esaf/packages/analysis/average_time_distri_60ns.root", "new");
    hprof->Write();
    
}