1 files changed, 8 insertions, 0 deletions
diff --git a/src/arap.h b/src/arap.h
index ca8967c..c4a24d7 100644
--- a/src/arap.h
+++ b/src/arap.h
@@ -1,6 +1,7 @@
 #pragma once
 
 #include "graphics/shape.h"
+#include "graphics/oceanshape.h"
 #include "Eigen/StdList"
 #include "Eigen/StdVector"
 #include "ocean/ocean.h"
@@ -17,6 +18,7 @@ class ARAP
 {
 private:
     Shape m_shape;
+    OceanShape m_oceanShape;
 
 public:
     ARAP();
@@ -38,6 +40,10 @@ public:
         m_shape.draw(shader, mode);
     }
 
+    void initGroundPlane(std::string texturePath, float depth, Shader* shader) {
+        m_shape.initGroundPlane(texturePath, depth, shader);
+    }
+
     SelectMode select(Shader *shader, int vertex)
     {
         return m_shape.select(shader, vertex);
@@ -72,5 +78,7 @@ public:
     ocean m_ocean;
     double m_time = 0.00;
     double m_timestep = 0.001;
+
+    Eigen::Vector3f minCorner, maxCorner;
 };