ThirdRatings.java

import edu.duke.*;
import java.util.*;
import org.apache.commons.csv.*;

/**
 * Write a description of ThirdRatings here.
 *
 * @author (your name)
 * @version (a version number or a date)
 */
public class ThirdRatings {

    private ArrayList<Rater> myRaters;

    public ThirdRatings() {
        // default constructor
        this("ratings.csv");
    }

    public ThirdRatings(String ratingsfile){
        //create a FirsrRatings object
        FirstRatings fr = new FirstRatings();
        // call the loadRaters
        myRaters = fr.loadRaters(ratingsfile);
    }


    public int getRaterSize(){
        return myRaters.size();
    }

    private double getAverageByID(String id,int minimalRaters){
        double average =0;
        double total=0;
        int countRaters= 0;
        for(Rater rater:myRaters){
            if(rater.hasRating(id)){
                countRaters++;
                total = total + rater.getRating(id);
            }
        }

        if(countRaters >= minimalRaters){
            average = total/countRaters;
        }
        return average;

    }

    public ArrayList<Rating> getAverageRatings(int minimalRaters){
        ArrayList<Rating> avgRatingList = new ArrayList<Rating>();
        ArrayList<String> movies = MovieDatabase.filterBy(new TrueFilter());
        for(String id:movies){
            double avg = getAverageByID(id,minimalRaters);
            if(avg>0.0){
                Rating currRating = new Rating(id,avg);
                avgRatingList.add(currRating);
            }
        }

        return avgRatingList;
    }

    public ArrayList<Rating>  getAverageRatingsByFilter(int minimalRaters,Filter filterCriteria)
    {
        ArrayList<Rating> avgRatingListByFilter = new ArrayList<Rating>();
        ArrayList<String> movies = MovieDatabase.filterBy(filterCriteria);
        for(String id:movies){
            double avg = getAverageByID(id, minimalRaters);
            if(avg>0.0){
                Rating currRating = new Rating(id,avg);
                avgRatingListByFilter.add(currRating);
            }
        }
        return avgRatingListByFilter;

    }

}