add check if null to ComponentSimulator

Siocnarff · Nov 5, 2020 · e7bd1f1 · e7bd1f1
1 parent e9605de
commit e7bd1f1
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/src/eng/buildings/ComponentSimulator.cpp b/src/eng/buildings/ComponentSimulator.cpp
@@ -3,9 +3,6 @@
 using namespace eng;
 
 void ComponentSimulator::testComponent(Component* component) {
-    if (!component) {
-        return;
-    }
 	int label = component->quality;
 	label += (rand()%5)-2;
 	if (label > 100) {
@@ -21,7 +18,9 @@ void ComponentSimulator::testComponent(Component* component) {
 
 void ComponentSimulator::testComponents(Car* car) {
 	for (Component* component : car->components) {
-		testComponent(component);
+		if (component) {
+		    testComponent(component);
+		}
 	}
 	pr::Doc::detail("Components of car ");
 	pr::Doc::detail(std::to_string(car->getId()));