KinematicState.h

Go to the documentation of this file.

#ifndef KinematicState_H
#define KinematicState_H
 
#include "RecoVertex/KinematicFitPrimitives/interface/KinematicParameters.h"
#include "RecoVertex/KinematicFitPrimitives/interface/KinematicParametersError.h"
#include "RecoVertex/KinematicFitPrimitives/interface/ParticleMass.h"
#include "TrackingTools/TrajectoryState/interface/FreeTrajectoryState.h"
#include "MagneticField/Engine/interface/MagneticField.h"
 
class KinematicState {
public:
  KinematicState() { vl = false; }
 
  KinematicState(const KinematicParameters& parameters,
                 const KinematicParametersError& error,
                 const TrackCharge& charge,
                 const MagneticField* field);
 
  KinematicState(const FreeTrajectoryState& state, const ParticleMass& mass, float m_sigma)
      : fts(state),
        param(state.position().x(),
              state.position().y(),
              state.position().z(),
              state.momentum().x(),
              state.momentum().y(),
              state.momentum().z(),
              mass),
        err(state.cartesianError(), m_sigma),
        vl(true) {}
 
  bool operator==(const KinematicState& other) const;
 
  ParticleMass mass() const { return param.vector()[6]; }
 
  KinematicParameters const& kinematicParameters() const { return param; }
 
  KinematicParametersError const& kinematicParametersError() const { return err; }
 
  GlobalTrajectoryParameters const& trajectoryParameters() const { return fts.parameters(); }
 
  GlobalVector globalMomentum() const { return fts.momentum(); }
 
  GlobalPoint globalPosition() const { return fts.position(); }
 
  TrackCharge particleCharge() const { return fts.charge(); }
 
  FreeTrajectoryState freeTrajectoryState() const { return fts; }
 
  bool isValid() const { return vl; }
 
  GlobalVector magneticFieldInInverseGeV(const GlobalPoint& x) const {
    return trajectoryParameters().magneticFieldInInverseGeV(x);
  }
  GlobalVector magneticFieldInInverseGeV() const { return trajectoryParameters().magneticFieldInInverseGeV(); }
 
  const MagneticField* magneticField() const { return &trajectoryParameters().magneticField(); }
 
private:
  FreeTrajectoryState fts;
  KinematicParameters param;
  KinematicParametersError err;
 
  bool vl;
};
#endif