be.fedict.eid.applet.maven.MyVertexTransformer.java Source code

Introduction

Here is the source code for be.fedict.eid.applet.maven.MyVertexTransformer.java

Source

/*
 * eID Applet Project.
 * Copyright (C) 2008-2010 FedICT.
 *
 * This is free software; you can redistribute it and/or modify it
 * under the terms of the GNU Lesser General Public License version
 * 3.0 as published by the Free Software Foundation.
 *
 * This software is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this software; if not, see 
 * http://www.gnu.org/licenses/.
 */

package be.fedict.eid.applet.maven;

import java.awt.Color;
import java.awt.Paint;
import java.util.List;

import org.apache.commons.collections15.Transformer;

public class MyVertexTransformer implements Transformer<String, Paint> {

    private final String startMessage;

    private final List<String> stopMessages;

    public MyVertexTransformer(String startMessage, List<String> stopMessages) {
        this.startMessage = startMessage;
        this.stopMessages = stopMessages;
    }

    public Paint transform(String vertexName) {
        if (this.startMessage.equals(vertexName)) {
            return Color.GREEN;
        }
        if (this.stopMessages.contains(vertexName)) {
            return Color.RED;
        }
        return Color.WHITE;
    }
}