1 files changed, 163 insertions, 0 deletions
diff --git a/Source/cmCreateTestSourceList.cxx b/Source/cmCreateTestSourceList.cxx
new file mode 100644
index 0000000..d8c629c
--- /dev/null
+++ b/Source/cmCreateTestSourceList.cxx
@@ -0,0 +1,163 @@
+/*============================================================================
+  CMake - Cross Platform Makefile Generator
+  Copyright 2000-2009 Kitware, Inc., Insight Software Consortium
+
+  Distributed under the OSI-approved BSD License (the "License");
+  see accompanying file Copyright.txt for details.
+
+  This software is distributed WITHOUT ANY WARRANTY; without even the
+  implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
+  See the License for more information.
+============================================================================*/
+#include "cmCreateTestSourceList.h"
+
+#include "cmSourceFile.h"
+
+// cmCreateTestSourceList
+bool cmCreateTestSourceList::InitialPass(std::vector<std::string> const& args,
+                                         cmExecutionStatus&)
+{
+  if (args.size() < 3) {
+    this->SetError("called with wrong number of arguments.");
+    return false;
+  }
+
+  std::vector<std::string>::const_iterator i = args.begin();
+  std::string extraInclude;
+  std::string function;
+  std::vector<std::string> tests;
+  // extract extra include and function ot
+  for (; i != args.end(); i++) {
+    if (*i == "EXTRA_INCLUDE") {
+      ++i;
+      if (i == args.end()) {
+        this->SetError("incorrect arguments to EXTRA_INCLUDE");
+        return false;
+      }
+      extraInclude = "#include \"";
+      extraInclude += *i;
+      extraInclude += "\"\n";
+    } else if (*i == "FUNCTION") {
+      ++i;
+      if (i == args.end()) {
+        this->SetError("incorrect arguments to FUNCTION");
+        return false;
+      }
+      function = *i;
+      function += "(&ac, &av);\n";
+    } else {
+      tests.push_back(*i);
+    }
+  }
+  i = tests.begin();
+
+  // Name of the source list
+
+  const char* sourceList = i->c_str();
+  ++i;
+
+  // Name of the test driver
+  // make sure they specified an extension
+  if (cmSystemTools::GetFilenameExtension(*i).size() < 2) {
+    this->SetError(
+      "You must specify a file extension for the test driver file.");
+    return false;
+  }
+  std::string driver = this->Makefile->GetCurrentBinaryDirectory();
+  driver += "/";
+  driver += *i;
+  ++i;
+
+  std::string configFile = cmSystemTools::GetCMakeRoot();
+
+  configFile += "/Templates/TestDriver.cxx.in";
+  // Create the test driver file
+
+  std::vector<std::string>::const_iterator testsBegin = i;
+  std::vector<std::string> tests_func_name;
+
+  // The rest of the arguments consist of a list of test source files.
+  // Sadly, they can be in directories. Let's find a unique function
+  // name for the corresponding test, and push it to the tests_func_name
+  // list.
+  // For the moment:
+  //   - replace spaces ' ', ':' and '/' with underscores '_'
+  std::string forwardDeclareCode;
+  for (i = testsBegin; i != tests.end(); ++i) {
+    if (*i == "EXTRA_INCLUDE") {
+      break;
+    }
+    std::string func_name;
+    if (!cmSystemTools::GetFilenamePath(*i).empty()) {
+      func_name = cmSystemTools::GetFilenamePath(*i) + "/" +
+        cmSystemTools::GetFilenameWithoutLastExtension(*i);
+    } else {
+      func_name = cmSystemTools::GetFilenameWithoutLastExtension(*i);
+    }
+    cmSystemTools::ConvertToUnixSlashes(func_name);
+    std::replace(func_name.begin(), func_name.end(), ' ', '_');
+    std::replace(func_name.begin(), func_name.end(), '/', '_');
+    std::replace(func_name.begin(), func_name.end(), ':', '_');
+    tests_func_name.push_back(func_name);
+    forwardDeclareCode += "int ";
+    forwardDeclareCode += func_name;
+    forwardDeclareCode += "(int, char*[]);\n";
+  }
+
+  std::string functionMapCode;
+  int numTests = 0;
+  std::vector<std::string>::iterator j;
+  for (i = testsBegin, j = tests_func_name.begin(); i != tests.end();
+       ++i, ++j) {
+    std::string func_name;
+    if (!cmSystemTools::GetFilenamePath(*i).empty()) {
+      func_name = cmSystemTools::GetFilenamePath(*i) + "/" +
+        cmSystemTools::GetFilenameWithoutLastExtension(*i);
+    } else {
+      func_name = cmSystemTools::GetFilenameWithoutLastExtension(*i);
+    }
+    functionMapCode += "  {\n"
+                       "    \"";
+    functionMapCode += func_name;
+    functionMapCode += "\",\n"
+                       "    ";
+    functionMapCode += *j;
+    functionMapCode += "\n"
+                       "  },\n";
+    numTests++;
+  }
+  if (!extraInclude.empty()) {
+    this->Makefile->AddDefinition("CMAKE_TESTDRIVER_EXTRA_INCLUDES",
+                                  extraInclude.c_str());
+  }
+  if (!function.empty()) {
+    this->Makefile->AddDefinition("CMAKE_TESTDRIVER_ARGVC_FUNCTION",
+                                  function.c_str());
+  }
+  this->Makefile->AddDefinition("CMAKE_FORWARD_DECLARE_TESTS",
+                                forwardDeclareCode.c_str());
+  this->Makefile->AddDefinition("CMAKE_FUNCTION_TABLE_ENTIRES",
+                                functionMapCode.c_str());
+  bool res = true;
+  if (!this->Makefile->ConfigureFile(configFile.c_str(), driver.c_str(), false,
+                                     true, false)) {
+    res = false;
+  }
+
+  // Construct the source list.
+  std::string sourceListValue;
+  {
+    cmSourceFile* sf = this->Makefile->GetOrCreateSource(driver);
+    sf->SetProperty("ABSTRACT", "0");
+    sourceListValue = args[1];
+  }
+  for (i = testsBegin; i != tests.end(); ++i) {
+    cmSourceFile* sf = this->Makefile->GetOrCreateSource(*i);
+    sf->SetProperty("ABSTRACT", "0");
+    sourceListValue += ";";
+    sourceListValue += *i;
+  }
+
+  this->Makefile->AddDefinition(sourceList, sourceListValue.c_str());
+  return res;
+}