storage_scp.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_SCP_HPP
#define PACS_SERVICES_STORAGE_SCP_HPP
 
#include "scp_service.h"
#include "storage_status.h"
#include "kcenon/pacs/core/dicom_dataset.h"
 
#include <atomic>
#include <functional>
#include <memory>
#include <string>
#include <vector>
 
namespace kcenon::pacs::security { class atna_service_auditor; }
 
namespace kcenon::pacs::services {
 
enum class duplicate_policy {
    reject,     
    replace,    
    ignore      
};
 
struct storage_scp_config {
    std::vector<std::string> accepted_sop_classes;
 
    duplicate_policy dup_policy = duplicate_policy::reject;
};
 
using storage_handler = std::function<storage_status(
    const core::dicom_dataset& dataset,
    const std::string& calling_ae,
    const std::string& sop_class_uid,
    const std::string& sop_instance_uid)>;
 
using pre_store_handler = std::function<bool(const core::dicom_dataset& dataset)>;
 
using post_store_handler = std::function<void(
    const core::dicom_dataset& dataset,
    const std::string& patient_id,
    const std::string& study_uid,
    const std::string& series_uid,
    const std::string& sop_instance_uid)>;
 
class storage_scp final : public scp_service {
public:
    // =========================================================================
    // Construction
    // =========================================================================
 
    explicit storage_scp(std::shared_ptr<di::ILogger> logger = nullptr);
 
    explicit storage_scp(const storage_scp_config& config,
                         std::shared_ptr<di::ILogger> logger = nullptr);
 
    ~storage_scp() override = default;
 
    // =========================================================================
    // Handler Registration
    // =========================================================================
 
    void set_handler(storage_handler handler);
 
    void set_pre_store_handler(pre_store_handler handler);
 
    void set_post_store_handler(post_store_handler handler);
 
    void set_audit_handler(
        std::shared_ptr<kcenon::pacs::security::atna_service_auditor> auditor);
 
    // =========================================================================
    // scp_service Interface Implementation
    // =========================================================================
 
    [[nodiscard]] std::vector<std::string> supported_sop_classes() const override;
 
    [[nodiscard]] network::Result<std::monostate> handle_message(
        network::association& assoc,
        uint8_t context_id,
        const network::dimse::dimse_message& request) override;
 
    [[nodiscard]] std::string_view service_name() const noexcept override;
 
    // =========================================================================
    // Statistics
    // =========================================================================
 
    [[nodiscard]] size_t images_received() const noexcept;
 
    [[nodiscard]] size_t bytes_received() const noexcept;
 
    void reset_statistics() noexcept;
 
private:
    // =========================================================================
    // Private Members
    // =========================================================================
 
    storage_scp_config config_;
 
    storage_handler handler_;
 
    pre_store_handler pre_store_handler_;
 
    post_store_handler post_store_handler_;
 
    std::shared_ptr<kcenon::pacs::security::atna_service_auditor> auditor_;
 
    std::atomic<size_t> images_received_{0};
 
    std::atomic<size_t> bytes_received_{0};
};
 
// =============================================================================
// Standard Storage SOP Class UIDs
// =============================================================================
 
 
inline constexpr std::string_view ct_image_storage_uid =
    "1.2.840.10008.5.1.4.1.1.2";
 
inline constexpr std::string_view enhanced_ct_image_storage_uid =
    "1.2.840.10008.5.1.4.1.1.2.1";
 
inline constexpr std::string_view mr_image_storage_uid =
    "1.2.840.10008.5.1.4.1.1.4";
 
inline constexpr std::string_view enhanced_mr_image_storage_uid =
    "1.2.840.10008.5.1.4.1.1.4.1";
 
inline constexpr std::string_view cr_image_storage_uid =
    "1.2.840.10008.5.1.4.1.1.1";
 
inline constexpr std::string_view dx_image_storage_presentation_uid =
    "1.2.840.10008.5.1.4.1.1.1.1";
 
inline constexpr std::string_view dx_image_storage_processing_uid =
    "1.2.840.10008.5.1.4.1.1.1.1.1";
 
inline constexpr std::string_view us_image_storage_uid =
    "1.2.840.10008.5.1.4.1.1.6.1";
 
inline constexpr std::string_view secondary_capture_image_storage_uid =
    "1.2.840.10008.5.1.4.1.1.7";
 
inline constexpr std::string_view rt_image_storage_uid =
    "1.2.840.10008.5.1.4.1.1.481.1";
 
inline constexpr std::string_view rt_dose_storage_uid =
    "1.2.840.10008.5.1.4.1.1.481.2";
 
inline constexpr std::string_view rt_structure_set_storage_uid =
    "1.2.840.10008.5.1.4.1.1.481.3";
 
inline constexpr std::string_view rt_plan_storage_uid =
    "1.2.840.10008.5.1.4.1.1.481.5";
 
 
[[nodiscard]] std::vector<std::string> get_standard_storage_sop_classes();
 
}  // namespace kcenon::pacs::services
 
#endif  // PACS_SERVICES_STORAGE_SCP_HPP