#include "ScriptSystem.h"

#pragma once

namespace Atomic
{

class ScriptVector : public RefCounted
{
    ATOMIC_REFCOUNTED(ScriptVector)

public:

    ScriptVector() : RefCounted()
    {

    }

    virtual ~ScriptVector()
    {

    }

    RefCounted* At(unsigned index) const
    {
        if (index >= refVector_.Size())
            return 0;

        return refVector_[index];
    }

    void Push(RefCounted* refCounted)
    {
        // TODO: check null?

        refVector_.Push(SharedPtr<RefCounted>(refCounted));
    }

    unsigned GetSize() const
    {
        return refVector_.Size();
    }

    template <class T>
    bool AdaptFromVector(PODVector<T> vectorIn)
    {

        Vector<SharedPtr<RefCounted>> keepAlive = refVector_;

        refVector_.Clear();

        for (unsigned i = 0; i < vectorIn.Size(); i++)
        {
            refVector_.Push(SharedPtr<RefCounted>(vectorIn[i]));
        }

        return true;
    }

    template <class T>
    bool AdaptFromVector(Vector<T> vectorIn)
    {
        Vector<SharedPtr<RefCounted>> keepAlive = refVector_;

        refVector_.Clear();

        for (unsigned i = 0; i < vectorIn.Size(); i++)
        {
            refVector_.Push((T) vectorIn[i]);
        }

        return true;

    }

    template <class T>
    bool AdaptFromVector(Vector<WeakPtr<T>> vectorIn)
    {
        Vector<SharedPtr<RefCounted>> keepAlive = refVector_;

        refVector_.Clear();

        for (unsigned i = 0; i < vectorIn.Size(); i++)
        {
            refVector_.Push(SharedPtr<RefCounted>((RefCounted*) vectorIn[i]));
        }

        return true;

    }

    template <class T>
    bool AdaptFromVector(Vector<SharedPtr<T>> vectorIn)
    {
        Vector<SharedPtr<RefCounted>> keepAlive = refVector_;

        refVector_.Clear();

        for (unsigned i = 0; i < vectorIn.Size(); i++)
        {
            refVector_.Push(SharedPtr<RefCounted>((RefCounted*) vectorIn[i]));
        }

        return true;

    }

    template <class T>
    bool AdaptToVector(PODVector<T> vectorOut)
    {
        vectorOut.Clear();

        for (unsigned i = 0; i < refVector_.Size(); i++)
        {
            vectorOut.Push(static_cast<T>(refVector_[i].Get()));
        }

        return true;

    }

    template <class T>
    bool AdaptToVector(Vector<T> vectorOut)
    {
        vectorOut.Clear();

        for (unsigned i = 0; i < refVector_.Size(); i++)
        {
            vectorOut.Push((T) refVector_[i]);
        }

        return true;

    }

    template <class T>
    bool AdaptToVector(Vector<SharedPtr<T>> vectorOut)
    {
        vectorOut.Clear();

        for (unsigned i = 0; i < refVector_.Size(); i++)
        {
            vectorOut.Push(SharedPtr<T>((T) refVector_[i]));
        }

        return true;

    }

    template <class T>
    bool AdaptToVector(Vector<WeakPtr<T>> vectorOut)
    {
        vectorOut.Clear();

        for (unsigned i = 0; i < refVector_.Size(); i++)
        {
            vectorOut.Push(WeakPtr<T>((T) refVector_[i]));
        }

        return true;
    }


private:

    Vector<SharedPtr<RefCounted>> refVector_;

};

}