TermsFacetCollationComparator.java

/*
 * Licensed to ElasticSearch and Shay Banon under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership. ElasticSearch licenses this
 * file to you 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.elasticsearch.search.facet.terms.comparator.icu;

import com.ibm.icu.text.Collator;

import org.elasticsearch.search.facet.icu.TermsFacet.Entry;

/**
 * Collation-aware terms facet comparator
 * 
 * @author joerg
 */
public class TermsFacetCollationComparator extends AbstractTermsFacetComparator {

    private Collator collator;
    private int decomp = -1;
    private int strength = -1;
    private String rules = null;

    public TermsFacetCollationComparator(String type, boolean reverse) {
        super(type, reverse);
    }

    public void setStrength(int strength) {
        this.strength = strength;
    }
    
    public void setDecomposition(int decomp) {
        this.decomp = decomp;
    }
    
    public void setRules(String rules) {
        this.rules = rules;
    }
    
    @Override
    public int getDecomposition() {
        return decomp;
    }
    
    @Override
    public String getRules() {
        return rules;
    }
    @Override
    public int getStrength() {
        return strength;
    }
 
    public TermsFacetCollationComparator setCollator(Collator collator) {
        this.collator = collator;
        return this;
    }

    public Collator getCollator() {
        return collator;
    }

    @Override
    public int compare(Entry o1, Entry o2) {
        return reverse ? collator.compare(o2.getTerm(), o1.getTerm()) : collator.compare(o1.getTerm(), o2.getTerm());
    }

}