Concrete implementation of a tool to fit tracks with TrackKalmanFitter. More...

Inheritance diagram for trkmkr::KalmanFilterFitTrackMaker:

Classes
struct	Config

struct	Options

Public Types
using	Parameters = art::ToolConfigTable< Config >

Public Member Functions
	KalmanFilterFitTrackMaker (Parameters const &p)
	Constructor from Parameters. More...

void	initEvent (const art::Event &e) override
	initialize event: get collection of recob::MCSFitResult More...

bool	makeTrackImpl (const detinfo::DetectorPropertiesData &detProp, const recob::TrackTrajectory &traj, const int tkID, const std::vector< art::Ptr< recob::Hit >> &inHits, const recob::tracking::SMatrixSym55 &covVtx, const recob::tracking::SMatrixSym55 &covEnd, recob::Track &outTrack, std::vector< art::Ptr< recob::Hit >> &outHits, OptionalOutputs &optionals) const
	function that actually calls the fitter More...

bool	makeTrack (const detinfo::DetectorPropertiesData &detProp, const recob::TrackTrajectory &traj, const int tkID, const std::vector< art::Ptr< recob::Hit >> &inHits, recob::Track &outTrack, std::vector< art::Ptr< recob::Hit >> &outHits, OptionalOutputs &optionals) const override

int	getParticleID (const recob::TrackTrajectory &traj, const int tkID) const
	set the particle ID hypothesis More...

double	getMomentum (const recob::TrackTrajectory &traj, const int pid, const bool isFlip, const int tkID) const
	set the initial momentum estimate More...

bool	isFlipDirection (const recob::TrackTrajectory &traj, const int tkID) const
	decide whether to flip the direction or not More...

void	restoreInputPoints (const recob::TrackTrajectory &traj, const std::vector< art::Ptr< recob::Hit >> &inHits, recob::Track &outTrack, std::vector< art::Ptr< recob::Hit >> &outHits, OptionalOutputs &optionals) const

bool	makeTrack (const detinfo::DetectorPropertiesData &detProp, const art::Ptr< recob::TrackTrajectory > ttraj, const std::vector< art::Ptr< recob::Hit >> &inHits, recob::Track &outTrack, std::vector< art::Ptr< recob::Hit >> &outHits, OptionalOutputs &optionals) const
	makeTrack functions with art::Ptr<recob::TrackTrajectory>; calls the purely virtual version with const recob::TrackTrajectory reference as argument. More...


bool	makeTrack (const detinfo::DetectorPropertiesData &detProp, const art::Ptr< recob::Track > track, const std::vector< art::Ptr< recob::Hit >> &inHits, recob::Track &outTrack, std::vector< art::Ptr< recob::Hit >> &outHits, OptionalOutputs &optionals) const

Private Attributes
Parameters	p_

const trkf::TrackStatePropagator	prop

const trkf::TrackKalmanFitter	kalmanFitter

const trkf::TrajectoryMCSFitter	mcsfitter

double	mom_def_

bool	momFromMCSColl_

art::InputTag	mcsInputTag_

bool	momFromCombAndPid_

art::InputTag	contInputTag_

bool	pidFromColl_

art::InputTag	pidInputTag_

double	mom_len_cut_

int	pid_def_

bool	alwaysFlip_

bool	dirFromVec_

recob::tracking::Vector_t	dirVec

const std::vector< recob::MCSFitResult > *	mcs = nullptr

const std::vector< anab::CosmicTag > *	cont = nullptr

const std::vector< anab::ParticleID > *	pid = nullptr

trkf::TrackMomentumCalculator	tmc

Detailed Description

Concrete implementation of a tool to fit tracks with TrackKalmanFitter.

Concrete implementation of a tool to fit tracks with trkf::TrackKalmanFitter; inherits from abstract class TrackMaker. It prepares the input needed by the fitter (momentum, particleId, direction), and returns a track with all outputs filled. If the flag keepInputTrajectoryPoints is set to true, the tracjetory points from the input track are copied into the output, so that only the covariance matrices, the chi2 and the ndof in the output track are resulting from the fit.

