FirefighterDeparmtmentDistanceCalculator.java

/*
 * To change this template, choose Tools | Templates
 * and open the template in the editor.
 */

package com.wordpress.salaboy.acc;

import com.wordpress.salaboy.model.Emergency;
import com.wordpress.salaboy.model.FirefightersDepartment;
import java.io.IOException;
import java.io.ObjectInput;
import java.io.ObjectOutput;
import java.io.Serializable;
import org.drools.runtime.rule.AccumulateFunction;

/**
 *
 * @author esteban
 */
public class FirefighterDeparmtmentDistanceCalculator implements AccumulateFunction, Serializable {

    //TODO: change this values to real firefighters deparment values
    private int[] xs = new int[]{1, 7, 13, 19, 25, 31, 37};
    private int[] ys = new int[]{1, 7, 13, 19, 25};
    
    public static class ContextData implements Serializable{
        public FirefightersDepartment selectedDepartment;
        public double minDistance;
    }
    
    @Override 
    public void writeExternal(ObjectOutput out) throws IOException {
    }

    @Override
    public void readExternal(ObjectInput in) throws IOException, ClassNotFoundException {
    }

    @Override
    public Serializable createContext() {
        return new ContextData();
    }

    @Override
    public void init(Serializable context) throws Exception {
        ContextData contextData = (ContextData)context;
        contextData.selectedDepartment = null;
        contextData.minDistance = 0;
    }

    @Override
    public void accumulate(Serializable context, Object value) {
        ContextData contextData = (ContextData)context;
        
        FirefightersDepartmentDistanceCalculationData data = (FirefightersDepartmentDistanceCalculationData)value;
        FirefightersDepartment currentDepartment = data.getDepartment();
        Emergency emergency = data.getEmergency();
        
        float difX = currentDepartment.getX() - xs[emergency.getLocation().getLocationX()];
        float difY = currentDepartment.getY() - ys[emergency.getLocation().getLocationY()];
        
        double difTotal = Math.sqrt(Math.pow(((double)difX),2d) + Math.pow(((double)difY),2d));
        
        if(contextData.selectedDepartment == null || difTotal < contextData.minDistance){ 
            contextData.minDistance = difTotal; 
            contextData.selectedDepartment = currentDepartment;
        }
        
    }

    @Override
    public void reverse(Serializable context, Object value) throws Exception {
    }

    @Override
    public Object getResult(Serializable context) throws Exception {
        return ((ContextData)context).selectedDepartment;
    }

    @Override
    public boolean supportsReverse() {
        return false;
    }

}