Belle II Software development
VXDDedxPIDModule.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#include <framework/gearbox/Const.h>
13#include <framework/datastore/StoreArray.h>
14
15#include <reconstruction/dataobjects/VXDDedxTrack.h>
16#include <reconstruction/dataobjects/VXDDedxLikelihood.h>
17#include <reconstruction/dataobjects/DedxConstants.h>
18
19#include <mdst/dataobjects/Track.h>
20#include <mdst/dataobjects/MCParticle.h>
21#include <tracking/dataobjects/RecoTrack.h>
22#include <svd/dataobjects/SVDCluster.h>
23#include <pxd/dataobjects/PXDCluster.h>
24
25#include <framework/database/DBObjPtr.h>
26#include <svd/dbobjects/SVDdEdxPDFs.h>
27#include <pxd/dbobjects/PXDdEdxPDFs.h>
28
29#include <vector>
30
31
32namespace Belle2 {
37 class HelixHelper;
38
52 class VXDDedxPIDModule : public Module {
53
54 public:
55
58
60 virtual ~VXDDedxPIDModule();
61
63 virtual void initialize() override;
64
66 virtual void event() override;
67
69 virtual void terminate() override;
70
71 private:
72
74 void checkPDFs();
75
83 void calculateMeans(double& mean, double& truncatedMean, double& truncatedMeanErr, const std::vector<double>& dedx) const;
84
86 template <class HitClass> static double getTraversedLength(const HitClass* hit, const RecoTrack* recoTrack, double& p);
87
89 template <class HitClass> void saveSiHits(VXDDedxTrack* track, const std::vector<HitClass*>& hits,
90 const RecoTrack* recoTrack) const;
91
92 // module steering parameters
94 bool m_usePXD;
95 bool m_useSVD;
97
98 // required input
101
102 // optional input
106
107 // output
110
111 // PDF's for PID
114
115 // other
118
119 };
120
121} // Belle2 namespace
Class for accessing objects in the database.
Definition DBObjPtr.h:21
Helper class representing a helical track.
Definition HelixHelper.h:28
Module()
Constructor.
Definition Module.cc:30
This is the Reconstruction Event-Data Model Track.
Definition RecoTrack.h:79
Accessor to arrays stored in the data store.
Definition StoreArray.h:113
DBObjPtr< SVDdEdxPDFs > m_SVDDedxPDFs
SVD DB object for dedx vs.
void calculateMeans(double &mean, double &truncatedMean, double &truncatedMeanErr, const std::vector< double > &dedx) const
Save arithmetic and truncated mean for the 'dedx' values.
int m_trackID
counter for tracks in this event
virtual void initialize() override
Initialize the module.
StoreArray< SVDCluster > m_svdClusters
Optional array of SVDClusters.
bool m_usePXD
use PXD data for likelihood
virtual void event() override
This method is called for each event.
StoreArray< MCParticle > m_mcparticles
Optional array of MCParticles.
StoreArray< VXDDedxLikelihood > m_dedxLikelihoods
Output array of VXDDedxLikelihoods.
void checkPDFs()
Check the pdfs for consistency every time they change in the database.
virtual void terminate() override
End of the event processing.
DBObjPtr< PXDdEdxPDFs > m_PXDDedxPDFs
PXD DB object for dedx vs.
StoreArray< VXDDedxTrack > m_dedxTracks
Output array of VXDDedxTracks.
virtual ~VXDDedxPIDModule()
Destructor.
int m_eventID
counter for events
bool m_useSVD
use SVD data for likelihood
void saveSiHits(VXDDedxTrack *track, const std::vector< HitClass * > &hits, const RecoTrack *recoTrack) const
save energy loss and hit information from SVD/PXDHits to track
StoreArray< Track > m_tracks
Required array of Tracks.
bool m_onlyPrimaryParticles
For MC only: if true, only save data for primary particles (as determined by MC truth)
static double getTraversedLength(const HitClass *hit, const RecoTrack *recoTrack, double &p)
returns traversed length through active medium of given hit
StoreArray< PXDCluster > m_pxdClusters
Optional array of PXDClusters.
StoreArray< RecoTrack > m_recoTracks
Required array of RecoTracks.
bool m_useIndividualHits
use individual hits (true) or truncated mean (false) to determine likelihoods
VXDDedxPIDModule()
Default constructor.
Debug output for VXDDedxPID module.
Abstract base class for different kinds of events.