trkf::KGTrack Class Reference

#include <KGTrack.h>

Public Member Functions
	KGTrack (int prefplane)
int	getPrefPlane () const
const std::multimap< double, KHitTrack > &	getTrackMap () const
	KHitTrack collection, indexed by path distance. More...
size_t	numHits () const
	Number of measurements in track. More...
const KHitTrack &	startTrack () const
	Track at start point. More...
const KHitTrack &	endTrack () const
	Track at end point. More...
bool	isValid () const
	Validity flag. More...
std::multimap< double, KHitTrack > &	getTrackMap ()
	Modifiable KHitTrack collection, indexed by path distance. More...
KHitTrack &	startTrack ()
	Modifiable track at start point. More...
KHitTrack &	endTrack ()
	Modifiable track at end point. More...
void	addTrack (const KHitTrack &trh)
	Add track. More...
void	recalibrate ()
	Recalibrate track map. More...
void	clear ()
	Clear track collection. More...
void	fillTrack (detinfo::DetectorPropertiesData const &detProp, recob::Track &track, int id) const
	Fill a recob::Track. More...
void	fillHits (art::PtrVector< recob::Hit > &hits, std::vector< unsigned int > &hittpindex) const
	Fill a PtrVector of Hits. More...
const std::multimap< double, KHitTrack >	TrackMap () const
std::ostream &	Print (std::ostream &out) const
	Printout. More...

Private Attributes
int	fPrefPlane
	Preferred plane. More...
std::multimap< double, KHitTrack >	fTrackMap
	KHitTrack collection, indexed by path distance. More...

Detailed Description

Definition at line 47 of file KGTrack.h.

Constructor & Destructor Documentation

trkf::KGTrack::KGTrack

(

int

prefplane

)

Definition at line 27 of file KGTrack.cxx.

: fPrefPlane(prefplane) {}

Member Function Documentation

void trkf::KGTrack::addTrack

(

const KHitTrack &

trh

)

Add track.

Definition at line 81 of file KGTrack.cxx.

  {
    if (!trh.isValid()) throw cet::exception("KGTrack") << "Adding invalid track to KGTrack.\n";
    fTrackMap.insert(std::make_pair(trh.getPath() + trh.getHit()->getPredDistance(), trh));
  }

void trkf::KGTrack::clear ( )

inline

Clear track collection.

Definition at line 107 of file KGTrack.h.

    {
      fTrackMap.clear();
    }

const KHitTrack & trkf::KGTrack::endTrack

(

)

const

Track at end point.

Throw exception if track is not valid.

Definition at line 42 of file KGTrack.cxx.

  {
    /// Throw exception if track is not valid.

    if (!isValid()) throw cet::exception("KGTrack") << "Ending track is not valid.\n";

    // Return track.

    return (*fTrackMap.rbegin()).second;
  }

KHitTrack & trkf::KGTrack::endTrack

(

)

Modifiable track at end point.

Throw exception if track is not valid.

Definition at line 68 of file KGTrack.cxx.

  {
    /// Throw exception if track is not valid.

    if (!isValid()) throw cet::exception("KGTrack") << "Ending track is not valid.\n";

    // Return track.

    return (*fTrackMap.rbegin()).second;
  }

void trkf::KGTrack::fillHits	(	art::PtrVector< recob::Hit > &	hits,
		std::vector< unsigned int > &	hittpindex
	)		const

Fill a PtrVector of Hits.

Fill a PtrVector of Hits.

Arguments:

hits - Hit vector to fill.

Definition at line 224 of file KGTrack.cxx.

  {
    hits.reserve(hits.size() + fTrackMap.size());

    // Loop over KHitTracks and fill hits belonging to this track.

    unsigned int counter = 0; //Index of corresponding trajectory point
    for (std::multimap<double, KHitTrack>::const_iterator it = fTrackMap.begin();
         it != fTrackMap.end();
         ++it) {
      const KHitTrack& track = (*it).second;
      ++counter;
      // Extrack Hit from track.
      const std::shared_ptr<const KHitBase>& hit = track.getHit();
      if (const KHitWireX* phit = dynamic_cast<const KHitWireX*>(&*hit)) {
        const art::Ptr<recob::Hit> prhit = phit->getHit();
        if (!prhit.isNull()) {
          hits.push_back(prhit);
          hittpindex.push_back(counter - 1);
        }
      }
      else if (const KHitWireLine* phit = dynamic_cast<const KHitWireLine*>(&*hit)) {
        const art::Ptr<recob::Hit> prhit = phit->getHit();
        if (!prhit.isNull()) {
          hits.push_back(prhit);
          hittpindex.push_back(counter - 1);
        }
      }
    }
  }

