/*
 *    MGQueryResult.java
 *    Copyright (C) 2002 New Zealand Digital Library, http://www.nzdl.org
 *
 *    This program 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 2 of the License, or
 *    (at your option) any later version.
 *
 *    This program 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 this program; if not, write to the Free Software
 *    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
 */
package org.greenstone.mg;


import java.util.Vector;


/** a java version of QueryResult for mg
 *
 * contains get methods for the java side, and simple
 * set methods to be called from c side
 *
 */
public class MGQueryResult
{
    /** the list of MGDocInfo */
    protected Vector docs_ = null;
    /** the list of MGTermInfo */
    protected Vector terms_ = null;
    /** the total number of docs found - not likely to be the size of docs_ */
    protected long total_num_docs_ = 0;


    MGQueryResult() {
	docs_ = new Vector();
	terms_ = new Vector();
    }


    /** clear the info from the last query
	should be called before setting any new docs/terms */
    public void clear() {
	total_num_docs_ = 0;
	docs_.clear();
	terms_.clear();
    }


    /** returns the result as a String - useful for printing out results */
    public String toString()
    {
	String result = "";
	result += "docs (ranks): ";
	for (int i = 0; i < docs_.size(); i++) {
	    if (i > 0) result += ", ";
	    result += ((MGDocInfo) docs_.elementAt(i)).toString();
	}
	result += "\nterms: ";
	for (int i = 0; i < terms_.size(); i++) {
	    if (i > 0) result += ", ";
	    result += ((MGTermInfo) terms_.elementAt(i)).toString();
	}
	result += "\nactual number of docs found = " + total_num_docs_;
	return result;
    }


    /** a shorter representation - just terms and total docs - not the 
	individual docnums and ranks */
    public String toShortString() {
	String result = "";
	result += "\nterms: ";
	for (int i = 0; i < terms_.size(); i++) {
	    if (i > 0) result += ", ";
	    result += ((MGTermInfo) terms_.elementAt(i)).toString();
	}
	result += "\nactual number of docs found = "+total_num_docs_;
	return result;
    }


    // set methods used by c++ code

    public void setTotalDocs(long num) {
	total_num_docs_=num;
    }


    public void addDoc(long doc, float rank) {
	MGDocInfo doc_info = new MGDocInfo(doc, rank);
	System.out.println("(Java) Added doc " + doc_info);

	docs_.add(doc_info);
    }


    // public void addTerm(String term, int stem, long match, long freq)
    public void addTerm(String term, String tag, int stem,
			long match, long freq, String[] equivs)
    {
	MGTermInfo ti = new MGTermInfo();
	ti.term_ = term;
	ti.tag_ = tag;
	ti.stem_method_ = stem;
	ti.match_docs_ = match;
	ti.term_freq_ = freq;
	if (equivs != null) {
	    for (int i = 0; i < equivs.length; i++) {
		ti.equiv_terms_.add(equivs[i]);
	    }
	}
	System.out.println("(Java) Added term " + ti);

	terms_.add(ti);
    }


    // Get methods for the java side - GS2MGSearch.java, GS2MGRetrieve.java
    public Vector getDocs() {
	return docs_;
    }


    public Vector getTerms() {
	return terms_;
    }


    public long getTotalDocs() {
	return total_num_docs_;
    }
}