/data3/calcul/jacquem/working_dir/Micromegas/micromegasFrameWork/src/analyse/root/addDifSynchro.cpp File Reference

#include <map>
#include <stdlib.h>
#include <string>
#include <set>
#include <list>
#include "TKey.h"
#include <TRint.h>
#include <TTree.h>
#include <TFile.h>
#include <iostream>
#include <sstream>
#include "tools/Log.hh"
#include "tools/Toolbox.hh"
#include "root/MTEvent.hh"
#include "MicroException.hh"

Include dependency graph for addDifSynchro.cpp:

Go to the source code of this file.

Functions
int	main (int argc, char **argv)

---

Function Documentation

int main	(	int	argc,
		char **	argv
	)

Definition at line 41 of file addDifSynchro.cpp.

                              {

   if ( argc !=2  ) {
       FILE_LOG(logERROR)  << "usage: addDifSynchro rootfile" << endl;
       exit(1);
   }

    // Get root file name from command arguments and open it
    string rootName;
    rootName.assign(argv[1]);
    TFile f(rootName.c_str(),"READONLY");
    TIter nextkey(f.GetListOfKeys());


    // Recherche du TTree Dif_synchro
    TKey *key;
    vector<string> slabTreeNames;

    TTree *difTree = NULL;
    MTEvent *findDifevt =  new MTEvent();

    while (key = (TKey*)nextkey())
    {
        TTree* tmp = (TTree*)key->ReadObj();
        TBranch *branch= tmp->GetBranch("MTEvent");
        branch->SetAddress(&findDifevt);
        tmp->GetEntry(0);

        // Si DifSynchro > 0 c'est un fichier DIF SYNCHRO
        if (  findDifevt->GetDifSynchro() > 0 )
        {
           difTree = tmp;
           cout << "Dif tree[" << (tmp->GetName()) << "]"<< endl;
        }
        else
        { // SINON C'EST UN FICHIER SLAB 
          slabTreeNames.push_back(string(tmp->GetName()));
        }
        
    }
    delete findDifevt;



    // Fill DifSynchro vector
    if ( difTree == NULL ) { cout << "Could not find DIF SYNCRO TTRee! "<< endl; exit(-1); }
    vector<ULong64_t> difSynchro;
    MTEvent *difEvt =  new MTEvent();
    difTree->SetBranchAddress("MTEvent", &difEvt);
    for ( int eventNum =0 ; eventNum < difTree->GetEntries() ; eventNum++ )
    {
      difTree->GetEntry(eventNum);
      difSynchro.push_back(difEvt->GetDifSynchro());
    }
    delete difEvt;

    string mergeFileName = rootName.substr(0,rootName.find("root")-1)+"_DS.root";
    TFile *newFile = new TFile(mergeFileName.c_str(), "RECREATE");
    newFile->cd();


    TTree* mergedTree = NULL;
    for(vector<string>::iterator slabNameIt=slabTreeNames.begin(); slabNameIt!=slabTreeNames.end(); slabNameIt++)
    {

      TTree* readTree = (TTree*)f.Get((*slabNameIt).c_str());
      if ( readTree->GetEntries() !=  difTree->GetEntries() ) { cout << "nb entries differ !!!" << endl; exit(0); }  

      FILE_LOG(logINFO  ) << "Add Dif Synchro to TTree[" << readTree->GetName() << "]" <<endl;
      mergedTree = readTree->CloneTree(0);

    // Set the Dif_Syncho parameter
      MTEvent *readEvt =  new MTEvent();
      readTree->SetBranchAddress("MTEvent",&readEvt);

      for ( int eventNum =0 ; eventNum < readTree->GetEntries() ; eventNum++ )
      {
        readTree->GetEntry(eventNum);
        readEvt->SetDifSynchro(difSynchro[eventNum]);
        mergedTree->Fill();
        if ( eventNum % 100 == 0)
        {
              FILE_LOG(logINFO  ) << "Done for entry " << eventNum << " / " << readTree->GetEntries() << " \r" << flush;
        }

      }
      delete readEvt;
      mergedTree->Write("", TObject::kOverwrite);
    }


    newFile->Close();
     FILE_LOG(logINFO  ) << endl << "Done." << endl;

}

---