Analysis Software
Documentation for
sPHENIX
simulation software
Home page
Related Pages
Modules
Namespaces
Classes
Files
Examples
External Links
File List
File Members
All
Classes
Namespaces
Files
Functions
Variables
Typedefs
Enumerations
Enumerator
Friends
Macros
Groups
Pages
VertexFitterAlgorithm.hpp
Go to the documentation of this file.
Or view
the newest version in sPHENIX GitHub for file VertexFitterAlgorithm.hpp
1
// This file is part of the Acts project.
2
//
3
// Copyright (C) 2016-2019 CERN for the benefit of the Acts project
4
//
5
// This Source Code Form is subject to the terms of the Mozilla Public
6
// License, v. 2.0. If a copy of the MPL was not distributed with this
7
// file, You can obtain one at http://mozilla.org/MPL/2.0/.
8
9
#pragma once
10
11
#include "
Acts/Definitions/Algebra.hpp
"
12
#include "
Acts/Definitions/Direction.hpp
"
13
#include "
Acts/Definitions/Units.hpp
"
14
#include "
Acts/EventData/TrackParameters.hpp
"
15
#include "
Acts/Geometry/GeometryIdentifier.hpp
"
16
#include "
Acts/MagneticField/MagneticFieldProvider.hpp
"
17
#include "
Acts/Propagator/EigenStepper.hpp
"
18
#include "
Acts/Propagator/Propagator.hpp
"
19
#include "
Acts/Utilities/Logger.hpp
"
20
#include "
Acts/Vertexing/FullBilloirVertexFitter.hpp
"
21
#include "
Acts/Vertexing/HelicalTrackLinearizer.hpp
"
22
#include "
Acts/Vertexing/Vertex.hpp
"
23
#include "
Acts/Vertexing/VertexingOptions.hpp
"
24
#include "
ActsExamples/EventData/ProtoVertex.hpp
"
25
#include "
ActsExamples/EventData/Track.hpp
"
26
#include "
ActsExamples/EventData/Trajectories.hpp
"
27
#include "
ActsExamples/Framework/DataHandle.hpp
"
28
#include "
ActsExamples/Framework/IAlgorithm.hpp
"
29
#include "
ActsExamples/Framework/ProcessCode.hpp
"
30
31
#include <algorithm>
32
#include <array>
33
#include <memory>
34
#include <string>
35
#include <tuple>
36
#include <utility>
37
#include <vector>
38
39
namespace
Acts {
40
class
MagneticFieldProvider;
41
}
// namespace Acts
42
43
namespace
ActsExamples {
44
struct
AlgorithmContext;
45
46
class
VertexFitterAlgorithm
final :
public
IAlgorithm
{
47
public
:
48
using
Propagator
=
Acts::Propagator<Acts::EigenStepper<>
>;
49
using
PropagatorOptions
=
Acts::PropagatorOptions<>
;
50
using
Linearizer
=
Acts::HelicalTrackLinearizer<Propagator>
;
51
using
VertexFitter
=
52
Acts::FullBilloirVertexFitter<Acts::BoundTrackParameters, Linearizer>
;
53
using
VertexFitterOptions
=
54
Acts::VertexingOptions<Acts::BoundTrackParameters>
;
55
56
using
VertexCollection
=
57
std::vector<Acts::Vertex<Acts::BoundTrackParameters>>;
58
59
struct
Config
{
61
std::string
inputTrackParameters
;
63
std::string
inputTrajectories
;
65
std::string
inputProtoVertices
;
67
std::string
outputVertices
;
69
std::shared_ptr<Acts::MagneticFieldProvider>
bField
;
71
bool
doConstrainedFit
=
false
;
73
Acts::Vector4
constraintPos
=
Acts::Vector4
(0, 0, 0, 0);
75
Acts::SquareMatrix4
constraintCov
=
76
Acts::Vector4
(3 *
Acts::UnitConstants::mm
*
Acts::UnitConstants::mm
,
77
3 * Acts::UnitConstants::mm * Acts::UnitConstants::mm,
78
10 * Acts::UnitConstants::mm * Acts::UnitConstants::mm,
79
1 *
Acts::UnitConstants::ns
*
Acts::UnitConstants::ns
)
80
.asDiagonal();
81
};
82
83
VertexFitterAlgorithm
(
const
Config
&
cfg
,
Acts::Logging::Level
lvl);
84
89
ProcessCode
execute
(
const
AlgorithmContext
& ctx)
const
final
;
90
92
const
Config
&
config
()
const
{
return
m_cfg
; }
93
94
private
:
95
Config
m_cfg
;
96
97
ReadDataHandle<std::vector<Acts::BoundTrackParameters>
>
98
m_inputTrackParameters
{
this
,
"InputTrackParameters"
};
99
100
ReadDataHandle<TrajectoriesContainer>
m_inputTrajectories
{
101
this
,
"InputTrajectories"
};
102
103
ReadDataHandle<ProtoVertexContainer>
m_inputProtoVertices
{
104
this
,
"InputProtoVertices"
};
105
106
WriteDataHandle<VertexCollection>
m_outputVertices
{
this
,
"OutputVertices"
};
107
};
108
109
}
// namespace ActsExamples
acts
blob
sPHENIX
Examples
Algorithms
Vertexing
include
ActsExamples
Vertexing
VertexFitterAlgorithm.hpp
Built by
Jin Huang
. updated:
Sat Feb 17 2024 22:17:37
using
1.8.2 with
sPHENIX GitHub integration