001/*
002 * #%L
003 * Netarchivesuite - common
004 * %%
005 * Copyright (C) 2005 - 2014 The Royal Danish Library, the Danish State and University Library,
006 *             the National Library of France and the Austrian National Library.
007 * %%
008 * This program is free software: you can redistribute it and/or modify
009 * it under the terms of the GNU Lesser General Public License as
010 * published by the Free Software Foundation, either version 2.1 of the
011 * License, or (at your option) any later version.
012 * 
013 * This program is distributed in the hope that it will be useful,
014 * but WITHOUT ANY WARRANTY; without even the implied warranty of
015 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
016 * GNU General Lesser Public License for more details.
017 * 
018 * You should have received a copy of the GNU General Lesser Public
019 * License along with this program.  If not, see
020 * <http://www.gnu.org/licenses/lgpl-2.1.html>.
021 * #L%
022 */
023package dk.netarkivet.common.utils;
024
025import java.io.IOException;
026import java.util.ArrayList;
027import java.util.List;
028
029import org.apache.lucene.index.AtomicReaderContext;
030import org.apache.lucene.search.Collector;
031import org.apache.lucene.search.ScoreDoc;
032import org.apache.lucene.search.Scorer;
033
034/** Simple Collector to collect all results from Lucene query. */
035public class AllDocsCollector extends Collector {
036
037    private List<ScoreDoc> docs = new ArrayList<ScoreDoc>();
038    private Scorer scorer;
039    private int docBase;
040
041    @Override
042    public boolean acceptsDocsOutOfOrder() {
043        return true;
044    }
045
046    @Override
047    public void setScorer(Scorer scorer) {
048        this.scorer = scorer;
049    }
050
051    @Override
052    public void collect(int doc) throws IOException {
053        docs.add(new ScoreDoc(doc + docBase, scorer.score()));
054    }
055
056    public List<ScoreDoc> getHits() {
057        return docs;
058    }
059
060    public void reset() {
061        docs.clear();
062    }
063
064    @Override
065    public void setNextReader(AtomicReaderContext arg0) throws IOException {
066        this.docBase = arg0.docBase;
067
068    }
069
070}