diff --git a/ProjectAirFighter/src/CountEngines.java b/ProjectAirFighter/src/CountEngines.java index 495ac36..cf1a723 100644 --- a/ProjectAirFighter/src/CountEngines.java +++ b/ProjectAirFighter/src/CountEngines.java @@ -4,9 +4,7 @@ public enum CountEngines { Four(4), Six(6); private int Count; - public int getCount() { - return Count; - } + public int getCount() { return Count; } CountEngines(int num){ this.Count = num; } diff --git a/ProjectAirFighter/src/DrawningAirFighter.java b/ProjectAirFighter/src/DrawningAirFighter.java index 7eda438..efc6a50 100644 --- a/ProjectAirFighter/src/DrawningAirFighter.java +++ b/ProjectAirFighter/src/DrawningAirFighter.java @@ -20,7 +20,7 @@ public class DrawningAirFighter { _pictureHeight = null; _startPosX = null; _startPosY = null; - DrawningEngines = new DrawningEngines(EntityAirFighter); + DrawningEngines = new DrawningEngines(); Random random = new Random(); int Count = 2 * random.nextInt(4); DrawningEngines.SetCountEngines(Count); diff --git a/ProjectAirFighter/src/DrawningEngines.java b/ProjectAirFighter/src/DrawningEngines.java index 4fa58a8..a25e1ed 100644 --- a/ProjectAirFighter/src/DrawningEngines.java +++ b/ProjectAirFighter/src/DrawningEngines.java @@ -2,21 +2,16 @@ import java.awt.*; public class DrawningEngines { private CountEngines CountEngines; - public CountEngines getCountEngines() { - return CountEngines; - } + private EntityAirFighter EntityAirFighter; public void SetCountEngines(int countEngines) { for (CountEngines num : CountEngines.values()) { if (num.getCount() == countEngines) { CountEngines = num; - return; } } } - public DrawningEngines(EntityAirFighter entityAirFighter) { - EntityAirFighter = entityAirFighter; - } + public void DrawEngines(Graphics2D g, Color color, int x, int y) { g.setColor(color); for (int i = CountEngines.getCount(); i > 0; i -= 2) {