TrackKalmanFilter.h

Go to the documentation of this file.

// $Id: TrackKalmanFilter.h,v 1.1 2005/06/29 15:35:02 erodrigu Exp $
#ifndef TRACKFITTER_TRACKKALMANFILTER_H 
#define TRACKFITTER_TRACKKALMANFILTER_H 1

// Include files
// -------------
// from Gaudi
#include "GaudiAlg/GaudiTool.h"

// from TrackInterfaces
#include "TrackInterfaces/ITrackExtrapolator.h"            
#include "TrackInterfaces/ITrackProjector.h"
#include "TrackInterfaces/ITrackFitter.h"

// from TrackEvent
#include "Event/Track.h"
#include "Event/FitNode.h"

class TrackKalmanFilter : public GaudiTool, public ITrackFitter {
public: 
  TrackKalmanFilter( const std::string& type, 
                     const std::string& name,
                     const IInterface* parent);

  virtual ~TrackKalmanFilter( );

  StatusCode initialize();
  
  StatusCode fit(Track& track) {
    //track.setHistoryFit( TrackKeys::Kalman );
    info() << " not implemented yet!" << track.nMeasurements() << endreq;
    return StatusCode::SUCCESS;
  }

  StatusCode fit(Track& track, const State& seed);
  
  StatusCode filter(Track& track, const State& seed);
  
  StatusCode filter(State& state, Measurement& meas);
  
protected:

  StatusCode iniKalman(Track& track);
  
  StatusCode predict(FitNode& node, State& state);
  
  StatusCode filter(FitNode& node);
  
  StatusCode smoother(Track& track);
  
  StatusCode smooth(FitNode& node0, const FitNode& node1);
  
  void computeChiSq(Track& track);
  
protected:
  
  // ! check that the contents of the cov matrix are fine
  StatusCode checkInvertMatrix(const HepSymMatrix& mat);

  // ! check that the contents of the cov matrix are fine
  StatusCode checkPositiveMatrix(HepSymMatrix& mat);

  // ! invert this matrix
  StatusCode invertMatrix(HepSymMatrix& mat);
  
  StatusCode cToG3(HepSymMatrix& C);
  
  StatusCode cToG4(HepSymMatrix& invC);

protected:

  State* m_state;

  // internal copy of the nodes of the track (track owns them)
  std::vector<FitNode*> m_nodes;

  // internal copy of the gain matrix in the filter of a state
  HepVector m_K;

protected:

  ITrackExtrapolator* m_extrapolator;

  ITrackProjector* m_projector;

private:

  // job options
  std::string m_extrapolatorName;   
  std::string m_projectorName;      
  std::vector<double> m_zPositions; 
  bool m_statesAtMeasZPos;          
  bool m_stateAtBeamLine;           
  int m_numFitIter;                 
  double m_chi2Outliers;            
  double m_storeTransport;          

  StatusCode failure(const std::string& comment);

  bool m_debugLevel;
  
  };
#endif // TRACKFITTER_TRACKKALMANFILTER_H

---