#include <TChain.h>
#include <TEntryList.h>

#include <RtypesCore.h>
#include <ROOT/RDataFrame.hxx>

#include <TTreeReader.h>

#include <iostream>
#include <limits>
#include <string>

auto LIMIT = std::numeric_limits<int>::max();
Long64_t OVERLIMIT = 2147483648; // LIMIT + 1
ULong64_t NENTRIES = LIMIT / 100;

void write_tree()
{
    ROOT::RDataFrame df{NENTRIES};
    df.Define("x", "rdfentry_").Snapshot<ULong64_t>("tree", "file.root", {"x"});
}

void repro()
{
    TChain chain{"tree"};
    TEntryList elists{};
    for (int i = 0; i < 101; i++){
        chain.Add("file.root?#tree", NENTRIES);
        TEntryList sublist{};
        sublist.SetTreeName("tree");
        sublist.SetFileName("file.root");
        sublist.EnterRange(0, NENTRIES);
        elists.AddSubList(&sublist);
    }

    chain.SetCacheEntryRange(0, OVERLIMIT);
    chain.SetEntryList(&elists, "sync"); // This sets the tree index in the sublist

    std::cout << "std::numeric_limits<int>::max(): " << LIMIT << "\n";
    std::cout << "Total entries in chain: " << chain.GetEntries() << "\n";
    std::cout << "Total entries in entrylist: " << elists.GetN() << "\n";

    int treenum_1 = -1;
    std::cout << "Giving input entry " << LIMIT << "\n";
    auto eindex_1 = elists.GetEntryAndTree(LIMIT, treenum_1);

    std::cout << "Got tree index " << treenum_1 << " and entry index " << eindex_1 << "\n";

    int treenum_2 = -1;
    std::cout << "Giving input entry " << OVERLIMIT << "\n";
    auto eindex_2 = elists.GetEntryAndTree(OVERLIMIT, treenum_2);

    std::cout << "Got tree index " << treenum_2 << " and entry index " << eindex_2 << "\n";

    TTreeReader r{&chain, chain.GetEntryList()};

    r.SetEntry(OVERLIMIT);
    std::string status = (r.GetEntryStatus() == TTreeReader::EEntryStatus::kEntryBeyondEnd) ? "kEntryBeyondEnd" : "not kEntryBeyondEnd";
    std::cout << "What is the entry status? " << status << "\n";
}

int main()
{
    write_tree();
    repro();
}