/
SecondRatings.java
78 lines (65 loc) · 2.04 KB
/
SecondRatings.java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
/**
* Write a description of class SecondRatings here.
*
* @author (your name)
* @version (a version number or a date)
*/
import java.util.*;
public class SecondRatings {
private ArrayList<Movie> myMovies;
private ArrayList<EfficientRater> myRaters;
public SecondRatings() {
// default constructor
this("ratedmoviesfull.csv", "ratings.csv");
}
public SecondRatings(String moviefile, String ratingsfile) {
FirstRatings fr = new FirstRatings();
myMovies = fr.loadMovies(moviefile);
myRaters = fr.loadRaters(ratingsfile);
}
public int getMovieSize() {
return myMovies.size();
}
public int getRaterSize() {
return myRaters.size();
}
private double getAverageByID(String id, int minimalRaters){
double count = 0.0;
double sum = 0.0;
for(EfficientRater r : myRaters){
if(r.hasRating(id)) {
sum += r.getRating(id);
count ++;
}
}
return (count >= minimalRaters) ? (double)(sum/count) : 0.0;
}
public ArrayList<Rating> getAverageRatings( int minimalRaters) {
ArrayList<Rating> averageRatings = new ArrayList<Rating>();
for (Movie m : myMovies) {
String id = m.getID();
double rating = getAverageByID(id, minimalRaters);
Rating movieAvgRate = new Rating(id, rating);
if(rating != 0){
averageRatings.add(movieAvgRate);
}
}
return averageRatings;
}
public String getTitle(String id){
for (Movie m : myMovies){
if(m.getID().equals(id)){
return m.getTitle();
}
}
return "Movie with ID " + id + " not found";
}
public String getId(String movieTitle){
for (Movie m : myMovies){
if(m.getTitle().equals(movieTitle)){
return m.getID();
}
}
return "NO SUCH TITLE";
}
}