#include "TFile.h"
#include "TTree.h"
#include <iostream>

#define CRASH // Toggle this to avoid crash

void shortjpsi()
{
  TFile* f               = new TFile("FillGenNtuple.root");
  TTree* NTupleTree     = (TTree*)f->Get("CollectionTree");

  // Declaring the ntuple variables:
  int              muon_nParticle; // Number of muons in an entry
  vector<double>*  l0;
  vector<double>*  l1;
  vector<double>*  l2;
  vector<double>*  l3;
  vector<int>*     muonCharge;
  vector<double>*  jE;
  vector<double>*  jPx;
  vector<double>*  jPy;
  vector<double>*  jPz;
  vector<double>*  jPt;
  // Linking the local variables to the tree branches
  NTupleTree->SetBranchAddress("nMuons", &muon_nParticle);
  NTupleTree->SetBranchAddress("muonPx", &l1);
  NTupleTree->SetBranchAddress("muonPy", &l2);
  NTupleTree->SetBranchAddress("muonPz", &l3);
  NTupleTree->SetBranchAddress("muonE",  &l0);
  NTupleTree->SetBranchAddress("muonQ",  &muonCharge);
  NTupleTree->SetBranchAddress("jpsiE",  &jE);
  NTupleTree->SetBranchAddress("jpsiPx", &jPx);
  NTupleTree->SetBranchAddress("jpsiPy", &jPy);
#if defined CRASH
  NTupleTree->SetBranchAddress("jpsiPz", &jPz);
#endif
  NTupleTree->SetBranchAddress("jpsiPt", &jPt);
  
  int nEntries = 10;//NTuppleTree->GetEntries(); // Get the number of entries in this tree
  for (int i = 0; i < nEntries; i++)
    {
      NTupleTree->GetEntry(i); // Gets the next entry (filling the linked variables)
    }

  f->Close();
}