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

Go to the documentation of this file.

/* @version $Revision: 1566 $ * @modifiedby $Author: jacquem $ * @lastmodified $Date: 2012-03-08 14:00:10 +0100 (Thu, 08 Mar 2012) $ */
#include <map>
#include <stdlib.h>
#include <string>
#include <set>
#include <list>
#include "TKey.h"
#include <TRint.h>
#include <TKey.h>
#include <TTree.h>
#include <TFile.h>
#include <iostream>
#include <sstream>
#include "tools/Log.hh"
#include "tools/Toolbox.hh"


//#include "root/MTRun.hh"
//#include "root/MTChannel.hh"
#include "root/MTEvent.hh"


#include "MicroException.hh"


using namespace std;

/*
  !!! ATTENTION  Le fichier root en entree doit contentir uniquement 2 TTree dont un etant un fichier DifSynchro
  -- Trouver le TTree contenant les info DIF_SYNCHRO
  -- Remplir un vector contenant les donneees dif_synchro ( car il est impossible de lire un donnee d'un TTree pendant que l'on Fill un autre)

  -- cloner le TTree Slab 
  -- Ajouter l'information Dif_Synchro
   
*/  

//---------------------------------------------
//
//---------------------------------------------
int main(int argc, char**argv){

   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;

}

---