X-Git-Url: https://git.creatis.insa-lyon.fr/pubgit/?a=blobdiff_plain;f=tests_dav%2FclitkMergeRootFiles.cxx;h=957a8ed8eac06a48adb5796ef6ad608bd6e270e1;hb=6848b1c025818a83ca5ddd5194910b87587bfc80;hp=5b62d84e7608235360b0459f96a66ae9c23dbce2;hpb=2546511d8aaa6f88c0abe39c84f9bdbce83a04d4;p=clitk.git

diff --git a/tests_dav/clitkMergeRootFiles.cxx b/tests_dav/clitkMergeRootFiles.cxx
index 5b62d84..957a8ed 100644
--- a/tests_dav/clitkMergeRootFiles.cxx
+++ b/tests_dav/clitkMergeRootFiles.cxx
@@ -17,6 +17,7 @@
 #include <TKey.h>
 #include <TFileMerger.h>
 #include <TTree.h>
+#include <TChain.h>
 #include <TH1.h>
 #include <TH2.h>
 #include <iostream>
@@ -29,6 +30,43 @@ typedef std::list<std::string> Strings;
 typedef std::list<TFile*> Handles;
 typedef std::set<std::string> StringsSet;
 
+void mergeSingleTree(TChain* chain, TFile* output_handle, const std::string& tree_name)
+{
+	assert(chain->FindBranch("runID"));
+	assert(chain->FindBranch("eventID"));
+	assert(chain->FindBranch("time"));
+
+	int eventID = -1;
+	int runID = -1;
+	double time = -1;
+
+	chain->SetBranchAddress("eventID",&eventID);
+	chain->SetBranchAddress("runID",&runID);
+	chain->SetBranchAddress("time",&time);
+
+	output_handle->cd();
+	TTree* output_tree = chain->CloneTree(0);
+
+	const long int nentries = chain->GetEntries();
+	for (long int kk=0; kk<nentries; kk++)
+	{
+		chain->GetEntry(kk);
+		//cout << kk << "/" << chain->GetTreeNumber() << endl;
+		output_tree->Fill();
+	}
+
+	output_tree->Write();
+	delete output_tree;
+}
+
+void mergeGateTree(TChain* chain, TFile* output_handle, const std::string& tree_name)
+{
+}
+
+void mergeCoinTree(TChain* chain, TFile* output_handle, const std::string& tree_name)
+{
+}
+
 void mergePetRoot(const Strings& input_filenames, const std::string& output_filename)
 {
 	TFile* output_handle = TFile::Open(output_filename.c_str(),"RECREATE");
@@ -50,21 +88,40 @@ void mergePetRoot(const Strings& input_filenames, const std::string& output_file
 			const bool is_h1 = key->ReadObj()->InheritsFrom("TH1");
 			const bool is_h2 = key->ReadObj()->InheritsFrom("TH2");
 			if (is_tree) tree_names.insert(name);
-			if (is_h1) h1_names.insert(name);
+			if (is_h1) { h1_names.insert(name); cout << name << "## " << static_cast<TH1D*>(key->ReadObj())->GetMean() << endl; }
 			if (is_h2) h2_names.insert(name);
 		}
 
 		input_handles.push_back(handle);
 	}
 
-	cout << "found " << tree_names.size() << " tree "<< h1_names.size() << " histo1d " << h2_names.size() << " histo2d" << endl;
-
-
 	for (Handles::const_iterator iter=input_handles.begin(); iter!=input_handles.end(); iter++)
 	{
 		(*iter)->Close();
 		delete *iter;
 	}
+
+	cout << "found " << tree_names.size() << " tree "<< h1_names.size() << " histo1d " << h2_names.size() << " histo2d" << endl;
+
+	{
+		cout << "merging trees" << endl;
+		for (StringsSet::const_iterator iter_tree=tree_names.begin(); iter_tree!=tree_names.end(); iter_tree++)
+		{
+			cout << "  tree " << *iter_tree;
+			TChain* chain = new TChain(iter_tree->c_str());
+			for (Strings::const_iterator iter_filename=input_filenames.begin(); iter_filename!=input_filenames.end(); iter_filename++)
+				chain->Add(iter_filename->c_str());
+			cout << " nentries " << chain->GetEntries() << endl;
+
+			if ((*iter_tree) == "Singles" || (*iter_tree) == "Hits") mergeSingleTree(chain,output_handle,*iter_tree);
+			if ((*iter_tree) == "Gate") mergeGateTree(chain,output_handle,*iter_tree);
+			if ((*iter_tree) == "Coincidences") mergeCoinTree(chain,output_handle,*iter_tree);
+
+			delete chain;
+		}
+	}
+
+
 }