---

trackhit.cpp

Go to the documentation of this file.

#include <XANADOO/domain/interface/XANAEsdEvent.h>
#include <XANADOO/XANAClusters/interface/XANACluster.h>
#include <XANADOO/XANAClusters/interface/XANACaloRecHit.h>
#include <XANADOO/XANAClusters/interface/XANAEmCluster.h>
#include <XANADOO/XANAClusters/interface/XANASuperCluster.h>
#include <XANADOO/XANAClusters/interface/XANAHadCluster.h>
#include <XANADOO/XANATracks/interface/XANATrack.h>
#include <XANADOO/XANATracks/interface/XANATrackHit.h>
#include <XANADOO/XANAMcInfo/interface/XANAGeneratorEvent.h>
#include <XANADOO/XANAMcInfo/interface/XANAGeantEvent.h>
#include <XANADOO/XANAMcInfo/interface/XANAGeneratorParticle.h>
#include <XANADOO/XANAElectronCandidate/interface/XANAElectronCandidate.h>
#include <XANADOO/XANAElectronCandidate/interface/XANAElectronTrack.h>
#include <XANADOO/XANAElectronCandidate/interface/XANAElectronSeed.h>
#include <XANADOO/XANAMuons/interface/XANAMuonCandidate.h>

#include "TTree.h"
#include "TFile.h"
#include "TBranch.h"
#include "TClonesArray.h"
#include "TObjArray.h"
#include "TH1.h"
#include "TH2.h"
#include "TSystem.h"
#include "TROOT.h"
#include "TStyle.h"
#include "TCanvas.h"
#include "TGraph.h" 
#include "TSelector.h"
#include "TApplication.h"


#include <string>


using namespace std;

