132 lines
4.1 KiB
Java
132 lines
4.1 KiB
Java
package laba1Loco;
|
||
|
||
import java.awt.Color;
|
||
import java.awt.Graphics2D;
|
||
import java.awt.image.BufferedImage;
|
||
|
||
public class TrainsGenericCollection<T extends DrawingTrain, U extends IMoveableObject>{
|
||
/// <summary>
|
||
/// Ширина окна прорисовки
|
||
/// </summary>
|
||
private int _pictureWidth;
|
||
/// <summary>
|
||
/// Высота окна прорисовки
|
||
/// </summary>
|
||
private int _pictureHeight;
|
||
/// <summary>
|
||
/// Размер занимаемого объектом места (ширина)
|
||
/// </summary>
|
||
private int _placeSizeWidth = 210;
|
||
/// <summary>
|
||
/// Размер занимаемого объектом места (высота)
|
||
/// </summary>
|
||
private int _placeSizeHeight = 90;
|
||
/// <summary>
|
||
/// Набор объектов
|
||
/// </summary>
|
||
private SetGeneric<T> _collection;
|
||
/// <summary>
|
||
/// Конструктор
|
||
/// </summary>
|
||
/// <param name="picWidth"></param>
|
||
/// <param name="picHeight"></param>
|
||
public TrainsGenericCollection(int picWidth, int picHeight)
|
||
{
|
||
int width = picWidth / _placeSizeWidth;
|
||
int height = picHeight / _placeSizeHeight;
|
||
_pictureWidth = picWidth;
|
||
_pictureHeight = picHeight;
|
||
_collection = new SetGeneric<T>(width * height);
|
||
}
|
||
/// <summary>
|
||
/// Перегрузка оператора сложения
|
||
/// </summary>
|
||
/// <param name="collect"></param>
|
||
/// <param name="obj"></param>
|
||
/// <returns></returns>
|
||
public int Add(T obj)
|
||
{
|
||
if (obj == null)
|
||
{
|
||
return -1;
|
||
}
|
||
return _collection.Insert(obj);
|
||
}
|
||
/// <summary>
|
||
/// Перегрузка оператора вычитания
|
||
/// </summary>
|
||
/// <param name="collect"></param>
|
||
/// <param name="pos"></param>
|
||
/// <returns></returns>
|
||
public T remove(int pos)
|
||
{
|
||
T obj = _collection.Get(pos);
|
||
if (obj != null)
|
||
{
|
||
_collection.Remove(pos);
|
||
}
|
||
return obj;
|
||
}
|
||
/// <summary>
|
||
/// Получение объекта IMoveableObject
|
||
/// </summary>
|
||
/// <param name="pos"></param>
|
||
/// <returns></returns>
|
||
public U GetU(int pos)
|
||
{
|
||
return (U)_collection.Get(pos).GetMoveableObject();
|
||
}
|
||
/// <summary>
|
||
/// Вывод всего набора объектов
|
||
/// </summary>
|
||
/// <returns></returns>
|
||
public BufferedImage ShowTrains()
|
||
{
|
||
BufferedImage bmp = new BufferedImage(_pictureWidth, _pictureHeight, BufferedImage.TYPE_INT_ARGB);
|
||
Graphics2D g = bmp.createGraphics();
|
||
DrawBackground(g);
|
||
DrawObjects(g);
|
||
g.dispose();
|
||
return bmp;
|
||
}
|
||
/// <summary>
|
||
/// Метод отрисовки фона
|
||
/// </summary>
|
||
/// <param name="g"></param>
|
||
private void DrawBackground(Graphics2D g)
|
||
{
|
||
g.setColor(Color.BLACK);
|
||
|
||
for (int i = 0; i < _pictureWidth / _placeSizeWidth; i++)
|
||
{
|
||
for (int j = 0; j < _pictureHeight / _placeSizeHeight + 1; ++j)
|
||
{//линия рамзетки места
|
||
g.drawLine( i * _placeSizeWidth, j * _placeSizeHeight, i * _placeSizeWidth + _placeSizeWidth / 2, j * _placeSizeHeight);
|
||
}
|
||
g.drawLine( i * _placeSizeWidth, 0, i * _placeSizeWidth, _pictureHeight / _placeSizeHeight * _placeSizeHeight);
|
||
}
|
||
}
|
||
/// <summary>
|
||
/// Метод прорисовки объектов
|
||
/// </summary>
|
||
/// <param name="g"></param>
|
||
private void DrawObjects(Graphics2D g)
|
||
{
|
||
int i = 0;
|
||
for (T train : _collection.GetTrains(100))
|
||
{
|
||
if (train != null)
|
||
{
|
||
train._pictureHeight = _pictureHeight;
|
||
train._pictureWidth = _pictureWidth;
|
||
train.SetPosition((i % (_pictureWidth / _placeSizeWidth)) * _placeSizeWidth, (i / (_pictureWidth / _placeSizeWidth)) * _placeSizeHeight);
|
||
if (train instanceof DrawingLoco)
|
||
((DrawingLoco)train).DrawTransport(g);
|
||
else
|
||
train.DrawTransport(g);
|
||
}
|
||
i++;
|
||
}
|
||
}
|
||
}
|