Skip to content
Node.java 2.03 KiB
Newer Older
Javier Costa's avatar
Javier Costa committed
package tfm.nodes;
Javier Costa's avatar
Javier Costa committed

Javier Costa's avatar
Javier Costa committed
import com.github.javaparser.ast.stmt.Statement;
import edg.graphlib.Arrow;
Javier Costa's avatar
Javier Costa committed
import edg.graphlib.Vertex;
Javier Costa's avatar
Javier Costa committed
import tfm.arcs.data.ArcData;
import tfm.graphs.Graph;
Javier Costa's avatar
Javier Costa committed

Javier Costa's avatar
Javier Costa committed
import java.util.Objects;
Javier Costa's avatar
Javier Costa committed
import java.util.stream.Collectors;

Javier Costa's avatar
Javier Costa committed
public class Node extends Vertex<String, ArcData> {
Javier Costa's avatar
Javier Costa committed

Javier Costa's avatar
Javier Costa committed
    private Statement statement;
Javier Costa's avatar
Javier Costa committed
//    public Node(Graph.NodeId id, String instruction) {
//        this(id, instruction, null);
//    }
Javier Costa's avatar
Javier Costa committed
    public Node(Graph.NodeId id, String representation, Statement statement) {
        super(id.toString(), representation);
Javier Costa's avatar
Javier Costa committed
        this.statement = statement;
Javier Costa's avatar
Javier Costa committed
    }

    public int getId() {
        return Integer.parseInt(getName());
    }

    public String toString() {
Javier Costa's avatar
Javier Costa committed
        return String.format("Node{id: %s, data: '%s', in: %s, out: %s}",
Javier Costa's avatar
Javier Costa committed
                getName(),
                getData(),
                getIncomingArrows().stream().map(arrow -> arrow.getFrom().getName()).collect(Collectors.toList()),
                getOutgoingArrows().stream().map(arc -> arc.getTo().getName()).collect(Collectors.toList()));
    }
Javier Costa's avatar
Javier Costa committed
    public Statement getStatement() {
        return statement;
Javier Costa's avatar
Javier Costa committed
    public Optional<Integer> getFileLineNumber() {
        return statement.getBegin().isPresent() ? Optional.of(statement.getBegin().get().line) : Optional.empty();
Javier Costa's avatar
Javier Costa committed

    @Override
    public boolean equals(Object o) {
        if (this == o)
            return true;

Javier Costa's avatar
Javier Costa committed
        if (!(o instanceof Node))
Javier Costa's avatar
Javier Costa committed
            return false;

Javier Costa's avatar
Javier Costa committed
        Node other = (Node) o;
Javier Costa's avatar
Javier Costa committed

        return Objects.equals(getData(), other.getData())
                && Objects.equals(getIncomingArrows(), other.getIncomingArrows())
                && Objects.equals(getOutgoingArrows(), other.getOutgoingArrows())
Javier Costa's avatar
Javier Costa committed
                && Objects.equals(statement, other.statement);
Javier Costa's avatar
Javier Costa committed
                // && Objects.equals(getName(), other.getName()) ID IS ALWAYS UNIQUE, SO IT WILL NEVER BE THE SAME
    }
Javier Costa's avatar
Javier Costa committed

    public String toGraphvizRepresentation() {
        return String.format("%s[label=\"%s: %s\"];", getId(), getId(), getData());
    }
Javier Costa's avatar
Javier Costa committed
}