For configuration options see KalmanFilterFitTrackMaker::Options and KalmanFilterFitTrackMaker::Config.

Author: G. Cerati (FNAL, MicroBooNE)

Date: 2017

Version: 1.0

Definition at line 47 of file KalmanFilterFitTrackMaker_tool.cc.

Member Typedef Documentation

using trkmkr::KalmanFilterFitTrackMaker::Parameters = art::ToolConfigTable<Config>

Definition at line 125 of file KalmanFilterFitTrackMaker_tool.cc.

Constructor & Destructor Documentation

trkmkr::KalmanFilterFitTrackMaker::KalmanFilterFitTrackMaker ( Parameters const & p )

inlineexplicit

Constructor from Parameters.

Definition at line 128 of file KalmanFilterFitTrackMaker_tool.cc.

References alwaysFlip_, contInputTag_, d, dirFromVec_, trkmkr::KalmanFilterFitTrackMaker::Options::dirVec, kalmanFitter, trkmkr::KalmanFilterFitTrackMaker::Options::keepInputTrajectoryPoints, mcsfitter, mcsInputTag_, mom_def_, mom_len_cut_, momFromCombAndPid_, momFromMCSColl_, p_, pid_def_, pidFromColl_, pidInputTag_, and prop.

       : p_(p)
       , prop{p_().propagator}
       , kalmanFitter{&prop, p_().fitter}
       , mcsfitter{p_().mcsfit}
       , mom_def_{p_().options().defaultMomInGeV()}
       , momFromMCSColl_{p_().options().momFromMCSCollection()}
       , momFromCombAndPid_{p_().options().momFromCombAndPid()}
       , pidFromColl_{p_().options().pidFromCollection()}
       , mom_len_cut_{p_().options().pidFromLengthCut()}
       , pid_def_{p_().options().defaultPdgId()}
       , alwaysFlip_{p_().options().alwaysInvertDir()}
       , dirFromVec_{p_().options().dirFromVec()}
     {
       if (momFromMCSColl_) { mcsInputTag_ = p_().options().mcsInputTag(); }
       if (momFromCombAndPid_) { contInputTag_ = p_().options().contInputTag(); }
       if (pidFromColl_) { pidInputTag_ = p_().options().pidInputTag(); }
       if (dirFromVec_) {
         auto d = p_().options().dirVec();
         dirVec = recob::tracking::Vector_t{d[0], d[1], d[2]};
       }
       //
       if (p_().options().keepInputTrajectoryPoints() &&
           (p_().fitter().sortHitsByPlane() || p_().fitter().sortOutputHitsMinLength() ||
            p_().fitter().skipNegProp())) {
         throw cet::exception("KalmanFilterFitTrackMaker")
           << "Incompatible configuration parameters: keepInputTrajectoryPoints "
              "needs the following fitter options all set to false: "
              "sortHitsByPlane, sortOutputHitsMinLength, skipNegProp."
           << "\n";
       }
       if (momFromMCSColl_ && momFromCombAndPid_) {
         throw cet::exception("KalmanFilterFitTrackMaker")
           << "Incompatible configuration parameters: momFromMCSCollection and "
              "momFromCombAndPid cannot be both true at the same time."
           << "\n";
       }
       if (pidFromColl_ && mom_len_cut_ > 0) {
         throw cet::exception("KalmanFilterFitTrackMaker")
           << "Incompatible configuration parameters: pidFromCollection and "
              "pidFromLengthCut cannot be respectively true and >0. at the same "
              "time."
           << "\n";
       }
       if (alwaysFlip_ && dirFromVec_) {
         throw cet::exception("KalmanFilterFitTrackMaker")
           << "Incompatible configuration parameters: alwaysInvertDir and "
              "dirFromVec cannot be both true at the same time."
           << "\n";
       }
       //
     }

