#include "MedicalLib/Patient.h"
#include "MedicalLib/Heart.h"
#include "MedicalLib/Lungs.h"
#include "MedicalLib/Brain.h"
#include "MedicalLib/Liver.h"
#include "MedicalLib/Kidneys.h"
#include "MedicalLib/Bladder.h"
#include "MedicalLib/Stomach.h"
#include "MedicalLib/Intestines.h"
#include "MedicalLib/Gallbladder.h"
#include "MedicalLib/Pancreas.h"
#include "MedicalLib/Esophagus.h"
#include "MedicalLib/Spleen.h"
#include "MedicalLib/SpinalCord.h"
#include <memory>

/**
 * @brief Initializes a new patient with a specific number of heart leads.
 * @param patientId The ID for the new patient.
 * @param numHeartLeads The number of EKG leads for the heart.
 * @return A Patient struct with default healthy values.
 */
Patient initializePatient(int patientId, int numHeartLeads) {
    Patient patient;
    patient.patientId = patientId;

    // Initialize default organs
    patient.organs.push_back(std::make_unique<Heart>(1, numHeartLeads));
    patient.organs.push_back(std::make_unique<Lungs>(2));
    patient.organs.push_back(std::make_unique<Brain>(3));
    patient.organs.push_back(std::make_unique<Liver>(4));
    patient.organs.push_back(std::make_unique<Kidneys>(5));
    patient.organs.push_back(std::make_unique<Bladder>(6));
    patient.organs.push_back(std::make_unique<Stomach>(7));
    patient.organs.push_back(std::make_unique<Intestines>(8));
    patient.organs.push_back(std::make_unique<Gallbladder>(9));
    patient.organs.push_back(std::make_unique<Pancreas>(10));
    patient.organs.push_back(std::make_unique<Esophagus>(11));
    patient.organs.push_back(std::make_unique<Spleen>(12));
    patient.organs.push_back(std::make_unique<SpinalCord>(13));

    return patient;
}

/**
 * @brief Initializes a new patient with baseline vital signs and a 12-lead heart.
 * @param patientId The ID for the new patient.
 * @return A Patient struct with default healthy values.
 */
Patient initializePatient(int patientId) {
    return initializePatient(patientId, 12);
}


/**
 * @brief Updates the patient's state by updating the state of all their organs.
 * @param patient The patient to update.
 * @param deltaTime_s The time elapsed in seconds.
 */
void updatePatient(Patient& patient, double deltaTime_s) {
    // Update all organs
    for (auto& organ : patient.organs) {
        organ->update(deltaTime_s);
    }
}

/**
 * @brief Gets a summary of a specific organ's vitals.
 * @param patient The patient to get the organ summary from.
 * @param organType The type of the organ to get the summary for.
 * @return A string containing the organ's vital signs, or an empty string if not found.
 */
std::string getOrganSummary(const Patient& patient, const std::string& organType) {
    for (const auto& organ : patient.organs) {
        if (organ->getType() == organType) {
            return organ->getSummary();
        }
    }
    return "";
}

/**
 * @brief Gets a consolidated summary of all the patient's organ vitals.
 * @param patient The patient to get the summary from.
 * @return A string containing the vital signs of all the patient's organs.
 */
std::string getPatientSummary(const Patient& patient) {
    std::string summary;
    for (const auto& organ : patient.organs) {
        summary += organ->getSummary() + "\n";
    }
    return summary;
}