Expose particle scatter ratio parameter

ogre2/src/Ogre2DepthCamera.cc

@@ -134,10 +134,6 @@ class ignition::rendering::Ogre2DepthCameraPrivate
   /// \brief standard deviation of particle noise
   public: double particleStddev = 0.01;
 
-  /// \brief Particle scatter ratio. This is used to determine the ratio of
-  /// particles that will detected by the depth camera
-  public: double particleScatterRatio = 0.65;
-
   /// \brief Listener for setting particle noise value based on particle
   /// emitter region
   public: std::unique_ptr<Ogre2ParticleNoiseListener> particleNoiseListener;
@@ -436,8 +432,6 @@ void Ogre2DepthCamera::CreateDepthTexture()
   psParams->setNamedConstant("backgroundColor", bg);
   psParams->setNamedConstant("particleStddev",
     static_cast<float>(this->dataPtr->particleStddev));
-  psParams->setNamedConstant("particleScatterRatio",
-    static_cast<float>(this->dataPtr->particleScatterRatio));
 
   std::string matDepthFinalName = "DepthCameraFinal";
   Ogre::MaterialPtr matDepthFinal =

ogre2/src/Ogre2GpuRays.cc

@@ -169,10 +169,6 @@ class ignition::rendering::Ogre2GpuRaysPrivate
   /// \brief standard deviation of particle noise
   public: double particleStddev = 0.01;
 
-  /// \brief Particle scatter ratio. This is used to determine the ratio of
-  /// particles that will detected by the depth camera
-  public: double particleScatterRatio = 0.65;
-
   /// \brief Listener for setting particle noise value based on particle
   /// emitter region
   public: std::unique_ptr<Ogre2ParticleNoiseListener> particleNoiseListener[6];
@@ -623,8 +619,6 @@ void Ogre2GpuRays::Setup1stPass()
       static_cast<float>(this->dataMinVal));
   psParams->setNamedConstant("particleStddev",
     static_cast<float>(this->dataPtr->particleStddev));
-  psParams->setNamedConstant("particleScatterRatio",
-    static_cast<float>(this->dataPtr->particleScatterRatio));
 
   // Create 1st pass compositor
   auto engine = Ogre2RenderEngine::Instance();

ogre2/src/Ogre2ParticleEmitter.cc

@@ -479,6 +479,9 @@ void Ogre2ParticleEmitter::CreateParticleSystem()
 {
   // Instantiate the particle system and default parameters.
   this->dataPtr->ps = this->scene->OgreSceneManager()->createParticleSystem();
+  this->dataPtr->ps->getUserObjectBindings().setUserAny(
+      Ogre::Any(this->Id()));
+
   this->dataPtr->ps->setCullIndividually(true);
   this->dataPtr->ps->setParticleQuota(500);
   this->dataPtr->ps->setSortingEnabled(true);

ogre2/src/Ogre2ParticleNoiseListener.cc

@@ -1,5 +1,5 @@
 /*
- * Copyright (C) 2020 Open Source Robotics Foundation
+ * Copyright (C) 2021 Open Source Robotics Foundation
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -15,9 +15,12 @@
  *
  */
 
+#include <string>
+
 #include "ignition/rendering/ogre2/Ogre2Includes.hh"
 #include "ignition/rendering/ogre2/Ogre2RenderTypes.hh"
 #include "ignition/rendering/ogre2/Ogre2Scene.hh"
+#include "ignition/rendering/ogre2/Ogre2Visual.hh"
 
 #include "Ogre2ParticleNoiseListener.hh"
 
@@ -82,6 +85,63 @@ void Ogre2ParticleNoiseListener::preRenderTargetUpdate(
           pass->getFragmentProgramParameters();
       psParams->setNamedConstant("particleStddev",
           static_cast<float>(particleStddev));
+
+      // get particle scatter ratio value from particle emitter user data
+      // set pass that to the shaders
+      Ogre::Any userAny = ps->getUserObjectBindings().getUserAny();
+      if (!userAny.isEmpty() && userAny.getType() == typeid(unsigned int))
+      {
+        VisualPtr result;
+        try
+        {
+          result = this->scene->VisualById(
+              Ogre::any_cast<unsigned int>(userAny));
+        }
+        catch(Ogre::Exception &e)
+        {
+          ignerr << "Ogre Error:" << e.getFullDescription() << "\n";
+        }
+        Ogre2VisualPtr ogreVisual =
+            std::dynamic_pointer_cast<Ogre2Visual>(result);
+
+        std::string particleScatterRatioKey = "particle_scatter_ratio";
+        Variant particleScatterRatioAny =
+            ogreVisual->UserData(particleScatterRatioKey);
+        if (particleScatterRatioAny.index() != std::variant_npos)
+        {
+          float ratio = -1.0;
+          try
+          {
+            ratio = std::get<float>(particleScatterRatioAny);
+          }
+          catch(...)
+          {
+            try
+            {
+              ratio = std::get<double>(particleScatterRatioAny);
+            }
+            catch(...)
+            {
+              try
+              {
+                ratio = std::get<int>(particleScatterRatioAny);
+              }
+              catch(std::bad_variant_access &e)
+              {
+                ignerr << "Error casting user data: " << e.what() << "\n";
+                ratio = -1.0;
+              }
+            }
+          }
+          if (ratio > 0)
+          {
+            this->particleScatterRatio = ratio;
+          }
+        }
+      }
+      psParams->setNamedConstant("particleScatterRatio",
+          static_cast<float>(this->particleScatterRatio));
+
       return;
     }
     itor.moveNext();

ogre2/src/Ogre2ParticleNoiseListener.hh

@@ -40,9 +40,19 @@ namespace ignition
       private: virtual void preRenderTargetUpdate(
           const Ogre::RenderTargetEvent &_evt) override;
 
+      /// \brief Pointer to scene
       private: Ogre2ScenePtr scene;
+
+      /// \brief Pointer to camera
       private: Ogre::Camera *ogreCamera = nullptr;
+
+      /// \brief Pointer to ogre matieral with shaders for applying particle
+      /// scattering effect to sensors
       private: Ogre::MaterialPtr ogreMaterial;
+
+      /// \brief Particle scatter ratio. This is used to determine the ratio of
+      /// particles that will detected by sensors
+      private: double particleScatterRatio = 0.65;
     };
     }
   }