Member Function Documentation

double trkmkr::KalmanFilterFitTrackMaker::getMomentum	(	const recob::TrackTrajectory &	traj,
		const int	pid,
		const bool	isFlip,
		const int	tkID
	)		const

set the initial momentum estimate

Definition at line 301 of file KalmanFilterFitTrackMaker_tool.cc.

References cont, trkf::TrajectoryMCSFitter::fitMcs(), trkf::TrackMomentumCalculator::GetTrackMomentum(), anab::kNotTagged, recob::TrackTrajectory::Length(), mcs, mcsfitter, mom_def_, momFromCombAndPid_, momFromMCSColl_, pid, recob::TrackTrajectory::StartMomentum(), and tmc.

Referenced by makeTrack(), and makeTrackImpl().

 {
   double mom = (mom_def_ > 0 ? mom_def_ : traj.StartMomentum());
   if (momFromMCSColl_) {
     double mcsmom = (isFlip ? mcs->at(tkID).bwdMomentum() : mcs->at(tkID).fwdMomentum());
     // make sure the mcs fit converged
     if (mcsmom > 0.01 && mcsmom < 7.) mom = mcsmom;
   }
   if (momFromCombAndPid_) {
     bool isContained = cont->at(tkID).CosmicType() == anab::kNotTagged;
     // for now momentum from range implemented only for muons and protons
     // so treat pions as muons (~MIPs) and kaons as protons
     int pidtmp = pid;
     if (pidtmp == 211 || pidtmp == 13)
       pidtmp = 13;
     else
       pidtmp = 2212;
     mom = tmc.GetTrackMomentum(traj.Length(), pidtmp);
     if (isContained == false) {
       auto mcsresult = mcsfitter.fitMcs(traj, pid);
       double mcsmom = (isFlip ? mcsresult.bwdMomentum() : mcsresult.fwdMomentum());
       // make sure the mcs fit converged, also the mcsmom should not be less
       // than the range!
       if (mcsmom > 0.01 && mcsmom < 7. && mcsmom > mom) mom = mcsmom;
     }
   }
   return mom;
 }

int trkmkr::KalmanFilterFitTrackMaker::getParticleID	(	const recob::TrackTrajectory &	traj,
		const int	tkID
	)		const

set the particle ID hypothesis

Definition at line 333 of file KalmanFilterFitTrackMaker_tool.cc.

References recob::TrackTrajectory::Length(), mom_len_cut_, pid_def_, and pidFromColl_.

Referenced by makeTrack(), and makeTrackImpl().

 {
   if (pidFromColl_) {
     return -1; //pid->at(tkID).Pdg();
   }
   if (mom_len_cut_ > 0.) { return (traj.Length() < mom_len_cut_ ? 2212 : 13); }
   return pid_def_;
 }

void trkmkr::KalmanFilterFitTrackMaker::initEvent ( const art::Event & e )

inlineoverridevirtual

initialize event: get collection of recob::MCSFitResult

Reimplemented from trkmkr::TrackMaker.

Definition at line 182 of file KalmanFilterFitTrackMaker_tool.cc.

References cont, contInputTag_, art::ProductRetriever::getValidHandle(), makeTrackImpl(), mcs, mcsInputTag_, momFromCombAndPid_, momFromMCSColl_, pidFromColl_, pidInputTag_, and lar::dump::vector().

     {
       if (momFromMCSColl_)
         mcs = e.getValidHandle<std::vector<recob::MCSFitResult>>(mcsInputTag_).product();
       if (momFromCombAndPid_) {
         cont = e.getValidHandle<std::vector<anab::CosmicTag>>(contInputTag_).product();
       }
       if (pidFromColl_) {
         pid = e.getValidHandle<std::vector<anab::ParticleID>>(pidInputTag_).product();
       }
     }

