pma::VtxCandidate Class Reference

#include "PmaVtxCandidate.h"

Public Member Functions
	VtxCandidate (double segMinLength=0.5)
bool	Has (pma::Track3D *trk) const
bool	Has (const VtxCandidate &other) const
bool	IsAttached (pma::Track3D *trk) const
bool	IsAttached (const VtxCandidate &other) const
bool	HasLoops () const
bool	Add (const pma::TrkCandidate &trk)
double	ComputeMse2D ()
double	Test (const VtxCandidate &other) const
double	MaxAngle (double minLength=0.0) const
size_t	Size () const
size_t	Size (double minLength) const
bool	MergeWith (const VtxCandidate &other)
double	Compute ()
bool	JoinTracks (detinfo::DetectorPropertiesData const &detProp, pma::TrkCandidateColl &tracks, pma::TrkCandidateColl &src)
const TVector3 &	Center () const
double	Mse () const
double	Mse2D () const
std::pair< pma::Track3D *, size_t >	Track (size_t i) const

Static Public Attributes
static constexpr double	kMaxDistToTrack {4.0}
static constexpr double	kMinDistToNode {2.0}

Private Member Functions
bool	has (const std::vector< int > &v, int id) const

Private Attributes
bool	tracksJoined
double	fSegMinLength
double	fMse
double	fMse2D
std::vector< std::pair< pma::TrkCandidate, size_t > >	fAssigned
TVector3	fCenter
TVector3	fErr

Detailed Description

Definition at line 24 of file PmaVtxCandidate.h.

Constructor & Destructor Documentation

pma::VtxCandidate::VtxCandidate ( double  segMinLength = 0.5 )

inline

Definition at line 29 of file PmaVtxCandidate.h.

References Add(), ComputeMse2D(), Has(), HasLoops(), IsAttached(), MaxAngle(), fhicl::other, and Test().

    : tracksJoined(false)
    , fSegMinLength(segMinLength)
    , fMse(0.0)
    , fMse2D(0.0)
    , fCenter(0., 0., 0.)
    , fErr(0., 0., 0.)
  {}

Member Function Documentation

bool pma::VtxCandidate::Add

(

const pma::TrkCandidate &

trk

)

Definition at line 86 of file PmaVtxCandidate.cxx.

References Compute(), ComputeMse2D(), d, fAssigned, fCenter, fMse, fMse2D, fSegMinLength, pma::Segment3D::GetDistance2To(), IsAttached(), kMaxDistToTrack, pma::Element3D::Length(), n, pma::Track3D::NextSegment(), pma::Track3D::Nodes(), and pma::TrkCandidate::Track().

Referenced by pma::PMAlgVertexing::firstPassCandidates(), pma::PMAlgVertexing::secondPassCandidates(), and VtxCandidate().

