lucene-java-3.5.0/lucene/contrib/analyzers/common/src/java/org/apache/lucene/analysis/payloads/NumericPayloadTokenFilter.java

   1 package org.apache.lucene.analysis.payloads;
   2 /**
   3  * Licensed to the Apache Software Foundation (ASF) under one or more
   4  * contributor license agreements.  See the NOTICE file distributed with
   5  * this work for additional information regarding copyright ownership.
   6  * The ASF licenses this file to You under the Apache License, Version 2.0
   7  * (the "License"); you may not use this file except in compliance with
   8  * the License.  You may obtain a copy of the License at
   9  *
  10  *     http://www.apache.org/licenses/LICENSE-2.0
  11  *
  12  * Unless required by applicable law or agreed to in writing, software
  13  * distributed under the License is distributed on an "AS IS" BASIS,
  14  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  15  * See the License for the specific language governing permissions and
  16  * limitations under the License.
  17  */
  18
  19
  20 import org.apache.lucene.analysis.TokenFilter;
  21 import org.apache.lucene.analysis.TokenStream;
  22 import org.apache.lucene.analysis.tokenattributes.PayloadAttribute;
  23 import org.apache.lucene.analysis.tokenattributes.TypeAttribute;
  24 import org.apache.lucene.index.Payload;
  25
  26 import java.io.IOException;
  27
  28
  29 /**
  30  * Assigns a payload to a token based on the {@link org.apache.lucene.analysis.Token#type()}
  31  *
  32  **/
  33 public class NumericPayloadTokenFilter extends TokenFilter {
  34
  35   private String typeMatch;
  36   private Payload thePayload;
  37
  38   private final PayloadAttribute payloadAtt = addAttribute(PayloadAttribute.class);
  39   private final TypeAttribute typeAtt = addAttribute(TypeAttribute.class);
  40
  41   public NumericPayloadTokenFilter(TokenStream input, float payload, String typeMatch) {
  42     super(input);
  43     //Need to encode the payload
  44     thePayload = new Payload(PayloadHelper.encodeFloat(payload));
  45     this.typeMatch = typeMatch;
  46   }
  47
  48   @Override
  49   public final boolean incrementToken() throws IOException {
  50     if (input.incrementToken()) {
  51       if (typeAtt.type().equals(typeMatch))
  52         payloadAtt.setPayload(thePayload);
  53       return true;
  54     } else {
  55       return false;
  56     }
  57   }
  58 }