Week3. Additional Starter Files for Assignment 2

Author: SkyrocketStan

Week3/src/AllFilters.java

@@ -0,0 +1,24 @@
+import java.util.ArrayList;
+
+public class AllFilters implements Filter {
+  ArrayList<Filter> filters;
+
+  public AllFilters() {
+    filters = new ArrayList<>();
+  }
+
+  public void addFilter(Filter f) {
+    filters.add(f);
+  }
+
+  @Override
+  public boolean satisfies(String id) {
+    for (Filter f : filters) {
+      if (!f.satisfies(id)) {
+        return false;
+      }
+    }
+
+    return true;
+  }
+}

Week3/src/Filter.java

@@ -0,0 +1,3 @@
+public interface Filter {
+  boolean satisfies(String id);
+}

Week3/src/FirstRatings.java

@@ -105,7 +105,7 @@ private HashMap<String, HashSet<String>> getDirectorsAndMovies() {
   }
 
   // Process every record from the CSV file.
-  private ArrayList<Movie> loadMovies(String filename) {
+  ArrayList<Movie> loadMovies(String filename) {
 
     // Create empty list of Movie's
     ArrayList<Movie> moviesList = new ArrayList<>();

Week3/src/MovieDatabase.java

@@ -0,0 +1,89 @@
+import java.util.ArrayList;
+import java.util.HashMap;
+
+public class MovieDatabase {
+  private static HashMap<String, Movie> ourMovies;
+
+  public static void initialize(String moviefile) {
+    if (ourMovies == null) {
+      ourMovies = new HashMap<>();
+      loadMovies("data/" + moviefile);
+    }
+  }
+
+  private static void initialize() {
+    if (ourMovies == null) {
+      ourMovies = new HashMap<>();
+      loadMovies("data/ratedmoviesfull.csv");
+    }
+  }
+
+  private static void loadMovies(String filename) {
+    FirstRatings fr = new FirstRatings();
+    ArrayList<Movie> list = fr.loadMovies(filename);
+    for (Movie m : list) {
+      ourMovies.put(m.getID(), m);
+    }
+  }
+
+  public static boolean containsID(String id) {
+    initialize();
+    return ourMovies.containsKey(id);
+  }
+
+  public static int getYear(String id) {
+    initialize();
+    return ourMovies.get(id).getYear();
+  }
+
+  public static String getGenres(String id) {
+    initialize();
+    return ourMovies.get(id).getGenres();
+  }
+
+  public static String getTitle(String id) {
+    initialize();
+    return ourMovies.get(id).getTitle();
+  }
+
+  public static Movie getMovie(String id) {
+    initialize();
+    return ourMovies.get(id);
+  }
+
+  public static String getPoster(String id) {
+    initialize();
+    return ourMovies.get(id).getPoster();
+  }
+
+  public static int getMinutes(String id) {
+    initialize();
+    return ourMovies.get(id).getMinutes();
+  }
+
+  public static String getCountry(String id) {
+    initialize();
+    return ourMovies.get(id).getCountry();
+  }
+
+  public static String getDirector(String id) {
+    initialize();
+    return ourMovies.get(id).getDirector();
+  }
+
+  public static int size() {
+    return ourMovies.size();
+  }
+
+  public static ArrayList<String> filterBy(Filter f) {
+    initialize();
+    ArrayList<String> list = new ArrayList<>();
+    for (String id : ourMovies.keySet()) {
+      if (f.satisfies(id)) {
+        list.add(id);
+      }
+    }
+
+    return list;
+  }
+}

Week3/src/TrueFilter.java

@@ -0,0 +1,6 @@
+public class TrueFilter implements Filter {
+  @Override
+  public boolean satisfies(String id) {
+    return true;
+  }
+}

Week3/src/YearAfterFilter.java

@@ -0,0 +1,12 @@
+public class YearAfterFilter implements Filter {
+  private final int myYear;
+
+  public YearAfterFilter(int year) {
+    myYear = year;
+  }
+
+  @Override
+  public boolean satisfies(String id) {
+    return MovieDatabase.getYear(id) >= myYear;
+  }
+}