/data3/calcul/jacquem/working_dir/Micromegas/micromegasFrameWork/src/lcio/AddDifSynchro.cpp

Go to the documentation of this file.

/* @version $Revision: 1328 $ * @modifiedby $Author: jacquem $ * @lastmodified $Date: 2011-10-03 17:04:17 +0200 (Mon, 03 Oct 2011) $ */
#include "lcio/AddDifSynchro.hh"
#include "lcio.h"

#include "IO/LCWriter.h"
#include "IO/LCReader.h"
#include "IO/LCEventListener.h"
#include "IO/LCRunListener.h"

#include "EVENT/LCIO.h"
#include "EVENT/TrackerRawData.h"

#include "IMPL/LCEventImpl.h" 
#include "IMPL/LCCollectionVec.h"
#include "IMPL/SimCalorimeterHitImpl.h"
#include "IMPL/CalorimeterHitImpl.h"
#include "IMPL/MCParticleImpl.h" 
#include "IMPL/TrackerHitImpl.h" 
#include "IMPL/TrackImpl.h" 
#include "IMPL/ClusterImpl.h" 
#include "IMPL/ReconstructedParticleImpl.h" 
#include "IMPL/VertexImpl.h" 
//#include "IMPL/ParticleIDImpl.h" 
#include "IMPL/LCFlagImpl.h" 
#include "UTIL/LCTOOLS.h"
#include "UTIL/PIDHandler.h"
//#include "UTIL/IndexMap.h"
#include "IMPL/LCRelationImpl.h"

#include "UTIL/LCRelationNavigator.h"

#include <iostream>
#include <algorithm>

using namespace std ;
using namespace lcio ;

//static std::string FILEN = "simjob.slcio" ;
//static std::string OUTFILEN = "recjob.slcio" ;
//static const int NHITS = 50 ;  // calorimeter hits per event
//static const int nRecP = 10 ; // number of reconstructed particles

/** Example of reading events from the file, add sth. to the event
 * and write it to a new file. This uses the listener mechanism to read 
 * the input file (run headers and events) record by record. <br>
 * 
 *  The RunEventProcessor class is defined for processing run and event records.
 *  This is our analysis module.
 *  For simplicity it is defined in the same file - in a real world application 
 *  it should of course be defined in separate header and source files.
 *  
 */

  
  AddDifSynchro::AddDifSynchro(const string _difSyncrhoFileName, const string _outputFile) : nEvent(0) , difSyncrhoFileName(_difSyncrhoFileName){

    

    // Get Event Number from DifSynchro file
    IO::LCReader* lcReader = IOIMPL::LCFactory::getInstance()->createLCReader() ;
    lcReader->open( difSyncrhoFileName ) ;
    LCEvent* evt = NULL;
    //----------- the event loop -----------
    while( (evt = lcReader->readNextEvent()) != 0 )
    {
                syncEvtId.push_back(evt->getParameters().getIntVal("Dif_syncro" ));
        }

    // open outputfile
    lcWrt = LCFactory::getInstance()->createLCWriter() ;

    try{ 
      lcWrt->setCompressionLevel( 9 ) ;
      lcWrt->open( _outputFile , LCIO::WRITE_NEW ) ; 
    } 
    
    catch(IOException& e){
      cout << "[RunEventProcessor()] Can't open file for writing -  " 
           << e.what()  << endl ;
      exit(1) ;
    }
    
  }
  
  AddDifSynchro::~AddDifSynchro(){

    // close outputfile
        cout << endl << "Done !" << endl;
    lcWrt->close()  ;
    delete lcWrt ;
  }
  
  void AddDifSynchro::processEvent( LCEvent * evt ) { /* used for 'read only' access*/ 

        // Write current read event and just modifing (add Dif_synchro parameter)
    nEvent ++ ;
    lcWrt->writeEvent( evt ) ;
  }

  void AddDifSynchro::modifyEvent( LCEvent * evt ) {

    // here we can modify existing objects that have been read from a stream:
        // get first list element
        int difEvtId = syncEvtId.front();
        // erase first list element
        syncEvtId.pop_front();
        evt->parameters().setValue("Dif_syncro",difEvtId ) ;
    cout << "Write Event[" << evt->getEventNumber() << "] \r" << flush;

  }
  

  void AddDifSynchro::processRunHeader( LCRunHeader* run){

    // just copy run headers to the outputfile
    lcWrt->writeRunHeader( run ) ;
        cout << "Write Run Header"<< endl;
  }

  void AddDifSynchro::modifyRunHeader(LCRunHeader* run){/*  we don't modify anything */;}

---