org.apache.lucene.search.SortField.getComparator java code examples

private FieldValueHitQueue(SortField[] fields, int size) {
 super(size);
 // When we get here, fields.length is guaranteed to be > 0, therefore no
 // need to check it again.
 
 // All these are required by this class's API - need to return arrays.
 // Therefore even in the case of a single comparator, create an array
 // anyway.
 this.fields = fields;
 int numComparators = fields.length;
 comparators = new FieldComparator<?>[numComparators];
 reverseMul = new int[numComparators];
 for (int i = 0; i < numComparators; ++i) {
  SortField field = fields[i];
  reverseMul[i] = field.reverse ? -1 : 1;
  comparators[i] = field.getComparator(size, i);
 }
}

public MergeSortQueue(Sort sort, TopDocs[] shardHits) {
 super(shardHits.length);
 this.shardHits = new ScoreDoc[shardHits.length][];
 for(int shardIDX=0;shardIDX<shardHits.length;shardIDX++) {
  final ScoreDoc[] shard = shardHits[shardIDX].scoreDocs;
  //System.out.println("  init shardIdx=" + shardIDX + " hits=" + shard);
  if (shard != null) {
   this.shardHits[shardIDX] = shard;
   // Fail gracefully if API is misused:
   for(int hitIDX=0;hitIDX<shard.length;hitIDX++) {
    final ScoreDoc sd = shard[hitIDX];
    if (!(sd instanceof FieldDoc)) {
     throw new IllegalArgumentException("shard " + shardIDX + " was not sorted by the provided Sort (expected FieldDoc but got ScoreDoc)");
    }
    final FieldDoc fd = (FieldDoc) sd;
    if (fd.fields == null) {
     throw new IllegalArgumentException("shard " + shardIDX + " did not set sort field values (FieldDoc.fields is null); you must pass fillFields=true to IndexSearcher.search on each shard");
    }
   }
  }
 }
 final SortField[] sortFields = sort.getSort();
 comparators = new FieldComparator[sortFields.length];
 reverseMul = new int[sortFields.length];
 for(int compIDX=0;compIDX<sortFields.length;compIDX++) {
  final SortField sortField = sortFields[compIDX];
  comparators[compIDX] = sortField.getComparator(1, compIDX);
  reverseMul[compIDX] = sortField.getReverse() ? -1 : 1;
 }
}

comparators[i] = fields[i].getComparator(1, i).getLeafComparator(readerContext);

MergeSortQueue(Sort sort, CollapseTopFieldDocs[] shardHits) {
  super(shardHits.length);
  this.shardHits = new ScoreDoc[shardHits.length][];
  for (int shardIDX = 0; shardIDX < shardHits.length; shardIDX++) {
    final ScoreDoc[] shard = shardHits[shardIDX].scoreDocs;
    if (shard != null) {
      this.shardHits[shardIDX] = shard;
      // Fail gracefully if API is misused:
      for (int hitIDX = 0; hitIDX < shard.length; hitIDX++) {
        final ScoreDoc sd = shard[hitIDX];
        final FieldDoc gd = (FieldDoc) sd;
        assert gd.fields != null;
      }
    }
  }
  final SortField[] sortFields = sort.getSort();
  comparators = new FieldComparator[sortFields.length];
  reverseMul = new int[sortFields.length];
  for (int compIDX = 0; compIDX < sortFields.length; compIDX++) {
    final SortField sortField = sortFields[compIDX];
    comparators[compIDX] = sortField.getComparator(1, compIDX);
    reverseMul[compIDX] = sortField.getReverse() ? -1 : 1;
  }
}

