io.github.repir.Strategy.Operator.ProximityOperatorOrdered Maven / Gradle / Ivy

Go to download
package io.github.repir.Strategy.Operator;

import io.github.repir.Strategy.Tools.MatchSetPositional;
import io.github.repir.Strategy.Tools.ProximityOccurrence;
import io.github.repir.Strategy.Tools.MatchSetLength;
import io.github.repir.Retriever.Document;
import io.github.repir.Strategy.GraphRoot;
import io.github.repir.Strategy.Tools.ProximityOrderedSet;
import io.github.repir.tools.lib.Log;
import java.util.ArrayList;

/**
 * This {@link ProximityOperator} counts ordered co-occurrence of the contained
 * {@link Operator}s in the same document. 
 * 
 * @author jeroen
 */
public class ProximityOperatorOrdered extends ProximityOperator {

   public static Log log = new Log(ProximityOperatorOrdered.class);
   ProximityOrderedSet termpositions;
   
   public ProximityOperatorOrdered(GraphRoot root, ArrayList list) {
      super(root, list);
   }

   @Override
   public Operator clone(GraphRoot newmodel) {
      ArrayList list = new ArrayList();
      for (Operator n : containednodes) {
         list.add(n.clone(newmodel));
      }
      ProximityOperatorOrdered f = new ProximityOperatorOrdered(newmodel, list);
      f.parent = parent;
      return f;
   }

   public void setMinimalSpan() {
      span = 0;
      for (Operator o : containednodes)
         span += o.span;
   }
   
   @Override
   public void process(Document doc) {
      if (termpositions.hasProximityMatches(doc)) {
         MatchSetLength matches = new MatchSetLength();
         do {
            if (termpositions.tpi[containednodes.size() - 1].current - termpositions.tpi[0].current
                 <= span - termpositions.tpi[containednodes.size() - 1].span) {
               matches.add(new ProximityOccurrence(termpositions.tpi[0].current, termpositions.tpi[containednodes.size() - 1].current - termpositions.tpi[0].current + 1));
            }
         } while( termpositions.next() );

         MatchSetPositional result = matches.purgeOccurrencesOccupyingSameSpace();

         pos = new int[result.size()];
         dist = new int[result.size()];
         int p = 0;
         for (ProximityOccurrence m : result) {
            pos[p] = m.pos;
            dist[p++] = m.span;
         }
         frequency = pos.length;
      } else {
         pos = ZEROPOS;
         dist = ZEROPOS;
         frequency = 0;
      }
   }

  @Override
   protected void prepareRetrieval() {
      termpositions = new ProximityOrderedSet(new ArrayList(containednodes));
      termpositions.setMaxSpan(span);
   }

   @Override
   public String postReformUnweighted() {
      StringBuilder sb = new StringBuilder();
      sb.append("(").append(super.postReformUnweighted()).append(")");
      return sb.toString();
   }
   
   @Override
   public String toTermString() {
      return postReformUnweighted();
   }
}