Set HOME_ENV according the OS. for windows is USERPROFILE for other i…

…s HOME
gazebosim · Jun 24, 2023 · 6ea7238 · 6ea7238
1 parent 17e09f5
commit 6ea7238
Show file tree

Hide file tree

Showing 14 changed files with 49 additions and 15 deletions.
diff --git a/gazebo/common/CommonIface.cc b/gazebo/common/CommonIface.cc
@@ -215,6 +215,13 @@ const char *common::getEnv(const char *_name)
 #endif
 }
 
+/////////////////////////////////////////////////
+#ifdef _WIN32
+  std::string common::HOME_ENV = "USERPROFILE";
+#else
+  std::string common::HOME_ENV = "HOME";
+#endif
+
 /////////////////////////////////////////////////
 std::vector<std::string> common::split(const std::string &_str,
     const std::string &_delim)

diff --git a/gazebo/common/CommonIface.hh b/gazebo/common/CommonIface.hh
@@ -85,6 +85,8 @@ namespace gazebo
     GZ_COMMON_VISIBLE
     const char *getEnv(const char *_name);
 
+    extern std::string HOME_ENV;
+
     /// \brief Get the current working directory
     /// \return Name of the current directory
     GZ_COMMON_VISIBLE

diff --git a/gazebo/common/Console.cc b/gazebo/common/Console.cc
@@ -23,6 +23,7 @@
 #include "gazebo/common/Exception.hh"
 #include "gazebo/common/Time.hh"
 #include "gazebo/common/Console.hh"
+#include "gazebo/common/CommonIface.hh"
 
 #include "gazebo/gazebo_config.h"
 
