net.automatalib.ts.powerset.DirectPowersetDTS Maven / Gradle / Ivy

Go to download
/* Copyright (C) 2013-2014 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.ts.powerset;

import java.util.Collection;
import java.util.HashSet;
import java.util.Set;

import net.automatalib.ts.PowersetViewTS;
import net.automatalib.ts.TransitionSystem;
import net.automatalib.ts.abstractimpl.AbstractDTS;


public class DirectPowersetDTS extends AbstractDTS, I, Set>
		implements PowersetViewTS, I, Set, S, T> {
	
	private final TransitionSystem ts;
	
	public DirectPowersetDTS(TransitionSystem ts) {
		this.ts = ts;
	}

	@Override
	public Set getInitialState() {
		return ts.getInitialStates();
	}

	@Override
	public Set getTransition(Set state, I input) {
		Set result = new HashSet<>();
		for(S s : state) {
			Collection transitions = ts.getTransitions(s, input);
			result.addAll(transitions);
		}
		return result;
	}

	@Override
	public Set getSuccessor(Set transition) {
		Set result = new HashSet();
		for(T trans : transition)
			result.add(ts.getSuccessor(trans));
		return result;
	}
	
	@Override
	public Set getSuccessor(Set state, I input) {
		Set result = new HashSet();
		for(S s : state) {
			Collection transitions = ts.getTransitions(s, input);
			for(T t : transitions) {
				result.add(ts.getSuccessor(t));
			}
		}
		
		return result;
	}

	@Override
	public Collection getOriginalStates(Set state) {
		return state;
	}

	@Override
	public Collection getOriginalTransitions(
			Set transition) {
		return transition;
	}
}