more root merger

[clitk.git] / tests_dav / clitkMergeRootFiles.cxx

diff --git a/tests_dav/clitkMergeRootFiles.cxx b/tests_dav/clitkMergeRootFiles.cxx
index cb352bec897f6370e66335c3db2b557205d8ba9e..957a8ed8eac06a48adb5796ef6ad608bd6e270e1 100644 (file)
--- a/tests_dav/clitkMergeRootFiles.cxx
+++ b/tests_dav/clitkMergeRootFiles.cxx
@@ -11,10 +11,119 @@
 #include "clitkMergeRootFiles_ggo.h"
 #include "clitkCommon.h"
 #include <string> 
-#include "TROOT.h"
-#include "TPluginManager.h"
-#include "TFile.h"
-#include "TFileMerger.h"
+#include <TROOT.h>
+#include <TPluginManager.h>
+#include <TFile.h>
+#include <TKey.h>
+#include <TFileMerger.h>
+#include <TTree.h>
+#include <TChain.h>
+#include <TH1.h>
+#include <TH2.h>
+#include <iostream>
+#include <set>
+
+using std::endl;
+using std::cout;
+
+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");
+
+       Handles input_handles;
+       StringsSet tree_names;
+       StringsSet h1_names;
+       StringsSet h2_names;
+       for (Strings::const_iterator iter=input_filenames.begin(); iter!=input_filenames.end(); iter++)
+       {
+               TFile* handle = TFile::Open(iter->c_str(),"READ");
+               assert(handle);
+
+               TIter key_next(handle->GetListOfKeys());
+               while (TKey* key = static_cast<TKey*>(key_next()))
+               {
+                       const std::string name = key->GetName();
+                       const bool is_tree = key->ReadObj()->InheritsFrom("TTree");
+                       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); cout << name << "## " << static_cast<TH1D*>(key->ReadObj())->GetMean() << endl; }
+                       if (is_h2) h2_names.insert(name);
+               }
+
+               input_handles.push_back(handle);
+       }
+
+       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;
+               }
+       }
+
+
+}
+
 
 //-----------------------------------------------------------------------------
 int main(int argc, char * argv[]) {
@@ -30,6 +139,33 @@ int main(int argc, char * argv[]) {
     FATAL("Error, please provide at least two inputs files");
   }
 
+  { // Detect Pet output
+         bool all_pet_output = true;
+         Strings input_filenames;
+         for (uint i=0; i<args_info.input_given; i++) 
+         {
+                 const char* filename = args_info.input_arg[i];
+                 input_filenames.push_back(filename);
+                 TFile* handle = TFile::Open(filename,"READ");
+                 assert(handle);
+                 TTree* hits = dynamic_cast<TTree*>(handle->Get("Hits"));
+                 TTree* singles = dynamic_cast<TTree*>(handle->Get("Singles"));
+                 const bool is_pet_output = (hits!=NULL) && (singles!=NULL);
+                 cout << "testing " << filename << " is_pet_output " << is_pet_output << endl;
+                 handle->Close();
+                 delete handle;
+                 all_pet_output &= is_pet_output;
+         }
+         cout << "all_pet_output " << all_pet_output << endl;
+
+         if (all_pet_output)
+         {
+                 mergePetRoot(input_filenames,args_info.output_arg);
+                 return 0;
+         }
+  }
+
+
   // Merge
   TFileMerger * merger = new TFileMerger;
   for (uint i=0; i<args_info.input_given; i++) merger->AddFile(args_info.input_arg[i]);