Belle II Software development
BeamParametersFitter Class Reference

Fitter calculating BeamParameters from CollisionBoostVector and CollisionInvariantMass. More...

#include <BeamParametersFitter.h>

Public Member Functions

 BeamParametersFitter ()
 Constructor.
 
 ~BeamParametersFitter ()
 Destructor.
 
void setIntervalOfValidity (const IntervalOfValidity &iov)
 Set interval of validity.
 
void setAngleHER (double angleHER)
 Set HER angle.
 
void setAngleLER (double angleLER)
 Set LER angle.
 
void setAngleError (double angleError)
 Set angle error.
 
void setBoostError (double boostError)
 Set boost error.
 
void setInvariantMassError (double invariantMassError)
 Set invariant-mass error.
 
void setVerbose (bool verbose)
 Set whether to be verbose (print Minuit output).
 
BeamParameters getBeamParameters () const
 Get beam parameters.
 
void setBeamParameters (const BeamParameters *beamParameters)
 Set beam parameters.
 
void fit ()
 Perform the fit.
 
void fillVertexData (double covarianceXX, double covarianceYY)
 Fill beam spot (vertex) data.
 
void importBeamParameters ()
 Import beam parameters.
 

Protected Member Functions

void setupDatabase ()
 Setup database.
 

Protected Attributes

IntervalOfValidity m_IntervalOfValidity
 Interval of validity.
 
double m_AngleHER = 0
 HER angle.
 
double m_AngleLER = 0
 LER angle.
 
double m_AngleError = 0
 Angle error.
 
double m_BoostError = 0.0001
 Boost error (use only if inverse error matrix is not available).
 
double m_InvariantMassError = 0.0001
 Invariant-mass error (use only if error is 0).
 
bool m_Verbose = false
 Whether to be verbose (print Minuit output).
 
BeamParameters m_BeamParameters
 Beam parameters.
 
DBObjPtr< BeamSpotm_BeamSpot
 Beam spot.
 
DBObjPtr< CollisionBoostVectorm_CollisionBoostVector
 Collision boost vector.
 
DBObjPtr< CollisionInvariantMassm_CollisionInvariantMass
 Collision invariant mass.
 

Detailed Description

Fitter calculating BeamParameters from CollisionBoostVector and CollisionInvariantMass.

Definition at line 31 of file BeamParametersFitter.h.

Constructor & Destructor Documentation

◆ BeamParametersFitter()

Constructor.

Definition at line 38 of file BeamParametersFitter.h.

39 {
40 }

◆ ~BeamParametersFitter()

~BeamParametersFitter ( )
inline

Destructor.

Definition at line 45 of file BeamParametersFitter.h.

46 {
47 }

Member Function Documentation

◆ fillVertexData()

void fillVertexData ( double covarianceXX,
double covarianceYY )

Fill beam spot (vertex) data.

The covariance matrix is rescaled to nominal XX and YY covariances. If covariances are negative, then no rescaling is performed.

Parameters
[in]covarianceXXCovariance-matrix XX element.
[in]covarianceYYCovariance-matrix YY element.

Definition at line 294 of file BeamParametersFitter.cc.

296{
298 m_BeamParameters.setVertex(ROOT::Math::XYZVector(m_BeamSpot->getIPPosition()));
299 TMatrixDSym beamSize = m_BeamSpot->getSizeCovMatrix();
300 double xScale, yScale;
301 if (covarianceXX < 0)
302 xScale = 1;
303 else
304 xScale = sqrt(covarianceXX / beamSize[0][0]);
305 if (covarianceYY < 0)
306 yScale = 1;
307 else
308 yScale = sqrt(covarianceYY / beamSize[1][1]);
309 for (int i = 0; i < 3; ++i) {
310 beamSize[0][i] *= xScale;
311 beamSize[i][0] *= xScale;
312 beamSize[1][i] *= yScale;
313 beamSize[i][1] *= yScale;
314 }
315 m_BeamParameters.setCovVertex(beamSize);
316}
BeamParameters m_BeamParameters
Beam parameters.
void setupDatabase()
Setup database.
DBObjPtr< BeamSpot > m_BeamSpot
Beam spot.
double sqrt(double a)
sqrt for double
Definition beamHelpers.h:28

◆ fit()

void fit ( )

Perform the fit.

Definition at line 189 of file BeamParametersFitter.cc.