{
  if (IsAttached(trk.Track())) return false;

  fAssigned.emplace_back(trk, 0);

  double d, d_best;
  double mse, min_mse = kMaxDistToTrack * kMaxDistToTrack;
  if (fAssigned.size() > 2) {
    size_t n_best = 0;
    d_best = kMaxDistToTrack;
    for (size_t n = 0; n < trk.Track()->Nodes().size() - 1; n++) {
      pma::Segment3D* seg = trk.Track()->NextSegment(trk.Track()->Nodes()[n]);
      if (seg->Length() < fSegMinLength) continue;

      fAssigned.back().second = n;

      mse = Compute();
      if (mse < min_mse) {
        d = sqrt(seg->GetDistance2To(fCenter));
        if (d < d_best) {
          min_mse = mse;
          n_best = n;
          d_best = d;
        }
      }
    }

    if (d_best < kMaxDistToTrack) {
      fAssigned.back().second = n_best;
      fMse = Compute();
      fMse2D = ComputeMse2D();
      return true;
    }
    else {
      fAssigned.pop_back();
      fMse = Compute();
      fMse2D = ComputeMse2D();
      return false;
    }
  }
  else if (fAssigned.size() == 2) {
    pma::Track3D* p0 = fAssigned.front().first.Track();

    size_t n_best = 0, m_best = 0;
    d_best = kMaxDistToTrack;

    double lm, ln, l_best = 0;
    for (size_t m = 0; m < p0->Nodes().size() - 1; m++) {
      pma::Segment3D* seg_m = p0->NextSegment(p0->Nodes()[m]);
      lm = seg_m->Length();
      if (lm < fSegMinLength) continue;

      fAssigned.front().second = m;

      for (size_t n = 0; n < trk.Track()->Nodes().size() - 1; n++) {
        pma::Segment3D* seg_n = trk.Track()->NextSegment(trk.Track()->Nodes()[n]);
        ln = seg_n->Length();
        if (ln < fSegMinLength) continue;

        fAssigned.back().second = n;

        mse = Compute(); // std::cout << mse << std::endl;

        d = sqrt(ComputeMse2D());

        if (d < d_best) {
          double d_dist = (d_best - d) / d_best;
          if (lm + ln > 0.8 * d_dist * l_best) // take closer if not much shorter
          {
            min_mse = mse;
            n_best = n;
            m_best = m;
            d_best = d;
            l_best = lm + ln;
          }
        }
      }
    }

    if (d_best < kMaxDistToTrack) {
      fAssigned.front().second = m_best;
      fAssigned.back().second = n_best;
      fMse = Compute();

      fMse2D = ComputeMse2D();

      return true;
    }
    else {
      fAssigned.pop_back();
      fCenter.SetXYZ(0., 0., 0.);
      fMse = 0;
      fMse2D = 0;
      return false;
    }
  }
  else {
    for (size_t n = 0; n < trk.Track()->Nodes().size() - 1; n++) {
      pma::Segment3D* seg = trk.Track()->NextSegment(trk.Track()->Nodes()[n]);
      if (seg->Length() >= fSegMinLength) { return true; }
    }
    fAssigned.pop_back();
    fCenter.SetXYZ(0., 0., 0.);
    fMse = 0;
    fMse2D = 0;
    return false;
  }
}

const TVector3& pma::VtxCandidate::Center ( ) const

inline

Definition at line 67 of file PmaVtxCandidate.h.

References fCenter.

{ return fCenter; }

double pma::VtxCandidate::Compute

(

)

Definition at line 330 of file PmaVtxCandidate.cxx.

References fAssigned, fCenter, fErr, fSegMinLength, pma::GetProjectionToSegment(), pma::Element3D::Length(), pma::SortedObjectBase::Next(), pma::Track3D::NextSegment(), pma::Track3D::Nodes(), pma::Node3D::Point3D(), pma::SolveLeastSquares3D(), and w.

Referenced by Add(), MergeWith(), and Size().

{
  std::vector<pma::Segment3D*> segments;
  std::vector<std::pair<TVector3, TVector3>> lines;
  std::vector<double> weights;
  for (const auto& v : fAssigned) {
    pma::Track3D* trk = v.first.Track();
    int vIdx = v.second;

    pma::Node3D* vtx1 = trk->Nodes()[vIdx];
    pma::Segment3D* seg = trk->NextSegment(vtx1);
    double segLength = seg->Length();
    if (segLength >= fSegMinLength) {
      pma::Node3D* vtx2 = static_cast<pma::Node3D*>(seg->Next());

      std::pair<TVector3, TVector3> endpoints(vtx1->Point3D(), vtx2->Point3D());
      double dy = endpoints.first.Y() - endpoints.second.Y();
      double fy_norm = asin(fabs(dy) / segLength) / (0.5 * TMath::Pi());
      double w = 1.0 - pow(fy_norm - 1.0, 12);
      if (w < 0.3) w = 0.3;

      lines.push_back(endpoints);
      segments.push_back(seg);
      weights.push_back(w);
    }
  }

  fCenter.SetXYZ(0., 0., 0.);
  fErr.SetXYZ(0., 0., 0.);

  TVector3 result;
  double resultMse = pma::SolveLeastSquares3D(lines, result);
  if (resultMse < 0.0) {
    mf::LogWarning("pma::VtxCandidate") << "Cannot compute crossing point.";
    return 1.0E+6;
  }

  TVector3 pproj;
  //double dx, dy, dz
  double wsum = 0.0;
  for (size_t s = 0; s < segments.size(); s++) {
    pma::Node3D* vprev = static_cast<pma::Node3D*>(segments[s]->Prev());
    pma::Node3D* vnext = static_cast<pma::Node3D*>(segments[s]->Next());

    pproj = pma::GetProjectionToSegment(result, vprev->Point3D(), vnext->Point3D());

    //dx = weights[s] * (result.X() - pproj.X());
    //dy = result.Y() - pproj.Y();
    //dz = result.Z() - pproj.Z();

    fErr[0] += weights[s] * weights[s];
    fErr[1] += 1.0;
    fErr[2] += 1.0;

    fCenter[0] += weights[s] * pproj.X();
    fCenter[1] += pproj.Y();
    fCenter[2] += pproj.Z();
    wsum += weights[s];
  }
  fCenter[0] /= wsum;
  fCenter[1] /= segments.size();
  fCenter[2] /= segments.size();

  fErr *= 1.0 / segments.size();
  fErr[0] = sqrt(fErr[0]);
  fErr[1] = sqrt(fErr[1]);
  fErr[2] = sqrt(fErr[2]);

  return resultMse;
}

