Mattione GlueX Kinematic Fitting

From GlueXWiki
Revision as of 07:29, 3 November 2017 by Pmatt (Talk | contribs) (Fit Setup - New Fit)

Jump to: navigation, search

Quick Start

  • The kinematic fitter (DKinFitter) is in the KINFITTER library, and is fully integrated into the analysis library (via DKinFitUtils_GlueX and the DKinFitResults_factory). So, to kinematically fit a DReaction, you only have to specify the desired kinematic fit type in the DReaction.
    • See sim-recon/src/programs/Analysis/plugins/b1pi_hists for an example of using the kinematic fitter in an analysis.
  • Details on the kinematic fitting equations and constraints are in GlueX-doc-2112.

Not Yet Supported

  • Spacetime constraints

Constraints and Unknowns

  • Up to one particle may be missing: 3 unknowns (px, py, pz)
  • Four-momentum constraint: 4 constraints (Initial p4 - Final p4 = 0)
  • Mass constraints: 1 constraint per particle (E2 - p2 - m2 = 0)
    • Note that the ANALYSIS library does not constrain the masses of the ω or φ mesons since their widths are non-negligible.
    • Mass constraints are automatically applied when a "P4" fit is chosen.
  • Vertex constraints: 3 unknowns per vertex (x, y, z), 2 constraints per constraining particle
    • Constraining particles: Tracks, beam (when uncertainties are defined), decaying particles (when their positions are defined by a different vertex constraint)
    • E.g. for no magnetic field, p x (Δx) = 0: The path (Δx) and momentum vectors must be parallel
      • Δx = xMeasured - xUnknown
  • Spacetime constraints (NOT YET SUPPORTED): 4 unknowns per spacetime point (x, y, z, t), 3 constraints per constraining particle (vertex + time)
    • Also, one constraint per constraining neutral shower (time)

Notes for those manually performing their own fits

DKinFitUtils_GlueX

  • These utilities are used to setup the kinematic fit that is indicated by the DReaction::dKinFitType variable. This variable is an enum type called DKinFitType, which is defined in libraries/ANALYSIS/DReaction.h
  • These utilities automatically determines what p4, vertex, and spacetime (not yet supported!) constraints to setup such that each the data in each DParticleCombo are maximally constrained.
    • It will also automatically create additional vertex constraints for decaying particles with long lifetimes, as long as the IsDetachedVertex() function in libraries/include/particleType.h is kept updated.
    • If an inclusive channel, or if a decaying particle of type Unknown is present in the DReaction, the fit will not constrain the overall p4 of the reaction, but it will still apply the other constraints.
  • To exclude decaying particles from the mass constraints of a kinematic fit, call the DReactionStep::Set_KinFitConstrainInitMassFlag(bool) function with an argument of true on the step where the decaying particle is in the initial state.

Detected Neutral Particles

  • The "measured" DNeutralParticleHypothesis objects in the combos sets the particle momentum assuming that the particle came from the position roughly defined during comboing (see DSourceComboVertexer).
  • The best way to accurately perform kinematic fits with neutral particles is to perform a simultaneous vertex & p4 fit, with the neutral particles included in the vertex constraints.
    • Instead of constraining the vertex, the neutral particle vertex will be defined by the latest iteration of the kinematic fit, and it's momentum will be recalculated at each iteration accordingly.
      • Technically the neutral shower momentum isn't directly included in the p4 fit if it's also in a vertex fit: the shower hit position and energy (and their uncertainties) are used along with the unknown vertex instead.

Vertex Fits

  • The initial vertex guesses must be supplied to the kinematic fitter, and it is extremely important to have good initial values for them.
  • It is recommended that the following procedure be used to determine initial guesses for the vertex fits:
    • Use the DOCA routines in DAnalysisUtilities to find an initial vertex guess for a set of tracks.
    • Prior to performing a simultaneous vertex-p4 fit, first kinematically fit each vertex individually using the above initial vertex guesses. Then use the results from these kinematic fit as inputs to the vertex-p4 fit.

