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/HoleStats.java b/src/main/java/design/model/statistics/HoleStats.java
new file mode 100644
index 0000000..2c023b4
--- /dev/null
+++ b/src/main/java/design/model/statistics/HoleStats.java
@@ -0,0 +1,22 @@
+package design.model.statistics;
+
+import java.util.Arrays;
+
+import design.model.Hole;
+import design.model.Round;
+
+public class HoleStats extends StatisticsDecorator{
+    private Hole target_hole;
+
+    public HoleStats(Statistics wrapped_statistics, Hole target_hole){
+        super(wrapped_statistics);
+        this.target_hole = target_hole;
+    }
+
+    @Override
+    public Round[] getRounds(){
+        return Arrays.stream(super.getRounds())
+                        .filter(round -> round.getCourse().getHoles().stream().anyMatch(hole -> hole.equals(target_hole)))
+                        .toArray(Round[]::new);
+    }
+}