time_based_studies.C

Go to the documentation of this file.

#include <iostream>

#include<TCanvas.h>
#include<TApplication.h>
#include<TROOT.h>
#include<TChain.h>
#include<TStyle.h>
#include<TGaxis.h>
#include<TColor.h>
#include<TClonesArray.h>
#include<TH1.h>
#include<TH2.h>

#include<PndLmdDim.h>
#include<PndMCTrack.h>
#include<PndSdsMCPoint.h>

using namespace std;

// change some stuff in the ROOT appearance
void Root_Appearance();

double last_percent(-1.);
// draw a progress bar only when the length changes significantly
void DrawProgressBar(int len, double percent);

// the application
// Assuming to find an mc file called
// ./Lumi_MC_0.root
int time_based_studies();

int main() {
        Root_Appearance();
        TApplication myapp("myapp", 0, 0);
        time_based_studies();
        myapp.Run();
        return 0;
}

int time_based_studies(){
        cout << " lmd time based studies " << endl;
        TCanvas* canvas = new TCanvas("canvas", "canvas", 600, 600);
        canvas->Divide(2,2);

        // load the MC File
        TChain tMC("pndsim");
        tMC.Add("./Lumi_MC_0.root");

        //--- assign MC info -----------------------------------------------------
        TClonesArray* true_tracks = new TClonesArray("PndMCTrack");
        tMC.SetBranchAddress("MCTrack", &true_tracks); //True Track to compare

        TClonesArray* true_points = new TClonesArray("PndSdsMCPoint");
        tMC.SetBranchAddress("LMDPoint", &true_points); //True Points to compare

        // get the lmd dim help class
        PndLmdDim& lmddim = PndLmdDim::Get_instance();

        // setup histograms
        /*
        TH1F* hist_eloss_total = new TH1F("hist_eloss_total", "energy deposit in 50#mum x 2cm x2cm sensors", 200, 0, 60);
        TH1F* hist_eloss = new TH1F("hist_eloss", "energy deposit scaled to 1#mum", 200, 0, 1200);
        hist_eloss_total->SetXTitle("E_{dep} [keV]");
        hist_eloss->SetXTitle("E_{dep} [eV]");

        TH2* plane3_hits = new TH2F("plane3_hits", "hits on plane 3", 200, -10, 10, 200, -10, 10); //lmddim.Get_histogram_Plane(0, 0);
        plane3_hits->SetXTitle("X [cm]");
        plane3_hits->SetYTitle("Y [cm]");
        plane3_hits->SetZTitle("# hits");
        TH2Poly* plane3_Edep = lmddim.Get_histogram_Plane(3, 0);
        plane3_Edep->SetNameTitle("plane3_Edep", "E dep on plane 3");
        plane3_Edep->SetZTitle("E_{dep} [J]");
        TH2Poly* plane3_rate = lmddim.Get_histogram_Plane(3, 0);
        plane3_rate->SetNameTitle("plane3_rate", "rate on plane 3");
        plane3_rate->SetZTitle("rate [kHz]");
        TH2Poly* plane3_rad = lmddim.Get_histogram_Plane(3, 0);
        plane3_rad->SetNameTitle("plane3_rad", "IEL dose on plane 3");
        plane3_rad->SetZTitle("IEL dose [Gy/a]");
*/
        // read events and analyze
        int nEvents = tMC.GetEntries();
        int nEvents_counted(0);
        int nhitstotal(0);

        cout << " reading " << nEvents << " Events " << endl;
        for (Int_t j = 0; j < nEvents ; j++) {
                nEvents_counted++;
                DrawProgressBar(50, (j + 1) / ((double) nEvents));
                tMC.GetEntry(j);

                int nHits = true_points->GetEntriesFast();
                int nSdsHits = 0;
                for (Int_t iHit = 0; iHit < nHits; iHit++) {
                        PndSdsMCPoint* mcpoint = (PndSdsMCPoint*) true_points->At(iHit);
                        //if (mcpoint->GetTrackID() < 0) continue;
                        // cout << mcpoint->GetEnergyLoss() << endl;
                        //hist_eloss_total->Fill(mcpoint->GetEnergyLoss()*1.e6); // in keV
                        //hist_eloss->Fill(mcpoint->GetEnergyLoss()*1.e9/50.); // in eV

                        int sensID = mcpoint->GetSensorID();
                        int ihalf, iplane, imodule, iside, idie, isensor;
                        lmddim.Get_sensor_by_id(sensID, ihalf, iplane, imodule, iside, idie, isensor);
                        TVector3 hit_pos = mcpoint->GetPosition();
                        TVector3 hit_pos_lmd = lmddim.Transform_global_to_lmd_local(hit_pos, false);

                        nhitstotal++;

                        cout << mcpoint->GetTime() << endl;
                        //cout << mcpoint->Get << endl << endl;

                        if (iplane == 3 && iside == 0){
                                //plane3_hits->Fill(hit_pos_lmd.X(), hit_pos_lmd.Y());
                                //plane3_rate->Fill(hit_pos_lmd.X(), hit_pos_lmd.Y());
                                //plane3_Edep->Fill(hit_pos_lmd.X(), hit_pos_lmd.Y(), mcpoint->GetEnergyLoss()*1.e9*1.602e-19); // in J
                                //plane3_rad->Fill(hit_pos_lmd.X(), hit_pos_lmd.Y(), mcpoint->GetEnergyLoss()*1.e9*1.602e-19); // in J
                        }
/*
                        if (mcpoint->GetTrackID() < 0) continue;
                        PndMCTrack *mctrk = (PndMCTrack*) true_tracks->At(
                                        mcpoint->GetTrackID());
                        if (mctrk->IsGeneratorCreated()) { // take only anti-protons from the generator
                                if (mcpoint) {
                                        TVector3 _mcpoint((mcpoint->GetPosition())); // position at the sensor entrance
                                        TVector3 _mctrack(mcpoint->GetPx(), mcpoint->GetPy(),
                                                        mcpoint->GetPz()); // momentum of the track at the entrance

                                        TVector3 momMC = mctrk->GetMomentum(); // momentum in the primary vertex
                                        TVector3 posMC = mctrk->GetStartVertex(); // position of the primary vertex
                                }
                        } */
                }
        }
        // cross section for N generated events at 1.5 GeV/c momentum was
        // sigm_coul = 36.683 mbarn
        // sigm_had_el = 25.3021 mbarn
        // sigm_inel = 62.71 mbarn according to Anastasias note
        // -> sig_total = 124.63 mbarn or 124.63 x 10-27 cm^-2
        double sig_total = 124.63e-27;//62e-27;//124.63e-27; // cm^2
        // the mean instantanious luminosity at high luminosity mode is 1.6 x 10^32
        // but we use here for reference 1 x 10^32 which can be scaled easily to any other number
        double lumi_ref = 1e32; // cm^-2 s^-1
        // therefore we simulated an interaction rate of
        double rate_total = sig_total * lumi_ref; // 1/s or Hz
        cout << "\n The total rate at 1.5 GeV/c and " << lumi_ref << " mean luminosity to normalize to is " << rate_total << " Hz" << endl;
        // The simulated time scale is then
        double tsim = nEvents_counted / rate_total; // s
        cout << nEvents_counted << " events correspond to a real time of " << tsim << " s " << endl;
        /*
        plane3_rate->Scale(1./tsim*1e-3); // in kHz
        //plane3_rate->SetMaximum(150);
        cout << " total rate of " << nhitstotal << " hits is " << nhitstotal/tsim*1e-6 << "MHz" << endl;
        plane3_rad->Scale(1./tsim*60.*60.*24.*365.*0.5/(2.33e-3*(2*2*50.e-4))); // 50% duty cycle over a year and normalized to the mass in kg of one sensor
        //canvas->cd(1);
        plane3_Edep->Draw("colz");
        //hist_eloss_total->Draw();
        //canvas->cd(2);
        plane3_hits->Draw("colz");
        canvas->Print("MC_hit_distr_1_5_plane_3_side_0.pdf");
        canvas->Print("MC_hit_distr_1_5_plane_3_side_0.root");
        hist_eloss->Draw();
        canvas->Print("MC_IEL_distrib_1_5.pdf");
        canvas->Print("MC_IEL_distrib_1_5.root");
        //canvas->cd(3);
        plane3_rate->Draw("colz");
        cout << " max rate is " << plane3_rate->GetBinContent(plane3_rate->GetMaximumBin()) << " kHz" << endl;
        canvas->Print("MC_hit_rate_1_5_plane_3_side_0.pdf");
        canvas->Print("MC_hit_rate_1_5_plane_3_side_0.root");
        //canvas->cd(4);
        plane3_rad->Draw("colz");
        cout << " max dose is " << plane3_rad->GetBinContent(plane3_rad->GetMaximumBin()) << " Gy" << endl;
        canvas->Print("MC_IEL_dose_1_5_plane_3_side_0.pdf");
        canvas->Print("MC_IEL_does_1_5_plane_3_side_0.root");
        */
  return 0;
}

