storage_scu.h

Go to the documentation of this file.

// BSD 3-Clause License
// Copyright (c) 2021-2025, 🍀☀🌕🌥 🌊
// See the LICENSE file in the project root for full license information.
 
#ifndef PACS_SERVICES_STORAGE_SCU_HPP
#define PACS_SERVICES_STORAGE_SCU_HPP
 
#include "storage_status.h"
#include "kcenon/pacs/core/dicom_dataset.h"
#include "kcenon/pacs/network/association.h"
#include "kcenon/pacs/network/dimse/dimse_message.h"
#include "kcenon/pacs/di/ilogger.h"
 
#include <atomic>
#include <cstdint>
#include <filesystem>
#include <functional>
#include <string>
#include <vector>
 
namespace kcenon::pacs::services {
 
struct store_result {
    std::string sop_instance_uid;
 
    uint16_t status{0};
 
    std::string error_comment;
 
    [[nodiscard]] bool is_success() const noexcept {
        return status == 0x0000;
    }
 
    [[nodiscard]] bool is_warning() const noexcept {
        return (status & 0xF000) == 0xB000;
    }
 
    [[nodiscard]] bool is_error() const noexcept {
        return !is_success() && !is_warning();
    }
};
 
using store_progress_callback = std::function<void(size_t completed, size_t total)>;
 
struct storage_scu_config {
    uint16_t default_priority = 0;
 
    std::chrono::milliseconds response_timeout{30000};
 
    bool continue_on_error = true;
};
 
class storage_scu {
public:
    // =========================================================================
    // Construction
    // =========================================================================
 
    explicit storage_scu(std::shared_ptr<di::ILogger> logger = nullptr);
 
    explicit storage_scu(const storage_scu_config& config,
                         std::shared_ptr<di::ILogger> logger = nullptr);
 
    ~storage_scu() = default;
 
    // Non-copyable, non-movable (due to atomic members)
    storage_scu(const storage_scu&) = delete;
    storage_scu& operator=(const storage_scu&) = delete;
    storage_scu(storage_scu&&) = delete;
    storage_scu& operator=(storage_scu&&) = delete;
 
    // =========================================================================
    // Single Image Operations
    // =========================================================================
 
    [[nodiscard]] network::Result<store_result> store(
        network::association& assoc,
        const core::dicom_dataset& dataset);
 
    // =========================================================================
    // Batch Operations
    // =========================================================================
 
    [[nodiscard]] std::vector<store_result> store_batch(
        network::association& assoc,
        const std::vector<core::dicom_dataset>& datasets,
        store_progress_callback progress_callback = nullptr);
 
    // =========================================================================
    // File-based Operations
    // =========================================================================
 
    [[nodiscard]] network::Result<store_result> store_file(
        network::association& assoc,
        const std::filesystem::path& file_path);
 
    [[nodiscard]] std::vector<store_result> store_files(
        network::association& assoc,
        const std::vector<std::filesystem::path>& file_paths,
        store_progress_callback progress_callback = nullptr);
 
    [[nodiscard]] std::vector<store_result> store_directory(
        network::association& assoc,
        const std::filesystem::path& directory,
        bool recursive = true,
        store_progress_callback progress_callback = nullptr);
 
    // =========================================================================
    // Statistics
    // =========================================================================
 
    [[nodiscard]] size_t images_sent() const noexcept;
 
    [[nodiscard]] size_t failures() const noexcept;
 
    [[nodiscard]] size_t bytes_sent() const noexcept;
 
    void reset_statistics() noexcept;
 
private:
    // =========================================================================
    // Private Implementation
    // =========================================================================
 
    [[nodiscard]] network::Result<store_result> store_impl(
        network::association& assoc,
        const core::dicom_dataset& dataset,
        uint16_t message_id);
 
    [[nodiscard]] uint16_t next_message_id() noexcept;
 
    [[nodiscard]] std::vector<std::filesystem::path> collect_dicom_files(
        const std::filesystem::path& directory,
        bool recursive) const;
 
    // =========================================================================
    // Private Members
    // =========================================================================
 
    std::shared_ptr<di::ILogger> logger_;
 
    storage_scu_config config_;
 
    std::atomic<uint16_t> message_id_counter_{1};
 
    std::atomic<size_t> images_sent_{0};
 
    std::atomic<size_t> failures_{0};
 
    std::atomic<size_t> bytes_sent_{0};
};
 
}  // namespace kcenon::pacs::services
 
#endif  // PACS_SERVICES_STORAGE_SCU_HPP