/*------------------------------------------------------------------------------
* Copyright (C) 2003-2006 Ben van Klinken and the CLucene Team
* 
* Distributable under the terms of either the Apache License (Version 2.0) or 
* the GNU Lesser General Public License, as specified in the COPYING file.
------------------------------------------------------------------------------*/
#include "CLucene/StdHeader.h"
#include "SegmentMergeInfo.h"

#include "SegmentTermEnum.h"
#include "SegmentHeader.h"

CL_NS_DEF(index)

SegmentMergeInfo::SegmentMergeInfo(const int32_t b, TermEnum* te, IndexReader* r):
    reader(r),termEnum(te),base(b), docMap(NULL) {
//Func - Constructor
//Pre  - b >= 0
//       te contains a valid reference to a SegmentTermEnum instance
//       r contains a valid reference to a SegmentReader instance
//Post - The instance has been created

    CND_PRECONDITION(b >= 0, "b is a negative number");

    postings=NULL;
	term   = te->term();
}

SegmentMergeInfo::~SegmentMergeInfo(){
//Func - Destructor
//Pre  - true
//Post - The instance has been destroyed

  close();
}

int32_t* SegmentMergeInfo::getDocMap(){
	if ( docMap == NULL ){
      	// build array which maps document numbers around deletions 
		if (reader->hasDeletions()) {
			//Get the total number of documents managed by the reader including the deleted ones
			int32_t maxDoc = reader->maxDoc();
			//Create a map for all documents
			docMap = _CL_NEWARRAY(int32_t,maxDoc);
			int32_t j = 0;
			//Iterate through all the document numbers
			for (int32_t i = 0; i < maxDoc; i++) {
            //Check if document i is marked deleted
				if (reader->isDeleted(i)){
					//Document i has not been marked deleted so assign -1
					docMap[i] = -1;
				}else{
					docMap[i] = j++;
				}
			}
		}
	}
	return docMap;
}

TermPositions* SegmentMergeInfo::getPositions() {
    if (postings == NULL) {
    	postings = reader->termPositions();
    }
    return postings;
}


bool SegmentMergeInfo::next() {
//Func - Moves the current term of the enumeration termEnum to the next and term
//       points to this new current term
//Pre  - true
//Post - Returns true if the term has been moved to the next otherwise false
	if (termEnum->next()) {
		_CLDECDELETE(term);
		term = termEnum->term();
		return true;
	} else {
		_CLDECDELETE(term); //TODO: test HighFreqTerms errors with this
		term = NULL;
		return false;
	}
}

void SegmentMergeInfo::close() {
//Func - Closes the the resources
//Pre  - true
//Post - The resources have been closed

    //First make sure posting has been closed
    if ( postings != NULL ){
        postings->close();
        _CLVDELETE(postings); //todo: not a clucene object... should be
    }

    if ( termEnum != NULL ){
        termEnum->close();
        _CLDELETE(termEnum);
    }
	_CLDECDELETE(term);
	_CLDELETE_ARRAY(docMap);
}

CL_NS_END