@@ -155,7 +156,7 @@ FileLogger::~FileLogger()
 /////////////////////////////////////////////////
 void FileLogger::Init(const std::string &_prefix, const std::string &_filename)
 {
-  if (!getenv("HOME"))
+  if (!getenv(HOME_ENV.c_str()))
   {
     gzerr << "Missing HOME environment variable."
           << "No log file will be generated.";
@@ -165,7 +166,7 @@ void FileLogger::Init(const std::string &_prefix, const std::string &_filename)
   FileLogger::Buffer *buf = static_cast<FileLogger::Buffer*>(
       this->rdbuf());
 
-  boost::filesystem::path logPath(getenv("HOME"));
+  boost::filesystem::path logPath(getenv(HOME_ENV.c_str()));
 
   // Create a subdirectory for the informational log. The name of the directory
   // will be <PREFIX><MASTER_PORT>. E.g.: server-11346. If the environment

diff --git a/gazebo/common/ModelDatabase.cc b/gazebo/common/ModelDatabase.cc
@@ -40,6 +40,7 @@
 #include "gazebo/common/ModelDatabasePrivate.hh"
 #include "gazebo/common/ModelDatabase.hh"
 #include "gazebo/common/SemanticVersion.hh"
+#include "gazebo/common/CommonIface.hh"
 
 using namespace gazebo;
 using namespace common;
@@ -503,7 +504,7 @@ std::string ModelDatabase::GetModelPath(const std::string &_uri,
         continue;
       }
 
-      std::string outputPath = getenv("HOME");
+      std::string outputPath = getenv(HOME_ENV.c_str());
       outputPath += "/.gazebo/models";
 
 #ifndef _WIN32

diff --git a/gazebo/common/SystemPaths.cc b/gazebo/common/SystemPaths.cc
@@ -38,6 +38,7 @@
 #include "gazebo/common/Exception.hh"
 #include "gazebo/common/ModelDatabase.hh"
 #include "gazebo/common/SystemPaths.hh"
+#include "gazebo/common/CommonIface.hh"
 
 using namespace gazebo;
 using namespace common;
@@ -81,7 +82,7 @@ SystemPaths::SystemPaths()
     return;
   }
 
-  char *homePath = getenv("HOME");
+  char *homePath = getenv(HOME_ENV.c_str());
   std::string home;
   if (!homePath)
     home = this->TmpPath() + "/gazebo";

diff --git a/gazebo/gui/DataLogger.cc b/gazebo/gui/DataLogger.cc
@@ -255,7 +255,7 @@ DataLogger::DataLogger(QWidget *_parent)
 
   // Fill the path with the home folder - duplicated from util/LogRecord
 #ifndef _WIN32
-  const char *homePath = common::getEnv("HOME");
+  const char *homePath = common::getEnv(HOME_ENV.c_str());
 #else
   const char *homePath = common::getEnv("HOMEPATH");
 #endif

diff --git a/gazebo/gui/MainWindow.cc b/gazebo/gui/MainWindow.cc
@@ -27,6 +27,7 @@
 #include "gazebo/common/Console.hh"
 #include "gazebo/common/Events.hh"
 #include "gazebo/common/Exception.hh"
+#include "gazebo/common/CommonIface.hh"
 
 #include "gazebo/msgs/msgs.hh"
 
@@ -463,7 +464,13 @@ void MainWindow::Open()
 /////////////////////////////////////////////////
 void MainWindow::SaveINI()
 {
-  char *home = getenv("HOME");
+#ifdef _WIN32
+  std::string HOME_ENV = "USERPROFILE";
+#else
+  std::string HOME_ENV = "HOME";
+#endif
+
+  char *home = getenv(HOME_ENV.c_str());
   if (!home)
   {
     gzerr << "HOME environment variable not found. "

diff --git a/gazebo/gui/SaveEntityDialog.cc b/gazebo/gui/SaveEntityDialog.cc
@@ -19,6 +19,7 @@
 
 #include "gazebo/common/SystemPaths.hh"
 #include "gazebo/common/Console.hh"
+#include "gazebo/common/CommonIface.hh"
 
 #include "gazebo/gui/GuiIface.hh"
 #include "gazebo/gui/SaveEntityDialog.hh"
@@ -408,7 +409,13 @@ void SaveEntityDialog::AddDirToModelPaths(const std::string &_path)
 
     // Save any changes that were made to the property tree
     // TODO: check gui.ini env variable
-    char *home = getenv("HOME");
+#ifdef _WIN32
+  std::string HOME_ENV = "USERPROFILE";
+#else
+  std::string HOME_ENV = "HOME";
+#endif
+
+    char *home = getenv(HOME_ENV.c_str());
     if (home)
     {
       boost::filesystem::path guiINIPath = home;

diff --git a/gazebo/rendering/Camera.cc b/gazebo/rendering/Camera.cc
@@ -44,6 +44,7 @@
 #include "gazebo/common/Console.hh"
 #include "gazebo/common/Exception.hh"
 #include "gazebo/common/VideoEncoder.hh"
+#include "gazebo/common/CommonIface.hh"
 
 #include "gazebo/rendering/ogre_gazebo.h"
 #include "gazebo/rendering/RTShaderSystem.hh"
@@ -101,7 +102,13 @@ Camera::Camera(const std::string &_name, ScenePtr _scene,
 
   this->sceneNode = NULL;
 
-  this->screenshotPath = getenv("HOME");
+#ifdef _WIN32
+  std::string HOME_ENV = "USERPROFILE";
+#else
+  std::string HOME_ENV = "HOME";
+#endif
+
+  this->screenshotPath = getenv(HOME_ENV.c_str());
   this->screenshotPath += "/.gazebo/pictures";
 
   // Connect to the render signal

diff --git a/gazebo/util/Diagnostics.cc b/gazebo/util/Diagnostics.cc
@@ -35,7 +35,7 @@ DiagnosticManager::DiagnosticManager()
 #ifndef _WIN32
   const char *homePath = common::getEnv("HOME");
 #else
-  const char *homePath = common::getEnv("HOMEPATH");
+  const char *homePath = common::getEnv("USERPROFILE");
 #endif
   this->dataPtr->logPath = homePath;
 

diff --git a/gazebo/util/LogRecord.cc b/gazebo/util/LogRecord.cc
@@ -72,7 +72,7 @@ LogRecord::LogRecord()
 
   // Get the user's home directory
 #ifndef _WIN32
-  const char *homePath = common::getEnv("HOME");
+  const char *homePath = common::getEnv(HOME_ENV.c_str());
 #else
   const char *homePath = common::getEnv("HOMEPATH");
 #endif

diff --git a/gazebo/util/LogRecord_TEST.cc b/gazebo/util/LogRecord_TEST.cc
@@ -34,7 +34,7 @@ TEST_F(LogRecord_TEST, Constructor)
   gazebo::util::LogRecord *recorder = gazebo::util::LogRecord::Instance();
 
 #ifndef _WIN32
-  const char *homePath = common::getEnv("HOME");
+  const char *homePath = common::getEnv(HOME_ENV.c_str());
 #else
   const char *homePath = common::getEnv("HOMEPATH");
 #endif

diff --git a/test/integration/plugin.cc b/test/integration/plugin.cc
@@ -17,6 +17,7 @@
 #include <boost/filesystem.hpp>
 
 #include "gazebo/test/ServerFixture.hh"
+#include "gazebo/common/CommonIface.hh"
 
 using namespace gazebo;
 class PluginTest : public ServerFixture
@@ -36,7 +37,7 @@ TEST_F(PluginTest, ModelExceptionConstructor)
 
   world->Step(100);
 
-  char *home = getenv("HOME");
+  char *home = getenv(HOME_ENV.c_str());
   ASSERT_TRUE(home);
 
   boost::filesystem::path path(home);
@@ -71,7 +72,7 @@ TEST_F(PluginTest, ModelExceptionInit)
 
   world->Step(100);
 
-  char *home = getenv("HOME");
+  char *home = getenv(HOME_ENV.c_str());
   ASSERT_TRUE(home);
 
   boost::filesystem::path path(home);
@@ -106,7 +107,7 @@ TEST_F(PluginTest, ModelExceptionLoad)
 
   world->Step(100);
 
-  char *home = getenv("HOME");
+  char *home = getenv(HOME_ENV.c_str());
   ASSERT_TRUE(home);
 
   boost::filesystem::path path(home);

diff --git a/test/integration/sdf_errors.cc b/test/integration/sdf_errors.cc
@@ -41,7 +41,7 @@ class SDFLogsTest : public ServerFixture
   public: void SetUp()
   {
 #ifndef _WIN32
-  const boost::filesystem::path home = common::getEnv("HOME");
+  const boost::filesystem::path home = common::getEnv(HOME_ENV.c_str());
 #else
   const boost::filesystem::path home = common::getEnv("HOMEPATH");
 #endif