feat: boundary conditions

particle-track-and-trace/src/advection/kernel/AdvectionKernel.h

@@ -3,7 +3,7 @@
 
 #include <tuple>
 
-#include "Vel.h"
+#include "../Vel.h"
 
 /*
  * Implement this class for every integration method.

particle-track-and-trace/src/advection/kernel/EulerAdvectionKernel.cpp

@@ -1,6 +1,6 @@
 
 #include "EulerAdvectionKernel.h"
-#include "interpolate.h"
+#include "../interpolate.h"
 
 using namespace std;
 

particle-track-and-trace/src/advection/kernel/EulerAdvectionKernel.h

@@ -2,7 +2,7 @@
 #define EULERADVECTIONKERNEL_H
 
 #include "AdvectionKernel.h"
-#include "UVGrid.h"
+#include "../UVGrid.h"
 #include <memory>
 
 /**

particle-track-and-trace/src/advection/kernel/RK4AdvectionKernel.cpp

@@ -1,5 +1,5 @@
 #include "RK4AdvectionKernel.h"
-#include "interpolate.h"
+#include "../interpolate.h"
 
 using namespace std;
 

particle-track-and-trace/src/advection/kernel/RK4AdvectionKernel.h

@@ -2,7 +2,7 @@
 #define RK4ADVECTIONKERNEL_H
 
 #include "AdvectionKernel.h"
-#include "UVGrid.h"
+#include "../UVGrid.h"
 #include <memory>
 
 /**

particle-track-and-trace/src/advection/kernel/SnapBoundaryConditionKernel.cpp

@@ -0,0 +1,26 @@
+#include "SnapBoundaryConditionKernel.h"
+
+SnapBoundaryConditionKernel::SnapBoundaryConditionKernel(std::unique_ptr<AdvectionKernel> kernel,
+                                                         std::shared_ptr<UVGrid> uvGrid) :
+                                                         kernel(std::move(kernel)),
+                                                         uvGrid(uvGrid) { }
+std::pair<double, double> SnapBoundaryConditionKernel::advect(int time, double latitude, double longitude, int dt) const {
+  auto [newLat, newLon] = kernel->advect(time, latitude, longitude, dt);
+  double minLat = uvGrid->lats.front();
+  double maxLat = uvGrid->lats.back();
+  double minLon = uvGrid->lons.front();
+  double maxLon = uvGrid->lons.back();
+  if (newLat < minLat) {
+    newLat = minLat;
+  }
+  if (newLat > maxLat) {
+    newLat = maxLat;
+  }
+  if (newLon < minLon) {
+    newLon = minLon;
+  }
+  if (newLon > maxLon) {
+    newLon = maxLon;
+  }
+  return {newLat, newLon};
+}

particle-track-and-trace/src/advection/kernel/SnapBoundaryConditionKernel.h

@@ -0,0 +1,22 @@
+#ifndef SNAPBOUNDARYCONDITIONKERNEL_H
+#define SNAPBOUNDARYCONDITIONKERNEL_H
+
+#include <memory>
+#include "AdvectionKernel.h"
+#include "../UVGrid.h"
+
+/**
+ * When
+ */
+class SnapBoundaryConditionKernel: public AdvectionKernel {
+  std::unique_ptr<AdvectionKernel> kernel;
+  std::shared_ptr<UVGrid> uvGrid;
+public:
+  SnapBoundaryConditionKernel(std::unique_ptr<AdvectionKernel> kernel, std::shared_ptr<UVGrid> uvGrid);
+
+private:
+  std::pair<double, double> advect(int time, double latitude, double longitude, int dt) const override;
+};
+
+
+#endif //SNAPBOUNDARYCONDITIONKERNEL_H