test_data_generator.h

Go to the documentation of this file.

 
#pragma once
 
#include <kcenon/pacs/core/dicom_dataset.h>
 
#include <atomic>
#include <cstdint>
#include <filesystem>
#include <string>
#include <string_view>
#include <vector>
 
namespace kcenon::pacs::samples {
 
// ============================================================================
// Data Structures
// ============================================================================
 
struct patient_info {
    std::string patient_id{"PAT001"};
    std::string patient_name{"DOE^JOHN"};
    std::string birth_date{"19800101"};
    std::string sex{"M"};
};
 
struct study_info {
    std::string study_uid;
    std::string study_date;
    std::string study_time;
    std::string accession_number;
    std::string description;
    std::string referring_physician;
};
 
struct image_params {
    uint16_t rows{512};
    uint16_t columns{512};
    uint16_t bits_allocated{16};
    uint16_t bits_stored{12};
    uint16_t high_bit{11};
    uint16_t pixel_representation{0};  // 0 = unsigned, 1 = signed
    std::string modality{"CT"};
    std::string photometric{"MONOCHROME2"};
    double window_center{40.0};
    double window_width{400.0};
};
 
// ============================================================================
// Test Data Generator
// ============================================================================
 
class test_data_generator {
public:
    // ========================================================================
    // Single Dataset Generation
    // ========================================================================
 
    [[nodiscard]] static auto create_ct_dataset(
        const patient_info& patient = {},
        const image_params& params = {}) -> core::dicom_dataset;
 
    [[nodiscard]] static auto create_mr_dataset(
        const patient_info& patient = {},
        image_params params = {}) -> core::dicom_dataset;
 
    [[nodiscard]] static auto create_cr_dataset(
        const patient_info& patient = {},
        image_params params = {}) -> core::dicom_dataset;
 
    [[nodiscard]] static auto create_dx_dataset(
        const patient_info& patient = {},
        image_params params = {}) -> core::dicom_dataset;
 
    // ========================================================================
    // Study Generation
    // ========================================================================
 
    [[nodiscard]] static auto create_study(
        const patient_info& patient,
        const study_info& study,
        int series_count = 1,
        int instances_per_series = 5) -> std::vector<core::dicom_dataset>;
 
    // ========================================================================
    // File Operations
    // ========================================================================
 
    static void save_to_directory(
        const std::vector<core::dicom_dataset>& datasets,
        const std::filesystem::path& directory);
 
    // ========================================================================
    // UID Generation
    // ========================================================================
 
    [[nodiscard]] static auto generate_uid(
        std::string_view root = "1.2.410.200001.1.1") -> std::string;
 
    // ========================================================================
    // Date/Time Utilities
    // ========================================================================
 
    [[nodiscard]] static auto current_date() -> std::string;
 
    [[nodiscard]] static auto current_time() -> std::string;
 
private:
    [[nodiscard]] static auto create_base_dataset(
        const patient_info& patient,
        const image_params& params) -> core::dicom_dataset;
 
    [[nodiscard]] static auto generate_pixel_data(
        uint16_t rows,
        uint16_t columns,
        uint16_t bits_allocated) -> std::vector<uint8_t>;
 
    static std::atomic<uint64_t> uid_counter_;
};
 
}  // namespace kcenon::pacs::samples