/*=========================================================================

  Program:   Insight Segmentation & Registration Toolkit
  Module:    $RCSfile$
  Language:  C++
  Date:      $Date$
  Version:   $Revision$

  Copyright (c) 2002 Insight Consortium. All rights reserved.
  See ITKCopyright.txt or http://www.itk.org/HTML/Copyright.htm for details.

     This software is distributed WITHOUT ANY WARRANTY; without even 
     the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR 
     PURPOSE.  See the above copyright notices for more information.

=========================================================================*/
#include "cmCreateTestSourceList.h"


// cmCreateTestSourceList
bool cmCreateTestSourceList::InitialPass(std::vector<std::string> const& argsIn)
{
  if (argsIn.size() < 3)
    {
      this->SetError("called with wrong number of arguments.");
      return false;
    }

  std::vector<std::string> args;
  cmSystemTools::ExpandListArguments(argsIn, args, true);
  
  std::vector<std::string>::iterator i = args.begin();

  // Name of the source list

  const char* sourceList = i->c_str();
  ++i;

  // Name of the test driver

  std::string driver = m_Makefile->GetCurrentOutputDirectory();
  driver += "/";
  driver += *i;
  driver += ".cxx";
  ++i;

  std::ofstream fout(driver.c_str());
  if(!fout)
    {
    std::string err = "Could not create file ";
    err += driver;
    err += " for cmCreateTestSourceList command.";
    this->SetError(err.c_str());
    return false;
    }

  // Create the test driver file

  fout << "#include <stdio.h>\n";
  fout << "#include <string.h>\n";
  fout << "// forward declare test functions\n";

  std::vector<std::string>::iterator testsBegin = i;
  std::vector<std::string> tests_filename;

  // The rest of the arguments consist of a list of test source files.
  // Sadly, they can be in directories. Let's modify each arg to get
  // a unique function name for the corresponding test, and push the 
  // real source filename to the tests_filename var (used at the end). 
  // For the moment:
  //   - replace spaces ' ', ':' and '/' with underscores '_'

  for(i = testsBegin; i != args.end(); ++i)
    {
    tests_filename.push_back(*i);
    cmSystemTools::ConvertToUnixSlashes(*i);
    cmSystemTools::ReplaceString(*i, " ", "_");
    cmSystemTools::ReplaceString(*i, "/", "_");
    cmSystemTools::ReplaceString(*i, ":", "_");
    fout << "int " << *i << "(int, char**);\n";
    }

  fout << "// Create map \n";
  fout << "typedef int (*MainFuncPointer)(int , char**);\n";
  fout << "struct functionMapEntry\n"
       << "{\n"
       << "const char* name;\n"
       << "MainFuncPointer func;\n"
       << "};\n\n";
  fout << "functionMapEntry cmakeGeneratedFunctionMapEntries[] = {\n";

  int numTests = 0;
  for(i = testsBegin; i != args.end(); ++i)
    {
    fout << "{\"" << *i << "\", " << *i << "},\n";
    numTests++;
    }

  fout << "};\n";
  fout << "int main(int ac, char** av)\n"
       << "{\n";
  fout << "  int NumTests = " << numTests << ";\n";
  fout << "  int i;\n";
  fout << "  if(ac < 2)\n";
  fout << "    {\n";
  fout << "    // if there is only one test, then run it with the arguments\n";
  fout << "    if(NumTests == 1)\n";
  fout << "      { return (*cmakeGeneratedFunctionMapEntries[0].func)(ac, av); }\n";
  fout << "    printf(\"Available tests:\\n\");\n";
  fout << "    for(i =0; i < NumTests; ++i)\n";
  fout << "      {\n";
  fout << "      printf(\"%d. %s\\n\", i, cmakeGeneratedFunctionMapEntries[i].name);\n";
  fout << "      }\n";
  fout << "    printf(\"To run a test, enter the test number: \");\n";
  fout << "    int testNum = 0;\n";
  fout << "    scanf(\"%d\", &testNum);\n";
  fout << "    if(testNum >= NumTests)\n";
  fout << "    {\n";
  fout << "    printf(\"%d is an invalid test number.\\n\", testNum);\n";
  fout << "    return -1;\n";
  fout << "    }\n";
  fout << "    return (*cmakeGeneratedFunctionMapEntries[testNum].func)(ac-1, av+1);\n";
  fout << "    }\n";
  fout << "  for(i =0; i < NumTests; ++i)\n";
  fout << "    {\n";
  fout << "    if(strcmp(cmakeGeneratedFunctionMapEntries[i].name, av[1]) == 0)\n";
  fout << "      {\n";
  fout << "      return (*cmakeGeneratedFunctionMapEntries[i].func)(ac-1, av+1);\n";
  fout << "      }\n";
  fout << "    }\n";
  fout << "  // if there is only one test, then run it with the arguments\n";
  fout << "  if(NumTests == 1)\n";
  fout << "    { return (*cmakeGeneratedFunctionMapEntries[0].func)(ac, av); }\n";
  fout << "  printf(\"Available tests:\\n\");\n";
  fout << "  for(i =0; i < NumTests; ++i)\n";
  fout << "    {\n";
  fout << "    printf(\"%d. %s\\n\", i, cmakeGeneratedFunctionMapEntries[i].name);\n";
  fout << "    }\n";
  fout << "  printf(\"Failed: %s is an invalid test name.\\n\", av[1]);\n";
  fout << "  return -1;\n";
  fout << "}\n";
  fout.close();

  // Create the source list

  cmSourceFile cfile;
  cfile.SetIsAnAbstractClass(false);
  cfile.SetName(args[1].c_str(), 
                m_Makefile->GetCurrentOutputDirectory(),
                "cxx", 
                false);
  m_Makefile->AddSource(cfile, sourceList);
  
  for(i = tests_filename.begin(); i != tests_filename.end(); ++i)
    {
    cmSourceFile cfile;
    cfile.SetIsAnAbstractClass(false);
    cfile.SetName(i->c_str(), 
                  m_Makefile->GetCurrentDirectory(),
                  "cxx", 
                  false);
    m_Makefile->AddSource(cfile, sourceList);
    }

  return true;
}