Solver.h File Reference

#include <vector>
#include "QuadExpr.h"

Go to the source code of this file.

Classes
class	WireHit
class	InductionWireHit
class	Neighbour
class	SpaceCharge
class	CollectionWireHit

Functions
double	Metric (const std::vector< SpaceCharge * > &scs, double alpha)
double	Metric (const std::vector< CollectionWireHit * > &cwires, double alpha)
QuadExpr	Metric (const SpaceCharge *sci, const SpaceCharge *scj, double alpha)
double	SolvePair (CollectionWireHit *cwire, SpaceCharge *sci, SpaceCharge *scj, double xmin, double xmax, double alpha)
void	Iterate (CollectionWireHit *cwire, double alpha)
void	Iterate (SpaceCharge *sc, double alpha)
void	Iterate (const std::vector< CollectionWireHit * > &cwires, const std::vector< SpaceCharge * > &orphanSCs, double alpha)

Function Documentation

void Iterate	(	CollectionWireHit *	cwire,
		double	alpha
	)

Definition at line 263 of file Solver.cxx.

References SpaceCharge::AddCharge(), CollectionWireHit::fCrossings, SolvePair(), and x.

Referenced by Iterate(), and reco3d::SpacePointSolver::Minimize().

{
  // Consider all pairs of crossings
  const unsigned int N = cwire->fCrossings.size();

  for(unsigned int i = 0; i+1 < N; ++i){
    SpaceCharge* sci = cwire->fCrossings[i];

    for(unsigned int j = i+1; j < N; ++j){
      SpaceCharge* scj = cwire->fCrossings[j];

      const double x = SolvePair(cwire, sci, scj, alpha);

      if(x == 0) continue;

      // Actually make the update
      sci->AddCharge(+x);
      scj->AddCharge(-x);
    } // end for j
  } // end for i
}

void Iterate	(	SpaceCharge *	sc,
		double	alpha
	)

Definition at line 286 of file Solver.cxx.

References SpaceCharge::AddCharge(), QuadExpr::Eval(), SpaceCharge::fPred, QuadExpr::Linear(), max, Metric(), QuadExpr::Quadratic(), and x.

{
  const QuadExpr chisq = Metric(sc, alpha);

  // Find the minimum of a quadratic expression
  double x = -chisq.Linear()/(2*chisq.Quadratic());

  // Don't allow the SpaceCharge to go negative
  const double xmin = -sc->fPred;

  // Clamp to allowed range
  x = std::max(xmin, x);

  const double chisq_new = chisq.Eval(x);

  // Should try here too, because the function might be convex not concave, so
  // d/dx=0 gives the max not the min, and the true min is at one extreme of
  // the range.
  const double chisq_n = chisq.Eval(xmin);

  if(chisq_n < chisq_new)
    sc->AddCharge(xmin);
  else
    sc->AddCharge(x);
}

void Iterate	(	const std::vector< CollectionWireHit * > &	cwires,
		const std::vector< SpaceCharge * > &	orphanSCs,
		double	alpha
	)

Definition at line 313 of file Solver.cxx.

References Iterate().

{
  // Visiting in a "random" order helps prevent local artefacts that are slow
  // to break up.
  unsigned int cwireIdx = 0;
  do{
    Iterate(cwires[cwireIdx], alpha);

    const unsigned int prime = 1299827;
    cwireIdx = (cwireIdx+prime)%cwires.size();
  } while(cwireIdx != 0);

  for(SpaceCharge* sc: orphanSCs) Iterate(sc, alpha);
}

double Metric	(	const std::vector< SpaceCharge * > &	scs,
		double	alpha
	)

Definition at line 83 of file Solver.cxx.

References Metric(), and sqr().