double pma::VtxCandidate::ComputeMse2D

(

)

Definition at line 196 of file PmaVtxCandidate.cxx.

References fAssigned, fCenter, pma::Segment3D::GetDistance2To(), pma::GetProjectionToPlane(), geo::kU, geo::kV, geo::kZ, pma::Track3D::NextSegment(), pma::Track3D::Nodes(), and geo::GeometryCore::TPC().

Referenced by Add(), MergeWith(), and VtxCandidate().

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

  double mse = 0.0;
  TVector2 center2d;
  for (const auto& t : fAssigned) {
    pma::Track3D* trk = t.first.Track();
    pma::Segment3D* seg = trk->NextSegment(trk->Nodes()[t.second]);

    int tpc = trk->Nodes()[t.second]->TPC();
    int cryo = trk->Nodes()[t.second]->Cryo();

    size_t k = 0;
    double m = 0.0;
    auto const& tpcgeom = geom->TPC(geo::TPCID(cryo, tpc));
    if (tpcgeom.HasPlane(geo::kU)) {
      center2d = GetProjectionToPlane(fCenter, geo::kU, tpc, cryo);
      m += seg->GetDistance2To(center2d, geo::kU);
      k++;
    }
    if (tpcgeom.HasPlane(geo::kV)) {
      center2d = GetProjectionToPlane(fCenter, geo::kV, tpc, cryo);
      m += seg->GetDistance2To(center2d, geo::kV);
      k++;
    }
    if (tpcgeom.HasPlane(geo::kZ)) {
      center2d = GetProjectionToPlane(fCenter, geo::kZ, tpc, cryo);
      m += seg->GetDistance2To(center2d, geo::kZ);
      k++;
    }
    mse += m / (double)k;
  }

  return mse / fAssigned.size();
}

bool pma::VtxCandidate::Has

(

pma::Track3D *

trk

)

const

Definition at line 26 of file PmaVtxCandidate.cxx.

References fAssigned.

Referenced by Has(), MergeWith(), and VtxCandidate().

{
  for (const auto& t : fAssigned)
    if (trk == t.first.Track()) return true;
  return false;
}

bool pma::VtxCandidate::Has

(

const VtxCandidate &

other

)

const

Definition at line 33 of file PmaVtxCandidate.cxx.

References fAssigned, and Has().

{
  for (const auto& t : other.fAssigned)
    if (!Has(t.first.Track())) return false;
  return true;
}

bool pma::VtxCandidate::has ( const std::vector< int > &  v, int  id  ) const

inlineprivate

Definition at line 77 of file PmaVtxCandidate.h.

Referenced by JoinTracks().

  {
    for (auto c : v)
      if (c == id) return true;
    return false;
  }

bool pma::VtxCandidate::HasLoops

(

)

const

Definition at line 61 of file PmaVtxCandidate.cxx.

References fAssigned, and pma::Track3D::IsAttachedTo().

Referenced by VtxCandidate().

{
  for (size_t t = 0; t < fAssigned.size(); t++) {
    pma::Track3D const* trk_t = fAssigned[t].first.Track()->GetRoot();
    if (!trk_t) throw cet::exception("pma::VtxCandidate") << "Broken track.";

    for (size_t u = 0; u < fAssigned.size(); u++)
      if (t != u) {
        pma::Track3D const* trk_u = fAssigned[u].first.Track()->GetRoot();
        if (!trk_u) throw cet::exception("pma::VtxCandidate") << "Broken track.";

        if (trk_t->IsAttachedTo(trk_u)) return true;
      }
  }
  return false;
}

