#include "CircularBuffer.h"
#include <assert.h>


bool CircularBuffer::Peek(OUT char* destbuf, size_t bytes) const
{
	assert(mBuffer != nullptr);

	if (mARegionSize + mBRegionSize < bytes)
		return false;

	size_t cnt = bytes;
	size_t aRead = 0;

	/// A, B 영역 둘다 데이터가 있는 경우는 A먼저 읽는다
	if (mARegionSize > 0)
	{
		aRead = (cnt > mARegionSize) ? mARegionSize : cnt;
		memcpy(destbuf, mARegionPointer, aRead);
		cnt -= aRead;
	}

	/// 읽기 요구한 데이터가 더 있다면 B 영역에서 읽는다
	if (cnt > 0 && mBRegionSize > 0)
	{
		assert(cnt <= mBRegionSize);

		/// 남은거 마저 다 읽기
		size_t bRead = cnt;

		memcpy(destbuf + aRead, mBRegionPointer, bRead);
		cnt -= bRead;
	}

	assert(cnt == 0);

	return true;

}

bool CircularBuffer::Read(OUT char* destbuf, size_t bytes)
{
	assert(mBuffer != nullptr);

	if (mARegionSize + mBRegionSize < bytes)
		return false;

	size_t cnt = bytes;
	size_t aRead = 0;


	/// A, B 영역 둘다 데이터가 있는 경우는 A먼저 읽는다
	if (mARegionSize > 0)
	{
		aRead = (cnt > mARegionSize) ? mARegionSize : cnt;
		memcpy(destbuf, mARegionPointer, aRead);
		mARegionSize -= aRead;
		mARegionPointer += aRead;
		cnt -= aRead;
	}

	/// 읽기 요구한 데이터가 더 있다면 B 영역에서 읽는다
	if (cnt > 0 && mBRegionSize > 0)
	{
		assert(cnt <= mBRegionSize);

		/// 남은거 마저 다 읽기
		size_t bRead = cnt;

		memcpy(destbuf + aRead, mBRegionPointer, bRead);
		mBRegionSize -= bRead;
		mBRegionPointer += bRead;
		cnt -= bRead;
	}

	assert(cnt == 0);

	/// A 버퍼가 비었다면 B버퍼를 맨 앞으로 당기고 A 버퍼로 지정 
	if (mARegionSize == 0)
	{
		if (mBRegionSize > 0)
		{
			if (mBRegionPointer != mBuffer)
				memmove(mBuffer, mBRegionPointer, mBRegionSize);

			mARegionPointer = mBuffer;
			mARegionSize = mBRegionSize;
			mBRegionPointer = nullptr;
			mBRegionSize = 0;
		}
		else
		{
			/// B에 아무것도 없는 경우 그냥 A로 스위치
			mBRegionPointer = nullptr;
			mBRegionSize = 0;
			mARegionPointer = mBuffer;
			mARegionSize = 0;
		}
	}

	return true;
}


bool CircularBuffer::Write(const char* data, size_t bytes)
{
	assert(mBuffer != nullptr);

	/// Read와 반대로 B가 있다면 B영역에 먼저 쓴다
	if (mBRegionPointer != nullptr)
	{
		if (GetBFreeSpace() < bytes)
			return false;

		memcpy(mBRegionPointer + mBRegionSize, data, bytes);
		mBRegionSize += bytes;

		return true;
	}

	/// A영역보다 다른 영역의 용량이 더 클 경우 그 영역을 B로 설정하고 기록
	if (GetAFreeSpace() < GetSpaceBeforeA())
	{
		AllocateB();

		if (GetBFreeSpace() < bytes)
			return false;

		memcpy(mBRegionPointer + mBRegionSize, data, bytes);
		mBRegionSize += bytes;

		return true;
	}
	/// A영역이 더 크면 당연히 A에 쓰기
	else
	{
		if (GetAFreeSpace() < bytes)
			return false;

		memcpy(mARegionPointer + mARegionSize, data, bytes);
		mARegionSize += bytes;

		return true;
	}
}


void CircularBuffer::Remove(size_t len)
{
	size_t cnt = len ;
	
	/// Read와 마찬가지로 A가 있다면 A영역에서 먼저 삭제

	if ( mARegionSize > 0 )
	{
		size_t aRemove = (cnt > mARegionSize) ? mARegionSize : cnt ;
		mARegionSize -= aRemove ;
		mARegionPointer += aRemove ;
		cnt -= aRemove ;
	}

	// 제거할 용량이 더 남은경우 B에서 제거 
	if ( cnt > 0 && mBRegionSize > 0 )
	{
		size_t bRemove = (cnt > mBRegionSize) ? mBRegionSize : cnt ;
		mBRegionSize -= bRemove ;
		mBRegionPointer += bRemove ;
		cnt -= bRemove ;
	}

	/// A영역이 비워지면 B를 A로 스위치 
	if ( mARegionSize == 0 )
	{
		if ( mBRegionSize > 0 )
		{
			/// 앞으로 당겨 붙이기
			if ( mBRegionPointer != mBuffer )
				memmove(mBuffer, mBRegionPointer, mBRegionSize) ;
	
			mARegionPointer = mBuffer ;
			mARegionSize = mBRegionSize ;
			mBRegionPointer = nullptr ;
			mBRegionSize = 0 ;
		}
		else
		{
			mBRegionPointer = nullptr ;
			mBRegionSize = 0 ;
			mARegionPointer = mBuffer ;
			mARegionSize = 0 ;
		}
	}
}