[src] Fix AdaptiveBeamMapping mapping at init with none deployed beam

epernod · Jul 17, 2024 · 5ae17a9 · 5ae17a9
1 parent 9e2dbec
commit 5ae17a9
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/src/BeamAdapter/component/mapping/AdaptiveBeamMapping.inl b/src/BeamAdapter/component/mapping/AdaptiveBeamMapping.inl
@@ -102,6 +102,8 @@ void AdaptiveBeamMapping< TIn, TOut>::init()
     {
         simulation::MainTaskSchedulerFactory::createInRegistry()->init();
     }
+
+    this->d_componentState.setValue(sofa::core::objectmodel::ComponentState::Valid);
 }
 
 
@@ -273,6 +275,14 @@ void AdaptiveBeamMapping< TIn, TOut>::apply(const MechanicalParams* mparams, Dat
 
     };
 
+
+    // HACK for init: In case the number of output points is bigger to the number of distribution, set all points to 0
+    for (int i = m_pointBeamDistribution.size(); i < d_points.getValue().size(); i++)
+    {
+        out[i] = Vec<3, InReal>(0.0, 0.0, 0.0);
+    }
+
+
     const bool multithreading = d_parallelMapping.getValue();
 
     const simulation::ForEachExecutionPolicy execution = multithreading ?