1 files changed, 80 insertions, 0 deletions
diff --git a/src/Utils.cxx b/src/Utils.cxx
new file mode 100644
index 0000000..bc556e3
--- /dev/null
+++ b/src/Utils.cxx
@@ -0,0 +1,80 @@
+/*
+  Copyright Kitware, Inc.
+
+  Licensed under the Apache License, Version 2.0 (the "License");
+  you may not use this file except in compliance with the License.
+  You may obtain a copy of the License at
+
+      http://www.apache.org/licenses/LICENSE-2.0
+
+  Unless required by applicable law or agreed to in writing, software
+  distributed under the License is distributed on an "AS IS" BASIS,
+  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+  See the License for the specific language governing permissions and
+  limitations under the License.
+*/
+
+#include "Utils.h"
+
+#include <cxsys/SystemTools.hxx>
+#include <llvm/Support/FileSystem.h>
+#include <fstream>
+
+static std::string castxmlResourceDir;
+
+//----------------------------------------------------------------------------
+static std::string GetMainExecutable(const char* argv0)
+{
+  return llvm::sys::fs::getMainExecutable
+    (argv0, (void*)(intptr_t)GetMainExecutable);
+}
+
+//----------------------------------------------------------------------------
+static bool tryBuildDir(std::string const& dir)
+{
+  // Build tree has
+  //   <build>/CMakeFiles/castxmlSourceDir.txt
+  std::string src_dir_txt = dir + "/CMakeFiles/castxmlSourceDir.txt";
+  std::ifstream src_fin(src_dir_txt.c_str());
+  std::string src_dir;
+  if(src_fin && cxsys::SystemTools::GetLineFromStream(src_fin, src_dir) &&
+     cxsys::SystemTools::FileIsDirectory(src_dir.c_str())) {
+    castxmlResourceDir = src_dir + "/share/castxml";
+    return true;
+  }
+  return false;
+}
+
+//----------------------------------------------------------------------------
+bool findResourceDir(const char* argv0, std::ostream& error)
+{
+  std::string exe = GetMainExecutable(argv0);
+  if(!cxsys::SystemTools::FileIsFullPath(exe.c_str())) {
+    error << "error: unable to locate " << argv0 << "\n";
+    return false;
+  }
+  std::string exe_dir = cxsys::SystemTools::GetFilenamePath(exe);
+
+  // Install tree has
+  //   <prefix>/bin/castxml
+  //   <prefix>/<CASTXML_INSTALL_DATA_DIR>
+  std::string dir = cxsys::SystemTools::GetFilenamePath(exe_dir);
+  castxmlResourceDir = dir + "/" + CASTXML_INSTALL_DATA_DIR;
+  if(!cxsys::SystemTools::FileIsDirectory(castxmlResourceDir.c_str())) {
+    // Build tree has
+    //   <build>/bin[/<config>]/castxml
+    if(!tryBuildDir(dir) &&
+       !tryBuildDir(cxsys::SystemTools::GetFilenamePath(dir))) {
+      error << "Unable to locate resources for " << exe << "\n";
+      return false;
+    }
+  }
+
+  return true;
+}
+
+//----------------------------------------------------------------------------
+std::string getResourceDir()
+{
+  return castxmlResourceDir;
+}