net.automatalib.util.ts.copy.TSCopyVisitor Maven / Gradle / Ivy

Go to download

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

This artifact provides various common utility operations for analyzing and manipulating automata and graphs, such as traversal, minimization and copying.

There is a newer version: 0.11.0

Show newest version

/* Copyright (C) 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.util.ts.copy;

import net.automatalib.automata.MutableAutomaton;
import net.automatalib.commons.util.Holder;
import net.automatalib.commons.util.mappings.Mapping;
import net.automatalib.commons.util.mappings.MutableMapping;
import net.automatalib.ts.TransitionPredicate;
import net.automatalib.ts.TransitionSystem;
import net.automatalib.util.ts.traversal.TSTraversalAction;
import net.automatalib.util.ts.traversal.TSTraversalVisitor;

import com.google.common.base.Function;
import com.google.common.base.Predicate;

public class TSCopyVisitor implements
		TSTraversalVisitor {
	
	private final MutableMapping stateMapping;
	private final MutableAutomaton out;
	
	private final Function inputMapping;
	private final Function spMapping;
	private final Function tpMapping;
	
	private final Predicate stateFilter;
	private final TransitionPredicate transFilter;
	
	public TSCopyVisitor(TransitionSystem in,
			MutableAutomaton out,
			Function inputMapping,
			Function spMapping,
			Function tpMapping,
			Predicate stateFilter,
			TransitionPredicate transFilter) {
		this.stateMapping = in.createStaticStateMapping();
		this.out = out;
		this.inputMapping = inputMapping;
		this.spMapping = spMapping;
		this.tpMapping = tpMapping;
		this.stateFilter = stateFilter;
		this.transFilter = transFilter;
	}
			
			

	@Override
	public TSTraversalAction processInitial(S1 state, Holder outData) {
		S2 s2 = stateMapping.get(state);
		if(s2 != null) {
			out.setInitial(s2, true);
			return TSTraversalAction.IGNORE;
		}
		else if(!stateFilter.apply(state)) {
			return TSTraversalAction.IGNORE;
		}
		
		SP2 sp = spMapping.apply(state);
		s2 = out.addInitialState(sp);
		
		stateMapping.put(state, s2);
		
		outData.value = s2;
		return TSTraversalAction.EXPLORE;
	}

	@Override
	public boolean startExploration(S1 state, S2 data) {
		return true;
	}

	@Override
	public TSTraversalAction processTransition(S1 source, S2 source2, I1 input,
			T1 transition, S1 succ, Holder outData) {
		if(!transFilter.apply(source, input, transition)) {
			return TSTraversalAction.IGNORE;
		}
		
		boolean ignore = false;
		
		S2 succ2 = stateMapping.get(succ);
		if(succ2 == null) {
			if(!stateFilter.apply(succ)) {
				return TSTraversalAction.IGNORE;
			}
			SP2 sp = spMapping.apply(succ);
			succ2 = out.addState(sp);
			stateMapping.put(succ, succ2);
		}
		else {
			ignore = true;
		}

		I2 input2 = inputMapping.apply(input);
		TP2 tp = tpMapping.apply(transition);
		
		out.addTransition(source2, input2, succ2, tp);
		outData.value = succ2;
		
		return (ignore) ? TSTraversalAction.IGNORE : TSTraversalAction.EXPLORE;
	}

	
	public Mapping getStateMapping() {
		return stateMapping;
	}
}