Support printing sdf poses in degrees and allow snapping to commonly used angles

include/sdf/Element.hh

@@ -199,7 +199,7 @@ namespace sdf
     /// \param[in] _prefix String value to prefix to the output.
     /// \param[in] _includeDefaultElements flag to include default elements.
     /// \param[in] _includeDefaultAttributes flag to include default attributes.
-    /// \param[in] _config Configuration for printing the values.
+    /// \param[in] _config Configuration for converting to string.
     /// \return The string representation.
     public: std::string ToString(
         const std::string &_prefix,
@@ -589,7 +589,7 @@ namespace sdf
     /// \param[in] _includeDefaultElements flag to include default elements.
     /// \param[in] _includeDefaultAttributes flag to include default attributes.
     /// \param[in] _config Configuration for printing values.
-    /// \param[out] _out the std::ostreamstream to write output to.
+    /// \param[out] _out the std::ostringstream to write output to.
     private: void PrintValuesImpl(const std::string &_prefix,
                                   bool _includeDefaultElements,
                                   bool _includeDefaultAttributes,
@@ -685,6 +685,14 @@ namespace sdf
 
     /// \brief XML path of this element.
     public: std::string xmlPath;
+
+    /// \brief Generate the string (XML) for the attributes.
+    /// \param[in] _includeDefaultAttributes flag to include default attributes.
+    /// \param[in] _config Configuration for printing attributes.
+    /// \param[out] _out the std::ostringstream to write output to.
+    public: void PrintAttributes(bool _includeDefaultAttributes,
+                                 const PrintConfig &_config,
+                                 std::ostringstream &_out) const;
   };
 
   ///////////////////////////////////////////////

include/sdf/Param.hh

@@ -401,17 +401,34 @@ namespace sdf
                                     const std::string &_valueStr,
                                     ParamVariant &_valueToSet) const;
 
-    /// \brief Method used to get the string representation from a ParamVariant
+    /// \brief Method used to get the string representation from a ParamVariant,
+    /// or the string that was used to set it.
     /// \param[in] _config Print configuration for the string output
     /// \param[in] _typeName The data type of the value
     /// \param[in] _value The value
-    /// \param[in] _valueStr The string representation of the value
-    /// \return True if the string was successfully retrieved from the value,
-    /// false otherwise.
-    public: bool StringFromValueImpl(const PrintConfig &_config,
-                                     const std::string &_typeName,
-                                     const ParamVariant &_value,
-                                     std::string &_valueStr) const;
+    /// \param[out] _valueStr The output string.
+    /// \return True if the string was successfully retrieved, false otherwise.
+    public: bool StringFromValueImpl(
+                const PrintConfig &_config,
+                const std::string &_typeName,
+                const ParamVariant &_value,
+                std::string &_valueStr) const;
+
+    /// \brief Method used to get the string representation from a ParamVariant,
+    /// or the string that was used to set it.
+    /// \param[in] _config Print configuration for the string output
+    /// \param[in] _typeName The data type of the value
+    /// \param[in] _value The value
+    /// \param[in] _orignalStr The original string that was used to set the
+    /// value. A nullopt can be passed in if it is not available.
+    /// \param[out] _valueStr The output string.
+    /// \return True if the string was successfully retrieved, false otherwise.
+    public: bool StringFromValueImpl(
+                const PrintConfig &_config,
+                const std::string &_typeName,
+                const ParamVariant &_value,
+                const std::optional<std::string> &_originalStr,
+                std::string &_valueStr) const;
 
     /// \brief Data type to string mapping
     /// \return The type as a string, empty string if unknown type

include/sdf/PrintConfig.hh

@@ -17,6 +17,7 @@
 #ifndef SDF_PRINTCONFIG_HH_
 #define SDF_PRINTCONFIG_HH_
 
+#include <optional>
 #include <ignition/utils/ImplPtr.hh>
 
 #include "sdf/sdf_config.h"
