#include <Atomic/Core/Thread.h>

#include <Atomic/Resource/ResourceCache.h>
#include <Atomic/Script/ScriptVariant.h>
#include <Atomic/Script/ScriptVariantMap.h>
#include <Atomic/IPC/IPC.h>

#include <Atomic/Graphics/VertexBuffer.h>
#include <Atomic/Graphics/Viewport.h>
#include <Atomic/Graphics/Graphics.h>
#include <Atomic/Graphics/RenderPath.h>
#include <Atomic/Graphics/Camera.h>
#include <Atomic/Graphics/Light.h>
#include <Atomic/Graphics/Octree.h>
#include <Atomic/Graphics/AnimatedModel.h>
#include <Atomic/Graphics/Material.h>

#include <Atomic/Navigation/NavigationMesh.h>
#include <Atomic/Navigation/CrowdManager.h>

#include <Atomic/Physics/PhysicsWorld.h>

#include <Atomic/Scene/ValueAnimation.h>

#include <AtomicNET/NETScript/CSComponent.h>

#include <Atomic/Input/Controls.h>

#include <Atomic/Atomic2D/PhysicsWorld2D.h>
#include <Atomic/Core/Profiler.h>
#include <Atomic/IO/Log.h>

#include "NETCore.h"

#ifdef ATOMIC_PLATFORM_WINDOWS
#define ATOMIC_EXPORT_API __declspec(dllexport)
#else
#define ATOMIC_EXPORT_API
#endif

// TODO: Split into separate module files

// IMPORTANT: methods here are prefaced with csi_ instead of csb_, the later being automatically generated bindings

namespace Atomic
{

#ifdef ATOMIC_PLATFORM_IOS
    static const char *sdlResourceDir = 0;
    static const char *sdlDocumentsDir = 0;
#endif