Code Documentation

Primary Classes

  • The DKinFitter C++ class is designed to be usable for any experiment, fixed-target or multi-beam, and with or without a magnetic field present. It's primary dependency is ROOT, from which it uses the vector and matrix classes. It also depends on DResettable and DResourcePool in sim-recon/src/libraries/include/ to help with memory management. The only assumption that the fitter makes is that the uncertainties in the input particle covariance matrices are in the following order:
    • Particles: px, py, pz, xx, xy, xz, t
    • Showers: E, xx, xy, xz, t
  • The DKinFitUtils C++ class is used to supplement the kinematic fitter with utility functions used to help manage the fit, and to perform some calculations.
    • It is used for creating the particles, constraints, and other objects used in the fit, and is responsible for managing their memory.
    • It is also used to provide access to information about the presence, strength, or direction of the magnetic field.
  • However, since the magnetic field can vary from experiment to experiment, the DKinFitUtils class itself knows nothing about it. Instead, it is an abstract class, and each experiment must define a class inheriting from it that defines the following member functions (including two that are useful for helping to setup constraints):
//FOR SETUP:
virtual bool Get_IncludeBeamlineInVertexFitFlag(void) const = 0;
 
//FOR FITS:
virtual bool Get_IsBFieldNearBeamline(void) const = 0;
virtual TVector3 Get_BField(const TVector3& locPosition) const = 0; //must return in units of Tesla!!
  • Thus DKinFitUtils is an abstract base class; it cannot be instantiated directly. Instead, for each experiment a class should inherit from DKinFitUtils and define these methods; for GlueX, this is the DKinFitUtils_GlueX class.

DKinFitter Data Classes

The following classes are used to hold and handle data for the kinematic fit. The memory for all of these classes is managed internally (with resource pools) by the DKinFitUtils class.

  • DKinFitParticle: One for each particle in the fit. There are several types of particles, which are differentiated by the DKinFitParticleType enum.
    • DKinFitParticleType: An enum used to distinguish different particle types. The different types of particles are: d_DetectedParticle, d_BeamParticle, d_TargetParticle, d_DecayingParticle, d_MissingParticle
  • DKinFitConstraint: This is an abstract base class that has the following derived classes:
    • DKinFitConstraint_P4: Constrains a set of initial- and final-state particles to conserve four-momentum.
    • DKinFitConstraint_Mass: Constrains the four-momentum of a decaying particle to have a fixed mass.
    • DKinFitConstraint_Vertex: Used for constraining a set of particles to a common vertex.
    • DKinFitConstraint_Spacetime: Used for constraining a set of particles to a common spacetime. This class inherits from DKinFitConstraint_Vertex.
  • DKinFitChain & DKinFitChainStep: Used for representing the reaction to be fit as a series of decay-steps in a decay chain, containing the created DKinFitParticle's.
    • These classes are not necessary to perform the fit, however, they can be useful for using some of the (optional) DKinFitUtils functions to help setup the constraints for your fits.

In addition, resource pools of TMatrixFSym objects are used to manage memory for the covariance matrices.

Fit Setup - New Run

  • Required: First, create pointers to the DKinFitter & DKinFitUtils_GlueX classes in your header file:
#include "KINFITTER/DKinFitter.h"
#include "ANALYSIS/DKinFitUtils_GlueX.h"
 
//in class definition:
DKinFitter* dKinFitter;
DKinFitUtils_GlueX* dKinFitUtils;
  • Required: Create the objects in your .cc file. This also sets the magnetic field.
dKinFitUtils = new DKinFitUtils_GlueX(locEventLoop);
dKinFitter = new DKinFitter(dKinFitUtils);
  • Optional: Set sizes for resource pools, set the debug flag, etc.

Fit Setup - New Event

  • At the beginning of each event, the kinematic fitter (and the utilities) need to be reset This recycles the memory (& output/input object mapping) of all objects for reuse. After the fit, these fit results objects are stored in DKinFitResults, so they must stay valid for the entire event. To reset everything, call:
void DKinFitUtils_GlueX::Reset_NewEvent(void);
void DKinFitter::Reset_NewEvent(void);

Fit Setup - New Particle Combination

  • Create DKinFitParticle objects for your particles. These can be created by calling one of the following methods of DKinUtils_GlueX:
shared_ptr<DKinFitParticle> Make_BeamParticle(const DBeamPhoton* locBeamPhoton);
shared_ptr<DKinFitParticle> Make_BeamParticle(const DBeamPhoton* locBeamPhoton, const DEventRFBunch* locEventRFBunch); //sets rf time for photon
shared_ptr<DKinFitParticle> Make_TargetParticle(Particle_t locPID);
 
shared_ptr<DKinFitParticle> Make_DetectedParticle(const DKinematicData* locKinematicData);
shared_ptr<DKinFitParticle> Make_DetectedShower(const DNeutralShower* locNeutralShower, Particle_t locPID); //DO NOT call this unless the neutral is also in a vertex fit!
 
shared_ptr<DKinFitParticle> Make_DecayingParticle(Particle_t locPID, const set<shared_ptr<DKinFitParticle>>& locFromInitialState, const set<shared_ptr<DKinFitParticle>>& locFromFinalState);
shared_ptr<DKinFitParticle> Make_MissingParticle(Particle_t locPID);
  • Note that for decaying particles, you must pass in the pre-created DKinFitParticle objects that are used to define the momentum of the decaying particle.

Fit Setup - New Fit

  • Create the constraints used for the fit. These can be created manually by calling the following functions on DKinFitUtils_GlueX:
shared_ptr<DKinFitConstraint_Mass> Make_MassConstraint(const shared_ptr<DKinFitParticle>& locDecayingParticle);
shared_ptr<DKinFitConstraint_P4> Make_P4Constraint(const set<shared_ptr<DKinFitParticle>>& locInitialParticles, const set<shared_ptr<DKinFitParticle>>& locFinalParticles);
shared_ptr<DKinFitConstraint_Vertex> Make_VertexConstraint(const set<shared_ptr<DKinFitParticle>>& locFullConstrainParticles, const set<shared_ptr<DKinFitParticle>>& locNoConstrainParticles, TVector3 locVertexGuess = TVector3());
shared_ptr<DKinFitConstraint_Spacetime> Make_SpacetimeConstraint(const set<shared_ptr<DKinFitParticle>>& locFullConstrainParticles, const set<shared_ptr<DKinFitParticle>>& locOnlyConstrainTimeParticles,
	const set<shared_ptr<DKinFitParticle>>& locNoConstrainParticles, TLorentzVector locSpacetimeGuess = TLorentzVector());
  • Or, the constraints can be created automatically by calling the following function:
set<shared_ptr<DKinFitConstraint>> Create_Constraints(const DReactionVertexInfo* locReactionVertexInfo, const DReaction* locReaction, const DParticleCombo* locParticleCombo, const shared_ptr<const DKinFitChain>& locKinFitChain, DKinFitType locKinFitType, vector<shared_ptr<DKinFitConstraint_Vertex>>& locSortedVertexConstraints);
  • Next, set initial guesses for any common vertices, common times, and the p3 of a missing or open-ended decaying particle, if present. Do this by calling:
void DKinFitConstraint_P4::Set_InitP3Guess(const TVector3& locInitP3Guess);
void DKinFitConstraint_Vertex::Set_InitVertexGuess(const TVector3& locInitVertexGuess);
void DKinFitConstraint_Spacetime::Set_InitVertexGuess(const TVector3& locInitVertexGuess);
void DKinFitConstraint_Spacetime::Set_InitTimeGuess(double locInitTimeGuess);

Performing the Kinematic Fit

  • Reset the kinematic fitter for the new fit. This clears the list of "current-fit" particles and constraints, gets new resources for the fit covariance matrices, etc.:
void DKinFitter::Reset_NewFit(void);
  • Of the constraints created above, set the constraints you would like to apply for the current fit, by calling either of:
void DKinFitter::Add_Constraint(const shared_ptr<DKinFitConstraint>& locKinFitConstraint){dKinFitConstraints.insert(locKinFitConstraint);}
void DKinFitter::Add_Constraints(const set<shared_ptr<DKinFitConstraint>>& locKinFitConstraints);
  • Perform the kinematic fit by calling:
bool DKinFitter::Fit_Reaction(void);
  • This function returns true if the fit was successful, and false if it failed. The fit can fail for one of several reasons:
    • The input constraints were not setup correctly. For example, if a detected particle is constrained to multiple vertices, or is in multiple four-momentum constraints, or if not enough particles are available to constrain a vertex (2 are needed), etc.
    • The fit fails to converge to a minimum χ2.
    • If during the kinematic fit, one of matrices is not invertible (e.g. the determinant is zero).

Retrieving Fit Results

  • The following methods can be called to retrieve information on the quality of the most-recently-completed fit.
double DKinFitter::Get_ChiSq(void) const;
double DKinFitter::Get_ConfidenceLevel(void) const;
unsigned int DKinFitter::Get_NDF(void) const;
  • Many different fits with different constraints can be placed on the same set of particles in a given event or particle combination. This includes re-using the DKinFitParticle and DKinFitConstraint objects that were created earlier. However, each fit needs to update the "active" DKinFitParticle and DKinFitConstraint objects as each iteration improves the particle/vertex/etc. properties. Thus, in order to avoid over-writing the user-created DKinFitParticle and DKinFitConstraint objects, clones of these objects are made when DKinFitter::Fit_Reaction() is called. The cloned objects are used in the fit and contain the fit results. Thus, to obtain the fit results for a given particle / constraint, the cloned, output object needs to be retrieved via:
set<DKinFitConstraint*> DKinFitter::Get_KinFitConstraints(void) const;
set<DKinFitParticle*> DKinFitter::Get_KinFitParticles(void) const;
  • The output DKinFitParticle's can be mapped to the input DKinFitParticle's via:
DKinFitParticle* DKinFitUtils::Get_InputKinFitParticle(DKinFitParticle* locKinFitParticle) const;
  • And, the input DKinFitParticle's can be mapped to the original JObject's via:
const JObject* DKinFitUtils_GlueX::Get_SourceJObject(DKinFitParticle* locInputKinFitParticle) const;
  • For a more detailed look at how the fit parameters have changed with respect to their input, measured values, the pulls can be retrieved:
//1st key is particle, 2nd key is param type (defined in DKinFitParticle.h)
void DKinFitter::Get_Pulls(map<DKinFitParticle*, map<DKinFitPullType, double> >& locPulls) const;
  • The full covariance matrices for the fit parameters can be retrieved via:
const TMatrixDSym* Get_VEta(void); //covariance matrix of the measured parameters used in the fit
const TMatrixDSym* Get_VXi(void); //covariance matrix of the unknown parameters used in the fit
const TMatrixDSym* Get_V(void); //combined covariance matrix of the measured and unknown parameters used in the fit.

Notes on Manual Setup

Magnetic Field

  • If the particle is charged, the magnetic field is taken into account if a common vertex is simultaneously kinematically fit
    • Magnetic field assumed to be constant over the propagation range
  • If a common vertex is not defined, the momentum is assumed to be constant
  • Energy loss between the common vertex and the particle position is ignored

Neutral Showers

  • For input neutral showers, the momentum is defined-by and updated-with the common vertex (see "Common Vertex Notes")
  • If a common vertex is not simultaneously kinematically fit for a neutral shower, the fit will fail
  • To include a neutral shower in a p4 fit without a vertex fit, create a neutral particle from it
    • With a full 7x7 covaraince matrix and input it instead of the neutral shower
  • Massive neutral showers (e.g. neutron) cannot be used in vertex constraints: only spacetime constraints. However, photons can.
    • This is because their momentum is defined by the vertex time