190{
191 int minuitResult;
193 /* Get p_HER and p_LER from a fit. */
194 double herMomentum, herThetaX, herThetaY;
195 double lerMomentum, lerThetaX, lerThetaY;
196 s_BoostVector = m_CollisionBoostVector->getBoost();
197 s_BoostVectorInverseCovariance = m_CollisionBoostVector->getBoostCovariance();
198 if (s_BoostVectorInverseCovariance.Determinant() == 0) {
199 B2WARNING("Determinant of boost covariance matrix is 0, "
200 "using generic inverse covariance matrix for fit.");
201 s_BoostVectorInverseCovariance[0][0] = 1.0 / (m_BoostError * m_BoostError);
202 s_BoostVectorInverseCovariance[0][1] = 0;
203 s_BoostVectorInverseCovariance[0][2] = 0;
204 s_BoostVectorInverseCovariance[1][0] = 0;
205 s_BoostVectorInverseCovariance[1][1] = 1.0 / (m_BoostError * m_BoostError);
206 s_BoostVectorInverseCovariance[1][2] = 0;
207 s_BoostVectorInverseCovariance[2][0] = 0;
208 s_BoostVectorInverseCovariance[2][1] = 0;
209 s_BoostVectorInverseCovariance[2][2] = 1.0 / (m_BoostError * m_BoostError);
210 } else {
211 s_BoostVectorInverseCovariance.Invert();
212 }
213 s_InvariantMass = m_CollisionInvariantMass->getMass();
214 s_InvariantMassError = m_CollisionInvariantMass->getMassError();
215 if (s_InvariantMassError == 0) {
216 B2WARNING("Invariant-mass errror is 0, using generic error for fit.");
217 s_InvariantMassError = m_InvariantMassError;
218 }
219 s_DirectionHER.SetX(0);
220 s_DirectionHER.SetY(0);
221 s_DirectionHER.SetZ(1);
222 s_DirectionHER.RotateY(m_AngleHER);
223 s_DirectionLER.SetX(0);
224 s_DirectionLER.SetY(0);
225 s_DirectionLER.SetZ(1);
226 s_DirectionLER.RotateY(m_AngleLER + M_PI);
227 s_AngleError = m_AngleError;
228 TMinuit minuit(6);
229 if (!m_Verbose)
230 minuit.SetPrintLevel(-1);
231 minuit.SetFCN(fcn);
232 minuit.mnparm(0, "PHER_E", 7, 0.01, 0, 0, minuitResult);
233 minuit.mnparm(1, "PHER_TX", 0, 0.01, 0, 0, minuitResult);
234 minuit.mnparm(2, "PHER_TY", 0, 0.01, 0, 0, minuitResult);
235 minuit.mnparm(3, "PLER_E", 4, 0.01, 0, 0, minuitResult);
236 minuit.mnparm(4, "PLER_TX", 0, 0.01, 0, 0, minuitResult);
237 minuit.mnparm(5, "PLER_TY", 0, 0.01, 0, 0, minuitResult);
238 minuit.mncomd("FIX 2 3 5 6", minuitResult);
239 minuit.mncomd("MIGRAD 10000", minuitResult);
240 minuit.mncomd("RELEASE 2 3 5 6", minuitResult);
241 minuit.mncomd("MIGRAD 10000", minuitResult);
242 double error;
243 minuit.GetParameter(0, herMomentum, error);
244 minuit.GetParameter(1, herThetaX, error);
245 minuit.GetParameter(2, herThetaY, error);
246 minuit.GetParameter(3, lerMomentum, error);
247 minuit.GetParameter(4, lerThetaX, error);
248 minuit.GetParameter(5, lerThetaY, error);
249 /* Calculate error. */
250 ROOT::Math::PxPyPzEVector pHER = getMomentum(herMomentum, herThetaX, herThetaY, false);
251 ROOT::Math::PxPyPzEVector pLER = getMomentum(lerMomentum, lerThetaX, lerThetaY, true);
252 ROOT::Math::PxPyPzEVector pBeam = pHER + pLER;
253 double fittedInvariantMass = pBeam.M();
254 B2RESULT("Initial invariant mass: " << s_InvariantMass <<
255 "; fitted invariant mass: " << fittedInvariantMass);
256 double cosThetaX = cos(herThetaX);
257 double sinThetaX = sin(herThetaX);
258 double cosThetaY = cos(herThetaY);
259 double sinThetaY = sin(herThetaY);
260 double herPartial =
261 (pBeam.E() - pHER.E() / pHER.P() *
262 (pBeam.Px() * cosThetaX * sinThetaY - pBeam.Py() * sinThetaX +
263 pBeam.Pz() * cosThetaX * cosThetaY)) / fittedInvariantMass;
264 cosThetaX = cos(lerThetaX);
265 sinThetaX = sin(lerThetaX);
266 cosThetaY = cos(lerThetaY + M_PI);
267 sinThetaY = sin(lerThetaY + M_PI);
268 double lerPartial =
269 (pBeam.E() - pLER.E() / pLER.P() *
270 (pBeam.Px() * cosThetaX * sinThetaY - pBeam.Py() * sinThetaX +
271 pBeam.Pz() * cosThetaX * cosThetaY)) / fittedInvariantMass;
272 double sigmaInvariantMass = m_CollisionInvariantMass->getMassSpread();
273 double k = sqrt(sigmaInvariantMass * sigmaInvariantMass /
274 (square(herPartial * pHER.E()) + square(lerPartial * pLER.E())));
275 double herSpread = k * pHER.E();
276 double lerSpread = k * pLER.E();
277 B2INFO("Invariant mass spread: " << sigmaInvariantMass);
278 B2RESULT("HER energy spread: " << herSpread <<
279 "; LER energy spread: " << lerSpread);
280 /* Fill beam parameters. */
281 m_BeamParameters.setHER(pHER);
282 m_BeamParameters.setLER(pLER);
283 TMatrixDSym covariance(3);
284 for (int i = 0; i < 3; ++i) {
285 for (int j = 0; j < 3; ++j)
286 covariance[i][j] = 0;
287 }
288 covariance[0][0] = herSpread * herSpread;
289 m_BeamParameters.setCovHER(covariance);
290 covariance[0][0] = lerSpread * lerSpread;
291 m_BeamParameters.setCovLER(covariance);
292}
double m_InvariantMassError
Invariant-mass error (use only if error is 0).
DBObjPtr< CollisionBoostVector > m_CollisionBoostVector
Collision boost vector.
bool m_Verbose
Whether to be verbose (print Minuit output).
DBObjPtr< CollisionInvariantMass > m_CollisionInvariantMass
Collision invariant mass.
double m_BoostError
Boost error (use only if inverse error matrix is not available).
constexpr T square(const T &x)
Calculate the square of the input.
Definition MathHelpers.h:21

