trunk/ruta-ep-textruler/src/main/java/org/apache/uima/ruta/textruler/learner/rapier/RapierRulePriorityQueue.java

/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF 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.apache.uima.ruta.textruler.learner.rapier;

import java.util.Collection;
import java.util.Comparator;
import java.util.Iterator;
import java.util.PriorityQueue;

public class RapierRulePriorityQueue implements Iterable<RapierRule> {

  private PriorityQueue<RapierRule> ruleList;

  private PriorityQueue<RapierRule> reverseRuleList;

  private int maxSize;

  public RapierRulePriorityQueue(int maxSize) {
    this.maxSize = maxSize;

    ruleList = new PriorityQueue<RapierRule>(maxSize, new Comparator<RapierRule>() {
      public int compare(RapierRule o1, RapierRule o2) {
        if (o1.getPriority() > o2.getPriority())
          return 1;
        else if (o1.getPriority() < o2.getPriority())
          return -1;
        else
          return 0;
      }
    });
    this.reverseRuleList = new PriorityQueue<RapierRule>(maxSize, new Comparator<RapierRule>() {
      public int compare(RapierRule o1, RapierRule o2) {
        if (o1.getPriority() < o2.getPriority())
          return 1;
        else if (o1.getPriority() > o2.getPriority())
          return -1;
        else
          return 0;
      }
    });
  }

  public Iterator<RapierRule> iterator() {
    return ruleList.iterator();
  }

  public void clear() {
    ruleList.clear();
    reverseRuleList.clear();
  }

  public void addAll(Collection<RapierRule> rules) {
    for (RapierRule r : rules)
      add(r);
  }

  public void add(RapierRule rule) {
    ruleList.add(rule);
    reverseRuleList.add(rule);
    while (ruleList.size() > maxSize) {
      Object tail = reverseRuleList.peek();
      ruleList.remove(tail);
      reverseRuleList.remove(tail);
    }
    // TextRulerToolkit.log("ADD TO RULE LIST, SIZE="+ruleList.size()+"    revSize="+reverseRuleList.size());
    // for (RapierRule r : ruleList)
    // {
    // TextRulerToolkit.log("Rule Value: "+r.getPriority()+"  peek: "+ruleList.peek().getPriority());
    // }
  }

  public RapierRule peek() {
    return ruleList.peek();
  }

}