public SearchAfterSortedDocQuery(Sort sort, FieldDoc after) {
  if (sort.getSort().length != after.fields.length) {
    throw new IllegalArgumentException("after doc  has " + after.fields.length + " value(s) but sort has "
        + sort.getSort().length + ".");
  }
  this.sort = sort;
  this.after = after;
  int numFields = sort.getSort().length;
  this.fieldComparators = new FieldComparator[numFields];
  this.reverseMuls = new int[numFields];
  for (int i = 0; i < numFields; i++) {
    SortField sortField = sort.getSort()[i];
    FieldComparator<?> fieldComparator = sortField.getComparator(1, i);
    @SuppressWarnings("unchecked")
    FieldComparator<Object> comparator = (FieldComparator<Object>) fieldComparator;
    comparator.setTopValue(after.fields[i]);
    fieldComparators[i] = fieldComparator;
    reverseMuls[i] = sortField.getReverse() ? -1 : 1;
  }
}

private FieldValueHitQueue(SortField[] fields, int size) {
 super(size);
 // When we get here, fields.length is guaranteed to be > 0, therefore no
 // need to check it again.
 
 // All these are required by this class's API - need to return arrays.
 // Therefore even in the case of a single comparator, create an array
 // anyway.
 this.fields = fields;
 int numComparators = fields.length;
 comparators = new FieldComparator<?>[numComparators];
 reverseMul = new int[numComparators];
 for (int i = 0; i < numComparators; ++i) {
  SortField field = fields[i];
  reverseMul[i] = field.reverse ? -1 : 1;
  comparators[i] = field.getComparator(size, i);
 }
}

private FieldValueHitQueue(SortField[] fields, int size) throws IOException {
 super(size);
 // When we get here, fields.length is guaranteed to be > 0, therefore no
 // need to check it again.
 
 // All these are required by this class's API - need to return arrays.
 // Therefore even in the case of a single comparator, create an array
 // anyway.
 this.fields = fields;
 int numComparators = fields.length;
 comparators = new FieldComparator<?>[numComparators];
 reverseMul = new int[numComparators];
 for (int i = 0; i < numComparators; ++i) {
  SortField field = fields[i];
  reverseMul[i] = field.reverse ? -1 : 1;
  comparators[i] = field.getComparator(size, i);
 }
}

private FieldValueHitQueue(SortField[] fields, int size) throws IOException {
 super(size);
 // When we get here, fields.length is guaranteed to be > 0, therefore no
 // need to check it again.
 
 // All these are required by this class's API - need to return arrays.
 // Therefore even in the case of a single comparator, create an array
 // anyway.
 this.fields = fields;
 int numComparators = fields.length;
 comparators = new FieldComparator<?>[numComparators];
 reverseMul = new int[numComparators];
 for (int i = 0; i < numComparators; ++i) {
  SortField field = fields[i];
  reverseMul[i] = field.reverse ? -1 : 1;
  comparators[i] = field.getComparator(size, i);
 }
}

@SuppressWarnings({"unchecked", "rawtypes"})
public GroupComparator(Sort groupSort) {
 final SortField[] sortFields = groupSort.getSort();
 comparators = new FieldComparator[sortFields.length];
 reversed = new int[sortFields.length];
 for (int compIDX = 0; compIDX < sortFields.length; compIDX++) {
  final SortField sortField = sortFields[compIDX];
  comparators[compIDX] = sortField.getComparator(1, compIDX);
  reversed[compIDX] = sortField.getReverse() ? -1 : 1;
 }
}

@SuppressWarnings({"unchecked", "rawtypes"})
public GroupComparator(Sort groupSort) throws IOException {
 final SortField[] sortFields = groupSort.getSort();
 comparators = new FieldComparator[sortFields.length];
 reversed = new int[sortFields.length];
 for (int compIDX = 0; compIDX < sortFields.length; compIDX++) {
  final SortField sortField = sortFields[compIDX];
  comparators[compIDX] = sortField.getComparator(1, compIDX);
  reversed[compIDX] = sortField.getReverse() ? -1 : 1;
 }
}

