Get the world name

include/sdf/Root.hh

@@ -58,6 +58,14 @@ namespace sdf
     /// \brief Default constructor
     public: Root();
 
+    /// \brief Get the name of the world without loading the entire world
+    /// \param[in] _filename Name of the SDF file to parse.
+    /// \param[out] _worldName The name of the world
+    /// \return Errors, which is a vector of Error objects. Each Error includes
+    /// an error code and message. An empty vector indicates no error.
+    public: Errors WorldName(const std::string &_filename,
+                             std::string &_worldName);
+
     /// \brief Parse the given SDF file, and generate objects based on types
     /// specified in the SDF file.
     /// \param[in] _filename Name of the SDF file to parse.

src/Root.cc

@@ -140,6 +140,40 @@ Root::Root()
 {
 }
 
+Errors Root::WorldName(const std::string &_filename, std::string &_worldName)
+{
+  Errors errors;
+
+  tinyxml2::XMLDocument doc;
+  if(doc.LoadFile(_filename.c_str()))
+  {
+    errors.push_back({ErrorCode::FILE_READ,
+                      "Unable to load file[" + _filename + "]:" +
+                      doc.ErrorStr()});
+    return errors;
+  }
+  tinyxml2::XMLElement * pRootElement = doc.RootElement();
+  if (nullptr != pRootElement)
+  {
+    tinyxml2::XMLElement * pworld = pRootElement->FirstChildElement("world");
+    if(pworld)
+    {
+      _worldName = pworld->Attribute("name");
+    }
+    else
+    {
+      errors.push_back({ErrorCode::ELEMENT_INVALID,
+                        "Failed to read the world tag."});
+    }
+  }
+  else
+  {
+    errors.push_back({ErrorCode::ELEMENT_INVALID,
+                      "Failed to read the root."});
+  }
+  return errors;
+}
+
 /////////////////////////////////////////////////
 Errors Root::Load(const std::string &_filename)
 {

src/Root_TEST.cc

@@ -26,6 +26,7 @@
 #include "sdf/World.hh"
 #include "sdf/Frame.hh"
 #include "sdf/Root.hh"
+#include "test_config.h"
 
 /////////////////////////////////////////////////
 TEST(DOMRoot, Construction)
@@ -65,6 +66,24 @@ TEST(DOMRoot, MoveAssignmentOperator)
   EXPECT_EQ("test_root", root2.Version());
 }
 
+TEST(DOMRoot, WorldName)
+{
+  const auto path = sdf::testing::TestFile("sdf", "basic_shapes.sdf");
+  sdf::Root root;
+  std::string worldName;
+  auto errors = root.WorldName(path, worldName);
+  EXPECT_TRUE(errors.empty());
+  EXPECT_EQ("shapes_world", worldName);
+
+  const auto path2 = sdf::testing::TestFile("sdf", "empty_invalid.sdf");
+  errors = root.WorldName(path2, worldName);
+  EXPECT_FALSE(errors.empty());
+
+  const auto path3 = sdf::testing::TestFile("sdf", "invalid_file_name.sdf");
+  errors = root.WorldName(path3, worldName);
+  EXPECT_FALSE(errors.empty());
+}
+
 /////////////////////////////////////////////////
 TEST(DOMRoot, StringModelSdfParse)
 {