2023-11-06 20:57:19 +04:00

66 lines
2.1 KiB
Java

package ProjectElectricLocomotive;
import java.util.ArrayList;
import java.util.Random;
public class DopClassParameters<T extends EntityLocomotive, U extends IDrawingWheels> {
public T[] _entityLocomotive;
public U[] _idrawningWheels;
int _locomotivesCount;
int _idrawningWheelsCount;
private int _pictureWidth;
private int _pictureHeight;
public DopClassParameters(int count, int width, int height) {
_entityLocomotive = (T[]) new EntityLocomotive[count];
_idrawningWheels = (U[]) new IDrawingWheels[count];
_pictureWidth = width;
_pictureHeight = height;
_locomotivesCount = 0;
_idrawningWheelsCount = 0;
}
public void Add(T entityLocoObj) {
if (_locomotivesCount < _entityLocomotive.length) {
_entityLocomotive[_locomotivesCount] = entityLocoObj;
_locomotivesCount++;
}
}
public void Add(U idrawingWheels) {
if (_idrawningWheelsCount < _idrawningWheels.length) {
_idrawningWheels[_idrawningWheelsCount] = idrawingWheels;
_idrawningWheelsCount++;
}
}
public DrawingLocomotive RandomLocomotive(int pictureWidth, int pictureHeight) {
Random rnd = new Random();
if (_entityLocomotive == null || _idrawningWheels == null) {
return null;
}
T entityLocomotive = _entityLocomotive[rnd.nextInt(_locomotivesCount)];
U idrawingWheels = _idrawningWheels[rnd.nextInt(_idrawningWheelsCount)];
DrawingLocomotive drawLocomotive;
if (entityLocomotive instanceof EntityElectricLocomotive) {
drawLocomotive = new DrawingElectricLocomotive(
(EntityElectricLocomotive) entityLocomotive,
idrawingWheels,
pictureWidth,
pictureHeight
);
} else {
drawLocomotive = new DrawingLocomotive(
entityLocomotive,
idrawingWheels,
pictureWidth,
pictureHeight
);
}
return drawLocomotive;
}
}