---

TestTrackViewer.cxx

Go to the documentation of this file.

// ================================================
// TestTrackViewer class Implementation
// ================================================
//
//
// Namespace ATLFast
//
#include "AtlfastAlgs/TestTrackViewer.h"
#include "AtlfastUtils/ITrackViewer.h"
#include "AtlfastAlgs/AtlfTrackViewer.h"
#include "AtlfastAlgs/ITrackParameterView.h"
#include "AtlfastAlgs/AtlfTrackParameterView.h"

#include "AtlfastEvent/ITrackParameters.h"
#include "AtlfastEvent/CollectionDefs.h"

// Utilities
#include <cmath> 
#include <assert.h> 
#include "AtlfastEvent/Phi.h"
#include "AtlfastUtils/HeaderPrinter.h"

// This is part of the STL and allows you to use
// generic sort algorithms (amongst other things)
#include <algorithm>
#include "AtlfastUtils/FunctionObjects.h"

// Athena/Gaudi includes
#include "GaudiKernel/DataSvc.h"

//#include "AtlfastEvent/MsgStreamDefs.h"

namespace Atlfast {
  using std::abs;
  using std::endl;
  
  
  //--------------------------------
  // Constructors and destructors
  //--------------------------------
  
  TestTrackViewer::TestTrackViewer ( const std::string& name, 
                               ISvcLocator* pSvcLocator ) 
    : Algorithm( name, pSvcLocator ){
    
    // Setting the parameter defaults.
    m_trackLocation      = "/Event/AtlfastTracks";
    declareProperty( "TrackLocation",    m_trackLocation ) ;
    
  }
  
  // Destructor
  TestTrackViewer::~TestTrackViewer() {
    MsgStream log( messageService(), name() ) ;
    log << MSG::INFO << "Destructor called" << endreq;
  } 
  
  
  //---------------------------------
  // initialise() 
  //---------------------------------
  
  StatusCode TestTrackViewer::initialize(){
    
    // We must here instantiate items which can only be made after
    // any job options have been set
    
    MsgStream log( messageService(), name() ) ;
    log << MSG::DEBUG << "Initialising" << endreq;
    HeaderPrinter hp("TestTrackViewer:",log);
    hp.add( "TrackLocation   ",   m_trackLocation ) ;
    hp.print();

    m_nevents = 0;
    m_ntracks = 0;
    //    m_tesIO = new TesIO(eventDataService());
    m_tesIO = new TesIO();
    
    return StatusCode::SUCCESS ;
  }
  
  
  //---------------------------------
  // finalise() 
  //---------------------------------
  
  StatusCode TestTrackViewer::finalize(){
    MsgStream log( messageService(), name() ) ;
    log << MSG::INFO << "Finalizing" << endreq;
    log<<  MSG::INFO  << "Events Processed: "<<m_nevents<<endreq;
    log<<  MSG::INFO  << "Tracks Processed: "<<m_ntracks<<endreq;
    return StatusCode::SUCCESS ;
  }
  
  
  //----------------------------------------------
  // execute() method called once per event
  //----------------------------------------------
  //
  
  
  StatusCode TestTrackViewer::execute( ){
    
    //................................
    // make a message logging stream 
    
    MsgStream log( messageService(), name() ) ;
    ++m_nevents;
    
    
    //.........................................................
    // This is how you extract the things you want to look at. 
    // Look at the userguide for details of ReconstructedParticle, Cell
    // and Cluster classes
    std::vector<Track*> tracks;
    if(!m_tesIO->copy<TrackCollection >(tracks, m_trackLocation)){
      log << MSG::DEBUG 
          << "No Tracks found in TES at " 
          << m_trackLocation 
          << endreq ;
    }else{
      log << MSG::DEBUG 
          << "No. of Tracks found in TES  = " 
          << tracks.size() 
          << endreq ;
    }
    
    //.........................................................
    // Here are example things you might do
    
    
    if( tracks.size() ) this->testViewer( log, tracks ) ;
    
    return StatusCode::SUCCESS ;
    
  }
  
  
  //-------------------------------------------
  // private: testViewer
  //------------------------------------------
  
  
  void TestTrackViewer::testViewer ( MsgStream& log, 
                                     std::vector<Track*>& tracks){
    
    log << MSG::DEBUG << "Starting track view tester" << endreq ;
    
    
    
    // traverse collection of particles
    std::vector<Track*>::const_iterator track;
    
    for( track = tracks.begin(); track < tracks.end(); ++track ){
      
      log << MSG::DEBUG << *track << endreq;
      ++m_ntracks;
      TrackTrajectory traj = (*track)->trajectory();
      AtlfAna::AtlfTrackViewer trackViewer(*track);
      AtlfAna::ITrackParameterView& parameterView =  trackViewer;
      log<<MSG::DEBUG<<parameterView.trackParameters()<<endl;
      const ITrackParameters& tp = parameterView.trackParameters();
      double eps = 0.0000001;
      TrackParameters testParam = (*track)->trajectory().parameters();
      assert(abs( tp.eta()-testParam.eta() )                          < eps);
      assert(abs( tp.phi()-testParam.phi() )                          < eps);
      assert(abs( tp.cotTheta()-testParam.cotTheta() )                < eps);
      assert(abs( tp.impactParameter()-testParam.impactParameter() )  < eps);
      assert(abs( tp.zPerigee()-testParam.zPerigee() )                < eps);
      assert(abs( tp.invPtCharge()-testParam.invPtCharge() )          < eps);
      
    }
    
  }
  
  
} // end of namespace bracket

---