#include <vector>
#include <iostream>
#include <fstream>
#include "TTree.h"
#include "TFile.h"
#include "TChain.h"
#include "Data.C"
#include "TH1.h"
#include "TLorentzVector.h"

void MakeFun()
{

    TH1D * h_deta_i_mdt = new TH1D("h_deta_i_mdt", " #Delta(#eta) Middle Distance == 0 ", 80, -.2, .2);
    TH1D * h_deta_i_tgc = new TH1D("h_deta_i_tgc", " #Delta(#eta) Middle Distance == 0 ", 80, -.2, .2);

    TTree *Tree1; 

    TChain * chain = new TChain("MUGIRL",""); 
    
    chain->Add("ntuple_test.root/MUGIRL");
    //Bs-->mu(6)mu(3)
    /*
    chain->Add("ntuple_Bs_m6m3_5_1.root/MUGIRL");
    chain->Add("ntuple_Bs_m6m3_5_2.root/MUGIRL");
    chain->Add("ntuple_Bs_m6m3_5_3.root/MUGIRL");
    chain->Add("ntuple_Bs_m6m3_5_4.root/MUGIRL");
    */
    //bb_bar-->mu(6)X
    
    /*chain->Add("ntuple_bbmu6X_104.root/MUGIRL");
    chain->Add("ntuple_bbmu6X_110_112.root/MUGIRL");
    chain->Add("ntuple_bbmu6X_111.root/MUGIRL");
    chain->Add("ntuple_bbmu6X_113_118_119.root/MUGIRL");
    chain->Add("ntuple_bbmu6X_115.root/MUGIRL");
    chain->Add("ntuple_bbmu6X_120.root/MUGIRL");
    chain->Add("ntuple_bbmu6X_122_128_129.root/MUGIRL");
    chain->Add("ntuple_bbmu6X_131.root/MUGIRL");
    chain->Add("ntuple_bbmu6X_132_134.root/MUGIRL");
    chain->Add("ntuple_bbmu6X_141_146.root/MUGIRL");
    chain->Add("ntuple_bbmu6X_147_149.root/MUGIRL");
    chain->Add("ntuple_bbmu6X_153_156.root/MUGIRL");
    chain->Add("ntuple_bbmu6X_162_168.root/MUGIRL");
    chain->Add("ntuple_bbmu6X_171.root/MUGIRL");
    chain->Add("ntuple_bbmu6X_179_180.root/MUGIRL");
    chain->Add("ntuple_bbmu6X_186_188.root/MUGIRL");
    chain->Add("ntuple_bbmu6X_193_198.root/MUGIRL");
    chain->Add("ntuple_bbmu6X_199.root/MUGIRL");*/
    
    //Higs-->4l
    //chain->Add("ntuple_H4l_5_1.root/MUGIRL");
    //Single muons
    /*
    chain->Add("ntuple_mu_minus6GeV_1_1000.root/MUGIRL");
    chain->Add("ntuple_mu_minus6GeV_2_1000.root/MUGIRL");
    */
    /*chain->Add("ntuple_mu_minus20GeV_1_1000_chi10.root/MUGIRL");
    chain->Add("ntuple_mu_minus20GeV_2_1000_chi10.root/MUGIRL");*/
    
    Tree1 = chain;
    Data * t = new Data(Tree1);
    if (t->fChain == 0) 
        return;
    Int_t nentries = Int_t(t->fChain->GetEntries());
    
    cout<<" Natasha's test: xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx"<<endl;
    cout<<" Reading data ..."<<nentries<<endl;
    
    Int_t nTen = nentries/10;
    Int_t nbytes = 0, nb = 0;
    for (Int_t jentry=0; jentry<nentries;jentry++) 
    { 
	    Int_t ientry = t->LoadTree(jentry);
	    if (ientry < 0) break;
	    nb = t->fChain->GetEntry(jentry);   nbytes += nb;
	    
        if (jentry%nTen==0) cout<<10*(jentry/nTen)<<"%-"<<flush;
	    if (jentry==nentries-1) cout<<endl;
        
        
        for (Int_t iCand = 0; iCand < t->NumCandidates; iCand++)
        {
            //remove muons in MS entrance
            int add_muon_ms	= 0;
			for(Int_t iTruth = 0; iTruth < t->NumTruthMuon;	iTruth++)
			{
				if(t->TruthMuType[iTruth] >	0)// MS	entrance
				{
					for(Int_t iSeg = 0;	iSeg < t->NumSegments[iCand]; iSeg++)
					{
						if (t->SegType[iCand][iSeg]	== 0)
						{
							if (fabs(t->TruthMuEta[iTruth] - t->SegPosEta[iCand][iSeg])<0.04)
								add_muon_ms++;
						}
					}
				}
			}
            //if(add_muon_ms > 0 && t->TruthPt[iCand] == -100)
            //{
            //if(add_muon_ms == 0 && t->TruthPt[iCand] == -100) // fakes
            //{
            if(t->TruthPt[iCand] != -100 )// muon
            {
                for(Int_t iIsec = 0; iIsec < t->NumIsections[iCand]; iIsec++)
                {
                    if(t->IsectRegion[iCand][iIsec]==1)// only end-cap
                    {
                        if(t->IsectDistance[iCand][iIsec] == 2)// only middle
                        {
                            if(t->IsectType[iCand][iIsec] == 1)// ID extrapolated
                            {
                                for(Int_t j1 = 0; j1 < t->NumIsections[iCand]; j1++)
                                {
                                    if(t->IsectRegion[iCand][j1]==1)// only end-cap
                                    {
                                        if(t->IsectDistance[iCand][j1] == 2)// only middle
                                        {
                                            if(t->IsectType[iCand][j1] == 3)// evaluated
                                            {
                                                if(t->IsectChamber[iCand][iIsec] == t->IsectChamber[iCand][j1])
                                                {
                                                    if(t->IsectOrigTech[iCand][j1] == 2)//TGC intesections
                                                    {
                                                        if(t->IsectDirEta[iCand][iIsec] - t->IsectDirEta[iCand][j1] != 0)
                                                            h_deta_i_tgc->Fill(t->IsectDirEta[iCand][iIsec] - t->IsectDirEta[iCand][j1]);
                                                    }
                                                    if(t->IsectOrigTech[iCand][j1] == 0)//MDT intersections
                                                    {
                                                        if(t->IsectDirEta[iCand][iIsec] - t->IsectDirEta[iCand][j1] != 0)
                                                            h_deta_i_mdt->Fill(t->IsectDirEta[iCand][iIsec] - t->IsectDirEta[iCand][j1]);
                                                    }
                                                }
                                            }
                                        }
                                    }
                                }
                            }
                        }
                    }
                }
            }
        }     
    }

    TCanvas  *c1=new TCanvas("c1","window",600,450);   
	gStyle->SetOptStat(111111);
	gStyle->SetOptFit(0);
	c1->SetFillColor(0);
	c1->SetBorderMode(0);
	c1->SetBorderSize(1);
	c1->SetFrameFillColor(1);
	c1->SetFrameFillStyle(0);

    c1->Divide(1,2);
    c1->cd(1);
    h_deta_i_mdt->Draw();
    h_deta_i_mdt->SetMarkerStyle(8);
    c1->cd(2);
    h_deta_i_tgc->Draw();
    h_deta_i_tgc->SetMarkerStyle(8);
}