feat: Update ITk seeding example (#1278)

* updates to itk_seeding.py * more updates * updates * update itk seeding py * clang format * clang format
acts-project · Jun 28, 2022 · e5e62f3 · e5e62f3
1 parent a6fa2ba
commit e5e62f3
Show file tree

Hide file tree

Showing 5 changed files with 234 additions and 69 deletions.
diff --git a/Core/include/Acts/Seeding/BinnedSPGroup.hpp b/Core/include/Acts/Seeding/BinnedSPGroup.hpp
@@ -194,7 +194,7 @@ class BinnedSPGroupIterator {
   BinnedSPGroupIterator(const SpacePointGrid<external_spacepoint_t>* spgrid,
                         BinFinder<external_spacepoint_t>* botBinFinder,
                         BinFinder<external_spacepoint_t>* tBinFinder,
-                        boost::container::small_vector<size_t, 20> bins = {}) {
+                        std::vector<size_t> bins = {}) {
     grid = spgrid;
     m_bottomBinFinder = botBinFinder;
     m_topBinFinder = tBinFinder;
@@ -216,7 +216,7 @@ class BinnedSPGroupIterator {
                         BinFinder<external_spacepoint_t>* botBinFinder,
                         BinFinder<external_spacepoint_t>* tBinFinder,
                         size_t phiInd, size_t zInd,
-                        boost::container::small_vector<size_t, 20> bins = {}) {
+                        std::vector<size_t> bins = {}) {
     m_bottomBinFinder = botBinFinder;
     m_topBinFinder = tBinFinder;
     grid = spgrid;
@@ -252,7 +252,7 @@ class BinnedSPGroupIterator {
   std::array<long unsigned int, 2ul> phiZbins;
   BinFinder<external_spacepoint_t>* m_bottomBinFinder;
   BinFinder<external_spacepoint_t>* m_topBinFinder;
-  boost::container::small_vector<size_t, 20> customZorder;
+  std::vector<size_t> customZorder;
   // 	bool start = true;
 };
 
@@ -298,7 +298,7 @@ class BinnedSPGroup {
   std::shared_ptr<BinFinder<external_spacepoint_t>> m_topBinFinder;
   std::shared_ptr<BinFinder<external_spacepoint_t>> m_bottomBinFinder;
 
-  boost::container::small_vector<size_t, 20> m_bins;
+  std::vector<size_t> m_bins;
 };
 
 }  // namespace Acts

diff --git a/Core/include/Acts/Seeding/SeedfinderConfig.hpp b/Core/include/Acts/Seeding/SeedfinderConfig.hpp
@@ -122,7 +122,7 @@ struct SeedfinderConfig {
   Acts::Vector2 beamPos{0 * Acts::UnitConstants::mm,
                         0 * Acts::UnitConstants::mm};
 
-  boost::container::small_vector<size_t, 20> zBinsCustomLooping = {};
+  std::vector<size_t> zBinsCustomLooping = {};
 
   // average radiation lengths of material on the length of a seed. used for
   // scattering.

diff --git a/Examples/Io/Csv/src/CsvSpacePointReader.cpp b/Examples/Io/Csv/src/CsvSpacePointReader.cpp
@@ -97,6 +97,7 @@ ActsExamples::ProcessCode ActsExamples::CsvSpacePointReader::read(
   ACTS_DEBUG("Created " << spacePoints.size() << " " << m_cfg.inputCollection
                         << " space points");
   ctx.eventStore.add("PixelSpacePoints", std::move(spacePoints));
+  ctx.eventStore.add("StripSpacePoints", std::move(spacePoints));
 
   return ProcessCode::SUCCESS;
 }
diff --git a/Examples/Python/src/TrackFinding.cpp b/Examples/Python/src/TrackFinding.cpp
@@ -123,9 +123,12 @@ void addTrackFinding(Context& ctx) {
     ACTS_PYTHON_MEMBER(useVariableMiddleSPRange);
     ACTS_PYTHON_MEMBER(deltaRMiddleMinSPRange);
     ACTS_PYTHON_MEMBER(deltaRMiddleMaxSPRange);
+    ACTS_PYTHON_MEMBER(binSizeR);
+    ACTS_PYTHON_MEMBER(forceRadialSorting);
     ACTS_PYTHON_MEMBER(seedConfirmation);
     ACTS_PYTHON_MEMBER(centralSeedConfirmationRange);
     ACTS_PYTHON_MEMBER(forwardSeedConfirmationRange);
+    ACTS_PYTHON_MEMBER(arithmeticAverageCotTheta);
     ACTS_PYTHON_MEMBER(useDetailedDoubleMeasurementInfo);
     ACTS_PYTHON_STRUCT_END();
     patchKwargsConstructor(c);