Removed one hack from grid_impl::search()

Fixed the navigator and propagator code accordingly

Author: Vakho Tsulaia

core/include/detray/navigation/navigator.hpp

@@ -674,7 +674,8 @@ class navigator {
     /// @return a heartbeat to indicate if the navigation is still alive
     template <typename propagator_state_t>
     DETRAY_HOST_DEVICE inline bool update(propagator_state_t &propagation,
-                                          const navigation::config &cfg) const {
+                                          const navigation::config &cfg,
+					  const context_type& ctx = {}) const {
 
         state &navigation = propagation._navigation;
 
@@ -707,7 +708,7 @@ class navigator {
             // navigation.run_inspector(cfg, track.pos(), track.dir(), "Volume
             // switch: ");
 
-            init(propagation, cfg);
+            init(propagation, cfg, ctx);
 
             // Fresh initialization, reset trust and hearbeat
             navigation.m_trust_level = navigation::trust_level::e_full;
@@ -717,7 +718,7 @@ class navigator {
         }
         // If no trust could be restored for the current state, (local)
         // navigation might be exhausted: re-initialize volume
-        navigation.m_heartbeat &= init(propagation, cfg);
+        navigation.m_heartbeat &= init(propagation, cfg, ctx);
 
         // Sanity check: Should never be the case after complete update call
         if (navigation.trust_level() != navigation::trust_level::e_full ||
@@ -738,7 +739,8 @@ class navigator {
     /// @param propagation contains the stepper and navigator states
     template <typename propagator_state_t>
     DETRAY_HOST_DEVICE inline void update_kernel(
-        propagator_state_t &propagation, const navigation::config &cfg) const {
+	propagator_state_t &propagation, const navigation::config &cfg,
+	const context_type& ctx = {}) const {
 
         state &navigation = propagation._navigation;
         const auto &det = navigation.detector();
@@ -819,7 +821,7 @@ class navigator {
         // Actor flagged cache as broken (other cases of 'no trust' are
         // handeled after volume switch was checked in 'update()')
         if (navigation.trust_level() == navigation::trust_level::e_no_trust) {
-            navigation.m_heartbeat &= init(propagation, cfg);
+	  navigation.m_heartbeat &= init(propagation, cfg, ctx);
             return;
         }
     }

core/include/detray/propagator/propagator.hpp

@@ -156,7 +156,7 @@ struct propagator {
 
             // Find next candidate
             propagation._heartbeat &=
-                m_navigator.update(propagation, m_cfg.navigation);
+	        m_navigator.update(propagation, m_cfg.navigation, m_cfg.context);
 
             // Run all registered actors/aborters after update
             run_actors(actor_state_refs, propagation);

core/include/detray/utils/grid/grid.hpp

@@ -273,12 +273,10 @@ class grid_impl {
     template <typename detector_t, typename track_t, typename config_t>
     DETRAY_HOST_DEVICE auto search(
         const detector_t &det, const typename detector_t::volume_type &volume,
-        const track_t &track, const config_t &cfg, const typename detector_t::geometry_context& /*ctx*/) const {
+        const track_t &track, const config_t &cfg, const typename detector_t::geometry_context& ctx) const {
 
         // Track position in grid coordinates
-//        const auto &trf = det.transform_store().at(volume.transform(),ctx);
-        typename detector_t::geometry_context cttx{0}; // Hack!
-        const auto &trf = det.transform_store().at(volume.transform(),cttx);
+        const auto &trf = det.transform_store().at(volume.transform(),ctx);
         const auto loc_pos = project(trf, track.pos(), track.dir());
 
         // Grid lookup