#include <SFML/Graphics.hpp>
#include <iostream>


class Troops
{
public:
    virtual sf::Texture& spawn() = 0;
    virtual sf::Sprite& sprite();
protected:
    sf::Texture _mob;
    sf::Sprite _sprite;
};
sf::Sprite& Troops::sprite()
{
    _sprite.setTexture(spawn());
    _sprite.setTextureRect(sf::IntRect(32,0,32,32));
    return _sprite;
}
class Yellowhair : public Troops
{
    sf::Texture& spawn();
    //sf::Sprite& sprite();
};

class Bandit : public Troops
{
    sf::Texture& spawn();
   // sf::Sprite& sprite();
};

class Doomknight : public Troops
{
    sf::Texture& spawn();
   // sf::Sprite& sprite();
};

class Rouge : public Troops
{
    sf::Texture& spawn();
   // sf::Sprite& sprite();
};

sf::Texture& Yellowhair::spawn()
{
    if (!_mob.loadFromFile("troops/yellowhair.png"))
        std::cout << "Error could not load the file";
    return _mob;

}

sf::Texture& Bandit::spawn()
{
    if(!_mob.loadFromFile("troops/bandit.png"));
        std::cout << "Error could not load the file";
    return _mob;
}

sf::Texture& Doomknight::spawn()
{
    if(!_mob.loadFromFile("troops/doomknight.png"));
        std::cout << "Error could not load the file";
    return _mob;
}

sf::Texture& Rouge::spawn()
{
    if(!_mob.loadFromFile("troops/rouge.png"));
        std::cout << "Error could not load the file";
    return _mob;
}



int main()
{
    sf::RenderWindow w(sf::VideoMode(500,500),"SFML WORKS!",sf::Style::Default);
    sf::Texture texture;
    if( !texture.loadFromFile("troops/trooplist.png"))return 0;
    sf::Sprite spr;
    spr.setTexture(texture);
    spr.setPosition(sf::Vector2f(0,w.getSize().y - texture.getSize().y));


    Troops *troops[4]{
    new Yellowhair,
    new Bandit,
    new Doomknight,
    new Rouge
    };

  //  enum dir { DOWN,LEFT,RIGHT,UP };
    int select = 0;
    w.draw(spr);
    while(w.isOpen())
    {
        sf::Event event;
        while(w.pollEvent(event))
        {
             if ( sf::Keyboard::isKeyPressed(sf::Keyboard::Num1) )
             {
                 select = 1;
             }
             else if ( sf::Keyboard::isKeyPressed(sf::Keyboard::Num2) )
             {
                 select = 2;
             }
             else if ( sf::Keyboard::isKeyPressed(sf::Keyboard::Num3) )
             {
                 select = 3;
             }
             else if ( sf::Keyboard::isKeyPressed(sf::Keyboard::Num4) )
             {
                 select = 4;
             }
             if ( select )
             {

                 if ( sf::Mouse::isButtonPressed(sf::Mouse::Left) )
                 {
                    //sprite[select-1].setPosition(sf::Vector2f(event.mouseButton.x,event.mouseButton.y));
                    //w.draw(sprite[select-1]);
                    troops[select-1]->sprite().setPosition(sf::Vector2f(event.mouseButton.x,event.mouseButton.y));
                    w.draw(troops[select-1]->sprite());

                 }
             }
        }

       w.display();
    }
}