tests/tools/toolTestRunner.cxx

   1 /*=========================================================================
   2   Program:   vv                     http://www.creatis.insa-lyon.fr/rio/vv
   3
   4   Authors belong to:
   5   - University of LYON              http://www.universite-lyon.fr/
   6   - Léon Bérard cancer center       http://www.centreleonberard.fr
   7   - CREATIS CNRS laboratory         http://www.creatis.insa-lyon.fr
   8
   9   This software is distributed WITHOUT ANY WARRANTY; without even
  10   the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
  11   PURPOSE.  See the copyright notices for more information.
  12
  13   It is distributed under dual licence
  14
  15   - BSD        See included LICENSE.txt file
  16   - CeCILL-B   http://www.cecill.info/licences/Licence_CeCILL-B_V1-en.html
  17 ===========================================================================*/
  18 #include <iostream>
  19 #include <sstream>
  20 #include <string>
  21 #include <fstream>
  22 #include <stdio.h>
  23 #include <stdlib.h>
  24 #include <itksys/SystemTools.hxx>
  25 const int NO_OUTPUT_OPTION=-1;
  26 const int TEST_EXITED=1;
  27 int getOutputOptionIndex(int argc, char** argv){
  28   for(int i=1; i<argc; i++){
  29       std::string s = argv[i];
  30       if(s=="-o"){
  31          return i+1;
  32       }
  33   }
  34   return NO_OUTPUT_OPTION;
  35 }
  36
  37 std::string getTmpFileName(){
  38 #ifdef _WIN32
  39   char fileName[L_tmpnam_s];
  40   errno_t err = tmpnam_s(fileName);
  41 #else
  42   char fileName[] = "/tmp/vvTempXXXXXX";
  43   int err=0;
  44   int fd = mkstemp(fileName);
  45   if(fd==-1) err=1;
  46 #endif
  47   if(err){
  48    std::cout<<"couldnot create file. Exiting"<<std::endl;
  49    exit(TEST_EXITED);
  50   }
  51   return std::string(fileName);
  52 }
  53
  54 void assertFalse(int fail, const std::string &message=""){
  55   if(fail){
  56     std::cout<<message<<std::endl;
  57     exit(1);
  58   }
  59 }
  60
  61 #ifdef _WIN32
  62 void dosToUnixFile(std::string dosFile, std::string unixedFile){
  63
  64         std::ifstream ifile(dosFile.c_str(),std::ios::binary);
  65         ifile.seekg(0,std::ios_base::end);
  66         long s=ifile.tellg();
  67         char *buffer=new char[s];
  68         ifile.seekg(0);
  69         ifile.read(buffer,s);
  70         ifile.close();
  71         std::string txt(buffer,s);
  72         delete[] buffer;
  73         size_t off=0;
  74         while ((off=txt.find("\r\n",off))!=std::string::npos)
  75                 txt.replace(off,sizeof("\r\n")-1,"\n");
  76         std::ofstream ofile(unixedFile.c_str());
  77         ofile.write(txt.c_str(),txt.size());
  78
  79 }
  80 #endif
  81
  82 std::string mhdToRawName(const std::string &mhdName){
  83         int found = mhdName.find_last_of(".");
  84   return mhdName.substr(0, found)+".raw";
  85 }
  86 /**
  87  * argv
  88  * [1] executable
  89  * [2] random options
  90  * [2.x] -o
  91  * [3] reference file
  92  *
  93  * [2.x] is optional. If set a temporary file will be generated. So NO need to pass a random outputFileName
  94  */
  95 int main(int argc, char** argv){
  96   //reference file must exist or we fail
  97   char* refFile = argv[argc-1];
  98         std::string strRefFile = std::string(refFile);
  99   assertFalse(!(itksys::SystemTools::FileExists(refFile, true)), "refFile "+strRefFile+" doesn't exist");
 100
 101   std::ostringstream cmd_line;
 102   cmd_line<<CLITK_TEST_TOOLS_PATH;
 103   for(int i=1; i<argc-1; i++){
 104       //we should ensure the file exists, find an -i index or a long file name maybe?
 105       cmd_line<<argv[i]<<" ";
 106   }
 107
 108   //look for the need of generating an output file
 109   int outputOptionIndex = getOutputOptionIndex(argc, argv);
 110   std::string outFile;
 111   if(NO_OUTPUT_OPTION==outputOptionIndex){
 112      outFile = getTmpFileName();
 113      cmd_line<<">"<<outFile;
 114   }else{
 115      outFile = argv[argc-2];
 116   }
 117   std::cout<<cmd_line.str()<<std::endl;;
 118   //run the command line
 119   system(cmd_line.str().c_str());
 120
 121
 122         //compare source files
 123 #ifdef _WIN32
 124         std::string unixedOutFile= getTmpFileName();
 125         //replace \r\n
 126         dosToUnixFile(outFile, unixedOutFile);
 127         assertFalse((itksys::SystemTools::FilesDiffer(unixedOutFile.c_str(), refFile)), "Generated mhd file != ref File");
 128   remove(unixedOutFile.c_str());
 129 #else
 130   assertFalse((itksys::SystemTools::FilesDiffer(outFile.c_str(), refFile)), "Generated mhd file != ref File");
 131 #endif
 132
 133   std::string refRawFile = mhdToRawName(strRefFile);
 134   std::string rawFile = mhdToRawName(outFile);
 135
 136   if((itksys::SystemTools::FileExists(refRawFile.c_str(), true))){
 137     //compare the raw stuff
 138     if((itksys::SystemTools::FileExists(rawFile.c_str(), true))){
 139        std::cout<<"Checking raws"<<std::endl;
 140        assertFalse(itksys::SystemTools::FilesDiffer(refRawFile.c_str(), rawFile.c_str()), "Raws are different");
 141     }
 142     //file is not removed if there is a fail
 143     remove(rawFile.c_str());
 144   }
 145   //neither the mhd is
 146   remove(outFile.c_str());
 147
 148   //success
 149   return 0;
 150 }
 151
 152