bool trkmkr::KalmanFilterFitTrackMaker::isFlipDirection	(	const recob::TrackTrajectory &	traj,
		const int	tkID
	)		const

decide whether to flip the direction or not

Definition at line 343 of file KalmanFilterFitTrackMaker_tool.cc.

References alwaysFlip_, dirFromVec_, trkmkr::KalmanFilterFitTrackMaker::Options::dirVec, and recob::TrackTrajectory::VertexDirection().

Referenced by makeTrack(), and makeTrackImpl().

 {
   if (alwaysFlip_) { return true; }
   else if (dirFromVec_) {
     auto tdir = traj.VertexDirection();
     if (tdir.Dot(dirVec) < 0.) return true;
   }
   return false;
 }

bool trkmkr::TrackMaker::makeTrack	(	const detinfo::DetectorPropertiesData &	detProp,
		const art::Ptr< recob::TrackTrajectory >	ttraj,
		const std::vector< art::Ptr< recob::Hit >> &	inHits,
		recob::Track &	outTrack,
		std::vector< art::Ptr< recob::Hit >> &	outHits,
		OptionalOutputs &	optionals
	)		const

inlineinherited

makeTrack functions with art::Ptr<recob::TrackTrajectory>; calls the purely virtual version with const recob::TrackTrajectory reference as argument.

Definition at line 207 of file TrackMaker.h.

References art::Ptr< T >::key(), track, recob::Track::Trajectory(), and lar::dump::vector().

     {
       return makeTrack(detProp, *ttraj, ttraj.key(), inHits, outTrack, outHits, optionals);
     }

bool trkmkr::KalmanFilterFitTrackMaker::makeTrack	(	const detinfo::DetectorPropertiesData &	detProp,
		const recob::TrackTrajectory &	traj,
		const int	tkID,
		const std::vector< art::Ptr< recob::Hit >> &	inHits,
		recob::Track &	outTrack,
		std::vector< art::Ptr< recob::Hit >> &	outHits,
		OptionalOutputs &	optionals
	)		const

inlineoverridevirtual

override of TrackMaker purely virtual function with recob::TrackTrajectory as argument

Implements trkmkr::TrackMaker.

Definition at line 207 of file KalmanFilterFitTrackMaker_tool.cc.

References getMomentum(), getParticleID(), isFlipDirection(), makeTrackImpl(), restoreInputPoints(), and lar::dump::vector().

     {
       return makeTrackImpl(detProp,
                            traj,
                            tkID,
                            inHits,
                            trkf::SMatrixSym55{},
                            trkf::SMatrixSym55{},
                            outTrack,
                            outHits,
                            optionals);
     }

bool trkmkr::TrackMaker::makeTrack	(	const detinfo::DetectorPropertiesData &	detProp,
		const art::Ptr< recob::Track >	track,
		const std::vector< art::Ptr< recob::Hit >> &	inHits,
		recob::Track &	outTrack,
		std::vector< art::Ptr< recob::Hit >> &	outHits,
		OptionalOutputs &	optionals
	)		const

inlineinherited

makeTrack functions with recob::Track as argument; calls the version with recob::TrackTrajectory.

Definition at line 230 of file TrackMaker.h.

     {
       return makeTrack(
         detProp, track->Trajectory(), track.key(), inHits, outTrack, outHits, optionals);
     }

bool trkmkr::KalmanFilterFitTrackMaker::makeTrackImpl	(	const detinfo::DetectorPropertiesData &	detProp,
		const recob::TrackTrajectory &	traj,
		const int	tkID,
		const std::vector< art::Ptr< recob::Hit >> &	inHits,
		const recob::tracking::SMatrixSym55 &	covVtx,
		const recob::tracking::SMatrixSym55 &	covEnd,
		recob::Track &	outTrack,
		std::vector< art::Ptr< recob::Hit >> &	outHits,
		OptionalOutputs &	optionals
	)		const

function that actually calls the fitter

Definition at line 268 of file KalmanFilterFitTrackMaker_tool.cc.

