added () operator to matrix

SCOREC · Jul 14, 2024 · 531e23c · 531e23c
1 parent b33651f
commit 531e23c
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/src/pmpo_MPMesh_assembly.hpp b/src/pmpo_MPMesh_assembly.hpp
@@ -117,7 +117,10 @@ void MPMesh::assemblyVtx1() {
         int vID = elm2VtxConn(elm,i+1)-1; //vID = vertex id
         for(int j=0;j<numEntries;j++){
           double fieldComponentVal = mpData(mp,j) * weight(mp, 0);
-          double[4] CoordDiffs = {1, vtxCoords(vID,0) - mpPositions(mp,0), vtxCoords(vID,1) - mpPositions(mp,1), vtxCoords(vID,2) - mpPositions(mp,2)};
+          double[4] CoordDiffs = {1, 
+                                  vtxCoords(vID,0) - mpPositions(mp,0), 
+                                  vtxCoords(vID,1) - mpPositions(mp,1),
+                                  vtxCoords(vID,2) - mpPositions(mp,2)};
           //add to the matrix
           for(int k=0;k<4;k++){
             for(int l=0;l<4;l++){

diff --git a/src/pmpo_utils.hpp b/src/pmpo_utils.hpp
@@ -263,6 +263,9 @@ class Matrix {
     return x;
   }
 
+  KOKKOS_INLINE_FUNCTION
+  double operator()(int i, int j) const { return data_[i][j];}
+
 
 };