bool pma::VtxCandidate::IsAttached

(

pma::Track3D *

trk

)

const

Definition at line 40 of file PmaVtxCandidate.cxx.

References fAssigned, pma::Track3D::GetRoot(), and pma::Track3D::IsAttachedTo().

Referenced by Add(), IsAttached(), MergeWith(), and VtxCandidate().

{
  pma::Track3D const* rootTrk = trk->GetRoot();
  if (!rootTrk) throw cet::exception("pma::VtxCandidate") << "Broken track.";

  for (const auto& t : fAssigned) {
    pma::Track3D const* rootAssn = t.first.Track()->GetRoot();
    if (!rootAssn) throw cet::exception("pma::VtxCandidate") << "Broken track.";

    if (rootTrk->IsAttachedTo(rootAssn)) return true;
  }
  return false;
}

bool pma::VtxCandidate::IsAttached

(

const VtxCandidate &

other

)

const

Definition at line 54 of file PmaVtxCandidate.cxx.

References fAssigned, and IsAttached().

{
  for (const auto& t : other.fAssigned)
    if (IsAttached(t.first.Track())) return true;
  return false;
}

bool pma::VtxCandidate::JoinTracks	(	detinfo::DetectorPropertiesData const &	detProp,
		pma::TrkCandidateColl &	tracks,
		pma::TrkCandidateColl &	src
	)

Definition at line 401 of file PmaVtxCandidate.cxx.

References pma::Track3D::AttachBackTo(), pma::Track3D::AttachTo(), pma::Track3D::CanFlip(), pma::Dist2(), pma::TrkCandidateColl::erase_at(), f, fAssigned, fCenter, pma::Track3D::Flip(), fMse, fMse2D, pma::Node3D::GetBranches(), pma::Track3D::GetBranches(), pma::TrkCandidateColl::getCandidateIndex(), pma::GetProjectionToSegment(), pma::Track3D::GetRoot(), pma::GetSegmentProjVector(), pma::TrkCandidateColl::getTreeCopy(), has(), pma::Track3D::InsertNode(), kMinDistToNode, pma::Track3D::MakeProjection(), pma::SortedBranchBase::Next(), pma::SortedBranchBase::NextCount(), pma::Track3D::NextSegment(), pma::Track3D::Nodes(), pma::Segment3D::Parent(), pma::Node3D::Point3D(), pma::SortedObjectBase::Prev(), proj, pma::TrkCandidateColl::push_back(), pma::Node3D::SetPoint3D(), pma::TrkCandidateColl::setTreeIds(), pma::TrkCandidateColl::size(), pma::Track3D::size(), pma::Track3D::Split(), pma::TrkCandidateColl::tracks(), tracksJoined, and pma::Track3D::TuneFullTree().

Referenced by Size().

