check if noise or distortion render pass is null (#211)

Signed-off-by: Ian Chen <[email protected]>
gazebosim · Mar 31, 2022 · eab2c87 · eab2c87
1 parent 836c306
commit eab2c87
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/src/ImageBrownDistortionModel.cc b/src/ImageBrownDistortionModel.cc
@@ -107,6 +107,12 @@ void ImageBrownDistortionModel::SetCamera(rendering::CameraPtr _camera)
     // add distortion pass
     rendering::RenderPassPtr distortionPass =
       rpSystem->Create<rendering::DistortionPass>();
+    if (!distortionPass)
+    {
+      ignwarn << "ImageBrownDistortionModel is not supported in "
+              << engine->Name() << std::endl;
+      return;
+    }
     this->dataPtr->distortionPass =
         std::dynamic_pointer_cast<rendering::DistortionPass>(distortionPass);
     this->dataPtr->distortionPass->SetK1(this->dataPtr->k1);

diff --git a/src/ImageGaussianNoiseModel.cc b/src/ImageGaussianNoiseModel.cc
@@ -93,6 +93,12 @@ void ImageGaussianNoiseModel::SetCamera(rendering::CameraPtr _camera)
     // add gaussian noise pass
     rendering::RenderPassPtr noisePass =
       rpSystem->Create<rendering::GaussianNoisePass>();
+    if (!noisePass)
+    {
+      ignwarn << "ImageGaussianNoiseModel is not supported in "
+             << engine->Name() << std::endl;
+      return;
+    }
     this->dataPtr->gaussianNoisePass =
         std::dynamic_pointer_cast<rendering::GaussianNoisePass>(noisePass);
     this->dataPtr->gaussianNoisePass->SetMean(this->dataPtr->mean);