package ac.essex.gp.markov;

import java.util.Vector;

/**
 * Initial probabilities of states
 * are recorded in the pi-vector.
 *
 * @author Olly Oechsle, University of Essex, Date: 05-Mar-2007
 * @version 1.0
 */
public class PiVector {

    protected double[] probabilities;

    public PiVector(Vector<HiddenState> states) {
        probabilities = new double[states.size()];
    }

    public void setProbability(HiddenState s, double probability) {
        probabilities[s.id] = probability;
    }

    public double getProbability(HiddenState s) {
        return probabilities[s.id];
    }
         
}