-
Notifications
You must be signed in to change notification settings - Fork 4.3k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Co-authored-by: Manos Vourliotis <[email protected]> Co-authored-by: Slava Krutelyov <[email protected]>
- Loading branch information
1 parent
ca8271a
commit 891eb11
Showing
27 changed files
with
1,290 additions
and
6 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,4 @@ | ||
import FWCore.ParameterSet.Config as cms | ||
|
||
# This modifier sets the iterative tracking to use a minimal set of iterations, first two | ||
trackingIters01 = cms.Modifier() |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,5 @@ | ||
import FWCore.ParameterSet.Config as cms | ||
|
||
# This modifier sets the LST (Phase-2 line segment tracking) used for track building | ||
trackingLST = cms.Modifier() | ||
|
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,10 @@ | ||
<use name="DataFormats/Common"/> | ||
<use name="alpaka"/> | ||
<use name="DataFormats/TrackerRecHit2D"/> | ||
<use name="HeterogeneousCore/AlpakaInterface"/> | ||
<use name="RecoTracker/LSTCore"/> | ||
<flags CXXFLAGS="-DLST_IS_CMSSW_PACKAGE"/> | ||
<flags ALPAKA_BACKENDS="1"/> | ||
<export> | ||
<lib name="1"/> | ||
</export> |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,36 @@ | ||
#ifndef RecoTracker_LST_LSTOutput_h | ||
#define RecoTracker_LST_LSTOutput_h | ||
|
||
#include <memory> | ||
#include <vector> | ||
|
||
class LSTOutput { | ||
public: | ||
LSTOutput() = default; | ||
~LSTOutput() = default; | ||
|
||
enum LSTTCType { T5 = 4, pT3 = 5, pT5 = 7, pLS = 8 }; | ||
|
||
void setLSTOutputTraits(std::vector<std::vector<unsigned int>> hitIdx, | ||
std::vector<unsigned int> len, | ||
std::vector<int> seedIdx, | ||
std::vector<short> trackCandidateType) { | ||
hitIdx_ = hitIdx; | ||
len_ = len; | ||
seedIdx_ = seedIdx; | ||
trackCandidateType_ = trackCandidateType; | ||
} | ||
|
||
std::vector<std::vector<unsigned int>> const& hitIdx() const { return hitIdx_; } | ||
std::vector<unsigned int> const& len() const { return len_; } | ||
std::vector<int> const& seedIdx() const { return seedIdx_; } | ||
std::vector<short> const& trackCandidateType() const { return trackCandidateType_; } | ||
|
||
private: | ||
std::vector<std::vector<unsigned int>> hitIdx_; | ||
std::vector<unsigned int> len_; | ||
std::vector<int> seedIdx_; | ||
std::vector<short> trackCandidateType_; | ||
}; | ||
|
||
#endif |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,40 @@ | ||
#ifndef RecoTracker_LST_LSTPhase2OTHitsInput_h | ||
#define RecoTracker_LST_LSTPhase2OTHitsInput_h | ||
|
||
#include <memory> | ||
#include <vector> | ||
|
||
#include "DataFormats/TrackerRecHit2D/interface/Phase2TrackerRecHit1D.h" | ||
|
||
class LSTPhase2OTHitsInput { | ||
public: | ||
LSTPhase2OTHitsInput() = default; | ||
~LSTPhase2OTHitsInput() = default; | ||
|
||
void setLSTPhase2OTHitsTraits(std::vector<unsigned int> detId, | ||
std::vector<float> x, | ||
std::vector<float> y, | ||
std::vector<float> z, | ||
std::vector<TrackingRecHit const*> hits) { | ||
detId_ = detId; | ||
x_ = x; | ||
y_ = y; | ||
z_ = z; | ||
hits_ = hits; | ||
} | ||
|
||
std::vector<unsigned int> const& detId() const { return detId_; } | ||
std::vector<float> const& x() const { return x_; } | ||
std::vector<float> const& y() const { return y_; } | ||
std::vector<float> const& z() const { return z_; } | ||
std::vector<TrackingRecHit const*> const& hits() const { return hits_; } | ||
|
||
private: | ||
std::vector<unsigned int> detId_; | ||
std::vector<float> x_; | ||
std::vector<float> y_; | ||
std::vector<float> z_; | ||
std::vector<TrackingRecHit const*> hits_; | ||
}; | ||
|
||
#endif |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,78 @@ | ||
#ifndef RecoTracker_LST_LSTPixelSeedInput_h | ||
#define RecoTracker_LST_LSTPixelSeedInput_h | ||
|
||
#include <memory> | ||
#include <vector> | ||
|
||
class LSTPixelSeedInput { | ||
public: | ||
LSTPixelSeedInput() = default; | ||
~LSTPixelSeedInput() = default; | ||
|
||
void setLSTPixelSeedTraits(std::vector<float> px, | ||
std::vector<float> py, | ||
std::vector<float> pz, | ||
std::vector<float> dxy, | ||
std::vector<float> dz, | ||
std::vector<float> ptErr, | ||
std::vector<float> etaErr, | ||
std::vector<float> stateTrajGlbX, | ||
std::vector<float> stateTrajGlbY, | ||
std::vector<float> stateTrajGlbZ, | ||
std::vector<float> stateTrajGlbPx, | ||
std::vector<float> stateTrajGlbPy, | ||
std::vector<float> stateTrajGlbPz, | ||
std::vector<int> q, | ||
std::vector<std::vector<int>> hitIdx) { | ||
px_ = px; | ||
py_ = py; | ||
pz_ = pz; | ||
dxy_ = dxy; | ||
dz_ = dz; | ||
ptErr_ = ptErr; | ||
etaErr_ = etaErr; | ||
stateTrajGlbX_ = stateTrajGlbX; | ||
stateTrajGlbY_ = stateTrajGlbY; | ||
stateTrajGlbZ_ = stateTrajGlbZ; | ||
stateTrajGlbPx_ = stateTrajGlbPx; | ||
stateTrajGlbPy_ = stateTrajGlbPy; | ||
stateTrajGlbPz_ = stateTrajGlbPz; | ||
q_ = q; | ||
hitIdx_ = hitIdx; | ||
} | ||
|
||
std::vector<float> const& px() const { return px_; } | ||
std::vector<float> const& py() const { return py_; } | ||
std::vector<float> const& pz() const { return pz_; } | ||
std::vector<float> const& dxy() const { return dxy_; } | ||
std::vector<float> const& dz() const { return dz_; } | ||
std::vector<float> const& ptErr() const { return ptErr_; } | ||
std::vector<float> const& etaErr() const { return etaErr_; } | ||
std::vector<float> const& stateTrajGlbX() const { return stateTrajGlbX_; } | ||
std::vector<float> const& stateTrajGlbY() const { return stateTrajGlbY_; } | ||
std::vector<float> const& stateTrajGlbZ() const { return stateTrajGlbZ_; } | ||
std::vector<float> const& stateTrajGlbPx() const { return stateTrajGlbPx_; } | ||
std::vector<float> const& stateTrajGlbPy() const { return stateTrajGlbPy_; } | ||
std::vector<float> const& stateTrajGlbPz() const { return stateTrajGlbPz_; } | ||
std::vector<int> const& q() const { return q_; } | ||
std::vector<std::vector<int>> const& hitIdx() const { return hitIdx_; } | ||
|
||
private: | ||
std::vector<float> px_; | ||
std::vector<float> py_; | ||
std::vector<float> pz_; | ||
std::vector<float> dxy_; | ||
std::vector<float> dz_; | ||
std::vector<float> ptErr_; | ||
std::vector<float> etaErr_; | ||
std::vector<float> stateTrajGlbX_; | ||
std::vector<float> stateTrajGlbY_; | ||
std::vector<float> stateTrajGlbZ_; | ||
std::vector<float> stateTrajGlbPx_; | ||
std::vector<float> stateTrajGlbPy_; | ||
std::vector<float> stateTrajGlbPz_; | ||
std::vector<int> q_; | ||
std::vector<std::vector<int>> hitIdx_; | ||
}; | ||
|
||
#endif |
Oops, something went wrong.