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

Go to download
Show more of this group Show more artifacts with this name
Show all versions of jenax-arq-aggregators Show documentation
The newest version!
package org.aksw.jenax.arq.aggregation;

import java.util.Collections;
import java.util.HashMap;
import java.util.Iterator;
import java.util.Map;

import org.aksw.jenax.arq.util.triple.TripleUtils;
import org.apache.jena.atlas.lib.Sink;
import org.apache.jena.graph.Node;
import org.apache.jena.graph.Triple;
import org.apache.jena.sparql.core.Var;
import org.apache.jena.sparql.engine.binding.Binding;
import org.apache.jena.sparql.expr.NodeValue;
import org.apache.jena.sparql.function.FunctionEnv;
import org.apache.jena.sparql.modify.TemplateLib;
import org.apache.jena.sparql.syntax.Template;

public class AccSinkTriples>
    implements Acc
{
    protected T sink;
    protected Template template;
    protected Node reverse;

    protected Map bNodeMap = new HashMap();


//    public AccSinkTriples(Template template) {
//       this(template, NodeValue.FALSE.asNode());
//    }

    public AccSinkTriples(T sink, Template template) {
        this(sink, template, NodeValue.FALSE.asNode());
    }
//    public AccSinkTriples(Template template, Node reverse) {
//        this(GraphFactory.createDefaultGraph(), template, reverse);
//    }

    public AccSinkTriples(T sink, Template template, Node reverse) {
        super();
        this.sink = sink;
        this.template = template;
        this.reverse = reverse;
    }

    public static boolean isTrue(Object o) {
        boolean result = Boolean.TRUE.equals(o) || (o instanceof Number && ((Number)o).intValue() == 1);
        return result;
    }

    @Override
    public void accumulate(Binding binding, FunctionEnv env) {
//        Set triples = new HashSet();
//        template.subst(triples, bNodeMap, binding);

        Node node = reverse.isVariable()
                ? binding.get((Var)reverse)
                : reverse
                ;

        boolean doReverse = node.isLiteral()
                ? isTrue(node.getLiteralValue())
                : false
                ;


        Iterator it = TemplateLib.calcTriples(template.getTriples(), Collections.singleton(binding).iterator());

        //for(Triple triple : triples) {
        while(it.hasNext()) {
            Triple triple = it.next();
            if(doReverse) {
                triple = TripleUtils.swap(triple);
            }

            // graph.add(triple);
            sink.send(triple);
        }
    }

    @Override
    public T getValue() {
        return sink;
    }

}