de.knowwe.compile.correction.IncrementalTermReferenceCorrectionProvider.java Source code

Introduction

Here is the source code for de.knowwe.compile.correction.IncrementalTermReferenceCorrectionProvider.java

Source

/*
 * Copyright (C) 2010 University Wuerzburg, Computer Science VI
 * 
 * This is free software; you can redistribute it and/or modify it under the
 * terms of the GNU Lesser General Public License as published by the Free
 * Software Foundation; either version 3 of the License, or (at your option) any
 * later version.
 * 
 * This software 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 Lesser General Public License for more
 * details.
 * 
 * You should have received a copy of the GNU Lesser General Public License
 * along with this software; if not, write to the Free Software Foundation,
 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA, or see the FSF
 * site: http://www.fsf.org.
 */
package de.knowwe.compile.correction;

import java.util.Collection;
import java.util.LinkedList;
import java.util.List;

import org.apache.commons.lang3.StringUtils;

import com.denkbares.strings.Identifier;
import de.knowwe.compile.IncrementalCompiler;
import de.knowwe.compile.ReferenceManager;
import de.knowwe.core.compile.terminology.TermCompiler;
import de.knowwe.core.correction.CorrectionProvider;
import de.knowwe.core.correction.DefaultSuggestion;
import de.knowwe.core.correction.Suggestion;
import de.knowwe.core.kdom.objects.SimpleDefinition;
import de.knowwe.core.kdom.objects.SimpleReference;
import de.knowwe.core.kdom.parsing.Section;
import de.knowwe.core.utils.KnowWEUtils;

public class IncrementalTermReferenceCorrectionProvider implements CorrectionProvider {

    @Override
    public List<Suggestion> getSuggestions(TermCompiler compiler, Section<?> section, int threshold) {
        List<Suggestion> suggestions = new LinkedList<>();
        if (!(section.get() instanceof SimpleReference)) {
            return suggestions;
        }

        ReferenceManager terminology = IncrementalCompiler.getInstance().getTerminology();

        if (terminology.isValid(new Identifier(section.getText()))) {
            // if reference is valid, no correction is proposed
            return suggestions;
        }

        Collection<Section<? extends SimpleDefinition>> defs = terminology.getAllTermDefinitions();

        String originalText = section.getText();

        for (Section<? extends SimpleDefinition> def : defs) {
            Identifier termIdentifier = KnowWEUtils.getTermIdentifier(def);
            String termIdentifierElement = termIdentifier.getLastPathElement();
            String originalTextRegex = originalText.replace(" ", ".*");

            /* levenstein test */
            double score = StringUtils.getLevenshteinDistance(originalText, termIdentifierElement, threshold);
            if (score >= 0) {
                suggestions.add(new DefaultSuggestion(termIdentifierElement, (int) score));
            }
            /* infix test */
            else if (termIdentifierElement.matches(".*" + originalTextRegex + ".*")) {
                int infixScore = termIdentifierElement.length() - originalText.length();
                suggestions.add(new DefaultSuggestion(termIdentifierElement, infixScore));
            }
        }

        return suggestions;
    }
}