KFitTrack.cxx

Go to the documentation of this file.

#include "lardata/RecoObjects/KFitTrack.h"
#include "cetlib_except/exception.h"

namespace trkf {

  KFitTrack::KFitTrack() :
    fPath(0.),
    fChisq(0.),
    fStat(INVALID)
  {}

  KFitTrack::KFitTrack(const KETrack& tre,
                       double s,
                       double chisq,
                       FitStatus stat) :
    KETrack(tre),
    fPath(s),
    fChisq(chisq),
    fStat(stat)
  {}

  KFitTrack::~KFitTrack()
  {}

  bool KFitTrack::combineFit(const KFitTrack& trf)
  {
    // Make sure that the two track surfaces are the same.
    // Throw an exception if they are not.

    if(!getSurface()->isEqual(*trf.getSurface()))
      throw cet::exception("KFitTrack") << "Track combination surfaces are not the same.\n";

    // Default result failure.

    bool result = false;

    // Make sure input statuses are compatible and figure out the
    // updated status.

    FitStatus stat1 = getStat();
    FitStatus stat2 = trf.getStat();
    FitStatus statu = UNKNOWN;
    if((stat1 == FORWARD && stat2 == BACKWARD_PREDICTED) ||
       (stat1 == FORWARD_PREDICTED && stat2 == BACKWARD) ||
       (stat1 == BACKWARD && stat2 == FORWARD_PREDICTED) ||
       (stat1 == BACKWARD_PREDICTED && stat2 == FORWARD))
      statu = OPTIMAL;
    else if(stat1 == FORWARD_PREDICTED && stat2 == BACKWARD_PREDICTED)
      statu = OPTIMAL_PREDICTED;
    else
      statu = UNKNOWN;
    if(statu != UNKNOWN) {

      // Update track parameters and error matrix.

      boost::optional<double> chisq = combineTrack(trf);

      // Update status and chisquare.

      if(!!chisq) {
        result = true;
        fStat = statu;
        fChisq = fChisq + trf.getChisq() + *chisq;
      }
    }

    // Done.

    return result;
  }

  std::ostream& KFitTrack::Print(std::ostream& out, bool doTitle) const
  {
    if(doTitle)
      out << "KFitTrack:\n";

    // Print information specific to this class.

    out << "  Distance = " << fPath << "\n";
    out << "  Chisquare = " << fChisq << "\n";
    out << "  Status = " << (fStat == INVALID ? "INVALID" :
                             (fStat == UNKNOWN ? "UNKNOWN" :
                              (fStat == FORWARD ? "FORWARD" :
                               (fStat == FORWARD_PREDICTED ? "FORWARD_PREDICTED" :
                                (fStat == BACKWARD ? "BACKWARD" :
                                 (fStat == BACKWARD_PREDICTED ? "BACKWARD_PREDICTED" :
                                  (fStat == OPTIMAL ? "OPTIMAL" : 
                                   "OPTIMAL_PREDICTED"))))))) << "\n";

    // Print base class.

    KETrack::Print(out, false);
    return out;
  }

} // end namespace trkf