P4 Constraints

  • Don't do if studying an inclusive channel.
  • Pick the step of the decay chain that contains the missing or open-ended-decaying particle, to make sure it is constrained
  • If no missing or open-ended decaying particle, pick the initial step, to make sure the beam information is used

Mass Constraints

  • Cannot use to constrain the mass of a missing particle: use a p4 constraint instead
  • Mass of decaying particle constrained by either missing mass or invariant mass:
    • If no missing particles, can constrain either way
      • By default, prefer using the invariant mass. This is to ensure consistency when combining with a p4 constraint.
      • Failure to do so could lead to uninvertible matrices, and a failed kinematic fit.
    • If this particle has a missing decay product (inclusive or otherwise): Constrain by missing mass
    • If computation by missing mass would have a missing particle (inclusive or otherwise): Constrain by invariant mass
  • However, the way it is calculated is not defined in these constraints; it is defined when creating the decaying DKinFitParticle
    • So be sure to create them correctly!

Vertex Constraints

  • You can include the beam particle in the common vertex fit, but only do so if it has a non-zero error matrix.
  • Neutral and missing particles included in the constraint will not be used to constrain the vertex (no-constrain), but will be set with the fit common vertex
    • This is necessary if you want the neutral particle momentum (from an input shower) to change with the reconstructed vertex
  • Decaying particles should only be used to constrain a fit if the position is defined in another vertex constraint
  • Massive neutral showers (e.g. neutron) cannot be used in vertex constraints: only spacetime constraints. However, photons can.
    • This is because their momentum is defined by the vertex time

Spacetime Constraints

  • THESE ARE CURRENTLY DISABLED
  • It is not possible to fit a common time without simultaneously fitting a common vertex.
    • Requiring a common time at a non-common vertex has no meaning, and the fitter is not setup to input a pre-fit common vertex with uncertainties.
  • You can include the RF time in the fit by calling the Make_BeamParticle function with the RF information
  • Missing particles included in the constraint will not be used to constrain the spacetime (no-constrain), but will be set with the fit common spacetime
  • Decaying particles should only be used to constrain a fit if the spacetime is defined in another vertex constraint

Objects & Memory

  • DKinFitParticle and DKinFitConstraint objects have private constructors: User cannot create them directly
    • Instead, they are managed by DKinFitUtils
  • User creates DKinFitParticles, adds them to the constraints
  • When the fit starts, INTERNALLY ONLY, the particles are cloned and the constraints are cloned to contain the clones
  • User can resuse particles & constraints objects (since they are cloned, they are not overwritten)

Duplicate Fit Results

The DKinFitResults_factory class will auto-detect whether the requested fit for the given particle combination is identical to a fit that has been previously performed that event. For example, fits can be identical when:

  • Two DParticleCombo's from the same DReaction differ only by a final-state photon, and a vertex-fit is requested (the photon is not used in the vertex fit).
  • Multiple DReaction's are used for the same DReactionStep contents. For example, each DReaction may have different actions but the same particles.
  • Two DReaction's are identical except for the inclusion of an ω or φ meson. For example: γ p → π+ π- π0 p vs. γ p → ω p; ω → π+ π- π0. The ω and φ mesons don't have a fixed mass so they are not constrained, nor do they have a detached vertex. Therefore, the kinematic fit results are independent of their inclusion.
  • Etc.

In these cases, the DKinFitResults_factory will simply assign the fit results from the previously-fit DParticleCombo to the new DParticleCombo.

Auto-detection of these cases is performed by creating the DKinFitConstraint's for the new DParticleCombo and comparing them to the previously-fit DParticleCombo.

Manual Kinematic Fit Example

  • To see an example on how to manually perform a kinematic fit (and cut on it's confidence level) inside of a DAnalysisAction, see DCustomAction_KinFitExample.* at Link