Instance.java

package ssmdp.instance;

import java.util.ArrayList;
import java.util.List;
import ssmdp.util.Weighted;

public class Instance {

    private double weights[][];
    private List<Node> nodes;
    private List<Weighted<Node>> nodesDistance;
    private int numSolutionNodes = 0;
    private String name;

    public Instance(int numSolutionNodes, double[][] weights) {
        this("Unknown",numSolutionNodes,weights);
    }

    public Instance(String name, int numSolutionNodes, double[][] weights) {
        this.name = name;
        this.numSolutionNodes = numSolutionNodes;
        this.weights = weights;
        nodes = new ArrayList<Node>();
        for (int i = 0; i < weights.length + 1; i++) {
            nodes.add(new Node(i, this));
        }
    }

    public Node getNode(int index) {
        return nodes.get(index);
    }

    public double getWeight(int i, int j) {
        if(i == j){
            return 0;
        } else {
            return weights[Math.max(i, j) - 1][Math.min(i, j)];
        }
    }

    public int getNumNodes() {
        return nodes.size();
    }

    public List<Node> getNodes() {
        return nodes;
    }

    public boolean isNodesDistanceCalculated() {
        return nodesDistance != null;
    }

    public List<Weighted<Node>> getNodesDistance() {
        if (nodesDistance == null) {
            nodesDistance = new ArrayList<Weighted<Node>>();

            for (Node n : getNodes()) {
                double weight = 0;
                for (Node on : getNodes()) {
                    weight += n.getDistanceTo(on);
                }
                nodesDistance.add(new Weighted<Node>(n, weight));
            }
        }

        return nodesDistance;
    }

    public String getName() {
        return name;
    }

    public int getNumSolutionNodes() {
        return numSolutionNodes;
    }

    public String getDescription() {
        return name + " (n=" + getNumNodes() + ", m=" + numSolutionNodes + ")";
    }

    @Override
    public String toString() {
        StringBuilder sb = new StringBuilder("Number of Nodes: "
                + getNumNodes() + "\n");
        int rowNumber = 1;
        for (double[] row : weights) {
            sb.append(rowNumber).append(": ");
            int col = 0;
            for (double value : row) {
                sb.append(value).append(" ").append("[" + col + "]");
                col++;
            }
            sb.append("\n");
            rowNumber++;
        }
        return sb.toString();
    }

}