{
  if (tracksJoined) {
    mf::LogError("pma::VtxCandidate") << "Tracks already attached to the vertex.";
    return false;
  }
  tracksJoined = true;

  mf::LogVerbatim("pma::VtxCandidate")
    << "JoinTracks (" << fAssigned.size() << ") at:"
    << " vx:" << fCenter.X() << " vy:" << fCenter.Y() << " vz:" << fCenter.Z();

  for (auto const& c : fAssigned) {
    size_t t = 0;
    while (t < src.size()) {
      if (c.first.Track() == src[t].Track()) { // move from "src" to "tracks"
        tracks.push_back(src[t]);
        src.erase_at(t);
        break;
      }
      else
        t++;
    }
  }
  // all involved tracks are in the same container, so:
  tracks.setTreeIds();
  for (auto& c : fAssigned) // update ids
    for (auto const& t : tracks.tracks())
      if (c.first.Track() == t.Track()) {
        c.first.SetTreeId(t.TreeId());
        break;
      }

  // backup in case of fitting problems
  std::vector<int> treeIds;
  pma::TrkCandidateColl backupTracks;

  pma::Node3D* vtxCenter = 0;
  bool hasInnerCenter = false;
  size_t nOK = 0;
  for (size_t i = 0; i < fAssigned.size(); i++) {
    mf::LogVerbatim("pma::VtxCandidate") << "----------> track #" << i;

    pma::Track3D* trk = fAssigned[i].first.Track();
    int key = fAssigned[i].first.Key();
    int tid = fAssigned[i].first.TreeId();
    size_t idx = fAssigned[i].second;

    mf::LogVerbatim("pma::VtxCandidate") << "  track tid:" << tid << ", size:" << trk->size()
                                         << " (nodes:" << trk->Nodes().size() << ")";

    if (!has(treeIds, tid)) // backup in case of fitting problems
    {
      treeIds.push_back(tid);
      int rootIdx = tracks.getCandidateIndex(trk->GetRoot());
      if (rootIdx >= 0)
        tracks.getTreeCopy(backupTracks, rootIdx);
      else
        mf::LogError("pma::VtxCandidate") << "Root of the tree not found in tracks collection.";
    }

    TVector3 p0(trk->Nodes()[idx]->Point3D());
    TVector3 p1(trk->Nodes()[idx + 1]->Point3D());

    int tpc0 = trk->Nodes()[idx]->TPC();
    int tpc1 = trk->Nodes()[idx + 1]->TPC();

    int cryo0 = trk->Nodes()[idx]->Cryo();
    int cryo1 = trk->Nodes()[idx + 1]->Cryo();

    double d0 = sqrt(pma::Dist2(p0, fCenter));
    double d1 = sqrt(pma::Dist2(p1, fCenter));
    double ds = sqrt(pma::Dist2(p0, p1));
    double f = pma::GetSegmentProjVector(fCenter, p0, p1);
    TVector3 proj = pma::GetProjectionToSegment(fCenter, p0, p1);

    if ((idx == 0) && (f * ds <= kMinDistToNode)) {
      if (i == 0) {
        mf::LogVerbatim("pma::VtxCandidate") << "  new at front";
        vtxCenter = trk->Nodes().front();
        vtxCenter->SetPoint3D(fCenter);
        nOK++;
      }
      else {
        mf::LogVerbatim("pma::VtxCandidate") << "  front to center";
        if (trk->AttachTo(vtxCenter)) nOK++;
      }
    }
    else if ((idx + 2 == trk->Nodes().size()) && ((1.0 - f) * ds <= kMinDistToNode)) {
      if (i == 0) {
        if (trk->CanFlip()) {
          mf::LogVerbatim("pma::VtxCandidate") << "  flip trk to make new center";
          trk->Flip();
          vtxCenter = trk->Nodes().front();
        }
        else {
          mf::LogVerbatim("pma::VtxCandidate") << "  new center at the endpoint";
          vtxCenter = trk->Nodes().back();
        }
        vtxCenter->SetPoint3D(fCenter);
        nOK++;
      }
      else {
        if (vtxCenter->Prev() && trk->CanFlip()) {
          mf::LogVerbatim("pma::VtxCandidate") << "  flip trk to attach to inner";
          trk->Flip();
          if (trk->AttachTo(vtxCenter)) nOK++;
        }
        else {
          mf::LogVerbatim("pma::VtxCandidate") << "  endpoint to center";
          if (trk->AttachBackTo(vtxCenter)) nOK++;
        }
      }
    }
    else {
      bool canFlipPrev = true;
      if (vtxCenter && vtxCenter->Prev()) {
        pma::Segment3D* seg = static_cast<pma::Segment3D*>(vtxCenter->Prev());
        if (seg->Parent()->NextSegment(vtxCenter))
          canFlipPrev = false;
        else
          canFlipPrev = seg->Parent()->CanFlip();
      }

      if (hasInnerCenter || !canFlipPrev) {
        mf::LogVerbatim("pma::VtxCandidate") << "  split track";

        if ((f >= 0.0F) && (f <= 1.0) && (f * ds > kMinDistToNode) &&
            ((1.0 - f) * ds > kMinDistToNode)) {
          mf::LogVerbatim("pma::VtxCandidate") << "  add center inside segment";

          int tpc, cryo;
          if (f < 0.5) {
            tpc = tpc0;
            cryo = cryo0;
          }
          else {
            tpc = tpc1;
            cryo = cryo1;
          }

          trk->InsertNode(detProp, fCenter, ++idx, tpc, cryo);
        }
        else {
          if (d1 < d0) {
            mf::LogVerbatim("pma::VtxCandidate") << "  add center at end of segment";
            ++idx;
          }
          else {
            mf::LogVerbatim("pma::VtxCandidate") << "  center at start of segment - no action";
          }
        }

        pma::Track3D* t0 = trk->Split(detProp, idx); // makes both tracks attached to each other

        if (t0) {
          mf::LogVerbatim("pma::VtxCandidate")
            << "  trk size:" << trk->size() << " (nodes:" << trk->Nodes().size() << ")";

          trk->MakeProjection();

          mf::LogVerbatim("pma::VtxCandidate")
            << "  t0 size:" << t0->size() << " (nodes:" << t0->Nodes().size() << ")";

          t0->MakeProjection();

          tracks.tracks().emplace_back(t0, key, tid);
          if (i == 0) {
            mf::LogVerbatim("pma::VtxCandidate") << "  center at trk0 back";
            vtxCenter = trk->Nodes().front();
            nOK += 2;
          }
          else {
            mf::LogVerbatim("pma::VtxCandidate") << "  attach trk to trk0";
            if (trk->AttachTo(vtxCenter)) nOK += 2;
          }
        }
        mf::LogVerbatim("pma::VtxCandidate") << "  done";
      }
      else {
        mf::LogVerbatim("pma::VtxCandidate") << "  inner center";
        hasInnerCenter = true;

        if ((f >= 0.0F) && (f <= 1.0) && (f * ds > kMinDistToNode) &&
            ((1.0 - f) * ds > kMinDistToNode)) {
          mf::LogVerbatim("pma::VtxCandidate") << "  add center inside segment";

          int tpc, cryo;
          if (f < 0.5) {
            tpc = tpc0;
            cryo = cryo0;
          }
          else {
            tpc = tpc1;
            cryo = cryo1;
          }

          trk->InsertNode(detProp, fCenter, ++idx, tpc, cryo);
        }
        else {
          if (d1 < d0) {
            mf::LogVerbatim("pma::VtxCandidate") << "  add center at end of segment";
            ++idx;
          }
          else {
            mf::LogVerbatim("pma::VtxCandidate") << "  center at start of segment - no action";
          }
        }

        pma::Node3D* innerCenter = trk->Nodes()[idx];
        if (i > 0) // already has vtxCenter
        {
          // prepare for prev...
          pma::Track3D* rootBranch = 0;
          pma::Segment3D* seg = static_cast<pma::Segment3D*>(vtxCenter->Prev());
          if (seg) {
            rootBranch = seg->Parent();
            rootBranch->Flip();
          }

          // ...but nexts reattached first, then...
          auto branches = vtxCenter->GetBranches();

          for (size_t j = 0; j < branches.size(); ++j) {
            if (branches[j]->AttachTo(innerCenter, true)) {}
          }
          vtxCenter = innerCenter; // vtxCenter is deleted after full reattach
        }
        else {
          vtxCenter = innerCenter; // vtx from inner center
        }                          // set vtxCenter on i == 0

        nOK++;

        mf::LogVerbatim("pma::VtxCandidate") << "  done";
      }
    }
  }

  bool result = false;
  if (vtxCenter) {
    pma::Segment3D* rootSeg = 0;
    if (vtxCenter->NextCount())
      rootSeg = static_cast<pma::Segment3D*>(vtxCenter->Next(0));
    else if (vtxCenter->Prev())
      rootSeg = static_cast<pma::Segment3D*>(vtxCenter->Prev());
    else
      throw cet::exception("pma::VtxCandidate") << "Vertex with no segments attached.";

    pma::Track3D* rootTrk = rootSeg->Parent()->GetRoot();
    if (!rootTrk) rootTrk = rootSeg->Parent();

    std::vector<pma::Track3D const*> branchesToRemove;     // change to a more simple check
    bool noLoops = rootTrk->GetBranches(branchesToRemove); // if there are loops

    bool tuneOK = true;
    if (noLoops && (nOK > 1)) {
      fAssigned.clear();
      fCenter = vtxCenter->Point3D();
      fMse = 0.0;
      fMse2D = 0.0;

      double g = rootTrk->TuneFullTree();

      if (g > -2.0) // -1.0: high value of g; -2.0: inf. value of g.
      {
        result = true; // all OK, new vertex added
      }
      else {
        tuneOK = false; // inf. g, remove tracks
      }
    }

    if (noLoops && tuneOK) {
      mf::LogVerbatim("pma::VtxCandidate") << "remove backup tracks";
      for (auto& c : backupTracks.tracks())
        c.DeleteTrack();
    }
    else {
      mf::LogVerbatim("pma::VtxCandidate") << "restore tracks from backup....";
      for (int tid : treeIds) {
        size_t t = 0;
        while (t < tracks.size()) {
          if (tracks[t].TreeId() == tid) {
            tracks[t].DeleteTrack();
            tracks.erase_at(t);
          }
          else
            t++;
        }
      }
      for (const auto& c : backupTracks.tracks())
        tracks.push_back(c);
      mf::LogVerbatim("pma::VtxCandidate") << "  done";
    }
  }
  else
    mf::LogError("pma::VtxCandidate") << "Cannot create common vertex";

  return result;
}

