#pragma once

#include "GL_integrator.hpp"
#include "TS_integrator.hpp"

// Gauss-Legendre quadrature for 1D integral over interval [a, b]
template<typename Func>
double gauss_integrate_1D(double a, double b, Func&& func, int q) {
    double sum = 0.0;
    double length = b - a;
    for (int i = 0; i < q; ++i) {
        double x = a + length * GLIntegrator<double>::x(q, i);
        double w = length * GLIntegrator<double>::w(q, i);
        sum += w * func(x);
    }
    return sum;
}

// Tanh-sinh quadrature for 1D integral over [a, b]
template<typename Func>
double tanh_sinh_integrate_1D(double a, double b, Func&& func, int q) {
    double result = 0.0;
    double length = b - a;
    for (int i = 0; i < q; ++i) {
        double x = a + length * TSIntegrator<double>::x(q, i, a, b);
        double w = length * TSIntegrator<double>::w(q, i, a, b);
        result += w * func(x);
    }
    return result;
}

template<typename Func>
double integrate_1D(double a, double b, Func&& func, int q, bool use_tanh_sinh = false) {
    if (use_tanh_sinh) {
        return tanh_sinh_integrate_1D(a, b, std::forward<Func>(func), q);
    } else {
        return gauss_integrate_1D(a, b, std::forward<Func>(func), q);       
    }
}