• Home
• org.aksw.jenax
• jenax-arq-aggregators
• 5.2.0-1
• source code
• AccMap2.java

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

org.aksw.jenax.arq.aggregation.AccMap2 Maven / Gradle / Ivy

package org.aksw.jenax.arq.aggregation;

import java.util.HashMap;
import java.util.Map;
import java.util.Map.Entry;
import java.util.function.BiFunction;
import java.util.function.Function;

import org.aksw.commons.collector.domain.Accumulator;
import org.aksw.commons.collector.domain.Aggregator;


/**
 *
 *
 * @author raven
 *
 * @param 
 * @param 
 * @param 
 * @param 
 */
public class AccMap2>
    implements Accumulator>
{
    protected BiFunction mapper;
    protected C subAgg;

    protected Map> state = new HashMap<>();

    public AccMap2(Function mapper, C subAgg) {
        this((binding, rowNum) -> mapper.apply(binding), subAgg);
    }

    public AccMap2(BiFunction mapper, C subAgg) {
        this.mapper = mapper;
        this.subAgg = subAgg;
    }

    @Override
    public void accumulate(B binding, E env) {
        // TODO Keep track of the relative binding index
        K k = mapper.apply(binding, -1l);
        Accumulator subAcc = state.get(k);
        if(subAcc == null) {
            subAcc = subAgg.createAccumulator();
            state.put(k, subAcc);
        }
        subAcc.accumulate(binding, env);
    }

    @Override
    public Map getValue() {
        Map result = new HashMap();

        for(Entry> entry : state.entrySet()) {
            K k = entry.getKey();
            V v = entry.getValue().getValue();

            result.put(k, v);
        }

        return result;
    }

    public static > AccMap2 create(Function mapper, C subAgg) {
        BiFunction fn = (binding, rowNum) -> mapper.apply(binding);
        AccMap2 result = new AccMap2<>(fn, subAgg);
        return result;
    }

    public static > AccMap2 create(BiFunction mapper, C subAgg) {
        AccMap2 result = new AccMap2<>(mapper, subAgg);
        return result;
    }

}