NtupleTrackSelector.h

Go to the documentation of this file.

/* emacs: this is -*- c++ -*- */
#ifndef TIDAEXAMPLE_NTUPLETRACKSELECTOR_H
#define TIDAEXAMPLE_NTUPLETRACKSELECTOR_H
 
#include "TrigInDetAnalysis/Track.h"
#include "TrigInDetAnalysis/TrackSelector.h"
#include "TrigInDetAnalysis/TrackFilter.h"
 
 
class NtupleTrackSelector : public TrackSelector { 
 
public:
 
  NtupleTrackSelector( TrackFilter* selector=0 ) : TrackSelector(selector) {  } 
 
  virtual TrackSelector* clone() override { return new NtupleTrackSelector(*this); } 
 
  void selectTrack( TIDA::Track* track ) {     
    // do the track extraction stuff here....
    if ( track ) { 
      TIDA::Track* t = new TIDA::Track( *track );
      if ( !addTrack( t ) ) delete t;
      // addTrack( track );
    }
  }
 
 
  void selectTrack( const TIDA::Track& track ) {
    // do the track extraction stuff here....
    TIDA::Track* t = new TIDA::Track( track );
    if ( !addTrack( t ) ) delete t;
    //  addTrack( &track );
  }
 
  void selectTracks( const std::vector<TIDA::Track>& tracks ) {     
    // do the track extraction stuff here....
    std::vector<TIDA::Track>::const_iterator trackitr = tracks.begin();
    std::vector<TIDA::Track>::const_iterator trackend = tracks.end();
    while ( trackitr!=trackend ) { 
      selectTrack( *trackitr );
      trackitr++;
    }
  }
 
  void selectTracks( const std::vector<TIDA::Track*>& tracks ) {     
    // do the track extraction stuff here....
    std::vector<TIDA::Track*>::const_iterator trackitr = tracks.begin();
    std::vector<TIDA::Track*>::const_iterator trackend = tracks.end();
    while ( trackitr!=trackend ) { 
      //      selectTrack( *(*trackitr) );
      selectTrack( *trackitr );
      trackitr++;
    }
  }
 
  virtual void clear() override {
    for ( int i=m_tracks.size() ; i-- ; ) delete m_tracks[i]; 
    m_tracks.clear(); 
  }
 
};
 
 
 
 
#endif // TIDAEXAMPLE_NTUPLETRACKSELECTOR_H