// Copyright (C) 2002-2015 Nikolaus Gebhardt, modified by Marianne Gagnon
// This file is part of the "Irrlicht Engine".
// For conditions of distribution and use, see copyright notice in irrlicht.h

#ifndef __C_GUI_MODIFIED_SPRITE_BANK_H_INCLUDED__
#define __C_GUI_MODIFIED_SPRITE_BANK_H_INCLUDED__

#include "IrrCompileConfig.h"
#ifdef _IRR_COMPILE_WITH_GUI_
#include <cassert>
#include "IGUISpriteBank.h"
#include "utils/leak_check.hpp"

namespace irr
{

namespace video
{
    class IVideoDriver;
    class ITexture;
}

namespace gui
{

    class IGUIEnvironment;

//! Sprite bank interface.
class STKModifiedSpriteBank : public IGUISpriteBank
{
public:

    LEAK_CHECK()

    STKModifiedSpriteBank(IGUIEnvironment* env);
    virtual ~STKModifiedSpriteBank();

    virtual core::array< core::rect<s32> >& getPositions();
    virtual core::array< SGUISprite >& getSprites();

    virtual u32 getTextureCount() const;
    virtual video::ITexture* getTexture(u32 index) const;
    virtual void addTexture(video::ITexture* texture);
    virtual void setTexture(u32 index, video::ITexture* texture);

    //! Add the texture and use it for a single non-animated sprite.
    virtual s32 addTextureAsSprite(video::ITexture* texture);

    //! clears sprites, rectangles and textures
    virtual void clear();

    //! Draws a sprite in 2d with position and color
    virtual void draw2DSprite(u32 index, const core::position2di& pos, const core::rect<s32>* clip=0,
                const video::SColor& color= video::SColor(255,255,255,255),
                u32 starttime=0, u32 currenttime=0, bool loop=true, bool center=false);

    //! Draws a sprite batch in 2d using an array of positions and a color
    virtual void draw2DSpriteBatch(const core::array<u32>& indices, const core::array<core::position2di>& pos,
            const core::rect<s32>* clip=0,
            const video::SColor& color= video::SColor(255,255,255,255),
            u32 starttime=0, u32 currenttime=0,
            bool loop=true, bool center=false);

    void setScale(float scale)
    {
        assert( m_magic_number == 0xCAFEC001 );
        m_scale = scale;
    }

    void scaleToHeight(int height);

    void setTargetIconSize(int width, int height)
                    { m_target_icon_size = core::dimension2du(width, height); }

protected:

    // this object was getting access after being freed, I wanna see when/why
    unsigned int m_magic_number;

    float m_scale;
    int m_height;

    struct SDrawBatch
    {
        core::array<core::position2di> positions;
        core::array<core::recti> sourceRects;
        u32 textureNumber;
    };

    core::dimension2du m_target_icon_size;
    //FIXME: ugly hack to work around irrLicht limitations, see STKModifiedSpriteBank::getPositions()
    // for all the gory details.
    core::array< core::rect<s32> > copy;

    core::array<SGUISprite> Sprites;
    core::array< core::rect<s32> > Rectangles;
    core::array<video::ITexture*> Textures;
    IGUIEnvironment* Environment;
    video::IVideoDriver* Driver;

    s32 getScaledWidth(s32 width) const;
    s32 getScaledHeight(s32 height) const;
};

} // end namespace gui
} // end namespace irr

#endif // _IRR_COMPILE_WITH_GUI_

#endif // __C_GUI_SPRITE_BANK_H_INCLUDED__