heros.flowfunc.Compose.java Source code

Introduction

Here is the source code for heros.flowfunc.Compose.java

Source

/*******************************************************************************
 * Copyright (c) 2012 Eric Bodden.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the GNU Lesser Public License v2.1
 * which accompanies this distribution, and is available at
 * http://www.gnu.org/licenses/old-licenses/gpl-2.0.html
 * 
 * Contributors:
 *     Eric Bodden - initial API and implementation
 ******************************************************************************/
package heros.flowfunc;

import heros.FlowFunction;

import java.util.Set;

import com.google.common.collect.Sets;

/**
 * Represents the ordered composition of a set of flow functions.
 */
public class Compose<D> implements FlowFunction<D> {

    private final FlowFunction<D>[] funcs;

    public Compose(FlowFunction<D>... funcs) {
        this.funcs = funcs;
    }

    public Set<D> computeTargets(D source) {
        Set<D> curr = Sets.newHashSet();
        curr.add(source);
        for (FlowFunction<D> func : funcs) {
            Set<D> next = Sets.newHashSet();
            for (D d : curr)
                next.addAll(func.computeTargets(d));
            curr = next;
        }
        return curr;
    }

}