MRDirectory.h

Go to the documentation of this file.

#pragma once
 
#include "MRMeshFwd.h"
#include <filesystem>
 
namespace MR
{
 

struct DirectoryIterator
{
    std::filesystem::directory_iterator it;
    std::error_code & ec;
    DirectoryIterator & operator ++() { it.increment( ec ); return * this; }
    auto operator *() const { return *it; }
};

struct Directory
{
    std::filesystem::path dir;
    std::error_code & ec;
 
    [[nodiscard]] friend DirectoryIterator begin( const Directory & sd )
    {
        return DirectoryIterator{ std::filesystem::directory_iterator( sd.dir, sd.ec ), sd.ec };
    }
 
    [[nodiscard]] friend std::filesystem::directory_iterator end( const Directory & )
    {
        return {};
    }
};

struct DirectoryRecursiveIterator
{
    std::filesystem::recursive_directory_iterator it;
    std::error_code & ec;
    DirectoryRecursiveIterator & operator ++() { it.increment( ec ); return * this; }
    auto operator *() const { return *it; }
};

struct DirectoryRecursive
{
    std::filesystem::path dir;
    std::error_code & ec;
 
    [[nodiscard]] friend DirectoryRecursiveIterator begin( const DirectoryRecursive & sd )
    {
        return DirectoryRecursiveIterator{ std::filesystem::recursive_directory_iterator( sd.dir, sd.ec ), sd.ec };
    }
 
    [[nodiscard]] friend std::filesystem::recursive_directory_iterator end( const DirectoryRecursive & )
    {
        return {};
    }
};

[[nodiscard]] MRMESH_API std::filesystem::path findPathWithExtension( const std::filesystem::path & pathWithoutExtension );
 
[[nodiscard]] inline bool operator !=( const DirectoryIterator & a, const std::filesystem::directory_iterator & b )
{
    return !a.ec && a.it != b;
}
 
[[nodiscard]] inline bool operator !=( const DirectoryRecursiveIterator & a, const std::filesystem::recursive_directory_iterator & b )
{
    return !a.ec && a.it != b;
}
 
}