#include "cpps/cpps.h"

namespace cpps
{

	void cpps_gc_add_barrier(C*c, cpps_regvar *v)
	{
		c->getbarrierlist()->insert(v);
	}

	void cpps_gc_remove_barrier(C*c, cpps_regvar *v)
	{
		c->getbarrierlist()->erase(v);
	}

	void cpps_gc_add_gen0(C*c,const cpps_value &v)
	{
		if (cpps_isclassvar(v.real()) || cpps_isstring(v.real()) || cpps_istuple(v.real()))
		{
			cpps_gc_add_gen0(c, cpps_to_cpps_cppsclassvar(v.real()));
			
		}
	}
	void cpps_gc_add_gen0(C*c, cpps_cppsclassvar *p)
	{
		//分配了内存时检测是否需要释放内存
		cpps_gc_check_step(c);

		c->_gen0lock->lock();
		c->setgen0size(c->getgen0size() + p->size());
		//新增到新生代
		c->getgen0()->insert(p);
		c->_gen0lock->unlock();
	}
	void cpps_gc_add_gen1(C*c, cpps_cppsclassvar *p)
	{
		c->setgen1size(c->getgen1size() + p->size());
		//新增到老生代
		c->getgen1()->insert(p);
	}
	void cpps_gc_check_gen_value(C*c, const cpps_value &v, bool checkchild, CLASSVARSET *oldgen, CLASSVARSET *newgen, size_t &size, CLASSVARSET &isCheck);
	void cpps_gc_check_child(const cpps_value &v, C* c, bool checkchild, CLASSVARSET * oldgen, CLASSVARSET * newgen, size_t &size, CLASSVARSET &isCheck)
	{	
		if (!v.value.domain->hasVar) return;

		for (VARLIST::iterator it = v.value.domain->varList->begin(); it != v.value.domain->varList->end(); ++it)
		{
			cpps_regvar *var = it->second;
			if (var->getval().tt == CPPS_TCLASSVAR|| var->getval().tt == CPPS_TSTRING)
			{
				cpps_gc_check_gen_value(c, var->getval(), checkchild, oldgen, newgen,size, isCheck);
			}
		}
	}
	//
	void cpps_gc_check_gen_value(C*c, const cpps_value &v, bool checkchild, CLASSVARSET *oldgen, CLASSVARSET *newgen, size_t &size, CLASSVARSET &isCheck)
	{
		if (v.tt == CPPS_TCLASSVAR)
		{
			if (isCheck.find((cpps_cppsclassvar*)v.value.domain) != isCheck.end()) return;//已经check过
			isCheck.insert((cpps_cppsclassvar*)v.value.domain);

			if (v.is_kindof<cpps_vector>())
			{
				cpps_cppsclassvar *pClsVar = (cpps_cppsclassvar *)v.value.domain;
				cpps_vector *pVector = (cpps_vector*)pClsVar->getclsptr();
				cpps_std_vector& realvector = pVector->realvector();//2019-12-14 防止外部正在调用begin 导致数据错乱
				for (cpps_std_vector::iterator it = realvector.begin(); it != realvector.end(); ++it)
				{
					const cpps_value& value = *it;
					cpps_gc_check_gen_value(c, value, checkchild, oldgen, newgen, size, isCheck);
				}
				CLASSVARSET::iterator it = oldgen->find(pClsVar);
				if (it != oldgen->end())
				{
					oldgen->erase(it);
					newgen->insert(pClsVar);
					size += pClsVar->size();
					//pClsVar->setgcLevel(1);//设置成老生代 哪怕也是老生代了 也设置一下
				}

			}
			else if (v.is_kindof<cpps_map>())
			{
				cpps_cppsclassvar *pClsVar = (cpps_cppsclassvar *)v.value.domain;
				cpps_map *pMap = (cpps_map*)pClsVar->getclsptr();
				cpps_hash_map& realmap = pMap->realmap();
				for (cpps_hash_map::iterator it = realmap.begin(); it != realmap.end(); ++it)
				{
					const cpps_value& value0 = it->first;
					const cpps_value& value1 = it->second;

					cpps_gc_check_gen_value(c, value0, checkchild, oldgen, newgen, size, isCheck);
					cpps_gc_check_gen_value(c, value1, checkchild, oldgen, newgen, size, isCheck);

				}
				CLASSVARSET::iterator it = oldgen->find(pClsVar);
				if (it != oldgen->end())
				{
					oldgen->erase(it);
					newgen->insert(pClsVar);
					size += pClsVar->size();
					//pClsVar->setgcLevel(1);//设置成老生代 哪怕也是老生代了 也设置一下
				}
			}
			else if (v.tt == CPPS_TCLASSVAR)
			{
				cpps_cppsclassvar *pClsVar = (cpps_cppsclassvar *)v.value.domain;
				CLASSVARSET::iterator it = oldgen->find(pClsVar);
				if (it != oldgen->end())
				{
					oldgen->erase(it);
					newgen->insert(pClsVar);
					size += pClsVar->size();
					//pClsVar->setgcLevel(1);//设置成老生代 哪怕也是老生代了 也设置一下
					
				}
				if (checkchild )
				{
					cpps_gc_check_child(v, c, checkchild, oldgen, newgen, size, isCheck);
				}
			}
		}
		else if(v.tt == CPPS_TSTRING)
		{
			cpps_cppsclassvar *pClsVar = static_cast<cpps_cppsclassvar *>(v.value.domain);
			CLASSVARSET::iterator it = oldgen->find(pClsVar);
			if (it != oldgen->end())
			{
				oldgen->erase(it);
				newgen->insert(pClsVar);
				size += pClsVar->size();
				//pClsVar->setgcLevel(1);//设置成老生代 哪怕也是老生代了 也设置一下
			}
		}
	}
	//检测新生代
	void cpps_gc_check_gen0(C *c)
	{

		c->_gen0lock->lock();
		CLASSVARSET isCheck;
	
		const cpps_value value = c->_G;
		size_t tmp = c->getgen1size();
		cpps_gc_check_child(value, c, true, c->getgen0(), c->getgen1(), tmp, isCheck);
		c->setgen1size(tmp);


		CLASSVARSET tempoldgen;
		size_t tempoldgensize = 0;

		for (phmap::flat_hash_set<cpps_regvar*>::iterator it = c->getbarrierlist()->begin();
			it != c->getbarrierlist()->end(); ++it)
		{
			cpps_regvar *v = *it;
			cpps_gc_check_gen_value(c, v->getval(), true, c->getgen0(), &tempoldgen, tempoldgensize, isCheck);
		}

		/*
		因为GC对象很多都是关联性,只有释放了主节点,子节点才会释放.
		所以牺牲一些效率,多遍历几次. 才可以真正清理出那些需要被释放的对象.
		*/
		bool hasdelete = true;
		usint32 cc = 0;
		while (hasdelete && cc < 10 ) {
			hasdelete = false;
			auto it = c->getgen0()->begin();
			while ( it != c->getgen0()->end() ) {
				cpps_cppsclassvar* pClsVar = *it;
				assert(pClsVar->count() >= 0);
				if (pClsVar->count() <= 0) {
					pClsVar->destory(c);
					pClsVar->release();
					hasdelete = true;
					it = c->getgen0()->erase(it);
				}
				else
					++it;
			}
			cc++;
		}

		for (auto it : *c->getgen0()) {
			cpps_cppsclassvar* pClsVar = it;
			tempoldgen.insert(pClsVar);
			tempoldgensize += pClsVar->size();
		}
		

		c->getgen0()->clear();
		c->getgen0()->swap( tempoldgen);
		c->setgen0size(tempoldgensize);

		c->_gen0lock->unlock();
	}

