com.googlecode.blaisemath.graph.modules.metrics.AdditiveSubsetMetric Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of blaise-graphtheory Show documentation

Link graph definitions, algorithms, and visualization.

The newest version!

/*
 * DerivedMetrics.java
 * Created May 18, 2010
 */

package com.googlecode.blaisemath.graph.modules.metrics;

/*
 * #%L
 * BlaiseGraphTheory
 * --
 * Copyright (C) 2009 - 2016 Elisha Peterson
 * --
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * 
 *      http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 * #L%
 */

import com.googlecode.blaisemath.graph.GraphSubsetMetric;
import com.googlecode.blaisemath.graph.GraphNodeMetric;
import java.util.Set;
import com.googlecode.blaisemath.graph.Graph;

/**
 * Provides a GraphSubsetMetric computed by adding together the
 * return values of a GraphNodeMetric for each node in the subset.
 *
 * @author Elisha Peterson
 */
public class AdditiveSubsetMetric implements GraphSubsetMetric {

    GraphNodeMetric baseMetric;

    /**
     * Constructs with provided base metric.
     * @param baseMetric the metric to use for computations on individual nodes.
     */
    public AdditiveSubsetMetric(GraphNodeMetric baseMetric) { 
        this.baseMetric = baseMetric;
    }

    public  N getValue(Graph graph, Set vertices) {
        Double result = 0.0;
        Number val = null;
        for (V v : vertices) {
            val = (Number) baseMetric.apply(graph, v);
            result += val.doubleValue();
        }
        if (val instanceof Integer) {
            return (N) (Integer) result.intValue();
        }
        else if (val instanceof Double) {
            return (N) result;
        }
        else if (val instanceof Float) {
            return (N) (Float) result.floatValue();
        }
        return null;
    }
}