dt::Chain Class Reference

Collection of overlapping Cands forming a 2D track. More...

#include "/cvmfs/nova-development.opensciencegrid.org/novasoft/releases/N21-02-25/DiscreteTracker/Chain.h"

Public Member Functions
	Chain ()
	Chain (const Cand &c)
rb::Track	ToTrack () const
std::vector< rb::Track >	ToDebugTracks () const
void	Add (const Cand &c)
void	Add (const Cand &c, Direction dir)
void	AddFront (const Cand &c)
unsigned int	FirstPlane () const
unsigned int	LastPlane () const
unsigned int	ExtremalPlane (Direction dir) const
unsigned int	ExtremalHitPlane (Direction dir) const
unsigned int	ExtentPlane () const
unsigned int	NCands () const
const Cand &	FirstCand () const
const Cand &	LastCand () const
const Cand &	ExtremalCand (Direction dir) const
const Chunk &	FirstChunk () const
const Chunk &	LastChunk () const
const Segment &	FirstSeg () const
const Segment &	LastSeg () const
const Segment &	ExtremalSeg (Direction dir) const
art::PtrVector< rb::CellHit >	AllHits () const
std::vector< Chunk >	AllChunks () const
int	NHitPlanes () const
void	TrimEnds ()
	Remove Cands that add no actual hits. After building but before writing. More...
void	Truncate (unsigned int plane, Direction end)
void	Add (const Chain &c, Direction dir)
dt::View::ChunkMap	AsChunkMap () const

Protected Attributes
std::list< Cand >	fCands

Detailed Description

Collection of overlapping Cands forming a 2D track.

Definition at line 29 of file Chain.h.

Constructor & Destructor Documentation

dt::Chain::Chain ( )

inline

Definition at line 32 of file Chain.h.

References plot_validation_datamc::c, and dir.

{}

dt::Chain::Chain

(

const Cand &

c

)

Definition at line 23 of file Chain.cxx.

References Add().

  {
    Add(c);
  }

Member Function Documentation

void dt::Chain::Add

(

const Cand &

c

)

Definition at line 142 of file Chain.cxx.

References ana::assert(), fCands, dt::Cand::NChunks(), and dt::Cand::NSegs().

Referenced by Add(), Chain(), dt::DiscreteTracker::ExtendChain(), and dt::DiscreteTracker::SpliceChains().

  {
    assert(c.NChunks() > 0);
    assert(c.NSegs() > 1);
    fCands.push_back(c);
  }

void dt::Chain::Add	(	const Cand &	c,
		Direction	dir
	)

Definition at line 150 of file Chain.cxx.

References Add(), AddFront(), and dt::kDownstream.

  {
    if(dir == kDownstream)
      Add(c);
    else
      AddFront(c);
  }

void dt::Chain::Add ( const Chain &  c, Direction  dir  )

inline

Definition at line 71 of file Chain.h.

References fCands, and dt::kDownstream.

    {
      if(dir == kDownstream)
        fCands.insert(fCands.end(), c.fCands.begin(), c.fCands.end());
      else
        fCands.insert(fCands.begin(), c.fCands.begin(), c.fCands.end());
    }

void dt::Chain::AddFront

(

const Cand &

c

)

Definition at line 159 of file Chain.cxx.

References ana::assert(), fCands, dt::Cand::NChunks(), and dt::Cand::NSegs().

Referenced by Add().

  {
    assert(c.NChunks() > 0);
    assert(c.NSegs() > 1);
    fCands.push_front(c);
  }

std::vector< Chunk > dt::Chain::AllChunks

(

)

const

Definition at line 245 of file Chain.cxx.

References make_goodruns_defs::chunks, fCands, it, and runNovaSAM::ret.

