FreeTrajectoryState.h

Go to the documentation of this file.

#ifndef _TRACKER_FREETRAJECTORYSTATE_H_
#define _TRACKER_FREETRAJECTORYSTATE_H_
 
// base trajectory state class
// track parameters and error covariance matrix
 
#include "TrackingTools/TrajectoryParametrization/interface/GlobalTrajectoryParameters.h"
#include "TrackingTools/TrajectoryParametrization/interface/CartesianTrajectoryError.h"
#include "TrackingTools/TrajectoryParametrization/interface/CurvilinearTrajectoryError.h"
#include "DataFormats/TrajectoryState/interface/TrackCharge.h"
#include "TrackingTools/TrajectoryParametrization/interface/TrajectoryStateExceptions.h"
 
#include <iosfwd>
 
#include "FWCore/Utilities/interface/Visibility.h"
#include "FWCore/Utilities/interface/Likely.h"
 
class FreeTrajectoryState {
public:
  // construst
  //default constructor - needed for Persistency
 
  FreeTrajectoryState() : theCurvilinearError(InvalidError()) {}
 
  FreeTrajectoryState(const GlobalTrajectoryParameters& aGlobalParameters)
      : theGlobalParameters(aGlobalParameters), theCurvilinearError(InvalidError()) {}
 
  FreeTrajectoryState(const GlobalPoint& aX,
                      const GlobalVector& aP,
                      TrackCharge aCharge,
                      const MagneticField* fieldProvider)
      : theGlobalParameters(aX, aP, aCharge, fieldProvider), theCurvilinearError(InvalidError()) {}
 
  FreeTrajectoryState(const GlobalPoint& aX,
                      const GlobalVector& aP,
                      TrackCharge aCharge,
                      const MagneticField* fieldProvider,
                      GlobalVector fieldValue)
      : theGlobalParameters(aX, aP, aCharge, fieldProvider, fieldValue), theCurvilinearError(InvalidError()) {}
 
  FreeTrajectoryState(const GlobalTrajectoryParameters& aGlobalParameters,
                      const CurvilinearTrajectoryError& aCurvilinearError)
      : theGlobalParameters(aGlobalParameters), theCurvilinearError(aCurvilinearError) {}
 
  FreeTrajectoryState(const GlobalTrajectoryParameters& aGlobalParameters,
                      const CartesianTrajectoryError& aCartesianError)
      : theGlobalParameters(aGlobalParameters) {
    createCurvilinearError(aCartesianError);
  }
 
  FreeTrajectoryState(const GlobalTrajectoryParameters& aGlobalParameters,
                      const CartesianTrajectoryError&,
                      const CurvilinearTrajectoryError& aCurvilinearError)
      : theGlobalParameters(aGlobalParameters), theCurvilinearError(aCurvilinearError) {}
 
  // access
  // propagate access to parameters
  GlobalPoint position() const { return theGlobalParameters.position(); }
  GlobalVector momentum() const { return theGlobalParameters.momentum(); }
  TrackCharge charge() const { return theGlobalParameters.charge(); }
  double signedInverseMomentum() const { return theGlobalParameters.signedInverseMomentum(); }
  double transverseCurvature() const { return theGlobalParameters.transverseCurvature(); }
 
  // direct access
 
  bool hasCurvilinearError() const { return theCurvilinearError.valid(); }
 
  bool hasError() const { return hasCurvilinearError(); }
 
  const GlobalTrajectoryParameters& parameters() const { return theGlobalParameters; }
 
  CartesianTrajectoryError cartesianError() const {
    if
      UNLIKELY(!hasError()) missingError();
    CartesianTrajectoryError aCartesianError;
    createCartesianError(aCartesianError);
    return aCartesianError;
  }
 
  const CurvilinearTrajectoryError& curvilinearError() const {
    if
      UNLIKELY(!hasError()) missingError();
    return theCurvilinearError;
  }
 
  void rescaleError(double factor);
 
  void setCartesianError(const CartesianTrajectoryError& err) { createCurvilinearError(err); }
  void setCartesianError(const AlgebraicSymMatrix66& err) { createCurvilinearError(CartesianTrajectoryError(err)); }
 
  CurvilinearTrajectoryError& setCurvilinearError() { return theCurvilinearError; }
 
  void setCurvilinearError(const CurvilinearTrajectoryError& err) { theCurvilinearError = err; }
  //  void setCurvilinearError(const AlgebraicSymMatrix55 &err) {
  //        theCurvilinearError = CurvilinearTrajectoryError(err);
  //  }
 
private:
  void missingError() const;  // dso_internal;
 
  // convert curvilinear errors to cartesian
  void createCartesianError(CartesianTrajectoryError& aCartesianError) const;  // dso_internal;
 
  // convert cartesian errors to curvilinear
  void createCurvilinearError(CartesianTrajectoryError const& aCartesianError) const;  // dso_internal;
 
private:
  GlobalTrajectoryParameters theGlobalParameters;
  mutable CurvilinearTrajectoryError theCurvilinearError;
};
 
std::ostream& operator<<(std::ostream& os, const FreeTrajectoryState& fts);
 
#endif