super-snuvi/src/me/hammerle/supersnuvi/gamelogic/pathfinding/LevelGraph.java

package me.hammerle.supersnuvi.gamelogic.pathfinding;

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

/**
 * level representation as graph
 * used to apply path finding for entities
 */
public class LevelGraph implements IGraph, IModifieableGraph {

    private Set<LevelGraphNode> nodes;

    public LevelGraph(){

        nodes = new HashSet<>();
    }


    @Override
    public Set<LevelGraphNode> getNodes() {
        return nodes;
    }

    @Override
    public void addNode(INode node) {

        if(node instanceof LevelGraphNode)
            nodes.add((LevelGraphNode) node);
    }

    @Override
    public void removeNode(INode node) {

        if(node instanceof LevelGraphNode)
            nodes.remove(node);
    }
}