/** 
 * @file lltrace.h
 * @brief Runtime statistics accumulation.
 *
 * $LicenseInfo:firstyear=2001&license=viewerlgpl$
 * Second Life Viewer Source Code
 * Copyright (C) 2012, 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$
 */

#ifndef LL_LLTRACE_H
#define LL_LLTRACE_H

#include "stdtypes.h"
#include "llpreprocessor.h"

#include "llmemory.h"
#include "llrefcount.h"
#include "lltraceaccumulators.h"
#include "llthreadlocalstorage.h"
#include "lltimer.h"

#include <list>

namespace LLTrace
{
class Recording;

template<typename T>
T storage_value(T val) { return val; }

template<typename UNIT_TYPE, typename STORAGE_TYPE> 
STORAGE_TYPE storage_value(LLUnit<STORAGE_TYPE, UNIT_TYPE> val) { return val.value(); }

template<typename UNIT_TYPE, typename STORAGE_TYPE> 
STORAGE_TYPE storage_value(LLUnitImplicit<STORAGE_TYPE, UNIT_TYPE> val) { return val.value(); }

class TraceBase
{
public:
	TraceBase(const char* name, const char* description);
	virtual ~TraceBase() {};
	virtual const char* getUnitLabel() const;

	const std::string& getName() const { return mName; }
	const std::string& getDescription() const { return mDescription; }

protected:
	std::string	mName;
	std::string	mDescription;
};

template<typename ACCUMULATOR>
class TraceType 
:	public TraceBase,
	public LLInstanceTracker<TraceType<ACCUMULATOR>, std::string>
{
public:
	TraceType(const char* name, const char* description = NULL)
	:	LLInstanceTracker<TraceType<ACCUMULATOR>, std::string>(name),
		TraceBase(name, description),
		mAccumulatorIndex(AccumulatorBuffer<ACCUMULATOR>::getDefaultBuffer()->reserveSlot())
	{}

	LL_FORCE_INLINE ACCUMULATOR& getCurrentAccumulator() const
	{
		ACCUMULATOR* accumulator_storage = LLThreadLocalSingletonPointer<ACCUMULATOR>::getInstance();
		return accumulator_storage ? accumulator_storage[mAccumulatorIndex] : (*AccumulatorBuffer<ACCUMULATOR>::getDefaultBuffer())[mAccumulatorIndex];
	}

	size_t getIndex() const { return mAccumulatorIndex; }
	static size_t getNumIndices() { return AccumulatorBuffer<ACCUMULATOR>::getNumIndices(); }

protected:
	const size_t		mAccumulatorIndex;
};


template<>
class TraceType<TimeBlockAccumulator::CallCountFacet>
:	public TraceType<TimeBlockAccumulator>
{
public:

	TraceType(const char* name, const char* description = "")
	:	TraceType<TimeBlockAccumulator>(name, description)
	{}
};

template<>
class TraceType<TimeBlockAccumulator::SelfTimeFacet>
	:	public TraceType<TimeBlockAccumulator>
{
public:

	TraceType(const char* name, const char* description = "")
		:	TraceType<TimeBlockAccumulator>(name, description)
	{}
};

template <typename T = F64>
class EventStatHandle
:	public TraceType<EventAccumulator>
{
public:
	typedef F64 storage_t;
	typedef TraceType<EventAccumulator> trace_t;
	typedef EventStatHandle<T> self_t;

	EventStatHandle(const char* name, const char* description = NULL)
	:	trace_t(name, description)
	{}

	/*virtual*/ const char* getUnitLabel() const { return LLGetUnitLabel<T>::getUnitLabel(); }

};

template<typename T, typename VALUE_T>
void record(EventStatHandle<T>& measurement, VALUE_T value)
{
	T converted_value(value);
	measurement.getCurrentAccumulator().record(storage_value(converted_value));
}

template <typename T = F64>
class SampleStatHandle
:	public TraceType<SampleAccumulator>
{
public:
	typedef F64 storage_t;
	typedef TraceType<SampleAccumulator> trace_t;
	typedef SampleStatHandle<T> self_t;

	SampleStatHandle(const char* name, const char* description = NULL)
	:	trace_t(name, description)
	{}

	/*virtual*/ const char* getUnitLabel() const { return LLGetUnitLabel<T>::getUnitLabel(); }
};

template<typename T, typename VALUE_T>
void sample(SampleStatHandle<T>& measurement, VALUE_T value)
{
	T converted_value(value);
	measurement.getCurrentAccumulator().sample(storage_value(converted_value));
}

template <typename T = F64>
class CountStatHandle
:	public TraceType<CountAccumulator>
{
public:
	typedef F64 storage_t;
	typedef TraceType<CountAccumulator> trace_t;
	typedef CountStatHandle<T> self_t;

	CountStatHandle(const char* name, const char* description = NULL) 
	:	trace_t(name, description)
	{}

	/*virtual*/ const char* getUnitLabel() const { return LLGetUnitLabel<T>::getUnitLabel(); }
};

template<typename T, typename VALUE_T>
void add(CountStatHandle<T>& count, VALUE_T value)
{
	T converted_value(value);
	count.getCurrentAccumulator().add(storage_value(converted_value));
}

template<>
class TraceType<MemStatAccumulator::AllocationFacet>
:	public TraceType<MemStatAccumulator>
{
public:

	TraceType(const char* name, const char* description = "")
	:	TraceType<MemStatAccumulator>(name, description)
	{}
};

template<>
class TraceType<MemStatAccumulator::DeallocationFacet>
:	public TraceType<MemStatAccumulator>
{
public:

	TraceType(const char* name, const char* description = "")
	:	TraceType<MemStatAccumulator>(name, description)
	{}
};

template<>
class TraceType<MemStatAccumulator::ShadowAllocationFacet>
	:	public TraceType<MemStatAccumulator>
{
public:

	TraceType(const char* name, const char* description = "")
		:	TraceType<MemStatAccumulator>(name, description)
	{}
};

template<>
class TraceType<MemStatAccumulator::ShadowDeallocationFacet>
	:	public TraceType<MemStatAccumulator>
{
public:

	TraceType(const char* name, const char* description = "")
		:	TraceType<MemStatAccumulator>(name, description)
	{}
};

template<>
class TraceType<MemStatAccumulator::ShadowMemFacet>
	:	public TraceType<MemStatAccumulator>
{
public:

	TraceType(const char* name, const char* description = "")
		:	TraceType<MemStatAccumulator>(name, description)
	{}
};

class MemStatHandle : public TraceType<MemStatAccumulator>
{
public:
	typedef TraceType<MemStatAccumulator> trace_t;
	MemStatHandle(const char* name)
	:	trace_t(name)
	{}