public MergeSortQueue(Sort sort, TopDocs[] shardHits) throws IOException {
 super(shardHits.length);
 this.shardHits = new ScoreDoc[shardHits.length][];
 for(int shardIDX=0;shardIDX<shardHits.length;shardIDX++) {
  final ScoreDoc[] shard = shardHits[shardIDX].scoreDocs;
  //System.out.println("  init shardIdx=" + shardIDX + " hits=" + shard);
  if (shard != null) {
   this.shardHits[shardIDX] = shard;
   // Fail gracefully if API is misused:
   for(int hitIDX=0;hitIDX<shard.length;hitIDX++) {
    final ScoreDoc sd = shard[hitIDX];
    if (!(sd instanceof FieldDoc)) {
     throw new IllegalArgumentException("shard " + shardIDX + " was not sorted by the provided Sort (expected FieldDoc but got ScoreDoc)");
    }
    final FieldDoc fd = (FieldDoc) sd;
    if (fd.fields == null) {
     throw new IllegalArgumentException("shard " + shardIDX + " did not set sort field values (FieldDoc.fields is null); you must pass fillFields=true to IndexSearcher.search on each shard");
    }
   }
  }
 }
 final SortField[] sortFields = sort.getSort();
 comparators = new FieldComparator[sortFields.length];
 reverseMul = new int[sortFields.length];
 for(int compIDX=0;compIDX<sortFields.length;compIDX++) {
  final SortField sortField = sortFields[compIDX];
  comparators[compIDX] = sortField.getComparator(1, compIDX);
  reverseMul[compIDX] = sortField.getReverse() ? -1 : 1;
 }
}

MergeSortQueue(Sort sort, CollapseTopFieldDocs[] shardHits) {
  super(shardHits.length);
  this.shardHits = new ScoreDoc[shardHits.length][];
  for (int shardIDX = 0; shardIDX < shardHits.length; shardIDX++) {
    final ScoreDoc[] shard = shardHits[shardIDX].scoreDocs;
    if (shard != null) {
      this.shardHits[shardIDX] = shard;
      // Fail gracefully if API is misused:
      for (int hitIDX = 0; hitIDX < shard.length; hitIDX++) {
        final ScoreDoc sd = shard[hitIDX];
        final FieldDoc gd = (FieldDoc) sd;
        assert gd.fields != null;
      }
    }
  }
  final SortField[] sortFields = sort.getSort();
  comparators = new FieldComparator[sortFields.length];
  reverseMul = new int[sortFields.length];
  for (int compIDX = 0; compIDX < sortFields.length; compIDX++) {
    final SortField sortField = sortFields[compIDX];
    comparators[compIDX] = sortField.getComparator(1, compIDX);
    reverseMul[compIDX] = sortField.getReverse() ? -1 : 1;
  }
}

MergeSortQueue(Sort sort, CollapseTopFieldDocs[] shardHits) {
  super(shardHits.length);
  this.shardHits = new ScoreDoc[shardHits.length][];
  for (int shardIDX = 0; shardIDX < shardHits.length; shardIDX++) {
    final ScoreDoc[] shard = shardHits[shardIDX].scoreDocs;
    if (shard != null) {
      this.shardHits[shardIDX] = shard;
      // Fail gracefully if API is misused:
      for (int hitIDX = 0; hitIDX < shard.length; hitIDX++) {
        final ScoreDoc sd = shard[hitIDX];
        final FieldDoc gd = (FieldDoc) sd;
        assert gd.fields != null;
      }
    }
  }
  final SortField[] sortFields = sort.getSort();
  comparators = new FieldComparator[sortFields.length];
  reverseMul = new int[sortFields.length];
  for (int compIDX = 0; compIDX < sortFields.length; compIDX++) {
    final SortField sortField = sortFields[compIDX];
    comparators[compIDX] = sortField.getComparator(1, compIDX);
    reverseMul[compIDX] = sortField.getReverse() ? -1 : 1;
  }
}

