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 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);
        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);}
        }

    }
}