Rework MP surface model

ViennaTools · Oct 28, 2024 · fc6eaa4 · fc6eaa4
1 parent 44ea4db
commit fc6eaa4
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/include/viennaps/models/psMultiParticleProcess.hpp b/include/viennaps/models/psMultiParticleProcess.hpp
@@ -35,10 +35,15 @@ class MultiParticleSurfaceModel : public viennaps::SurfaceModel<NumericType> {
     auto velocity =
         SmartPointer<std::vector<NumericType>>::New(materialIds.size(), 0.);
 
+    std::vector<std::vector<NumericType> *> fluxPtrs;
+    for (const auto &label : fluxDataLabels_) {
+      fluxPtrs.push_back(rates->getScalarData(label));
+    }
+
+    std::vector<NumericType> fluxes(fluxPtrs.size());
     for (std::size_t i = 0; i < velocity->size(); i++) {
-      std::vector<NumericType> fluxes;
-      for (auto &fluxDataLabel : fluxDataLabels_) {
-        fluxes.push_back(rates->getScalarData(fluxDataLabel)->at(i));
+      for (std::size_t j = 0; j < fluxPtrs.size(); j++) {
+        fluxes[j] = fluxPtrs[j]->at(i);
       }
       velocity->at(i) =
           rateFunction_(fluxes, MaterialMap::mapToMaterial(materialIds[i]));