Merge pull request #4 from RIT-SWEN-262/statistic-dev

Statistic dev

1 files changed, 22 insertions, 0 deletions

diff --git a/src/main/java/design/model/statistics/CourseStats.java b/src/main/java/design/model/statistics/CourseStats.java
new file mode 100644
index 0000000..98d9529
--- /dev/null
+++ b/src/main/java/design/model/statistics/CourseStats.java
@@ -0,0 +1,22 @@
+package design.model.statistics;
+
+import java.util.Arrays;
+
+import design.model.Course;
+import design.model.Round;
+
+public class CourseStats extends StatisticsDecorator{
+    private Course course;
+
+    public CourseStats(Statistics wrapped_statistics, Course course){
+        super(wrapped_statistics);
+        this.course = course;
+    }
+
+    @Override
+    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);
+    }
+}