org.terrier.matching.FatQueryResultSet Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of terrier-learning Show documentation
The newest version!
/*
 * Terrier - Terabyte Retriever 
 * Webpage: http://terrier.org 
 * Contact: terrier{a.}dcs.gla.ac.uk
 * University of Glasgow - School of Computing Science
 * http://www.gla.ac.uk/
 * 
 * The contents of this file are subject to the Mozilla Public License
 * Version 1.1 (the "License"); you may not use this file except in
 * compliance with the License. You may obtain a copy of the License at
 * http://www.mozilla.org/MPL/
 *
 * Software distributed under the License is distributed on an "AS IS"
 * basis, WITHOUT WARRANTY OF ANY KIND, either express or implied. See
 * the License for the specific language governing rights and limitations
 * under the License.
 *
 * The Original Code is FatQueryResultSet.java.
 *
 * The Original Code is Copyright (C) 2004-2020 the University of Glasgow.
 * All Rights Reserved.
 *
 * Contributor(s):
 *   Craig Macdonald 
 */

package org.terrier.matching;

import gnu.trove.TIntIntHashMap;

import java.io.DataInput;
import java.io.DataOutput;
import java.io.IOException;
import java.util.Set;

import org.terrier.structures.CollectionStatistics;
import org.terrier.structures.EntryStatistics;
import org.terrier.structures.postings.WritablePosting;
import org.terrier.structures.postings.Posting;
import org.terrier.utility.StableSort;

/** An implementation of {@link FatResultSet}.
 * 
 * @author Craig Macdonald
 * @since 4.0
 */
public class FatQueryResultSet extends QueryResultSet implements FatResultSet {

	
	CollectionStatistics collStats;
	WritablePosting[][] postings;
	String[] queryTerms;
	double[] keyFrequency;
	EntryStatistics[] entryStats;
	Set[] tags;
	
	public FatQueryResultSet(
			int numberOfDocuments,
			CollectionStatistics cs,
			String[] qs,
			EntryStatistics[] es,
			double[] ks,
			Set[] ts
			) 
	{
		super(numberOfDocuments);
		collStats = cs;
		queryTerms = qs;
		entryStats = es;
		keyFrequency = ks;
		postings = new WritablePosting[numberOfDocuments][];
		tags = ts;
	}

	private static final long serialVersionUID = 1L;

	/** {@inheritDoc} */
	@Override
	public WritablePosting[][] getPostings() {
		return postings;
	}
	
	/** {@inheritDoc} */
	@Override
	public double[] getKeyFrequencies() {
		return keyFrequency;
	}
	
	/** {@inheritDoc} */
	@Override
	public EntryStatistics[] getEntryStatistics() {
		return entryStats;
	}
	
	/** {@inheritDoc} */
	@Override
	public CollectionStatistics getCollectionStatistics()
	{
		return collStats;
	}
	
	/** {@inheritDoc} */	
	@Override
	public String[] getQueryTerms() {
		return queryTerms;
	}

	/** {@inheritDoc} */
	@Override
	public void readFields(DataInput in) throws IOException {
		FatUtils.readFields(this, in);
	}

	/** {@inheritDoc} */
	@Override
	public void write(DataOutput out) throws IOException {
		FatUtils.write(this, out);
	}

	/** {@inheritDoc} */
	@Override
	public FatResultSet getResultSet(int start, int length) {
		length = length < docids.length ? length : docids.length;
		FatQueryResultSet resultSet = new FatQueryResultSet(
				length, 
				this.getCollectionStatistics(), 
				this.getQueryTerms(), 
				this.getEntryStatistics(), 
				this.getKeyFrequencies(),
				this.getTags());
		resultSet.setExactResultSize(this.exactResultSize);
		System.arraycopy(docids, start, resultSet.getDocids(), 0, length);
		System.arraycopy(scores, start, resultSet.getScores(), 0, length);
		System.arraycopy(occurrences, start, resultSet.getOccurrences(), 0, length);
		System.arraycopy(postings, start, resultSet.getPostings(), 0, length);
		resultSet.metadata = new String[this.metadata.length][];
		resultSet.metaMap = this.metaMap.clone();
		for(int i=0;i[] getTags() {
		return tags;
	}

	@Override
	public void setTags(Set[] ks) {
		this.tags = ks;
	}
}