de.tla2b.output.PrologPrinter Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of tla2bAST Show documentation
Translator from TLA+ to ProB's AST representation.
There is a newer version: 1.4.0
package de.tla2b.output;

import java.io.File;
import java.io.IOException;
import java.io.PrintWriter;
import java.util.ArrayList;
import java.util.HashSet;
import java.util.Hashtable;
import java.util.List;
import java.util.Map;

import de.be4.classicalb.core.parser.BParser;
import de.be4.classicalb.core.parser.analysis.prolog.ASTProlog;
import de.be4.classicalb.core.parser.analysis.prolog.RecursiveMachineLoader;
import de.be4.classicalb.core.parser.node.Node;
import de.be4.classicalb.core.parser.node.Start;
import de.hhu.stups.sablecc.patch.PositionedNode;
import de.prob.prolog.output.IPrologTermOutput;
import de.prob.prolog.output.PrologTermOutput;
import de.tla2b.types.TLAType;

public class PrologPrinter {
	RecursiveMachineLoader rml;
	BParser bParser;
	String moduleName;

	//private final Map positions = new HashMap();
	private HashSet positions;
	private final List files = new ArrayList();
	private final Hashtable typeTable;

	public PrologPrinter(RecursiveMachineLoader rml, BParser bParser,
			File mainFile, String moduleName, Hashtable typeTable) {
		this.rml = rml;
		this.bParser = bParser;
		this.moduleName = moduleName;
		this.typeTable = typeTable;
		files.add(mainFile);
	}

	public void setPositions( HashSet sourcePositions) {
		positions = sourcePositions;
	}

	public void printAsProlog(final PrintWriter out, final boolean useIndention) {
		final IPrologTermOutput pout = new PrologTermOutput(out, useIndention);
		printAsProlog(pout);
	}

	public void printAsProlog(final IPrologTermOutput pout) {
		// final ClassicalPositionPrinter pprinter = new
		// ClassicalPositionPrinter(
		// rml.getNodeIdMapping());

		final TlaTypePrinter pprinter = new TlaTypePrinter(
				rml.getNodeIdMapping(), typeTable);
		pprinter.setSourcePositions(positions);
		
		final ASTProlog prolog = new ASTProlog(pout, pprinter);

		// parser version
		pout.openTerm("parser_version");
		pout.printAtom(BParser.getBuildRevision());
		pout.closeTerm();
		pout.fullstop();

		// machine
		pout.openTerm("classical_b");
		pout.printAtom(moduleName);
		pout.openList();
		for (final File file : files) {
			try {
				pout.printAtom(file.getCanonicalPath());
			} catch (IOException e) {
				pout.printAtom(file.getPath());
			}
		}
		pout.closeList();
		pout.closeTerm();
		pout.fullstop();
		for (final Map.Entry entry : rml.getParsedMachines()
				.entrySet()) {
			pout.openTerm("machine");
			//final SourcePositions src = positions.get(entry.getKey());
			//pprinter.setSourcePositions(src);
			entry.getValue().apply(prolog);
			pout.closeTerm();
			pout.fullstop();
		}

		pout.flush();
	}
}