{
  double ret = 0;

  std::set<InductionWireHit*> iwires;
  for(const SpaceCharge* sc: scs){
    if(sc->fWire1) iwires.insert(sc->fWire1);
    if(sc->fWire2) iwires.insert(sc->fWire2);

    if(alpha != 0){
      ret -= alpha*sqr(sc->fPred);
      // "Double-counting" of the two ends of the connection is
      // intentional. Otherwise we'd have a half in the line above.
      ret -= alpha * sc->fPred * sc->fNeiPotential;
    }
  }

  for(const InductionWireHit* iwire: iwires){
    ret += Metric(iwire->fCharge, iwire->fPred);
  }

  return ret;
}

double Metric	(	const std::vector< CollectionWireHit * > &	cwires,
		double	alpha
	)

Definition at line 108 of file Solver.cxx.

References Metric().

{
  std::vector<SpaceCharge*> scs;
  for(CollectionWireHit* cwire: cwires)
    scs.insert(scs.end(), cwire->fCrossings.begin(), cwire->fCrossings.end());
  return Metric(scs, alpha);
}

QuadExpr Metric	(	const SpaceCharge *	sci,
		const SpaceCharge *	scj,
		double	alpha
	)

Definition at line 117 of file Solver.cxx.

References InductionWireHit::fCharge, Neighbour::fCoupling, SpaceCharge::fNeighbours, SpaceCharge::fNeiPotential, InductionWireHit::fPred, SpaceCharge::fPred, Neighbour::fSC, SpaceCharge::fWire1, SpaceCharge::fWire2, Metric(), n, sqr(), x, and QuadExpr::X().

{
  QuadExpr ret = 0;

  // How much charge moves from scj to sci
  QuadExpr x = QuadExpr::X();

  if(alpha != 0){
    const double scip = sci->fPred;
    const double scjp = scj->fPred;

    // Self energy. SpaceCharges are never the same object
    ret -= alpha*sqr(scip + x);
    ret -= alpha*sqr(scjp - x);

    // Interaction. We're only seeing one end of the double-ended connection
    // here, so multiply by two.
    ret -= 2 * alpha * (scip + x) * sci->fNeiPotential;
    ret -= 2 * alpha * (scjp - x) * scj->fNeiPotential;

    // This miscounts if i and j are neighbours of each other
    for(const Neighbour& n: sci->fNeighbours){
      if(n.fSC == scj){
        // If we detect that case, remove the erroneous terms
        ret += 2 * alpha * (scip + x) * scjp * n.fCoupling;
        ret += 2 * alpha * (scjp - x) * scip * n.fCoupling;

        // And replace with the correct interaction terms
        ret -= 2 * alpha * (scip + x) * (scjp - x) * n.fCoupling;
        break;
      }
    }
  }


  const InductionWireHit* iwire1 = sci->fWire1;
  const InductionWireHit* jwire1 = scj->fWire1;

  const double qi1 = iwire1 ? iwire1->fCharge : 0;
  const double pi1 = iwire1 ? iwire1->fPred : 0;

  const double qj1 = jwire1 ? jwire1->fCharge : 0;
  const double pj1 = jwire1 ? jwire1->fPred : 0;

  if(iwire1 == jwire1){
    // Same wire means movement of charge cancels itself out
    if(iwire1) ret += Metric(qi1, pi1);
  }
  else{
    if(iwire1) ret += Metric(qi1, pi1 + x);
    if(jwire1) ret += Metric(qj1, pj1 - x);
  }

  const InductionWireHit* iwire2 = sci->fWire2;
  const InductionWireHit* jwire2 = scj->fWire2;

  const double qi2 = iwire2 ? iwire2->fCharge : 0;
  const double pi2 = iwire2 ? iwire2->fPred : 0;

  const double qj2 = jwire2 ? jwire2->fCharge : 0;
  const double pj2 = jwire2 ? jwire2->fPred : 0;

  if(iwire2 == jwire2){
    if(iwire2) ret += Metric(qi2, pi2);
  }
  else{
    if(iwire2) ret += Metric(qi2, pi2 + x);
    if(jwire2) ret += Metric(qj2, pj2 - x);
  }

  return ret;
}

double SolvePair	(	CollectionWireHit *	cwire,
		SpaceCharge *	sci,
		SpaceCharge *	scj,
		double	xmin,
		double	xmax,
		double	alpha
	)