#include <iostream>
#include <ROOT/RDataFrame.hxx>
#include <TH1.h>
#include <vector>
using namespace std;

using RNode = ROOT::RDF::RNode;

void fillTH(vector<double>TH_)
{
	TH_.clear();
	TH_.push_back(400);//#1
	TH_.push_back(400);//#2
	TH_.push_back(400);//#3
	TH_.push_back(400);//#4
	TH_.push_back(400);//#5
	TH_.push_back(400);//#6
	TH_.push_back(600);//#7
	TH_.push_back(600);//#8
	TH_.push_back(400);//#9
	TH_.push_back(400);//#10
	TH_.push_back(400);//#1:
	TH_.push_back(400);//#12
	TH_.push_back(400);//#13
	TH_.push_back(400);//#14
	TH_.push_back(400);//#15
}

RNode Filter_hit(RNode d,int N)
{
	auto HitCut = [](int x) {return x==1 || x==3;};
	stringstream hit;
	hit << "hit"<<N;
	auto df=d.Filter(HitCut,{hit.str().c_str()});
	hit.str("");
	return df;
}

RNode Filter_Ch(RNode d,int N)
{
	vector<double>TH;
	fillTH(TH);
	auto ChCut = [=](int x) {return x>TH[N-1];};
	stringstream depositEnergy_temp;
	depositEnergy_temp << "depositEnergyt"<<N;
	auto df=d.Filter(ChCut,{depositEnergy_temp.str().c_str()});
	depositEnergy_temp.str("");
	return df;
}

void coincidence_2F_test()
{
	ROOT::EnableImplicitMT(15);
	TChain chaint("treet");

	stringstream path;
	path<<"./20221011_4_fixed.root";
	chaint.Add(path.str().c_str());
	ROOT::RDataFrame dt(chaint);

	auto dt_hitf=Filter_Ch(dt,7);
//	auto dt_hitf=Filter_hit(dt,7);

	auto dt_hitf_c=dt_hitf.Count();
	cout<<dt_hitf_c.GetValue()<<endl;
}

int main() {
   coincidence_2F_test();
}