MergeSortQueue(Sort sort, CollapseTopFieldDocs[] shardHits) {
  super(shardHits.length);
  this.shardHits = new ScoreDoc[shardHits.length][];
  for (int shardIDX = 0; shardIDX < shardHits.length; shardIDX++) {
    final ScoreDoc[] shard = shardHits[shardIDX].scoreDocs;
    if (shard != null) {
      this.shardHits[shardIDX] = shard;
      // Fail gracefully if API is misused:
      for (int hitIDX = 0; hitIDX < shard.length; hitIDX++) {
        final ScoreDoc sd = shard[hitIDX];
        final FieldDoc gd = (FieldDoc) sd;
        assert gd.fields != null;
      }
    }
  }
  final SortField[] sortFields = sort.getSort();
  comparators = new FieldComparator[sortFields.length];
  reverseMul = new int[sortFields.length];
  for (int compIDX = 0; compIDX < sortFields.length; compIDX++) {
    final SortField sortField = sortFields[compIDX];
    comparators[compIDX] = sortField.getComparator(1, compIDX);
    reverseMul[compIDX] = sortField.getReverse() ? -1 : 1;
  }
}

protected SortingGroupHead(Sort sort, T groupValue, int doc, LeafReaderContext context, Scorer scorer) throws IOException {
 super(groupValue, doc, context.docBase);
 final SortField[] sortFields = sort.getSort();
 comparators = new FieldComparator[sortFields.length];
 leafComparators = new LeafFieldComparator[sortFields.length];
 for (int i = 0; i < sortFields.length; i++) {
  comparators[i] = sortFields[i].getComparator(1, i);
  leafComparators[i] = comparators[i].getLeafComparator(context);
  leafComparators[i].setScorer(scorer);
  leafComparators[i].copy(0, doc);
  leafComparators[i].setBottom(0);
 }
}

@SuppressWarnings({"unchecked","rawtypes"})
private GroupHead(MutableValue groupValue, Sort sort, int doc) throws IOException {
 super(groupValue, doc + readerContext.docBase);
 final SortField[] sortFields = sort.getSort();
 comparators = new FieldComparator[sortFields.length];
 leafComparators = new LeafFieldComparator[sortFields.length];
 for (int i = 0; i < sortFields.length; i++) {
  comparators[i] = sortFields[i].getComparator(1, i);
  leafComparators[i] = comparators[i].getLeafComparator(readerContext);
  leafComparators[i].setScorer(scorer);
  leafComparators[i].copy(0, doc);
  leafComparators[i].setBottom(0);
 }
}

public SearchAfterSortedDocQuery(Sort sort, FieldDoc after) {
  if (sort.getSort().length != after.fields.length) {
    throw new IllegalArgumentException("after doc  has " + after.fields.length + " value(s) but sort has "
        + sort.getSort().length + ".");
  }
  this.sort = sort;
  this.after = after;
  int numFields = sort.getSort().length;
  this.fieldComparators = new FieldComparator[numFields];
  this.reverseMuls = new int[numFields];
  for (int i = 0; i < numFields; i++) {
    SortField sortField = sort.getSort()[i];
    FieldComparator<?> fieldComparator = sortField.getComparator(1, i);
    @SuppressWarnings("unchecked")
    FieldComparator<Object> comparator = (FieldComparator<Object>) fieldComparator;
    comparator.setTopValue(after.fields[i]);
    fieldComparators[i] = fieldComparator;
    reverseMuls[i] = sortField.getReverse() ? -1 : 1;
  }
}

@SuppressWarnings({"unchecked", "rawtypes"})
GroupHead(BytesRef groupValue, Sort sort, int doc) throws IOException {
 super(groupValue, doc + readerContext.docBase);
 final SortField[] sortFields = sort.getSort();
 comparators = new FieldComparator[sortFields.length];
 leafComparators = new LeafFieldComparator[sortFields.length];
 for (int i = 0; i < sortFields.length; i++) {
  comparators[i] = sortFields[i].getComparator(1, i);
  leafComparators[i] = comparators[i].getLeafComparator(readerContext);
  leafComparators[i].setScorer(scorer);
  leafComparators[i].copy(0, doc);
  leafComparators[i].setBottom(0);
 }
}

