hu.bme.mit.theta.cfa.dsl.CfaWriter Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of theta-cfa Show documentation
Cfa subproject in the Theta model checking framework
There is a newer version: 6.8.5
/*
 *  Copyright 2024 Budapest University of Technology and Economics
 *
 *  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 hu.bme.mit.theta.cfa.dsl;

import hu.bme.mit.theta.cfa.CFA;
import hu.bme.mit.theta.cfa.CFA.Edge;
import hu.bme.mit.theta.cfa.CFA.Loc;
import hu.bme.mit.theta.core.decl.Decl;
import hu.bme.mit.theta.core.dsl.CoreDslManager;
import hu.bme.mit.theta.core.stmt.Stmt;

import java.io.BufferedWriter;
import java.io.IOException;
import java.io.OutputStream;
import java.io.OutputStreamWriter;

public final class CfaWriter {

    private CfaWriter() {
    }

    public static void write(final CFA cfa, final OutputStream outStream) throws IOException {
        final BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(outStream));

        bw.write("main process cfa {");
        bw.newLine();

        for (final Decl var : cfa.getVars()) {
            final String varName = var.getName();
            final String varType = var.getType().toString().toLowerCase();
            bw.write(String.format("\tvar %s : %s", varName, varType));
            bw.newLine();
        }

        bw.newLine();

        for (final Loc loc : cfa.getLocs()) {
            final String locName = "L" + loc.getName();
            String locPrefix = "";
            if (cfa.getErrorLoc().isPresent() && loc == cfa.getErrorLoc().get()) {
                locPrefix += "error ";
            }
            if (loc == cfa.getInitLoc()) {
                locPrefix += "init ";
            }
            if (cfa.getFinalLoc().isPresent() && loc == cfa.getFinalLoc().get()) {
                locPrefix += "final ";
            }
            bw.write(String.format("\t%sloc %s", locPrefix, locName));
            bw.newLine();
        }

        bw.newLine();

        for (final Edge edge : cfa.getEdges()) {
            final String sourceLoc = "L" + edge.getSource().getName();
            final String targetLoc = "L" + edge.getTarget().getName();
            bw.write(String.format("\t%s -> %s {", sourceLoc, targetLoc));
            bw.newLine();
            bw.write(String.format("\t\t%s", writeStmt(edge.getStmt())));
            bw.newLine();
            bw.write("\t}");
            bw.newLine();
        }

        bw.write("}");
        bw.close();
    }

    private static String writeStmt(final Stmt stmt) {
        final CoreDslManager dslManager = new CoreDslManager();
        return dslManager.writeStmt(stmt);
    }
}