PIbd-23_Mochalov_D.V._Locom.../ExtraRoundWheelDraw.java

58 lines
1.6 KiB
Java

import java.awt.*;
public class ExtraRoundWheelDraw implements IDrawningExtra{
private WheelsCount wheelsCount = WheelsCount.Two;
private ExtraWheelsDraw extraWheelsDraw;
private Color color;
public void setExtraNum(int num) {
switch (num) {
case 3: {
wheelsCount = WheelsCount.Three;
break;
}
case 4: {
wheelsCount = WheelsCount.Four;
break;
}
default:
break;
}
}
public String TypeString() {
return "Round";
}
public int getWheelsCount() {
if (wheelsCount == WheelsCount.Two) return 2;
if (wheelsCount == WheelsCount.Three) return 3;
else return 4;
}
public ExtraRoundWheelDraw (int num, Color bodyColor) {
setExtraNum(num);
extraWheelsDraw = new ExtraWheelsDraw(num, bodyColor);
}
public void SetColor(Color color) {
this.color = color;
}
public void DrawExtra(int startPosX, int startPosY, Graphics2D g) {
extraWheelsDraw.DrawExtra(startPosX, startPosY, g);
g.setColor(color);
g.fillOval(startPosX + 5, startPosY + 35, 10, 10);
g.fillOval(startPosX + 95, startPosY + 35, 10, 10);
switch (wheelsCount) {
case Four: {
g.fillOval(startPosX + 75, startPosY + 35, 10, 10);
}
case Three: {
g.fillOval(startPosX + 25, startPosY + 35, 10, 10);
break;
}
default:
break;
}
}
}