Belle II Software development
B2BIIConvertMdstModule.h
1/**************************************************************************
2 * basf2 (Belle II Analysis Software Framework) *
3 * Author: The Belle II Collaboration *
4 * *
5 * See git log for contributors and copyright holders. *
6 * This file is licensed under LGPL-3.0, see LICENSE.md. *
7 **************************************************************************/
8
9#pragma once
10
11#include <framework/core/Module.h>
12
13#define BELLE_TARGET_H "belle-x86_64-unknown-linux-gnu-g++.h"
14#include "belle_legacy/panther/panther.h"
15#include "belle_legacy/panther/panther_group.h"
16
17// Belle objects (Panther tables)
18#include "belle_legacy/tables/belletdf.h"
19#include "belle_legacy/tables/mdst.h"
20
21#include "belle_legacy/helix/Helix.h"
22
23// Belle2 objects
24#include <mdst/dataobjects/V0.h>
25#include <mdst/dataobjects/ECLCluster.h>
26#include <mdst/dataobjects/KLMCluster.h>
27#include <analysis/dataobjects/Particle.h>
28#include <analysis/dataobjects/ParticleList.h>
29#include <mdst/dataobjects/MCParticle.h>
30#include <mdst/dataobjects/MCParticleGraph.h>
31#include <mdst/dataobjects/Track.h>
32#include <mdst/dataobjects/PIDLikelihood.h>
33#include <framework/dataobjects/EventExtraInfo.h>
34#include <b2bii/dataobjects/BelleTrkExtra.h>
35
36// Replace BeamParameters
37#include <mdst/dbobjects/BeamSpot.h>
38#include <mdst/dbobjects/CollisionBoostVector.h>
39#include <mdst/dbobjects/CollisionInvariantMass.h>
40#include <mdst/dbobjects/CollisionAxisCMS.h>
41
42#include <framework/datastore/StoreArray.h>
43#include <framework/database/DBObjPtr.h>
44#include <framework/gearbox/Const.h>
45
46#include "CLHEP/Vector/ThreeVector.h"
47#include "CLHEP/Vector/LorentzVector.h"
48#include "CLHEP/Matrix/SymMatrix.h"
49#include "CLHEP/Geometry/Point3D.h"
50
51typedef HepGeom::Point3D<double> HepPoint3D;
52
53#include <string>
54#include <map>
55
56//enable ACC conversion (needs externals >= v00-06-01)
57#define HAVE_KID_ACC
58
59//enable EID conversion (needs externals > v00-07-01)
60#define HAVE_EID
61
62// enable findKs (needs externals > v00-07-01)
63#define HAVE_FINDKS
64
65// enable nisKsFinder (needs externals > v00-07-01)
66#define HAVE_NISKSFINDER
67
68// enable goodLambda (needs externals >= v01-08-00)
69#define HAVE_GOODLAMBDA
70
71namespace Belle2 {
76
91
93
106
107 // Public functions
108 public:
109
113 virtual ~B2BIIConvertMdstModule() override;
114
116 virtual void initialize() override;
117
119 virtual void beginRun() override;
120 virtual void event() override;
121 virtual void endRun() override;
122 virtual void terminate() override;
123
124 // Data members
125 private:
126
130 void initializeDataStore();
131
133 // true = real data
134 // false = mc
136
139
141 // true = use 6x6 (position, momentum) covariance matrix
142 // false = use 5x5 (helix parameters) covariance matrix
144
147
150
152
154
156
158
160
168
171
172 //-----------------------------------------------------------------------------
173 // CONVERT TABLES
174 //-----------------------------------------------------------------------------
175
179 void convertRecTrgTable();
180
184 void convertEvtclsTable();
185
190
194 void convertMdstECLTable();
195
199 void convertMdstKLMTable();
200
205
210
214 void copyNbarFromGamma();
215
220
224 void convertMdstPi0Table();
225
231
233 void convertBeamEnergy();
234
236 void convertIPProfile(bool beginRun = false);
237
238 //-----------------------------------------------------------------------------
239 // CONVERT OBJECTS
240 //-----------------------------------------------------------------------------
241
245 void convertGenHepevtObject(const Belle::Gen_hepevt& genHepevt, MCParticleGraph::GraphParticle* mcParticle);
246
251 void convertMdstECLObject(const Belle::Mdst_ecl& ecl, const Belle::Mdst_ecl_aux& eclAux, ECLCluster* eclCluster);
252
256 double computeTrkMinDistanceBelle(ECLCluster* eclCluster);
257
262 void convertMdstKLMObject(const Belle::Mdst_klm_cluster& klm, KLMCluster* klmCluster);
263
268 void convertMdstChargedObject(const Belle::Mdst_charged& belleTrack, Track* track);
269
273 TrackFitResult createTrackFitResult(const CLHEP::HepLorentzVector& momentum,
274 const HepPoint3D& position,
275 const CLHEP::HepSymMatrix& error,
276 const short int charge,
277 const Const::ParticleType& pType,
278 const float pValue,
279 const uint64_t hitPatternCDCInitializer,
280 const uint32_t hitPatternVXDInitializer,
281 const uint16_t ndf);
282
286 int getHelixParameters(const Belle::Mdst_trk_fit& trk_fit,
287 const double mass,
288 const HepPoint3D& newPivot,
289 std::vector<float>& helixParams,
290 CLHEP::HepSymMatrix& error5x5,
291 CLHEP::HepLorentzVector& momentum,
292 HepPoint3D& position,
293 CLHEP::HepSymMatrix& error7x7,
294 const double dPhi = 0.0);
295
299 void convertHelix(Belle::Helix& helix, std::vector<float>& helixParams, CLHEP::HepSymMatrix& error5x5);
300
304 void convertHelix(const Belle::Mdst_trk_fit& trk_fit, const HepPoint3D& newPivot, std::vector<float>& helixParams,
305 std::vector<float>& helixError);
306
310 void belleVeeDaughterToCartesian(const Belle::Mdst_vee2& vee, const int charge, const Const::ParticleType& pType,
311 CLHEP::HepLorentzVector& momentum, HepPoint3D& position, CLHEP::HepSymMatrix& error);
312
316 void belleVeeDaughterHelix(const Belle::Mdst_vee2& vee, const int charge, std::vector<float>& helixParam,
317 std::vector<float>& helixError);
318
320 void convertPIDData(const Belle::Mdst_charged& belleTrack, const Track* track);
321
323 const static int c_nHyp = 5;
326
328 void setLikelihoods(PIDLikelihood* pid, Const::EDetector det, double likelihoods[c_nHyp], bool discard_allzero = false);
329
330#ifdef HAVE_KID_ACC
332 static double acc_pid(const Belle::Mdst_charged& chg, int idp);
334 double cdc_pid(const Belle::Mdst_charged& chg, int idp);
335#endif
336
338 double atcPID(const PIDLikelihood* pid, int sigHyp, int bkgHyp);
339
340 //-----------------------------------------------------------------------------
341 // RELATIONS
342 //-----------------------------------------------------------------------------
343
348
353
354 //-----------------------------------------------------------------------------
355 // MISC
356 //-----------------------------------------------------------------------------
357
361 int recoverMoreThan24bitIDHEP(int id);
362
366 void testMCRelation(const Belle::Gen_hepevt& belleMC, const MCParticle* mcP, const std::string& objectName);
367
370
372 std::map<int, int> genHepevtToMCParticle;
374 std::map<int, int> mdstEclToECLCluster;
376 std::map<int, int> mdstGammaToParticle;
378 std::map<int, int> mdstKlmToKLMCluster;
380 std::map<int, int> mdstKlongToParticle;
381
384
387
390
393
396
399
402
405
408
411
415
419
423
427
437 const int ERRMCONV[7] = {3, 4, 5, -1, 0, 1, 2};
438
440 const double BFIELD = 1.5;
441
443 const double KAPPA2OMEGA = 1.5 * TMath::C() * 1E-11;
444
445 };
446
448} // end namespace Belle2
R E
internal precision of FFTW codelets
void belleVeeDaughterHelix(const Belle::Mdst_vee2 &vee, const int charge, std::vector< float > &helixParam, std::vector< float > &helixError)
obtains the helix parameters of the vee daughters
double atcPID(const PIDLikelihood *pid, int sigHyp, int bkgHyp)
calculates atc_pid(3,1,5,sigHyp,bkgHyp).prob() from converted PIDLikelihood
void convertMdstGammaTable()
Reads all entries of Mdst_Gamma Panther table, creates a particle list 'gamma:mdst' and adds them to ...
std::map< int, int > mdstKlmToKLMCluster
map of Mdst_klm Panther IDs and corresponding KLMCluster StoreArray indices
void copyNbarFromGamma()
Copies Particles in 'gamma:mdst' with energy > 0.5 GeV to be anti-n0:mdst.
void convertMdstChargedTable()
Reads and converts all entries of Mdst_charged (Mdst_trk and Mdst_trk_fit) Panther table to Track (Tr...
void setTracksToECLClustersRelations()
Sets Track -> ECLCluster relations.
OptionalDBObjPtr< CollisionBoostVector > m_collisionBoostVectorDB
CollisionBoostVector for boost vector.
StoreObjPtr< EventExtraInfo > m_evtInfo
Event Extra Info.
const int ERRMCONV[7]
CONVERSION OF TRACK ERROR MATRIX ELEMENTS.
CollisionAxisCMS m_collisionAxisCMS
CollisionAxisCMS of the beam.
StoreArray< KLMCluster > m_klmClusters
KLM clusters.
void belleVeeDaughterToCartesian(const Belle::Mdst_vee2 &vee, const int charge, const Const::ParticleType &pType, CLHEP::HepLorentzVector &momentum, HepPoint3D &position, CLHEP::HepSymMatrix &error)
Fills 4-momentum, position and 7x7 error matrix from Belle Vee daughter.
@ c_GeneratorLevel
Match to generator-level particles.
virtual void initialize() override
Initialize the module.
void setLikelihoods(PIDLikelihood *pid, Const::EDetector det, double likelihoods[c_nHyp], bool discard_allzero=false)
Add given Belle likelihoods (not log-likelihoods, in Belle hypothesis order) for given detector to pi...
OptionalDBObjPtr< CollisionAxisCMS > m_collisionAxisCMSDB
CollisionAxisCMS.
StoreArray< V0 > m_v0s
V0-particles.
virtual void event() override
Called for each event.
void convertGenHepEvtTable()
Reads and converts all entries of Gen_hepevt Panther table to MCParticle dataobjects and adds them to...
double cdc_pid(const Belle::Mdst_charged &chg, int idp)
Returns CDC likelihood for given hypothesis idp.
void convertRecTrgTable()
Reads and converts m_final from rectrg_summary3.
void setKLMClustersRelations()
Sets KLMCluster -> Track and ECLCluster relations.
bool m_convertEvtcls
Flag to switch on conversion of Evtcls table.
const double KAPPA2OMEGA
Conversion factor for Kappa -> Omega helix parameters.
double computeTrkMinDistanceBelle(ECLCluster *eclCluster)
calculate the minimal distance between a cluster and a set of tracks on the ECL surface.
virtual void endRun() override
Called when the current run is finished.
bool m_nisEnable
Flag to switch on conversion of nisKsFinder info.
StoreArray< TrackFitResult > m_trackFitResults
Track fir results.
BeamSpot m_beamSpot
Interaction Point of the beam.
StoreArray< Particle > m_particles
Particles.
virtual void terminate() override
Terminates the module.
OptionalDBObjPtr< CollisionInvariantMass > m_collisionInvMDB
CollisionInvariantMass for Invariant Mass of Beam.
void convertHelix(Belle::Helix &helix, std::vector< float > &helixParams, CLHEP::HepSymMatrix &error5x5)
Converts Belle's Helix parameters and it's covariance matrix to Belle II's version.
void convertMdstPi0Table()
Reads all entries of Mdst_Pi0 Panther table, creates a particle list 'pi0:mdst' and adds them to Stor...
std::map< int, int > genHepevtToMCParticle
map of Gen_hepevt Panther IDs and corresponding MCParticle StoreArray indices
void testMCRelation(const Belle::Gen_hepevt &belleMC, const MCParticle *mcP, const std::string &objectName)
Checks if the reconstructed object (Track, ECLCluster, ...) was matched to the same MCParticle.
bool m_convertNbar
Flag to create anti-n0:mdst list from gamma:mdst.
void convertIPProfile(bool beginRun=false)
Stores the IPProfiles in BeamSpot (currently in DataStore)
void convertMdstECLObject(const Belle::Mdst_ecl &ecl, const Belle::Mdst_ecl_aux &eclAux, ECLCluster *eclCluster)
Converts Mdst_ecl(_aux) record to ECLCluster object.
CollisionInvariantMass m_collisionInvM
CollisionInvariantMass for the invariant mass of the beam.
bool m_convertBeamParameters
Convert beam parameters or use information stored in Belle II database.
static double acc_pid(const Belle::Mdst_charged &chg, int idp)
Returns ACC likelihood for given hypothesis idp.
virtual void beginRun() override
Module functions to be called from event process.
StoreArray< Track > m_tracks
Tracks.
void convertBeamEnergy()
Stores beam parameters (energy, angles) in CollisionInvariantMass and CollisionBoostVector (currently...
void convertPIDData(const Belle::Mdst_charged &belleTrack, const Track *track)
Get PID information for belleTrack and add it to PIDLikelihood (with relation from track).
void convertMdstKLMObject(const Belle::Mdst_klm_cluster &klm, KLMCluster *klmCluster)
Converts Mdst_klm_cluster record to KLMCluster object.
Belle2::MCParticleGraph m_particleGraph
MCParticle Graph to build Belle2 MC Particles.
void convertMdstKLongTable()
Reads all entries of Mdst_Klong Panther table, creates a particle list 'K_L0:mdst' and adds them to S...
OptionalDBObjPtr< BeamSpot > m_beamSpotDB
BeamSpot for IP.
bool m_realData
flag that tells whether given data sample is for real data or MC
StoreArray< ECLCluster > m_eclClusters
ECL clusters.
void convertGenHepevtObject(const Belle::Gen_hepevt &genHepevt, MCParticleGraph::GraphParticle *mcParticle)
Converts Gen_hepevt record to MCParticleGraph object.
void initializeDataStore()
Initializes Belle II DataStore.
void convertMdstECLTable()
Reads and converts all entries of Mdst_ecl(_aux) Panther table to ECLCluster dataobjects and adds the...
CollisionBoostVector m_collisionBoostVector
CollisionBoostVector for bosst vector of the beam.
std::string m_mcMatchingModeString
MC matching mode.
int m_lastIPProfileBin
variable to tell us which IPProfile bin was active last time we looked
void convertMdstKLMTable()
Reads and converts all entries of Mdst_klm_cluster Panther table to KLMCluster dataobjects and adds t...
bool m_convertRecTrg
Flag to switch on conversion of rectrg_summary3.
std::map< int, int > mdstGammaToParticle
map of gamma Panther IDs and corresponding Particle StoreArray indices
static const Const::ChargedStable c_belleHyp_to_chargedStable[c_nHyp]
maps Belle hypotheses to Const::ChargedStable (from http://belle.kek.jp/secured/wiki/doku....
std::map< int, int > mdstKlongToParticle
map of Klong Panther IDs and corresponding Particle StoreArray indices
double m_matchType2E9oE25Threshold
E9/E25 threshold value clusters with a value above this threshold are classified as neutral even if t...
void convertMdstChargedObject(const Belle::Mdst_charged &belleTrack, Track *track)
Converts Mdst_charged (Mdst_trk(_fit)) record to Track (TrackFitResult) object.
bool m_evtgenProcessing
Flag to switch on only evtgen processing.
StoreArray< MCParticle > m_mcParticles
MC particles.
void convertMdstVee2Table()
Reads and converts all entries of Mdst_vee2 Panther table to V0 dataobjects and adds them to StoreArr...
StoreArray< PIDLikelihood > m_pidLikelihoods
output PIDLikelihood array.
std::map< int, int > mdstEclToECLCluster
map of Mdst_ecl Panther IDs and corresponding ECLCluster StoreArray indices
TrackFitResult createTrackFitResult(const CLHEP::HepLorentzVector &momentum, const HepPoint3D &position, const CLHEP::HepSymMatrix &error, const short int charge, const Const::ParticleType &pType, const float pValue, const uint64_t hitPatternCDCInitializer, const uint32_t hitPatternVXDInitializer, const uint16_t ndf)
Creates TrackFitResult and fills it.
virtual ~B2BIIConvertMdstModule() override
Destructor.
const double BFIELD
B filed in TESLA.
MCMatchingMode m_mcMatchingMode
C matching mode.
StoreArray< BelleTrkExtra > m_belleTrkExtra
Belle CDC extra information.
bool m_use6x6CovarianceMatrix4Tracks
flag that tells which form of covariance matrix should be used in the conversion of charged tracks
int recoverMoreThan24bitIDHEP(int id)
Helper function to recover falsely set idhep info in GenHepEvt list.
bool m_convertTrkExtra
Flag to switch on conversion of first(last)_{x,y,z} of mdst_trk_fit.
int getHelixParameters(const Belle::Mdst_trk_fit &trk_fit, const double mass, const HepPoint3D &newPivot, std::vector< float > &helixParams, CLHEP::HepSymMatrix &error5x5, CLHEP::HepLorentzVector &momentum, HepPoint3D &position, CLHEP::HepSymMatrix &error7x7, const double dPhi=0.0)
Fills Helix parameters (converted to Belle II version), 5x5 error matrix, 4-momentum,...
static const int c_nHyp
Number of Belle track hypotheses (see c_belleHyp_to_chargedStable).
void convertEvtclsTable()
Reads and converts all entries of evtcls Panther table.
This class contains the beam spot position and size modeled as a gaussian distribution in space.
Definition BeamSpot.h:22
This class contains the measured values of the orientation of the collision axis in the CM system obt...
This class contains the measured average boost vector vec(beta) = (beta_x, beta_y,...
This class contains the measured average center-of-mass energy, which is equal to the invariant mass ...
Provides a type-safe way to pass members of the chargedStableSet set.
Definition Const.h:589
The ParticleType class for identifying different particle types.
Definition Const.h:408
EDetector
Enum for identifying the detector components (detector and subdetector).
Definition Const.h:42
ECL cluster data.
Definition ECLCluster.h:27
KLM cluster data.
Definition KLMCluster.h:29
Class to represent Particle data in graph.
Class to build, validate and sort a particle decay chain.
A Class to store the Monte Carlo particle information.
Definition MCParticle.h:32
Module()
Constructor.
Definition Module.cc:30
Optional DBObjPtr: This class behaves the same as the DBObjPtr except that it will not raise errors w...
Definition DBObjPtr.h:48
Class to collect log likelihoods from TOP, ARICH, dEdx, ECL and KLM aimed for output to mdst includes...
Accessor to arrays stored in the data store.
Definition StoreArray.h:113
Type-safe access to single objects in the data store.
Definition StoreObjPtr.h:96
Values of the result of a track fit with a given particle hypothesis.
Class that bundles various TrackFitResults.
Definition Track.h:25
Abstract base class for different kinds of events.