FEAT: needed for grid to grid MP

Author: aaronjridley

include/grid.h

@@ -11,6 +11,33 @@
 // Grid class
 // ----------------------------------------------------------------------------
 
+  struct interp_coef_t
+  {
+    // The point is inside the cube of [iRow, iRow+1], [iCol, iCol+1], [iAlt, iAlt+1]
+    uint64_t iRow;
+    uint64_t iCol;
+    uint64_t iAlt;
+    // The coefficients along row, column and altitude
+    precision_t rRow;
+    precision_t rCol;
+    precision_t rAlt;
+    // Whether the point is within this grid or not
+    bool in_grid;
+    // If this is set to true:
+    bool above_grid, below_grid;
+    // do interpolation in lat and lon, but extrapolate in altitude
+  };
+
+  struct grid_to_grid_t {
+    int64_t iProcTo;
+    int64_t nPts;
+    int64_t nPtsReceive;
+    std::vector<struct interp_coef_t> interpCoefs;
+    std::vector<precision_t *> valueToSend;
+    std::vector<precision_t *> valueToReceive;
+  };
+
+
 class Grid
 {
 
@@ -20,13 +47,24 @@ class Grid
   const int iDipole_ = 3;
   int iGridShape_ = -1;
 
-  // Armidillo Cube Versions:
+  // The index and coefficient used for interpolation
+  // Each point is processed by the function set_interpolation_coefs and stored
+  // in the form of this structure.
+  // If the point is out of the grid, in_grid = false and all other members are undefined
+
+  std::vector<struct grid_to_grid_t> gridToGridCoefs;
+  arma::Cube<int> gridToGridMap;
+
+  // Armadillo Cube Versions:
   // Cell Center Coordinates
   arma_cube geoLon_scgc, geoX_scgc;
   arma_cube geoLat_scgc, geoY_scgc;
   arma_cube geoAlt_scgc, geoZ_scgc;
   arma_cube geoLocalTime_scgc;
 
+  // This is an array for testing things:
+  arma_cube test_scgc;
+
   // Reference coordinate
   arma_cube refx_scgc, refy_scgc;
 
@@ -434,6 +472,22 @@ class Grid
   bool set_interpolation_coefs(const std::vector<precision_t> &Lons,
                                const std::vector<precision_t> &Lats,
                                const std::vector<precision_t> &Alts);
+
+  /**
+   * \brief Set the interpolation coefficients
+   * \param Lons The longitude of points
+   * \param Lats The latitude of points
+   * \param Alts The altitude of points
+   * \pre This instance is an geo grid
+   * \pre Lons, Lats and Alts have the same size
+   * \return list of interpolation coefficients
+   */
+  
+  std::vector<struct interp_coef_t> get_interpolation_coefs(
+                                    const std::vector<precision_t> &Lons,
+                                    const std::vector<precision_t> &Lats,
+                                    const std::vector<precision_t> &Alts);
+  
   /**
    * \brief Create a map of geographic locations to data and do the interpolation
    * \param data The value at the positions of geoLon, geoLat, and geoAlt
@@ -443,6 +497,8 @@ class Grid
    *         an empty vector if the data is not the same size as the geo grid.
    */
   std::vector<precision_t> get_interpolation_values(const arma_cube &data) const;
+  std::vector<precision_t> get_interpolation_values(arma_cube data,
+                                                    std::vector<struct interp_coef_t> coefArray);
 
 private:
   bool IsGeoGrid;
@@ -513,24 +569,6 @@ class Grid
     bool col_max_exclusive;
   };
 
-  // The index and coefficient used for interpolation
-  // Each point is processed by the function set_interpolation_coefs and stored
-  // in the form of this structure.
-  // If the point is out of the grid, in_grid = false and all other members are undefined
-  struct interp_coef_t
-  {
-    // The point is inside the cube of [iRow, iRow+1], [iCol, iCol+1], [iAlt, iAlt+1]
-    uint64_t iRow;
-    uint64_t iCol;
-    uint64_t iAlt;
-    // The coefficients along row, column and altitude
-    precision_t rRow;
-    precision_t rCol;
-    precision_t rAlt;
-    // Whether the point is within this grid or not
-    bool in_grid;
-  };
-
   // Return the index of the last element that has altitude smaller than or euqal to the input
   uint64_t search_altitude(const precision_t alt_in) const;
 
@@ -540,11 +578,11 @@ class Grid
   void get_cubesphere_grid_range(struct cubesphere_range &cr) const;
 
   // Helper function for set_interpolation_coefs
-  void set_interp_coef_sphere(const sphere_range &sr,
+  struct interp_coef_t get_interp_coef_sphere(const sphere_range &sr,
                               const precision_t lon_in,
                               const precision_t lat_in,
                               const precision_t alt_in);
-  void set_interp_coef_cubesphere(const cubesphere_range &cr,
+  struct interp_coef_t get_interp_coef_cubesphere(const cubesphere_range &cr,
                                   const precision_t lon_in,
                                   const precision_t lat_in,
                                   const precision_t alt_in);