source: DCWoRMS/branches/coolemall/src/schedframe/resources/computing/ComputingResource.java @ 1207

package schedframe.resources.computing;

import gridsim.GridSimTags;
import gridsim.dcworms.DCWormsTags;

import java.util.ArrayList;
import java.util.LinkedList;
import java.util.List;
import java.util.Properties;

import schedframe.events.Event;
import schedframe.events.EventHandler;
import schedframe.events.ResourceEventCommand;
import schedframe.events.scheduling.EventReason;
import schedframe.events.scheduling.SchedulingEvent;
import schedframe.events.scheduling.SchedulingEventCommand;
import schedframe.exceptions.ResourceException;
import schedframe.resources.ResourceStatus;
import schedframe.resources.ResourceType;
import schedframe.resources.computing.description.ComputingResourceDescription;
import schedframe.resources.computing.extensions.Extension;
import schedframe.resources.computing.extensions.ExtensionListImpl;
import schedframe.resources.computing.profiles.energy.EnergyEvent;
import schedframe.resources.computing.profiles.energy.EnergyEventType;
import schedframe.resources.computing.profiles.energy.EnergyExtension;
import schedframe.resources.computing.properties.DefaultPropertiesBuilder;
import schedframe.resources.computing.properties.PropertiesDirector;
import schedframe.resources.computing.validator.ResourceNameValidator;
import schedframe.resources.computing.validator.ResourcePropertiesValidator;
import schedframe.resources.computing.validator.ResourceStatusValidator;
import schedframe.resources.computing.validator.ResourceTypeValidator;
import schedframe.resources.computing.validator.ResourceValidator;
import schedframe.resources.devices.Device;
import schedframe.resources.devices.PhysicalResource;
import schedframe.scheduling.Scheduler;

public class ComputingResource extends PhysicalResource{

        protected ComputingResource parent;
        protected List<ComputingResource> children;
        
        protected Scheduler scheduler;
        //protected ResourceCharacteristics resourceCharacteristic;


        public ComputingResource(ComputingResourceDescription resDesc) {
                this.type = resDesc.getType();
                this.name = resDesc.getId();
                this.category = resDesc.getCategory();
                this.status = ResourceStatus.FREE;
                this.extensionList = new ExtensionListImpl(1);
                initCharacteristics(resDesc);
                accept(new EnergyExtension.Builder().resource(this).powerProfile(resDesc.getPowerProfile()).airFlowProfile(resDesc.getAirThroughputProfile()).thermalProfile(resDesc.getThermalProfile()).build());     
        }

        protected void initCharacteristics(ComputingResourceDescription resDesc){
                resourceCharacteristic = ComputingResourceCharacteristics.builder().resourceUnits(resDesc.getResourceUnits()).location(resDesc.getLocation()).parameters(resDesc.getParameters()).device(resDesc.getDevices()).build();
                for(Device device: ((ComputingResourceCharacteristics)resourceCharacteristic).getDevices()){
                        device.setComputingResource(this);
                }
        }
        
        public ComputingResource getParent() {
                return parent;
        }

        public void setParent(ComputingResource newParent) {
                parent = newParent;
        }

        public List<ComputingResource> getChildren() {
                if (children == null)
                        return new ArrayList<ComputingResource>(1);
                return children;
        }

        public void addChild(ComputingResource child) {
                child.setParent(this);
                if (children == null)
                        children = new ArrayList<ComputingResource>();
                children.add(child);
        }

        public String getFullName() {
                if(this.getResourceCharacteristic().getParameters().get("fullPath") != null){
                        String fullPath = this.getResourceCharacteristic().getParameters().get("fullPath").get(0).getContent();
                        return fullPath;
                }
                if(parent!= null){
                        return parent.getFullName() + "/" + name;
                } else 
                        return name;
        }

        public ComputingResourceCharacteristics getResourceCharacteristic() {
                return (ComputingResourceCharacteristics)resourceCharacteristic;
        }

        public void setStatus(ResourceStatus newStatus) {
                if(newStatus != status){
                        status = newStatus;
                        if(children != null) {
                                for(ComputingResource child: children){
                                        child.setStatus(status);
                                }
                        }       
                }
        }
        
        private void triggerEventUp(Event event) {
                if(parent != null)
                        parent.handleEvent(event);
        }

        public void handleEvent(Event event){
                ResourceEventCommand rec = new ResourceEventCommand(this);
                rec.execute(event);
                SchedulingEventCommand sec = new SchedulingEventCommand(this);
                sec.execute(event);

                //old, correctly working method
                /*if (extensionList != null) {
                        for (Extension extension : extensionList) {
                                if (extension.supportsEvent(event)) {
                                        extension.handleEvent(event);
                                }
                        }
                }*/
                //TODO - delete, check in advance
                //if(scheduler != null && (parent != null && scheduler != parent.getScheduler())/*scheduler.getResources().contains(this)*/){
                //      String src = event.getSource() != null ? event.getSource() : name;
                //      scheduler.sendInternal(GridSimTags.SCHEDULE_NOW, DCWormsTags.UPDATE, src);
                //}
                //triggerEventUp(event);
        }
        
        public List <? extends ComputingResource> getDescendantsByType(ResourceType type) {
                List<ResourceValidator> validators = new ArrayList<ResourceValidator>();
                validators.add(new ResourceTypeValidator(type));
                return searchDescendants(validators, true);
        }