References trkf::TrackKalmanFitter::fitTrack(), getMomentum(), getParticleID(), isFlipDirection(), kalmanFitter, trkmkr::KalmanFilterFitTrackMaker::Options::keepInputTrajectoryPoints, p_, and restoreInputPoints().

Referenced by initEvent(), and makeTrack().

 {
   const int pid = getParticleID(traj, tkID);
   const bool flipDirection = isFlipDirection(traj, tkID);
   const double mom = getMomentum(traj, pid, flipDirection, tkID); // what about mom uncertainty?
   bool fitok = kalmanFitter.fitTrack(detProp,
                                      traj,
                                      tkID,
                                      covVtx,
                                      covEnd,
                                      inHits,
                                      mom,
                                      pid,
                                      flipDirection,
                                      outTrack,
                                      outHits,
                                      optionals);
   if (!fitok) return false;
   if (p_().options().keepInputTrajectoryPoints()) {
     restoreInputPoints(traj, inHits, outTrack, outHits, optionals);
   }
   return true;
 }

void trkmkr::KalmanFilterFitTrackMaker::restoreInputPoints	(	const recob::TrackTrajectory &	traj,
		const std::vector< art::Ptr< recob::Hit >> &	inHits,
		recob::Track &	outTrack,
		std::vector< art::Ptr< recob::Hit >> &	outHits,
		OptionalOutputs &	optionals
	)		const

restore the TrajectoryPoints in the Track to be the same as those in the input TrackTrajectory (but keep covariance matrices and chi2 from fit).

Definition at line 354 of file KalmanFilterFitTrackMaker_tool.cc.

References recob::Track::Chi2(), recob::Track::Covariances(), DEFINE_ART_CLASS_TOOL, recob::Track::FlagsAtPoint(), recob::TrajectoryPointFlags::fromHit(), recob::Trajectory::HasMomentum(), recob::Track::ID(), trkmkr::OptionalOutputs::isTrackFitInfosInit(), recob::TrackTrajectory::LocationAtPoint(), recob::TrajectoryPointFlags::mask(), recob::TrackTrajectory::MomentumVectorAtPoint(), recob::TrajectoryPointFlagTraits::NoPoint, recob::Track::NumberTrajectoryPoints(), recob::Track::ParticleId(), and util::flags::BitMask< Storage >::unset().

Referenced by makeTrack(), and makeTrackImpl().

 {
   if (optionals.isTrackFitInfosInit()) {
     throw cet::exception("KalmanFilterFitTrackMaker")
       << "Option keepInputTrajectoryPoints not compatible with "
          "doTrackFitHitInfo, please set doTrackFitHitInfo to false in the "
          "track producer.\n";
   }
   const auto np = outTrack.NumberTrajectoryPoints();
   trkmkr::TrackCreationBookKeeper tcbk(
     outHits, optionals, outTrack.ID(), outTrack.ParticleId(), traj.HasMomentum());
   //
   std::vector<unsigned int> flagsmap(np, -1);
   for (unsigned int i = 0; i < np; ++i)
     flagsmap[outTrack.FlagsAtPoint(i).fromHit()] = i;
   //
   for (unsigned int p = 0; p < np; ++p) {
     auto mask = outTrack.FlagsAtPoint(flagsmap[p]).mask();
     if (mask.isSet(recob::TrajectoryPointFlagTraits::NoPoint))
       mask.unset(recob::TrajectoryPointFlagTraits::NoPoint);
     tcbk.addPoint(traj.LocationAtPoint(p),
                   traj.MomentumVectorAtPoint(p),
                   inHits[p],
                   recob::TrajectoryPointFlags(p, mask),
                   0);
   }
   auto covs = outTrack.Covariances();
   tcbk.setTotChi2(outTrack.Chi2());
   outTrack = tcbk.finalizeTrack(std::move(covs.first), std::move(covs.second));
   //
 }