/data3/calcul/jacquem/working_dir/Micromegas/micromegasFrameWork/src/analyse/root/process_example.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)  << ":process_example 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();
  treeName = treeName + "_preprocessed" ;

  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
    run->SetProcessed(true);

    // Save MTRun infiormation to new processed TTree
    updateTree->GetUserInfo()->Add(run);
    MTEvent *evt =  new MTEvent();
    TBranch *branch= tree->GetBranch("MTEvent");
    branch->SetAddress(&evt);


    MTChannel* channel =NULL;
    int nbEntries = tree->GetEntries();
    TRandom random;
    //for ( int evtNum = 0; evtNum < nbEntries ; evtNum++)
    for ( int evtNum = 0; evtNum < 10 ; evtNum++)
    {
      tree->GetEntry(evtNum);

      // Copy original MTEvent to new processed MTEvent
      *updateEvt = *evt;

      // Processed MTEvents
      Float_t  ax =0.;
      Float_t  bx =0.;
      Float_t  ay =0.;
      Float_t  by=0.;
      ax = random.Rndm();
      bx = random.Rndm();
      ay = random.Rndm();
      by = random.Rndm();

      MTTrack *track = new MTTrack(true, // fromPad
                                  false, // fromStrip
                                  true , // straight 
                                  false, // singleHit
                                  ax,  // ax 
                                  bx,   // bx 
                                  ay,  // ay
                                  by); // by

      updateEvt->SetTimeInfo(false);
      updateEvt->SetSquareEvt(true);
      updateEvt->SetTrack(track);

      // Write MTEvent into Branch
      cout  << "Write Event[" << evtNum << "] \t\t\t\t\t\t\t\t\t\t\t\r" << flush;
      updateTree->Fill();
      delete track;

    }

    // 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();


  //delete updateEvt;
  // Replace original root file by the new processed one
  rename(outputfile.c_str(),inputfile.c_str());
  
  cout << endl << "Done." << endl;


  // Set run processed information into database
  Mysql mysql;
  mysql.updateRunProcessInfo(10211, // runId
                             12.2, // offsetX
                             45.3, // offsetY 
                             123.1, // pressure
                             37.4, // temperature
                             45 );// deltaTMax 

  mysql.addRunChamberProcessInfo(10211,
                                 10,  // chamberId,
                                 1.2, // xmean, 
                                 2.3, // ymean,
                                 99.3,// inefficiency, 
                                 15.5,// nHitMean,
                                 8) ;// deltaTMax;

  
    ui32 runId = 10211;
    ui16 chamberId = 10;
  
    cout <<  "OffsetX: " <<  mysql.getRunOffsetX(  runId) << endl;
    cout <<  "OffsetY: " << mysql.getRunOffsetY(  runId) << endl;
    cout <<  "Pressure: " <<  mysql.getRunPressure(  runId) << endl;
    cout <<  "Temperature: " <<  mysql.getRunTemperature(  runId) << endl;
    cout <<  "DeltaTMax: " <<  mysql.getRunDeltaTMax(  runId) << endl;

    cout <<  "XMean: " <<  mysql.getRunChamberXMean( runId,  chamberId) << endl;
    cout <<  "YMean: " <<  mysql.getRunChamberYMean( runId,  chamberId) << endl;
    cout <<  "Inefficiency: " <<  mysql.getRunChamberInefficiency( runId,  chamberId) << endl;
    cout <<  "NHitMean: " <<  mysql.getRunChamberNHitMean( runId,  chamberId) << endl;
    cout <<  "DeltaTMax: " <<  mysql.getRunChamberDeltaTMax( runId,  chamberId) << endl;


return 0;
}

---