order.java

SSD3 Exercise Four Ex 4

import java.util.*;

public class Order implements Iterable<OrderItem>{
	
	private ArrayList<OrderItem> items;

	public Order()
	{
		items = new ArrayList<OrderItem>();
	}
	
	public void addItem(OrderItem orderItem)
	{
		items.add(orderItem);
	}
	
	public void removeItem(OrderItem orderItem)
	{
		items.remove(orderItem);
	}
	
	public Iterator<OrderItem> iterator()
	{
		return items.iterator();
	}
	
	public OrderItem getItem(Product product)
	{
		for(Iterator<OrderItem> itr = items.iterator(); itr.hasNext();)
		{   
			
			OrderItem orderItem = itr.next();   
			
			if(product.getCode().equals(product))   
				return orderItem;   
		}  
		
		return null;
	}
	
	public int getNumberOfItems()
	{
		return items.size();
	}
	
	public double getTotalCost()
	{
		double total = 0.0D;   
	
		for(Iterator<OrderItem> itr = items.iterator(); itr.hasNext();){   
			total += itr.next().getValue();   
		}   
	
		return total;
	}
	
}