lucene-java-3.4.0/lucene/contrib/queryparser/src/java/org/apache/lucene/queryParser/standard/builders/MultiPhraseQueryNodeBuilder.java

   1 package org.apache.lucene.queryParser.standard.builders;
   2
   3 /**
   4  * Licensed to the Apache Software Foundation (ASF) under one or more
   5  * contributor license agreements.  See the NOTICE file distributed with
   6  * this work for additional information regarding copyright ownership.
   7  * The ASF licenses this file to You under the Apache License, Version 2.0
   8  * (the "License"); you may not use this file except in compliance with
   9  * the License.  You may obtain a copy of the License at
  10  *
  11  *     http://www.apache.org/licenses/LICENSE-2.0
  12  *
  13  * Unless required by applicable law or agreed to in writing, software
  14  * distributed under the License is distributed on an "AS IS" BASIS,
  15  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  16  * See the License for the specific language governing permissions and
  17  * limitations under the License.
  18  */
  19
  20 import java.util.LinkedList;
  21 import java.util.List;
  22 import java.util.TreeMap;
  23
  24 import org.apache.lucene.index.Term;
  25 import org.apache.lucene.queryParser.core.QueryNodeException;
  26 import org.apache.lucene.queryParser.core.builders.QueryTreeBuilder;
  27 import org.apache.lucene.queryParser.core.nodes.FieldQueryNode;
  28 import org.apache.lucene.queryParser.core.nodes.QueryNode;
  29 import org.apache.lucene.queryParser.standard.nodes.MultiPhraseQueryNode;
  30 import org.apache.lucene.search.MultiPhraseQuery;
  31 import org.apache.lucene.search.TermQuery;
  32
  33 /**
  34  * Builds a {@link MultiPhraseQuery} object from a {@link MultiPhraseQueryNode}
  35  * object.
  36  */
  37 public class MultiPhraseQueryNodeBuilder implements StandardQueryBuilder {
  38
  39   public MultiPhraseQueryNodeBuilder() {
  40     // empty constructor
  41   }
  42
  43   public MultiPhraseQuery build(QueryNode queryNode) throws QueryNodeException {
  44     MultiPhraseQueryNode phraseNode = (MultiPhraseQueryNode) queryNode;
  45
  46     MultiPhraseQuery phraseQuery = new MultiPhraseQuery();
  47
  48     List<QueryNode> children = phraseNode.getChildren();
  49
  50     if (children != null) {
  51       TreeMap<Integer, List<Term>> positionTermMap = new TreeMap<Integer, List<Term>>();
  52
  53       for (QueryNode child : children) {
  54         FieldQueryNode termNode = (FieldQueryNode) child;
  55         TermQuery termQuery = (TermQuery) termNode
  56             .getTag(QueryTreeBuilder.QUERY_TREE_BUILDER_TAGID);
  57         List<Term> termList = positionTermMap.get(termNode
  58             .getPositionIncrement());
  59
  60         if (termList == null) {
  61           termList = new LinkedList<Term>();
  62           positionTermMap.put(termNode.getPositionIncrement(), termList);
  63
  64         }
  65
  66         termList.add(termQuery.getTerm());
  67
  68       }
  69
  70       for (int positionIncrement : positionTermMap.keySet()) {
  71         List<Term> termList = positionTermMap.get(positionIncrement);
  72
  73         phraseQuery.add(termList.toArray(new Term[termList.size()]),
  74             positionIncrement);
  75
  76       }
  77
  78     }
  79
  80     return phraseQuery;
  81
  82   }
  83
  84 }