hitfit::Fourvec_Constrainer Class Reference

Do a kinematic fit for a set of four-momenta, given a set of mass constraints. More...

#include <Fourvec_Constrainer.h>

Public Member Functions
void	add_constraint (std::string s)
	Specify an additional constraint s for the problem. The format for s is described in the class description. More...
double	constrain (Fourvec_Event &ev, double &m, double &sigm, Column_Vector &pullx, Column_Vector &pully)
	Do a constrained fit for event ev. Returns the requested mass and its uncertainty in m and sigm, and the pull quantities in pullx and pully. Returns the , the value will be negative if the fit failed to converge. More...
	Fourvec_Constrainer (const Fourvec_Constrainer_Args &args)
	Constructor. More...
void	mass_constraint (std::string s)
	Specify a combination of objects that will be returned by the constrain() method as mass. The format of s is the same as for normal constraints. The left-hand side specifies the mass to calculate, the right-hand side should be zero. This combination of objects will be called only once. More...

Private Attributes
const Fourvec_Constrainer_Args	_args
std::vector< Constraint >	_constraints
std::vector< Constraint >	_mass_constraint

Friends
std::ostream &	operator<< (std::ostream &s, const Fourvec_Constrainer &c)
	Output stream operator, print the content of this Fourvec_Constrainer to an output stream. More...

Detailed Description

Do a kinematic fit for a set of four-momenta, given a set of mass constraints.

The input is in a Fourvec_Event instance. This consists of a collection of objects, each of which has a four-momentum, an uncertainty on the four-momentum, and an integer label.

There are also a set of mass constraints, based on sums of objects with specified labels. A constraint can either require that the invariant mass of a set of objects is constant, or that the mass of two sets of objects be equal to each other. For example, the constraint

says that the sum of all objects with labels 1 and 2 should have a mass of 80, where the unit of mass is always in GeV. And the constraint

says that the sum of all objects with labels 1 and 2 should have an invariant mass equal as the invariant mass of the sum of all objects with label 3 and 4.

All the objects are fixed to constant masses for the fit. These masses are attributes of the objects in the Fourvec_Event. This is done by scaling either the four-momentum's three-momentum or energy, depending on the setting of the use_e parameter.

If there is no neutrino present in the event, two additional constraints are automatically added for total balance, unless the parameter ignore_met has been set to TRUE.

