PIbd-12_Karamushko_M.K._AirFighter._Hard/DrawingEngines.java

import java.awt.*;

public class DrawingEngines {
    private EnginesCount enginesCount;
    private Color color;

    public void Init(int count, Color bodyColor) {
        setCount(count);
        color = bodyColor;
    }

    public void setCount(int count) {
        if(count <= 2) enginesCount = EnginesCount.Two;
        else if(count >= 6) enginesCount = EnginesCount.Six;
        else enginesCount = EnginesCount.Four;
    }

    public void draw(Graphics2D g, int startPosX, int startPosY) {
        g.setPaint(color);
        g.fillOval(startPosX + 90, startPosY + 10, 30, 15);
        g.fillOval(startPosX + 90, startPosY + 141, 30, 15);

        if(enginesCount == EnginesCount.Two) return;

        g.fillOval(startPosX + 90, startPosY + 30, 30, 15);
        g.fillOval(startPosX + 90, startPosY + 121, 30, 15);

        if(enginesCount == EnginesCount.Four) return;

        g.fillOval(startPosX + 90, startPosY + 50, 30, 15);
        g.fillOval(startPosX + 90, startPosY + 101, 30, 15);
    }
}