laba1Loco/SetGeneric.java

@@ -2,90 +2,120 @@ package laba1Loco;
 
 import java.util.ArrayList;
 import java.util.Iterator;
+import java.util.NoSuchElementException;
 
 public class SetGeneric<T extends Object> {
     /// <summary>
-        /// Массив объектов, которые храним
-        /// </summary>
-        private ArrayList<T>_places;
-        /// <summary>
-        /// Количество объектов в массиве
-        /// </summary>
-        public int Count () { return _places.size();};
-        /// <summary>
-        /// Максимальное количество объектов в списке
-        /// </summary>
-        private int _maxCount;
-        /// <summary>
-        /// Конструктор
-        /// </summary>
-        /// <param name="count"></param>
-        public SetGeneric(int count)
-        {
-            _maxCount = count;
-            _places = new ArrayList<T>(count);
-        }
-        /// <summary>
-        /// Добавление объекта в набор
-        /// </summary>
-        /// <param name="train">Добавляемый поезд</param>
-        /// <returns></returns>
-        public int Insert(T train)
-        {
-            if (_places.size() >= _maxCount)
-                return -1;
-            _places.add(0, train);
-            return 0;
-        }
-        /// <summary>
-        /// Добавление объекта в набор на конкретную позицию
-        /// </summary>
-        /// <param name="train">Добавляемый поезд</param>
-        /// <param name="position">Позиция</param>
-        /// <returns></returns>
-        public boolean Insert(T train, int position)
-        {
-            if (_places.size() >= _maxCount)
-                return false;
+    /// Массив объектов, которые храним
+    /// </summary>
+    private ArrayList<T>_places;
+    /// <summary>
+    /// Количество объектов в массиве
+    /// </summary>
+    public int Count () { return _places.size();};
+    /// <summary>
+    /// Максимальное количество объектов в списке
+    /// </summary>
+    private int _maxCount;
+    /// <summary>
+    /// Конструктор
+    /// </summary>
+    /// <param name="count"></param>
+    public SetGeneric(int count)
+    {
+        _maxCount = count;
+        _places = new ArrayList<T>(count);
+    }
+    /// <summary>
+    /// Добавление объекта в набор
+    /// </summary>
+    /// <param name="train">Добавляемый поезд</param>
+    /// <returns></returns>
+    public int Insert(T train)
+    {
+        if (_places.size() >= _maxCount)
+            return -1;
+        _places.add(0, train);
+        return 0;
+    }
+    /// <summary>
+    /// Добавление объекта в набор на конкретную позицию
+    /// </summary>
+    /// <param name="train">Добавляемый поезд</param>
+    /// <param name="position">Позиция</param>
+    /// <returns></returns>
+    public boolean Insert(T train, int position)
+    {
+        if (_places.size() >= _maxCount)
+            return false;
 
-            if (position < 0 || position > _places.size())
-                return false;
+        if (position < 0 || position > _places.size())
+            return false;
 
-            if (position == _places.size())
-                _places.add(train);
-            else
-                _places.add(position, train);
+        if (position == _places.size())
+            _places.add(train);
+        else
+            _places.add(position, train);
 
-            return true;
-        }
-        /// <summary>
-        /// Удаление объекта из набора с конкретной позиции
-        /// </summary>
-        /// <param name="position"></param>
-        /// <returns></returns>
-        public boolean Remove(int position)
-        {
-            if (position < 0 || position >= _places.size())
-                return false;
-            _places.remove(position);
-            return true;
-        }
-        /// <summary>
-        /// Получение объекта из набора по позиции
-        /// </summary>
-        /// <param name="position"></param>
-        /// <returns></returns>
-        public T Get(int position)
-        {
-            if (position < 0 || position >= _places.size())
-                return null;
-            return _places.get(position);
-        }
-        /// <summary>
-        /// Проход по списку
-        /// </summary>
-        /// <returns></returns>
-        public Iterator<T> iterator() {
-            return _places.iterator();
-        }
+        return true;
+    }
+    /// <summary>
+    /// Удаление объекта из набора с конкретной позиции
+    /// </summary>
+    /// <param name="position"></param>
+    /// <returns></returns>
+    public boolean Remove(int position)
+    {
+        if (position < 0 || position >= _places.size())
+            return false;
+        _places.remove(position);
+        return true;
+    }
+    /// <summary>
+    /// Получение объекта из набора по позиции
+    /// </summary>
+    /// <param name="position"></param>
+    /// <returns></returns>
+    public T Get(int position)
+    {
+        if (position < 0 || position >= _places.size())
+            return null;
+        return _places.get(position);
+    }
+
+    /// <summary>
+    /// Проход по списку
+    /// </summary>
+    /// <returns></returns>
+
+    public Iterable<T> GetTrains(final Integer maxTrains) {
+        return new Iterable<T>() {
+            @Override
+            public Iterator<T> iterator() {
+                return new Iterator<T>() {
+                    private int currentIndex = 0;
+                    private int count = 0;
+
+                    @Override
+                    public boolean hasNext() {
+                        return currentIndex < _places.size() && (maxTrains == null || count < maxTrains);
+                    }
+
+                    @Override
+                    public T next() {
+                        if (hasNext()) {
+                            count++;
+                            return _places.get(currentIndex++);
+                        }
+                        throw new NoSuchElementException();
+                    }
+
+                    @Override
+                    public void remove() {
+                        throw new UnsupportedOperationException();
+                    }
+                };
+            }
+        };
+    }
 }

laba1Loco/TrainsGenericCollection.java

@@ -112,17 +112,18 @@ public class TrainsGenericCollection<T extends DrawingTrain, U extends IMoveable
     /// <param name="g"></param>
     private void DrawObjects(Graphics2D g)
     {
-        for (int i = 0; i < _collection.Count(); i++)
+        int i = 0;
+        for (T train : _collection.GetTrains(100))
         {
-            T t = _collection.Get(i);
-            if (t != null)
+            if (train != null)
             {
-                t.SetPosition((i % (_pictureWidth / _placeSizeWidth)) * _placeSizeWidth, (i / (_pictureWidth / _placeSizeWidth)) * _placeSizeHeight);
-                if (t instanceof DrawingLoco)
-                    ((DrawingLoco) t).DrawTransport(g);
+                train.SetPosition((i % (_pictureWidth / _placeSizeWidth)) * _placeSizeWidth, (i / (_pictureWidth / _placeSizeWidth)) * _placeSizeHeight);
+                if (train instanceof DrawingLoco)
+                    ((DrawingLoco)train).DrawTransport(g);
                 else
-                    t.DrawTransport(g);
+                    train.DrawTransport(g);
             }
+            i++;
         }
     }
 }