double pma::VtxCandidate::MaxAngle

(

double

minLength = 0.0

)

const

Definition at line 243 of file PmaVtxCandidate.cxx.

References fAssigned, pma::Track3D::Nodes(), and pma::Node3D::Point3D().

Referenced by VtxCandidate().

{
  TVector3 dir_i;
  size_t max_l_idx = 0;
  double l, max_l = 0.0;
  for (size_t i = 0; i < fAssigned.size() - 1; i++) {
    l = fAssigned[i].first.Track()->Length();
    if (l > max_l) {
      max_l = l;
      max_l_idx = i;
      pma::Track3D* trk_i = fAssigned[i].first.Track();
      pma::Node3D* vtx_i0 = trk_i->Nodes()[fAssigned[i].second];
      pma::Node3D* vtx_i1 = trk_i->Nodes()[fAssigned[i].second + 1];
      dir_i = vtx_i1->Point3D() - vtx_i0->Point3D();
      dir_i *= 1.0 / dir_i.Mag();
    }
  }

  double a, min = 1.0;
  for (size_t j = 0; j < fAssigned.size(); j++)
    if ((j != max_l_idx) && (fAssigned[j].first.Track()->Length() > minLength)) {
      pma::Track3D* trk_j = fAssigned[j].first.Track();
      pma::Node3D* vtx_j0 = trk_j->Nodes()[fAssigned[j].second];
      pma::Node3D* vtx_j1 = trk_j->Nodes()[fAssigned[j].second + 1];
      TVector3 dir_j = vtx_j1->Point3D() - vtx_j0->Point3D();
      dir_j *= 1.0 / dir_j.Mag();
      a = fabs(dir_i * dir_j);
      if (a < min) min = a;
    }

  return 180.0 * acos(min) / TMath::Pi();
}

