Staff.java

import java.util.ArrayList;
import java.util.Random;

public class Staff {
	String name;
	ArrayList<Integer> occSlots = new ArrayList<Integer>();
	ArrayList<Integer> daysOff = new ArrayList<Integer>();
	ArrayList<Integer> preferences = new ArrayList<Integer>();
	
	public Staff( String name ) {
		this.name = name;
	}
	
	public Staff( String name, ArrayList<Integer> daysOff) {
		this.name = name;
		this.daysOff.addAll(daysOff);
	}
	
	public void addSlot( int slot ) {
		if( !occSlots.contains(slot) ) 
			occSlots.add(slot);
	}
	
	public void genDaysOff() {
		Random random = new Random();
		int dayOff = random.nextInt(3);
		boolean found = false;
		int count = 0;
		
		while( !found && count < 6 ) {
			int i = 0;
			while( i < occSlots.size() ) {
				if( (occSlots.get(i) / 5) == dayOff)
					break;
				i++;
			}
			if( i == occSlots.size() ) {
				found = true;
				daysOff.add(dayOff);
			}
			dayOff = (dayOff + 1) % 6;
			count++;
		}
		
	}
	
	public void genPreference() {
		int preference = 0;
		boolean found = false;
		
		while( !found && preference < 6 ) {
			int i = 0;
			while( i < occSlots.size() ) {
				if( (occSlots.get(i) / 5) == preference || daysOff.contains(preference) )
					break;
				i++;
			}
			if( i == occSlots.size() ) {
				found = true;
				preferences.add(preference);
			}
			preference++;
		}
	}
	
	public String toString() {
		String fact = "staff('";
		fact += name + "',[";
		
		for( int i = 0; i < occSlots.size(); i++ ) {
			fact += occSlots.get(i) + ",";
		}
		
		fact = fact.substring(0,fact.length()-1);
		fact += "]";
		
		if( daysOff.size() > 0 ) {
			fact += ",[";
			for( int i = 0; i < daysOff.size(); i++ ) {
				fact += daysOff.get(i) + ",";
			}
			
			fact = fact.substring(0,fact.length()-1);
			fact += "]";
		} else {
			fact += ",[]";
		}
		
		if( preferences.size() > 0 ) {
			fact += ",[";
			for( int i = 0; i < preferences.size(); i++ ) {
				fact += preferences.get(i) + ",";
			}
			
			fact = fact.substring(0,fact.length()-1);
			fact += "]";
		} else {
			fact += ",[]";
		}
		
		fact +=	").\n";
		
		return fact;
	}

}