package test.rewolucja.resources;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Properties;

import schedframe.resources.units.ResourceUnit;
import schedframe.scheduling.utils.ResourceParameterName;


public class ResourceCharacteristics extends Properties{

	private static final long serialVersionUID = 2719535186621622647L;

	protected Map<ResourceParameterName, List<ResourceUnit>> resUnits;

	public ResourceCharacteristics(Map<ResourceParameterName, List<ResourceUnit>> resUnits){
		this.resUnits = resUnits;
	}

	public ResourceCharacteristics(){
		this.resUnits = null;
	}
	
	public Map<ResourceParameterName, List<ResourceUnit>> getResourceUnits() {
		return resUnits;
	}
	
	public ResourceUnit getResourceUnit(ResourceParameterName unitName){
		return getResourceUnits().get(unitName).get(0);
	}
	
	public void addResourceUnit(ResourceUnit unit){ 
		if(resUnits == null){
			resUnits = new HashMap<ResourceParameterName, List<ResourceUnit>>(2);
		}
		List<ResourceUnit> list = null;
		if(resUnits.containsKey(unit.getName())){
			list = resUnits.get(unit.getName());
		} else {
			list = new ArrayList<ResourceUnit>(1);
			resUnits.put(unit.getName(), list);
		}
		list.add(unit);
	}

}