/* Web Polygraph       http://www.web-polygraph.org/
 * (C) 2003-2006 The Measurement Factory
 * Licensed under the Apache License, Version 2.0 */

#include "base/polygraph.h"

#include "base/OLog.h"
#include "base/ILog.h"
#include "base/BStream.h"
#include "base/ILog.h"
#include "base/ObjId.h"
#include "runtime/ObjSelector.h"
#include "runtime/ObjWorld.h"


ObjWorld::ObjWorld() {
	reset();
}

void ObjWorld::reset() {
	theId.clear();
	theHotSet.reset();
	theSize = 0;
	theWss = -1;
}

bool ObjWorld::newer(const ObjWorld &w) const {
	Assert(id() == w.id());
	// we lack info to tell whether hotPos() is newer
	return size() > w.size() || wss() > w.wss();
}

bool ObjWorld::canRepeat() const {
	return theSize > 0 && theWss != 0;
}

bool ObjWorld::canProduce() const {
	return theWss != 0;
}

void ObjWorld::repeat(ObjId &oid, ObjSelector *sel) {
	// since selecting oid name affects oid type, type must not be pre-set
	Assert(oid.type() < 0);
	Assert(sel);

	oid.repeat(true);
	oid.world(theId);
	sel->choose(theSize, theWss, theHotSet.pos(), oid);
}

void ObjWorld::produce(ObjId &oid, RndGen &) {
	oid.repeat(false);
	oid.hot(false);
	oid.world(theId);
	oid.name(theSize+1);
	produced();
}

void ObjWorld::freezeWss() {
	theWss = theSize;
	theHotSet.syncPos(theSize, theWss);
}

void ObjWorld::produced() {
	++theSize;
	theHotSet.syncPos(theSize, theWss);
}

void ObjWorld::incWss(int &count) {
	if (theWss < 0)
		count++;
}

void ObjWorld::store(OLog &ol) const {
	ol << theSize << theWss;
}

void ObjWorld::load(ILog &il) {
	il >> theSize >> theWss;
}

void ObjWorld::store(OBStream &os) const {
	os << theId << theSize << theWss;
	os.puti(theHotSet.pos());
}

void ObjWorld::load(IBStream &is) {
	is >> theId >> theSize >> theWss;
	theHotSet.pos(is.geti());
}

ostream &ObjWorld::print(ostream &os) const {
	return os << theId 
		<< ' ' << theWss << '/' << theSize 
		<< ' ' << theHotSet.pos();
}