From 531e23cc7d2775d10d17e424b20c9cfae45a38f2 Mon Sep 17 00:00:00 2001
From: Eamonn <glynneamonn@gmail.com>
Date: Sun, 14 Jul 2024 15:56:44 -0400
Subject: [PATCH] added () operator to matrix

---
 src/pmpo_MPMesh_assembly.hpp | 5 ++++-
 src/pmpo_utils.hpp           | 3 +++
 2 files changed, 7 insertions(+), 1 deletion(-)

diff --git a/src/pmpo_MPMesh_assembly.hpp b/src/pmpo_MPMesh_assembly.hpp
index 429c89e..8137179 100644
--- 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
index 1498b78..2fdb02b 100644
--- 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];}
+  
 
 };