RefitProcessor.h

#ifndef RefitProcessor_h
#define RefitProcessor_h 1

#include "marlin/Processor.h"
#include "lcio.h"
#include <string>

#include <UTIL/LCRelationNavigator.h>

#include <EVENT/TrackerHit.h>

namespace MarlinTrk{
  class IMarlinTrkSystem ;
}


class RefitProcessor : public marlin::Processor {
  
public:
  
  virtual marlin::Processor*  newProcessor() { return new RefitProcessor ; }
  
  RefitProcessor() ;
  
  virtual void init() ;
  
  virtual void processRunHeader( lcio::LCRunHeader* run ) ;
  
  virtual void processEvent( lcio::LCEvent * evt ) ; 
  
  
  virtual void check( lcio::LCEvent * evt ) ; 
  
  
  virtual void end() ;
  
  
  
  struct compare_r {
    bool operator()( EVENT::TrackerHit* a, EVENT::TrackerHit* b)  const { 
      double r_a_sqd = a->getPosition()[0] * a->getPosition()[0] + a->getPosition()[1] * a->getPosition()[1] ; 
      double r_b_sqd = b->getPosition()[0] * b->getPosition()[0] + b->getPosition()[1] * b->getPosition()[1] ; 
      return ( r_a_sqd < r_b_sqd ) ; 
    }
  } ;
  
  
  
protected:
  
  /* helper function to get collection using try catch block */
  lcio::LCCollection* GetCollection( lcio::LCEvent * evt, std::string colName ) ;
  
  /* helper function to get relations using try catch block */
  lcio::LCRelationNavigator* GetRelations(lcio::LCEvent * evt, std::string RelName ) ;
  
  std::string _input_track_col_name ;
  
  std::string _input_track_rel_name ;
  
  std::string _output_track_col_name ;
  
  std::string _output_track_rel_name ;
  
  MarlinTrk::IMarlinTrkSystem* _trksystem ;
  
  bool _MSOn ;
  bool _ElossOn ;
  bool _SmoothOn ;
  
  float _initialTrackError_d0;
  float _initialTrackError_phi0;
  float _initialTrackError_omega;
  float _initialTrackError_z0;
  float _initialTrackError_tanL;
  float _maxChi2PerHit;
  double _mass ;

  int _n_run ;
  int _n_evt ;

  int _initialTrackState;
  int _fitDirection ; 

  std::string _trkSystemName ;
  
  float _bField;
  
} ;

#endif