◆ getBeamParameters()

BeamParameters getBeamParameters ( ) const
inline

Get beam parameters.

Definition at line 108 of file BeamParametersFitter.h.

109 {
110 return m_BeamParameters;
111 }

◆ importBeamParameters()

void importBeamParameters ( )

Import beam parameters.

Definition at line 318 of file BeamParametersFitter.cc.

319{
320 DBImportObjPtr<BeamParameters> beamParametersImport;
321 beamParametersImport.construct(m_BeamParameters);
322 beamParametersImport.import(m_IntervalOfValidity);
323}
IntervalOfValidity m_IntervalOfValidity
Interval of validity.
bool import(const IntervalOfValidity &iov)
Import the object to database.
void construct(Args &&... params)
Construct an object of type T in this DBImportObjPtr using the provided constructor arguments.

◆ setAngleError()

void setAngleError ( double angleError)
inline

Set angle error.

Definition at line 76 of file BeamParametersFitter.h.

77 {
78 m_AngleError = angleError;
79 }

◆ setAngleHER()

void setAngleHER ( double angleHER)
inline

Set HER angle.

Definition at line 60 of file BeamParametersFitter.h.

61 {
62 m_AngleHER = angleHER;
63 }

◆ setAngleLER()

void setAngleLER ( double angleLER)
inline

Set LER angle.

Definition at line 68 of file BeamParametersFitter.h.

69 {
70 m_AngleLER = angleLER;
71 }

◆ setBeamParameters()

void setBeamParameters ( const BeamParameters * beamParameters)
inline

Set beam parameters.

Definition at line 116 of file BeamParametersFitter.h.

117 {
118 m_BeamParameters = *beamParameters;
119 }

◆ setBoostError()

void setBoostError ( double boostError)
inline

