[ML/MPL] check if element has medium

MaterialLib/MPL/MaterialSpatialDistributionMap.cpp

@@ -21,4 +21,18 @@ Medium* MaterialSpatialDistributionMap::getMedium(std::size_t const element_id)
 
     return _media.at(material_id).get();
 }
+
+void MaterialSpatialDistributionMap::checkElementHasMedium(std::size_t const element_id)
+{
+    auto const material_id =
+            _material_ids == nullptr ? 0 : (*_material_ids)[element_id];
+    if (_media.find(material_id) == _media.end())
+    {
+        OGS_FATAL(
+                "There is no medium definition for element %d with material "
+                "ID %d. Please define a medium for each material.",
+                element_id, material_id);
+    }
+}
+
 }  // namespace MaterialPropertyLib

MaterialLib/MPL/MaterialSpatialDistributionMap.h

@@ -36,6 +36,7 @@ public:
     }
 
     Medium* getMedium(std::size_t element_id);
+    void checkElementHasMedium(std::size_t const element_id);
 
 private:
     std::map<int, std::shared_ptr<Medium>> const& _media;