void trkf::KGTrack::fillTrack	(	detinfo::DetectorPropertiesData const &	detProp,
		recob::Track &	track,
		int	id
	)		const

Fill a recob::Track.

Fill a recob::Track.

Arguments:

track - Track to fill.

Definition at line 126 of file KGTrack.cxx.

  {

    // Make propagator for propating to standard track surface.

    PropXYZPlane prop(detProp, 0., false);

    // Fill collections of trajectory points and direction vectors.

    std::vector<recob::tracking::Point_t> xyz;
    std::vector<recob::tracking::Vector_t> pxpypz;
    std::vector<recob::tracking::SMatrixSym55> cov;
    std::vector<recob::TrajectoryPointFlags> outFlags;

    xyz.reserve(fTrackMap.size());
    pxpypz.reserve(fTrackMap.size());
    outFlags.reserve(fTrackMap.size());

    // Loop over KHitTracks.

    int ndof = 0;
    float totChi2 = 0.;
    unsigned int n = 0;
    for (std::multimap<double, KHitTrack>::const_iterator itr = fTrackMap.begin();
         itr != fTrackMap.end();
         ++itr, ++n) {
      const KHitTrack& trh = (*itr).second;

      // Get position.

      double pos[3];
      trh.getPosition(pos);
      xyz.push_back({pos[0], pos[1], pos[2]});

      // Get momentum vector.
      // Fill direction unit vector and momentum.

      double mom[3];
      trh.getMomentum(mom);
      double p = std::sqrt(mom[0] * mom[0] + mom[1] * mom[1] + mom[2] * mom[2]);
      if (p == 0.) throw cet::exception("KGTrack") << __func__ << ": null momentum\n";
      pxpypz.push_back({mom[0], mom[1], mom[2]});

      ndof += 1;
      totChi2 += trh.getChisq();
      outFlags.emplace_back(n, recob::TrajectoryPointFlags::makeMask());

      // Fill error matrix.

      recob::tracking::SMatrixSym55 covar;

      // Construct surface perpendicular to track momentun, and
      // propagate track to that surface (zero distance).

      const std::shared_ptr<const Surface> psurf(
        new SurfXYZPlane(pos[0], pos[1], pos[2], mom[0], mom[1], mom[2]));
      KETrack tre(trh);
      std::optional<double> dist = prop.err_prop(tre, psurf, Propagator::UNKNOWN, false);
      if (!dist) throw cet::exception("KGTrack") << __func__ << ": error propagation failed\n";
      for (int i = 0; i < 5; ++i) {
        for (int j = 0; j < 5; ++j)
          covar(i, j) = tre.getError()(i, j);
      }

      // Only save first and last error matrix.

      if (cov.size() < 2)
        cov.push_back(covar);
      else
        cov.back() = covar;
    }

    // Fill track.

    ndof = ndof - 4; //fit measures 4 parameters: position and direction on plane
    if (xyz.size() >= 2) {
      track = recob::Track(std::move(xyz),
                           std::move(pxpypz),
                           std::move(outFlags),
                           true,
                           this->startTrack().PdgCode(),
                           totChi2,
                           ndof,
                           std::move(cov.front()),
                           std::move(cov.back()),
                           id);
    }
  }

int trkf::KGTrack::getPrefPlane ( ) const

inline

Definition at line 52 of file KGTrack.h.

    {
      return fPrefPlane;
    }

const std::multimap<double, KHitTrack>& trkf::KGTrack::getTrackMap ( ) const

inline

KHitTrack collection, indexed by path distance.

Definition at line 59 of file KGTrack.h.

    {
      return fTrackMap;
    }

std::multimap<double, KHitTrack>& trkf::KGTrack::getTrackMap ( )

inline

Modifiable KHitTrack collection, indexed by path distance.

Definition at line 88 of file KGTrack.h.

    {
      return fTrackMap;
    }

bool trkf::KGTrack::isValid ( ) const

inline

Validity flag.

Definition at line 79 of file KGTrack.h.

    {
      return fTrackMap.size() > 0;
    }

