X-Git-Url: http://git.salome-platform.org/gitweb/?a=blobdiff_plain;f=src%2FBasics%2FBasics_DirUtils.cxx;h=01d32f935d8902f17bf74715fdbf756af696504b;hb=4d61f5b8863253a259cde301c39b60909a6b18fa;hp=2c7f3338ce643605f1b4b69831f734b646986713;hpb=63414a08d9492c25c206579c1953ec6f390679fd;p=modules%2Fkernel.git diff --git a/src/Basics/Basics_DirUtils.cxx b/src/Basics/Basics_DirUtils.cxx index 2c7f3338c..01d32f935 100644 --- a/src/Basics/Basics_DirUtils.cxx +++ b/src/Basics/Basics_DirUtils.cxx @@ -1,4 +1,4 @@ -// Copyright (C) 2007-2014 CEA/DEN, EDF R&D, OPEN CASCADE +// Copyright (C) 2007-2021 CEA/DEN, EDF R&D, OPEN CASCADE // // This library is free software; you can redistribute it and/or // modify it under the terms of the GNU Lesser General Public @@ -18,19 +18,25 @@ // // File : Basics_DirUtils.cxx -// Autor : Alexander A. BORODIN +// Author : Alexander A. BORODIN // Module : SALOME // #include "Basics_DirUtils.hxx" +#include "Basics_Utils.hxx" #include #include #include +#include + #ifndef WIN32 # include # include # include #else +#include +#define F_OK 0 +#define access _access # include # include #endif @@ -41,89 +47,116 @@ # define _separator_ '/' #endif +#define _extension_ ".hdf" + namespace Kernel_Utils { - std::string GetBaseName( const std::string& file_path ) + std::string GetBaseName( const std::string& file_path, const bool with_extension ) { - int pos = file_path.rfind( _separator_ ); - if ( pos >= 0 ) - return pos < (int)file_path.size()-1 ? file_path.substr( pos+1 ) : ""; - return file_path; + std::string tmp_str = file_path; + auto pos = file_path.rfind( _separator_ ); + if ( pos != std::string::npos ) + tmp_str = pos < file_path.size()-1 ? file_path.substr( pos+1 ) : ""; + + pos = tmp_str.rfind( _extension_ ); + if( !with_extension && pos != std::string::npos ) + tmp_str = pos < tmp_str.size()-1 ? tmp_str.substr( 0, pos ) : ""; + + return tmp_str; } std::string GetDirName( const std::string& file_path ) { - int pos = file_path.rfind( _separator_ ); - if ( pos >= 0 ) - return pos < (int)file_path.size()-1 ? file_path.substr(0, pos ) : ""; + auto pos = file_path.rfind( _separator_ ); + if ( pos != std::string::npos ) + return pos < file_path.size()-1 ? file_path.substr(0, pos ) : ""; return std::string("."); } std::string GetTmpDirByEnv( const std::string& tmp_path_env ) { +#if defined WIN32 && defined UNICODE + std::wstring w_tmp_path_env = utf8_decode_s( tmp_path_env ); + wchar_t* val = _wgetenv( w_tmp_path_env.c_str() ); + std::string dir = val ? utf8_encode_s(val) : ""; +#else char* val = getenv( tmp_path_env.c_str() ); - std::string dir = val ? val : ""; + std::string dir = val ? val : ""; +#endif return GetTmpDirByPath( dir ); } std::string GetTmpDirByPath( const std::string& tmp_path ) { std::string aTmpDir = tmp_path; - if ( aTmpDir == "" ) - { + if ( aTmpDir == "" || !IsExists( aTmpDir )) + { #ifdef WIN32 - char *Tmp_dir = getenv("TEMP"); - if( Tmp_dir == NULL ) - { - Tmp_dir = getenv("TMP"); - if (Tmp_dir == NULL) - aTmpDir = std::string("C:\\"); - else - aTmpDir = std::string(Tmp_dir); - } +#ifdef UNICODE + wchar_t *wTmp_dir = _wgetenv( L"TEMP" ); + if ( wTmp_dir == NULL ) { + wTmp_dir = _wgetenv(L"TMP"); + if (wTmp_dir == NULL) + wTmp_dir = L"C:\\"; + } + aTmpDir = utf8_encode_s( wTmp_dir ); +#else + char *Tmp_dir = getenv("TEMP"); + if ( Tmp_dir == NULL ) + { + Tmp_dir = getenv("TMP"); + if ( Tmp_dir == NULL ) + aTmpDir = "C:\\"; else - aTmpDir = std::string(Tmp_dir); + aTmpDir = Tmp_dir; + } + else + aTmpDir = Tmp_dir; +#endif #else - aTmpDir = std::string("/tmp/"); + aTmpDir = "/tmp/"; #endif - } - - if(aTmpDir[aTmpDir.size()-1] != _separator_) - aTmpDir+=_separator_; - - srand((unsigned int)time(NULL)); + } + + if ( aTmpDir.back() != _separator_ ) + aTmpDir += _separator_; + + srand( (unsigned int)time( NULL )); int aRND = 999 + (int)(100000.0*rand()/(RAND_MAX+1.0)); //Get a random number to present a name of a sub directory char buffer[127]; - sprintf(buffer, "%d", aRND); - std::string aSubDir(buffer); - if(aSubDir.size() <= 1) aSubDir = std::string("123409876"); - + sprintf( buffer, "%d", aRND ); + std::string aSubDir( buffer ); + if ( aSubDir.size() <= 1 ) aSubDir = "123409876"; + aTmpDir += aSubDir; //Get RND sub directory - + std::string aDir = aTmpDir; - - if(IsExists(aDir)) { - for(aRND = 0; IsExists(aDir); aRND++) { - sprintf(buffer, "%d", aRND); - aDir = aTmpDir+buffer; //Build a unique directory name - } + + for ( aRND = 0; IsExists( aDir ); aRND++ ) + { + sprintf( buffer, "%d", aRND ); + aDir = aTmpDir + buffer; //Build a unique directory name } - - if(aDir[aDir.size()-1] != _separator_) aDir += _separator_; - -#ifdef WIN32 - CreateDirectory(aDir.c_str(), NULL); + + if ( aDir.back() != _separator_ ) aDir += _separator_; + +#ifdef WIN32 +#ifdef UNICODE + std::wstring aDirToCreate = utf8_decode_s(aDir); +#else + std::string aDirToCreate = aDir; +#endif + CreateDirectory( aDirToCreate.c_str(), NULL ); #else - mkdir(aDir.c_str(), 0x1ff); + mkdir( aDir.c_str(), 0x1ff ); #endif - return aDir; } - + //============================================================================ // function : GetTempDir - // purpose : Returns a temp directory to store created files like "/tmp/sub_dir/" - //============================================================================ + // purpose : Returns a temp directory to store created files like "/tmp/sub_dir/" + //============================================================================ std::string GetTmpDir() { return GetTmpDirByPath( "" ); @@ -154,22 +187,27 @@ namespace Kernel_Utils return aFilePath; } + std::string AddExtension( const std::string& name ) + { + std::string tmp_str = name; + auto pos = tmp_str.rfind( _extension_ ); + if( pos == std::string::npos ) + return tmp_str.append( _extension_ ); + return tmp_str; + } + //============================================================================ // function : IsExists // purpose : Returns True(False) if the path (not)exists //============================================================================ bool IsExists(const std::string& thePath) { -#ifdef WIN32 - if ( GetFileAttributes ( thePath.c_str() ) == 0xFFFFFFFF ) { - if ( GetLastError () == ERROR_FILE_NOT_FOUND ) { - return false; - } - } -#else +#if defined WIN32 && defined UNICODE + int status = _waccess( utf8_decode_s( thePath).c_str(), F_OK ); +#else int status = access ( thePath.c_str() , F_OK ); - if (status != 0) return false; #endif + if (status != 0) return false; return true; } @@ -177,10 +215,15 @@ namespace Kernel_Utils // function : IsWritable // purpose : Returns True(False) if the path is (not) writable //============================================================================ - bool IsWritable(const std::string& thePath) + bool IsWritable(const std::string& thePath) { -#ifdef WIN32 - if ( GetFileAttributes ( thePath.c_str() ) == 0xFFFFFFFF ) { +#ifdef WIN32 +#ifdef UNICODE + std::wstring aPathToCheck = utf8_decode_s( thePath ); +#else + std::string aPathToCheck = thePath; +#endif + if ( GetFileAttributes ( aPathToCheck.c_str() ) == 0xFFFFFFFF ) { if ( GetLastError () == FILE_ATTRIBUTE_READONLY ) { return false; } @@ -260,7 +303,13 @@ namespace Kernel_Utils #ifdef WIN32 WIN32_FIND_DATA aFileData; - HANDLE hFile = FindFirstFile( thePath.c_str(), &aFileData ); +#ifdef UNICODE + std::wstring aPathToCheck = utf8_decode_s(thePath); +#else + std::string aPathToCheck = thePath; +#endif + + HANDLE hFile = FindFirstFile( aPathToCheck.c_str(), &aFileData ); if ( hFile == INVALID_HANDLE_VALUE ) { //empty dir @@ -292,4 +341,28 @@ namespace Kernel_Utils #endif return result; } + + //============================================================================ + // function : BackSlashToSlash + // purpose : Convert back slash to slash + //============================================================================ + std::string BackSlashToSlash(const std::string& path) { + std::string res = path; + std::replace(res.begin(), res.end(), '\\', '/'); + return res; + } + + + //============================================================================ + // function : BackSlashToSlash + // purpose : Convert back slash to slash + //============================================================================ + std::string HomePath() { +#ifdef WIN32 + std::string homedir = getenv("USERPROFILE"); +#else + std::string homedir = getenv("HOME"); +#endif + return homedir; + } }