Referenced by AsChunkMap(), dt::DiscreteTracker::CheckAndFixSparseChain(), ExtremalHitPlane(), NHitPlanes(), and Truncate().

  {
    // TODO: this doesn't take into account the possibility that chunks on the
    // same plane don't entirely correspond (due to dead channels?)
    std::map<unsigned int, Chunk> retMap;

    for(std::list<Cand>::const_iterator it = fCands.begin(); it != fCands.end(); ++it){
      const std::list<Chunk> chunks = it->AllChunks();
      for(std::list<Chunk>::const_iterator it = chunks.begin(); it != chunks.end(); ++it){
        retMap[it->Plane()] = *it;
      }
    }

    // Come out sorted in plane order
    std::vector<Chunk> ret;
    for(std::map<unsigned int, Chunk>::iterator it = retMap.begin(); it != retMap.end(); ++it)
      ret.push_back(it->second);
    return ret;
  }

art::PtrVector< rb::CellHit > dt::Chain::AllHits

(

)

const

Definition at line 220 of file Chain.cxx.

References rb::HitMap::Add(), rb::CellHit::Cell(), rb::HitMap::CellExists(), fCands, hits(), it, rb::CellHit::Plane(), art::PtrVector< T >::push_back(), runNovaSAM::ret, and art::PtrVector< T >::size().

Referenced by dt::DiscreteTracker::CheckAndFixSparseChain(), dt::DiscreteTracker::RecoDiscrete(), dt::DiscreteTracker::SpliceChains(), ToTrack(), TrimEnds(), and Truncate().

  {
    art::PtrVector<rb::CellHit> ret;

    // To eliminate duplicates
    rb::HitMap hmap;

    for(std::list<Cand>::const_iterator it = fCands.begin();
        it != fCands.end(); ++it){
      const art::PtrVector<rb::CellHit>& hits = it->AllHits();
      for(unsigned int hitIdx = 0; hitIdx < hits.size(); ++hitIdx){
        const art::Ptr<rb::CellHit> hit = hits[hitIdx];
        if(!hmap.CellExists(hit->Plane(), hit->Cell())){
          ret.push_back(hit);
        }
        hmap.Add(hit);
      }
    } // end for it

    // TODO
    //    assert(!ret.empty());
    return ret;
  }

dt::View::ChunkMap dt::Chain::AsChunkMap

(

)

const

Definition at line 337 of file Chain.cxx.

References AllChunks(), make_goodruns_defs::chunks, getGoodRuns4SAM::n, and runNovaSAM::ret.

Referenced by dt::DiscreteTracker::SpliceChains().

  {
    dt::View::ChunkMap ret;

    const std::vector<Chunk> chunks = AllChunks();
    for(unsigned int n = 0; n < chunks.size(); ++n)
      ret[chunks[n].Plane()].push_back(chunks[n]);

    return ret;
  }

unsigned int dt::Chain::ExtentPlane ( ) const

inline

Definition at line 43 of file Chain.h.

Referenced by dt::compareByLength(), and ToTrack().

{return LastPlane()-FirstPlane();}

const Cand& dt::Chain::ExtremalCand ( Direction  dir ) const

inline

Definition at line 47 of file Chain.h.

References dt::kUpstream.

Referenced by dt::DiscreteTracker::ExtendChain(), ExtremalPlane(), and dt::DiscreteTracker::SpliceChains().

    {
      if(dir == kUpstream) return FirstCand();
      return LastCand();
    }

unsigned int dt::Chain::ExtremalHitPlane

(

Direction

dir

)

const

Definition at line 185 of file Chain.cxx.

References AllChunks(), make_goodruns_defs::chunks, dt::kUpstream, and site_stats_from_log::reverse.

  {
    std::vector<Chunk> chunks = AllChunks();
    if(dir == kUpstream) std::reverse(chunks.begin(), chunks.end());

    while(!chunks.empty() && chunks.back().AllDead()) chunks.pop_back();

    return chunks.back().Plane();
  }

unsigned int dt::Chain::ExtremalPlane

(

Direction

dir

)

const

