eu.interedition.collatex.simple.SimpleToken Maven / Gradle / Ivy

Go to download
/*
 * Copyright (c) 2015 The Interedition Development Group.
 *
 * This file is part of CollateX.
 *
 * CollateX is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * CollateX is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with CollateX.  If not, see .
 */

package eu.interedition.collatex.simple;

import eu.interedition.collatex.Token;
import eu.interedition.collatex.Witness;
import eu.interedition.collatex.util.VertexMatch;

import java.util.SortedSet;
import java.util.function.Function;
import java.util.stream.Collectors;
import java.util.stream.StreamSupport;

public class SimpleToken implements Token, Comparable {
    private final SimpleWitness witness;
    private final String content;
    private final String normalized;

    public SimpleToken(SimpleWitness witness, String content, String normalized) {
        this.witness = witness;
        this.content = content;
        this.normalized = normalized;
    }

    public String getContent() {
        return content;
    }

    @Override
    public Witness getWitness() {
        return witness;
    }

    public String getNormalized() {
        return normalized;
    }

    @Override
    public String toString() {
        return new StringBuilder(witness.toString()).append(":").append(witness.getTokens().indexOf(this)).append(":'").append(normalized).append("'").toString();
    }

    public static String toString(Iterable tokens) {
        return StreamSupport.stream(tokens.spliterator(), false)
            .filter(t -> SimpleToken.class.isAssignableFrom(t.getClass()))
            .map(t -> (SimpleToken) t)
            .map(SimpleToken::getContent)
            .collect(Collectors.joining())
            .trim();
    }

    @Override
    public int compareTo(SimpleToken o) {
        return witness.compare(this, o);
    }

    public static final Function, Integer> TOKEN_MATCH_EVALUATOR = input -> {
        int value = 0;
        for (VertexMatch.WithToken match : input) {
            value += ((SimpleToken) match.token).getContent().length();
        }
        return value;
    };
}