formatted_index_size.hpp

Go to the documentation of this file.

// SPDX-FileCopyrightText: 2006-2024 Knut Reinert & Freie Universität Berlin
// SPDX-FileCopyrightText: 2016-2024 Knut Reinert & MPI für molekulare Genetik
// SPDX-License-Identifier: BSD-3-Clause
 
#pragma once
 
#include <cassert>
#include <cstddef>
#include <filesystem>
#include <string>
 
#include <raptor/argument_parsing/formatted_bytes.hpp>
 
namespace raptor
{
 
[[nodiscard]] inline size_t index_size_in_KiB(std::filesystem::path index_path, uint8_t const parts)
{
    size_t index_size_in_bytes{};
    if (parts == 1u)
    {
        index_size_in_bytes = std::filesystem::file_size(index_path);
    }
    else
    {
        for (size_t part = 0u; part < parts; ++part)
        {
            index_size_in_bytes += std::filesystem::file_size(index_path.string() + "_" + std::to_string(part));
        }
    }
 
    return index_size_in_bytes >> 10;
}
 
[[nodiscard]] inline std::string formatted_index_size(std::filesystem::path index_path, uint8_t const parts)
{
    return formatted_bytes(index_size_in_KiB(index_path, parts) << 10);
}
 
} // namespace raptor