image_transfer_shape.hpp

// SPDX-License-Identifier: LGPL-2.1-or-later
 
#pragma once
 
#include <vitrio/export.hpp>
#include <vitrio/span.hpp>
 
#include <cstddef>
#include <vector>
 
namespace vitrio
{
 
class image_transfer_shape
{
public:
    VITRIO_API
    image_transfer_shape(
        std::vector<std::size_t> extents,
        std::size_t file_rank,
        std::size_t array_rank
    );
 
    VITRIO_API
    image_transfer_shape(const image_transfer_shape &other);
    VITRIO_API
    image_transfer_shape(image_transfer_shape &&other) noexcept;
    VITRIO_API
    ~image_transfer_shape();
 
    VITRIO_API
    image_transfer_shape& operator=(const image_transfer_shape &other);
    VITRIO_API
    image_transfer_shape& operator=(image_transfer_shape &&other) noexcept;
 
    VITRIO_API
    span<const std::size_t> get_extents() const noexcept;
 
    VITRIO_API
    std::size_t get_rank() const noexcept;
 
    VITRIO_API
    std::size_t get_file_rank() const noexcept;
 
    VITRIO_API
    std::size_t get_array_rank() const noexcept;
 
    VITRIO_API
    std::size_t get_leading_rank(std::size_t rank) const noexcept;
 
    VITRIO_API
    std::size_t get_extent(std::size_t rank, std::size_t axis) const noexcept;
 
private:
    std::vector<std::size_t> m_extents;
    std::size_t m_file_rank;
    std::size_t m_array_rank;
};
 
} // namespace vitrio