#include "PublicScript.h"
#include "TMath.h"
#include "TFile.h"
#include "TTree.h"
#include "TCanvas.h"
#include "TROOT.h"
#include "TH1F.h"
#include "TH2F.h"
#include <iostream>
#include "TString.h"
#include "TVector3.h"
using namespace std;



void tree1r(TChain *chainB,TChain *chainV,TString output_filename)
{

  //----------------------------------------------------------  
  //This initializes both Trees, meaning it sets all the branch addresses, etc. This Initialize function is defined in PublicScript.h
  //----------------------------------------------------------
  InitializeB(chainB);
  InitializeV(chainV);

  chainB->SetBranchStatus("*",0);
  chainB->SetBranchStatus("mus_pt",1);

  chainV->SetBranchStatus("*",0);
  chainV->SetBranchStatus("HLT_Mu3",1);


  //Initialize(t1, t2);

  TFile *test = new TFile(output_filename.Data(),"RECREATE");

  test->cd();
  //----------------------------------------------------------
  //Histogram Declarations
  //----------------------------------------------------------
   TCanvas *c0 = new TCanvas("c0","",800,800);


   TH1F *h_muon_pt = new TH1F("h_muon_pt","Muon Pt",200,0,200);

  //----------------------------------------------------------     
  //Number of events to loop over
  //----------------------------------------------------------
  Int_t nentries = (Int_t)chainB->GetEntries();
  //  Int_t nentries = (Int_t)t1->GetEntries();
  cout<<"The number of entries is: "<<nentries<<endl;

  //----------------------------------------------------------
  //Main Event loop
  //----------------------------------------------------------
    for(int ia = 0; ia<nentries;ia++){

      chainB->GetEntry(ia);
      chainV->GetEntry(ia);
      
      if(HLT_Mu3>0){

	for(int it = 0;it<mus_pt->size();it++)
	  {
		h_muon_pt->Fill(mus_pt->at(it));


	  }
      }
      
      
    }//end over loop over all the events
      
      
    
    test->cd();
    test->Write();
    
}