bool pma::VtxCandidate::MergeWith

(

const VtxCandidate &

other

)

Definition at line 276 of file PmaVtxCandidate.cxx.

References Compute(), ComputeMse2D(), d, pma::Dist2(), fAssigned, fCenter, fMse, fMse2D, Has(), IsAttached(), Size(), and Test().

Referenced by Size().

{
  double d = sqrt(pma::Dist2(fCenter, other.fCenter));
  if (d > 10.0) {
    mf::LogVerbatim("pma::VtxCandidate") << "too far..";
    return false;
  }

  double dw = Test(other);

  size_t ntrk = 0;
  for (const auto& t : other.fAssigned) {
    if (IsAttached(t.first.Track())) {
      mf::LogVerbatim("pma::VtxCandidate") << "already attached..";
      return false;
    }
    if (!Has(t.first.Track())) {
      fAssigned.push_back(t);
      ntrk++;
    }
  }
  if (ntrk) {
    double mse0 = fMse, mse1 = other.fMse;
    mf::LogVerbatim("pma::VtxCandidate")
      << "try: " << d << " mse0:" << sqrt(mse0) << " mse1:" << sqrt(mse1);
    //std::cout << "try: " << d << " mse0:" << sqrt(mse0) << " mse1:" << sqrt(mse1) << std::endl;

    double mse = Compute();
    mf::LogVerbatim("pma::VtxCandidate")
      << "out: " << Size() << " mse:" << sqrt(mse) << " dw:" << dw;
    //std::cout << "out: " << Size() << " mse:" << sqrt(mse) << " dw:" << dw << std::endl;

    if (mse < 1.0) // kMaxDistToTrack * kMaxDistToTrack)
    {
      fMse = mse;
      fMse2D = ComputeMse2D();
      return true;
    }
    else {
      mf::LogVerbatim("pma::VtxCandidate") << "high mse..";
      while (ntrk--) {
        fAssigned.pop_back();
      }
      fMse = Compute();
      fMse2D = ComputeMse2D();
      return false;
    }
  }
  else {
    mf::LogVerbatim("pma::VtxCandidate") << "no tracks..";
    return false;
  }
}