	void setName(const char* name)
	{
		mName = name;
		setKey(name);
	}

	/*virtual*/ const char* getUnitLabel() const { return "KB"; }

	TraceType<MemStatAccumulator::AllocationFacet>& allocations() 
	{ 
		return static_cast<TraceType<MemStatAccumulator::AllocationFacet>&>(*(TraceType<MemStatAccumulator>*)this);
	}

	TraceType<MemStatAccumulator::DeallocationFacet>& deallocations() 
	{ 
		return static_cast<TraceType<MemStatAccumulator::DeallocationFacet>&>(*(TraceType<MemStatAccumulator>*)this);
	}

	TraceType<MemStatAccumulator::ShadowAllocationFacet>& shadowAllocations() 
	{ 
		return static_cast<TraceType<MemStatAccumulator::ShadowAllocationFacet>&>(*(TraceType<MemStatAccumulator>*)this);
	}

	TraceType<MemStatAccumulator::ShadowDeallocationFacet>& shadowDeallocations() 
	{ 
		return static_cast<TraceType<MemStatAccumulator::ShadowDeallocationFacet>&>(*(TraceType<MemStatAccumulator>*)this);
	}

	TraceType<MemStatAccumulator::ShadowMemFacet>& shadowMem() 
	{ 
		return static_cast<TraceType<MemStatAccumulator::ShadowMemFacet>&>(*(TraceType<MemStatAccumulator>*)this);
	}
};

inline void claim_footprint(MemStatHandle& measurement, S32 size)
{
	MemStatAccumulator& accumulator = measurement.getCurrentAccumulator();
	accumulator.mSize.sample(accumulator.mSize.hasValue() ? accumulator.mSize.getLastValue() + (F64)size : (F64)size);
	accumulator.mAllocated.add(1);
}

inline void disclaim_footprint(MemStatHandle& measurement, S32 size)
{
	MemStatAccumulator& accumulator = measurement.getCurrentAccumulator();
	accumulator.mSize.sample(accumulator.mSize.hasValue() ? accumulator.mSize.getLastValue() - (F64)size : -(F64)size);
	accumulator.mDeallocated.add(1);
}

inline void claim_shadow(MemStatHandle& measurement, S32 size)
{
	MemStatAccumulator& accumulator = measurement.getCurrentAccumulator();
	accumulator.mShadowSize.sample(accumulator.mSize.hasValue() ? accumulator.mSize.getLastValue() + (F64)size : (F64)size);
	accumulator.mShadowAllocated.add(1);
}

inline void disclaim_shadow(MemStatHandle& measurement, S32 size)
{
	MemStatAccumulator& accumulator = measurement.getCurrentAccumulator();
	accumulator.mShadowSize.sample(accumulator.mSize.hasValue() ? accumulator.mSize.getLastValue() - (F64)size : -(F64)size);
	accumulator.mShadowDeallocated.add(1);
}

// measures effective memory footprint of specified type
// specialize to cover different types

template<typename T, typename IS_MEM_TRACKABLE = void>
struct MeasureMem
{
	static size_t measureFootprint(const T& value)
	{
		return sizeof(T);
	}

