aira_assessment_manager.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_AI_AIRA_ASSESSMENT_MANAGER_HPP
#define PACS_AI_AIRA_ASSESSMENT_MANAGER_HPP
 
#include "kcenon/pacs/ai/aira_assessment.h"
#include "kcenon/pacs/core/dicom_dataset.h"
 
#include <map>
#include <optional>
#include <string>
#include <vector>
 
namespace kcenon::pacs::ai {
 
// =============================================================================
// Assessment Manager
// =============================================================================
 
class assessment_manager {
public:
    assessment_manager() = default;
 
    [[nodiscard]] bool store_assessment(
        const core::dicom_dataset& assessment_sr);
 
    [[nodiscard]] std::optional<core::dicom_dataset> retrieve_assessment(
        const std::string& assessment_uid) const;
 
    [[nodiscard]] std::vector<assessment_info> find_by_ai_result(
        const std::string& ai_result_uid) const;
 
    [[nodiscard]] std::vector<assessment_info> find_by_assessor(
        const std::string& assessor_name) const;
 
    [[nodiscard]] std::vector<assessment_info> find_by_type(
        assessment_type type) const;
 
    [[nodiscard]] std::optional<assessment_info> get_info(
        const std::string& assessment_uid) const;
 
    [[nodiscard]] bool exists(const std::string& assessment_uid) const;
 
    bool remove(const std::string& assessment_uid);
 
    [[nodiscard]] size_t count() const noexcept;
 
    [[nodiscard]] std::map<assessment_type, size_t> get_statistics() const;
 
private:
    assessment_info parse_assessment_info(
        const core::dicom_dataset& sr) const;
 
    std::map<std::string, core::dicom_dataset> datasets_;
    std::map<std::string, assessment_info> metadata_;
};
 
}  // namespace kcenon::pacs::ai
 
#endif  // PACS_AI_AIRA_ASSESSMENT_MANAGER_HPP