double pma::VtxCandidate::Mse ( ) const

inline

Definition at line 68 of file PmaVtxCandidate.h.

References fMse.

Referenced by pma::PMAlgVertexing::firstPassCandidates(), and pma::PMAlgVertexing::secondPassCandidates().

{ return fMse; }

double pma::VtxCandidate::Mse2D ( ) const

inline

Definition at line 69 of file PmaVtxCandidate.h.

References fMse2D.

{ return fMse2D; }

size_t pma::VtxCandidate::Size ( ) const

inline

Definition at line 56 of file PmaVtxCandidate.h.

References Compute(), fAssigned, JoinTracks(), and MergeWith().

Referenced by MergeWith().

{ return fAssigned.size(); }

size_t pma::VtxCandidate::Size

(

double

minLength

)

const

Definition at line 78 of file PmaVtxCandidate.cxx.

References fAssigned, and n.

{
  size_t n = 0;
  for (auto const& c : fAssigned)
    if (c.first.Track()->Length() > minLength) n++;
  return n;
}

double pma::VtxCandidate::Test

(

const VtxCandidate &

other

)

const

Definition at line 233 of file PmaVtxCandidate.cxx.

References fCenter, and fErr.

Referenced by MergeWith(), and VtxCandidate().

{
  double dx = fCenter[0] - other.fCenter[0];
  double dy = fCenter[1] - other.fCenter[1];
  double dz = fCenter[2] - other.fCenter[2];
  double dw = fErr[0] * other.fErr[0] * dx * dx + fErr[1] * other.fErr[1] * dy * dy +
              fErr[2] * other.fErr[2] * dz * dz;
  return sqrt(dw);
}

std::pair<pma::Track3D*, size_t> pma::VtxCandidate::Track ( size_t  i ) const

inline

Definition at line 71 of file PmaVtxCandidate.h.

References fAssigned.

  {
    return std::pair<pma::Track3D*, size_t>(fAssigned[i].first.Track(), fAssigned[i].second);
  }

Member Data Documentation

std::vector<std::pair<pma::TrkCandidate, size_t> > pma::VtxCandidate::fAssigned

private

Definition at line 86 of file PmaVtxCandidate.h.

Referenced by Add(), Compute(), ComputeMse2D(), Has(), HasLoops(), IsAttached(), JoinTracks(), MaxAngle(), MergeWith(), Size(), and Track().

TVector3 pma::VtxCandidate::fCenter

private

Definition at line 87 of file PmaVtxCandidate.h.

Referenced by Add(), Center(), Compute(), ComputeMse2D(), JoinTracks(), MergeWith(), and Test().

TVector3 pma::VtxCandidate::fErr

private

Definition at line 87 of file PmaVtxCandidate.h.

Referenced by Compute(), and Test().

double pma::VtxCandidate::fMse

private

Definition at line 85 of file PmaVtxCandidate.h.

Referenced by Add(), JoinTracks(), MergeWith(), and Mse().

double pma::VtxCandidate::fMse2D

private

Definition at line 85 of file PmaVtxCandidate.h.

Referenced by Add(), JoinTracks(), MergeWith(), and Mse2D().

double pma::VtxCandidate::fSegMinLength

private

Definition at line 85 of file PmaVtxCandidate.h.

Referenced by Add(), and Compute().

constexpr double pma::VtxCandidate::kMaxDistToTrack {4.0}

static

Definition at line 26 of file PmaVtxCandidate.h.

Referenced by Add().

constexpr double pma::VtxCandidate::kMinDistToNode {2.0}

static

Definition at line 27 of file PmaVtxCandidate.h.

Referenced by JoinTracks().

bool pma::VtxCandidate::tracksJoined

private

Definition at line 84 of file PmaVtxCandidate.h.

Referenced by JoinTracks().

---

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

• larreco/v09_25_00/source/larreco/RecoAlg/PMAlg/PmaVtxCandidate.h
• larreco/v09_25_00/source/larreco/RecoAlg/PMAlg/PmaVtxCandidate.cxx