Definition at line 179 of file Chain.cxx.

References ExtremalCand(), and dt::Cand::ExtremalPlane().

Referenced by dt::DiscreteTracker::CheckAndFixSparseChain(), and dt::DiscreteTracker::SpliceChains().

  {
    return ExtremalCand(dir).ExtremalPlane(dir);
  }

const Segment& dt::Chain::ExtremalSeg ( Direction  dir ) const

inline

Definition at line 56 of file Chain.h.

References febshutoff_auto::end, dt::kUpstream, and NDAPDHVSetting::plane.

    {
      if(dir == kUpstream) return FirstSeg();
      return LastSeg();
    }

const Cand& dt::Chain::FirstCand ( ) const

inline

Definition at line 45 of file Chain.h.

Referenced by FirstChunk(), FirstPlane(), FirstSeg(), dt::DiscreteTracker::RecoDiscrete(), dt::DiscreteTracker::SpliceIsSane(), ToTrack(), and Truncate().

{return fCands.front();}

const Chunk & dt::Chain::FirstChunk

(

)

const

Definition at line 196 of file Chain.cxx.

References FirstCand(), and dt::Cand::FirstChunk().

  {
    return FirstCand().FirstChunk();
  }

unsigned int dt::Chain::FirstPlane

(

)

const

Definition at line 167 of file Chain.cxx.

References FirstCand(), dt::Cand::FirstChunk(), and dt::Chunk::Plane().

Referenced by dt::compareByStart(), dt::DiscreteTracker::SpliceChains(), dt::DiscreteTracker::SpliceIsSane(), and Truncate().

  {
    return FirstCand().FirstChunk().Plane();
  }

const Segment & dt::Chain::FirstSeg

(

)

const

Definition at line 208 of file Chain.cxx.

References FirstCand(), and dt::Cand::FirstSeg().

Referenced by Truncate().

  {
    return FirstCand().FirstSeg();
  }

const Cand& dt::Chain::LastCand ( ) const

inline

Definition at line 46 of file Chain.h.

Referenced by LastChunk(), LastPlane(), LastSeg(), dt::DiscreteTracker::RecoDiscrete(), dt::DiscreteTracker::SpliceIsSane(), and Truncate().

{return fCands.back();}

const Chunk & dt::Chain::LastChunk

(

)

const

Definition at line 202 of file Chain.cxx.

References LastCand(), and dt::Cand::LastChunk().

  {
    return LastCand().LastChunk();
  }

unsigned int dt::Chain::LastPlane

(

)

const

Definition at line 173 of file Chain.cxx.

References LastCand(), dt::Cand::LastChunk(), and dt::Chunk::Plane().

Referenced by dt::compareByEnd(), dt::DiscreteTracker::SpliceChains(), dt::DiscreteTracker::SpliceIsSane(), and Truncate().

  {
    return LastCand().LastChunk().Plane();
  }

const Segment & dt::Chain::LastSeg

(

)

const

Definition at line 214 of file Chain.cxx.

References LastCand(), and dt::Cand::LastSeg().

Referenced by Truncate().

  {
    return LastCand().LastSeg();
  }

unsigned int dt::Chain::NCands ( ) const

inline

Definition at line 44 of file Chain.h.

{return fCands.size();}

int dt::Chain::NHitPlanes

(

)

const

Definition at line 266 of file Chain.cxx.

References AllChunks(), make_goodruns_defs::chunks, getGoodRuns4SAM::n, and runNovaSAM::ret.

Referenced by dt::DiscreteTracker::SpliceChains().

  {
    std::vector<Chunk> chunks = AllChunks();
    int ret = 0;
    for(unsigned int n = 0; n < chunks.size(); ++n)
      if(!chunks[n].AllDead()) ++ret;

    return ret;
  }

std::vector< rb::Track > dt::Chain::ToDebugTracks

(

)

const

Definition at line 129 of file Chain.cxx.

