// Copyright (c) 2017-2024, Mudita Sp. z.o.o. All rights reserved.
// For licensing, see https://github.com/mudita/MuditaOS/blob/master/LICENSE.md

#include "CalculatorInputProcessor.hpp"
#include "CalculatorUtility.hpp"
#include <widgets/CalculatorStyle.hpp>
#include <i18n/i18n.hpp>
#include <Utils.hpp>
#include <tinyexpr.h>
#include <cmath>

namespace calc
{
    Result Calculator::calculate(std::string source)
    {
        using namespace calc::limits;

        source = prepareEquationForParser(source);

        if (source.empty()) {
            return Result{source, {}, false};
        }

        int error;
        const auto result = te_interp(source.c_str(), &error);
        if (error == 0 && !std::isinf(result) && !std::isnan(result)) {
            auto output = utils::to_string(result);
            if (output.length() > MaxStringLength) {
                output = getValueThatFitsOnScreen(result);
            }
            if (symbols::strings::decimal_separator_str() == symbols::strings::comma) {
                output = replaceAllOccurrences(output, symbols::strings::full_stop, symbols::strings::comma);
            }
            return Result{source, output, false};
        }
        return Result{source, symbols::strings::error_str(), true};
    }

    std::string Calculator::prepareEquationForParser(std::string input)
    {
        input = replaceAllOccurrences(input, symbols::strings::division, symbols::strings::solidus);
        input = replaceAllOccurrences(input, symbols::strings::multiplication, symbols::strings::asterisk);
        input = replaceAllOccurrences(input, symbols::strings::comma, symbols::strings::full_stop);
        return input;
    }

    std::string Calculator::replaceAllOccurrences(std::string input, const std::string &from, const std::string &to)
    {
        size_t index = 0;
        while (true) {
            index = input.find(from, index);
            if (index == std::string::npos) {
                break;
            }
            input.replace(index, from.length(), to);
            index += to.length();
        }
        return input;
    }

    std::string Calculator::getValueThatFitsOnScreen(double result)
    {
        using namespace calc::limits;

        const auto base = static_cast<long long>(result);
        auto length     = utils::to_string(base).length();
        if (base < 0) {
            length -= 1;
        }
        if (length > ExpLength + 1) {
            return convertToNumberWithPositiveExponent(result, length - 1);
        }
        else if (length == ExpLength + 1) {
            if (result < 0) {
                return utils::to_string(getCoefficient(result, VeryLowPrecision));
            }
            return utils::to_string(getCoefficient(result, LowPrecision));
        }
        else if (length == 1 && result < -1) {
            return utils::to_string(getCoefficient(result, LowPrecision));
        }
        else if (result > 1) {
            return utils::to_string(getCoefficient(result, Precision));
        }
        else {
            return convertToNumberWithNegativeExponent(result, base);
        }
    }

    std::string Calculator::convertToNumberWithPositiveExponent(double result, uint32_t exponent)
    {
        using namespace calc::limits;

        result /= pow(10, exponent);
        auto exponentLength = utils::to_string(exponent).length();
        auto decimalPlace   = Precision - exponentLength - ExpLength;
        if (result < 0) {
            decimalPlace -= 1;
        }
        return utils::to_string(getCoefficient(result, decimalPlace)) + "e" + utils::to_string(exponent);
    }

    std::string Calculator::convertToNumberWithNegativeExponent(double result, long long base)
    {
        using namespace calc::limits;

        double frac = (result - base) * pow(10, HighPrecision);
        if (result < 0) {
            frac *= -1;
        }
        auto fractionalPart = static_cast<unsigned long int>(round(frac));
        auto fracLength     = utils::to_string(fractionalPart).length();
        auto exponent       = HighPrecision - fracLength + 1;
        if (exponent > MinusExpLength + 1) {
            result *= pow(10, exponent);
            auto exponentLength = utils::to_string(exponent).length();
            auto decimalPlace   = Precision - exponentLength - MinusExpLength;
            if (result < 0) {
                decimalPlace -= 1;
            }
            return utils::to_string(getCoefficient(result, decimalPlace)) + "e-" + utils::to_string(exponent);
        }
        else if (result < 0) {
            return utils::to_string(getCoefficient(result, LowPrecision));
        }
        return utils::to_string(getCoefficient(result, Precision));
    }

    long double Calculator::getCoefficient(double result, uint32_t precision)
    {
        return std::roundl(result * pow(10, precision)) / pow(10, precision);
    }

} // namespace calc