/*
* All or portions of this file Copyright (c) Amazon.com, Inc. or its affiliates or
* its licensors.
*
* For complete copyright and license terms please see the LICENSE at the root of this
* distribution (the "License"). All use of this software is governed by the License,
* or, if provided, by the license below or the license accompanying this file. Do not
* remove or modify any license notices. This file is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*
*/
// Original file Copyright Crytek GMBH or its affiliates, used under license.

#ifndef CRYINCLUDE_CRYANIMATION_FACIALANIMATION_FACEJOYSTICK_H
#define CRYINCLUDE_CRYANIMATION_FACIALANIMATION_FACEJOYSTICK_H
#pragma once


#include "IJoystick.h"
#include "smartptr.h"

struct IFacialAnimChannel;

class IFacialJoystickSerializeContext
{
public:
    virtual ~IFacialJoystickSerializeContext(){}
    virtual IFacialAnimChannel* FindChannel(const char* szName) = 0;
};

class CFacialJoystickChannel
    : public IJoystickChannel
{
public:
    CFacialJoystickChannel(const string& channelName);
    CFacialJoystickChannel(IFacialAnimChannel* pChannel);

    // IJoystickChannel
    virtual void AddRef();
    virtual void Release();

    virtual const char* GetName() const;

    virtual void* GetTarget();
    virtual int GetSplineCount();
    virtual ISplineInterpolator* GetSpline(int splineIndex);

    virtual void SetFlipped(bool flipped);
    virtual bool GetFlipped() const;

    virtual void SetVideoMarkerOffset(float offset);
    virtual float GetVideoMarkerOffset() const;
    virtual void SetVideoMarkerScale(float scale);
    virtual float GetVideoMarkerScale() const;

    virtual void CleanupKeys(float fErrorMax);

    const char* GetPath() const;

    IFacialAnimChannel* GetFacialChannel();

    void Bind(IFacialJoystickSerializeContext* pContext);

private:
    IFacialAnimChannel* m_pChannel;
    string m_channelName;
    int m_refCount;
    bool m_bFlipped;
    float m_videoMarkerOffset;
    float m_videoMarkerScale;
};

class CFacialJoystick
    : public IJoystick
{
public:
    CFacialJoystick(uint64 id);

    // IJoystick
    virtual void AddRef();
    virtual void Release();

    virtual uint64 GetID() const;

    virtual void SetName(const char* szName);
    virtual const char* GetName();

    virtual IJoystickChannel* GetChannel(ChannelType type);
    virtual void SetChannel(ChannelType type, IJoystickChannel* pChannel);

    virtual const Vec2& GetCentre() const;
    virtual void SetCentre(const Vec2& vCentre);
    virtual const Vec2& GetDimensions() const;
    virtual void SetDimensions(const Vec2& vDimensions);

    CFacialJoystickChannel* GetDerivedChannel(ChannelType type);

    void Bind(IFacialJoystickSerializeContext* pContext);

    virtual void SetColor(const Color& colour);
    virtual IJoystick::Color GetColor() const;

private:
    _smart_ptr<CFacialJoystickChannel> m_channels[2];
    uint64 m_id;
    Vec2 m_vCentre;
    Vec2 m_vDimensions;
    int m_refCount;
    string m_name;
    Color m_colour;
};

class CFacialJoystickSet
    : public IJoystickSet
{
public:
    CFacialJoystickSet();

    void HandleRemovedChannel(IFacialAnimChannel* pChannel);

    // IJoystickSet
    virtual void AddRef();
    virtual void Release();

    virtual void SetName(const char* name);
    virtual const char* GetName() const;
    virtual void AddJoystick(IJoystick* pJoystick);
    virtual void RemoveJoystick(IJoystick* pJoystick);
    virtual int GetJoystickCount() const;
    virtual IJoystick* GetJoystick(int index);
    virtual IJoystick* GetJoystickAtPoint(const Vec2& vPosition);
    virtual IJoystick* GetJoystickByID(uint64 id);

    virtual void Serialize(XmlNodeRef& node, bool bLoading);

    void Bind(IFacialJoystickSerializeContext* pContext);

private:
    typedef std::vector<_smart_ptr<CFacialJoystick> > JoystickContainer;
    JoystickContainer m_joysticks;
    int m_refCount;
    string m_name;
};

#endif // CRYINCLUDE_CRYANIMATION_FACIALANIMATION_FACEJOYSTICK_H