/*****************************************************************************************
 *                                                                                       *
 * OpenSpace                                                                             *
 *                                                                                       *
 * Copyright (c) 2014-2025                                                               *
 *                                                                                       *
 * Permission is hereby granted, free of charge, to any person obtaining a copy of this  *
 * software and associated documentation files (the "Software"), to deal in the Software *
 * without restriction, including without limitation the rights to use, copy, modify,    *
 * merge, publish, distribute, sublicense, and/or sell copies of the Software, and to    *
 * permit persons to whom the Software is furnished to do so, subject to the following   *
 * conditions:                                                                           *
 *                                                                                       *
 * The above copyright notice and this permission notice shall be included in all copies *
 * or substantial portions of the Software.                                              *
 *                                                                                       *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED,   *
 * INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A         *
 * PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT    *
 * HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF  *
 * CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE  *
 * OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.                                         *
 ****************************************************************************************/

#include <modules/space/translation/gptranslation.h>

#include <modules/space/kepler.h>
#include <openspace/documentation/verifier.h>
#include <filesystem>
#include <optional>

namespace {
    struct [[codegen::Dictionary(GPTranslation)]] Parameters {
        // Specifies the filename of the general pertubation file
        std::filesystem::path file;

        enum class [[codegen::map(openspace::kepler::Format)]] Format {
            // A NORAD-style Two-Line element
            TLE,
            // Orbit Mean-Elements Message in the KVN notation
            OMM,
            // JPL's Small Bodies Database
            SBDB
        };
        // The file format that is contained in the file
        Format format;

        // Specifies the element within the file that should be used in case the file
        // provides multiple general pertubation elements. Defaults to 1.
        std::optional<int> element [[codegen::greater(0)]];
    };
#include "gptranslation_codegen.cpp"

    ghoul::Dictionary gpDictionaryToKepler(const ghoul::Dictionary& dictionary) {
        using namespace openspace;

        const Parameters p = codegen::bake<Parameters>(dictionary);
        if (!std::filesystem::is_regular_file(p.file)) {
            throw ghoul::RuntimeError("The provided TLE file must exist");
        }

        int element = p.element.value_or(1);

        std::vector<kepler::Parameters> parameters = kepler::readFile(
            p.file,
            codegen::map<kepler::Format>(p.format)
        );

        if (element > static_cast<int>(parameters.size())) {
            throw ghoul::RuntimeError(std::format(
                "Requested element {} but only {} are available",
                element, parameters.size()
            ));
        }

        const kepler::Parameters& param = parameters[element - 1];

        // We copy the old dictionary to make sure we keep values intact that we don't
        // want to touch here (for example the 'Type')
        ghoul::Dictionary res = dictionary;
        res.setValue("Eccentricity", param.eccentricity);
        res.setValue("SemiMajorAxis", param.semiMajorAxis);
        res.setValue("Inclination", param.inclination);
        res.setValue("AscendingNode", param.ascendingNode);
        res.setValue("ArgumentOfPeriapsis", param.argumentOfPeriapsis);
        res.setValue("MeanAnomaly", param.meanAnomaly);
        res.setValue("Period", param.period);
        res.setValue("Epoch", param.epoch);
        return res;
    }
} // namespace

namespace openspace {

documentation::Documentation GPTranslation::Documentation() {
    return codegen::doc<Parameters>("space_translation_gp");
}

GPTranslation::GPTranslation(const ghoul::Dictionary& dictionary)
    : KeplerTranslation(gpDictionaryToKepler(dictionary))
{}

} // namespace openspace