/data3/calcul/jacquem/working_dir/Micromegas/micromegasFrameWork/src/ahcalReco.cpp

Go to the documentation of this file.

/* @version $Revision: 1456 $ * @modifiedby $Author: chefdevi $ * @lastmodified $Date: 2012-02-02 16:58:26 +0100 (Thu, 02 Feb 2012) $ */
#include "lcio.h"

#include "IO/LCWriter.h"
#include "EVENT/LCIO.h"
#include <EVENT/CalorimeterHit.h>
#include "EVENT/LCGenericObject.h"
#include "IMPL/LCCollectionVec.h"
#include "IMPL/CalorimeterHitImpl.h"

#include "TFile.h"
#include "TTree.h"

#include "TreeClass.hh"


#include <cstdlib>
#include <iostream>
#include <sstream>


using namespace std ;
using namespace lcio ;



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


  if (argc==3)
  {

    static string lcioFile = argv[1] ;
    TString rootfilename = argv[2];

    IO::LCReader* lcReader = IOIMPL::LCFactory::getInstance()->createLCReader() ;
    lcReader->open( lcioFile ) ;

    bool display = 0;
    bool display_dif = 0;

    int i = 0;
    unsigned int n = i + 1e9 ;

    LCEvent* evt = NULL;

    int nthEvent = 0;//500;//23500;
    //lcReader->skipNEvents(nthEvent) ;

    TFile * tf = new TFile(rootfilename.Data(),"RECREATE");

    float energy = 0;
    float xpos = 0;
    float ypos = 0;
    float zpos = 0;
    int dif_trg = 0;

    TString collec_name;
    TString collec_type;

    cout<<endl;
    cout<<"Read file "<<lcioFile<<endl;
    cout<<endl;

    MTEvent *ahcalEvt =  new MTEvent();
    //TTree *mt = new TTree(lcioFile.c_str(),"AHCAL MT format");
    TTree *mt = new TTree("AHCAL" ,"AHCAL MT format");
    mt->SetMaxTreeSize(4000000000); // 4 Mb
    mt->Branch("MTEvent",&ahcalEvt,16000,2);// second argument is class of evt!!!

    unsigned int energyFill = 0;
    unsigned int difFill = 0;

    while( (evt = lcReader->readNextEvent()) != 0 )
    {
      bool energyFilled = false;
      bool difFilled = false;
        
      i++;
      // Fill MT Event
      ahcalEvt->SetGlobalTriggerCounter(i);

      if (i%100==0){cout<<"Read event "<<i<<" \r"<<flush;}

      //LCTOOLS::dumpEventDetailed( evt );

      const vector<string>* collNames = evt->getCollectionNames();
    
      vector<string>::const_iterator it;
    
      for ( it = collNames->begin(); it!=collNames->end(); ++it) 
      {
        LCCollection*  collec = evt->getCollection(*it);
        if (display){cout<<endl;cout << " collection "<<*it<<" of type " << collec->getTypeName() << endl;}

        collec_name = *it;
        collec_type = collec->getTypeName();

        //Calo Hits
        if (collec_type == "CalorimeterHit")
        {
          for ( int element = 0; element < collec->getNumberOfElements(); element++)
          {
            CalorimeterHit *hit = dynamic_cast<CalorimeterHit*>(collec->getElementAt(element));
            if (hit != NULL)
            {
              if(display)
              {
                cout<<"Hit with energy "<<hit->getEnergy()<<" GeV at position "<<hit->getPosition()[0]<<" "<<hit->getPosition()[1]<<" "<<hit->getPosition()[2]<<endl; 
              }
              energy = hit->getEnergy();
              if  ( element == 0 ) { energyFill++; energyFilled =true; }
              xpos = hit->getPosition()[0];
              ypos = hit->getPosition()[1];
              zpos = hit->getPosition()[2];
              MTChannel mtHit;
              mtHit.SetX(xpos);
              mtHit.SetY(ypos);
              mtHit.SetZ(zpos);
              mtHit.SetEnergy( energy);
              ahcalEvt->AddChannel(&mtHit);
            }
          } 
        }

        //DIF trigger
        if (collec_name == "CALDAQ2_DifTrigger")
        {
          for ( int element = 0; element < collec->getNumberOfElements(); element++)
          {
            LCGenericObject *obj = dynamic_cast<LCGenericObject*>(collec->getElementAt(element));
            if (obj != NULL)
            {       
              dif_trg = obj->getIntVal(0);
              //if ( dif_trg < 0 ) { cout << "dif_trg[" << dif_trg << "] pour i[" << i+nthEvent << "]" << endl; dif_trg = 0;}

              // Fill MT Event dif synchro
              ahcalEvt->SetDifSynchro(dif_trg);
              difFill++;
              difFilled = true;
              ahcalEvt->SetEventId(i);
    
              if (display_dif){cout<<"  dif trigger number of evt["<<i+nthEvent<<"] = "<<dif_trg<<endl;}
            }
          }
        }
      }
      //if ( energyFilled && difFilled ) 
      if ( difFilled ) 
      {
         mt->Fill();
      }  
      ahcalEvt->SetNchannel(0); 
      if (i==n){break;}
    } // end while event

    mt->Fill();
    lcReader->close() ;
    tf->Write("", TObject::kOverwrite);
    tf->Close();
  
    delete lcReader ;
  }
  else
  {
    cout<<"argument 1 is the slcio file - arg 2 is the final rootfile with position, energy and trigger number"<<endl;
  }


  return 0;
}

---