import java.awt.Color; import java.awt.Graphics; public class DrawingWheels{ Graphics g; Color color; private CountWheels _countWheels; public void setNumWheels(int numRollers) { if (numRollers < 4 || numRollers > 6 ){ _countWheels = CountWheels.fromNumberToEnum(numRollers); return; } _countWheels = CountWheels.fromNumberToEnum(numRollers); } public DrawingWheels(Color color, int countWheels){ this.color = color; setNumWheels(countWheels); } public void Draw(Graphics g, int _startPosX, int _startPosY){ switch(_countWheels){ case Min: g.setColor(Color.BLACK); g.drawOval(_startPosX + 20, _startPosY + 35, 15, 15); g.drawOval(_startPosX + 110, _startPosY + 35, 15, 15); g.setColor(color); g.fillOval(_startPosX + 20, _startPosY + 35, 15, 15); g.fillOval(_startPosX + 110, _startPosY + 35, 15, 15); break; case Mid: g.setColor(Color.BLACK); g.drawOval(_startPosX + 20, _startPosY + 35, 15, 15); g.drawOval(_startPosX + 65, _startPosY + 35, 15, 15); g.drawOval(_startPosX + 110, _startPosY + 35, 15, 15); g.setColor(color); g.fillOval(_startPosX + 20, _startPosY + 35, 15, 15); g.fillOval(_startPosX + 65, _startPosY + 35, 15, 15); g.fillOval(_startPosX + 110, _startPosY + 35, 15, 15); break; case Max: g.setColor(Color.BLACK); g.drawOval(_startPosX + 20, _startPosY + 35, 15, 15); g.drawOval(_startPosX + 50, _startPosY + 35, 15, 15); g.drawOval( _startPosX + 80, _startPosY + 35, 15, 15); g.drawOval(_startPosX + 110, _startPosY + 35, 15, 15); g.setColor(color); g.fillOval(_startPosX + 20, _startPosY + 35, 15, 15); g.fillOval(_startPosX + 50, _startPosY + 35, 15, 15); g.fillOval(_startPosX + 80, _startPosY + 35, 15, 15); g.fillOval(_startPosX + 110, _startPosY + 35, 15, 15); break; } } }