MyHtautauAnalysis/run/cut_eff.cxx

#include "iomanip"

int cut_flow(TString file){
    
    TFile *_file0 = TFile::Open(file);
    
    vector<TString> v_cut;
    
    _file0->cd("cut_info");
    TTree *cut_order = (TTree*)gDirectory->Get("cut_order");
    
     //Declaration of leaves types
    Char_t          cut_01[24];
    Char_t          cut_02[24];
    Char_t          cut_03[24];
    Char_t          cut_04[24];
    Char_t          cut_05[24];
    Char_t          cut_06[24];
    Char_t          cut_07[24];
    Char_t          cut_08[18];
    Char_t          cut_09[18];
    Char_t          cut_10[18];
    Char_t          cut_11[18];
    Char_t          cut_12[18];
    Char_t          cut_13[17];
    Char_t          cut_14[17];
    Char_t          cut_15[17];
    Char_t          cut_16[17];
    Char_t          cut_17[17];
    Char_t          cut_18[12];
    Char_t          cut_19[11];
    Int_t           nb_cuts;

    // Set branch addresses.
    cut_order->SetBranchAddress("cut_01",cut_01);
    cut_order->SetBranchAddress("cut_02",cut_02);
    cut_order->SetBranchAddress("cut_03",cut_03);
    cut_order->SetBranchAddress("cut_04",cut_04);
    cut_order->SetBranchAddress("cut_05",cut_05);
    cut_order->SetBranchAddress("cut_06",cut_06);
    cut_order->SetBranchAddress("cut_07",cut_07);
    cut_order->SetBranchAddress("cut_08",cut_08);
    cut_order->SetBranchAddress("cut_09",cut_09);
    cut_order->SetBranchAddress("cut_10",cut_10);
    cut_order->SetBranchAddress("cut_11",cut_11);
    cut_order->SetBranchAddress("cut_12",cut_12);
    cut_order->SetBranchAddress("cut_13",cut_13);
    cut_order->SetBranchAddress("cut_14",cut_14);
    cut_order->SetBranchAddress("cut_15",cut_15);
    cut_order->SetBranchAddress("cut_16",cut_16);
    cut_order->SetBranchAddress("cut_17",cut_17);
    cut_order->SetBranchAddress("cut_18",cut_18);
    cut_order->SetBranchAddress("cut_19",cut_19);
    cut_order->SetBranchAddress("nb_cuts",&nb_cuts);
    cut_order->GetEntry(0);
    v_cut.push_back(TString(cut_01));
    v_cut.push_back(TString(cut_02));
    v_cut.push_back(TString(cut_03));
    v_cut.push_back(TString(cut_04));
    v_cut.push_back(TString(cut_05));
    v_cut.push_back(TString(cut_06));
    v_cut.push_back(TString(cut_07));
    v_cut.push_back(TString(cut_08));
    v_cut.push_back(TString(cut_09));
    v_cut.push_back(TString(cut_10));
    v_cut.push_back(TString(cut_11));
    v_cut.push_back(TString(cut_12));
    v_cut.push_back(TString(cut_13));
    v_cut.push_back(TString(cut_14));
    v_cut.push_back(TString(cut_15));
    v_cut.push_back(TString(cut_16));
    v_cut.push_back(TString(cut_17));
    v_cut.push_back(TString(cut_18));
    v_cut.push_back(TString(cut_19));
    v_cut.push_back("");v_cut.push_back("");v_cut.push_back("");
    v_cut.push_back("");v_cut.push_back("");v_cut.push_back("");
    v_cut.push_back("");v_cut.push_back("");v_cut.push_back("");
    v_cut.push_back("");v_cut.push_back("");v_cut.push_back("");
    
    _file0->cd();
    analysis->cd();
//     for(int k=3; k<24; k++){
//         //cout<<"Bin "<<k<<": "<<endl;
//         cout<<"Cut "<<k-3<<" "<<v_cut.at(k-3)<<":  \t"<<cut_evolution->GetBinContent(k)<<endl;
//      //cout<<"                Cut "<<k-2<<": "<<cut_evolution->GetBinContent(k+1)*100/cut_evolution->GetBinContent(k)<<"%"<<endl;
//      
//     }
//     cout<<"Overall eff: "<<cut_evolution->GetBinContent(18)*100/cut_evolution->GetBinContent(3)<<"%"<<endl;

    double nb_init(0);
    double nb_before(0);
    double nb_current(0);
    
    double csec;
    cout<<"Enter cross section: ";
    cin>>csec;

    cout<<endl<<endl<<"====== Cut evolution ======"<<endl;
    cout<<"cut                                        eff   cumeff    events"<<endl;
    cout.width(38);
    cout<<"All events: ";
    cout.width(27);
    int tmp_bin;
    tmp_bin = cut_evolution->FindBin(0.);
    cout<</*right<<*/cut_evolution->GetBinContent(tmp_bin)<<endl;
    nb_init=cut_evolution->GetBinContent(tmp_bin);
    nb_before=nb_init;
    double rel_eff, cum_eff;
    for(int m=1; m<=19; m++){
        tmp_bin = cut_evolution->FindBin(m);
        nb_current=cut_evolution->GetBinContent(tmp_bin);
        if(nb_before==0) rel_eff=0;
        else rel_eff=/*(int)*/double((int)(10000.*nb_current/nb_before))/100.;
        if(nb_init==0) cum_eff=0;
        else cum_eff=/*(int)*/double((int)(10000.*nb_current/nb_init))/100.;

    //  cout<<"rel eff: "<<nb_current/nb_before<<endl;
    //  cout<<"cumulated eff: "<<nb_current/nb_init<<endl;
        cout<<"cut ";
        cout.width(2);
        cout<</*right<<*/m;
        cout<<": ";
        cout.width(25);
        //cout<<left<<cuts[m];//
        cout<</*left<<*/v_cut.at(m-1);
        cout.width(12);cout.precision(4);
        cout<</*right<<*/rel_eff<<"%";
        cout.width(8);
        cout<</*right<<*/cum_eff<<"%";
        cout.width(10);
        cout<</*right<<*/(int)(nb_current);
        cout.width(10);
        cout<</*right<<*/(csec*nb_current/nb_init)<<endl;
        nb_before=nb_current;
    }


    return 0;
}

---