Code cleanup.

[javainspect.git] / src / main / java / eu / svjatoslav / inspector / java / methods / ProjectScanner.java

diff --git a/src/main/java/eu/svjatoslav/inspector/java/methods/ProjectScanner.java b/src/main/java/eu/svjatoslav/inspector/java/methods/ProjectScanner.java
index cf776c4..b90c0e0 100755 (executable)
--- a/src/main/java/eu/svjatoslav/inspector/java/methods/ProjectScanner.java
+++ b/src/main/java/eu/svjatoslav/inspector/java/methods/ProjectScanner.java
@@ -1,7 +1,7 @@
 /*
  * JavaInspect - Utility to visualize java software
  * Copyright (C) 2013-2017, Svjatoslav Agejenko, svjatoslav@svjatoslav.eu
- * 
+ *
  * This program is free software; you can redistribute it and/or
  * modify it under the terms of version 3 of the GNU Lesser General Public License
  * or later as published by the Free Software Foundation.
@@ -21,9 +21,9 @@ import java.util.Map;
 
 public class ProjectScanner {
 
+    public final List<JavaFile> javaFiles = new ArrayList<>();
     private final File scanPath;
-    public List<JavaFile> javaFiles = new ArrayList<JavaFile>();
-    Map<File, Project> projects = new HashMap<File, Project>();
+    Map<File, Project> projects = new HashMap<>();
 
     public ProjectScanner(final File projectPath) {
         scanPath = projectPath;
@@ -31,11 +31,10 @@ public class ProjectScanner {
     }
 
     public List<Clazz> getAllClasses() {
-        final List<Clazz> result = new ArrayList<Clazz>();
+        final List<Clazz> result = new ArrayList<>();
 
         for (final JavaFile file : javaFiles)
-            for (final Clazz clazz : file.classes)
-                result.add(clazz);
+            result.addAll(file.classes);
 
         return result;
     }