	static size_t measureFootprint()
	{
		return sizeof(T);
	}

	static size_t measureShadow(const T& value)
	{
		return 0;
	}

	static size_t measureShadow()
	{
		return 0;
	}
};

template<typename T>
struct MeasureMem<T, typename T::mem_trackable_tag_t>
{
	static size_t measureFootprint(const T& value)
	{
		return sizeof(T) + value.getMemFootprint();
	}

	static size_t measureFootprint()
	{
		return sizeof(T);
	}

	static size_t measureShadow(const T& value)
	{
		return value.getMemShadow();
	}

	static size_t measureShadow()
	{
		return MeasureMem<T>::measureShadow();
	}
};


template<typename T, typename IS_MEM_TRACKABLE>
struct MeasureMem<T*, IS_MEM_TRACKABLE>
{
	static size_t measureFootprint(const T* value)
	{
		if (!value)
		{
			return 0;
		}
		return MeasureMem<T>::measureFootprint(*value);
	}

	static size_t measureFootprint()
	{
		return MeasureMem<T>::measureFootprint();
	}

	static size_t measureShadow(const T* value)
	{
		return MeasureMem<T>::measureShadow(*value);
	}

	static size_t measureShadow()
	{
		return MeasureMem<T>::measureShadow();
	}
};

template<typename T, typename IS_MEM_TRACKABLE>
struct MeasureMem<std::basic_string<T>, IS_MEM_TRACKABLE>
{
	static size_t measureFootprint(const std::basic_string<T>& value)
	{
		return value.capacity() * sizeof(T);
	}

	static size_t measureFootprint()
	{
		return sizeof(std::basic_string<T>);
	}

	static size_t measureShadow(const std::basic_string<T>& value)
	{
		return 0;
	}

	static size_t measureShadow()
	{
		return 0;
	}
};

template<typename DERIVED, size_t ALIGNMENT = LL_DEFAULT_HEAP_ALIGN>
class MemTrackable
{
public:
	typedef void mem_trackable_tag_t;

	enum EMemType
	{
		MEM_FOOTPRINT, 
		MEM_SHADOW
	};

	MemTrackable()
	:	mMemFootprint(0),
		mMemShadow(0)
	{
		static bool name_initialized = false;
		if (!name_initialized)
		{
			name_initialized = true;
			sMemStat.setName(typeid(DERIVED).name());
		}
	}

	virtual ~MemTrackable()
	{
		disclaimMem(mMemFootprint, MEM_FOOTPRINT);
		disclaimMem(mMemShadow, MEM_SHADOW);
	}

	static MemStatHandle& getMemStatHandle()
	{
		return sMemStat;
	}

