Week3. printAverageRatingsByYear

Author: SkyrocketStan

Week3/src/MovieRunnerWithFilters.java

@@ -3,13 +3,24 @@
 
 public class MovieRunnerWithFilters {
 
+  private final ThirdRatings thirdRatings;
+
+  public MovieRunnerWithFilters(String moviesFileName, String ratingsFileName) {
+    this.thirdRatings = new ThirdRatings(moviesFileName, ratingsFileName);
+  }
+
+  public MovieRunnerWithFilters() {
+    this("ratedmovies_short.csv", "ratings_short.csv");
+  }
+
   /**
    * Print a list of movies and their average ratings sorted by averages
    *
    * @param minimalRatings int specified number of ratings
    */
-  public static void printAverageRatings(int minimalRatings) {
-    ThirdRatings thirdRatings = new ThirdRatings("ratings_short.csv");
+  public void printAverageRatings(int minimalRatings) {
+    //    ThirdRatings thirdRatings = new ThirdRatings("ratedmovies_short.csv",
+    // "ratings_short.csv");
     ArrayList<Rating> ratedList = thirdRatings.getAverageRatings(minimalRatings);
 
     Collections.sort(ratedList);
@@ -31,23 +42,30 @@ public static void printAverageRatings(int minimalRatings) {
     // Print out how many movies with ratings are returned,
     // then sort them, and print out the rating and title of each movie
     System.out.printf(
-        "How many movies with ratings %d are returned: %d\n",
+        "How many movies with ratings %d are returned: %d%n",
         minimalRatings, averageRatings.size());
 
+    printRatingsList(averageRatings);
+  }
+
+  private void printRatingsList(ArrayList<Rating> averageRatings) {
     averageRatings.stream()
         .sorted()
         .forEach(
             rating ->
                 System.out.printf(
-                    "%-4s %s%n", rating.getValue(), MovieDatabase.getTitle(rating.getItem())));
-    //    System.out.printf(
-    //            "The name of the movie that has the lowest rating is \"%s\"\n",
-    //            thirdRatings.getTitle(ratedList.get(0).getItem()));
-    //    for (Rating ratedObj : ratedList) {
-    //      double rating = ratedObj.getValue();
-    //      String movieID = ratedObj.getItem();
-    //      String movieTitle = secondRatings.getTitle(movieID);
-    //      System.out.println(rating + " " + movieTitle);
-    //    }
+                    "%-4s %s %s%n",
+                    rating.getValue(),
+                    MovieDatabase.getYear(rating.getItem()),
+                    MovieDatabase.getTitle(rating.getItem())));
+  }
+
+  public void printAverageRatingsByYear(int minimalRatings, int year) {
+    System.out.println("number of raters " + thirdRatings.getRaterSize());
+    System.out.println("number of movies " + MovieDatabase.size());
+    ArrayList<Rating> aveRating =
+        thirdRatings.getAverageRatingsByFilter(minimalRatings, new YearAfterFilter(year));
+    System.out.printf("found %d movies%n", aveRating.size());
+    printRatingsList(aveRating);
   }
 }

Week3/src/ThirdRatings.java

@@ -1,17 +1,15 @@
 import java.util.ArrayList;
+import java.util.Collections;
 
 public class ThirdRatings {
 
   private final ArrayList<Rater> myRaters;
   private final ArrayList<String> allMoviesIDs;
 
-  public ThirdRatings() {
-    this("ratings.csv");
-  }
-
-  public ThirdRatings(String ratingsFileName) {
+  public ThirdRatings(String moviesFileName, String ratingsFileName) {
     this.myRaters = FirstRatings.loadRaters(ratingsFileName);
-    this.allMoviesIDs = MovieDatabase.filterBy(new TrueFilter());
+    MovieDatabase.initialize(moviesFileName);
+    allMoviesIDs = MovieDatabase.filterBy(new TrueFilter());
   }
 
   public ArrayList<Rating> getAverageRatings(int minimalRaters) {
@@ -42,4 +40,25 @@ Double getAverageByID(String movieID, Integer minimalRaters) {
     }
     return 0.0;
   }
+
+  public int getRaterSize() {
+    return myRaters.size();
+  }
+
+  public ArrayList<Rating> getAverageRatingsByFilter(Integer minimalRaters, Filter filterCriteria) {
+    ArrayList<Rating> rating = new ArrayList<>();
+    ArrayList<String> movies = MovieDatabase.filterBy(filterCriteria);
+    // System.out.println(movies);
+    Rating rat;
+    for (String movie_id : movies) {
+      if (getAverageByID(movie_id, minimalRaters) != 0) {
+        rat = new Rating(movie_id, getAverageByID(movie_id, minimalRaters));
+        rating.add(rat);
+      }
+    }
+
+    Collections.sort(rating);
+    // System.out.println(rating);
+    return rating;
+  }
 } // class

Week3/src/Week3.java

@@ -3,6 +3,12 @@ public static void main(String[] args) {
     //    ThirdRatings week3 = new ThirdRatings();
     //    ArrayList<Rating> averageRatings = week3.getAverageRatings(1);
     //    averageRatings.forEach(System.out::println);
-    MovieRunnerWithFilters.printAverageRatings(1);
+    MovieRunnerWithFilters filters = new MovieRunnerWithFilters();
+    filters.printAverageRatingsByYear(1, 2000);
+    System.out.println("---");
+
+    //    MovieRunnerWithFilters filtersFull =
+    //            new MovieRunnerWithFilters("ratedmoviesfull.csv", "ratings.csv");
+    //    filtersFull.printAverageRatingsByYear(20, 2000);
   }
 }