SetReplacer.java

/*
 * Copyright 2013, 2014 Deutsche Nationalbibliothek
 *
 * 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 org.culturegraph.mf.commons.tries;

import java.util.Collections;
import java.util.Comparator;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;

import org.culturegraph.mf.commons.tries.SetMatcher.Match;

/**
 * @author Markus Michael Geipel
 *
 */
public final class SetReplacer {
	private final SetMatcher<String> matcher = new SetMatcher<String>();

	public void addReplacement(final String key, final String with) {
		matcher.put(key, with);
	}

	public void addReplacements(final Map<String, String> replacements) {
		for (Entry<String, String> entry : replacements.entrySet()) {
			addReplacement(entry.getKey(), entry.getValue());
		}
	}

	public String replaceIn(final String text) {
		final List<SetMatcher.Match<String>> matches = matcher.match(text);
		final StringBuilder builder = new StringBuilder();

		int lastCut = 0;

		Collections.sort(matches, new Comparator<SetMatcher.Match<String>>() {
			@Override
			public int compare(final Match<String> o1, final Match<String> o2) {
				final int result;
				final int delta = o1.getStart() - o2.getStart();
				if (delta < 0) {
					result = -1;
				} else if (delta > 0) {
					result = 1;
				} else {
					if (o1.getLength() > o2.getLength()) {
						result = -1;
					} else {
						result = 1;
					}
				}
				return result;
			}

		});

		for (SetMatcher.Match<String> match : matches) {

			if (match.getStart() < lastCut) {
				continue;
			}

			// System.out.println(match.getStart() + " "+ match.getValue() +" "+
			// match.getLength());

			builder.append(text.substring(lastCut, match.getStart()));
			builder.append(match.getValue());

			lastCut = match.getStart() + match.getLength();
		}
		builder.append(text.substring(lastCut, text.length()));

		return builder.toString();
	}

}