#include "EGPTrack.hh"
#include "CLHEP/Units/SystemOfUnits.h"

void analyzeStage1()
{
  std::string fileName = "CosmicsStage1_EGP_0.root";

  TFile *file = new TFile(fileName.c_str(),"READ");
  if(file==NULL) return;  //File can't be opened
  TTree *tree = (TTree*)file->FindObjectAny("CosmicsStage1");
  if(tree==NULL) return;  //File doesn't contain the tree

  int nEntries = tree->GetEntries();  //number of events in the tree

  //all variable which we want to read
  //variables that are not needed can be removed
  EGPTrack               *primaryMuon     = new EGPTrack;
  std::map<int,EGPTrack> *containerTracks = new std::map<int,EGPTrack>;
//  int                     productionPlane;
//  bool                    muonPassedThroughPyramid;

  //set the branch to the address of the variables above
  //variables that are not needed can be removed
  tree->SetBranchAddress("primaryMuon",&primaryMuon);
  tree->SetBranchAddress("stage2Tracks",&containerTracks);
//  tree->SetBranchAddress("productionPlane",&productionPlane);
//  tree->SetBranchAddress("muonPassedThroughPyramid",&muonPassedThroughPyramid);
      
  //loop over all tree entries (=events)
  for(int i=0; i<nEntries; i++)
  {
    tree->GetEntry(i);  //read the i-th entry(=i-th event). the variables above are filled at this point

    const TString &endVolume = primaryMuon->_endVolume;
    if(!endVolume.EqualTo("Container")) continue; //skip all muons which don't hit the container

    //get some information about the position/direction of the primary muon when it hits the container
    const CLHEP::Hep3Vector &endPos = primaryMuon->_endPos;
    const CLHEP::Hep3Vector &endDir = primaryMuon->_endDir;
    float endEnergy = primaryMuon->_endEnergy;

    //print out some information for the first 10 events
    if(i<10) 
    {
      //primary muon information
      std::cout<<"Event #"<<i<<std::endl;
      std::cout<<"Primary muon:"<<std::endl;
      std::cout<<"Pos "<<endPos/CLHEP::m<<" m    Dir "<<endDir<<"    E "<<endEnergy/CLHEP::GeV<<" Gev"<<std::endl;

      //information about the tracks hitting the container
      std::cout<<"Container tracks:"<<std::endl;
      for(auto track=containerTracks->begin(); track!=containerTracks->end(); ++track)
      {
        std::cout<<"TrackID "<<track->second._trackID<<"  PDG code "<<track->second._particleID<<"   E "<<track->second._endEnergy/CLHEP::GeV<<" GeV"<<std::endl;
      }

      std::cout<<std::endl;
    }
  }

  delete file;
  delete primaryMuon;
  delete containerTracks;
}