#include <iostream>
#include <vector>
#include <ROOT/RDataFrame.hxx>

// scalar fill
template <typename T>
void agg(std::vector<int> &a, const T &b) {
        std::cout << "f1" << std::endl;
	a.push_back(b);
}
// vector fill
template <typename T>
void agg(std::vector<int> &a, const std::vector<T> &b) {
  std::cout << "f2" << " "<<b.size()<<std::endl;
  a.insert(a.end(), b.begin(), b.end());

}

//this would work

template <typename T>
void aggscalar(std::vector<int> &a, const T &b) {
        std::cout << "f1" << std::endl;
	a.push_back(b);
}
// vector fill
template <typename T>
void aggvector(std::vector<int> &a, const T &b) {
  std::cout << "f2" << " "<<b.size()<<std::endl;
  a.insert(a.end(), b.begin(), b.end());
}


void print(std::vector <int> const &a) {
   std::cout << "The vector elements are : ";

   for(int i=0; i < a.size(); i++)
   std::cout << a.at(i) << ' ';

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

std::vector<int> add_all(std::vector<int> a, std::vector<int> b)
{
  std::vector<int> c =a;
  c.insert(c.end(), b.begin(), b.end());
  return c;
}

int main(int argc, char **argv)
                    
{
  std::vector<int> all1;
  ROOT::RDataFrame df("tree1","pokus1.root");
  //this doesn't work
  //auto total1 = df.Aggregate(agg<int>,add_all,"x",all1);
  //auto total2 = df.Aggregate(agg<std::vector<int>>,add_all,"y",all1);

  auto total3 = df.Aggregate(aggscalar<int>,add_all,"x",all1);
  auto total4 = df.Aggregate(aggvector<std::vector<int>>,add_all,"vecy",all1);
  //print(*total1);
  //print(*total2);
  print(*total3);
  print(*total4);
  
  return 0;
};