#include "ResourceBuffer.hpp"

#include "FilamentViewer.hpp"
#include "filament/LightManager.h"
#include "Log.hpp"
#include "ThreadPool.hpp"

#include <thread>
#include <functional>

using namespace polyvox;

extern "C"
{

#include "FlutterFilamentApi.h"

    FLUTTER_PLUGIN_EXPORT const void *create_filament_viewer(const void *context, const ResourceLoaderWrapper *const loader, void *const platform, const char *uberArchivePath)
    {
        return (const void *)new FilamentViewer(context, loader, platform, uberArchivePath);
    }

    FLUTTER_PLUGIN_EXPORT ResourceLoaderWrapper *make_resource_loader(LoadFilamentResourceFromOwner loadFn, FreeFilamentResourceFromOwner freeFn, void *const owner)
    {
        return new ResourceLoaderWrapper(loadFn, freeFn, owner);
    }

    FLUTTER_PLUGIN_EXPORT void create_render_target(const void *const viewer, intptr_t texture, uint32_t width, uint32_t height)
    {
        ((FilamentViewer *)viewer)->createRenderTarget(texture, width, height);
    }

    FLUTTER_PLUGIN_EXPORT void destroy_filament_viewer(const void *const viewer)
    {
        delete ((FilamentViewer *)viewer);
    }

    FLUTTER_PLUGIN_EXPORT void set_background_color(const void *const viewer, const float r, const float g, const float b, const float a)
    {
        ((FilamentViewer *)viewer)->setBackgroundColor(r, g, b, a);
    }

    FLUTTER_PLUGIN_EXPORT void clear_background_image(const void *const viewer)
    {
        ((FilamentViewer *)viewer)->clearBackgroundImage();
    }

    FLUTTER_PLUGIN_EXPORT void set_background_image(const void *const viewer, const char *path, bool fillHeight)
    {
        ((FilamentViewer *)viewer)->setBackgroundImage(path, fillHeight);
    }

    FLUTTER_PLUGIN_EXPORT void set_background_image_position(const void *const viewer, float x, float y, bool clamp)
    {
        ((FilamentViewer *)viewer)->setBackgroundImagePosition(x, y, clamp);
    }

    FLUTTER_PLUGIN_EXPORT void set_tone_mapping(const void *const viewer, int toneMapping)
    {
        ((FilamentViewer *)viewer)->setToneMapping((ToneMapping)toneMapping);
    }

    FLUTTER_PLUGIN_EXPORT void set_bloom(const void *const viewer, float strength)
    {
        Log("Setting bloom to %f", strength);
        ((FilamentViewer *)viewer)->setBloom(strength);
    }

    FLUTTER_PLUGIN_EXPORT void load_skybox(const void *const viewer, const char *skyboxPath)
    {
        ((FilamentViewer *)viewer)->loadSkybox(skyboxPath);
    }

    FLUTTER_PLUGIN_EXPORT void load_ibl(const void *const viewer, const char *iblPath, float intensity)
    {
        ((FilamentViewer *)viewer)->loadIbl(iblPath, intensity);
    }

    FLUTTER_PLUGIN_EXPORT void remove_skybox(const void *const viewer)
    {
        ((FilamentViewer *)viewer)->removeSkybox();
    }

    FLUTTER_PLUGIN_EXPORT void remove_ibl(const void *const viewer)
    {
        ((FilamentViewer *)viewer)->removeIbl();
    }

    EntityId add_light(const void *const viewer, uint8_t type, float colour, float intensity, float posX, float posY, float posZ, float dirX, float dirY, float dirZ, bool shadows)
    {
        return ((FilamentViewer *)viewer)->addLight((LightManager::Type)type, colour, intensity, posX, posY, posZ, dirX, dirY, dirZ, shadows);
    }

    FLUTTER_PLUGIN_EXPORT void remove_light(const void *const viewer, int32_t entityId)
    {
        ((FilamentViewer *)viewer)->removeLight(entityId);
    }

    FLUTTER_PLUGIN_EXPORT void clear_lights(const void *const viewer)
    {
        ((FilamentViewer *)viewer)->clearLights();
    }

    FLUTTER_PLUGIN_EXPORT EntityId load_glb(void *assetManager, const char *assetPath, bool unlit)
    {
        return ((AssetManager *)assetManager)->loadGlb(assetPath, unlit);
    }

    FLUTTER_PLUGIN_EXPORT EntityId load_gltf(void *assetManager, const char *assetPath, const char *relativePath)
    {
        return ((AssetManager *)assetManager)->loadGltf(assetPath, relativePath);
    }

    FLUTTER_PLUGIN_EXPORT bool set_camera(const void *const viewer, EntityId asset, const char *nodeName)
    {
        return ((FilamentViewer *)viewer)->setCamera(asset, nodeName);
    }

    const double *const get_camera_model_matrix(const void *const viewer)
    {
        const auto &modelMatrix = ((FilamentViewer *)viewer)->getCameraModelMatrix();
        double *array = (double *)calloc(16, sizeof(double));
        memcpy(array, modelMatrix.asArray(), 16 * sizeof(double));
        return array;
    }

    const double *const get_camera_view_matrix(const void *const viewer)
    {
        const auto &modelMatrix = ((FilamentViewer *)viewer)->getCameraViewMatrix();
        return modelMatrix.asArray();
    }

    FLUTTER_PLUGIN_EXPORT void set_camera_manipulator_options(const void *const viewer, _ManipulatorMode mode, double orbitSpeedX, double orbitSpeedY, double zoomSpeed)
    {
        ((FilamentViewer *)viewer)->setCameraManipulatorOptions((filament::camutils::Mode)mode, orbitSpeedX, orbitSpeedY, zoomSpeed);
    }

    FLUTTER_PLUGIN_EXPORT void set_view_frustum_culling(const void *const viewer, bool enabled)
    {
        ((FilamentViewer *)viewer)->setViewFrustumCulling(enabled);
    }

    FLUTTER_PLUGIN_EXPORT void move_camera_to_asset(const void *const viewer, EntityId asset)
    {
        ((FilamentViewer *)viewer)->moveCameraToAsset(asset);
    }

    FLUTTER_PLUGIN_EXPORT void set_camera_focus_distance(const void *const viewer, float distance)
    {
        ((FilamentViewer *)viewer)->setCameraFocusDistance(distance);
    }

    FLUTTER_PLUGIN_EXPORT void set_camera_exposure(const void *const viewer, float aperture, float shutterSpeed, float sensitivity)
    {
        ((FilamentViewer *)viewer)->setCameraExposure(aperture, shutterSpeed, sensitivity);
    }

    FLUTTER_PLUGIN_EXPORT void set_camera_position(const void *const viewer, float x, float y, float z)
    {
        ((FilamentViewer *)viewer)->setCameraPosition(x, y, z);
    }

    FLUTTER_PLUGIN_EXPORT void set_camera_rotation(const void *const viewer, float rads, float x, float y, float z)
    {
        ((FilamentViewer *)viewer)->setCameraRotation(rads, x, y, z);
    }

    FLUTTER_PLUGIN_EXPORT void set_camera_model_matrix(const void *const viewer, const float *const matrix)
    {
        ((FilamentViewer *)viewer)->setCameraModelMatrix(matrix);
    }

    FLUTTER_PLUGIN_EXPORT void set_camera_focal_length(const void *const viewer, float focalLength)
    {
        ((FilamentViewer *)viewer)->setCameraFocalLength(focalLength);
    }

    FLUTTER_PLUGIN_EXPORT void render(
        const void *const viewer,
        uint64_t frameTimeInNanos,
        void *pixelBuffer,
        void (*callback)(void *buf, size_t size, void *data),
        void *data)
    {
        ((FilamentViewer *)viewer)->render(frameTimeInNanos, pixelBuffer, callback, data);
    }

    FLUTTER_PLUGIN_EXPORT void set_frame_interval(
        const void *const viewer,
        float frameInterval)
    {
        ((FilamentViewer *)viewer)->setFrameInterval(frameInterval);
    }

    FLUTTER_PLUGIN_EXPORT void destroy_swap_chain(const void *const viewer)
    {
        ((FilamentViewer *)viewer)->destroySwapChain();
    }

    FLUTTER_PLUGIN_EXPORT void create_swap_chain(const void *const viewer, const void *const window, uint32_t width, uint32_t height)
    {
        ((FilamentViewer *)viewer)->createSwapChain(window, width, height);
    }

    FLUTTER_PLUGIN_EXPORT void update_viewport_and_camera_projection(const void *const viewer, uint32_t width, uint32_t height, float scaleFactor)
    {
        return ((FilamentViewer *)viewer)->updateViewportAndCameraProjection(width, height, scaleFactor);
    }

    FLUTTER_PLUGIN_EXPORT void scroll_update(const void *const viewer, float x, float y, float delta)
    {
        ((FilamentViewer *)viewer)->scrollUpdate(x, y, delta);
    }

    FLUTTER_PLUGIN_EXPORT void scroll_begin(const void *const viewer)
    {
        ((FilamentViewer *)viewer)->scrollBegin();
    }

    FLUTTER_PLUGIN_EXPORT void scroll_end(const void *const viewer)
    {
        ((FilamentViewer *)viewer)->scrollEnd();
    }

    FLUTTER_PLUGIN_EXPORT void grab_begin(const void *const viewer, float x, float y, bool pan)
    {
        ((FilamentViewer *)viewer)->grabBegin(x, y, pan);
    }

    FLUTTER_PLUGIN_EXPORT void grab_update(const void *const viewer, float x, float y)
    {
        ((FilamentViewer *)viewer)->grabUpdate(x, y);
    }

    FLUTTER_PLUGIN_EXPORT void grab_end(const void *const viewer)
    {
        ((FilamentViewer *)viewer)->grabEnd();
    }

    FLUTTER_PLUGIN_EXPORT void *get_asset_manager(const void *const viewer)
    {
        return (void *)((FilamentViewer *)viewer)->getAssetManager();
    }

    FLUTTER_PLUGIN_EXPORT void apply_weights(
        void *assetManager,
        EntityId asset,
        const char *const entityName,
        float *const weights,
        int count)
    {
        // ((AssetManager*)assetManager)->setMorphTargetWeights(asset, entityName, weights, count);
    }

    FLUTTER_PLUGIN_EXPORT void set_morph_target_weights(
        void *assetManager,
        EntityId asset,
        const char *const entityName,
        const float *const weights,
        const int numWeights)
    {

        return ((AssetManager *)assetManager)->setMorphTargetWeights(asset, entityName, weights, numWeights);
    }

    bool set_morph_animation(
        void *assetManager,
        EntityId asset,
        const char *const entityName,
        const float *const morphData,
        const int *const morphIndices,
        int numMorphTargets,
        int numFrames,
        float frameLengthInMs)
    {

        return ((AssetManager *)assetManager)->setMorphAnimationBuffer(asset, entityName, morphData, morphIndices, numMorphTargets, numFrames, frameLengthInMs);
    }

    FLUTTER_PLUGIN_EXPORT void set_bone_animation(
        void *assetManager,
        EntityId asset,
        const float *const frameData,
        int numFrames,
        int numBones,
        const char **const boneNames,
        const char **const meshNames,
        int numMeshTargets,
        float frameLengthInMs)
    {
        ((AssetManager *)assetManager)->setBoneAnimationBuffer(asset, frameData, numFrames, numBones, boneNames, meshNames, numMeshTargets, frameLengthInMs);
    }

    FLUTTER_PLUGIN_EXPORT void set_post_processing(void *const viewer, bool enabled)
    {
        ((FilamentViewer *)viewer)->setPostProcessing(enabled);
    }

    //   void set_bone_transform(
    //     EntityId asset,
    //     const char* boneName,
    //     const char* entityName,
    //     float transX,
    //     float transY,
    //     float transZ,
    //     float quatX,
    //     float quatY,
    //     float quatZ,
    //     float quatW
    // ) {
    //     ((AssetManager*)assetManager)->setBoneTransform(
    //         boneName,
    //         entityName,
    //         transX,
    //         transY,
    //         transZ,
    //         quatX,
    //         quatY,
    //         quatZ,
    //         quatW,
    //         false
    //     );

    //   }

    FLUTTER_PLUGIN_EXPORT void play_animation(
        void *assetManager,
        EntityId asset,
        int index,
        bool loop,
        bool reverse,
        bool replaceActive,
        float crossfade)
    {
        ((AssetManager *)assetManager)->playAnimation(asset, index, loop, reverse, replaceActive, crossfade);
    }

    FLUTTER_PLUGIN_EXPORT void set_animation_frame(
        void *assetManager,
        EntityId asset,
        int animationIndex,
        int animationFrame)
    {
        // ((AssetManager*)assetManager)->setAnimationFrame(asset, animationIndex, animationFrame);
    }

    float get_animation_duration(void *assetManager, EntityId asset, int animationIndex)
    {
        return ((AssetManager *)assetManager)->getAnimationDuration(asset, animationIndex);
    }

    int get_animation_count(
        void *assetManager,
        EntityId asset)
    {
        auto names = ((AssetManager *)assetManager)->getAnimationNames(asset);
        return (int)names->size();
    }

    FLUTTER_PLUGIN_EXPORT void get_animation_name(
        void *assetManager,
        EntityId asset,
        char *const outPtr,
        int index)
    {
        auto names = ((AssetManager *)assetManager)->getAnimationNames(asset);
        string name = names->at(index);
        strcpy(outPtr, name.c_str());
    }

    FLUTTER_PLUGIN_EXPORT int get_morph_target_name_count(void *assetManager, EntityId asset, const char *meshName)
    {
        unique_ptr<vector<string>> names = ((AssetManager *)assetManager)->getMorphTargetNames(asset, meshName);
        return (int)names->size();
    }

    FLUTTER_PLUGIN_EXPORT void get_morph_target_name(void *assetManager, EntityId asset, const char *meshName, char *const outPtr, int index)
    {
        unique_ptr<vector<string>> names = ((AssetManager *)assetManager)->getMorphTargetNames(asset, meshName);
        string name = names->at(index);
        strcpy(outPtr, name.c_str());
    }

    FLUTTER_PLUGIN_EXPORT void remove_asset(const void *const viewer, EntityId asset)
    {
        ((FilamentViewer *)viewer)->removeAsset(asset);
    }

    FLUTTER_PLUGIN_EXPORT void clear_assets(const void *const viewer)
    {
        ((FilamentViewer *)viewer)->clearAssets();
    }

    bool set_material_color(void *assetManager, EntityId asset, const char *meshName, int materialIndex, const float r, const float g, const float b, const float a)
    {
        return ((AssetManager *)assetManager)->setMaterialColor(asset, meshName, materialIndex, r, g, b, a);
    }

    FLUTTER_PLUGIN_EXPORT void transform_to_unit_cube(void *assetManager, EntityId asset)
    {
        ((AssetManager *)assetManager)->transformToUnitCube(asset);
    }

    FLUTTER_PLUGIN_EXPORT void set_position(void *assetManager, EntityId asset, float x, float y, float z)
    {
        ((AssetManager *)assetManager)->setPosition(asset, x, y, z);
    }

    FLUTTER_PLUGIN_EXPORT void set_rotation(void *assetManager, EntityId asset, float rads, float x, float y, float z)
    {
        ((AssetManager *)assetManager)->setRotation(asset, rads, x, y, z);
    }

    FLUTTER_PLUGIN_EXPORT void set_scale(void *assetManager, EntityId asset, float scale)
    {
        ((AssetManager *)assetManager)->setScale(asset, scale);
    }

    FLUTTER_PLUGIN_EXPORT void stop_animation(void *assetManager, EntityId asset, int index)
    {
        ((AssetManager *)assetManager)->stopAnimation(asset, index);
    }

    FLUTTER_PLUGIN_EXPORT int hide_mesh(void *assetManager, EntityId asset, const char *meshName)
    {
        return ((AssetManager *)assetManager)->hide(asset, meshName);
    }

    FLUTTER_PLUGIN_EXPORT int reveal_mesh(void *assetManager, EntityId asset, const char *meshName)
    {
        return ((AssetManager *)assetManager)->reveal(asset, meshName);
    }

    FLUTTER_PLUGIN_EXPORT void pick(void *const viewer, int x, int y, EntityId *entityId)
    {
        ((FilamentViewer *)viewer)->pick(static_cast<uint32_t>(x), static_cast<uint32_t>(y), static_cast<int32_t *>(entityId));
    }

    FLUTTER_PLUGIN_EXPORT const char *get_name_for_entity(void *const assetManager, const EntityId entityId)
    {
        return ((AssetManager *)assetManager)->getNameForEntity(entityId);
    }

    FLUTTER_PLUGIN_EXPORT void ios_dummy()
    {
        Log("Dummy called");
    }
}