SingleColorScheme.java

// License: GPL. For details, see LICENSE file.
package org.openstreetmap.josm.plugins.graphview.core.visualisation;

import java.awt.Color;

import org.openstreetmap.josm.plugins.graphview.core.graph.GraphNode;
import org.openstreetmap.josm.plugins.graphview.core.transition.Segment;

/**
 * scheme giving the same color to all segments and the same color to all nodes
 * (but possibly different colors for one of them)
 */
public class SingleColorScheme implements ColorScheme {

    private final Color nodeColor;
    private final Color segmentColor;

    public SingleColorScheme(Color nodeColor, Color segmentColor) {
        this.nodeColor = nodeColor;
        this.segmentColor = segmentColor;
    }

    @Override
    public Color getNodeColor(GraphNode node) {
        return nodeColor;
    }

    @Override
    public Color getSegmentColor(Segment segment) {
        return segmentColor;
    }

}