src/ParticleEmitter_TEST.cc

@@ -91,6 +91,9 @@ void ParticleEmitterTest::CheckBasicAPI()
   math::Color    expectedColorEnd        = ignition::math::Color::White;
   double         expectedScaleRate       = 1;
   std::string    expectedColorRangeImage = "";
+  // Particle scatter ratio is stored in user data
+  // \todo(anyone) Add API to set scatter ratio
+  Variant        emptyVariant;
 
   // Check default expectations.
   EXPECT_EQ(expectedEmitterType,      particleEmitter->Type());
@@ -107,6 +110,7 @@ void ParticleEmitterTest::CheckBasicAPI()
   EXPECT_EQ(expectedColorEnd,         particleEmitter->ColorEnd());
   EXPECT_DOUBLE_EQ(expectedScaleRate, particleEmitter->ScaleRate());
   EXPECT_EQ(expectedColorRangeImage,  particleEmitter->ColorRangeImage());
+  EXPECT_EQ(emptyVariant,  particleEmitter->UserData("particle_scatter_ratio"));
 
   // Modify values.
   expectedEmitterType     = EmitterType::EM_BOX;
@@ -123,6 +127,9 @@ void ParticleEmitterTest::CheckBasicAPI()
   expectedColorEnd        = ignition::math::Color::Blue;
   expectedScaleRate       = 10;
   expectedColorRangeImage = common::joinPaths(TEST_MEDIA_PATH, "texture.png");
+  // Particle scatter ratio is stored in user data
+  // \todo(anyone) Add API to set scatter ratio
+  double expectedScatterRatio  = 0.24;
 
   // Modify attributes.
   particleEmitter->SetType(expectedEmitterType);
@@ -137,6 +144,7 @@ void ParticleEmitterTest::CheckBasicAPI()
   particleEmitter->SetColorRange(expectedColorStart, expectedColorEnd);
   particleEmitter->SetScaleRate(expectedScaleRate);
   particleEmitter->SetColorRangeImage(expectedColorRangeImage);
+  particleEmitter->SetUserData("particle_scatter_ratio", expectedScatterRatio);
 
   // Check getters.
   EXPECT_EQ(expectedEmitterType,      particleEmitter->Type());
@@ -153,6 +161,8 @@ void ParticleEmitterTest::CheckBasicAPI()
   EXPECT_EQ(expectedColorEnd,         particleEmitter->ColorEnd());
   EXPECT_DOUBLE_EQ(expectedScaleRate, particleEmitter->ScaleRate());
   EXPECT_EQ(expectedColorRangeImage,  particleEmitter->ColorRangeImage());
+  Variant v = particleEmitter->UserData("particle_scatter_ratio");
+  EXPECT_DOUBLE_EQ(expectedScatterRatio,  std::get<double>(v));
 }
 
 /////////////////////////////////////////////////

test/integration/depth_camera.cc

