import java.awt.*;

public class DrawWheelClassic implements IWheelDraw{
    private WheelCounter wheelCounter;
    public WheelCounter getWheelCounter()
    {
        return wheelCounter;
    }
    public void setWheelCount(int count)
    {
        if (count % 3 == 0)
            wheelCounter = WheelCounter.THREE;
        else if (count % 3 == 1)
            wheelCounter = WheelCounter.FOUR;
        else if (count % 3 == 2)
            wheelCounter = WheelCounter.TWO;
    }
    public int getWheelCount()
    {
        switch (wheelCounter)
        {
            case TWO -> {return 2;}
            case THREE -> {return 3;}
            case FOUR -> {return 4;}
        }
        return 2;
    }
    public void DrawWheels(int _startPosX, int _startPosY, Color bodyColor, Graphics2D g2d)
    {
        g2d.setColor(bodyColor);
        g2d.fillOval(10 + _startPosX, 60 + _startPosY, 20, 20);
        g2d.fillOval(80 + _startPosX, 60 + _startPosY, 20, 20);
        if (wheelCounter == null)
            return;
        switch (wheelCounter)
        {
            case THREE -> {g2d.fillOval(45 + _startPosX, 60 + _startPosY ,20, 20);}
            case FOUR -> {g2d.fillOval(30 + _startPosX, 60 + _startPosY ,20, 20); g2d.fillOval(60 + _startPosX, 60 + _startPosY ,20, 20);}
        }

    }
}