@@ -29,9 +30,41 @@ namespace sdf
   /// This class contains configuration options for printing elements.
   class SDFORMAT_VISIBLE PrintConfig
   {
-    /// \brief Default constructor.
+    /// \brief Default constructor. All options are set to false by default.
     public: PrintConfig();
 
+    /// \brief Sets the option for printing pose rotations in degrees if true,
+    /// otherwise they will be printed as radians by default.
+    /// \param[in] _value Whether to print pose rotations in degrees.
+    public: void SetRotationInDegrees(bool _value);
+
+    /// \brief Returns whether or not pose rotations should be printed in
+    /// degrees.
+    /// \return True if pose rotations are printed in degrees, false otherwise.
+    public: bool RotationInDegrees() const;
+
+    /// \brief Sets the option for printing pose rotation in degrees as well as
+    /// snapping the rotation to the desired interval, with the provided
+    /// tolerance.
+    /// \param[in] _interval Degrees interval to snap to, this value must be
+    /// larger than 0, and less than or equal to 360.
+    /// \param[in] _tolerance Tolerance which snapping occurs, this value must
+    /// be larger than 0, less than 360, and less than the provided interval.
+    /// \return True, unless any of the provided values are not valid.
+    public: bool SetRotationSnapToDegrees(unsigned int _interval,
+                                          double _tolerance);
+
+    /// \brief Returns the current degree value that pose rotations will snap to
+    /// when printed.
+    /// \return The assigned degrees interval value to snap to. If it has not
+    /// been assigned, a nullopt will be returned.
+    public: std::optional<unsigned int> RotationSnapToDegrees() const;
+
+    /// \brief Returns the tolerance for snapping degree values when printed.
+    /// \return The assigned tolerance value which allows snapping to happen. If
+    /// it has not been assigned, a nullopt will be returned.
+    public: std::optional<double> RotationSnapTolerance() const;
+
     /// \brief Set print config to preserve <include> tags.
     /// \param[in] _preserve True to preserve <include> tags.
     /// False to expand included model.
@@ -42,6 +75,11 @@ namespace sdf
     /// False if they are to be expanded.
     public: bool PreserveIncludes() const;
 
+    /// \brief Return true if both PrintConfig objects contain the same values.
+    /// \param[in] _config PrintConfig to compare.
+    /// \return True if 'this' == _config.
+    public: bool operator==(const PrintConfig &_config) const;
+
     /// \brief Private data pointer.
     IGN_UTILS_IMPL_PTR(dataPtr)
   };

src/CMakeLists.txt

@@ -136,8 +136,8 @@ if (BUILD_SDF_TEST)
     Pbr_TEST.cc
     Physics_TEST.cc
     Plugin_TEST.cc
-    PrintConfig_TEST.cc
     Plane_TEST.cc
+    PrintConfig_TEST.cc
     Root_TEST.cc
     Scene_TEST.cc
     SemanticPose_TEST.cc

src/Element.cc

@@ -499,6 +499,7 @@ void Element::PrintDocLeftPane(std::string &_html, int _spacing,
   _html += "</div>\n";
 }
 
+/////////////////////////////////////////////////
 void Element::PrintValuesImpl(const std::string &_prefix,
                               bool _includeDefaultElements,
                               bool _includeDefaultAttributes,
@@ -513,22 +514,7 @@ void Element::PrintValuesImpl(const std::string &_prefix,
   {
     _out << _prefix << "<" << this->dataPtr->name;
 
-    Param_V::const_iterator aiter;
-    for (aiter = this->dataPtr->attributes.begin();
-         aiter != this->dataPtr->attributes.end(); ++aiter)
-    {
-      // Only print attribute values if they were set
-      // TODO(anyone): GetRequired is added here to support up-conversions where
-      // a new required attribute with a default value is added. We would have
-      // better separation of concerns if the conversion process set the
-      // required attributes with their default values.
-      if ((*aiter)->GetSet() || (*aiter)->GetRequired() ||
-          _includeDefaultAttributes)
-      {
-        _out << " " << (*aiter)->GetKey() << "='"
-             << (*aiter)->GetAsString(_config) << "'";
-      }
-    }
+    this->dataPtr->PrintAttributes(_includeDefaultAttributes, _config, _out);
 
     if (this->dataPtr->elements.size() > 0)
     {
@@ -560,6 +546,50 @@ void Element::PrintValuesImpl(const std::string &_prefix,
   }
 }
 
+/////////////////////////////////////////////////
+void ElementPrivate::PrintAttributes(bool _includeDefaultAttributes,
+                                     const PrintConfig &_config,
+                                     std::ostringstream &_out) const
+{
+  // Attribute exceptions are used in the event of a non-default PrintConfig
+  // which modifies the Attributes of this Element that are printed out. The
+  // modifications to an Attribute by a PrintConfig will overwrite the original
+  // existing Attribute when this Element is printed.
+  std::set<std::string> attributeExceptions;
+  if (this->name == "pose")
+  {
+    if (_config.RotationInDegrees() || _config.RotationSnapToDegrees())
+    {
+      attributeExceptions.insert("degrees");
+      _out << " " << "degrees='true'";
+
+      attributeExceptions.insert("rotation_format");
+      _out << " " << "rotation_format='euler_rpy'";
+    }
+  }
+
+  Param_V::const_iterator aiter;
+  for (aiter = this->attributes.begin();
+       aiter != this->attributes.end(); ++aiter)
+  {
+    // Only print attribute values if they were set
+    // TODO(anyone): GetRequired is added here to support up-conversions where
+    // a new required attribute with a default value is added. We would have
+    // better separation of concerns if the conversion process set the
+    // required attributes with their default values.
+    if ((*aiter)->GetSet() || (*aiter)->GetRequired() ||
+        _includeDefaultAttributes)
+    {
+      const std::string key = (*aiter)->GetKey();
+      const auto it = attributeExceptions.find(key);
+      if (it == attributeExceptions.end())
+      {
+        _out << " " << key << "='" << (*aiter)->GetAsString(_config) << "'";
+      }
+    }
+  }
+}
+
 /////////////////////////////////////////////////
 void Element::PrintValues(std::string _prefix, const PrintConfig &_config) const
 {