/** 
 * @file llfloatermemleak.cpp
 * @brief LLFloatermemleak class definition
 *
 * $LicenseInfo:firstyear=2007&license=viewerlgpl$
 * Second Life Viewer Source Code
 * Copyright (C) 2010, Linden Research, Inc.
 * 
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation;
 * version 2.1 of the License only.
 * 
 * This library 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
 * Lesser General Public License for more details.
 * 
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
 * 
 * Linden Research, Inc., 945 Battery Street, San Francisco, CA  94111  USA
 * $/LicenseInfo$
 */

#include "llviewerprecompiledheaders.h"

#include "llfloatermemleak.h"

#include "lluictrlfactory.h"
#include "llbutton.h"
#include "llspinctrl.h"
#include "llresmgr.h"

#include "llmath.h"
#include "llviewerwindow.h"

U32 LLFloaterMemLeak::sMemLeakingSpeed = 0 ; //bytes leaked per frame
U32 LLFloaterMemLeak::sMaxLeakedMem = 0 ; //maximum allowed leaked memory
U32 LLFloaterMemLeak::sTotalLeaked = 0 ;
S32 LLFloaterMemLeak::sStatus = LLFloaterMemLeak::STOP ;
BOOL LLFloaterMemLeak::sbAllocationFailed = FALSE ;

extern BOOL gSimulateMemLeak;

LLFloaterMemLeak::LLFloaterMemLeak(const LLSD& key)
	: LLFloater(key)
{
	setTitle("Memory Leaking Simulation Floater");
	mCommitCallbackRegistrar.add("MemLeak.ChangeLeakingSpeed",	boost::bind(&LLFloaterMemLeak::onChangeLeakingSpeed, this));
	mCommitCallbackRegistrar.add("MemLeak.ChangeMaxMemLeaking",	boost::bind(&LLFloaterMemLeak::onChangeMaxMemLeaking, this));
	mCommitCallbackRegistrar.add("MemLeak.Start",	boost::bind(&LLFloaterMemLeak::onClickStart, this));
	mCommitCallbackRegistrar.add("MemLeak.Stop",	boost::bind(&LLFloaterMemLeak::onClickStop, this));
	mCommitCallbackRegistrar.add("MemLeak.Release",	boost::bind(&LLFloaterMemLeak::onClickRelease, this));
	mCommitCallbackRegistrar.add("MemLeak.Close",	boost::bind(&LLFloaterMemLeak::onClickClose, this));
}
//----------------------------------------------

bool LLFloaterMemLeak::postBuild(void)
{	
	F32 a, b ;
	a = getChild<LLUICtrl>("leak_speed")->getValue().asReal();
	if(a > (F32)(0xFFFFFFFF))
	{
		sMemLeakingSpeed = 0xFFFFFFFF ;
	}
	else
	{
		sMemLeakingSpeed = (U32)a ;
	}
	b = getChild<LLUICtrl>("max_leak")->getValue().asReal();
	if(b > (F32)0xFFF)
	{
		sMaxLeakedMem = 0xFFFFFFFF ;
	}
	else
	{
		sMaxLeakedMem = ((U32)b) << 20 ;
	}
	
	sbAllocationFailed = FALSE ;
	return true ;
}
LLFloaterMemLeak::~LLFloaterMemLeak()
{
	release() ;
		
	sMemLeakingSpeed = 0 ; //bytes leaked per frame
	sMaxLeakedMem = 0 ; //maximum allowed leaked memory	
}

void LLFloaterMemLeak::release()
{
	if(mLeakedMem.empty())
	{
		return ;
	}

	for(S32 i = 0 ; i < (S32)mLeakedMem.size() ; i++)
	{
		delete[] mLeakedMem[i] ;
	}
	mLeakedMem.clear() ;

	sStatus = STOP ;
	sTotalLeaked = 0 ;
	sbAllocationFailed = FALSE ;
	gSimulateMemLeak = FALSE;
}

void LLFloaterMemLeak::stop()
{
	sStatus = STOP ;
	sbAllocationFailed = TRUE ;
}

void LLFloaterMemLeak::idle()
{
	if(STOP == sStatus)
	{
		return ;
	}

	sbAllocationFailed = FALSE ;
	
	if(RELEASE == sStatus)
	{
		release() ;
		return ;
	}

	char* p = NULL ;
	if(sMemLeakingSpeed > 0 && sTotalLeaked < sMaxLeakedMem)
	{
		p = new char[sMemLeakingSpeed] ;

		if(p)
		{
			mLeakedMem.push_back(p) ;
			sTotalLeaked += sMemLeakingSpeed ;
		}
	}
	if(!p)
	{
		stop();
	}
}

//----------------------
void LLFloaterMemLeak::onChangeLeakingSpeed()
{
	F32 tmp ;
	tmp =getChild<LLUICtrl>("leak_speed")->getValue().asReal();

	if(tmp > (F32)0xFFFFFFFF)
	{
		sMemLeakingSpeed = 0xFFFFFFFF ;
	}
	else
	{
		sMemLeakingSpeed = (U32)tmp ;
	}

}

void LLFloaterMemLeak::onChangeMaxMemLeaking()
{

	F32 tmp ;
	tmp =getChild<LLUICtrl>("max_leak")->getValue().asReal();
	if(tmp > (F32)0xFFF)
	{
		sMaxLeakedMem = 0xFFFFFFFF ;
	}
	else
	{
		sMaxLeakedMem = ((U32)tmp) << 20 ;
	}
	
}

void LLFloaterMemLeak::onClickStart()
{
	sStatus = START ;
	gSimulateMemLeak = TRUE;
}

void LLFloaterMemLeak::onClickStop()
{
	sStatus = STOP ;
}

void LLFloaterMemLeak::onClickRelease()
{
	sStatus = RELEASE ;
}

void LLFloaterMemLeak::onClickClose()
{
	setVisible(FALSE);
}

void LLFloaterMemLeak::draw()
{
	//show total memory leaked
	if(sTotalLeaked > 0)
	{
		std::string bytes_string;
		LLResMgr::getInstance()->getIntegerString(bytes_string, sTotalLeaked >> 10 );
		getChild<LLUICtrl>("total_leaked_label")->setTextArg("[SIZE]", bytes_string);
	}
	else
	{
		getChild<LLUICtrl>("total_leaked_label")->setTextArg("[SIZE]", LLStringExplicit("0"));
	}

	if(sbAllocationFailed)
	{
		getChild<LLUICtrl>("note_label_1")->setTextArg("[NOTE1]", LLStringExplicit("Memory leaking simulation stops. Reduce leaking speed or"));
		getChild<LLUICtrl>("note_label_2")->setTextArg("[NOTE2]", LLStringExplicit("increase max leaked memory, then press Start to continue."));
	}
	else
	{
		getChild<LLUICtrl>("note_label_1")->setTextArg("[NOTE1]", LLStringExplicit(""));
		getChild<LLUICtrl>("note_label_2")->setTextArg("[NOTE2]", LLStringExplicit(""));
	}

	LLFloater::draw();
}