com.jaeksoft.searchlib.result.collector.join.JoinDocCollector Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of opensearchserver Show documentation
Show all versions of opensearchserver Show documentation
OpenSearchServer is a powerful, enterprise-class, search engine program. Using the web user interface, the crawlers (web, file, database, ...) and the REST/RESTFul API you will be able to integrate quickly and easily advanced full-text search capabilities in your application. OpenSearchServer runs on Windows and Linux/Unix/BSD.
The newest version!
/**
* License Agreement for OpenSearchServer
*
* Copyright (C) 2012-2015 Emmanuel Keller / Jaeksoft
*
* http://www.open-search-server.com
*
* This file is part of OpenSearchServer.
*
* OpenSearchServer is free software: you can redistribute it and/or
* modify it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* OpenSearchServer is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with OpenSearchServer.
* If not, see implements
JoinCollectorInterface, JoinDocInterface, DocIdInterface {
private final int maxDoc;
private final ReaderAbstract[] foreignReaders;
final int[] srcIds;
final int joinResultSize;
private final int[][] foreignDocIdsArray;
private RoaringBitmap bitSet = null;
private final static int[][] EMPTY = new int[0][0];
JoinDocCollector() {
srcIds = ArrayUtils.EMPTY_INT_ARRAY;
foreignDocIdsArray = EMPTY;
joinResultSize = 0;
foreignReaders = null;
maxDoc = 0;
}
JoinDocCollector(DocIdInterface docs, int joinResultSize) {
this.srcIds = ArrayUtils.clone(docs.getIds());
this.foreignDocIdsArray = new int[srcIds.length][];
if (docs instanceof JoinDocCollector)
((JoinDocCollector) docs).copyForeignDocIdsArray(this);
this.joinResultSize = joinResultSize;
this.foreignReaders = new ReaderAbstract[joinResultSize];
this.maxDoc = docs.getMaxDoc();
}
private void copyForeignDocIdsArray(final JoinDocCollector joinDocCollector) {
if (foreignDocIdsArray == null)
return;
int i = 0;
for (int[] ids : foreignDocIdsArray)
joinDocCollector.foreignDocIdsArray[i++] = ids;
}
/**
* Copy only the valid item (other than -1)
*
* @param src
*/
private JoinDocCollector(final JoinDocCollector src) {
this.maxDoc = src.maxDoc;
this.joinResultSize = src.joinResultSize;
this.srcIds = new int[validSize(src.srcIds)];
this.foreignDocIdsArray = new int[this.srcIds.length][];
this.foreignReaders = new ReaderAbstract[joinResultSize];
int i1 = 0;
int i2 = 0;
for (int id : src.srcIds) {
if (id != -1) {
srcIds[i1] = id;
foreignDocIdsArray[i1++] = ArrayUtils
.clone(src.foreignDocIdsArray[i2]);
}
i2++;
}
System.arraycopy(src.getForeignReaders(), 0, foreignReaders, 0,
foreignReaders.length);
}
@Override
public JoinDocCollector duplicate(final AbstractBaseCollector base) {
return new JoinDocCollector((JoinDocCollector) base);
}
protected static final int validSize(final int[] ids) {
int i = 0;
for (int id : ids)
if (id != -1)
i++;
return i;
}
final public static int[][] copyForeignDocIdsArray(
final int[][] foreignDocIdsArray) {
int[][] neworeignDocIdsArray = new int[foreignDocIdsArray.length][];
int i = 0;
for (int[] foreignIds : foreignDocIdsArray)
neworeignDocIdsArray[i++] = ArrayUtils.clone(foreignIds);
return neworeignDocIdsArray;
}
final public static void swap(final int[][] foreignDocIdsArray,
final int pos1, final int pos2) {
int[] foreignDocIds = foreignDocIdsArray[pos1];
foreignDocIdsArray[pos1] = foreignDocIdsArray[pos2];
foreignDocIdsArray[pos2] = foreignDocIds;
}
@Override
final public void doSwap(final int pos1, final int pos2) {
int id = srcIds[pos1];
srcIds[pos1] = srcIds[pos2];
srcIds[pos2] = id;
swap(foreignDocIdsArray, pos1, pos2);
}
@Override
final public void doSetForeignDoc(final int pos, final int joinResultPos,
final int foreignDocId, float foreignScore) {
int[] foreignDocIds = foreignDocIdsArray[pos];
if (foreignDocIds == null) {
foreignDocIds = new int[joinResultSize];
Arrays.fill(foreignDocIds, -1);
foreignDocIdsArray[pos] = foreignDocIds;
}
foreignDocIds[joinResultPos] = foreignDocId;
}
final void setForeignDoc(final int pos, final int joinResultPos,
final int foreignDocId, float foreignScore) {
lastCollector.doSetForeignDoc(pos, joinResultPos, foreignDocId,
foreignScore);
}
final public static int getForeignDocIds(final int[][] foreignDocIdsArray,
int pos, int joinPosition) {
int[] foreignDocIds = foreignDocIdsArray[pos];
if (foreignDocIds == null)
return -1;
if (joinPosition >= foreignDocIds.length)
return -1;
return foreignDocIds[joinPosition];
}
@Override
final public int getForeignDocId(final int pos, final int joinPosition) {
return getForeignDocIds(foreignDocIdsArray, pos, joinPosition);
}
final public static DocIdInterface getDocIdInterface(int maxDoc,
int joinPosition, JoinDocCollector joinDocCollector)
throws IOException {
DocIdCollector docIdCollector = new DocIdCollector(maxDoc,
joinDocCollector.srcIds.length);
for (int[] foreignDocs : joinDocCollector.getForeignDocIdsArray())
if (foreignDocs != null)
docIdCollector.collectDoc(foreignDocs[joinPosition]);
return docIdCollector;
}
@Override
final public int[][] getForeignDocIdsArray() {
return foreignDocIdsArray;
}
@Override
public ReaderAbstract[] getForeignReaders() {
return foreignReaders;
}
@Override
public int getSize() {
return srcIds.length;
}
@Override
public int getMaxDoc() {
return maxDoc;
}
@Override
public int[] getIds() {
return srcIds;
}
@Override
public RoaringBitmap getBitSet() {
if (bitSet != null)
return bitSet;
bitSet = new RoaringBitmap();
for (int srcId : srcIds)
bitSet.add(srcId);
return bitSet;
}
}