public SearchAfterSortedDocQuery(Sort sort, FieldDoc after) {
  if (sort.getSort().length != after.fields.length) {
    throw new IllegalArgumentException("after doc  has " + after.fields.length + " value(s) but sort has "
        + sort.getSort().length + ".");
  }
  this.sort = sort;
  this.after = after;
  int numFields = sort.getSort().length;
  this.fieldComparators = new FieldComparator[numFields];
  this.reverseMuls = new int[numFields];
  for (int i = 0; i < numFields; i++) {
    SortField sortField = sort.getSort()[i];
    FieldComparator<?> fieldComparator = sortField.getComparator(1, i);
    @SuppressWarnings("unchecked")
    FieldComparator<Object> comparator = (FieldComparator<Object>) fieldComparator;
    comparator.setTopValue(after.fields[i]);
    fieldComparators[i] = fieldComparator;
    reverseMuls[i] = sortField.getReverse() ? -1 : 1;
  }
}

private void initializeFieldComparators(Sort sort, int numberOfHits, IndexReader indexReader, float[] scores) {
 try {
  scorer = new PredefinedScorer(scores == null ? new float[0] : scores);
  if (sort.getSort().length == 1 && scores != null) {
   fieldComparators[0] = new FloatValueFieldComparator(scores);
   descending[0] = sort.getSort()[0].getReverse();
   return;
  }
  for (int i = 0; i < sort.getSort().length; i++) {
   SortField sortField = sort.getSort()[i];
   fieldComparators[i] = sortField.getComparator(numberOfHits, i);
   descending[i] = sortField.getReverse();
   fieldComparators[i].setNextReader(indexReader, 0);
   fieldComparators[i].setScorer(scorer);
  }
 } catch (IOException e) {
  throw new RuntimeException(e);
 }
}

Javadoc

Returns the FieldComparator to use for sorting.

Popular methods of SortField

<init>
Creates a sort, possibly in reverse, by terms in the given field where the type of term value is det
getField
Returns the name of the field. Could return null if the sort is by SCORE or DOC.
getReverse
Returns whether the sort should be reversed.
getType
Returns the type of contents in the field.
setMissingValue
Set the value to use for documents that don't have a value.
equals
Returns true if o is equal to this. If a FieldComparatorSource was provided, it must properly implem
getComparatorSource
Returns the FieldComparatorSource used for custom sorting
getMissingValue
Return the value to use for documents that don't have a value. A value of null indicates that defaul
toString
hashCode
Returns a hash code for this SortField instance. If a FieldComparatorSource was provided, it must pr
needsScores
Whether the relevance score is needed to sort documents.
getFactory

Popular in Java

Running tasks concurrently on multiple threads
requestLocationUpdates (LocationManager)
getOriginalFilename (MultipartFile)
Return the original filename in the client's filesystem.This may contain path information depending
addToBackStack (FragmentTransaction)
FileOutputStream (java.io)
An output stream that writes bytes to a file. If the output file exists, it can be replaced or appen
InputStream (java.io)
A readable source of bytes.Most clients will use input streams that read data from the file system (
Runnable (java.lang)
Represents a command that can be executed. Often used to run code in a different Thread.
URLConnection (java.net)
A connection to a URL for reading or writing. For HTTP connections, see HttpURLConnection for docume
Enumeration (java.util)
A legacy iteration interface.New code should use Iterator instead. Iterator replaces the enumeration
TreeSet (java.util)
TreeSet is an implementation of SortedSet. All optional operations (adding and removing) are support
Top 17 Plugins for Android Studio

How to use getComparatormethodin org.apache.lucene.search.SortField

Best Java code snippets using org.apache.lucene.search.SortField.getComparator (Showing top 20 results out of 315)

How to use
getComparator
method
in
org.apache.lucene.search.SortField