← Back to branch summary

~slub.team/goobi-indexserver/3.x

« back to all changes in this revision

Viewing changes to lucene/src/java/org/apache/lucene/index/FieldSortedTermVectorMapper.java

  • Committer: Sebastian Meyer
  • Date: 2012-08-03 09:12:40 UTC
  • Revision ID: sebastian.meyer@slub-dresden.de-20120803091240-x6861b0vabq1xror
https://launchpad.net/bugs/985487
Remove Lucene and Solr source code and add patches instead
Fix Bug #985487: Auto-suggestion for the search interface

Show diffs side-by-side

added added

removed removed

Lines of Context:
lucene/src/java/org/apache/lucene/index/FieldSortedTermVectorMapper.java
1
 
package org.apache.lucene.index;
2
 
 
3
 
import java.util.*;
4
 
 
5
 
/**
6
 
 * Copyright 2007 The Apache Software Foundation
7
 
 * <p/>
8
 
 * Licensed under the Apache License, Version 2.0 (the "License");
9
 
 * you may not use this file except in compliance with the License.
10
 
 * You may obtain a copy of the License at
11
 
 * <p/>
12
 
 * http://www.apache.org/licenses/LICENSE-2.0
13
 
 * <p/>
14
 
 * Unless required by applicable law or agreed to in writing, software
15
 
 * distributed under the License is distributed on an "AS IS" BASIS,
16
 
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
17
 
 * See the License for the specific language governing permissions and
18
 
 * limitations under the License.
19
 
 */
20
 
 
21
 
/**
22
 
 * For each Field, store a sorted collection of {@link TermVectorEntry}s
23
 
 * <p/>
24
 
 * This is not thread-safe.
25
 
 */
26
 
public class FieldSortedTermVectorMapper extends TermVectorMapper{
27
 
  private Map<String,SortedSet<TermVectorEntry>> fieldToTerms = new HashMap<String,SortedSet<TermVectorEntry>>();
28
 
  private SortedSet<TermVectorEntry> currentSet;
29
 
  private String currentField;
30
 
  private Comparator<TermVectorEntry> comparator;
31
 
 
32
 
  /**
33
 
   *
34
 
   * @param comparator A Comparator for sorting {@link TermVectorEntry}s
35
 
   */
36
 
  public FieldSortedTermVectorMapper(Comparator<TermVectorEntry> comparator) {
37
 
    this(false, false, comparator);
38
 
  }
39
 
 
40
 
 
41
 
  public FieldSortedTermVectorMapper(boolean ignoringPositions, boolean ignoringOffsets, Comparator<TermVectorEntry> comparator) {
42
 
    super(ignoringPositions, ignoringOffsets);
43
 
    this.comparator = comparator;
44
 
  }
45
 
 
46
 
  @Override
47
 
  public void map(String term, int frequency, TermVectorOffsetInfo[] offsets, int[] positions) {
48
 
    TermVectorEntry entry = new TermVectorEntry(currentField, term, frequency, offsets, positions);
49
 
    currentSet.add(entry);
50
 
  }
51
 
 
52
 
  @Override
53
 
  public void setExpectations(String field, int numTerms, boolean storeOffsets, boolean storePositions) {
54
 
    currentSet = new TreeSet<TermVectorEntry>(comparator);
55
 
    currentField = field;
56
 
    fieldToTerms.put(field, currentSet);
57
 
  }
58
 
 
59
 
  /**
60
 
   * Get the mapping between fields and terms, sorted by the comparator
61
 
   *
62
 
   * @return A map between field names and {@link java.util.SortedSet}s per field.  SortedSet entries are {@link TermVectorEntry}
63
 
   */
64
 
  public Map<String,SortedSet<TermVectorEntry>> getFieldToTerms() {
65
 
    return fieldToTerms;
66
 
  }
67
 
 
68
 
 
69
 
  public Comparator<TermVectorEntry> getComparator() {
70
 
    return comparator;
71
 
  }
72
 
}