eckit::geo: exception handling and macros

ecmwf · Dec 20, 2024 · c59047f · c59047f
1 parent 2af6a89
commit c59047f
Show file tree

Hide file tree

Showing 3 changed files with 21 additions and 16 deletions.
diff --git a/src/geo/GeoIterator.cc b/src/geo/GeoIterator.cc
@@ -31,15 +31,21 @@ GeoIterator::GeoIterator(grib_handle* h, unsigned long flags) :
     CODES_CHECK(codes_get_size(h_, "values", &nv_), "");
     ECCODES_ASSERT(nv_ > 0);
 
-    data_ = (flags_ & GRIB_GEOITERATOR_NO_VALUES) ? nullptr : static_cast<double*>(grib_context_malloc(h_->context, nv_ * sizeof(double)));
-    ECCODES_ASSERT(data_ != nullptr);
-
-    auto size = nv_;
-    CODES_CHECK(codes_get_double_array(h_, "values", data_, &size), "");
-    ECCODES_ASSERT(nv_ == size);
+    //long numberOfPoints = 0;
+    //grib_get_long_internal(h, "numberOfPoints", &numberOfPoints);
+
+    if (flags_ & GRIB_GEOITERATOR_NO_VALUES) {
+        data_ = nullptr;
+    } else {
+        data_ = static_cast<double*>(grib_context_malloc(h_->context, nv_ * sizeof(double)));
+        ECCODES_ASSERT(data_ != nullptr);
+        auto size = nv_;
+        CODES_CHECK(codes_get_double_array(h_, "values", data_, &size), "");
+        // Check numberOfPoints equals nv_
+        // if not, throw an exception
+    }
 }
 
-
 int GeoIterator::init(grib_handle*, grib_arguments*)
 {
     NOTIMP;

diff --git a/src/geo/GeoIterator.h b/src/geo/GeoIterator.h
@@ -15,18 +15,12 @@
 
 #include "eckit/geo/Grid.h"
 
-// eccodes macros conflict with eckit
-#ifdef Assert
-    #undef Assert
-#endif
-
 #include "geo/iterator/grib_iterator.h"
 
 
 namespace eccodes::geo
 {
 
-
 class GeoIterator : public geo_iterator::Iterator
 {
 public:
@@ -47,5 +41,4 @@ class GeoIterator : public geo_iterator::Iterator
     geo_iterator::Iterator* create() const override;
 };
 
-
 }  // namespace eccodes::geo
diff --git a/src/geo/iterator/grib_iterator.cc b/src/geo/iterator/grib_iterator.cc
@@ -122,8 +122,14 @@ grib_iterator* grib_iterator_new(const grib_handle* ch, unsigned long flags, int
                 }
             }
         } static const init_main;
-
-        i->iterator = new eccodes::geo::GeoIterator(const_cast<grib_handle*>(ch), flags);
+        try {
+            i->iterator = new eccodes::geo::GeoIterator(const_cast<grib_handle*>(ch), flags);
+        }
+        catch(std::exception& e) {
+            grib_context_log(ch->context, GRIB_LOG_ERROR, "grib_iterator_new: Exception thrown (%s)", e.what());
+            *error = GRIB_GEOCALCULUS_PROBLEM;
+            return NULL;
+        }
     }
     else
     #endif