de.tudarmstadt.ukp.dkpro.argumentation.io.annotations.SpanAnnotationMatrices.java Source code

Introduction

Here is the source code for de.tudarmstadt.ukp.dkpro.argumentation.io.annotations.SpanAnnotationMatrices.java

Source

/*
 * Copyright 2016
 * Ubiquitous Knowledge Processing (UKP) Lab
 * Technische Universitt Darmstadt
 *
 * 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.
 */
package de.tudarmstadt.ukp.dkpro.argumentation.io.annotations;

import java.util.Collection;
import java.util.Map;
import java.util.stream.Stream;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;

import de.tudarmstadt.ukp.math.Sparse3DObjectMatrix;
import it.unimi.dsi.fastutil.ints.Int2ObjectOpenHashMap;

/**
 * @author Todd Shore
 * @since May 30, 2016
 *
 */
@SuppressWarnings("deprecation")
final class SpanAnnotationMatrices {

    private static final Log LOG = LogFactory.getLog(SpanAnnotationMatrices.class);

    public static <T extends SpanTextLabel> Sparse3DObjectMatrix<String, T> createMatrix(
            final Collection<T> spanAnnotationVector) {
        final Sparse3DObjectMatrix<String, T> result = new Sparse3DObjectMatrix<>(
                new Int2ObjectOpenHashMap<>(spanAnnotationVector.size() + 1));
        putAnnotations(result, spanAnnotationVector.stream());
        return result;

    }

    public static <T extends SpanTextLabel> Sparse3DObjectMatrix<String, T> createMatrix(
            final Collection<T> spanAnnotationVector, final int estimatedSpanBeginToEndMapMaxCapacity,
            final int estimatedAnnotationMapMaxCapacity) {
        final Sparse3DObjectMatrix<String, T> result = new Sparse3DObjectMatrix<>(
                new Int2ObjectOpenHashMap<>(spanAnnotationVector.size() + 1), estimatedSpanBeginToEndMapMaxCapacity,
                estimatedAnnotationMapMaxCapacity);
        putAnnotations(result, spanAnnotationVector.stream());
        return result;
    }

    public static <T extends SpanTextLabel> void putAnnotations(
            final Sparse3DObjectMatrix<? super String, T> result, final Stream<T> spanAnnotationVector) {
        spanAnnotationVector.forEach(spanAnnotation -> {
            final SpanText spanText = spanAnnotation.getSpanText();
            final Span span = spanText.getSpan();
            final int begin = span.getBegin();
            final int end = span.getEnd();
            final Map<? super String, T> spanAnnotations = result.fetch3DMap(begin, end);
            final String label = spanAnnotation.getLabel();
            final T oldSpanAnnotation = spanAnnotations.put(label, spanAnnotation);
            if (oldSpanAnnotation != null) {
                LOG.warn(String.format("Annotation label \"%s\" already exists for span [%d, %d]; Overwriting.",
                        label, begin, end));
            }
        });
    }

    private SpanAnnotationMatrices() {
    }

}