add particle scatter ratio?

Signed-off-by: Ian Chen <[email protected]>

include/sdf/ParticleEmitter.hh

@@ -276,6 +276,17 @@ namespace sdf
  /// \param[in] _topic The topic used to update the particle emitter.
  public: void SetTopic(const std::string &_topic);
 
+ /// \brief Get the particle scatter ratio. This is used to determine the
+ /// ratio of particles that will be detected by sensors.
+ /// \return Particle scatter ratio
+ /// \sa SetScatterRatio
+ public: float ScatterRatio() const;
+
+ /// \brief Set the particle scatter ratio. This is used to determine the
+ /// ratio of particles that will be detected by sensors.
+ /// \param[in] _ratio Scatter raito. This value should be > 0.
+ public: void SetScatterRatio(float _ratio);
+
  /// \brief Get the pose of the particle emitter. This is the pose of the
  /// emitter as specified in SDF
  /// (<particle_emitter><pose> ... </pose></particle_emitter>).

sdf/1.7/particle_emitter.sdf

@@ -104,6 +104,16 @@
  </description>
  </element>
 
+ <element name="scatter_ratio" type="float" default="0.65" required="0">
+ <description>
+ This is used to determine the ratio of particles that will be detected
+ by sensors. Increasing the ratio means there is a higher chance of
+ particles reflecting and interfering with depth sensing, making the
+ emitter appear more dense. Decreasing the ratio decreases making it
+ appear less dense. This value should be > 0.
+ </description>
+ </element>
+
  <include filename="pose.sdf" required="0"/>
  <include filename="material.sdf" required="0"/>
 </element>

src/ParticleEmitter.cc

@@ -96,6 +96,13 @@ class sdf::ParticleEmitterPrivate
  /// \brief Topic used to update particle emitter properties at runtime.
  public: std::string topic = "";
 
+ /// \brief Particle scatter ratio. This is used to determine the ratio of
+ /// particles that will be detected by sensors. Increasing the ratio
+ /// means there is a higher chance of particles reflecting and interfering
+ /// with depth sensing, making the emitter appear more dense. Decreasing the
+ /// ratio decreases making it appear less dense. This value should be > 0.
+ public: float scatterRatio = 0.65f;
+
  /// \brief Pose of the emitter
  public: ignition::math::Pose3d pose = ignition::math::Pose3d::Zero;
 
@@ -136,6 +143,7 @@ ParticleEmitterPrivate::ParticleEmitterPrivate(
  colorEnd(_private.colorEnd),
  colorRangeImage(_private.colorRangeImage),
  topic(_private.topic),
+ scatterRatio(_private.scatterRatio),
  pose(_private.pose),
  poseRelativeTo(_private.poseRelativeTo),
  xmlParentName(_private.xmlParentName),
@@ -272,6 +280,9 @@ Errors ParticleEmitter::Load(ElementPtr _sdf)
  this->dataPtr->topic = _sdf->Get<std::string>(
  "topic", this->dataPtr->topic).first;
 
+ this->dataPtr->scatterRatio = _sdf->Get<float>(
+ "scatter_ratio", this->dataPtr->scatterRatio).first;
+
  if (_sdf->HasElement("material"))
  {
  this->dataPtr->material.reset(new sdf::Material());
@@ -487,6 +498,18 @@ void ParticleEmitter::SetTopic(const std::string &_topic)
  this->dataPtr->topic = _topic;
 }
 
+/////////////////////////////////////////////////
+void ParticleEmitter::SetScatterRatio(float _ratio)
+{
+ this->dataPtr->scatterRatio = _ratio;
+}
+
+/////////////////////////////////////////////////
+float ParticleEmitter::ScatterRatio() const
+{
+ return this->dataPtr->scatterRatio;
+}
+
 /////////////////////////////////////////////////
 const ignition::math::Pose3d &ParticleEmitter::RawPose() const
 {

src/ParticleEmitter_TEST.cc

@@ -104,6 +104,10 @@ TEST(DOMParticleEmitter, Construction)
  emitter.SetTopic("/test/topic");
  EXPECT_EQ("/test/topic", emitter.Topic());
 
+ EXPECT_FLOAT_EQ(0.65f, emitter.ScatterRatio());
+ emitter.SetScatterRatio(0.5f);
+ EXPECT_FLOAT_EQ(0.5f, emitter.ScatterRatio());
+
  EXPECT_EQ(ignition::math::Pose3d::Zero, emitter.RawPose());
  emitter.SetRawPose(ignition::math::Pose3d(1, 2, 3, 0, 0, 1.5707));
  EXPECT_EQ(ignition::math::Pose3d(1, 2, 3, 0, 0, 1.5707), emitter.RawPose());

test/integration/particle_emitter_dom.cc

@@ -78,6 +78,7 @@ TEST(DOMWorld, LoadParticleEmitter)
  EXPECT_DOUBLE_EQ(0.2, linkEmitter->MaxVelocity());
  EXPECT_DOUBLE_EQ(0.5, linkEmitter->ScaleRate());
  EXPECT_DOUBLE_EQ(5, linkEmitter->Rate());
+ EXPECT_FLOAT_EQ(0.2f, linkEmitter->ScatterRatio());
 
  sdf::Material *mat = linkEmitter->Material();
  ASSERT_NE(nullptr, mat);

test/sdf/world_complete.sdf

@@ -90,6 +90,7 @@
  </pbr>
  </material>
  <color_range_image>materials/textures/fogcolors.png</color_range_image>
+ <scatter_ratio>0.2</scatter_ratio>
  </particle_emitter>
 
  <light name="spot_light" type="spot">