diff options
Diffstat (limited to 'src/armnnUtils')
-rw-r--r-- | src/armnnUtils/Filesystem.cpp | 53 | ||||
-rw-r--r-- | src/armnnUtils/Filesystem.hpp | 5 | ||||
-rw-r--r-- | src/armnnUtils/ModelAccuracyChecker.cpp | 1 |
3 files changed, 18 insertions, 41 deletions
diff --git a/src/armnnUtils/Filesystem.cpp b/src/armnnUtils/Filesystem.cpp index 4a2e2ed393..ac9a414ae4 100644 --- a/src/armnnUtils/Filesystem.cpp +++ b/src/armnnUtils/Filesystem.cpp @@ -5,51 +5,30 @@ #include "Filesystem.hpp" -#if defined(__unix__) -#include <sys/stat.h> -#include <stdio.h> -#elif defined(_MSC_VER) -#include "WindowsWrapper.hpp" -#endif - namespace armnnUtils { namespace Filesystem { -long long GetFileSize(const char* path) +/** + * @brief Construct a temporary file name. + * + * Given a specified file name construct a path to that file in the + * system temporary directory. If the file already exists it is deleted. This + * could throw filesystem_error exceptions. + * + * @param fileName the file name required in the temporary directory. + * @return path consisting of system temporary directory and file name. + */ +fs::path NamedTempFile(const char* fileName) { -#if defined(__ANDROID__) - struct stat statusBuffer; - if (stat(path, & statusBuffer) != 0) - { - return -1; - } - return statusBuffer.st_size; -#elif defined(__unix__) - struct stat statusBuffer; - if (stat(path, & statusBuffer) != 0) - { - return -1; - } - return static_cast<long long>(statusBuffer.st_size); -#elif defined(_MSC_VER) - WIN32_FILE_ATTRIBUTE_DATA attr; - if (::GetFileAttributesEx(path, GetFileExInfoStandard, &attr) == 0) + fs::path tmpDir = fs::temp_directory_path(); + fs::path namedTempFile{tmpDir / fileName}; + if (fs::exists(namedTempFile)) { - return -1; + fs::remove(namedTempFile); } - return attr.nFileSizeLow; -#endif -} - -bool Remove(const char* path) -{ -#if defined(__unix__) - return remove(path) == 0; -#elif defined(_MSC_VER) - return ::DeleteFile(path); -#endif + return namedTempFile; } } diff --git a/src/armnnUtils/Filesystem.hpp b/src/armnnUtils/Filesystem.hpp index 1b12502b40..869b0c1b5e 100644 --- a/src/armnnUtils/Filesystem.hpp +++ b/src/armnnUtils/Filesystem.hpp @@ -13,9 +13,8 @@ namespace armnnUtils namespace Filesystem { -long long GetFileSize(const char* path); - -bool Remove(const char* path); +/// Returns a path to a file in the system temporary folder. If the file existed it will be deleted. +fs::path NamedTempFile(const char* fileName); } } diff --git a/src/armnnUtils/ModelAccuracyChecker.cpp b/src/armnnUtils/ModelAccuracyChecker.cpp index d197dc86ba..418737c0f6 100644 --- a/src/armnnUtils/ModelAccuracyChecker.cpp +++ b/src/armnnUtils/ModelAccuracyChecker.cpp @@ -7,7 +7,6 @@ #include <armnn/Logging.hpp> -#include <boost/filesystem.hpp> #include <map> #include <vector> |