source: xssim/trunk/src/schedframe/resources/ResourceStateDescription.java @ 217

package schedframe.resources;

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

import schedframe.resources.units.ResourceUnit;
import schedframe.scheduling.utils.ResourceParameterName;
import test.rewolucja.resources.physical.base.ComputingResource;

/**
 * 
 * @author Marcin Krystek
 *
 */
public class ResourceStateDescription extends ResourceDescription {

        protected ResourceType type;
        
        protected Map<String, Integer> queue_size;
        
        protected List<ComputingResource> compResources;

        public ResourceStateDescription(ResourceProvider provider){
                super();
                this.provider = provider;
                this.queue_size = new HashMap<String, Integer>();
                this.compResources = new ArrayList<ComputingResource>();
        }
        
        public ResourceStateDescription(ResourceDescription r){
                super(r);
        }
        
        public void addResourceUnit(ResourceUnit unit){
                List<ResourceUnit> list = null;
                
                if(this.resUnits.containsKey(unit.getName())){
                        list = this.resUnits.get(unit.getName());
                } else {
                        list = new ArrayList<ResourceUnit>();
                        this.resUnits.put(unit.getName(), list);
                }
                list.add(unit);
        }
        
        public void addResourceUnit(Map<ResourceParameterName, ResourceUnit> allUnits){
                Iterator<ResourceUnit> itr = allUnits.values().iterator();
                while(itr.hasNext()){
                        addResourceUnit(itr.next());
                }
        }
        
        public void addResourceUnitList(Map<ResourceParameterName, List<ResourceUnit>> allUnits){
                this.resUnits.putAll(allUnits);
        }
        
        public void setType(ResourceType type){
                this.type = type;
        }
        
        public ResourceType getType() {
                return type;
        }
        
        public void setQueuesSize(Map<String, Integer> queue_size){
                this.queue_size = queue_size;
        }
        
        public long getQueueLoad(String queueName) throws NoSuchFieldException{
                long load = 0;
                if(queueName == null){
                        for(String queue: queue_size.keySet()){
                                load += queue_size.get(queue);
                        }
                        return load;
                }
                else if(queue_size.containsKey(queueName))
                        return queue_size.get(queueName);
                else
                        throw new NoSuchFieldException("Queue " + queueName + 
                                                        " is not available in resource " + provider.id);
        }
        
        public void addComputingResources(List<ComputingResource> compRes){
                compResources.addAll(compRes);
        }
}