#define A1_cxx
#include "A1.h"
#include <TH2.h>
#include <TStyle.h>
#include <TCanvas.h>
#include <iostream>

void A1::Loop()
{
//   In a ROOT session, you can do:
//      Root > .L A1.C
//      Root > A1 t
//      Root > t.GetEntry(12); // Fill t data members with entry number 12
//      Root > t.Show();       // Show values of entry 12
//      Root > t.Show(16);     // Read and show values of entry 16
//      Root > t.Loop();       // Loop on all entries
//

//     This is the loop skeleton where:
//    jentry is the global entry number in the chain
//    ientry is the entry number in the current Tree
//  Note that the argument to GetEntry must be:
//    jentry for TChain::GetEntry
//    ientry for TTree::GetEntry and TBranch::GetEntry
//
//       To read only selected branches, Insert statements like:
// METHOD1:
//    fChain->SetBranchStatus("*",0);  // disable all branches
//    fChain->SetBranchStatus("branchname",1);  // activate branchname
// METHOD2: replace line
//    fChain->GetEntry(jentry);       //read all branches
//by  b_branchname->GetEntry(ientry); //read only this branch
  c1 = new TCanvas( 3 );
  c1->Divide( 2, 2 );
  hTOF1 = new TH1F( "hTOF1", "First ToF;Time of Flight [ns];Counts", 100, 0, 100 );
  hTOF2 = new TH1F( "hTOF2", "Second ToF;Time of Flight [ns];Counts", 100, 0, 100 );
  hTOFDiff = new TH1F( "hTOFDiff", "ToF2 - ToF1;Time [ns];Counts", 100, 0, 100 );
  hTrigger = new TH1F( "hTrigger", "Trigger;Time 0.1 ns;Counts;", 100000, 10000, 100000 );
  if (fChain == 0) return;

   Long64_t nentries = fChain->GetEntriesFast();

   Long64_t nbytes = 0, nb = 0;
   for (Long64_t jentry=0; jentry<nentries;jentry++) {
      Long64_t ientry = LoadTree(jentry);
      if (ientry < 0) break;
      nb = fChain->GetEntry(jentry);   
      nbytes += nb;
      // if (Cut(ientry) < 0) continue;
      //Module2_NUMBER_OF_CHANNELS;
      vector<unsigned int> * HITS_PER_CHANNEL = HitsPerChannel[ 1 ];
      vector<unsigned int> * CH32 = _Data[ 1 ][ 32 ];//Module2_LE_CH32;
      vector<unsigned int> * CH96 = _Data[ 1 ][ 96 ];//Module2_LE_CH96;
      vector<unsigned int> * CH64 = _Data[ 1 ][ 64 ];//Module2_LE_CH64;
      //std::cout << "HITS_PER_CHANNEL 32: " << HITS_PER_CHANNEL->at( 32 ) << std::endl;
      if( HITS_PER_CHANNEL->at( 32 ) > 0 ){
        long trig = CH32->at( 0 );
        hTrigger->Fill( trig );
        long tof1( 0 ), tof2( 0 );
        //std::cout << "trigger : " << trig << std::endl;
        if( HITS_PER_CHANNEL->at( 96 ) > 0 ){
          tof1 = CH96->at( 0 ) - trig;
          hTOF1->Fill( tof1/10 );
//          std::cout << "TOF1 = " << tof1 / 10<< "ns " << std::endl;
        }
        
        if( HITS_PER_CHANNEL->at( 64 ) > 0 ){
          tof2 = CH64->at( 0 ) - trig;
          hTOF2->Fill( tof2/10 );
//          std::cout << "TOF2 = " << tof2 / 10 << " ns "<< std::endl;
        }
        if( (tof1 != 0 ) && (tof2 != 0 ) ){
          //std::cout << "TOF2 -  TOF1 = " << tof2 - tof1 << std::endl;
          hTOFDiff->Fill( ( tof2 - tof1 )/10 );
        }
      } 
      
   }
   c1->cd( 1 );
   hTOF1->Draw();
   c1->cd( 2 );
   hTOF2->Draw();
   c1->cd( 3 );
   hTOFDiff->Draw();
   c1->cd( 4 );
   hTrigger->Draw();
}

void A1::PlotChannel( unsigned aModule, unsigned aChannel ){
  if (fChain == 0) return;
  if( c1 ) delete c1;
  c1 = new TCanvas( 3 );
  stringstream stmp;
  stmp << "Module " << aModule + 1 << ", Channel #" << aChannel << ";Time [0.1 ns]; Counts";


  TH1I * hChannel = new TH1I( "hChannel", stmp.str().c_str(), 100000, 10000, 100000 );
  Long64_t nentries = fChain->GetEntriesFast();
  Long64_t nbytes = 0, nb = 0;
  
  for (Long64_t jentry=0; jentry<nentries;jentry++) {
    Long64_t ientry = LoadTree(jentry);
    if (ientry < 0) break;
    nb = fChain->GetEntry(jentry);   
    nbytes += nb;
      // if (Cut(ientry) < 0) continue;
    vector<unsigned int> * HITS_PER_CHANNEL = HitsPerChannel[ aModule ];
    vector<unsigned int> * CH = _Data[ aModule ][ aChannel ];//Module2_LE_CH32;
      //std::cout << "HITS_PER_CHANNEL 32: " << HITS_PER_CHANNEL->at( 32 ) << std::endl;
    if( HITS_PER_CHANNEL->at( aChannel ) > 0 ){
      long trig = CH->at( 0 );
      hChannel->Fill( trig );
    } 
      
  }

  hChannel->Draw();

}