import java.awt.*;
public class NumberOfWheels {
    public int numOfWheels;
    public void setNumOfWheels(String num){
        switch (Integer.valueOf(num)){
            case 2:
                numOfWheels = NumberOfWheelsEnum.wheel_2.value;
                break;
            case 3:
                numOfWheels = NumberOfWheelsEnum.wheel_3.value;
                break;
            case 4:
                numOfWheels = NumberOfWheelsEnum.wheel_4.value;
                break;
            default:
                numOfWheels = NumberOfWheelsEnum.wheel_2.value;
        }
    }
    public void drawWheels(Graphics g, int _startPosX, int _startPosY, Color c) {
        g.setColor(Color.black);
        if (numOfWheels == 2) {
            g.fillOval(_startPosX+20, _startPosY  + 20, 15, 15);
            g.fillOval(_startPosX + 45, _startPosY + 20, 15, 15);
        } else if (numOfWheels == 3) {
            g.fillOval(_startPosX, _startPosY + 20, 15, 15);
            g.fillOval(_startPosX + 25, _startPosY  + 20, 15, 15);
            g.fillOval(_startPosX + 55, _startPosY  + 20, 15, 15);
        } else {
            g.fillOval(_startPosX, _startPosY  + 20, 15, 15);
            g.fillOval(_startPosX + 15, _startPosY  + 20, 15, 15);
            g.fillOval(_startPosX + 65 , _startPosY  + 20, 15, 15);
            g.fillOval(_startPosX + 85, _startPosY + 20, 15, 15);
        }
    }
}