        public List<? extends ComputingResource> getDescendantsByTypeAndStatus(ResourceType type, ResourceStatus status) {
                List<ResourceValidator> validators = new ArrayList<ResourceValidator>();
                validators.add(new ResourceStatusValidator(status));
                validators.add(new ResourceTypeValidator(type));
                return searchDescendants(validators, true);
        }

        public ComputingResource getDescendantByName(String resourceName) throws ResourceException {
                List<ResourceValidator> validators = new ArrayList<ResourceValidator>();
                validators.add(new ResourceNameValidator(resourceName));
                List<? extends ComputingResource> resources = searchDescendants(validators, true);
                return resources.size() == 0 ? null : resources.get(0);
        }
        
        public List<? extends ComputingResource> filterDescendants(Properties properties)  {
                List<ResourceValidator> validators = new ArrayList<ResourceValidator>();
                validators.add(new ResourcePropertiesValidator(properties));
                return searchDescendants(validators, false);
        }
        
        protected List<? extends ComputingResource> searchDescendants(List<ResourceValidator> validators, boolean cutOff) {

                List<ComputingResource> descendants = new ArrayList<ComputingResource>();
                if (children != null) {
                        LinkedList<ComputingResource> toExamine = new LinkedList<ComputingResource>();
                        toExamine.push(this);

                        while (!toExamine.isEmpty()) {
                                ComputingResource resource = toExamine.pop();
                                List<ComputingResource> resources = resource.getChildren();
                                int numberOfRes = resources.size();
                                for (int i = 0; i < numberOfRes; i++) {
                                        ComputingResource resourceChild = resources.get(i);
                                        if (resourceChild.match(validators)) {
                                                descendants.add(resourceChild);
                                                if(cutOff == false) {
                                                        toExamine.addLast(resourceChild);
                                                }
                                        } else
                                                //toExamine.insertElementAt(resourceChild, 0);
                                                toExamine.addLast(resourceChild);
                                }
                        }
                }
                return descendants;
        }
        
        protected boolean match(List<ResourceValidator> validators){
                for(ResourceValidator validator: validators){
                        if(validator.validate(this) == false)
                                return false;
                }
                return true;
        }
        
        public Properties getProperties(){
                PropertiesDirector propDirector = new PropertiesDirector();
                propDirector.setPropertiesBuilder(new DefaultPropertiesBuilder());
                propDirector.constructProperties(this);
                return propDirector.getProperties();
        }
        
        public Scheduler getScheduler() {
                return scheduler;
        }
        
        public void setScheduler(Scheduler scheduler) {
                this.scheduler = scheduler;
                if(children != null){
                        for(ComputingResource child: children){
                                child.setScheduler(scheduler);
                        }       
                }
        }

        class ComputingResourceEventHandler implements EventHandler{
                
                public void handleResourceEvent(Event event){
                        if (extensionList != null) {
                                for (Extension extension : extensionList) {
                                        if (extension.supportsEvent(event)) {
                                                extension.handleEvent(event);
                                        }
                                }
                                
                                for (Device device: ((ComputingResourceCharacteristics)resourceCharacteristic).getDevices()) {
                                        for (Extension extension : device.getExtensionList()) {
                                                if (extension.supportsEvent(event)) {
                                                        //extension.handleEvent(event);
                                                }
                                        }
                                }
                        }
                        if(event.getReason() != EventReason.SIM_INIT)
                                triggerEventUp(event);
                }
                
                public void handleSchedulingEvent(SchedulingEvent event){

                        if(scheduler != null && (parent != null && scheduler != parent.getScheduler())/*scheduler.getResources().contains(this)*/){
                                String src = event.getSource() != null ? event.getSource() : name;
                                scheduler.sendInternal(GridSimTags.SCHEDULE_NOW, DCWormsTags.UPDATE_PROCESSING, src);
                        } else if(parent != null)
                                parent.getEventHandler().handleSchedulingEvent(event);
                        
                        //TODO - check if needed
                        //if(event.getReason() != EventReason.SIM_INIT)
                        //      triggerEventUp(event);
                }
        }
        
        public EventHandler getEventHandler(){
                return new ComputingResourceEventHandler();
        }
        
        public final void initiate(){
                for(Device dev: this.getResourceCharacteristic().getDevices()){
                        dev.initiate();
                }
                ResourceEventCommand rec = new ResourceEventCommand(this);
                EnergyEvent event = new EnergyEvent(EnergyEventType.AIRFLOW_STATE_CHANGED, "Resource controller");
                event.setReason(EventReason.SIM_INIT);
                rec.execute(event);
                /*ResourceEventCommand*/ rec = new ResourceEventCommand(this);
                /*EnergyEvent*/ event = new EnergyEvent(EnergyEventType.POWER_STATE_CHANGED, "Resource controller");
                event.setReason(EventReason.SIM_INIT);
                rec.execute(event);
                
                rec = new ResourceEventCommand(this);
                event = new EnergyEvent(EnergyEventType.TEMPERATURE_CHANGED, "Resource controller");
                event.setReason(EventReason.SIM_INIT);
                rec.execute(event);
                
                //alternative way
                //getEventHandler().handleResourceEvent(new EnergyEvent(EnergyEventType.AIRFLOW_STATE_CHANGED, "Resource controller"));
                //getEventHandler().handleResourceEvent(new EnergyEvent(EnergyEventType.POWER_STATE_CHANGED, "Resource controller"));
        }


}