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

Go to download

Show more of this group Show more artifacts with this name
Show all versions of automata-core Show documentation

The core artifact of AutomataLib. This library contains concrete implementations for many of the automaton model interfaces defined in the API artifact, as well as abstract base classes that facilitate implementing new automaton model classes. Note that concrete algorithms (traversal, reachability analysis etc.) are not in the scope of this artifact.

There is a newer version: 0.11.0

Show newest version

/* Copyright (C) 2013-2018 TU Dortmund
 * This file is part of AutomataLib, http://www.automatalib.net/.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package net.automatalib.automata.transout.impl;

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

public class FastProbMealy
        extends AbstractFastMutableNondet, 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<>(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));
    }

}