source: xssim/trunk/src/test/rewolucja/resources/ResourceCharacteristics.java @ 104

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);
        }

}