net.automatalib.automata.transout.impl.FastProbMealy Maven / Gradle / Ivy

Go to download
/* Copyright (C) 2013 TU Dortmund
 * This file is part of AutomataLib, http://www.automatalib.net/.
 * 
 * AutomataLib is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License version 3.0 as published by the Free Software Foundation.
 * 
 * AutomataLib is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 * 
 * You should have received a copy of the GNU Lesser General Public
 * License along with AutomataLib; if not, see
 * http://www.gnu.de/documents/lgpl.en.html.
 */
package net.automatalib.automata.transout.impl;

import net.automatalib.automata.base.fast.FastMutableNondet;
import net.automatalib.automata.transout.probabilistic.MutableProbabilisticMealy;
import net.automatalib.automata.transout.probabilistic.ProbabilisticOutput;
import net.automatalib.words.Alphabet;

public class FastProbMealy
		extends
		FastMutableNondet, I, ProbMealyTransition, O>, Void, ProbabilisticOutput>
		implements
		MutableProbabilisticMealy, I, ProbMealyTransition, O>, O> {

	public FastProbMealy(Alphabet inputAlphabet) {
		super(inputAlphabet);
	}

	@Override
	public FastProbMealyState getSuccessor(
			ProbMealyTransition, O> transition) {
		return transition.getSuccessor();
	}



	@Override
	public O getTransitionOutput(
			ProbMealyTransition, O> transition) {
		return transition.getOutput();
	}


	@Override
	public Void getStateProperty(FastProbMealyState state) {
		return null;
	}

	@Override
	public ProbabilisticOutput getTransitionProperty(
			ProbMealyTransition, O> transition) {
		return new ProbabilisticOutput(transition.getProbability(), transition.getOutput());
	}

	@Override
	public void setTransitionOutput(
			ProbMealyTransition, O> transition, O output) {
		transition.setOutput(output);
	}

	@Override
	public void setTransitionProbability(
			ProbMealyTransition, O> transition,
			float probability) {
		transition.setProbability(probability);
	}

	@Override
	public float getTransitionProbability(
			ProbMealyTransition, O> transition) {
		return transition.getProbability();
	}

	@Override
	public void setStateProperty(FastProbMealyState state, Void property) {
	}

	@Override
	public void setTransitionProperty(
			ProbMealyTransition, O> transition,
			ProbabilisticOutput property) {
		float prob;
		O output;
		if(property == null) {
			prob = 0.0f;
			output = null;
		}
		else {
			prob = property.getProbability();
			output = property.getOutput();
		}
		transition.setProbability(prob);
		transition.setOutput(output);
	}

	@Override
	public ProbMealyTransition, O> createTransition(
			FastProbMealyState successor, ProbabilisticOutput properties) {
		float prob;
		O output;
		if(properties == null) {
			prob = 0.0f;
			output = null;
		}
		else {
			prob = properties.getProbability();
			output = properties.getOutput();
		}
		return new ProbMealyTransition, O>(successor, output, prob);
	}


	@Override
	protected FastProbMealyState createState(Void property) {
		return new FastProbMealyState(inputAlphabet.size());
	}

	
	public void addTransition(FastProbMealyState src, I input, FastProbMealyState successor, O output, float prob) {
		addTransition(src, input, successor, new ProbabilisticOutput(prob, output));
	}

}