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

Statistic dev

9 files changed, 161 insertions, 0 deletions

diff --git a/src/main/java/design/model/Hole.java b/src/main/java/design/model/Hole.java
index 23dc783..f79cc3d 100644
--- a/src/main/java/design/model/Hole.java
+++ b/src/main/java/design/model/Hole.java
@@ -18,5 +18,14 @@ public class Hole {
         return par;
     }
 
+    @Override
+    public boolean equals(Object obj){
+        if(!(obj instanceof Hole)){
+            return false;
+        }
+        Hole object = (Hole)obj;
+        return (object.getNumber() == this.getNumber() && object.getPar() == this.getPar());
+    }
+
     // TODO: Maybe add a get yardage method
 }
diff --git a/src/main/java/design/model/Round.java b/src/main/java/design/model/Round.java
index bc4914e..e4442d3 100644
--- a/src/main/java/design/model/Round.java
+++ b/src/main/java/design/model/Round.java
@@ -40,6 +40,12 @@ public class Round {
                 .reduce(0, Integer::sum);
     }
 
+    public double getTotalDistance() {
+        return plays.stream()
+                    .mapToDouble(design.model.Play::getDistance)
+                    .sum();
+    }
+
     public Course getCourse() {
         return course;
     }
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);
+    }
+}
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);
+    }
+}
diff --git a/src/main/java/design/model/statistics/LifetimeStats.java b/src/main/java/design/model/statistics/LifetimeStats.java
new file mode 100644
index 0000000..aee4e66
--- /dev/null
+++ b/src/main/java/design/model/statistics/LifetimeStats.java
@@ -0,0 +1,31 @@
+package design.model.statistics;
+
+import design.model.Golfer;
+import design.model.Round;
+
+public class LifetimeStats implements Statistics{
+    private final Golfer golfer;
+    public LifetimeStats(Golfer golfer){
+        this.golfer = golfer;
+    }
+    @Override
+    public Round[] getRounds(){
+        return golfer.getRounds();
+    }
+    @Override
+    public int get_score(){
+        int score = 0;
+        for(Round round : getRounds()){
+            score += round.getTotalSwings();
+        }
+        return score;
+    }
+    @Override
+    public double get_distance(){
+        double distance = 0;
+        for(Round round : getRounds()){
+            distance += round.getTotalDistance();
+        }
+        return distance;
+    }
+}
diff --git a/src/main/java/design/model/statistics/RoundStats.java b/src/main/java/design/model/statistics/RoundStats.java
new file mode 100644
index 0000000..598ca59
--- /dev/null
+++ b/src/main/java/design/model/statistics/RoundStats.java
@@ -0,0 +1,21 @@
+package design.model.statistics;
+
+import java.util.Arrays;
+
+import design.model.Round;
+
+public class RoundStats extends StatisticsDecorator{
+    private Round target_round;
+
+    public RoundStats(Statistics wrapped_statistics, Round target_round){
+        super(wrapped_statistics);
+        this.target_round = target_round;
+    }
+
+    @Override
+    public Round[] getRounds(){
+        return Arrays.stream(super.getRounds())
+                        .filter(round -> round == target_round)
+                        .toArray(Round[]::new);
+    }
+}
diff --git a/src/main/java/design/model/statistics/Statistics.java b/src/main/java/design/model/statistics/Statistics.java
new file mode 100644
index 0000000..53764a1
--- /dev/null
+++ b/src/main/java/design/model/statistics/Statistics.java
@@ -0,0 +1,7 @@
+package design.model.statistics;
+
+public interface Statistics {
+    public int get_score();
+    public double get_distance();
+    design.model.Round[] getRounds();
+}
diff --git a/src/main/java/design/model/statistics/StatisticsDecorator.java b/src/main/java/design/model/statistics/StatisticsDecorator.java
new file mode 100644
index 0000000..91ca452
--- /dev/null
+++ b/src/main/java/design/model/statistics/StatisticsDecorator.java
@@ -0,0 +1,22 @@
+package design.model.statistics;
+
+import design.model.Round;
+
+public abstract class StatisticsDecorator implements Statistics{
+    protected Statistics wrapped_statistics;
+    public StatisticsDecorator(Statistics wrapped_statistics){
+        this.wrapped_statistics = wrapped_statistics;
+    }
+    @Override
+    public Round[] getRounds() {
+        return wrapped_statistics.getRounds();
+    }
+    @Override
+    public int get_score(){
+        return wrapped_statistics.get_score();
+    }
+    @Override
+    public double get_distance(){
+        return wrapped_statistics.get_distance();
+    }
+}
diff --git a/src/main/java/design/model/statistics/YearlyStats.java b/src/main/java/design/model/statistics/YearlyStats.java
new file mode 100644
index 0000000..4c5d078
--- /dev/null
+++ b/src/main/java/design/model/statistics/YearlyStats.java
@@ -0,0 +1,21 @@
+package design.model.statistics;
+
+import java.util.Arrays;
+
+import design.model.Round;
+
+public class YearlyStats extends StatisticsDecorator{
+    private int year;
+
+    public YearlyStats(Statistics wrapped_statistics, int year){
+        super(wrapped_statistics);
+        this.year = year;
+    }
+
+    @Override
+    public Round[] getRounds(){
+        return Arrays.stream(super.getRounds())
+                        .filter(round -> round.getDateTime().getYear() == year)
+                        .toArray(Round[]::new);
+    }
+}