import java.awt.*;
import java.util.Random;

public class DrawWheelCircle 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);
        Random rnd = new Random();
        g2d.fillOval(10 + _startPosX, 60 + _startPosY, 20, 20);
        g2d.fillOval(80 + _startPosX, 60 + _startPosY, 20, 20);
        Color stringColor = new Color(rnd.nextInt(0, 256), rnd.nextInt(0, 256), rnd.nextInt(0, 256));
        g2d.setColor(stringColor);
        g2d.drawString("S",15 + _startPosX, 75 + _startPosY);
        g2d.drawString("S",85 + _startPosX, 75 + _startPosY);
        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);}
        }
        switch (wheelCounter)
        {
            case THREE -> {g2d.drawString("S",50 + _startPosX, 75 + _startPosY);}
            case FOUR -> {g2d.drawString("S",35 + _startPosX, 75 + _startPosY); g2d.drawString("S",65 + _startPosX, 75 + _startPosY);}
        }
    }
}