    extern "C"
    {
        ATOMIC_EXPORT_API void csi_Atomic_VariantMap_SetBool(VariantMap* vmap, const char* key, bool value)
        {
            if (!vmap)
                return;

            (*vmap)[key] = value;
        }

        ATOMIC_EXPORT_API ClassID csi_Atomic_RefCounted_GetClassID(RefCounted* refCounted)
        {
            if (!refCounted)
                return 0;

            return refCounted->GetClassID();
        }

        ATOMIC_EXPORT_API void csi_AtomicEngine_AddRef(RefCounted* refCounted)
        {
            if (!NETCore::EnsureMainThread("csi_AtomicEngine_AddRef - not on main thread"))
                return;

            if (!refCounted)
                return;

            refCounted->AddRef();
        }


        ATOMIC_EXPORT_API void csi_AtomicEngine_AddRefSilent(RefCounted* refCounted)
        {
            if (!NETCore::EnsureMainThread("csi_AtomicEngine_AddRefSilent - not on main thread"))
                return;

            if (!refCounted)
                return;

            refCounted->AddRefSilent();
        }


        ATOMIC_EXPORT_API void csi_AtomicEngine_ReleaseRef(RefCounted* refCounted)
        {
            if (!NETCore::EnsureMainThread("csi_AtomicEngine_ReleaseRef - not on main thread"))
                return;

            if (!refCounted)
                return;

            refCounted->ReleaseRef();
        }

        ATOMIC_EXPORT_API void csi_AtomicEngine_ReleaseRefSilent(RefCounted* refCounted)
        {
            if (!NETCore::EnsureMainThread("csi_AtomicEngine_ReleaseRefSilent - not on main thread"))
                return;

            if (!refCounted)
                return;

            refCounted->ReleaseRefSilent();
        }

        ATOMIC_EXPORT_API bool csi_AtomicEngine_IsMainThread()
        {
            return Thread::IsMainThread();
        }

        ATOMIC_EXPORT_API const char* csi_Atomic_RefCounted_GetTypeName(RefCounted* self)
        {
            return self ? self->GetTypeName().CString() : "(NULL)";
        }

        ATOMIC_EXPORT_API int csi_Atomic_RefCounted_Refs(RefCounted* self)
        {
            if (!self)
                return 0;

            return self->Refs();
        }


        ATOMIC_EXPORT_API void csi_Atomic_AObject_SendEvent(Object* obj, const char* eventType, ScriptVariantMap* vmap)
        {
            obj->SendEvent(eventType, vmap ? vmap->GetVariantMap() : obj->GetEventDataMap());
        }

        ATOMIC_EXPORT_API void csi_Atomic_AObject_UnsubscribeFromAllEvents(Object* obj)
        {
            if (!obj)
                return;

            obj->UnsubscribeFromAllEvents();
        }


        ATOMIC_EXPORT_API ClassID csi_Atomic_NETCore_Initialize(NETCoreEventDispatchFunction eventDispatch, 
            NETCoreUpdateDispatchFunction updateDispatch, 
            NETCoreRefCountedDeletedFunction refCountedDeleted,
            NETCoreThrowManagedExceptionFunction throwManagedException)
        {
            Context* context = new Context();

            // NOTE: We don't simply marshal the NETCoreDelegates structure due to iOS "reverse callback" limitation
            NETCoreDelegates delegates;
            delegates.eventDispatch = eventDispatch;
            delegates.updateDispatch = updateDispatch;
            delegates.refCountedDeleted = refCountedDeleted;
            delegates.throwManagedException = throwManagedException;

            NETCore* netCore = new NETCore(context, &delegates);
            context->RegisterSubsystem(netCore);
            return netCore;
        }

        ATOMIC_EXPORT_API unsigned csi_Atomic_AtomicNET_StringToStringHash(const char* str)
        {
            unsigned hash = 0;

            if (!str)
                return hash;

            while (*str)
            {
                // Perform the actual hashing as case-insensitive
                char c = *str;
                hash = SDBMHash(hash, (unsigned char)tolower(c));
                ++str;
            }

            return hash;
        }

        ATOMIC_EXPORT_API void csi_Atomic_AtomicNET_ScriptVariantMapCopyVariantMap(ScriptVariantMap* svm, VariantMap* vm)
        {
            if (!svm)
                return;

            if (!vm)
            {
                svm->CopySourceVariantMap(Variant::emptyVariantMap);
                return;
            }

            svm->CopySourceVariantMap(*vm);

        }

        ATOMIC_EXPORT_API void* csi_Atomic_AtomicNET_ScriptVariantMap_GetVoidPtr(ScriptVariantMap* svm, const char* key)
        {
            if (!svm || !key || !strlen(key))
                return nullptr;

            return svm->GetVoidPtr(key);

        }

        // IPC
        ATOMIC_EXPORT_API void csi_Atomic_IPC_SendEventToBrokerWithEventData(IPC* ipc, const char* eventType, ScriptVariantMap* variantMap)
        {

#ifdef ATOMIC_PLATFORM_DESKTOP
            if (variantMap)
                ipc->SendEventToBroker(eventType, variantMap->GetVariantMap());
            else
                ipc->SendEventToBroker(eventType);
#endif

        }

        // Graphics

        ATOMIC_EXPORT_API void* csi_Atomic_Graphics_GetSDLWindow()
        {
            if (!NETCore::GetContext())
                return 0;

            if (!NETCore::GetContext()->GetSubsystem<Graphics>())
                return 0;

            return NETCore::GetContext()->GetSubsystem<Graphics>()->GetSDLWindow();

        }

        ATOMIC_EXPORT_API void* csi_Atomic_VertexBuffer_Lock(VertexBuffer* vb, unsigned start, unsigned count, bool discard)
        {
            if (!vb)
                return nullptr;

            return vb->Lock(start, count, discard);

        }

        ATOMIC_EXPORT_API void csi_Atomic_Graphics_SetShaderParameter_Matrix3x4(Graphics* graphics, const char* param, Matrix3x4* matrix)
        {
            if (!graphics || !param || !strlen(param))
                return;

            graphics->SetShaderParameter(param, *matrix);
        }

        // RenderPath

        ATOMIC_EXPORT_API void csi_Atomic_RenderPath_SetShaderParameter(RenderPath* renderPath, const char* name, ScriptVariant* value)
        {
            if (!renderPath || !name || !value)
                return;

            Vector2 v2 = value->GetVariant().GetVector2();

            renderPath->SetShaderParameter(name, value->GetVariant());
        }

        ATOMIC_EXPORT_API void csi_Atomic_RenderPath_GetShaderParameter(RenderPath* renderPath, const char* name, ScriptVariant* value)
        {
            if (!renderPath || !name || !value)
                return;

            value->SetVariant(renderPath->GetShaderParameter(name));
        }

        // Light

        ATOMIC_EXPORT_API void csi_Atomic_Light_SetShadowBias(Light* light, BiasParameters* parameters)
        {
            if (!parameters)
                return;

            light->SetShadowBias(*parameters);

        }

        ATOMIC_EXPORT_API void csi_Atomic_Light_GetShadowBias(Light* light, BiasParameters* parameters)
        {
            if (!parameters)
                return;

            *parameters = light->GetShadowBias();

        }


        ATOMIC_EXPORT_API void csi_Atomic_Light_SetShadowCascade(Light* light, CascadeParameters* parameters)
        {
            if (!parameters)
                return;

            light->SetShadowCascade(*parameters);

        }

        ATOMIC_EXPORT_API void csi_Atomic_Light_GetShadowCascade(Light* light, CascadeParameters* parameters)
        {
            if (!parameters)
                return;

            *parameters = light->GetShadowCascade();

        }


        ATOMIC_EXPORT_API void csi_Atomic_Light_SetShadowFocus(Light* light, FocusParameters* parameters)
        {
            if (!parameters)
                return;

            light->SetShadowFocus(*parameters);

        }

        ATOMIC_EXPORT_API void csi_Atomic_Light_GetShadowFocus(Light* light, FocusParameters* parameters)
        {
            if (!parameters)
                return;

            *parameters = light->GetShadowFocus();
        }

        // Camera

        ATOMIC_EXPORT_API void csi_Atomic_Camera_GetScreenRay(Camera* camera, float x, float y, Ray* ray)
        {
            if (!camera || !ray)
                return;

            *ray = camera->GetScreenRay(x, y);
        }


        ATOMIC_EXPORT_API void csi_Atomic_Camera_SetClipPlane(Camera* camera, Plane* plane)
        {
            if (!camera || !plane)
                return;

            camera->SetClipPlane(*plane);
        }

        ATOMIC_EXPORT_API void csi_Atomic_Camera_GetClipPlane(Camera* camera, Plane* plane)
        {
            if (!camera || !plane)
                return;

            *plane = camera->GetClipPlane();
        }

        ATOMIC_EXPORT_API void csi_Atomic_Camera_SetReflectionPlane(Camera* camera, Plane* plane)
        {
            if (!camera || !plane)
                return;

            camera->SetReflectionPlane(*plane);
        }


        ATOMIC_EXPORT_API void csi_Atomic_Camera_GetReflectionPlane(Camera* camera, Plane* plane)
        {
            if (!camera || !plane)
                return;

            *plane = camera->GetReflectionPlane();
        }

        // Octree

        ATOMIC_EXPORT_API void csi_Atomic_Octree_Raycast_FreeResult(PODVector<RayQueryResult>* resultVector)
        {
            delete resultVector;
        }
    

        // Any result vector must be freed with csi_Atomic_Octree_Raycast_FreeResult
        ATOMIC_EXPORT_API RayQueryResult* csi_Atomic_Octree_Raycast(Octree *octree, const Ray& ray, const RayQueryLevel& level, float maxDistance, unsigned int flags, unsigned int viewMask,
            bool single, void** resultVector, int *count) 
        {
            PODVector<RayQueryResult>* results = new PODVector<RayQueryResult>();

            *count = 0;
            *resultVector = 0;

            RayOctreeQuery query(*results, ray, level, maxDistance, flags, viewMask);
            
            if (single)
                octree->RaycastSingle(query);
            else
                octree->Raycast(query);

            if (results->Size() == 0)
            {
                delete results;
                return NULL;
            }

            *count = results->Size();
            *resultVector = results;
            return &(*results)[0];
        }

        
        // NavigationMesh

        ATOMIC_EXPORT_API void csi_Atomic_NavigationMesh_FindPath_FreeResult(PODVector<Vector3>* resultVector)
        {
            delete resultVector;
        }


        // Any result vector must be freed with csi_Atomic_NavigationMesh_FindPath_FreeResult
        ATOMIC_EXPORT_API Vector3* csi_Atomic_NavigationMesh_FindPath(NavigationMesh *navMesh, Vector3* start, Vector3* end, Vector3 *extents, void** resultVector, int *count)
        {
            PODVector<Vector3>* results = new PODVector<Vector3>();

            navMesh->FindPath(*results, *start, *end, *extents);

            *count = 0;
            *resultVector = 0;

            if (results->Size() == 0)
            {
                delete results;
                return NULL;
            }

            *count = results->Size();
            *resultVector = results;
            return &(*results)[0];
        }

        // AnimatedModel

        ATOMIC_EXPORT_API Skeleton* csi_Atomic_AnimatedModel_GetSkeleton(AnimatedModel *animatedModel)
        {
            if (!animatedModel)
                return 0;

            return &animatedModel->GetSkeleton();
        }

        // Skeleton

        ATOMIC_EXPORT_API void csi_Atomic_Skeleton_SetRootBoneIndex(Skeleton *skeleton, unsigned index)
        {
            if (!skeleton)
                return;

            skeleton->SetRootBoneIndex(index);
        }

        ATOMIC_EXPORT_API void csi_Atomic_Skeleton_Define(Skeleton *skeleton, Skeleton *src)
        {
            if (!skeleton || !src)
                return;

            skeleton->Define(*src);
        }


        ATOMIC_EXPORT_API void csi_Atomic_Skeleton_Reset(Skeleton *skeleton)
        {
            if (!skeleton)
                return;

            skeleton->Reset();
        }

        ATOMIC_EXPORT_API void csi_Atomic_Skeleton_ResetSilent(Skeleton *skeleton)
        {
            if (!skeleton)
                return;

            skeleton->ResetSilent();
        }

        ATOMIC_EXPORT_API void csi_Atomic_Skeleton_ClearBones(Skeleton *skeleton)
        {
            if (!skeleton)
                return;

            skeleton->ClearBones();
        }

        ATOMIC_EXPORT_API Bone* csi_Atomic_Skeleton_GetRootBone(Skeleton *skeleton)
        {
            if (!skeleton)
                return 0;

            return skeleton->GetRootBone();
        }


        ATOMIC_EXPORT_API Bone* csi_Atomic_Skeleton_GetBone_ByIndex(Skeleton *skeleton, unsigned index)
        {
            if (!skeleton)
                return 0;

            if (index >= skeleton->GetNumBones())
                return 0;

            return skeleton->GetBone(index);            
        }

        ATOMIC_EXPORT_API Bone* csi_Atomic_Skeleton_GetBone_ByName(Skeleton *skeleton, const char* name)
        {
            if (!skeleton || !name || !strlen(name))
                return 0;

            return skeleton->GetBone(name);
        }

        ATOMIC_EXPORT_API unsigned csi_Atomic_Skeleton_GetNumBones(Skeleton *skeleton)
        {
            if (!skeleton)
                return 0;

            return skeleton->GetNumBones();
        }

        // PhysicsWorld

        ATOMIC_EXPORT_API void csi_Atomic_PhysicsWorld_RaycastSingle(PhysicsWorld* world, Ray* ray, float maxDistance, unsigned collisionMask, PhysicsRaycastResult* result)
        {
            if (!world || !ray || !result)
                return;

            world->RaycastSingle(*result, *ray, maxDistance, collisionMask);

        }

        // PhysicsWorld2D

        ATOMIC_EXPORT_API void csi_Atomic_PhysicsWorld2D_RaycastSingle(PhysicsWorld2D* world2D, Vector2* startPoint, Vector2* endPoint, unsigned collisionMask, PhysicsRaycastResult2D* result)
        {
            if (!world2D || !result)
                return;
            world2D->RaycastSingle(*result, *startPoint, *endPoint, collisionMask);
        }

        // Controls
        
        ATOMIC_EXPORT_API unsigned csi_Atomic_Controls_GetButtons(Controls *controls)
        {
            return controls->buttons_;
        }

        ATOMIC_EXPORT_API void csi_Atomic_Controls_SetButtons(Controls *controls, unsigned value)
        {
            controls->buttons_ = value;
        }

        ATOMIC_EXPORT_API float csi_Atomic_Controls_GetYaw(Controls *controls)
        {
            return controls->yaw_;
        }

        ATOMIC_EXPORT_API void csi_Atomic_Controls_SetYaw(Controls *controls, float value)
        {
            controls->yaw_ = value;
        }

        ATOMIC_EXPORT_API float csi_Atomic_Controls_GetPitch(Controls *controls)
        {
            return controls->pitch_;
        }

        ATOMIC_EXPORT_API void csi_Atomic_Controls_SetPitch(Controls *controls, float value)
        {
            controls->pitch_ = value;
        }

        ATOMIC_EXPORT_API void csi_Atomic_Controls_Reset(Controls *_target)
        {
            _target->Reset();
        }

        ATOMIC_EXPORT_API void csi_Atomic_Controls_Set(Controls *_target, unsigned int buttons, int down)
        {
            _target->Set(buttons, down);
        }

        ATOMIC_EXPORT_API int csi_Atomic_Controls_IsDown(Controls *_target, unsigned int button)
        {
            return _target->IsDown(button);
        }

        ATOMIC_EXPORT_API Controls*  csi_Atomic_Controls_Create()
        {
            return new Controls();
        }

        ATOMIC_EXPORT_API void  csi_Atomic_Controls_Destroy(Controls *controls)
        {
            if (!controls)
                return;

            delete controls;
        }

        // CrowdManager

        ATOMIC_EXPORT_API void  csi_Atomic_CrowdManager_GetObstacleAvoidanceParams(CrowdManager *manager, unsigned obstacleAvoidanceType, CrowdObstacleAvoidanceParams* parms)
        {
            if (!manager || !parms)
                return;

            *parms = manager->GetObstacleAvoidanceParams(obstacleAvoidanceType);
        }

        ATOMIC_EXPORT_API void  csi_Atomic_CrowdManager_SetObstacleAvoidanceParams(CrowdManager *manager, unsigned obstacleAvoidanceType, CrowdObstacleAvoidanceParams* parms)
        {
            if (!manager || !parms)
                return;

            manager->SetObstacleAvoidanceParams(obstacleAvoidanceType, *parms);
        }

        // ValueAnimation

        ATOMIC_EXPORT_API bool csi_Atomic_ValueAnimation_SetKeyFrame(ValueAnimation* self, float time, ScriptVariant* variant)
        {
            if (!self || !variant)
                return false;

            return self->SetKeyFrame(time, variant->GetVariant());
        }

        // File

        ATOMIC_EXPORT_API unsigned csi_Atomic_File_Read(File* self, void* dest, unsigned size)
        {
            return self->Read(dest, size);
        }

        ATOMIC_EXPORT_API unsigned csi_Atomic_File_Write(File* self, void* data, unsigned offset, unsigned size)
        {
            unsigned char* bytes = (unsigned char*) data;
            bytes += offset;
            return self->Write(bytes, size);
        }

        // Material

        ATOMIC_EXPORT_API void csi_Atomic_Material_SetShaderParameter(Material* self, const char* name, ScriptVariant* variant)
        {

            if (!self || !variant || !name || !strlen(name))
                return;

            self->SetShaderParameter(name, variant->GetVariant());
        }

        // CSComponent

        ATOMIC_EXPORT_API CSComponent* csi_Atomic_CSComponent_Constructor()
        {

            return new CSComponent(NETCore::GetContext());
        }

        // Frustum

        ATOMIC_EXPORT_API void csi_Atomic_Frustum_Define(Frustum* self, float fov, float aspectRatio, float zoom, float nearZ, float farZ, Vector3* position, Quaternion* rotation)
        {
            if (!self || !position || !rotation)
            {
                return;
            }

            self->Define(fov, aspectRatio, zoom, nearZ, farZ, Matrix3x4(*position, *rotation, 1.0f));

        }

        ATOMIC_EXPORT_API void csi_Atomic_Frustum_DefineOrtho(Frustum* self, float orthoSize, float aspectRatio, float zoom, float nearZ, float farZ, Vector3* position, Quaternion* rotation)
        {
            if (!self || !position || !rotation)
            {
                return;
            }

            self->DefineOrtho(orthoSize, aspectRatio, zoom, nearZ, farZ, Matrix3x4(*position, *rotation, 1.0f));

        }

        ATOMIC_EXPORT_API Intersection csi_Atomic_Frustum_IsInside_Vector3(Frustum* self, Vector3* point)
        {
            if (!self || !point)
            {
                return OUTSIDE;
            }

            return self->IsInside(*point);
        }

        ATOMIC_EXPORT_API Intersection csi_Atomic_Frustum_IsInside_BoundingBox(Frustum* self, BoundingBox* bbox)
        {
            if (!self || !bbox)
            {
                return OUTSIDE;
            }

            return self->IsInside(*bbox);
        }

        ATOMIC_EXPORT_API Intersection csi_Atomic_Frustum_IsInsideFast_BoundingBox(Frustum* self, BoundingBox* bbox)
        {
            if (!self || !bbox)
            {
                return OUTSIDE;
            }

            return self->IsInsideFast(*bbox);
        }

        ATOMIC_EXPORT_API float csi_Atomic_Frustum_Distance(Frustum* self, Vector3* point)
        {
            if (!self || !point)
            {
                return 0.0f;
            }

            return self->Distance(*point);
        }

        // Native method is suppressed from automated bindings, for additional GC logic in C# version of method
        ATOMIC_EXPORT_API void csi_Atomic_ResourceCache_ReleaseAllResources(ResourceCache* self, bool force)
        {
            if (!self)
            {
                return;
            }

            self->ReleaseAllResources(force);
        }

#ifdef ATOMIC_PLATFORM_IOS
        ATOMIC_EXPORT_API void SDL_IOS_Init(const char *resourceDir, const char *documentsDir)
        {
            sdlResourceDir = resourceDir;
            sdlDocumentsDir = documentsDir;
        }
#endif

        ATOMIC_EXPORT_API void csi_Atomic_Profiler_BeginBlock(Profiler* profiler, const char* name, const char* file, int line, unsigned int argb, unsigned char status)
        {
#if ATOMIC_PROFILING
            if (!profiler)
                return;

            profiler->BeginBlock(name, file, line, argb, status);
#else
            static bool warned = false;
            if (!warned)
            {
                warned = true;
                ATOMIC_LOGWARNING("Engine is built without profiler support.");
            }
#endif
        }

        ATOMIC_EXPORT_API void csi_Atomic_Profiler_EndBlock(Profiler* profiler)
        {
#if ATOMIC_PROFILING
            if (!profiler)
                return;

            profiler->EndBlock();
#endif
        }
    }
}

#ifdef ATOMIC_PLATFORM_IOS

//FileSystem.cpp uses these functions as external.
const char* SDL_IOS_GetResourceDir()
{
    return Atomic::sdlResourceDir;
}

const char* SDL_IOS_GetDocumentsDir()
{
    return Atomic::sdlDocumentsDir;
}

#endif