CMSSW_4_4_3_patch1/src/TrackingTools/TrajectoryState/interface/BasicSingleTrajectoryState.h

Go to the documentation of this file.

#ifndef BasicSingleTrajectoryState_H
#define BasicSingleTrajectoryState_H

#include "TrackingTools/TrajectoryState/interface/BasicTrajectoryState.h"
#include "TrackingTools/TrajectoryState/interface/FreeTrajectoryState.h"
#include "TrackingTools/TrajectoryParametrization/interface/LocalTrajectoryParameters.h"
#include "TrackingTools/TrajectoryParametrization/interface/LocalTrajectoryError.h"

#include "DataFormats/GeometrySurface/interface/ReferenceCounted.h"

#include "DataFormats/GeometryCommonDetAlgo/interface/DeepCopyPointer.h"
#include "DataFormats/GeometrySurface/interface/Surface.h"
#include "TrackingTools/TrajectoryParametrization/interface/TrajectoryStateExceptions.h"

#include "FWCore/Utilities/interface/Visibility.h"
#include "FWCore/Utilities/interface/Likely.h"

#include <iostream>

//class MagneticField;
#include "MagneticField/Engine/interface/MagneticField.h"

class BasicSingleTrajectoryState : public BasicTrajectoryState {
public:

  BasicSingleTrajectoryState( const FreeTrajectoryState& fts,
                              const Surface& aSurface,
                              const SurfaceSide side = SurfaceSideDefinition::atCenterOfSurface);
  BasicSingleTrajectoryState( const GlobalTrajectoryParameters& par,
                              const Surface& aSurface,
                              const SurfaceSide side = SurfaceSideDefinition::atCenterOfSurface);
  BasicSingleTrajectoryState( const GlobalTrajectoryParameters& par,
                              const CartesianTrajectoryError& err,
                              const Surface& aSurface,
                              const SurfaceSide side = SurfaceSideDefinition::atCenterOfSurface);
  BasicSingleTrajectoryState( const GlobalTrajectoryParameters& par,
                              const CurvilinearTrajectoryError& err,
                              const Surface& aSurface,
                              const SurfaceSide side = SurfaceSideDefinition::atCenterOfSurface,
                              double weight = 1.);
  BasicSingleTrajectoryState( const GlobalTrajectoryParameters& par,
                              const CurvilinearTrajectoryError& err,
                              const Surface& aSurface,
                              double weight);
  BasicSingleTrajectoryState( const LocalTrajectoryParameters& par,
                              const Surface& aSurface,
                              const MagneticField* field,
                              const SurfaceSide side = SurfaceSideDefinition::atCenterOfSurface);
  BasicSingleTrajectoryState( const LocalTrajectoryParameters& par,
                              const LocalTrajectoryError& err,
                              const Surface& aSurface,
                              const MagneticField* field,
                              const SurfaceSide side = SurfaceSideDefinition::atCenterOfSurface,
                              double weight = 1.);
  BasicSingleTrajectoryState( const LocalTrajectoryParameters& par,
                              const LocalTrajectoryError& err,
                              const Surface& aSurface,
                              const MagneticField* field,
                              double weight);

  explicit BasicSingleTrajectoryState(const Surface& aSurface);

  virtual ~BasicSingleTrajectoryState();

  bool isValid() const {
    return theFreeState || theLocalParametersValid;
  }


// access global parameters/errors
  const GlobalTrajectoryParameters& globalParameters() const {
    return freeTrajectoryState(false)->parameters();
  }
  GlobalPoint globalPosition() const {
    return freeTrajectoryState(false)->position();
  }
  GlobalVector globalMomentum() const {
    return freeTrajectoryState(false)->momentum();
  }
  GlobalVector globalDirection() const {
    return freeTrajectoryState(false)->momentum().unit();
  }  
  TrackCharge charge() const {
    return freeTrajectoryState(false)->charge();
  }
  double signedInverseMomentum() const {
    return freeTrajectoryState(false)->signedInverseMomentum();
  }
  double transverseCurvature() const {
    return freeTrajectoryState(false)->transverseCurvature();
  }
  const CartesianTrajectoryError& cartesianError() const {
    if unlikely(!hasError()) missingError(" accesing cartesian error.");
    return freeTrajectoryState()->cartesianError();
  }
  const CurvilinearTrajectoryError& curvilinearError() const {
    if unlikely(!hasError()) missingError(" accesing curvilinearerror.");
    return freeTrajectoryState()->curvilinearError();
  }


  FreeTrajectoryState* freeTrajectoryState(bool withErrors = true) const;
  
  const MagneticField *magneticField() const { return theField; }

// access local parameters/errors
  const LocalTrajectoryParameters& localParameters() const {
    if unlikely(!isValid()) notValid();
    if unlikely(!theLocalParametersValid)
      createLocalParameters();
    return theLocalParameters;
  }
  LocalPoint localPosition() const {
    return localParameters().position();
  }
  LocalVector localMomentum() const {
    return localParameters().momentum();
  }
  LocalVector localDirection() const {
    return localMomentum().unit();
  }

  const LocalTrajectoryError& localError() const {
    if unlikely(!hasError()) missingError(" accessing local error.");
    if unlikely(!theLocalErrorValid)
      createLocalError();
    return theLocalError;
  }

  const Surface& surface() const {
    return *theSurfaceP;
  }

  virtual double weight() const {return theWeight;} 

  void rescaleError(double factor);


  BasicSingleTrajectoryState* clone() const {
    return new BasicSingleTrajectoryState(*this);
  }

  virtual SurfaceSide surfaceSide() const {
    return theSurfaceSide;
  }

  virtual bool canUpdateLocalParameters() const { return true; }
  virtual void update( const LocalTrajectoryParameters& p,
                       const Surface& aSurface,
                       const MagneticField* field,
                       const SurfaceSide side ) ;
  virtual void update( const LocalTrajectoryParameters& p,
                       const LocalTrajectoryError& err,
                       const Surface& aSurface,
                       const MagneticField* field,
                       const SurfaceSide side,
                       double weight ) ;
private:

  static void notValid() dso_internal;

  bool hasError() const;
  //bool hasError() const {
  //  return (theFreeState && theFreeState->hasError()) || theLocalErrorValid;
  //}

  
  void missingError(char const * where) const; // dso_internal;

// create global parameters and errors from local
  void checkGlobalParameters() const dso_internal;
  void checkCurvilinError() const  dso_internal;
  void checkCartesianError() const  dso_internal;

// create local parameters and errors from global
  void createLocalParameters() const  dso_internal;
  // create local errors from global
  void createLocalError() const  dso_internal;
  void createLocalErrorFromCartesianError() const  dso_internal;
  void createLocalErrorFromCurvilinearError() const  dso_internal;

private:

  mutable DeepCopyPointer<FreeTrajectoryState> theFreeState;

  mutable LocalTrajectoryError      theLocalError;
  mutable LocalTrajectoryParameters theLocalParameters;
  mutable bool                      theLocalParametersValid;
  mutable bool                      theLocalErrorValid;

  mutable bool theGlobalParamsUp2Date;
  mutable bool theCartesianErrorUp2Date;
  mutable bool theCurvilinErrorUp2Date;

 
  SurfaceSide theSurfaceSide;
  ConstReferenceCountingPointer<Surface> theSurfaceP;

  double theWeight;
  const MagneticField* theField;

};

#endif