#include <array>
#include <bitset>
#include <iostream>
#include <booluarray.hpp>


#include <cstddef>
#include <iostream>
#include <iomanip>
#include <fstream>
#include <vector>
#include "bernstein.hpp"
#include "multiloop.hpp"
#include "quadrature_multipoly.hpp"
#include "binomial.hpp"

#include "real.hpp"
#include "sparkstack.hpp"
#include "uvector.hpp"
#include "vector"
#include "xarray.hpp"
#include <chrono>
#include <cmath>
#include <memory>

#include "organizer/primitive.hpp"

namespace algoim::Organizer
{

bool keepQuadraturePoint(std::vector<tensor3>& originTensors, const uvector3& originPt)
{
    // TODO: using blobtree to filter tensors
    for (auto& t : originTensors) {
        if (evalPower(t, originPt) >= 0) { return false; }
    }
    return true;
}

class BasicTask
{
public:
    // std::vector<std::shared_ptr<PrimitiveDesc>> primitives;

    // BasicTask(std::vector<std::shared_ptr<PrimitiveDesc>> ps) {};


    BasicTask(std::shared_ptr<PrimitiveDesc> p)
    {
        int      q         = 20;
        real     volume    = 0;
        real     xmin      = -1;
        real     xmax      = 1;
        auto     integrand = [](const uvector<real, 3>& x) { return 1.0; };
        uvector3 range     = xmax - xmin;


        if (auto pt = std::dynamic_pointer_cast<SphereDesc>(p)) {
            tensor3 tensor(nullptr, 3);
            algoim_spark_alloc(real, tensor);
            makeSphere(*pt, tensor);
            detail::powerTransformation(range, xmin, tensor);

            tensor3 phi(nullptr, 3);
            algoim_spark_alloc(real, phi);
            detail::power2BernsteinTensor(tensor, phi);

            uvector<real, 3> testX(0., 0., 0.25);
            real             testEvalBernstein = bernstein::evalBernsteinPoly(phi, testX);
            // auto             vec1              = xarray2StdVector(phi);
            std::cout << "eval bernstein without interpolation:" << testEvalBernstein << std::endl;

            ImplicitPolyQuadrature<3> ipquad(phi);
            ipquad.integrate(AutoMixed, q, [&](const uvector<real, 3>& x, real w) {
                if (isInsideBernstein(phi, x)) volume += w * integrand(xmin + x * (xmax - xmin));
            });
        } else if (auto pt = std::dynamic_pointer_cast<MeshDesc>(p)) {
            const int faceCount = pt->indexInclusiveScan.size();
            assert(faceCount > 1);
            std::vector<tensor3> planeTensors(faceCount, tensor3(nullptr, 2));
            algoim_spark_alloc(real, planeTensors);

            tensor3 compositeTensor(nullptr, 1 + faceCount);
            algoim_spark_alloc(real, compositeTensor);

            makeMesh(*pt, compositeTensor, planeTensors);
            detail::powerTransformation(range, xmin, compositeTensor);

            auto compositeTensorStdVector = xarray2StdVector(compositeTensor);

            uvector<real, 3> testX(0.8, 0.8, 0.8);
            real             textEvalPower = evalPower(compositeTensor, testX);

            tensor3 phi(nullptr, 1 + faceCount);
            algoim_spark_alloc(real, phi);
            detail::power2BernsteinTensor(compositeTensor, phi);

            int                       quadraturePointCount = 0;
            ImplicitPolyQuadrature<3> ipquad(phi);
            ipquad.integrate(AutoMixed, q, [&](const uvector<real, 3>& x, real w) {
                quadraturePointCount++;
                auto realX = x * range + xmin;
                if (isInsidePowers(planeTensors, realX)) volume += w * integrand(realX);
            });
            std::cout << "quadraturePointCount: " << quadraturePointCount << std::endl;
        }
        volume *= pow(xmax - xmin, 3);
        std::cout << "Volume xxx: " << volume << std::endl;
    };

    BasicTask(std::vector<std::shared_ptr<PrimitiveDesc>> primitives)
    {
        std::vector<SparkStack<real>*> phiStacks;
        std::vector<tensor3>           phis;

        std::vector<SparkStack<real>*> originTensorStacks;
        std::vector<tensor3>           originTensors;
        int                            q = 10;
        real                           volume;
        uvector3                       xmin      = 0;
        uvector3                       xmax      = 1;
        auto                           integrand = [](const uvector<real, 3>& x) { return 1.0; };
        uvector3                       range     = xmax - xmin;

        for (int i = 0; i < primitives.size(); i++) {
            if (auto pt = std::dynamic_pointer_cast<SphereDesc>(primitives[i])) {
                tensor3 originTensor(nullptr, 3), transformedTensor(nullptr, 3);
                originTensorStacks.emplace_back(algoim_spark_alloc_heap(real, originTensor)); // 记录，用以最后bool
                algoim_spark_alloc(real, transformedTensor);
                makeSphere(*pt, originTensor);
                originTensors.emplace_back(originTensor);
                detail::powerTransformation(range, xmin, originTensor, transformedTensor);

                tensor3 phi(nullptr, 3);
                phiStacks.emplace_back(algoim_spark_alloc_heap(real, phi));
                detail::power2BernsteinTensor(transformedTensor, phi);
                phis.emplace_back(phi);
            } else if (auto pt = std::dynamic_pointer_cast<MeshDesc>(primitives[i])) {
                const int faceCount = pt->indexInclusiveScan.size();
                assert(faceCount > 1);
                std::vector<tensor3> planeTensors(faceCount, tensor3(nullptr, 2));
                algoimSparkAllocHeapVector(originTensorStacks, planeTensors);

                tensor3 compositeTensor(nullptr, 1 + faceCount);
                algoim_spark_alloc(real, compositeTensor);

                makeMesh(*pt, compositeTensor, planeTensors);
                detail::powerTransformation(range, xmin, compositeTensor);
                originTensors.insert(originTensors.end(), planeTensors.begin(), planeTensors.end());

                tensor3 phi(nullptr, 1 + faceCount);
                phiStacks.emplace_back(algoim_spark_alloc_heap(real, phi));
                detail::power2BernsteinTensor(compositeTensor, phi);
                phis.emplace_back(phi);
            }
        }
        ImplicitPolyQuadrature<3> ipquad(phis);
        ipquad.integrate(AutoMixed, q, [&](const uvector<real, 3>& x, real w) {
            auto realX = x * range + xmin;
            if (keepQuadraturePoint(originTensors, realX)) volume += w * integrand(realX);
        });
        std::cout << "Volume xxx: " << volume << std::endl;

        // free memory
        for (auto& p : phiStacks) delete p;
        for (auto& p : originTensorStacks) delete p;
    };
};
}; // namespace algoim::Organizer