Implemented all decorators, replaced BaseStatistics with LifetimeStatistics since they would have been redundant

1 files changed, 7 insertions, 7 deletions

diff --git a/src/main/java/design/model/statistics/CourseStats.java b/src/main/java/design/model/statistics/CourseStats.java
index 47cc2c9..98d9529 100644
--- a/src/main/java/design/model/statistics/CourseStats.java
+++ b/src/main/java/design/model/statistics/CourseStats.java
@@ -1,6 +1,9 @@
 package design.model.statistics;
 
+import java.util.Arrays;
+
 import design.model.Course;
+import design.model.Round;
 
 public class CourseStats extends StatisticsDecorator{
     private Course course;
@@ -11,12 +14,9 @@ public class CourseStats extends StatisticsDecorator{
     }
 
     @Override
-    public int get_score(){
-        return super.get_score();
-    }
-
-    @Override
-    public double get_distance(){
-        return super.get_distance();
+    public Round[] getRounds(){
+        return Arrays.stream(super.getRounds())
+                        .filter(rounds -> (rounds.getCourse().getName().equals(course.getName()) && rounds.getCourse().getLocation().equals(course.getLocation())))
+                        .toArray(Round[]::new);
     }
 }