TrackAnalysisAlgorithm.cc

Go to the documentation of this file.

#include "PhysicsTools/ParallelAnalysis/interface/TrackAnalysisAlgorithm.h"
#include "DataFormats/TrackReco/interface/Track.h"
#include "DataFormats/TrackReco/interface/TrackFwd.h"
#include "DataFormats/Common/interface/Handle.h"
#include "FWCore/Framework/interface/Event.h"
#include <TCanvas.h>
#include <TList.h>
#include <TH1.h>
#include <iostream>
using namespace examples;
using namespace std;
using namespace edm;
using namespace reco;

const char * TrackAnalysisAlgorithm::kPt = "pt";
const char * TrackAnalysisAlgorithm::kEta = "eta";

TrackAnalysisAlgorithm::TrackAnalysisAlgorithm( const TList *, TList& out )  {
  cout << ">> booking histograms" << endl;
  out.Add( h_pt  = new TH1F( kPt , "pt"  , 100,  0, 20 ) );
  out.Add( h_eta = new TH1F( kEta, "#eta", 100, -3,    3 ) );
}

void TrackAnalysisAlgorithm::process( const Event & event ) {
  cout << ">> processing event " << endl;
  Handle<TrackCollection> tracks;
  event.getByLabel( "ctfWithMaterialTracks", tracks );

  cout << ">> tracks found:" << tracks->size() << endl;
  for ( size_t i = 0; i < tracks->size(); ++i ) {
    const Track & track = ( * tracks )[ i ];
    h_pt ->Fill( track.pt() );
    h_eta->Fill( track.eta() );
    cout << ">> pt, eta:  " << track.pt() << ", " << track.eta() << endl;
  }
}

void TrackAnalysisAlgorithm::postProcess( TList & ) {
  cout << ">> nothing to be done in post-processing" << endl;
}

void TrackAnalysisAlgorithm::terminate( TList & out ) {
  cout << ">> terminating" << endl;
  TCanvas canvas;
  draw( out, canvas,  kPt );
  draw( out, canvas, kEta );
}

void TrackAnalysisAlgorithm::draw( const TList & out, TCanvas & canvas, const char * k ) {
  TObject * hist = out.FindObject( k );
  if( nullptr != hist ) {
    hist->Draw();
    canvas.SaveAs( ( string( k ) + ".jpg" ).c_str() );
  } else {
    cerr <<">> no '" << k << "' histogram" << endl;
  }
}