Set boost error.

Definition at line 84 of file BeamParametersFitter.h.

85 {
86 m_BoostError = boostError;
87 }

◆ setIntervalOfValidity()

void setIntervalOfValidity ( const IntervalOfValidity & iov)
inline

Set interval of validity.

Definition at line 52 of file BeamParametersFitter.h.

53 {
54 m_IntervalOfValidity = iov;
55 }

◆ setInvariantMassError()

void setInvariantMassError ( double invariantMassError)
inline

Set invariant-mass error.

Definition at line 92 of file BeamParametersFitter.h.

93 {
94 m_InvariantMassError = invariantMassError;
95 }

◆ setupDatabase()

void setupDatabase ( )
protected

Setup database.

Definition at line 96 of file BeamParametersFitter.cc.

97{
98 /* DataStore. */
100 StoreObjPtr<EventMetaData> eventMetaData;
101 eventMetaData.registerInDataStore();
103 /* Database. */
104 if (eventMetaData.isValid()) {
105 eventMetaData->setExperiment(m_IntervalOfValidity.getExperimentLow());
106 eventMetaData->setRun(m_IntervalOfValidity.getRunLow());
107 } else {
108 eventMetaData.construct(1, m_IntervalOfValidity.getRunLow(),
109 m_IntervalOfValidity.getExperimentLow());
110 }
111 DBStore& dbStore = DBStore::Instance();
112 dbStore.update();
113 dbStore.updateEvent();
114}
static DataStore & Instance()
Instance of singleton Store.
Definition DataStore.cc:53
void setInitializeActive(bool active)
Setter for m_initializeActive.
Definition DataStore.cc:93
bool registerInDataStore(DataStore::EStoreFlags storeFlags=DataStore::c_WriteOut)
Register the object/array in the DataStore.
bool isValid() const
Check whether the object was created.
bool construct(Args &&... params)
Construct an object of type T in this StoreObjPtr, using the provided constructor arguments.
static DBStore & Instance()
Instance of a singleton DBStore.
Definition DBStore.cc:26
void updateEvent()
Updates all intra-run dependent objects.
Definition DBStore.cc:140
void update()
Updates all objects that are outside their interval of validity.
Definition DBStore.cc:77

◆ setVerbose()

void setVerbose ( bool verbose)
inline

Set whether to be verbose (print Minuit output).

Definition at line 100 of file BeamParametersFitter.h.

101 {
102 m_Verbose = verbose;
103 }

Member Data Documentation

◆ m_AngleError

double m_AngleError = 0
protected

Angle error.

Definition at line 157 of file BeamParametersFitter.h.

◆ m_AngleHER

double m_AngleHER = 0
protected

HER angle.

Definition at line 151 of file BeamParametersFitter.h.

◆ m_AngleLER

double m_AngleLER = 0
protected

LER angle.

Definition at line 154 of file BeamParametersFitter.h.

◆ m_BeamParameters

BeamParameters m_BeamParameters
protected

Beam parameters.

Definition at line 169 of file BeamParametersFitter.h.

◆ m_BeamSpot

DBObjPtr<BeamSpot> m_BeamSpot
protected

Beam spot.

Definition at line 172 of file BeamParametersFitter.h.

◆ m_BoostError

double m_BoostError = 0.0001
protected

Boost error (use only if inverse error matrix is not available).

Definition at line 160 of file BeamParametersFitter.h.

◆ m_CollisionBoostVector

DBObjPtr<CollisionBoostVector> m_CollisionBoostVector
protected

Collision boost vector.

Definition at line 175 of file BeamParametersFitter.h.

◆ m_CollisionInvariantMass

DBObjPtr<CollisionInvariantMass> m_CollisionInvariantMass
protected

Collision invariant mass.

Definition at line 178 of file BeamParametersFitter.h.

◆ m_IntervalOfValidity

IntervalOfValidity m_IntervalOfValidity
protected

Interval of validity.

Definition at line 148 of file BeamParametersFitter.h.

◆ m_InvariantMassError

double m_InvariantMassError = 0.0001
protected

Invariant-mass error (use only if error is 0).

Definition at line 163 of file BeamParametersFitter.h.

◆ m_Verbose

bool m_Verbose = false
protected

Whether to be verbose (print Minuit output).

Definition at line 166 of file BeamParametersFitter.h.


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