size_t trkf::KGTrack::numHits ( ) const

inline

Number of measurements in track.

Definition at line 66 of file KGTrack.h.

    {
      return fTrackMap.size();
    }

std::ostream & trkf::KGTrack::Print

(

std::ostream &

out

)

const

Printout.

Printout

Definition at line 259 of file KGTrack.cxx.

  {

    int n = 0;

    double oldxyz[3] = {0., 0., 0.};
    double len = 0.;
    bool first = true;
    for (auto const& ele : fTrackMap) {
      double s = ele.first;
      const KHitTrack& trh = ele.second;
      double xyz[3];
      double mom[3];
      trh.getPosition(xyz);
      trh.getMomentum(mom);
      double tmom = std::sqrt(mom[0] * mom[0] + mom[1] * mom[1] + mom[2] * mom[2]);
      if (tmom != 0.) {
        mom[0] /= tmom;
        mom[1] /= tmom;
        mom[2] /= tmom;
      }
      if (!first) {
        double dx = xyz[0] - oldxyz[0];
        double dy = xyz[1] - oldxyz[1];
        double dz = xyz[2] - oldxyz[2];
        len += std::sqrt(dx * dx + dy * dy + dz * dz);
      }
      const KHitBase& hit = *(trh.getHit());
      int plane = hit.getMeasPlane();
      std::ios_base::fmtflags f = out.flags();
      out << "State " << std::setw(4) << n << ", path=" << std::setw(8) << std::fixed
          << std::setprecision(2) << s << ", length=" << std::setw(8) << len
          << ", x=" << std::setw(8) << xyz[0] << ", y=" << std::setw(8) << xyz[1]
          << ", z=" << std::setw(8) << xyz[2] << ", dx=" << std::setw(8) << mom[0]
          << ", dy=" << std::setw(8) << mom[1] << ", dz=" << std::setw(8) << mom[2]
          << ", plane=" << std::setw(1) << plane << "\n";
      out.flags(f);

      oldxyz[0] = xyz[0];
      oldxyz[1] = xyz[1];
      oldxyz[2] = xyz[2];

      ++n;
      first = false;
    }
    return out;
  }

void trkf::KGTrack::recalibrate

(

)

Recalibrate track map.

Recalibrate track map.

Loop over contents of track map. Copy each KHitTrack into a new multimap track map. Offset the distance stored in the KHitTracks such that the minimum distance is zero. Also update multimap keys to agree with distance stored in track.

Definition at line 94 of file KGTrack.cxx.

  {
    std::multimap<double, KHitTrack> newmap;

    // Loop over old track map.

    bool first = true;
    double s0 = 0.;
    for (std::multimap<double, KHitTrack>::iterator i = fTrackMap.begin(); i != fTrackMap.end();
         ++i) {
      KHitTrack& trh = (*i).second;
      if (first) {
        first = false;
        s0 = trh.getPath();
      }
      double s = trh.getPath() - s0;
      trh.setPath(s);
      newmap.insert(std::make_pair(s, trh));
    }

    // Update data member track map.

    fTrackMap.swap(newmap);
  }

const KHitTrack & trkf::KGTrack::startTrack

(

)

const

Track at start point.

Definition at line 31 of file KGTrack.cxx.

  {
    if (!isValid()) throw cet::exception("KGTrack") << "Starting track is not valid.\n";

    // Return track.

    return (*fTrackMap.begin()).second;
  }

KHitTrack & trkf::KGTrack::startTrack

(

)

Modifiable track at start point.

Throw exception if track is not valid.

Definition at line 55 of file KGTrack.cxx.

  {
    /// Throw exception if track is not valid.

    if (!isValid()) throw cet::exception("KGTrack") << "Starting track is not valid.\n";

    // Return track.

    return (*fTrackMap.begin()).second;
  }

const std::multimap<double, KHitTrack> trkf::KGTrack::TrackMap ( ) const

inline

Definition at line 123 of file KGTrack.h.

    {
      return fTrackMap;
    }

Member Data Documentation

int trkf::KGTrack::fPrefPlane

private

Preferred plane.

Definition at line 133 of file KGTrack.h.

std::multimap<double, KHitTrack> trkf::KGTrack::fTrackMap

private

KHitTrack collection, indexed by path distance.

Definition at line 136 of file KGTrack.h.

---

The documentation for this class was generated from the following files:

• KGTrack.h
• KGTrack.cxx