int main(int argc, char** argv)
{
  //   Connect to file generated
  TFile *f;
  if (argc==1) f = new TFile("./SE04_E25_Pall_PT30_step1.root");
  else f = new TFile(*++argv);
 
  //   Read Tree named "T" in memory. Tree pointer is assigned the same name
  TTree *T = (TTree*)f->Get("ESD");
   
  // Reconstructed event
  XANAEsdEvent *event = 0;  
  T->SetBranchAddress("Event",&event);  
  Short_t ne = 0, nt=0, nth =0, neth =0; 
  TObjArray *tracks = 0;
  TObjArray *trackhits = 0;
  TObjArray *electrons = 0;
  TObjArray *electronTracks = 0;
  TObjArray *electronTrackhits = 0;

  //   Start main loop on all events
  Int_t nb = 0; 
  Int_t nevent = (Int_t)T->GetEntries();
  for (Int_t i=0; i<nevent; i++) {
    nb += T->GetEntry(i); 

    cout << "================ Event: " << i << " ================" << endl;    
    
    // Reconstructed event
    cout << "Reconstructed:" ;

    tracks = event->getTracks();
    nt  = event->getNumberOfTracks(); 
    trackhits = event->getTrackHits();        
    nth = event->getNumberOfTrackHits();  
    electrons = event->getElectronCandidates();    
    ne  = event->getNumberOfElectronCandidates(); 
    electronTracks = event->getElectronTracks();
    electronTrackhits = event->getEleTrackHits();
    neth = event->getNumberOfEleTrackHits();      
 
    // tracks
     XANATrack *tra;
     if ( nt != 0 ){
       cout << "tracks number: " << nt << endl; 

       // loop on tracks  
       int hitcount =0;
       for (int it =0; it < nt; it++){ 
       tra = (XANATrack*)((*tracks)[it]);

       cout << "track " << it << ": number of hits = "<< tra->getNumberOfHits() << endl;

       for (int ithit = hitcount; ithit < (tra->getNumberOfHits() + hitcount); ithit++){ 
       XANATrackHit *trahit = (XANATrackHit*)((*trackhits)[ithit]); 

       cout << "standard track: hit number " << ithit << endl;
       cout << "st: new hit" << endl;
       cout << trahit->getHitPosition().x() << endl;
       cout << trahit->getHitPosition().y() << endl;
       cout << trahit->getHitPosition().z() << endl;
       cout << "st: new updated state: position" << endl;
       cout << trahit->getUpdatedPosition().x() << endl;
       cout << trahit->getUpdatedPosition().y() << endl;
       cout << trahit->getUpdatedPosition().z() << endl;
       cout << "st: new updated state: momentum" << endl;
       cout << trahit->getUpdatedMomentum().x() << endl;
       cout << trahit->getUpdatedMomentum().y() << endl;
       cout << trahit->getUpdatedMomentum().z() << endl;
       cout << "st: isStereo" << endl;
       cout << trahit->getIsStereo() << endl;
 
       // checking first and last hit
       if ( ithit == hitcount){
          cout << "standard" << endl; 
          cout << "position" << endl;
          cout << "first hit, x" << trahit->getUpdatedPosition().x() << endl;  
          cout << "first hit, y" << trahit->getUpdatedPosition().y() << endl;  
          cout << "first hit, z" << trahit->getUpdatedPosition().z() << endl;  
          cout << "momentum" << endl;
          cout << "first hit, px" << trahit->getUpdatedMomentum().x() << endl;  
          cout << "first hit, py" << trahit->getUpdatedMomentum().y() << endl;  
          cout << "first hit, pz" << trahit->getUpdatedMomentum().z() << endl;  
        }
       if ( ithit == (tra->getNumberOfHits() + hitcount -1)){
         cout << "position" << endl;
         cout << "last hit, x" << trahit->getUpdatedPosition().x() << endl;  
         cout << "last hit, y" << trahit->getUpdatedPosition().y() << endl;  
         cout << "last hit, z" << trahit->getUpdatedPosition().z() << endl;  
         cout << "momenutm" << endl;
         cout << "last hit, px" << trahit->getUpdatedMomentum().x() << endl;  
         cout << "last hit, py" << trahit->getUpdatedMomentum().y() << endl;  
         cout << "last hit, pz" << trahit->getUpdatedMomentum().z() << endl;  
        }
       } // end loop over hits
      
       // checking first and last hit  
       cout << "standard method:" << endl;  
       cout << "position" << endl;
       cout << "first hit, x" << tra->getPositionAtFirstPoint().x() << endl;  
       cout << "first hit, y" << tra->getPositionAtFirstPoint().y() << endl;  
       cout << "first hit, z" << tra->getPositionAtFirstPoint().z() << endl;  
       cout << "last hit, x"  << tra->getPositionAtLastPoint().x() << endl;  
       cout << "last hit, y"  << tra->getPositionAtLastPoint().y() << endl;  
       cout << "last hit, z"  << tra->getPositionAtLastPoint().z() << endl;  
       cout << "momenutm" << endl;
       cout << "first hit, px" << tra->getMomentumAtFirstPoint().x() << endl;  
       cout << "first hit, py" << tra->getMomentumAtFirstPoint().y() << endl;  
       cout << "first hit, pz" << tra->getMomentumAtFirstPoint().z() << endl;  
       cout << "last hit, px"  << tra->getMomentumAtLastPoint().x() << endl;  
       cout << "last hit, py"  << tra->getMomentumAtLastPoint().y() << endl;  
       cout << "last hit, pz"  << tra->getMomentumAtLastPoint().z() << endl;  
      

       hitcount = hitcount + tra->getNumberOfHits(); 
       } // end loop on tracks
     } // checking tracks
    

    // electrons
    XANAElectronTrack *eletra;
    int elehitcount =0;

    for (Int_t ie=0; ie<ne; ie++) {
      XANAElectronCandidate *elec = (XANAElectronCandidate*)((*electrons)[ie]);
      cout << "new electron candidate : " << elec << endl;

      if ( elec->getElectronTrack() ){
       eletra = elec->getElectronTrack();

       cout << "new electron track: number of hits = "<< eletra->getNumberOfHits() << endl;

       // loop on all elec track hits        
       for (int eleithit = elehitcount; eleithit < (eletra->getNumberOfHits() + elehitcount); eleithit++){ 
       XANATrackHit *eletrahit = (XANATrackHit*)((*electronTrackhits)[eleithit]); 

       cout << "Electronic track: hit number = " << eleithit << endl;
       cout << "new ele hit" << endl;
       cout << eletrahit->getHitPosition().x() << endl;
       cout << eletrahit->getHitPosition().y() << endl;
       cout << eletrahit->getHitPosition().z() << endl;
       cout << "new ele updated state: position" << endl;
       cout << eletrahit->getUpdatedPosition().x() << endl;
       cout << eletrahit->getUpdatedPosition().y() << endl;
       cout << eletrahit->getUpdatedPosition().z() << endl;
       cout << "new ele updated state: momentum" << endl;
       cout << eletrahit->getUpdatedMomentum().x() << endl;
       cout << eletrahit->getUpdatedMomentum().y() << endl;
       cout << eletrahit->getUpdatedMomentum().z() << endl;
       cout << "new ele hit: isStereo" << endl;
       cout << eletrahit->getIsStereo().x() << endl;

       // checking first and last hit
       if ( eleithit == elehitcount){
          cout << "electronic track" << endl;
          cout << "position" << endl; 
          cout << "first hit, x" << eletrahit->getUpdatedPosition().x() << endl;  
          cout << "first hit, y" << eletrahit->getUpdatedPosition().y() << endl;  
          cout << "first hit, z" << eletrahit->getUpdatedPosition().z() << endl;  
          cout << "momentum" << endl; 
          cout << "first hit, px" << eletrahit->getUpdatedMomentum().x() << endl;  
          cout << "first hit, py" << eletrahit->getUpdatedMomentum().y() << endl;  
          cout << "first hit, pz" << eletrahit->getUpdatedMomentum().z() << endl;  
        }
       if ( eleithit == (eletra->getNumberOfHits() + elehitcount -1)){
         cout << "position" << endl; 
         cout << "last hit, x" << eletrahit->getUpdatedPosition().x() << endl;  
         cout << "last hit, y" << eletrahit->getUpdatedPosition().y() << endl;  
         cout << "last hit, z" << eletrahit->getUpdatedPosition().z() << endl;  
         cout << "momentum" << endl; 
         cout << "last hit, px" << eletrahit->getUpdatedMomentum().x() << endl;  
         cout << "last hit, py" << eletrahit->getUpdatedMomentum().y() << endl;  
         cout << "last hit, pz" << eletrahit->getUpdatedMomentum().z() << endl;  
        }

       } // end loop over hits

       // checking first and last hit  
       cout << "standard method:" << endl;  
       cout << "position" << endl; 
       cout << "first hit, x" << eletra->getPositionAtFirstPoint().x() << endl;  
       cout << "first hit, y" << eletra->getPositionAtFirstPoint().y() << endl;  
       cout << "first hit, z" << eletra->getPositionAtFirstPoint().z() << endl;  
       cout << "last hit, x"  << eletra->getPositionAtLastPoint().x() << endl;  
       cout << "last hit, y"  << eletra->getPositionAtLastPoint().y() << endl;  
       cout << "last hit, z"  << eletra->getPositionAtLastPoint().z() << endl;  
       cout << "momentum" << endl; 
       cout << "first hit, px" << eletra->getMomentumAtFirstPoint().x() << endl;  
       cout << "first hit, py" << eletra->getMomentumAtFirstPoint().y() << endl;  
       cout << "first hit, pz" << eletra->getMomentumAtFirstPoint().z() << endl;  
       cout << "last hit, px"  << eletra->getMomentumAtLastPoint().x() << endl;  
       cout << "last hit, py"  << eletra->getMomentumAtLastPoint().y() << endl;  
       cout << "last hit, pz"  << eletra->getMomentumAtLastPoint().z() << endl;  

       elehitcount = elehitcount + eletra->getNumberOfHits();
      } // ok electron track
    } // loop over electron candidates

  event->clear(); 
  }
  
  cout << endl;
  cout << nevent << " events and " << nb << " bytes read " << endl;  
  cout << endl;
  
  f->Close();  
 
}

---