---

XKalman.h

Go to the documentation of this file.

#ifndef ATLFAST_XKALMAN_H
#define ATLFAST_XKALMAN_H

#include "AtlfastCode/TrackTrajectory.h"
#include "CLHEP/Matrix/Matrix.h"
#include <vector>
#include <string>
#include "CLHEP/Random/JamesRandom.h"
#include "AtlfastCode/CorrelatedData.h"
class MsgStream;

namespace Atlfast {
  class CorrelatedData;
//==============================================
// data structure for d0/phi fit parameters
class Displacement{
 public:
  Displacement() {}
  ~Displacement() {}
  Displacement(double off, double pow,
               vector<double> average, vector<double> rms):
    m_off(off), m_pow(pow), m_average(average), m_rms(rms) {}
  Displacement(const Displacement& disp);
  Displacement operator=(const Displacement& disp);

  double off() const {return m_off;}
  double pow() const {return m_pow;}
  double average(int i) const {return m_average[i];}
  double rms(int i) const {return m_rms[i];}
 private:
  double m_off;
  double m_pow;
  vector<double> m_average;
  vector<double> m_rms;
};

//============================================



class XKalman
{
 public:
  XKalman(int, MsgStream&);
  ~XKalman() {delete m_randomEngine; delete m_correlatedData;}
  TrackTrajectory doBremFit(const TrackTrajectory&) const;
 private:

  Displacement m_d0Disp;
  Displacement m_phiDisp;

  string m_file;
  // variables from file
  int m_nRBins, m_nEtaBins, m_nPBins;
  double m_rMin, m_rMax;
  double m_pMin, m_pMax;
  double m_p1scal, m_p2scal;
  HepMatrix m_bremR;
  vector<HepMatrix> m_pTDist;

  MsgStream* m_log;
  CorrelatedData* m_correlatedData;
  HepRandomEngine* m_randomEngine;

  // private methods
  double scaleDPhi(double pT, double old,
                   const Displacement& axis,
                   int iRBrem, double deltaP) const;
  double scalePT(double pT, double pscale) const;

};

}//end of namespace  

#endif

---