Disp increment in polyMPO

Author: dhyan1272

src/pmpo_mesh.cpp

@@ -190,22 +190,34 @@ namespace polyMPO{
     });
   }
 
+
   void Mesh::aggregateDeluDyn(){
     std::cout<<__FUNCTION__<<std::endl;
 
     int numVtx = getNumVertices();
+    auto elasticTimeStep = getElasticTimeStep(); 
 
-    constexpr MeshFieldIndex mfIndex1 = MeshF_OnSurfDispIncr; 
-    auto meshField1 = getMeshField<mfIndex1>();
+    constexpr MeshFieldIndex mfIndex_surfDispIncr = MeshF_OnSurfDispIncr; 
+    auto meshField_surfDispIncr = getMeshField<mfIndex_surfDispIncr>();
 
-    constexpr MeshFieldIndex mfIndex2 = MeshF_TanLatVertexRotatedOverRadius;
-    auto meshField2 = getMeshField<mfIndex2>();
-
-    constexpr MeshFieldIndex mfIndex3 = MeshF_Vel;
-    auto meshField3 = getMeshField<mfIndex3>();
+    constexpr MeshFieldIndex mfIndex_metric = MeshF_TanLatVertexRotatedOverRadius;
+    auto meshField_metric = getMeshField<mfIndex_metric>();
+
+    constexpr MeshFieldIndex mfIndex_vel = MeshF_Vel;
+    auto meshFieldVel = getMeshField<mfIndex_vel>();
+
+    Kokkos::parallel_for("update_velocity", numVtx, KOKKOS_LAMBDA(int vtx) {
+      auto u =  meshFieldVel(vtx, 0);
+      auto v =  meshFieldVel(vtx, 1);
+       
+      auto Del = elasticTimeStep * u * meshField_metric(vtx, 0);
+      meshFieldVel(vtx, 0) =  cos(Del) * u + sin(Del) * v ;
+      meshFieldVel(vtx, 1) = -sin(Del) * u + cos(Del) * v ;    
+    });
 
-    Kokkos::parallel_for("calcVelIncr", numVtx, KOKKOS_LAMBDA(int i) {
-              
+    Kokkos::parallel_for("calcVelIncr", numVtx, KOKKOS_LAMBDA(int vtx) {
+      meshField_surfDispIncr(vtx, 0) = meshField_surfDispIncr(vtx, 0) + meshFieldVel(vtx, 0);
+      meshField_surfDispIncr(vtx, 1) = meshField_surfDispIncr(vtx, 1) + meshFieldVel(vtx, 1);
     });
   }