From 20775f7f260875b33a5eca3a29642f9b12ebc491 Mon Sep 17 00:00:00 2001
From: ekallin <elina_kallin@mail.ru>
Date: Tue, 19 Dec 2023 12:47:07 +0400
Subject: [PATCH] fix bugs, ready lab 6

---
 ProjectElectricLocomotive/LocomotiveGenericCollection.java | 4 ++++
 ProjectElectricLocomotive/LocomotivesGenericStorage.java   | 2 +-
 ProjectElectricLocomotive/SetGeneric.java                  | 3 +++
 3 files changed, 8 insertions(+), 1 deletion(-)

diff --git a/ProjectElectricLocomotive/LocomotiveGenericCollection.java b/ProjectElectricLocomotive/LocomotiveGenericCollection.java
index 25f45b4..e58fc66 100644
--- a/ProjectElectricLocomotive/LocomotiveGenericCollection.java
+++ b/ProjectElectricLocomotive/LocomotiveGenericCollection.java
@@ -91,4 +91,8 @@ public class LocomotiveGenericCollection<T extends DrawingLocomotive,U extends I
             i++;
         }
     }
+
+    public void clear() {
+        _collection.clear();
+    }
 }
diff --git a/ProjectElectricLocomotive/LocomotivesGenericStorage.java b/ProjectElectricLocomotive/LocomotivesGenericStorage.java
index da835a6..58cf697 100644
--- a/ProjectElectricLocomotive/LocomotivesGenericStorage.java
+++ b/ProjectElectricLocomotive/LocomotivesGenericStorage.java
@@ -96,7 +96,7 @@ public class LocomotivesGenericStorage {
             LocomotiveGenericCollection<DrawingLocomotive, DrawingObjectLocomotive> collection;
             if (_locomotiveStorage.containsKey(key)){
                 collection = _locomotiveStorage.get(key);
-//                collection.clear();  // what is the wtf???
+                collection.clear();
             }
             else
                 collection = new LocomotiveGenericCollection<>(_pictureWidth, _pictureHeight);
diff --git a/ProjectElectricLocomotive/SetGeneric.java b/ProjectElectricLocomotive/SetGeneric.java
index f731887..062252a 100644
--- a/ProjectElectricLocomotive/SetGeneric.java
+++ b/ProjectElectricLocomotive/SetGeneric.java
@@ -66,4 +66,7 @@ public class SetGeneric<T extends DrawingLocomotive>{
         return _places;
     }
 
+    public void clear() {
+        _places.clear();
+    }
 }
\ No newline at end of file