source: DCWoRMS/branches/coolemall/src/schedframe/scheduling/SchedulerDescription.java @ 1207

package schedframe.scheduling;

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

import schedframe.resources.StandardResourceType;
import schedframe.resources.computing.description.ExecutingResourceDescription;
import schedframe.resources.computing.description.ResourceDescription;
import schedframe.resources.providers.ResourceProvider;
import schedframe.resources.units.ResourceUnit;
import schedframe.resources.units.ResourceUnitName;
import schedframe.scheduling.queue.QueueDescription;

public class SchedulerDescription extends ResourceDescription implements ExecutingResourceDescription{

        protected ResourceProvider provider;
        protected List<QueueDescription> accesQueues;
        protected Map<ResourceUnitName, List<ResourceUnit>> resUnits;

        public SchedulerDescription(ResourceProvider provider){
                super(StandardResourceType.ResourceProvider);
                this.id = provider.getProviderId();
                this.provider = provider;
                this.resUnits = new HashMap<ResourceUnitName, List<ResourceUnit>>();
                this.accesQueues = new ArrayList<QueueDescription>(1);
        }
        
        public SchedulerDescription(String id){
                super(StandardResourceType.ResourceProvider);
                this.id = id;
                this.provider = new ResourceProvider(id, null, null, null);
                this.resUnits = new HashMap<ResourceUnitName, List<ResourceUnit>>();
                this.accesQueues = new ArrayList<QueueDescription>(1);
        }
        
        public void addQueueDescription(QueueDescription queue){
                accesQueues.add(queue);
        }
        
        public void addQueuesDescription(List<QueueDescription> queue){
                accesQueues.addAll(queue);
        }
        
        public List<QueueDescription> getAvailableQueues(){
                return this.accesQueues;
        }
        
        public void addResourceUnit(Map<ResourceUnitName, ResourceUnit> allUnits){
                Iterator<ResourceUnit> itr = allUnits.values().iterator();
                while(itr.hasNext()){
                        addResourceUnit(itr.next());
                }
        }
        
        public void addResourceUnitList(Map<ResourceUnitName, List<ResourceUnit>> allUnits){
                this.resUnits.putAll(allUnits);
        }

        public ResourceProvider getProvider() {
                return provider;
        }
        
        public void addResourceUnit(ResourceUnit unit) {
                if (this.resUnits == null)
                        this.resUnits = new HashMap<ResourceUnitName, List<ResourceUnit>>(1);
                List<ResourceUnit> list = null;
                if (this.resUnits.containsKey(unit.getName())) {
                        list = this.resUnits.get(unit.getName());
                } else {
                        list = new ArrayList<ResourceUnit>(1);
                        this.resUnits.put(unit.getName(), list);
                }
                list.add(unit);
        }
        
        public ResourceUnit getResourceUnit(ResourceUnitName unitName) throws NoSuchFieldException {
                return getResourceUnitList(unitName).get(0);
        }

        public List<ResourceUnit> getResourceUnitList(ResourceUnitName unitName) throws NoSuchFieldException {
                if (resUnits.containsKey(unitName))
                        return resUnits.get(unitName);
                else
                        throw new NoSuchFieldException("Resource unit " + unitName + " is not available in resource " + this.id);
        }

        public Collection<ResourceUnit> getResourceUnit() {
                if (resUnits == null)
                        return null;
                List<ResourceUnit> values = new ArrayList<ResourceUnit>();
                Collection<List<ResourceUnit>> lists = resUnits.values();
                Iterator<List<ResourceUnit>> itr = lists.iterator();

                while (itr.hasNext()) {
                        List<ResourceUnit> list = itr.next();
                        values.addAll(list);
                }

                return values;
        }

        public Map<ResourceUnitName, List<ResourceUnit>> getResourceUnits() {
                return resUnits;
        }
}