References fCands, and runNovaSAM::ret.

  {
    std::vector<rb::Track> ret;

    for(std::list<Cand>::const_iterator candIt = fCands.begin();
        candIt != fCands.end(); ++candIt){
      ret.push_back(candIt->ToTrack());
    }

    return ret;
  }

rb::Track dt::Chain::ToTrack

(

)

const

Definition at line 29 of file Chain.cxx.

References AllHits(), rb::Track::AppendTrajectoryPoint(), ana::assert(), dt::Chunk::BottomHitCell(), make_associated_cosmic_defs::chunk, geo::GeometryBase::DetLength(), dir, dz, e, allTimeWatchdog::endl, ExtentPlane(), fCands, FirstCand(), dt::Cand::FirstChunk(), geom(), makeTrainCVSamples::int, std::max(), std::min(), getGoodRuns4SAM::n, PandAna.Demos.pi0_spectra::p0, plot_validation_datamc::p1, dt::Chunk::Plane(), geo::GeometryBase::Plane(), gen_hdf5record::size, fillBadChanDBTables::step, dt::Chunk::TopHitCell(), make_root_from_grid_output::tr, POTSpillRate::view, dt::Cand::View(), ana::weight, y1, submit_syst::y2, test::z, and Z.

Referenced by dt::DiscreteTracker::RecoDiscrete().

  {
    art::ServiceHandle<geo::Geometry> geom;

    // Special case for vertical cands (who'll otherwise estimate themselves
    // arbitrarily at a slope depending if they're marked "up" or "down").
    if(ExtentPlane() == 0 && AllHits().size() > 1){
      const Chunk& chunk = FirstCand().FirstChunk();
      double xyz0[3];
      geom->Plane(chunk.Plane())->Cell(chunk.TopHitCell())->GetCenter(xyz0);
      double xyz1[3];
      geom->Plane(chunk.Plane())->Cell(chunk.BottomHitCell())->GetCenter(xyz1);

      geo::View_t view = FirstCand().View();
      rb::Track tr(AllHits(), FirstCand().View(),
                   xyz0[view], xyz0[2],
                   xyz1[view]-xyz0[view], xyz1[2]-xyz0[2]);

      tr.AppendTrajectoryPoint(xyz1[view], xyz1[2]);
      return tr;
    }

    // The algorithm is to take the weighted average of the y-positions of all
    // relevant candidates at each z-position. (Convention is that the non-z
    // coordinate is called "y").

    // Estimate the start and end points of all the component cands upfront
    std::vector<TVector3> p0s, p1s;
    for(std::list<Cand>::const_iterator candIt = fCands.begin();
        candIt != fCands.end(); ++candIt){
      double z1, y1, z2, y2;
      candIt->EstimateStraightLine(z1, y1, z2, y2);
      p0s.push_back(TVector3(0, y1, z1));
      p1s.push_back(TVector3(0, y2, z2));
    }

    std::vector<TVector3> pts;

    // TODO, figure out where -1e10 Z's come from
    const double minZ = std::max(0., p0s[0].Z());
    const double maxZ = std::min(p1s.back().Z(), geom->DetLength());

    const double kZStep = 5; // cm
    const int stepMax = int((maxZ-minZ)/kZStep);

    for(int step = 0; step <= stepMax; ++step){
      const double z = minZ+(double(step)/stepMax)*(maxZ-minZ);

      TVector3 meanp;
      double totweight = 0;

      for(unsigned int pIdx = 0; pIdx < p0s.size(); ++pIdx){
        const TVector3& p0 = p0s[pIdx];
        const TVector3& p1 = p1s[pIdx];
        const double dz = p1.Z()-p0.Z();

        const TVector3 p = p0*((p1.Z()-z)/dz) + p1*((z-p0.Z())/dz);

        // Give more weight to cands near their centres
        double weight = (z-p0.Z())/dz;
        if(weight > 0.5) weight = 1-weight;
        weight += 1e-3; // Avoid zeros at very ends
        if(weight < 0) continue;

        meanp += p*weight;
        totweight += weight;
      }

      // This can happen when two cands are connected because they intersect at
      // a shallow angle inside a large dead region. There's probably a neater
      // way to illustrate this in the trajectory, but it's rare, so this will
      // do.
      if(totweight == 0) continue; // No cands cover this range

      meanp *= 1/totweight;

      pts.push_back(meanp);
    } // end for step

    // TODO
    //    assert(pts.size() >= 2);
    assert(!pts.empty());
    if(pts.size() == 1){
      mf::LogWarning("DiscreteTracker") << "Only one trajectory point. Fabricating a second one." << std::endl;
      pts.push_back(pts[0]+TVector3(0, 0, 1));
    }

    const TVector3 dir = pts[1]-pts[0];

    rb::Track tr(AllHits(),
                 fCands.begin()->View(),
                 pts[0].Y(), pts[0].Z(), dir.Y(), dir.Z());

    for(unsigned int n = 1; n < pts.size(); ++n)
      tr.AppendTrajectoryPoint(pts[n].Y(), pts[n].Z());

    return tr;
  }

