initial rough working

vtk/src/layers/EGlyphLayer.cpp

@@ -11,43 +11,20 @@
 #include <vtkProperty.h>
 #include <vtkProperty2D.h>
 #include <vtkVertexGlyphFilter.h>
-#include <netcdf>
 #include <vtkArrowSource.h>
 #include "../CartographicTransformation.h"
+#include "../advection/readdata.h"
+#include "../advection/interpolate.h"
 
-using namespace netCDF;
 using namespace std;
 
-template <typename T>
-vector<T> getVarVector(const NcVar &var) {
-    int length = 1;
-    for (NcDim dim : var.getDims()) {
-        length *= dim.getSize();
-    }
-
-    vector<T> vec(length);
-
-    var.getVar(vec.data());
-
-    return vec;
-}
-
-tuple<vector<int>, vector<double>, vector<double>> readGrid() {
-    netCDF::NcFile data("../../../../data/grid.h5", netCDF::NcFile::read);
-    multimap< string, NcVar > vars = data.getVars();
-    vector<int> time = getVarVector<int>(vars.find("times")->second);
-    vector<double> longitude = getVarVector<double>(vars.find("longitude")->second);
-    vector<double> latitude = getVarVector<double>(vars.find("latitude")->second);
-
-    return {time, latitude, longitude};
-}
-
-
-EGlyphLayer::EGlyphLayer() {
+EGlyphLayer::EGlyphLayer(std::shared_ptr<UVGrid> uvGrid) {
   this->ren = vtkSmartPointer<vtkRenderer>::New();
   this->ren->SetLayer(1);
   this->ren->InteractiveOff();
 
+  this->uvGrid = uvGrid;
+
   this->data = vtkSmartPointer<vtkPolyData>::New();
   this->direction = vtkSmartPointer<vtkDoubleArray>::New();
   this->direction->SetName("direction");
@@ -57,25 +34,28 @@ EGlyphLayer::EGlyphLayer() {
 
 void EGlyphLayer::readCoordinates() {
   vtkNew<vtkPoints> points;
-  auto [times, lats, lons] = readGrid(); // FIXME: import Robin's readData function and use it
-  this->numLats = lats.size();
-  this->numLons = lons.size();
+  this->numLats = uvGrid->lats.size();
+  this->numLons = uvGrid->lons.size();
 
   this->direction->SetNumberOfComponents(3);
-  this->direction->SetNumberOfTuples(numLats*numLons); 
-  points->Allocate(numLats*numLons);
+  this->direction->SetNumberOfTuples(numLats * numLons);
+  points->Allocate(numLats * numLons);
 
   auto camera = createNormalisedCamera();
   ren->SetActiveCamera(camera);
 
   int i = 0;
-  for (double lat : lats) {
-    for (double lon : lons) {
-        cout << "lon: " << lon << " lat: " << lat << endl;
-      direction->SetTuple3(i, 0.45, 0.90, 0); //FIXME: read this info from file
+  int latIndex = 0;
+  for (double lat: uvGrid->lats) {
+    int lonIndex = 0;
+    for (double lon: uvGrid->lons) {
+      auto [u, v] = (*uvGrid)[0, latIndex, lonIndex];
+      direction->SetTuple3(i, 2*v, 2*u, 0);
       points->InsertPoint(i++, lon, lat, 0);
       // see also https://vtk.org/doc/nightly/html/classvtkPolyDataMapper2D.html
+      lonIndex++;
     }
+    latIndex++;
   }
   this->data->SetPoints(points);
   this->data->GetPointData()->AddArray(this->direction);
@@ -94,8 +74,8 @@ void EGlyphLayer::readCoordinates() {
   glyph2D->SetInputConnection(transformFilter->GetOutputPort());
   glyph2D->OrientOn();
   glyph2D->ClampingOn();
-  glyph2D->SetScaleModeToScaleByVector(); 
-  glyph2D->SetVectorModeToUseVector(); 
+  glyph2D->SetScaleModeToScaleByVector();
+  glyph2D->SetVectorModeToUseVector();
   glyph2D->Update();
 
 //  vtkNew<vtkCoordinate> coordinate;
@@ -109,16 +89,21 @@ void EGlyphLayer::readCoordinates() {
   vtkNew<vtkActor> actor;
   actor->SetMapper(mapper);
 
-  actor->GetProperty()->SetColor(0,0,0);
+  actor->GetProperty()->SetColor(0, 0, 0);
   actor->GetProperty()->SetOpacity(0.2);
 
-  this->ren->AddActor(actor)  ;
+  this->ren->AddActor(actor);
 }
 
 
 void EGlyphLayer::updateData(int t) {
-  for (int i=0; i < numLats*numLons; i++) {
-    this->direction->SetTuple3(i, std::cos(t), std::sin(t), 0); // FIXME: fetch data from file.
+  int i = 0;
+  for (int lat = 0; lat < uvGrid->latSize; lat++) {
+    for (int lon = 0; lon < uvGrid->lonSize; lon++) {
+      auto [u, v] = (*uvGrid)[t/3600, lat, lon];
+      this->direction->SetTuple3(i, 5*v, 5*u, 0); // FIXME: fetch data from file.
+      i++;
+    }
   }
   this->direction->Modified();
 }

vtk/src/layers/EGlyphLayer.h

@@ -4,6 +4,8 @@
 #include "Layer.h"
 #include <vtkPolyData.h>
 
+#include "../advection/UVGrid.h"
+
 /** Implements the Layer class for the case of a Eulerian visualization.
   * Specifically, this class models the eulerian flow-fields of the simulation using the 'glyph' mark and 'direction' and 'form' channels to denote direction and strength of velocities.
   */
@@ -11,6 +13,7 @@ class EGlyphLayer : public Layer {
 private:
   vtkSmartPointer<vtkPolyData> data;
   vtkSmartPointer<vtkDoubleArray> direction;
+  std::shared_ptr<UVGrid> uvGrid;
   int numLats;
   int numLons;
 
@@ -22,7 +25,7 @@ private:
 public:
   /** Constructor.
     */
-  EGlyphLayer();
+  EGlyphLayer(std::shared_ptr<UVGrid> uvGrid);
 
   /** updates the glyphs to reflect the given timestamp in the dataset.
     * @param t : the time at which to fetch the data.

vtk/src/layers/LGlyphLayer.cpp

@@ -31,7 +31,7 @@ vtkSmartPointer<SpawnPointCallback> LGlyphLayer::createSpawnPointCallback() {
 //
 // TODO: modelling all this in vtkClasses is workable, but ideally i would want to work with a native C++ class. See if this is doable and feasible.
 
-LGlyphLayer::LGlyphLayer() {
+LGlyphLayer::LGlyphLayer(std::unique_ptr<AdvectionKernel> advectionKernel) {
     this->ren = vtkSmartPointer<vtkRenderer>::New();
     this->ren->SetLayer(2);
 
@@ -39,6 +39,8 @@ LGlyphLayer::LGlyphLayer() {
     this->data = vtkSmartPointer<vtkPolyData>::New();
     this->data->SetPoints(this->points);
 
+    advector = std::move(advectionKernel);
+
     auto camera = createNormalisedCamera();
     ren->SetActiveCamera(camera);
 
@@ -88,7 +90,7 @@ void LGlyphLayer::updateData(int t) {
     double point[3];
     for (vtkIdType n = 0; n < this->points->GetNumberOfPoints(); n++) {
         this->points->GetPoint(n, point);
-        auto [xNew, yNew] = advect(n, point[0], point[1]);
+        auto [yNew, xNew] = advector->advect(t, point[1], point[0]);
         this->points->SetPoint(n, xNew, yNew, 0);
     }
     this->points->Modified();

vtk/src/layers/LGlyphLayer.h

@@ -2,6 +2,7 @@
 #define LGLYPHLAYER_H
 
 #include "Layer.h"
+#include "../advection/AdvectionKernel.h"
 #include "../commands/SpawnPointCallback.h"
 #include <vtkPolyData.h>
 #include <vtkInteractorStyle.h>
@@ -13,13 +14,14 @@ class LGlyphLayer : public Layer {
 private:
   vtkSmartPointer<vtkPoints> points;
   vtkSmartPointer<vtkPolyData> data;
+  std::unique_ptr<AdvectionKernel> advector;
 
 
 
 public:
   /** Constructor.
     */
-  LGlyphLayer();
+  LGlyphLayer(std::unique_ptr<AdvectionKernel> advectionKernel);
 
   /** This function spoofs a few points in the dataset. Mostly used for testing.
     */