package pr03;

import java.util.*;

public class Franchise {
	
    private String name;
	private ArrayList<Restaurant> restaurants;
	private final int DISCOUNT_CAPACITY = 50;
	
	public Franchise(String name) {
		this.name = name;
		restaurants = new ArrayList<Restaurant>();
	}
	
	public int getNumRestaurants() {
		return restaurants.size();
	}
	
	public Restaurant getRestaurant(int number) {
		if ((number < 1) || (number > restaurants.size()))
	    	return null;
		return restaurants.get(number-1);
	}
	
	public Restaurant getRestaurant(String name) {
		for (int i=0;i<restaurants.size();i++)
			if (name.equals(restaurants.get(i).getName()))
				return restaurants.get(i);
		return null;
	}
	
	public void addRestaurant(Restaurant rest) {
		restaurants.add(rest);
	}
	
	public double getAverageEntreePrice() {
		double sum = 0;
		int numEntrees;
		int totalEntrees = 0;
		for (int i=0;i<restaurants.size();i++) {
			numEntrees = restaurants.get(i).getMenuSize();
			sum += numEntrees*restaurants.get(i).getAveragePrice();
			totalEntrees += numEntrees;
		}
		if (totalEntrees == 0)
			return 0;
		return sum/totalEntrees;
	}
	
	public double getAverageDiningInEntreePrice() {
		double sum = 0;
		int numEntrees;
		int totalEntrees = 0;
		for (int i=0;i<restaurants.size();i++) 
			if (restaurants.get(i) instanceof DiningIn ) {
		    	numEntrees = restaurants.get(i).getMenuSize();
			    sum += numEntrees*restaurants.get(i).getAveragePrice();
			    totalEntrees += numEntrees;
		    }
		if (totalEntrees == 0)
			return 0;
		return sum/totalEntrees;
	}
	
	public double getAverageTakeOutInEntreePrice() {
		double sum = 0;
		int numEntrees;
		int totalEntrees = 0;
		for (int i=0;i<restaurants.size();i++) 
			if (restaurants.get(i) instanceof TakeOut ) {
		    	numEntrees = restaurants.get(i).getMenuSize();
			    sum += numEntrees*restaurants.get(i).getAveragePrice();
			    totalEntrees += numEntrees;
		    }
		if (totalEntrees == 0)
			return 0;
		return sum/totalEntrees;
	}
	
	public double getAverageDiningInRestaurantEntreePrice() {
		double sum = 0;
		int count = 0;
		for (int i=0;i<restaurants.size();i++)
			if (restaurants.get(i) instanceof DiningIn) {
			   sum += restaurants.get(i).getAveragePrice();
			   count++;
		}
		if (count == 0)
			return 0;
		return sum/count;
		
	}
	
	public void printDiscountRestaurants() {
		double average = getAverageDiningInRestaurantEntreePrice();
		System.out.println("Discount restaurants:");
		for (int i=0;i<restaurants.size();i++)
			if ((restaurants.get(i) instanceof DiningIn) &&
				  (((DiningIn)(restaurants.get(i))).getCapacity() >= DISCOUNT_CAPACITY) &&
				  (restaurants.get(i).getAveragePrice() < average))
				System.out.println("  "+restaurants.get(i));
	}
	
	public void printLargestDiningInRestaurant() {
		int pos = -1;
		DiningIn biggest;
		for (int i=0;i<restaurants.size();i++)
			if (restaurants.get(i) instanceof DiningIn) {
				if (pos == -1)
					pos = i;
				else if ( ((DiningIn)(restaurants.get(i))).getCapacity() >
				          ((DiningIn)(restaurants.get(pos))).getCapacity())
					pos = i;
			}
		if (pos == -1)
			System.out.println("No Dining In Restaurants");
		else
		    System.out.println("Largest Dining In: "+restaurants.get(pos));
	}
	
	public void printDiningInByCapacity() {
		ArrayList<DiningIn> diningIn = new ArrayList<DiningIn>();
		for (int i=0;i<restaurants.size();i++)
			if (restaurants.get(i) instanceof DiningIn)
				diningIn.add((DiningIn)(restaurants.get(i)));
		Collections.sort(diningIn);
		System.out.println("Dining In Restaurants sorted by capacity:");
		for (int i=0;i<diningIn.size();i++)
			System.out.println("  " +diningIn.get(i));
	}
	
	public void printTakeOut() {
		System.out.println("Take Out Restaurants:");
		for (int i=0;i<restaurants.size();i++)
			if (restaurants.get(i) instanceof TakeOut)
				System.out.println("  "+restaurants.get(i));
	}
	
	public String toString() {
		StringBuffer sb = new StringBuffer();
		sb.append("Franchise "+name + ":\n");
		for (int i=0;i<restaurants.size();i++)
			sb.append("  "+restaurants.get(i) + "\n");
		return sb.toString();
	}

}