TrackAnalysis.h

Go to the documentation of this file.

/* emacs: this is -*- c++ -*- */
#ifndef TIDA_TRACKANALYSIS_H
#define TIDA_TRACKANALYSIS_H
 
#include <iostream>
#include <vector>
#include <string>
#include <map>
 
 
#include "TrigInDetAnalysis/TrackAssociator.h"
#include "TrigInDetAnalysis/Track.h"
#include "TrigInDetAnalysis/TIDAEvent.h"
#include "TrigInDetAnalysis/TIDAVertex.h"
#include "TrigInDetAnalysis/TIDARoiDescriptor.h"
#include "TrigInDetAnalysis/TrigObjectMatcher.h"
#include "TrigInDetAnalysis/TIDAFeatureStore.h"
 
#include "TH1.h"
 
 
class TrackAnalysis { 
 
public:
 
  TrackAnalysis( const std::string& name ) : 
    m_name(name), 
    m_xBeamReference(0), m_yBeamReference(0), m_zBeamReference(0),
    m_xBeamTest(0),      m_yBeamTest(0),      m_zBeamTest(0),
    m_event(0), m_roi(0)
  { } 
  
  virtual ~TrackAnalysis() { } 
 
  std::string name() const { return m_name; }  
    
  virtual void initialise() = 0;
 
  virtual void execute( const std::vector<TIDA::Track*>& tracks1,
            const std::vector<TIDA::Track*>& tracks2,
            TrackAssociator*   matcher ) = 0;
  
  virtual void execute( const std::vector<TIDA::Track*>& tracks1,
            const std::vector<TIDA::Track*>& tracks2,
            TrackAssociator*   matcher, 
            TrigObjectMatcher* /* objects */ ) {
    execute( tracks1, tracks2, matcher );
  }
  
  virtual void execute( const std::vector<TIDA::Track*>& ,
            const std::vector<TIDA::Track*>& ,
            TrackAssociator* , 
            const TIDA::Event* ) { }
 
  virtual void execute( const std::vector<TIDA::Track*>& tracks1,
            const std::vector<TIDA::Track*>& tracks2,
            TrackAssociator* matcher, 
            const TIDA::Event* event, 
            double* ) { 
    execute( tracks1, tracks2, matcher, event );
  }
 
  virtual void execute( const std::vector<TIDA::Track*>& ,
            const std::vector<TIDA::Track*>& ,
            TrackAssociator* , 
            const TIDA::Event* , 
            double* , 
            TIDARoiDescriptor* ) { }
 
 
  virtual void execute_vtx( const std::vector<TIDA::Vertex*>& ,
                const std::vector<TIDA::Vertex*>& , 
                const TIDA::Event* =0 ) { }  
               
  
  virtual void finalise() = 0;
 
  // histogram map access - add a histogram
  void addHistogram( TH1* h ) { 
    std::string name = h->GetName();
    m_histos.insert( std::map<std::string, TH1*>::value_type( name, h) );
  }
  
  const std::map<std::string, TH1*>& getHistograms() const { return m_histos; }
  
  std::map<std::string, TH1*>::const_iterator begin() const { return m_histos.begin(); }
  std::map<std::string, TH1*>::const_iterator end()   const { return m_histos.end(); }
 
  void  setBeamRef( double x, double y, double z=0 ) {  m_xBeamReference = x;  m_yBeamReference = y; m_zBeamReference = z; } 
  void setBeamTest( double x, double y, double z=0 ) {  m_xBeamTest      = x;       m_yBeamTest = y;      m_zBeamTest = z; }
 
  void  setBeamRef( const std::vector<double>& v ) {  
    if ( v.size()>0 ) m_xBeamReference = v[0]; 
    if ( v.size()>1 ) m_yBeamReference = v[1]; 
    if ( v.size()>2 ) m_zBeamReference = v[2]; 
  }
 
  void setBeamTest(  const std::vector<double>& v ) { 
    if ( v.size()>0 ) m_xBeamTest = v[0]; 
    if ( v.size()>1 ) m_yBeamTest = v[1]; 
    if ( v.size()>2 ) m_zBeamTest = v[2]; 
  }
 
  double  beamRefx() const { return m_xBeamReference; }
  double  beamRefy() const { return m_yBeamReference; }
  double  beamRefz() const { return m_zBeamReference; }
 
  double  beamTestx() const { return  m_xBeamTest; }
  double  beamTesty() const { return  m_yBeamTest; }
  double  beamTestz() const { return  m_zBeamTest; }
 
 
  TIDA::FeatureStore& store() { return m_store; }
 
  const TIDA::Event* event() const         { return m_event; }
  void            setevent(TIDA::Event* e) { m_event=e; }
 
  const TIDARoiDescriptor* roi() const               { return m_roi; }
  void                  setroi(TIDARoiDescriptor* r) { m_roi=r; }
 
protected:
 
  std::string   m_name;
 
  std::map<std::string, TH1*> m_histos;
 
  double m_xBeamReference;
  double m_yBeamReference;
  double m_zBeamReference;
 
  double m_xBeamTest;
  double m_yBeamTest;
  double m_zBeamTest;
 
  TIDA::FeatureStore m_store;
  
  TIDA::Event*        m_event;
  TIDARoiDescriptor*  m_roi;
  
};
 
 
inline std::ostream& operator<<(std::ostream& s, const TrackAnalysis& ta) { 
  return s << ta.name();
}
 
 
#endif  // TIDA_TRACKANALYSIS_H