When the fit completes, this object can compute an invariant mass out of some combination of the objects, including and uncertainty. The particular combination is specified through the method mass_constraint((); it gets a constraint string like normal with the LHS of the constraint i s the mass which will be calculater, and the RHS of the oncstrant should be zero.

Definition at line 220 of file Fourvec_Constrainer.h.

Constructor & Destructor Documentation

Fourvec_Constrainer()

hitfit::Fourvec_Constrainer::Fourvec_Constrainer

(

const Fourvec_Constrainer_Args &

args

)

Constructor.

Parameters

args	Parameter settings for this instance.

Definition at line 174 of file Fourvec_Constrainer.cc.

      : _args(args) {}

Member Function Documentation

add_constraint()

void hitfit::Fourvec_Constrainer::add_constraint

(

std::string

s

)

Specify an additional constraint s for the problem. The format for s is described in the class description.

Parameters

s	The constraint to add.

Definition at line 183 of file Fourvec_Constrainer.cc.

References _constraints, and alignCSCRings::s.

Referenced by hitfit::Constrained_Top::Constrained_Top(), and hitfit::Constrained_Z::Constrained_Z().

  {
    _constraints.push_back(Constraint(s));
  }

constrain()

double hitfit::Fourvec_Constrainer::constrain	(	Fourvec_Event &	ev,
		double &	m,
		double &	sigm,
		Column_Vector &	pullx,
		Column_Vector &	pully
	)

Do a constrained fit for event ev. Returns the requested mass and its uncertainty in m and sigm, and the pull quantities in pullx and pully. Returns the , the value will be negative if the fit failed to converge.

Parameters

ev	The event to be fitted (input), and later after fitting (output).
m	The requested invariant mass to calculate.
sigm	The uncertainty in the requested invariant mass.
pullx	Pull quantities for well-measured variables.
pully	Pull quantities for poorly-measured variables.

Input:

• ev (before fit).

Output:

• ev (after fit).
• m.
• sigm.
• pullx
• pully.

Returns:

The fit , this value will be negative if the fit failed to converge.

Definition at line 1612 of file Fourvec_Constrainer.cc.

References _args, _constraints, _mass_constraint, gpuPixelDoublets::cc, hitfit::Fourvec_Constrainer_Args::chisq_constrainer_args(), makeMEIFBenchmarkPlots::ev, hitfit::Fourvec_Constrainer_Args::ignore_met(), visualization-live-secondInstance_cfg::m, dttmaxenums::R, hitfit::Fourvec_Constrainer_Args::use_e(), x, and beamSpotPI::Y.

Referenced by hitfit::Constrained_Z::constrain(), and hitfit::Constrained_Top::constrain().

  {
    adjust_fourvecs(ev, _args.use_e());

    bool use_kt = ev.has_neutrino() || !_args.ignore_met();
    int nobjs = ev.nobjs();
    int n_measured_vars = nobjs * 3;
    int n_unmeasured_vars = 0;

    if (use_kt) {
      n_measured_vars += 2;

      if (ev.has_neutrino())
        n_unmeasured_vars = 1;
    }

    Matrix G_i(n_measured_vars, n_measured_vars);
    Diagonal_Matrix Y(n_unmeasured_vars);
    Column_Vector x(n_measured_vars);
    Column_Vector y(n_unmeasured_vars);
    pack_event(ev, _args.use_e(), use_kt, x, y, G_i, Y);

    Column_Vector xm = x;
    Column_Vector ym = y;

    // ??? Should allow for using a different underlying fitter here.
    Chisq_Constrainer fitter(_args.chisq_constrainer_args());

    Fourvec_Constraint_Calculator cc(ev, _constraints, _args);

    Matrix Q;
    Matrix R;
    Matrix S;
    double chisq = fitter.fit(cc, xm, x, ym, y, G_i, Y, pullx, pully, Q, R, S);

    unpack_event(ev, x, y, _args.use_e(), use_kt);

    calculate_mass(ev, _mass_constraint, _args, chisq, Q, R, S, m, sigm);

    return chisq;
  }

mass_constraint()

void hitfit::Fourvec_Constrainer::mass_constraint

(

std::string

s

)

Specify a combination of objects that will be returned by the constrain() method as mass. The format of s is the same as for normal constraints. The left-hand side specifies the mass to calculate, the right-hand side should be zero. This combination of objects will be called only once.

Parameters

s	The constraint defining the mass.

Definition at line 195 of file Fourvec_Constrainer.cc.

References _mass_constraint, cms::cuda::assert(), and alignCSCRings::s.

Referenced by hitfit::Constrained_Top::Constrained_Top().

  {
    assert(_mass_constraint.empty());
    _mass_constraint.push_back(Constraint(s));
  }

Friends And Related Function Documentation

operator<<

std::ostream& operator<< ( std::ostream &  s, const Fourvec_Constrainer &  c  )

friend

Output stream operator, print the content of this Fourvec_Constrainer to an output stream.

Parameters

s	The output stream to which to write.
c	The instance of Fourvec_Constrainer to be printed.

Definition at line 220 of file Fourvec_Constrainer.cc.

  {
    s << "Constraints: (e_com = " << c._args.e_com() << ") ";
    if (c._args.use_e())
      s << "(E)";
    s << "\n";

    for (std::vector<Constraint>::size_type i = 0; i < c._constraints.size(); i++)
      s << "  " << c._constraints[i] << "\n";

    if (!c._mass_constraint.empty()) {
      s << "Mass constraint:\n";
      s << c._mass_constraint[0] << "\n";
    }
    return s;
  }

Member Data Documentation

_args

const Fourvec_Constrainer_Args hitfit::Fourvec_Constrainer::_args

private

Parameter settings for this instance.

Definition at line 306 of file Fourvec_Constrainer.h.

Referenced by constrain().

_constraints

std::vector<Constraint> hitfit::Fourvec_Constrainer::_constraints

private

The constraints for this problem.

Definition at line 312 of file Fourvec_Constrainer.h.

Referenced by add_constraint(), and constrain().

_mass_constraint

std::vector<Constraint> hitfit::Fourvec_Constrainer::_mass_constraint

private

The constraint giving the mass to be calculated. This should have no more than one entry.

Definition at line 320 of file Fourvec_Constrainer.h.

Referenced by constrain(), and mass_constraint().