void Root_Appearance(){
        TPad foo; // never remove this line :-)))
        if(1){
                gROOT->SetStyle("Plain");
                const Int_t NRGBs = 5;
                const Int_t NCont = 255;
                Double_t stops[NRGBs] = { 0.00, 0.34, 0.61, 0.84, 1.00 };
                Double_t red[NRGBs]   = { 0.00, 0.00, 0.87, 1.00, 0.51 };
                Double_t green[NRGBs] = { 0.00, 0.81, 1.00, 0.20, 0.00 };
                Double_t blue[NRGBs]  = { 0.51, 1.00, 0.12, 0.00, 0.00 };
                TColor::CreateGradientColorTable(NRGBs, stops, red, green, blue, NCont);
                gStyle->SetNumberContours(NCont);
                gStyle->SetTitleFont(10*13+2,"xyz");
                gStyle->SetTitleSize(0.06, "xyz");
                gStyle->SetTitleOffset(1,"xy");
                gStyle->SetTitleOffset(1.3,"z");
                gStyle->SetLabelFont(10*13+2,"xyz");
                gStyle->SetLabelSize(0.06,"xyz");
                gStyle->SetLabelOffset(0.009,"xyz");
                gStyle->SetPadBottomMargin(0.2);
                gStyle->SetPadTopMargin(0.15);
                gStyle->SetPadLeftMargin(0.15);
                gStyle->SetPadRightMargin(0.20);
                gStyle->SetOptTitle(0);
                gStyle->SetOptStat(0);
                gROOT->ForceStyle();
                gStyle->SetFrameFillColor(0);
                gStyle->SetFrameFillStyle(0);
                TGaxis::SetMaxDigits(3);
        }
}

void DrawProgressBar(int len, double percent) {
        if ((int) (last_percent * 100) == (int) (percent * 100))
                return;
        //cout << " drawing " << endl;
        cout << "\x1B[2K"; // Erase the entire current line.
        cout << "\x1B[0E"; // Move to the beginning of the current line.
        string progress;
        for (int i = 0; i < len; ++i) {
                if (i < static_cast<int> (len * percent)) {
                        progress += "=";
                } else {
                        progress += " ";
                }
        }
        cout << "[" << progress << "] " << (static_cast<int> (100 * percent))
                        << "%";
        flush( cout); // Required.
        last_percent = percent;
}