package poo;

import java.util.Iterator;
import java.util.SortedSet;
import java.util.TreeSet;

public class CompilationAlbumClass extends AbstractAlbumClass implements CompilationAlbum{

	private SortedSet<AlbumTrack> tracks;
	
	public CompilationAlbumClass(String code,String title,double price){
		super(code,title,price);
		tracks= new TreeSet<AlbumTrack>();
	}
	
	public void addLineItem(String composer,String music,String performer, double duration){
		AlbumTrack track = new AlbumTrackClass(composer,music,performer,duration);
		tracks.add(track);
	}
	
	public int getNumberOfTracks(){
		return tracks.size();
	}
	
	public double getDuration(){
		double totalDuration=0;
		for(int i=0;i<getNumberOfTracks();i++){
			AlbumTrack t = getTrack(i);
			totalDuration+=t.getDuration();
		}
		return totalDuration;
	}
	
	public AlbumTrack getTrack(int trackNumber){
		Iterator<AlbumTrack> it = this.trackIterator();
		for(int i=0;i<trackNumber;i++){
			it.next();
		}
		return it.next();
	}
	
	public Iterator<AlbumTrack> trackIterator(){
		return tracks.iterator();
	}
}