diff options
Diffstat (limited to '')
| -rw-r--r-- | src/main/java/design/model/statistics/CourseStats.java | 14 |
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); } } |
