///
/// Repro the ResetBranch bug
///

#include <TFile.h>
#include <TTree.h>
#include <TChain.h>

#include <iostream>
#include <string>

using namespace std;

/// The value we will store in the ttree
static int gValue = 0;

///
/// Create a single file with a tree with an integer in it
///
void create_file (const string &fname)
{
	cout << "Making ntuple file " << fname << endl;
	auto f = new TFile (fname.c_str(), "RECREATE");

	auto t = new TTree ("ntup", "Ntuple of one");
	t->Branch("val", &gValue);

	for (int i = 0; i < 100; i++)
	{
		gValue = i;
		t->Fill();
	}

	f->Write();
	f->Close();
}

void resetbranchtest()
{
	/// Create the two files
	create_file ("n1.root");
	create_file ("n2.root");

	/// Create the chain that we will be reading in
	auto c = new TChain("ntup");
	c->AddFile("n1.root");
	c->AddFile("n2.root");

	TBranch *b;
	c->SetBranchAddress("val", &gValue, &b);

	int ent = c->GetEntries();
	cout << "Total Entries " << ent << endl;

	/// Run through most of the index
	int index = 0;
	for (index = 0; index < ent - 50; index++) {
		c->GetEntry(index);
	}

	/// Now, reset the address and then re-go
	// b->ResetAddress(); - doesn't seem to do anything
	//c->ResetBranchAddress(b); // This doesn't either
	c->ResetBranchAddresses(); // THis actuall works!!
	for (index = 0; index < ent - 50; index++) {
		c->GetEntry(index);
		cout << gValue << endl;
	}

}