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).
 
const BeamParametersgetBeamParameters () 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 296 of file BeamParametersFitter.cc.

298{
300 m_BeamParameters.setVertex(ROOT::Math::XYZVector(m_BeamSpot->getIPPosition()));
301 TMatrixDSym beamSize = m_BeamSpot->getSizeCovMatrix();
302 double xScale, yScale;
303 if (covarianceXX < 0)
304 xScale = 1;
305 else
306 xScale = sqrt(covarianceXX / beamSize[0][0]);
307 if (covarianceYY < 0)
308 yScale = 1;
309 else
310 yScale = sqrt(covarianceYY / beamSize[1][1]);
311 for (int i = 0; i < 3; ++i) {
312 beamSize[0][i] *= xScale;
313 beamSize[i][0] *= xScale;
314 beamSize[1][i] *= yScale;
315 beamSize[i][1] *= yScale;
316 }
317 m_BeamParameters.setCovVertex(beamSize);
318}
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 191 of file BeamParametersFitter.cc.

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

const 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 320 of file BeamParametersFitter.cc.

321{
322 DBImportObjPtr<BeamParameters> beamParametersImport;
323 beamParametersImport.construct(m_BeamParameters);
324 beamParametersImport.import(m_IntervalOfValidity);
325}
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 98 of file BeamParametersFitter.cc.

99{
100 /* DataStore. */
102 StoreObjPtr<EventMetaData> eventMetaData;
103 eventMetaData.registerInDataStore();
105 /* Database. */
106 if (eventMetaData.isValid()) {
107 eventMetaData->setExperiment(m_IntervalOfValidity.getExperimentLow());
108 eventMetaData->setRun(m_IntervalOfValidity.getRunLow());
109 } else {
110 eventMetaData.construct(1, m_IntervalOfValidity.getRunLow(),
111 m_IntervalOfValidity.getExperimentLow());
112 }
113 DBStore& dbStore = DBStore::Instance();
114 dbStore.update();
115 dbStore.updateEvent();
116}
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: