test_data_generator.h

Go to the documentation of this file.

 
#ifndef PACS_INTEGRATION_TESTS_TEST_DATA_GENERATOR_HPP
#define PACS_INTEGRATION_TESTS_TEST_DATA_GENERATOR_HPP
 
#include "kcenon/pacs/core/dicom_dataset.h"
#include "kcenon/pacs/core/dicom_tag_constants.h"
#include "kcenon/pacs/services/sop_classes/us_storage.h"
#include "kcenon/pacs/services/sop_classes/xa_storage.h"
 
#include <cstdint>
#include <string>
#include <vector>
 
namespace kcenon::pacs::integration_test {
 
// =============================================================================
// Forward Declarations and Types
// =============================================================================
 
enum class invalid_dataset_type {
    missing_sop_class_uid,      
    missing_sop_instance_uid,   
    missing_patient_id,         
    missing_study_instance_uid, 
    invalid_vr,                 
    corrupted_pixel_data,       
    oversized_value             
};
 
struct multi_modal_study {
    std::string patient_id;                      
    std::string patient_name;                    
    std::string study_uid;                       
    std::vector<core::dicom_dataset> datasets;   
 
    [[nodiscard]] std::vector<core::dicom_dataset>
    get_by_modality(const std::string& modality) const;
 
    [[nodiscard]] size_t series_count() const;
};
 
// =============================================================================
// Test Data Generator Class
// =============================================================================
 
class test_data_generator {
public:
    // =========================================================================
    // Single Modality Generators
    // =========================================================================
 
    [[nodiscard]] static core::dicom_dataset ct(const std::string& study_uid = "");
 
    [[nodiscard]] static core::dicom_dataset mr(const std::string& study_uid = "");
 
    [[nodiscard]] static core::dicom_dataset xa(const std::string& study_uid = "");
 
    [[nodiscard]] static core::dicom_dataset us(const std::string& study_uid = "");
 
    // =========================================================================
    // Multi-frame Generators
    // =========================================================================
 
    [[nodiscard]] static core::dicom_dataset xa_cine(
        uint32_t frames = 30,
        const std::string& study_uid = "");
 
    [[nodiscard]] static core::dicom_dataset us_cine(
        uint32_t frames = 60,
        const std::string& study_uid = "");
 
    [[nodiscard]] static core::dicom_dataset enhanced_ct(
        uint32_t frames = 100,
        const std::string& study_uid = "");
 
    [[nodiscard]] static core::dicom_dataset enhanced_mr(
        uint32_t frames = 50,
        const std::string& study_uid = "");
 
    // =========================================================================
    // Clinical Workflow Generators
    // =========================================================================
 
    [[nodiscard]] static multi_modal_study patient_journey(
        const std::string& patient_id,
        const std::vector<std::string>& modalities = {"CT", "MR", "XA"});
 
    [[nodiscard]] static core::dicom_dataset worklist(
        const std::string& patient_id = "",
        const std::string& modality = "CT");
 
    // =========================================================================
    // Edge Case Generators
    // =========================================================================
 
    [[nodiscard]] static core::dicom_dataset large(size_t target_size_mb);
 
    [[nodiscard]] static core::dicom_dataset unicode();
 
    [[nodiscard]] static core::dicom_dataset with_private_tags(
        const std::string& creator_id = "TEST PRIVATE CREATOR");
 
    [[nodiscard]] static core::dicom_dataset invalid(invalid_dataset_type type);
 
    // =========================================================================
    // Utility Functions
    // =========================================================================
 
    [[nodiscard]] static std::string generate_uid(
        const std::string& root = "1.2.826.0.1.3680043.9.9999");
 
    [[nodiscard]] static std::string current_date();
 
    [[nodiscard]] static std::string current_time();
 
private:
    static void add_patient_module(
        core::dicom_dataset& ds,
        const std::string& patient_name,
        const std::string& patient_id,
        const std::string& birth_date = "19800101",
        const std::string& sex = "M");
 
    static void add_study_module(
        core::dicom_dataset& ds,
        const std::string& study_uid,
        const std::string& accession_number,
        const std::string& study_id,
        const std::string& description);
 
    static void add_series_module(
        core::dicom_dataset& ds,
        const std::string& series_uid,
        const std::string& modality,
        const std::string& series_number,
        const std::string& description);
 
    static void add_image_pixel_module(
        core::dicom_dataset& ds,
        uint16_t rows,
        uint16_t columns,
        uint16_t bits_allocated,
        uint16_t bits_stored,
        uint16_t samples_per_pixel = 1,
        const std::string& photometric = "MONOCHROME2");
 
    static void add_pixel_data(
        core::dicom_dataset& ds,
        uint16_t rows,
        uint16_t columns,
        uint16_t bits_allocated,
        uint16_t samples_per_pixel = 1,
        uint16_t fill_value = 512);
 
    static void add_multiframe_pixel_data(
        core::dicom_dataset& ds,
        uint16_t rows,
        uint16_t columns,
        uint16_t bits_allocated,
        uint32_t num_frames,
        uint16_t samples_per_pixel = 1);
};
 
}  // namespace kcenon::pacs::integration_test
 
#endif  // PACS_INTEGRATION_TESTS_TEST_DATA_GENERATOR_HPP