#include "TTree.h"
#include "TFile.h"
#include "Math/LorentzVector.h"
#include "Math/PtEtaPhiM4D.h"
#include "TInterpreter.h"
#include <vector>

void test() {
  typedef ROOT::Math::LorentzVector<ROOT::Math::PtEtaPhiM4D<Double32_t> > LV;

  gInterpreter->GenerateDictionary("vector<ROOT::Math::LorentzVector<ROOT::Math::PtEtaPhiM4D<double> > >","vector;Math/LorentzVector.h");
  gInterpreter->GenerateDictionary("vector<ROOT::Math::LorentzVector<ROOT::Math::PtEtaPhiM4D<Double32_t> > >","vector;Math/LorentzVector.h");

  LV lv,lv32;
  LV* lvp = &lv;
  LV* lv32p = &lv32;
  std::vector<LV> lvv,lvv32;
  std::vector<LV>* lvvp = &lvv;
  std::vector<LV>* lvv32p = &lvv32;
  
  TTree *tree = new TTree("tree","tree");
  tree->Branch("lv","ROOT::Math::LorentzVector<ROOT::Math::PtEtaPhiM4D<double> >", &lvp);
  tree->Branch("lv32","ROOT::Math::LorentzVector<ROOT::Math::PtEtaPhiM4D<Double32_t> >", &lv32p);

  tree->Branch("lvv","vector<ROOT::Math::LorentzVector<ROOT::Math::PtEtaPhiM4D<double> > >", &lvvp);
  tree->Branch("lvv32","vector<ROOT::Math::LorentzVector<ROOT::Math::PtEtaPhiM4D<Double32_t> > >", &lvv32p);

  tree->Fill();
  TFile* f = TFile::Open("test.root","RECREATE");
  tree->Write();
  f->Close();
  
  return;
}