	S32 getMemFootprint() const	{ return mMemFootprint; }
	S32 getMemShadow() const	{ return mMemShadow; }

	void* operator new(size_t size) 
	{
		claim_footprint(sMemStat, size);
		return ll_aligned_malloc(ALIGNMENT, size);
	}

	void operator delete(void* ptr, size_t size)
	{
		disclaim_footprint(sMemStat, size);
		ll_aligned_free(ALIGNMENT, ptr);
	}

	void *operator new [](size_t size)
	{
		claim_footprint(sMemStat, size);
		return ll_aligned_malloc(ALIGNMENT, size);
	}

	void operator delete[](void* ptr, size_t size)
	{
		disclaim_footprint(sMemStat, size);
		ll_aligned_free(ALIGNMENT, ptr);
	}

	// claim memory associated with other objects/data as our own, adding to our calculated footprint
	template<typename CLAIM_T>
	CLAIM_T& claimMem(CLAIM_T& value, EMemType mem_type = MEM_FOOTPRINT)
	{
		trackAlloc(MeasureMem<CLAIM_T>::measureFootprint(value), mem_type);
		trackAlloc(MeasureMem<CLAIM_T>::measureShadow(value), MEM_SHADOW);
		return value;
	}

	template<typename CLAIM_T>
	const CLAIM_T& claimMem(const CLAIM_T& value, EMemType mem_type = MEM_FOOTPRINT)
	{
		trackAlloc(MeasureMem<CLAIM_T>::measureFootprint(value), mem_type);
		trackAlloc(MeasureMem<CLAIM_T>::measureShadow(value), MEM_SHADOW);
		return value;
	}

	size_t& claimMem(size_t& size, EMemType mem_type = MEM_FOOTPRINT)
	{
		trackAlloc(size, mem_type);
		return size;
	}

	S32& claimMem(S32& size, EMemType mem_type = MEM_FOOTPRINT)
	{
		trackAlloc(size, mem_type);
		return size;
	}

	// remove memory we had claimed from our calculated footprint
	template<typename CLAIM_T>
	CLAIM_T& disclaimMem(CLAIM_T& value, EMemType mem_type = MEM_FOOTPRINT)
	{
		trackDealloc(MeasureMem<CLAIM_T>::measureFootprint(value), mem_type);
		trackDealloc(MeasureMem<CLAIM_T>::measureShadow(value), MEM_SHADOW);
		return value;
	}

	template<typename CLAIM_T>
	const CLAIM_T& disclaimMem(const CLAIM_T& value, EMemType mem_type = MEM_FOOTPRINT)
	{
		trackDealloc(MeasureMem<CLAIM_T>::measureFootprint(value), mem_type);
		trackDealloc(MeasureMem<CLAIM_T>::measureShadow(value), MEM_SHADOW);
		return value;
	}

	size_t& disclaimMem(size_t& size, EMemType mem_type = MEM_FOOTPRINT)
	{
		trackDealloc(size, mem_type);
		return size;
	}

	S32& disclaimMem(S32& size, EMemType mem_type = MEM_FOOTPRINT)
	{
		trackDealloc(size, mem_type);
		return size;
	}

private:

	void trackAlloc(S32 size, EMemType mem_type)
	{
		if (mem_type == MEM_FOOTPRINT)
		{
			claim_footprint(sMemStat, size);
			mMemFootprint += size;
		}
		else
		{
			claim_shadow(sMemStat, size);
			mMemShadow += size;
		}
	}

	void trackDealloc(S32 size, EMemType mem_type)
	{
		if (mem_type == MEM_FOOTPRINT)
		{
			disclaim_footprint(sMemStat, size);
			mMemFootprint -= size;
		}
		else
		{
			disclaim_shadow(sMemStat, size);
			mMemShadow -= size;
		}
	}

private:
	// use signed values so that we can temporarily go negative
	// and reconcile in destructor
	// NB: this assumes that no single class is responsible for > 2GB of allocations
	S32 mMemFootprint,
		mMemShadow;
	
	static	MemStatHandle	sMemStat;
};

template<typename DERIVED, size_t ALIGNMENT>
MemStatHandle MemTrackable<DERIVED, ALIGNMENT>::sMemStat("");

}
#endif // LL_LLTRACE_H