/data3/calcul/jacquem/working_dir/Micromegas/micromegasFrameWork/src/analyse/root/setRunId.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 "tools/Log.hh"
#include "tools/Toolbox.hh"

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

#include "mysql/Mysql.hh"


#include "mysql/Mysql.hh"

#include "TFile.h"
#include "TTree.h"
#include "TKey.h"
#include "TBranch.h"
#include "TRandom.h"


#include <sstream>
#include <sys/types.h>
#include <unistd.h>
#include <stdio.h>


using namespace std;

/*
 Modifying a existing TTree is often not possible because the tree is in a read-only
 file and you do not have permission to save the modified tree .
 Even if you do have the permission, you risk losing the
 original tree with an unsuccessful attempt to save  the modification.
*/
//---------------------------------------------
//
//---------------------------------------------
int main(int argc, char**argv){

   if ( argc != 2  ) {
       FILE_LOG(logERROR)  << "usage:" << endl;
       FILE_LOG(logERROR)  << "setRunId fileToProcess" << endl;
       exit(1);
   }
  string inputfile = argv[1] ;

  // build temporaly fill path and name
  pid_t pid = getpid(); 
  std::ostringstream oss;
  oss << "/tmp/" << pid << ".root";
  string outputfile = oss.str();

  // Open Root TFile in read only mode
  TFile infile(inputfile.c_str(),"READ");

  // Open Root TFile in write mode
  TFile oufile(outputfile.c_str(),"RECREATE");
  // Change current directory to "this" directory
  // Mandatory to use updateTree->GetCurrentFile() later
  oufile.cd();



  // Loop over original TTrees
  TIter nextkey(infile.GetListOfKeys());
  TKey *key;
  MTEvent *updateEvt = new MTEvent();

  while (key = (TKey*)nextkey()) 
  {
    TTree *tree = (TTree*)key->ReadObj();

  // Create nwew TTree which will contain processed MTRun and MTEvent
  string treeName = tree->GetName();

  TTree *updateTree = new TTree(treeName.c_str(),"MicroMegas event");
  updateTree->Branch("MTEvent",&updateEvt,16000,2);

    MTRun * run  = (MTRun*)tree->GetUserInfo()->FindObject("MTRun");

    // Set Processed flag to MTRun
    ui32 runId = run->GetRunId();
    ui32 sqlrunId = UNDEFINED_ID;
    if ( runId != 0 && runId != UNDEFINED_ID )
    {
      cout << "Run id alread set[" << runId << "]" << endl;
    }
    else
    {
      size_t last_slash = inputfile.find_last_of("/")  +1 ;
      string name = inputfile.substr(last_slash);
      string path = inputfile.substr(0,last_slash-1);
      cout << "search file FromRawFile: path[" << path << "] , name[" << name <<"]" << endl; 

      Mysql mysql;
      try {
        sqlrunId = mysql.getRunIdFromRawFile(path,name);
      }
      catch  (MicroException &e) {; } 

      if ( sqlrunId == UNDEFINED_ID)
      {
  //      cout << "search file RebuildFile: path[" << path << "] , name[" << name <<"]" << endl; 
        try {
          sqlrunId = mysql.getRunIdFromRebuildFile(path,name);
        }
        catch  (MicroException &e) {; } 
        if ( sqlrunId == UNDEFINED_ID )
        {
   //       cout << "search file FromMergeFile: path[" << path << "] , name[" << name <<"]" << endl; 
          try {
            sqlrunId = mysql.getRunIdFromMergeFile(path,name);
          }
          catch  (MicroException &e) {; } 
        }
      }
   // cout << "after reseacrh RunId[" << sqlrunId << "] [0x" << hex << sqlrunId << "]" << endl;
    }
      
    if (  sqlrunId == UNDEFINED_ID )
    {
       cout << "No run correspond to this fuile in database !" << endl; 
       exit(-1); 
    }

    // Save MTRun infiormation to new processed TTree
    run->SetRunId(sqlrunId);
    cout << "set  RunId[" << dec << sqlrunId  << endl;
    updateTree->GetUserInfo()->Add(run);
    MTEvent *evt =  new MTEvent();
    TBranch *branch= tree->GetBranch("MTEvent");
    branch->SetAddress(&evt);

    int nbEntries = tree->GetEntries();
    for ( int evtNum = 0; evtNum < 10 ; evtNum++)
    {
      tree->GetEntry(evtNum);
      updateTree->Fill();
    }

    // Write Root Object to TFile
    TFile *curFile =updateTree->GetCurrentFile();
    updateTree->Write("", TObject::kOverwrite);
    updateTree->SetDirectory(0);
  
    // free heap memory
    delete evt;
}

  infile.Close();
  oufile.Close();


  // Replace original root file by the new processed one
  string command = "cp -f ";
  command = command + outputfile + " "  + inputfile ;
  int result = system(command.c_str());
  if ( result == -1 ) { cout <<  strerror(errno) << endl; }
  command = "rm -f " + outputfile;
  result = system(command.c_str());
  if ( result == -1 ) { cout <<  strerror(errno) << endl; }

  
  cout << endl << "Done." << endl;


return 0;
}

---