#include <iomanip>
#include <iostream>
#include <vector>
#include "MatlabCppSharedLib.hpp"
#include "cppstruct.hpp"     //include user-defined C++ structs

// Inform MATLAB about the custom C++ types to be used
USING_TYPE_WITH_MATLAB(cppstruct::OutputStruct, "data");
USING_TYPE_WITH_MATLAB(cppstruct::InputStruct, "data");
USING_TYPE_WITH_MATLAB(cppstruct::FieldStats, "data");


// Start MATLAB Runtime, initialize it, and return an object to it
std::shared_ptr<matlab::cpplib::MATLABApplication> setup() {
    auto mode = matlab::cpplib::MATLABApplicationMode::IN_PROCESS; // must be started in IN_PROCESS mode to pass C++ structs
    std::vector<std::u16string> options = { u"-nojvm" };
    std::shared_ptr<matlab::cpplib::MATLABApplication> matlabApplication =
        matlab::cpplib::initMATLABApplication(mode, options);
    return matlabApplication;
}

// Function to print the MATLAB computation results
void printResults(const cppstruct::OutputStruct& output) {
    std::cout << "Temperatures - Mean: " << std::fixed << std::setprecision(4)
        << output.temperatures.mean
        << ", Std: " << output.temperatures.std
        << ", Max: " << output.temperatures.max << std::endl;
    std::cout << "Pressures - Mean: " << std::fixed << std::setprecision(4)
        << output.pressures.mean
        << ", Std: " << output.pressures.std
        << ", Max: " << output.pressures.max << std::endl;
}

// Initialize the code archive (.ctf file), specify input arguments, call the MATLAB function, 
// and print the result
int mainFunc(std::shared_ptr<matlab::cpplib::MATLABApplication> app, int argc, const char* argv[]) {
    try {
        std::cout << "Executing MATLAB function ..." << std::endl;
        auto libPtr = matlab::cpplib::initMATLABLibrary(app, u"analyzeData.ctf");
        if (!libPtr) {
            std::cerr << "Failed to initialize MATLAB library." << std::endl;
            return -1;
        }

        // Initialize InputStruct with temperatures and pressures
        cppstruct::InputStruct input;
        input.temperatures = { 72, 75, 69, 68, 70 };
        input.pressures = { 30, 29.5, 30.2, 29.9, 30.1 };

        // Set the pointer to input
        cppstruct::InputStruct* inputPtr = &input;
        

        if (!inputPtr) {
            std::cerr << "Invalid input pointer." << std::endl;
            return -1;
        }

        // Call the MATLAB function with the input struct
        cppstruct::OutputStruct* output =
            libPtr->feval<cppstruct::OutputStruct*>("analyzeData", inputPtr);
        if (!output) {
            std::cerr << "Failed to obtain valid output from MATLAB function." << std::endl;
            return -1;
        }

        // Print the results
        printResults(*output);
    }
    catch (const std::exception& e) {
        std::cerr << "An error occurred: " << e.what() << std::endl;
        return -1;
    }
    return 0;
}


int main(int argc, const char* argv[]) {
    int ret = 0;
    try {
        auto matlabApplication = setup();
        ret = matlab::cpplib::runMain(mainFunc, std::move(matlabApplication), argc, argv);
        matlabApplication.reset();
        return ret;
    }
    catch (const std::exception& e) {
        std::cerr << "An error occurred during setup: " << e.what() << std::endl;
        return -1;
    }
}