	//检测老生代 这样检测估计后面就卡死了。 还得想办法优化!!!
	void cpps_gc_check_gen1(C *c)
	{

		c->_gen0lock->lock();
		CLASSVARSET newgen;
		size_t newgensize = 0;
		CLASSVARSET isCheck1;
		CLASSVARSET isCheck;
		cpps_value value = c->_G;
		cpps_gc_check_child(value, c, true, c->getgen0(), &newgen, newgensize, isCheck);
		cpps_gc_check_child(value, c, true, c->getgen1(), &newgen, newgensize, isCheck1);



		CLASSVARSET tempoldgen;
		size_t tempoldgensize = 0;

		//先把新生代的检测了
		for (phmap::flat_hash_set<cpps_regvar*>::iterator it = c->getbarrierlist()->begin();
			it != c->getbarrierlist()->end(); ++it)
		{
			cpps_regvar *v = *it;
			cpps_gc_check_gen_value(c, v->getval(), true, c->getgen0(), &tempoldgen, tempoldgensize, isCheck);
			cpps_gc_check_gen_value(c, v->getval(), true, c->getgen1(), &tempoldgen, tempoldgensize, isCheck1);
		}



		bool hasdelete = true;
		usint32 cc = 0;
		while (hasdelete && cc < 5) {
			hasdelete = false;
			auto it = c->getgen0()->begin();
			while (it != c->getgen0()->end()) {
				cpps_cppsclassvar* pClsVar = *it;
				assert(pClsVar->count() >= 0);
				if (pClsVar->count() <= 0) {
					pClsVar->destory(c);
					pClsVar->release();
					hasdelete = true;
					it = c->getgen0()->erase(it);
				}
				else
					++it;
			}
			cc++;
		}
		for (auto it : *c->getgen0()) {
			cpps_cppsclassvar* pClsVar = it;
			tempoldgen.insert(pClsVar);
			tempoldgensize += pClsVar->size();
		}

		c->getgen0()->clear();
		c->getgen0()->swap(tempoldgen);
		c->setgen0size(tempoldgensize);


		//释放gen1里面的内存
		hasdelete = true;
		cc = 0;
		while (hasdelete && cc < 5) {
			hasdelete = false;
			auto it = c->getgen1()->begin();
			while (it != c->getgen1()->end()) {
				cpps_cppsclassvar* pClsVar = *it;
				assert(pClsVar->count() >= 0);
				if (pClsVar->count() <= 0) {
					pClsVar->destory(c);
					pClsVar->release();
					hasdelete = true;
					it = c->getgen1()->erase(it);
				}
				else
					++it;
			}
			cc++;
		}
		for (auto it : *c->getgen1()) {
			cpps_cppsclassvar* pClsVar = it;
			newgen.insert(pClsVar);
			newgensize += pClsVar->size();
		}
		c->getgen1()->clear();
		c->getgen1()->swap(newgen);
		c->setgen1size(newgensize);
		c->setlastgensize(c->getgen1size());



		c->_gen0lock->unlock();
	}
	void		gc_cleanup(C *c,bool cleanall )
	{
		

		bool hasdelete = true;
		int cc = 0;
		while (hasdelete && cc < 500) {
			hasdelete = false;
			auto it = c->getgen0()->begin();
			while (it != c->getgen0()->end()) {
				cpps_cppsclassvar* pClsVar = *it;
				assert(pClsVar->count() >= 0);
				if (pClsVar->count() <= 0) {
					pClsVar->destory(c);
					pClsVar->release();
					hasdelete = true;
					it = c->getgen0()->erase(it);
				}
				else
					++it;
			}
			cc++;
		}
		if (!c->getgen0()->empty()) {
#ifdef _DEBUG
			if(cleanall)printf("gen0 no cleanup:%d\r\n", (int)c->getgen0()->size());
#endif
			for (auto item : *c->getgen0()) {
				c->getgen1()->insert(item);
			}
		}

		c->getgen0()->clear();
		c->setgen0size(0);

	

		hasdelete = true;
		cc = 0;
		while (hasdelete && cc < 500) {
			hasdelete = false;
			auto it = c->getgen1()->begin();
			while (it != c->getgen1()->end()) {
				cpps_cppsclassvar* pClsVar = *it;
				assert(pClsVar->count() >= 0);
				if (pClsVar->count() <= 0) {
					pClsVar->destory(c);
					pClsVar->release();
					hasdelete = true;
					it = c->getgen1()->erase(it);
				}
				else
					++it;
			}
			cc++;
		}
		if (!c->getgen1()->empty()) {
#ifdef _DEBUG
			if(cleanall)printf("no cleanup:%d\r\n", (int)c->getgen1()->size());
#endif
			if (cleanall)
			{
				std::vector< cpps_cppsclassvar* > tmp;
				for (auto item : *c->getgen1()) {
					item->destory(c);
					item->release();
				}
			}
		}
		if (cleanall) {
			c->getgen1()->clear();
			c->setgen1size(0);
			c->setlastgensize(0);
			c->getbarrierlist()->clear();
		}
	}
	void gc_swap(C* src,C* dest) {
		for (auto item : *src->getgen1()) {
			cpps_gc_add_gen0(dest, item);
		}
	}
	std::string gcinfo(C *c)
	{
		//c->gclock.lock();
		std::string ret = "";
#ifdef _WIN32
#ifdef _WIN64
		char buffer[1024];
		sprintf(buffer,"gen0 memory %I64d b\n", c->getgen0size());
		ret += buffer;
		sprintf(buffer, "gen1 memory %I64d b\n", c->getgen1size());	//测试 200字节进行清理年轻代
		ret += buffer;
		sprintf(buffer, "current memory %I64d b\n", c->getgen0size() + c->getgen1size());
		ret += buffer;
		sprintf(buffer, "c->barrierList.size(): %I64d \n", c->getbarrierlist()->size());
		ret += buffer;
		sprintf(buffer, "c->gen1.size(): %I64d \n", c->getgen1()->size());
		ret += buffer;
		sprintf(buffer, "c->gen0.size(): %I64d \n", c->getgen0()->size());
		ret += buffer;

		for (CLASSVARSET::iterator it = c->getgen1()->begin(); it != c->getgen1()->end(); it++)
		{
			auto v = *it;
			if (v->domainname == "String")
			{
				cpps::string *tmpStr = (cpps::string *)v->getclsptr();
				printf("%s\r\n", tmpStr->__str.c_str());
			}
		}
#endif
#endif
		return ret;
	}
	void cpps_reggc(C *c)
	{
		cpps::_module(c,"GC")[
			def_inside("Collect", cpps_gc_check_gen0),
			def_inside("CollectAll", cpps_gc_check_gen1),
			def_inside("gcinfo", gcinfo)
		];
	}

}