src/Core/Utils.cxx

   1 //  Copyright (C) 2007-2013  CEA/DEN, EDF R&D, OPEN CASCADE
   2 //
   3 //  Copyright (C) 2003-2007  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
   4 //  CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
   5 //
   6 //  This library is free software; you can redistribute it and/or
   7 //  modify it under the terms of the GNU Lesser General Public
   8 //  License as published by the Free Software Foundation; either
   9 //  version 2.1 of the License.
  10 //
  11 //  This library is distributed in the hope that it will be useful,
  12 //  but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 //  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14 //  Lesser General Public License for more details.
  15 //
  16 //  You should have received a copy of the GNU Lesser General Public
  17 //  License along with this library; if not, write to the Free Software
  18 //  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  19 //
  20 //  See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  21 //
  22 /*
  23  * Utils.cxx
  24  *
  25  *  Created on: 30 jan. 2012
  26  *  Author : Renaud BARATE - EDF R&D
  27  */
  28
  29 #include <cstdlib>
  30 #include <cstdio>
  31 #include <iostream>
  32 #include <fstream>
  33
  34 #ifdef WIN32
  35 #include <Windows.h>
  36 #include <time.h>
  37 #include <io.h>
  38 #else
  39 #include <sys/stat.h>
  40 #include <unistd.h>
  41 #endif
  42
  43 #include <config.h>
  44 #include "Utils.hxx"
  45 #include "RunTimeException.hxx"
  46
  47 #ifdef MSVC
  48 #define popen _popen
  49 #define pclose _pclose
  50 #endif
  51
  52 using namespace std;
  53 namespace Batch {
  54
  55 int Utils::getCommandOutput(const string & command, string & output)
  56 {
  57   // Reinitialize output
  58   output = "";
  59
  60   // Call command
  61   FILE * fp = popen(command.c_str(), "r");
  62   if (fp == NULL) {
  63     return -1;
  64   }
  65
  66   // Read the output and store it
  67   char buf[1024];
  68   while (fgets(buf, sizeof(buf), fp) != NULL) {
  69     output += buf;
  70   }
  71
  72   // close and get status
  73   int status = pclose(fp);
  74   return status;
  75 }
  76
  77 bool Utils::isAbsolutePath(const string & path)
  78 {
  79   if (path.size() == 0)
  80     return false;
  81 #ifdef WIN32
  82   // On Windows, absolute paths may begin with something like "C:"
  83   if (path[1] == ':')
  84     return true;
  85 #endif
  86   return path[0] == '/';
  87 }
  88
  89 string Utils::createAndOpenTemporaryFile(const string & prefix, ofstream & outputStream)
  90 {
  91   if (outputStream.is_open())
  92     outputStream.close();
  93
  94   // Find directory for temporary files
  95   const char * tmpDirName = getenv("TEMP");
  96   if (tmpDirName == NULL) tmpDirName = getenv("TMP");
  97   if (tmpDirName == NULL) tmpDirName = getenv("TEMPDIR");
  98   if (tmpDirName == NULL) tmpDirName = getenv("TMPDIR");
  99   if (tmpDirName == NULL) tmpDirName = "/tmp";
 100
 101   string fileName = (string)tmpDirName + "/libbatch-" + prefix + "-XXXXXX";
 102
 103 #ifdef WIN32
 104
 105   char randstr[7];
 106   srand(time(NULL));
 107
 108   do {
 109     sprintf(randstr, "%06d", rand() % 1000000);
 110     fileName.replace(fileName.size()-6, 6, randstr);
 111   } while (_access_s(fileName.c_str(), 0) == 0);
 112
 113   // Open the file as binary to avoid problems with Windows newlines
 114   outputStream.open(fileName.c_str(), ios_base::binary | ios_base::out);
 115
 116 #else
 117
 118   char * buf = new char[fileName.size()+1];
 119   fileName.copy(buf, fileName.size());
 120   buf[fileName.size()] = '\0';
 121
 122   int fd = mkstemp(buf);
 123   if (fd == -1) {
 124     delete[] buf;
 125     throw RunTimeException(string("Can't create temporary file ") + fileName);
 126   }
 127   fileName = buf;
 128   delete[] buf;
 129
 130   outputStream.open(fileName.c_str());
 131   close(fd);  // Close the file descriptor so that the file is not opened twice
 132
 133 #endif
 134
 135   if (outputStream.fail())
 136     throw RunTimeException(string("Can't open temporary file ") + fileName);
 137
 138   return fileName;
 139 }
 140
 141 int Utils::chmod(const char *path, int mode)
 142 {
 143 #ifdef WIN32
 144   return _chmod(path, mode);
 145 #else
 146   return ::chmod(path, mode);
 147 #endif
 148 }
 149
 150 void Utils::sleep(unsigned int seconds)
 151 {
 152 #ifdef WIN32
 153   Sleep((seconds)*1000);
 154 #else
 155   ::sleep(seconds);
 156 #endif
 157 }
 158
 159 string Utils::fixPath(const string & path)
 160 {
 161   string fixedPath = path;
 162 #ifdef WIN32
 163   for (unsigned int i=0 ; i<fixedPath.size() ; i++) {
 164     if (fixedPath[i] == '/') fixedPath[i] = '\\';
 165   }
 166 #endif
 167   return fixedPath;
 168 }
 169
 170 }