@@ -678,6 +678,69 @@ void DepthCameraTest::DepthCameraParticles(
     EXPECT_LT(pointParticleAvg, pointAvg);
     EXPECT_LT(depthParticleAvg, depthAvg);
 
+    // test setting particle scatter ratio
+    // reduce particle scatter ratio - this creates a "less dense" particle
+    // emitter so we should have larger depth values on avg since fewers
+    // depth readings are occluded by particles
+    emitter->SetUserData("particle_scatter_ratio", 0.1);
+
+    g_depthCounter = 0u;
+    g_pointCloudCounter = 0u;
+    for (unsigned int i = 0; i < 100; ++i)
+    {
+      depthCamera->Update();
+    }
+    EXPECT_EQ(100u, g_depthCounter);
+    EXPECT_EQ(100u, g_pointCloudCounter);
+
+    double pointParticleLowScatterAvg = 0.0;
+    double depthParticleLowScatterAvg = 0.0;
+
+    // Verify depth and point cloud data after setting particle scatter ratio
+    for (unsigned int i = 0u; i < depthCamera->ImageHeight(); ++i)
+    {
+      unsigned int step =
+          i * depthCamera->ImageWidth() * pointCloudChannelCount;
+      for (unsigned int j = 0u; j < depthCamera->ImageWidth(); ++j)
+      {
+        float x = pointCloudData[step + j * pointCloudChannelCount];
+        float y = pointCloudData[step + j * pointCloudChannelCount + 1];
+        float z = pointCloudData[step + j * pointCloudChannelCount + 2];
+
+        double xd = static_cast<double>(x);
+        // depth camera sees only certain percentage of particles
+        // so the values should be either
+        //   * box depth (depth camera does not see particles), or
+        //   * noisy particle depth (depth camera see particles but values
+        //     are affected by noise)
+        EXPECT_TRUE(
+            ignition::math::equal(expectedParticleDepth, xd, depthNoiseTol) ||
+            ignition::math::equal(expectedDepth, xd, DEPTH_TOL))
+            << "actual vs expected particle depth: "
+            << xd << " vs " << expectedParticleDepth;
+        float depth = scan[i * depthCamera->ImageWidth() + j];
+        double depthd = static_cast<double>(depth);
+        EXPECT_TRUE(
+            ignition::math::equal(expectedParticleDepth, depthd, depthNoiseTol)
+            || ignition::math::equal(expectedDepth, depthd, DEPTH_TOL))
+            << "actual vs expected particle depth: "
+            << depthd << " vs " << expectedParticleDepth;
+
+        pointParticleLowScatterAvg +=
+            ignition::math::Vector3d(x, y, z).Length();
+        depthParticleLowScatterAvg += depthd;
+      }
+    }
+
+    // compare point and depth data before and after setting particle scatter
+    // ratio. The avg point length and depth values in the image with low
+    // particle scatter ratio should be should be higher than the previous
+    // images with particle effects
+    pointParticleLowScatterAvg /= pixelCount;
+    depthParticleLowScatterAvg /= pixelCount;
+    EXPECT_LT(pointParticleAvg, pointParticleLowScatterAvg);
+    EXPECT_LT(depthParticleAvg, depthParticleLowScatterAvg);
+
     // Clean up
     connection.reset();
     delete [] scan;

test/integration/gpu_rays.cc

@@ -558,7 +558,7 @@ void GpuRaysTest::RaysParticles(const std::string &_renderEngine)
   emitter->SetRate(100);
   emitter->SetLifetime(2);
   emitter->SetVelocityRange(0.1, 0.1);
-  emitter->SetScaleRate(0.1);
+  emitter->SetScaleRate(0.0);
   emitter->SetColorRange(ignition::math::Color::Red,
       ignition::math::Color::Black);
   emitter->SetEmitting(true);
@@ -621,6 +621,49 @@ void GpuRaysTest::RaysParticles(const std::string &_renderEngine)
   // there should be at least one miss
   EXPECT_GT(particleMissCount, 0u);
 
+
+  // test setting particle scatter ratio
+  // reduce particle scatter ratio - this creates a "less dense" particle
+  // emitter so we should have larger range values on avg since fewer
+  // rays are occluded by particles
+  emitter->SetUserData("particle_scatter_ratio", 0.1);
+
+  unsigned int particleHitLowScatterCount = 0u;
+  unsigned int particleMissLowScatterCount = 0u;
+  for (unsigned int i = 0u; i < 100u; ++i)
+  {
+    gpuRays->Update();
+
+    // sensor should see ether a particle or box01
+    double particleRange = static_cast<double>(scan[mid]);
+    bool particleHit = ignition::math::equal(
+        expectedParticleRange, particleRange, laserNoiseTol);
+    bool particleMiss = ignition::math::equal(
+        expectedRangeAtMidPointBox1, particleRange, LASER_TOL);
+    EXPECT_TRUE(particleHit || particleMiss)
+        << "actual vs expected particle range: "
+        << particleRange << " vs " << expectedParticleRange;
+
+    particleHitLowScatterCount += particleHit ? 1u : 0u;
+    particleMissLowScatterCount += particleMiss ? 1u : 0u;
+
+    // sensor should see box02 without noise or scatter effect
+    EXPECT_NEAR(expectedRangeAtMidPointBox2, scan[0], LASER_TOL);
+
+    // sensor should not see box03 as it is out of range
+    EXPECT_DOUBLE_EQ(ignition::math::INF_D, scan[last]);
+  }
+
+  // there should be at least one hit
+  EXPECT_GT(particleHitLowScatterCount, 0u);
+  // there should be at least one miss
+  EXPECT_GT(particleMissLowScatterCount, 0u);
+
+  // there should be more misses than the previous particle emitter setting
+  // i.e. more rays missing the particles because of low scatter ratio / density
+  EXPECT_GT(particleHitCount, particleHitLowScatterCount);
+  EXPECT_LT(particleMissCount, particleMissLowScatterCount);
+
   c.reset();
 
   delete [] scan;