void dt::Chain::TrimEnds

(

)

Remove Cands that add no actual hits. After building but before writing.

Definition at line 277 of file Chain.cxx.

References AllHits(), ana::assert(), fCands, and art::PtrVector< T >::size().

Referenced by dt::DiscreteTracker::RecoDiscrete(), and Truncate().

  {
    const unsigned int origSize = AllHits().size();

    assert(origSize > 0);

    while(true){
      const Cand cand = fCands.front();
      fCands.pop_front();
      if(fCands.empty() || AllHits().size() < origSize){
        fCands.push_front(cand);
        break;
      }
    }

    while(true){
      const Cand cand = fCands.back();
      fCands.pop_back();
      if(fCands.empty() || AllHits().size() < origSize){
        fCands.push_back(cand);
        break;
      }
    }

    fCands.front().TrimFront();
    fCands.back().TrimBack();
  }

void dt::Chain::Truncate	(	unsigned int	plane,
		Direction	end
	)

Definition at line 306 of file Chain.cxx.

References AllChunks(), AllHits(), fCands, FirstCand(), FirstPlane(), FirstSeg(), dt::kUpstream, LastCand(), LastPlane(), LastSeg(), and TrimEnds().

Referenced by dt::DiscreteTracker::CheckAndFixSparseChain().

  {
    if(end == kUpstream){
      while(FirstPlane() < plane){
        fCands.begin()->PopFirstChunk();
        if(FirstCand().AllChunks().empty()) fCands.pop_front();
      }
      while(FirstSeg().plane < int(plane)){
        if(fCands.begin()->NSegs() > 1)
          fCands.begin()->PopFirstSeg();
        else
          fCands.pop_front();
      }
    }
    else{
      while(LastPlane() > plane){
        fCands.rbegin()->PopLastChunk();
        if(LastCand().AllChunks().empty()) fCands.pop_back();
      }
      while(LastSeg().plane > int(plane)){
        if(fCands.rbegin()->NSegs() > 1)
          fCands.rbegin()->PopLastSeg();
        else
          fCands.pop_back();
      }
    }

    if(!AllHits().empty()) TrimEnds();
  }

Member Data Documentation

std::list<Cand> dt::Chain::fCands

protected

Definition at line 80 of file Chain.h.

Referenced by Add(), AddFront(), AllChunks(), AllHits(), ToDebugTracks(), ToTrack(), TrimEnds(), and Truncate().

---

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

• /cvmfs/nova-development.opensciencegrid.org/novasoft/releases/N21-02-25/DiscreteTracker/Chain.h
• /cvmfs/nova-development.opensciencegrid.org/novasoft/releases/N21-02-25/DiscreteTracker/Chain.cxx