ArchiveReader.h

Go to the documentation of this file.

#pragma once
 
#include <bit>
#include <filesystem>
#include <memory>
#include <string>
#include <vector>
 
#include "storm-config.h"
 
#ifdef STORM_HAVE_LIBARCHIVE
#include <archive.h>
#include <archive_entry.h>
#endif
 
namespace storm {
 
namespace storage {
class BitVector;
}
 
namespace io {
static_assert(std::endian::native == std::endian::little || std::endian::native == std::endian::big, "This code is not supported for mixed endian systems.");
 
class ArchiveReader {
   private:
#ifdef STORM_HAVE_LIBARCHIVE
    struct ArchiveDeleter {
        void operator()(archive* arch) const noexcept;
    };
#endif
   public:
    class ArchiveReadEntry {
       public:
#ifdef STORM_HAVE_LIBARCHIVE
        ArchiveReadEntry(archive_entry* currentEntry, archive* archive);
#endif
 
        std::filesystem::path name() const;
 
        bool isDir() const;
 
        template<typename T>
        using VectorType = std::conditional_t<std::is_same_v<T, bool>, storm::storage::BitVector, std::vector<T>>;
 
        template<typename T, std::endian Endianness = std::endian::little>
            requires(std::is_arithmetic_v<T>)
        VectorType<T> toVector();
 
        std::string toString();
 
       private:
#ifdef STORM_HAVE_LIBARCHIVE
        static constexpr size_t BufferSize = 8192;
        archive_entry* const _currentEntry;
        archive* _archive;
#endif
    };
    class Iterator {
       public:
        Iterator() = default;
#ifdef STORM_HAVE_LIBARCHIVE
        Iterator(std::filesystem::path const& filename);
#endif
 
        bool operator==(Iterator const& other) const;
        bool operator!=(Iterator const& other) const;
 
        Iterator& operator++();
 
        ArchiveReadEntry operator*() const;
 
       private:
#ifdef STORM_HAVE_LIBARCHIVE
        std::unique_ptr<archive, ArchiveDeleter> _archive;
        archive_entry* _currentEntry{nullptr};
#endif
    };
 
    ArchiveReader(std::filesystem::path const& file);
 
    bool isReadableArchive() const;
 
    Iterator begin() const;
 
    Iterator end() const;
 
   private:
    std::filesystem::path const file;
};
 
ArchiveReader openArchive(std::filesystem::path const& file);
}  // namespace io
}  // namespace storm